fix: set score_threshold only when score_threshold_enabled is true. (#14221)

This commit is contained in:
Fei He 2025-03-12 20:55:57 +08:00 committed by GitHub
parent 74f66d3119
commit 36ae0e5476
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -89,7 +89,9 @@ class DatasetConfigManager:
dataset_configs["retrieval_model"]
),
top_k=dataset_configs.get("top_k", 4),
score_threshold=dataset_configs.get("score_threshold"),
score_threshold=dataset_configs.get("score_threshold")
if dataset_configs.get("score_threshold_enabled", False)
else None,
reranking_model=dataset_configs.get("reranking_model"),
weights=dataset_configs.get("weights"),
reranking_enabled=dataset_configs.get("reranking_enabled", True),