gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(knowledge): 简化知识库问答管理器的使用

Browse Source 
将 `QAManager` 类实例化改为直接导入和使用单例 `qa_manager`，以简化代码并提高效率。同时，移除了对 `get_knowledge` 方法中不再需要的参数，使其调用更加清晰。
This commit is contained in:
minecraft1024a
2025-10-18 21:29:44 +08:00
committed by Windpicker-owo
parent 8549778f9c
commit 823d432152
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/chat/utils/prompt.py
View File

@@ -764,20 +764,15 @@ class Prompt:
return {"knowledge_prompt": ""} return {"knowledge_prompt": ""}
try: try:
from src.chat.knowledge.knowledge_lib import QAManager from src.chat.knowledge.knowledge_lib import qa_manager
# 获取问题文本（当前消息） # 获取问题文本（当前消息）
question = self.parameters.target or "" question = self.parameters.target or ""
if not question: if not question or not qa_manager:
return {"knowledge_prompt": ""} return {"knowledge_prompt": ""}
# 创建QA管理器
qa_manager = QAManager()
# 搜索相关知识 # 搜索相关知识
knowledge_results = await qa_manager.get_knowledge( knowledge_results = await qa_manager.get_knowledge(question=question)
question=question, chat_id=self.parameters.chat_id, max_results=5, min_similarity=0.5
)
# 构建知识块 # 构建知识块
if knowledge_results and knowledge_results.get("knowledge_items"): if knowledge_results and knowledge_results.get("knowledge_items"):