diff --git a/api/controllers/console/datasets/datasets.py b/api/controllers/console/datasets/datasets.py index d1bc960bd8..36e2d6cd2d 100644 --- a/api/controllers/console/datasets/datasets.py +++ b/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