From 370120fd4154ce6c5f69d16a4a343a016cf2e816 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期二, 04 三月 2025 09:53:17 +0800 Subject: [PATCH] 完善问题 --- app/models/v2/chat.py | 16 ++++++++++++++++ 1 files changed, 16 insertions(+), 0 deletions(-) diff --git a/app/models/v2/chat.py b/app/models/v2/chat.py new file mode 100644 index 0000000..a4818e9 --- /dev/null +++ b/app/models/v2/chat.py @@ -0,0 +1,16 @@ +from pydantic import BaseModel + + + +class RetrievalSetting(BaseModel): + top_k: int + score_threshold: float + + +class RetrievalRequest(BaseModel): + knowledge_id: str + query: str + retrieval_setting: RetrievalSetting + + + -- Gitblit v1.8.0