jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(core): Fix incorrect type hints. (#5427)

Browse source
This commit is contained in:
-LAN- 2024-06-20 15:16:21 +08:00 • committed by GitHub
commit 23fa3dedc4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 166 additions and 117 deletions
Download patch file Download diff file Expand all files Collapse all files

9
api/core/model_runtime/entities/provider_entities.py
View file

@ -1,10 +1,11 @@
from collections.abc import Sequence
from enum import Enum
from typing import Optional
from pydantic import BaseModel, ConfigDict
from core.model_runtime.entities.common_entities import I18nObject
from core.model_runtime.entities.model_entities import AIModelEntity, ModelType, ProviderModel
from core.model_runtime.entities.model_entities import ModelType, ProviderModel
class ConfigurateMethod(Enum):
@ -93,8 +94,8 @@ class SimpleProviderEntity(BaseModel):
label: I18nObject
icon_small: Optional[I18nObject] = None
icon_large: Optional[I18nObject] = None
supported_model_types: list[ModelType]
models: list[AIModelEntity] = []
supported_model_types: Sequence[ModelType]
models: list[ProviderModel] = []
class ProviderHelpEntity(BaseModel):
@ -116,7 +117,7 @@ class ProviderEntity(BaseModel):
icon_large: Optional[I18nObject] = None
background: Optional[str] = None
help: Optional[ProviderHelpEntity] = None
supported_model_types: list[ModelType]
supported_model_types: Sequence[ModelType]
configurate_methods: list[ConfigurateMethod]
models: list[ProviderModel] = []
provider_credential_schema: Optional[ProviderCredentialSchema] = None