From 823d432152eeff4eefd541ac6e52c787c42640af Mon Sep 17 00:00:00 2001 From: minecraft1024a Date: Sat, 18 Oct 2025 21:29:44 +0800 Subject: [PATCH] =?UTF-8?q?refactor(knowledge):=20=E7=AE=80=E5=8C=96?= =?UTF-8?q?=E7=9F=A5=E8=AF=86=E5=BA=93=E9=97=AE=E7=AD=94=E7=AE=A1=E7=90=86?= =?UTF-8?q?=E5=99=A8=E7=9A=84=E4=BD=BF=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 将 `QAManager` 类实例化改为直接导入和使用单例 `qa_manager`,以简化代码并提高效率。同时,移除了对 `get_knowledge` 方法中不再需要的参数,使其调用更加清晰。 --- src/chat/utils/prompt.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/src/chat/utils/prompt.py b/src/chat/utils/prompt.py index 7f58b77a1..aca806475 100644 --- a/src/chat/utils/prompt.py +++ b/src/chat/utils/prompt.py @@ -764,20 +764,15 @@ class Prompt: return {"knowledge_prompt": ""} try: - from src.chat.knowledge.knowledge_lib import QAManager + from src.chat.knowledge.knowledge_lib import qa_manager # 获取问题文本(当前消息) question = self.parameters.target or "" - if not question: + if not question or not qa_manager: return {"knowledge_prompt": ""} - # 创建QA管理器 - qa_manager = QAManager() - # 搜索相关知识 - knowledge_results = await qa_manager.get_knowledge( - question=question, chat_id=self.parameters.chat_id, max_results=5, min_similarity=0.5 - ) + knowledge_results = await qa_manager.get_knowledge(question=question) # 构建知识块 if knowledge_results and knowledge_results.get("knowledge_items"):