opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixed single retrival

This commit is contained in:
jyong 2024-03-29 19:29:27 +08:00
parent 704cb42869
commit ca6acf2650
1 changed files with 2 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

@ -238,7 +238,8 @@ class KnowledgeRetrievalNode(BaseNode):
# get retrieval method
retrival_method = retrieval_model_config['search_method']
# 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
# get score threshold
score_threshold = .0
score_threshold_enabled = retrieval_model_config.get("score_threshold_enabled")