AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 10:45:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: dataset tool missing in n-to-1 retrieve mode (#1617)

Browse Source
This commit is contained in:
takatost 2023-11-25 16:04:22 +08:00 committed by GitHub
parent ea526d0822
commit 93467cb363
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/orchestrator_rule_parser.py
View File

@ -213,16 +213,16 @@ class OrchestratorRuleParser:
continue continue
dataset_ids.append(dataset.id) dataset_ids.append(dataset.id)
if retrieval_model == 'single': if retrieval_model == 'single':
retrieval_model = dataset.retrieval_model if dataset.retrieval_model else default_retrieval_model retrieval_model_config = dataset.retrieval_model if dataset.retrieval_model else default_retrieval_model
top_k = retrieval_model['top_k'] top_k = retrieval_model_config['top_k']
# dynamically adjust top_k when the remaining token number is not enough to support top_k # dynamically adjust top_k when the remaining token number is not enough to support top_k
# top_k = self._dynamic_calc_retrieve_k(dataset=dataset, top_k=top_k, rest_tokens=rest_tokens) # top_k = self._dynamic_calc_retrieve_k(dataset=dataset, top_k=top_k, rest_tokens=rest_tokens)
score_threshold = None score_threshold = None
score_threshold_enable = retrieval_model.get("score_threshold_enable") score_threshold_enable = retrieval_model_config.get("score_threshold_enable")
if score_threshold_enable: if score_threshold_enable:
score_threshold = retrieval_model.get("score_threshold") score_threshold = retrieval_model_config.get("score_threshold")
tool = DatasetRetrieverTool.from_dataset( tool = DatasetRetrieverTool.from_dataset(
dataset=dataset, dataset=dataset,