Unverified Commit e7199826 authored by Jyong's avatar Jyong Committed by GitHub

embedding model available check (#1009)

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 70e24b75
...@@ -87,10 +87,13 @@ class DatasetListApi(Resource): ...@@ -87,10 +87,13 @@ class DatasetListApi(Resource):
# raise ProviderNotInitializeError( # raise ProviderNotInitializeError(
# f"No Embedding Model available. Please configure a valid provider " # f"No Embedding Model available. Please configure a valid provider "
# f"in the Settings -> Model 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) data = marshal(datasets, dataset_detail_fields)
for item in data: 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 item['embedding_available'] = True
else: else:
item['embedding_available'] = False item['embedding_available'] = False
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment