fix: missed rerank_mode when convert to DatasetEntity (#7269)

1 year ago · 101db126c8
parent ba79088ffc
commit 101db126c8
1 changed files with 1 additions and 0 deletions
--- a/api/core/app/app_config/easy_ui_based_app/dataset/manager.py
+++ b/api/core/app/app_config/easy_ui_based_app/dataset/manager.py
@ -93,6 +93,7 @@ class DatasetConfigManager:
                    reranking_model=dataset_configs.get('reranking_model'),
                    weights=dataset_configs.get('weights'),
                    reranking_enabled=dataset_configs.get('reranking_enabled', True),
+                    rerank_mode=dataset_configs["reranking_mode"],
                )
            )