From c8c27a14836e01c67d424e16465b71e7070d2ce6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com> Date: Wed, 30 Apr 2025 18:22:06 +0800 Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E4=BF=AE=E6=AD=A3=E7=9F=A5?= =?UTF-8?q?=E8=AF=86=E8=8E=B7=E5=8F=96=E9=80=BB=E8=BE=91=EF=BC=8C=E8=B0=83?= =?UTF-8?q?=E6=95=B4=E7=9F=A5=E8=AF=86=E7=9B=B8=E5=85=B3=E6=80=A7=E8=BE=93?= =?UTF-8?q?=E5=87=BA=E9=A1=BA=E5=BA=8F=EF=BC=8C=E5=B9=B6=E6=B7=BB=E5=8A=A0?= =?UTF-8?q?=E7=9F=A5=E8=AF=86=E9=95=BF=E5=BA=A6=E9=99=90=E5=88=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/knowledge/src/qa_manager.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/plugins/knowledge/src/qa_manager.py b/src/plugins/knowledge/src/qa_manager.py index 9cb5c0186..6d84dc8e7 100644 --- a/src/plugins/knowledge/src/qa_manager.py +++ b/src/plugins/knowledge/src/qa_manager.py @@ -11,6 +11,8 @@ from .lpmmconfig import global_config from .utils.dyn_topk import dyn_select_top_k +MAX_KNOWLEDGE_LENGTH = 10000 # 最大知识长度 + class QAManager: def __init__( self, @@ -112,8 +114,10 @@ class QAManager: for res in query_res ] 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 else: logger.info("LPMM知识库并未初始化,使用旧版数据库进行检索")