From 370120fd4154ce6c5f69d16a4a343a016cf2e816 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 04 三月 2025 09:53:17 +0800
Subject: [PATCH] 完善问题
---
app/api/v2/chat.py | 9 ++-------
1 files changed, 2 insertions(+), 7 deletions(-)
diff --git a/app/api/v2/chat.py b/app/api/v2/chat.py
index 6f78896..b65541e 100644
--- a/app/api/v2/chat.py
+++ b/app/api/v2/chat.py
@@ -136,13 +136,8 @@
-@chat_router_v2.post("/multi/retrieval")
-async def retrieve_chunks(request_data: RetrievalRequest, api_key: str = Depends(get_api_key)):
- records = await service_chunk_retrieval(request_data.query, request_data.retrieval_setting.top_k, request_data.retrieval_setting.score_threshold, api_key)
- return {"records": records}
-
-
@chat_router_v2.post("/retrieval")
async def retrieve_chunks(request_data: RetrievalRequest, api_key: str = Depends(get_api_key)):
- records = await service_base_chunk_retrieval(request_data.query,request_data.knowledge_id, request_data.retrieval_setting.top_k, request_data.retrieval_setting.score_threshold, api_key)
+ records = await service_chunk_retrieval(request_data.query, request_data.knowledge_id, request_data.retrieval_setting.top_k, request_data.retrieval_setting.score_threshold, api_key)
return {"records": records}
+
--
Gitblit v1.8.0