BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix：修正知识获取逻辑，调整知识相关性输出顺序，并添加知识长度限制

pull/895/head
墨梓柒 2025-04-30 18:22:06 +08:00
parent 5963214d95
commit c8c27a1483
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/plugins/knowledge/src/qa_manager.py
View File

@ -11,6 +11,8 @@ from .lpmmconfig import global_config
from .utils.dyn_topk import dyn_select_top_k from .utils.dyn_topk import dyn_select_top_k
MAX_KNOWLEDGE_LENGTH = 10000 # 最大知识长度
class QAManager: class QAManager:
def __init__( def __init__(
self, self,
@ -112,8 +114,10 @@ class QAManager:
for res in query_res for res in query_res
] ]
found_knowledge = "\n".join( found_knowledge = "\n".join(
[f"第{i + 1}条知识：{k[1]}\n 该条知识对于问题的相关性：{k[0]}" for i, k in enumerate(knowledge)] [f"第{i + 1}条知识：{k[0]}\n 该条知识对于问题的相关性：{k[1]}" for i, k in enumerate(knowledge)]
) )
if len(found_knowledge) > MAX_KNOWLEDGE_LENGTH:
found_knowledge = found_knowledge[:MAX_KNOWLEDGE_LENGTH] + "\n"
return found_knowledge return found_knowledge
else: else:
logger.info("LPMM知识库并未初始化，使用旧版数据库进行检索") logger.info("LPMM知识库并未初始化，使用旧版数据库进行检索")