fix: multiple retrieve reranking_enabled switch (#19958)

11 months ago · 8cb3b4aef2
parent 09547b4c8d
commit 8cb3b4aef2
1 changed files with 1 additions and 1 deletions
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@ -190,7 +190,7 @@ class DatasetRetrieval:
                retrieve_config.rerank_mode or "reranking_model",
                retrieve_config.reranking_model,
                retrieve_config.weights,
-                retrieve_config.reranking_enabled or True,
+                True if retrieve_config.reranking_enabled is None else retrieve_config.reranking_enabled,
                message_id,
                metadata_filter_document_ids,
                metadata_condition,