Procházet zdrojové kódy

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

tags/0.4.3
takatost před 1 rokem
rodič
revize
6a2eb5f442
Žádný účet není propojen s e-mailovou adresou tvůrce revize
1 změnil soubory, kde provedl 13 přidání a 6 odebrání
  1. 13
    6
      api/core/entities/provider_configuration.py

+ 13
- 6
api/core/entities/provider_configuration.py Zobrazit soubor

@@ -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

Načítá se…
Zrušit
Uložit