Sfoglia il codice sorgente

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

takatost 1 anno fa
parent
commit
6a2eb5f442
1 ha cambiato i file con 13 aggiunte e 6 eliminazioni
  1. 13 6
      api/core/entities/provider_configuration.py

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

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