From 84d7cbf916d00a3c51880fb6dd94e0ad37d82928 Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Tue, 9 Apr 2024 13:20:51 +0800 Subject: [PATCH] fix economy index search in workflow (#3205) --- .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 5 ++++- 1 file changed, 4 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 e9aa655401..a1dce8b09d 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -242,7 +242,10 @@ class KnowledgeRetrievalNode(BaseNode): # get top k top_k = retrieval_model_config['top_k'] # get retrieval method - retrival_method = retrieval_model_config['search_method'] + if dataset.indexing_technique == "economy": + retrival_method = 'keyword_search' + else: + retrival_method = retrieval_model_config['search_method'] # get reranking model reranking_model=retrieval_model_config['reranking_model'] \ if retrieval_model_config['reranking_enable'] else None