Преглед на файлове

fixed score threshold is none (#8342)

Jyong преди 7 месеца
родител
ревизия
49cee773c5

+ 1 - 1
api/core/rag/retrieval/dataset_retrieval.py

@@ -429,7 +429,7 @@ class DatasetRetrieval:
                         top_k=top_k,
                         top_k=top_k,
                         score_threshold=retrieval_model.get("score_threshold", 0.0)
                         score_threshold=retrieval_model.get("score_threshold", 0.0)
                         if retrieval_model["score_threshold_enabled"]
                         if retrieval_model["score_threshold_enabled"]
-                        else None,
+                        else 0.0,
                         reranking_model=retrieval_model.get("reranking_model", None)
                         reranking_model=retrieval_model.get("reranking_model", None)
                         if retrieval_model["reranking_enable"]
                         if retrieval_model["reranking_enable"]
                         else None,
                         else None,

+ 1 - 1
api/core/tools/tool/dataset_retriever/dataset_multi_retriever_tool.py

@@ -179,7 +179,7 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
                         top_k=self.top_k,
                         top_k=self.top_k,
                         score_threshold=retrieval_model.get("score_threshold", 0.0)
                         score_threshold=retrieval_model.get("score_threshold", 0.0)
                         if retrieval_model["score_threshold_enabled"]
                         if retrieval_model["score_threshold_enabled"]
-                        else None,
+                        else 0.0,
                         reranking_model=retrieval_model.get("reranking_model", None)
                         reranking_model=retrieval_model.get("reranking_model", None)
                         if retrieval_model["reranking_enable"]
                         if retrieval_model["reranking_enable"]
                         else None,
                         else None,

+ 1 - 1
api/core/tools/tool/dataset_retriever/dataset_retriever_tool.py

@@ -72,7 +72,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
                     top_k=self.top_k,
                     top_k=self.top_k,
                     score_threshold=retrieval_model.get("score_threshold", 0.0)
                     score_threshold=retrieval_model.get("score_threshold", 0.0)
                     if retrieval_model["score_threshold_enabled"]
                     if retrieval_model["score_threshold_enabled"]
-                    else None,
+                    else 0.0,
                     reranking_model=retrieval_model.get("reranking_model", None)
                     reranking_model=retrieval_model.get("reranking_model", None)
                     if retrieval_model["reranking_enable"]
                     if retrieval_model["reranking_enable"]
                     else None,
                     else None,

+ 1 - 1
api/services/hit_testing_service.py

@@ -42,7 +42,7 @@ class HitTestingService:
             top_k=retrieval_model.get("top_k", 2),
             top_k=retrieval_model.get("top_k", 2),
             score_threshold=retrieval_model.get("score_threshold", 0.0)
             score_threshold=retrieval_model.get("score_threshold", 0.0)
             if retrieval_model["score_threshold_enabled"]
             if retrieval_model["score_threshold_enabled"]
-            else None,
+            else 0.0,
             reranking_model=retrieval_model.get("reranking_model", None)
             reranking_model=retrieval_model.get("reranking_model", None)
             if retrieval_model["reranking_enable"]
             if retrieval_model["reranking_enable"]
             else None,
             else None,