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