From 167fd3105358e7fbb7a3559c29f01b105470fb64 Mon Sep 17 00:00:00 2001
From: arkohut <39525455+arkohut@users.noreply.github.com>
Date: Tue, 10 Sep 2024 15:47:36 +0800
Subject: [PATCH] feat: add builtin plugins for default library

---
 memos/commands.py | 3 +++
 memos/config.py   | 4 +++-
 memos/models.py   | 4 ++--
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/memos/commands.py b/memos/commands.py
index 744bd9d..7167234 100644
--- a/memos/commands.py
+++ b/memos/commands.py
@@ -797,6 +797,9 @@ def scan_default_library(force: bool = False):
             return
         default_library = response.json()
 
+    for plugin in settings.default_plugins:
+        bind(default_library["id"], plugin)
+
     # Check if the library is empty
     if not default_library["folders"]:
         # Add the screenshots directory to the library
diff --git a/memos/config.py b/memos/config.py
index aa57145..b979852 100644
--- a/memos/config.py
+++ b/memos/config.py
@@ -1,6 +1,6 @@
 import os
 from pathlib import Path
-from typing import Tuple, Type
+from typing import Tuple, Type, List
 from pydantic_settings import (
     BaseSettings,
     PydanticBaseSettingsSource,
@@ -78,6 +78,8 @@ class Settings(BaseSettings):
     auth_username: str = "admin"
     auth_password: SecretStr = SecretStr("changeme")
 
+    default_plugins: List[str] = ["builtin_vlm", "builtin_ocr"]
+
     @classmethod
     def settings_customise_sources(
         cls,
diff --git a/memos/models.py b/memos/models.py
index 740e8e6..72bc91c 100644
--- a/memos/models.py
+++ b/memos/models.py
@@ -178,10 +178,10 @@ def init_database():
 def initialize_default_plugins(session):
     default_plugins = [
         PluginModel(
-            name="buildin_vlm", description="VLM Plugin", webhook_url="/plugins/vlm"
+            name="builtin_vlm", description="VLM Plugin", webhook_url="/plugins/vlm"
         ),
         PluginModel(
-            name="buildin_ocr", description="OCR Plugin", webhook_url="/plugins/ocr"
+            name="builtin_ocr", description="OCR Plugin", webhook_url="/plugins/ocr"
         ),
     ]