From dd621b59b91653c3e177ed691e98b9f0c1be8d51 Mon Sep 17 00:00:00 2001 From: zhangzhanwei Date: Thu, 3 Jul 2025 20:21:35 +0800 Subject: [PATCH] Application model use --- apps/models_provider/tools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/apps/models_provider/tools.py b/apps/models_provider/tools.py index 7227ad977..8626b58f4 100644 --- a/apps/models_provider/tools.py +++ b/apps/models_provider/tools.py @@ -107,8 +107,8 @@ def is_valid_credential(provider, model_type, model_name, model_credential: Dict def get_model_by_id(_id, workspace_id): model = QuerySet(Model).filter(id=_id).first() get_authorized_model = DatabaseModelManage.get_model("get_authorized_model") - if get_authorized_model is not None: - model = get_authorized_model(model, workspace_id) + if model and model.workspace_id!=workspace_id and get_authorized_model is not None: + model = get_authorized_model(QuerySet(Model).filter(id=_id), workspace_id).first() if model is None: raise Exception(_("Model does not exist")) return model