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'