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