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

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