fix score threshold limit be None (#6900)

This commit is contained in:
Jyong
2024-08-02 12:10:51 +08:00
committed by GitHub
parent 56af1a0adf
commit 44801df8f8
4 changed files with 10 additions and 5 deletions

View File

@@ -158,10 +158,11 @@ class DatasetRetrieveConfigEntity(BaseModel):
retrieve_strategy: RetrieveStrategy
top_k: Optional[int] = None
score_threshold: Optional[float] = None
score_threshold: Optional[float] = .0
rerank_mode: Optional[str] = 'reranking_model'
reranking_model: Optional[dict] = None
weights: Optional[dict] = None
reranking_enabled: Optional[bool] = True