Skip to content

D
dify
Unverified Commit 790b885d authored by Jyong's avatar Jyong Committed by GitHub
Browse files
Options

fix multi-dataset retrieve score limit (#2050) 

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 1a2eacc5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
api/core/features/dataset_retrieval.py
View file @ 790b885d
......@@ -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(),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment