zhaoqingang
2025-03-04 370120fd4154ce6c5f69d16a4a343a016cf2e816
1
2
3
4
5
6
7
8
9
10
11
12
13
from pydantic import BaseModel
 
 
 
class RetrievalSetting(BaseModel):
    top_k: int
    score_threshold: float
 
 
class RetrievalRequest(BaseModel):
    knowledge_id: str
    query: str
    retrieval_setting: RetrievalSetting