Procházet zdrojové kódy

Merge branch 'feature/xueyiming/2025-09-29-update-res' of Server/rag_server into master

xueyiming před 1 týdnem
rodič
revize
1c9411297d

+ 1 - 1
applications/utils/chat/rag_chat_agent.py

@@ -127,7 +127,7 @@ class RAGChatAgent:
         
          请返回以下格式的 JSON 结果:
         {{
-            "result": "<最终的答案>",
+            "result": "<最终的答案,输出markdown格式>",
             "relevance_score": <0到1之间的小数,表示总结与问题的相关度>,
             "status": <1代表回答的好,0代表回答的不好>
         }}

+ 4 - 3
routes/buleprint.py

@@ -348,7 +348,7 @@ async def query_search(
                     "datasetId": content_chunk["dataset_id"],
                 }
             )
-    return res
+    return res[:limit]
 
 
 @server_bp.route("/query", methods=["GET"])
@@ -483,6 +483,7 @@ async def delete_task():
     await build_graph_task.deal(doc_id)
     return jsonify({"status_code": 200, "detail": "success", "data": {}})
 
+
 @server_bp.route("/rag/search", methods=["POST"])
 async def rag_search():
     body = await request.get_json()
@@ -495,8 +496,8 @@ async def rag_search():
         query_text=query_text,
         filters={"dataset_id": dataset_ids},
         search_type=search_type,
+        limit=5,
     )
-
     resource = get_resource_manager()
     chat_result_mapper = ChatResult(resource.mysql_client)
     rag_chat_agent = RAGChatAgent()
@@ -519,4 +520,4 @@ async def rag_search():
         llm_search["source"],
         llm_search["status"],
     )
-    return jsonify({"status_code": 200, "detail": "success", "data": data})
+    return jsonify({"status_code": 200, "detail": "success", "data": data})