瀏覽代碼

embedding model available check (#1009)

Co-authored-by: jyong <jyong@dify.ai>
tags/0.3.19
Jyong 2 年之前
父節點
當前提交
e7199826cc
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 5 行新增2 行删除
  1. 5
    2
      api/controllers/console/datasets/datasets.py

+ 5
- 2
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

Loading…
取消
儲存