ソースを参照

fix embedding model name translate issue (#16111)

Jyong 1 ヶ月 前
コミット
b65f2eb55f
1 ファイル変更4 行追加0 行削除
  1. 4 0
      api/controllers/console/datasets/datasets.py

+ 4 - 0
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})