From b65f2eb55f14352c44826f1d0c67d7944ea46a4b Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Tue, 18 Mar 2025 16:41:35 +0800 Subject: [PATCH] fix embedding model name translate issue (#16111) --- api/controllers/console/datasets/datasets.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/api/controllers/console/datasets/datasets.py b/api/controllers/console/datasets/datasets.py index ba1a30cb03..6532e046c9 100644 --- a/api/controllers/console/datasets/datasets.py +++ b/api/controllers/console/datasets/datasets.py @@ -184,6 +184,10 @@ class DatasetApi(Resource): except services.errors.account.NoPermissionError as e: raise Forbidden(str(e)) data = marshal(dataset, dataset_detail_fields) + if dataset.indexing_technique == "high_quality": + if dataset.embedding_model_provider: + provider_id = ModelProviderID(dataset.embedding_model_provider) + data["embedding_model_provider"] = str(provider_id) if data.get("permission") == "partial_members": part_users_list = DatasetPermissionService.get_dataset_partial_member_list(dataset_id_str) data.update({"partial_member_list": part_users_list})