estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

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
parent e4259a8f13
commit 23fa3dedc4
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