fix embedding model name translate issue (#16111)

This commit is contained in:
Jyong 2025-03-18 16:41:35 +08:00 committed by GitHub
parent 7d620ffd5e
commit b65f2eb55f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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})