fixed score threshold is none (#8342)

2024-09-13 10:21:58 +08:00
parent c78828ab7c
commit 49cee773c5
4 changed files with 4 additions and 4 deletions
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@@ -429,7 +429,7 @@ class DatasetRetrieval:
                        top_k=top_k,
                        score_threshold=retrieval_model.get("score_threshold", 0.0)
                        if retrieval_model["score_threshold_enabled"]
-                        else None,
+                        else 0.0,
                        reranking_model=retrieval_model.get("reranking_model", None)
                        if retrieval_model["reranking_enable"]
                        else None,