test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-02 03:07:59 +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

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

@ -242,6 +242,9 @@ class KnowledgeRetrievalNode(BaseNode):
# get top k
top_k = retrieval_model_config['top_k']
# get retrieval 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'] \