Unverified Commit 6a2eb5f4 authored by takatost's avatar takatost Committed by GitHub

fix: customize model schema fetch failed raise error (#1900)

parent 0c5892bc
import datetime import datetime
import json import json
import logging
import time import time
from json import JSONDecodeError from json import JSONDecodeError
from typing import Optional, List, Dict, Tuple, Iterator from typing import Optional, List, Dict, Tuple, Iterator
...@@ -18,6 +19,8 @@ from core.model_runtime.utils import encoders ...@@ -18,6 +19,8 @@ from core.model_runtime.utils import encoders
from extensions.ext_database import db from extensions.ext_database import db
from models.provider import ProviderType, Provider, ProviderModel, TenantPreferredModelProvider from models.provider import ProviderType, Provider, ProviderModel, TenantPreferredModelProvider
logger = logging.getLogger(__name__)
class ProviderConfiguration(BaseModel): class ProviderConfiguration(BaseModel):
""" """
...@@ -544,13 +547,17 @@ class ProviderConfiguration(BaseModel): ...@@ -544,13 +547,17 @@ class ProviderConfiguration(BaseModel):
if model_configuration.model_type not in model_types: if model_configuration.model_type not in model_types:
continue continue
custom_model_schema = ( try:
provider_instance.get_model_instance(model_configuration.model_type) custom_model_schema = (
.get_customizable_model_schema_from_credentials( provider_instance.get_model_instance(model_configuration.model_type)
model_configuration.model, .get_customizable_model_schema_from_credentials(
model_configuration.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: if not custom_model_schema:
continue continue
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment