Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 11:42:29 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix economy index search in workflow (#3205)

Browse Source
This commit is contained in:
Jyong 2024-04-09 13:20:51 +08:00 committed by GitHub
parent f514fd2182
commit 84d7cbf916
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -242,7 +242,10 @@ class KnowledgeRetrievalNode(BaseNode):
# get top k # get top k
top_k = retrieval_model_config['top_k'] top_k = retrieval_model_config['top_k']
# get retrieval method # 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 # 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 if retrieval_model_config['reranking_enable'] else None