From ea7b8547be697fb28822bc66eb2f4709611c415a Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Wed, 25 Jun 2025 20:37:27 +0800 Subject: [PATCH] fix: user permission (#3390) --- apps/common/auth/handle/impl/user_token.py | 3 ++- apps/models_provider/tools.py | 2 -- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/apps/common/auth/handle/impl/user_token.py b/apps/common/auth/handle/impl/user_token.py index 992c68aa7..96e397ca9 100644 --- a/apps/common/auth/handle/impl/user_token.py +++ b/apps/common/auth/handle/impl/user_token.py @@ -171,7 +171,8 @@ def get_permission_list(user, workspace_user_permission_list = QuerySet(WorkspaceUserResourcePermission).filter( workspace_id__in=[workspace_user_role.workspace_id for workspace_user_role in - workspace_user_role_mapping_list]) + workspace_user_role_mapping_list], + user_id=user_id) # 资源权限 workspace_resource_permission_list = get_workspace_resource_permission_list(workspace_user_permission_list, diff --git a/apps/models_provider/tools.py b/apps/models_provider/tools.py index 3c28a7a66..10ec4a292 100644 --- a/apps/models_provider/tools.py +++ b/apps/models_provider/tools.py @@ -105,8 +105,6 @@ 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() - # 手动关闭数据库连接 - connection.close() if model is None: raise Exception(_('Model does not exist')) if model.workspace_id: