|
|
|
@@ -7,6 +7,7 @@ from controllers.service_api import api |
|
|
|
from controllers.service_api.dataset.error import DatasetInUseError, DatasetNameDuplicateError |
|
|
|
from controllers.service_api.wraps import DatasetApiResource |
|
|
|
from core.model_runtime.entities.model_entities import ModelType |
|
|
|
from core.plugin.entities.plugin import ModelProviderID |
|
|
|
from core.provider_manager import ProviderManager |
|
|
|
from fields.dataset_fields import dataset_detail_fields |
|
|
|
from libs.login import current_user |
|
|
|
@@ -48,7 +49,8 @@ class DatasetListApi(DatasetApiResource): |
|
|
|
|
|
|
|
data = marshal(datasets, dataset_detail_fields) |
|
|
|
for item in data: |
|
|
|
if item["indexing_technique"] == "high_quality": |
|
|
|
if item["indexing_technique"] == "high_quality" and item["embedding_model_provider"]: |
|
|
|
item["embedding_model_provider"] = str(ModelProviderID(item["embedding_model_provider"])) |
|
|
|
item_model = f"{item['embedding_model']}:{item['embedding_model_provider']}" |
|
|
|
if item_model in model_names: |
|
|
|
item["embedding_available"] = True |