From 2b2eb7e6abac38021522b2c37e32efe41e2d90c7 Mon Sep 17 00:00:00 2001 From: martsforever Date: Mon, 24 Nov 2025 19:38:51 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E7=9F=A5=E8=AF=86=E5=BA=93=E6=A3=80?= =?UTF-8?q?=E7=B4=A2=E6=8E=A5=E5=8F=A3=E6=94=AF=E6=8C=81=E8=AE=BE=E7=BD=AE?= =?UTF-8?q?=E6=8F=90=E7=A4=BA=E8=AF=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/controller/add_knowledge_route.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/app/controller/add_knowledge_route.py b/app/controller/add_knowledge_route.py index 3e5fcc4..02a4da3 100644 --- a/app/controller/add_knowledge_route.py +++ b/app/controller/add_knowledge_route.py @@ -33,8 +33,12 @@ def add_knowledge_route(app): # 先把检索结果返回前端 yield f'data: {json.dumps({"type": "retrieve", "data": json.dumps(search_response.model_dump())}, ensure_ascii=False)}\n\n' + external_prompt = body.get('input', {}).get('prompt', '') + + # print("external_prompt", external_prompt) + async for chunk in chain.astream([ - SystemMessage(content=f"""你需要根据如下内容来回答用户的问题：{search_response.answer}，如果内容为'在提供的资料中未找到相关信息'，你需要根据你自己的知识来回答用户问题"""), + SystemMessage(content=f"""你需要根据如下内容来回答用户的问题：{search_response.answer}，如果内容为'在提供的资料中未找到相关信息'，你需要根据你自己的知识来回答用户问题。""" + external_prompt), HumanMessage(content=param.question) ]): yield f'data: {json.dumps({"type": "messages", "data": chunk}, ensure_ascii=False)}\n\n'