fix: can not find model bug (#12051)

Signed-off-by: yihong0618 <[email protected]>
langgenius · Dec 25, 2024 · 7da4fb6 · 7da4fb6
1 parent 7a24c95
commit 7da4fb6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/core/entities/provider_configuration.py b/api/core/entities/provider_configuration.py
@@ -872,7 +872,7 @@ def _get_system_provider_models(
             # 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

diff --git a/api/services/entities/model_provider_entities.py b/api/services/entities/model_provider_entities.py
@@ -7,7 +7,6 @@
 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())