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