fix economy index search in workflow (#3205)

2 years ago · 84d7cbf916
parent f514fd2182
commit 84d7cbf916
1 changed files with 4 additions and 1 deletions
--- 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