diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py index a3f9bcd539..7c2d26bb82 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -238,7 +238,8 @@ class KnowledgeRetrievalNode(BaseNode): # get retrieval method retrival_method = retrieval_model_config['search_method'] # get reranking model - reranking_model = retrieval_model_config['reranking_model'] + reranking_model=retrieval_model_config['reranking_model'] \ + if retrieval_model_config['reranking_enable'] else None # get score threshold score_threshold = .0 score_threshold_enabled = retrieval_model_config.get("score_threshold_enabled")