|
|
|
|
@ -14,6 +14,7 @@ from controllers.console.wraps import account_initialization_required, enterpris
|
|
|
|
|
from core.errors.error import LLMBadRequestError, ProviderTokenNotInitError
|
|
|
|
|
from core.indexing_runner import IndexingRunner
|
|
|
|
|
from core.model_runtime.entities.model_entities import ModelType
|
|
|
|
|
from core.plugin.entities.plugin import ModelProviderID
|
|
|
|
|
from core.provider_manager import ProviderManager
|
|
|
|
|
from core.rag.datasource.vdb.vector_type import VectorType
|
|
|
|
|
from core.rag.extractor.entity.extract_setting import ExtractSetting
|
|
|
|
|
@ -72,6 +73,8 @@ class DatasetListApi(Resource):
|
|
|
|
|
|
|
|
|
|
data = marshal(datasets, dataset_detail_fields)
|
|
|
|
|
for item in data:
|
|
|
|
|
# convert embedding_model_provider to plugin standard format
|
|
|
|
|
item["embedding_model_provider"] = str(ModelProviderID(item["embedding_model_provider"]))
|
|
|
|
|
if item["indexing_technique"] == "high_quality":
|
|
|
|
|
item_model = f"{item['embedding_model']}:{item['embedding_model_provider']}"
|
|
|
|
|
if item_model in model_names:
|
|
|
|
|
|