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

embedding model available check (#1009)

Co-authored-by: jyong <jyong@dify.ai>
Jyong 1 год назад
Родитель
Сommit
e7199826cc
1 измененных файлов с 5 добавлено и 2 удалено
  1. 5 2
      api/controllers/console/datasets/datasets.py

+ 5 - 2
api/controllers/console/datasets/datasets.py

@@ -87,10 +87,13 @@ class DatasetListApi(Resource):
         #     raise ProviderNotInitializeError(
         #         f"No Embedding Model available. Please configure a valid provider "
         #         f"in the Settings -> Model Provider.")
-        model_names = [item['model_name'] for item in valid_model_list]
+        model_names = []
+        for valid_model in valid_model_list:
+            model_names.append(f"{valid_model['model_name']}:{valid_model['model_provider']['provider_name']}")
         data = marshal(datasets, dataset_detail_fields)
         for item in data:
-            if item['embedding_model'] in model_names:
+            item_model = f"{item['embedding_model']}:{item['embedding_model_provider']}"
+            if item_model in model_names:
                 item['embedding_available'] = True
             else:
                 item['embedding_available'] = False