Browse Source

fix: can not find model bug (#12051)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
yihong 4 tháng trước cách đây
mục cha
commit
7da4fb68da

+ 1 - 1
api/core/entities/provider_configuration.py

@@ -872,7 +872,7 @@ class ProviderConfiguration(BaseModel):
             # if llm name not in restricted llm list, remove it
             restrict_model_names = [rm.model for rm in restrict_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
                 elif not quota_configuration.is_valid:
                     model.status = ModelStatus.QUOTA_EXCEEDED

+ 2 - 2
api/services/entities/model_provider_entities.py

@@ -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())