Browse Source

fix economy index search in workflow (#3205)

Jyong 1 year ago
parent
commit
84d7cbf916

+ 4 - 1
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