fix: multiple retrieve reranking_enabled switch

1 year ago · 3eac8fd61c
parent 276c02f341
commit 3eac8fd61c
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,