fix dataset reranking mode miss (#15643)

This commit is contained in:
Jyong 2025-03-12 18:44:10 +08:00 committed by GitHub
parent 087bb60b31
commit d37490adc3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -88,6 +88,7 @@ class RetrievalModel(BaseModel):
search_method: Literal["hybrid_search", "semantic_search", "full_text_search"]
reranking_enable: bool
reranking_model: Optional[RerankingModel] = None
reranking_mode: Optional[str] = None
top_k: int
score_threshold_enabled: bool
score_threshold: Optional[float] = None