fix:修复qa
This commit is contained in:
@@ -31,12 +31,12 @@ class QAManager:
|
|||||||
request_type="lpmm.qa"
|
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
|
# 生成问题的Embedding
|
||||||
part_start_time = time.perf_counter()
|
part_start_time = time.perf_counter()
|
||||||
question_embedding = get_embedding(question)
|
question_embedding = await get_embedding(question)
|
||||||
if question_embedding is None:
|
if question_embedding is None:
|
||||||
logger.error("生成问题Embedding失败")
|
logger.error("生成问题Embedding失败")
|
||||||
return None
|
return None
|
||||||
@@ -103,10 +103,10 @@ class QAManager:
|
|||||||
else:
|
else:
|
||||||
return None
|
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:
|
if processed_result is not None:
|
||||||
query_res = processed_result[0]
|
query_res = processed_result[0]
|
||||||
knowledge = [
|
knowledge = [
|
||||||
|
|||||||
@@ -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}")
|
logger.debug(f"知识库查询结果: {knowledge_info}")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user