소스 검색

fix multi-dataset retrieve score limit (#2050)

Co-authored-by: jyong <jyong@dify.ai>
Jyong 1 년 전
부모
커밋
790b885d0a
1개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 2
      api/core/features/dataset_retrieval.py

+ 1 - 2
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(),