AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 17:58:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
takatost 2024-01-04 10:53:50 +08:00 committed by GitHub
parent 0c5892bcb6
commit 6a2eb5f442
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
api/core/entities/provider_configuration.py
View File

@ -1,5 +1,6 @@
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
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):
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