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

2025-11-24 19:38:51 +08:00
parent ed449686fe
commit 2b2eb7e6ab
1 changed files with 5 additions and 1 deletions
@@ -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'