Selaa lähdekoodia

fixed score threshold is none (#8342)

tags/0.8.2
Jyong 1 vuosi sitten
vanhempi
commit
49cee773c5
No account linked to committer's email address

+ 1
- 1
api/core/rag/retrieval/dataset_retrieval.py Näytä tiedosto

@@ -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,

+ 1
- 1
api/core/tools/tool/dataset_retriever/dataset_multi_retriever_tool.py Näytä tiedosto

@@ -179,7 +179,7 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
top_k=self.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,

+ 1
- 1
api/core/tools/tool/dataset_retriever/dataset_retriever_tool.py Näytä tiedosto

@@ -72,7 +72,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
top_k=self.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,

+ 1
- 1
api/services/hit_testing_service.py Näytä tiedosto

@@ -42,7 +42,7 @@ class HitTestingService:
top_k=retrieval_model.get("top_k", 2),
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,

Loading…
Peruuta
Tallenna