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