|
@@ -213,16 +213,16 @@ class OrchestratorRuleParser:
|
|
|
continue
|
|
|
dataset_ids.append(dataset.id)
|
|
|
if retrieval_model == 'single':
|
|
|
- retrieval_model = dataset.retrieval_model if dataset.retrieval_model else default_retrieval_model
|
|
|
- top_k = retrieval_model['top_k']
|
|
|
+ retrieval_model_config = dataset.retrieval_model if dataset.retrieval_model else default_retrieval_model
|
|
|
+ top_k = retrieval_model_config['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)
|
|
|
|
|
|
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:
|
|
|
- score_threshold = retrieval_model.get("score_threshold")
|
|
|
+ score_threshold = retrieval_model_config.get("score_threshold")
|
|
|
|
|
|
tool = DatasetRetrieverTool.from_dataset(
|
|
|
dataset=dataset,
|