Pārlūkot izejas kodu

fix multi-dataset retrieve score limit (#2050)

Co-authored-by: jyong <jyong@dify.ai>
tags/0.4.8
Jyong pirms 1 gada
vecāks
revīzija
790b885d0a
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 1 papildinājumiem un 2 dzēšanām
  1. 1
    2
      api/core/features/dataset_retrieval.py

+ 1
- 2
api/core/features/dataset_retrieval.py Parādīt failu

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

Notiek ielāde…
Atcelt
Saglabāt