From 9927322bf9894b2949bb385420ed7f1f2ce7b51f Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 15 Jul 2025 22:35:38 +0800 Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E4=BF=AE=E5=A4=8Dqa?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/chat/knowledge/qa_manager.py | 8 ++++---- src/tools/not_using/lpmm_get_knowledge.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/chat/knowledge/qa_manager.py b/src/chat/knowledge/qa_manager.py index 8940dbb55..c83683b79 100644 --- a/src/chat/knowledge/qa_manager.py +++ b/src/chat/knowledge/qa_manager.py @@ -31,12 +31,12 @@ class QAManager: request_type="lpmm.qa" ) - def process_query(self, question: str) -> Tuple[List[Tuple[str, float, float]], Optional[Dict[str, float]]]: + async def process_query(self, question: str) -> Tuple[List[Tuple[str, float, float]], Optional[Dict[str, float]]]: """处理查询""" # 生成问题的Embedding part_start_time = time.perf_counter() - question_embedding = get_embedding(question) + question_embedding = await get_embedding(question) if question_embedding is None: logger.error("生成问题Embedding失败") return None @@ -103,10 +103,10 @@ class QAManager: else: return None - def get_knowledge(self, question: str) -> str: + async def get_knowledge(self, question: str) -> str: """获取知识""" # 处理查询 - processed_result = self.process_query(question) + processed_result = await self.process_query(question) if processed_result is not None: query_res = processed_result[0] knowledge = [ diff --git a/src/tools/not_using/lpmm_get_knowledge.py b/src/tools/not_using/lpmm_get_knowledge.py index 80b9b617b..180c5e699 100644 --- a/src/tools/not_using/lpmm_get_knowledge.py +++ b/src/tools/not_using/lpmm_get_knowledge.py @@ -43,7 +43,7 @@ class SearchKnowledgeFromLPMMTool(BaseTool): # 调用知识库搜索 - knowledge_info = qa_manager.get_knowledge(query) + knowledge_info = await qa_manager.get_knowledge(query) logger.debug(f"知识库查询结果: {knowledge_info}")