feat: 优化知识库检索接口返回内容

This commit is contained in:
martsforever
2025-11-24 19:30:18 +08:00
parent 25d9dce5f1
commit ed449686fe
+1 -1
View File
@@ -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'