From 790b885d0a11b3bc25ae34ae7d16eb58a6d806ef Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Tue, 16 Jan 2024 14:14:34 +0800 Subject: [PATCH] fix multi-dataset retrieve score limit (#2050) Co-authored-by: jyong --- api/core/features/dataset_retrieval.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/api/core/features/dataset_retrieval.py b/api/core/features/dataset_retrieval.py index 2accbafbdd..90ca6c42ed 100644 --- a/api/core/features/dataset_retrieval.py +++ b/api/core/features/dataset_retrieval.py @@ -166,8 +166,7 @@ class DatasetRetrievalFeature: dataset_ids=[dataset.id for dataset in available_datasets], tenant_id=tenant_id, top_k=retrieve_config.top_k or 2, - score_threshold=(retrieve_config.score_threshold or 0.5) - if retrieve_config.reranking_model.get('score_threshold_enabled', False) else None, + score_threshold=retrieve_config.score_threshold, hit_callbacks=[hit_callback], return_resource=return_resource, retriever_from=invoke_from.to_source(),