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

fix: can not find model bug (#12051)

Browse source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong 2024-12-25 08:42:52 +08:00 • committed by GitHub
commit 7da4fb68da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/entities/provider_configuration.py
View file

@ -872,7 +872,7 @@ class ProviderConfiguration(BaseModel):
# if llm name not in restricted llm list, remove it # if llm name not in restricted llm list, remove it
restrict_model_names = [rm.model for rm in restrict_models] restrict_model_names = [rm.model for rm in restrict_models]
for model in provider_models: for model in provider_models:
if model.model_type == ModelType.LLM and m.model not in restrict_model_names: if model.model_type == ModelType.LLM and model.model not in restrict_model_names:
model.status = ModelStatus.NO_PERMISSION model.status = ModelStatus.NO_PERMISSION
elif not quota_configuration.is_valid: elif not quota_configuration.is_valid:
model.status = ModelStatus.QUOTA_EXCEEDED model.status = ModelStatus.QUOTA_EXCEEDED

4
api/services/entities/model_provider_entities.py
View file

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