fix: can not find model bug (#12051)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong
2024-12-25 08:42:52 +08:00
committed by GitHub
parent 7a24c957bd
commit 7da4fb68da
2 changed files with 3 additions and 3 deletions

View File

@@ -7,7 +7,6 @@ from configs import dify_config
from core.entities.model_entities import (
ModelWithProviderEntity,
ProviderModelWithStatusEntity,
SimpleModelProviderEntity,
)
from core.entities.provider_entities import QuotaConfiguration
from core.model_runtime.entities.common_entities import I18nObject
@@ -152,7 +151,8 @@ class ModelWithProviderEntityResponse(ModelWithProviderEntity):
Model with provider entity.
"""
provider: SimpleModelProviderEntity
# FIXME type error ignore here
provider: SimpleProviderEntityResponse # type: ignore
def __init__(self, model: ModelWithProviderEntity) -> None:
super().__init__(**model.model_dump())