| @@ -1,5 +1,6 @@ | |||
| import datetime | |||
| import json | |||
| import logging | |||
| import time | |||
| from json import JSONDecodeError | |||
| from typing import Optional, List, Dict, Tuple, Iterator | |||
| @@ -18,6 +19,8 @@ from core.model_runtime.utils import encoders | |||
| from extensions.ext_database import db | |||
| from models.provider import ProviderType, Provider, ProviderModel, TenantPreferredModelProvider | |||
| logger = logging.getLogger(__name__) | |||
| class ProviderConfiguration(BaseModel): | |||
| """ | |||
| @@ -544,13 +547,17 @@ class ProviderConfiguration(BaseModel): | |||
| if model_configuration.model_type not in model_types: | |||
| continue | |||
| custom_model_schema = ( | |||
| provider_instance.get_model_instance(model_configuration.model_type) | |||
| .get_customizable_model_schema_from_credentials( | |||
| model_configuration.model, | |||
| model_configuration.credentials | |||
| try: | |||
| custom_model_schema = ( | |||
| provider_instance.get_model_instance(model_configuration.model_type) | |||
| .get_customizable_model_schema_from_credentials( | |||
| model_configuration.model, | |||
| model_configuration.credentials | |||
| ) | |||
| ) | |||
| ) | |||
| except Exception as ex: | |||
| logger.warning(f'get custom model schema failed, {ex}') | |||
| continue | |||
| if not custom_model_schema: | |||
| continue | |||