mirror of https://github.com/Mai-with-u/MaiBot.git
fix:修复qa
parent
a150aa7b2b
commit
9927322bf9
|
|
@ -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}")
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue