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