diff --git a/app/controller/add_knowledge_route.py b/app/controller/add_knowledge_route.py index 759ae21..3e5fcc4 100644 --- a/app/controller/add_knowledge_route.py +++ b/app/controller/add_knowledge_route.py @@ -34,7 +34,7 @@ def add_knowledge_route(app): yield f'data: {json.dumps({"type": "retrieve", "data": json.dumps(search_response.model_dump())}, ensure_ascii=False)}\n\n' async for chunk in chain.astream([ - SystemMessage(content=f"""你需要根据如下内容来回答用户的问题:{search_response.answer}"""), + SystemMessage(content=f"""你需要根据如下内容来回答用户的问题:{search_response.answer},如果内容为'在提供的资料中未找到相关信息',你需要根据你自己的知识来回答用户问题"""), HumanMessage(content=param.question) ]): yield f'data: {json.dumps({"type": "messages", "data": chunk}, ensure_ascii=False)}\n\n'