✨新增控制加载内置工具包的配置项

LiteyukiStudio · Nov 23, 2024 · aebd6d7 · aebd6d7
1 parent 8e0af47
commit aebd6d7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -143,6 +143,7 @@ _✨ 使用 OpenAI 标准格式 API 的聊天机器人插件 ✨_
 |   MARSHOAI_ENABLE_NICKNAME_TIP    | 否  |                 `true`                  |                                       启用后用户未设置昵称时提示用户设置                                       |
 |      MARSHOAI_ENABLE_PRAISES      | 否  |                 `true`                  |                                          是否启用夸赞名单功能                                           |
 |    MARSHOAI_ENABLE_TOOLS    | 否  |                 `true`                  |                                是否启用小棉工具(MarshoTools)                                 |
+|    MARSHOAI_LOAD_BUILTIN_TOOLS   | 否  |            `true`               |                                  是否加载内置工具包                                            |
 |      MARSHOAI_AZURE_ENDPOINT      | 否  | `https://models.inference.ai.azure.com` |                                  OpenAI 标准格式 API 端点                                  |
 |       MARSHOAI_TEMPERATURE        | 否  |                    无                    |                                          进行推理时的温度参数                                           |
 |          MARSHOAI_TOP_P           | 否  |                    无                    |                                          进行推理时的核采样参数                                          |

diff --git a/nonebot_plugin_marshoai/azure.py b/nonebot_plugin_marshoai/azure.py
@@ -67,7 +67,8 @@
 async def _preload_tools():
     tools_dir = store.get_plugin_data_dir() / "tools"
     os.makedirs(tools_dir, exist_ok=True)
-    tools.load_tools(Path(__file__).parent / "tools")
+    if config.marshoai_load_builtin_tools:
+        tools.load_tools(Path(__file__).parent / "tools")
     tools.load_tools(store.get_plugin_data_dir() / "tools")
 
 @add_usermsg_cmd.handle()

diff --git a/nonebot_plugin_marshoai/config.py b/nonebot_plugin_marshoai/config.py
@@ -25,6 +25,7 @@ class ConfigModel(BaseModel):
     marshoai_enable_praises: bool = True
     marshoai_enable_time_prompt: bool = True
     marshoai_enable_tools: bool = True
+    marshoai_load_builtin_tools: bool = True
     marshoai_azure_endpoint: str = "https://models.inference.ai.azure.com"
     marshoai_temperature: float | None = None
     marshoai_max_tokens: int | None = None