Просмотр исходного кода

fix: set score_threshold only when score_threshold_enabled is true. (#14221)

tags/1.1.0
Fei He 7 месяцев назад
Родитель
Сommit
36ae0e5476
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 3 добавлений и 1 удалений
  1. 3
    1
      api/core/app/app_config/easy_ui_based_app/dataset/manager.py

+ 3
- 1
api/core/app/app_config/easy_ui_based_app/dataset/manager.py Просмотреть файл

@@ -89,7 +89,9 @@ class DatasetConfigManager:
dataset_configs["retrieval_model"]
),
top_k=dataset_configs.get("top_k", 4),
score_threshold=dataset_configs.get("score_threshold"),
score_threshold=dataset_configs.get("score_threshold")
if dataset_configs.get("score_threshold_enabled", False)
else None,
reranking_model=dataset_configs.get("reranking_model"),
weights=dataset_configs.get("weights"),
reranking_enabled=dataset_configs.get("reranking_enabled", True),

Загрузка…
Отмена
Сохранить