文档和tool适配

src/plugins/built_in/knowledge/get_knowledge.py

@@ -1,10 +1,12 @@
-from src.plugin_system.base.base_tool import BaseTool
+import json  # Added for parsing embedding
+import math  # Added for cosine similarity
+from typing import Any, Union, List  # Added List
+
 from src.chat.utils.utils import get_embedding
 from src.common.database.database_model import Knowledges  # Updated import
 from src.common.logger import get_logger
-from typing import Any, Union, List  # Added List
-import json  # Added for parsing embedding
-import math  # Added for cosine similarity
+from src.plugin_system import BaseTool, ToolParamType
+
 
 logger = get_logger("get_knowledge_tool")
 
@@ -15,8 +17,8 @@ class SearchKnowledgeTool(BaseTool):
     name = "search_knowledge"
     description = "使用工具从知识库中搜索相关信息"
     parameters = [
-        ("query", "string", "搜索查询关键词", True),
-        ("threshold", "float", "相似度阈值，0.0到1.0之间", False),
+        ("query", ToolParamType.STRING, "搜索查询关键词", True, None),
+        ("threshold", ToolParamType.FLOAT, "相似度阈值，0.0到1.0之间", False, None),
     ]
 
     async def execute(self, function_args: dict[str, Any]) -> dict[str, Any]:

src/plugins/built_in/knowledge/lpmm_get_knowledge.py

@@ -1,10 +1,8 @@
-from src.plugin_system.base.base_tool import BaseTool
-
-# from src.common.database import db
-from src.common.logger import get_logger
 from typing import Dict, Any
-from src.chat.knowledge.knowledge_lib import qa_manager
 
+from src.common.logger import get_logger
+from src.chat.knowledge.knowledge_lib import qa_manager
+from src.plugin_system import BaseTool, ToolParamType
 
 logger = get_logger("lpmm_get_knowledge_tool")
 
@@ -15,8 +13,8 @@ class SearchKnowledgeFromLPMMTool(BaseTool):
     name = "lpmm_search_knowledge"
     description = "从知识库中搜索相关信息，如果你需要知识，就使用这个工具"
     parameters = [
-        ("query", "string", "搜索查询关键词", True),
-        ("threshold", "float", "相似度阈值，0.0到1.0之间", False),
+        ("query", ToolParamType.STRING, "搜索查询关键词", True, None),
+        ("threshold", ToolParamType.FLOAT, "相似度阈值，0.0到1.0之间", False, None),
     ]
 
     async def execute(self, function_args: Dict[str, Any]) -> Dict[str, Any]: