|
@@ -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})
|