From e7199826ccabf15180cc671cb1763098653b160a Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Fri, 25 Aug 2023 00:25:16 +0800 Subject: [PATCH] embedding model available check (#1009) Co-authored-by: jyong --- api/controllers/console/datasets/datasets.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) 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