From ca6acf26503d0b8c5a073dfa30f627c6b56a1f1d Mon Sep 17 00:00:00 2001 From: jyong Date: Fri, 29 Mar 2024 19:29:27 +0800 Subject: [PATCH] fixed single retrival --- .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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")