wuzhiqiang/ai-admin-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 知识库检索接口支持设置提示词

Browse Source
This commit is contained in:
martsforever
2025-11-24 19:38:51 +08:00
parent ed449686fe
commit 2b2eb7e6ab
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/controller/add_knowledge_route.py
+5 -1
View File
@@ -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' 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([ async for chunk in chain.astream([
SystemMessage(content=f"""你需要根据如下内容来回答用户的问题：{search_response.answer}，如果内容为'在提供的资料中未找到相关信息'，你需要根据你自己的知识来回答用户问题"""), SystemMessage(content=f"""你需要根据如下内容来回答用户的问题：{search_response.answer}，如果内容为'在提供的资料中未找到相关信息'，你需要根据你自己的知识来回答用户问题。""" + external_prompt),
HumanMessage(content=param.question) HumanMessage(content=param.question)
]): ]):
yield f'data: {json.dumps({"type": "messages", "data": chunk}, ensure_ascii=False)}\n\n' yield f'data: {json.dumps({"type": "messages", "data": chunk}, ensure_ascii=False)}\n\n'