fixed score threshold is none

langgenius · Sep 13, 2024 · 3e9782a · 3e9782a
1 parent 43ac320
commit 3e9782a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/rag/datasource/retrieval_service.py b/api/core/rag/datasource/retrieval_service.py
@@ -28,7 +28,7 @@ def retrieve(
         dataset_id: str,
         query: str,
         top_k: int,
-        score_threshold: Optional[float] = .0,
+        score_threshold: Optional[float] = 0.0,
         reranking_model: Optional[dict] = None,
         reranking_mode: Optional[str] = "reranking_model",
         weights: Optional[dict] = None,