fix score threshold limit be None (#6900)

2024-08-02 12:10:51 +08:00
parent 56af1a0adf
commit 44801df8f8
4 changed files with 10 additions and 5 deletions
--- a/api/core/app/app_config/entities.py
+++ b/api/core/app/app_config/entities.py
@@ -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