Explorar el Código

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

tags/0.3.32
takatost hace 1 año
padre
commit
93467cb363
No account linked to committer's email address
Se han modificado 1 ficheros con 4 adiciones y 4 borrados
  1. 4
    4
      api/core/orchestrator_rule_parser.py

+ 4
- 4
api/core/orchestrator_rule_parser.py Ver fichero

@@ -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,

Cargando…
Cancelar
Guardar