zhaoqingang
2025-03-03 f95f801f35aa201cbaffd7d881c07edc9398b570
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