正确使用lpmm构建prompt

This commit is contained in:
UnCLAS-Prommer
2025-08-03 19:52:31 +08:00
parent 9a63a8030e
commit 1e5db5d7e1
12 changed files with 141 additions and 249 deletions

View File

@@ -1,6 +1,7 @@
from typing import Dict, Any
from src.common.logger import get_logger
from src.config.config import global_config
from src.chat.knowledge.knowledge_lib import qa_manager
from src.plugin_system import BaseTool, ToolParamType
@@ -16,6 +17,7 @@ class SearchKnowledgeFromLPMMTool(BaseTool):
("query", ToolParamType.STRING, "搜索查询关键词", True, None),
("threshold", ToolParamType.FLOAT, "相似度阈值0.0到1.0之间", False, None),
]
available_for_llm = global_config.lpmm_knowledge.enable
async def execute(self, function_args: Dict[str, Any]) -> Dict[str, Any]:
"""执行知识库搜索