From 5f10b70e24353e85b7a90b8b3d2081015265a545 Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Mon, 9 Jun 2025 15:26:21 +0800
Subject: [PATCH] refactor: tools add default workspace_id

---
 apps/common/constants/permission_constants.py | 17 +++++++++++++++++
 apps/tools/serializers/tool.py                |  2 ++
 2 files changed, 19 insertions(+)

diff --git a/apps/common/constants/permission_constants.py b/apps/common/constants/permission_constants.py
index 4c068a643..2617472ce 100644
--- a/apps/common/constants/permission_constants.py
+++ b/apps/common/constants/permission_constants.py
@@ -842,6 +842,23 @@ class PermissionConstants(Enum):
         parent_group=[SystemGroup.SHARED_KNOWLEDGE]
     )
 
+    SHARED_MODEL_CREATE = Permission(
+        group=Group.MODEL, operate=Operate.CREATE, role_list=[RoleConstants.ADMIN, RoleConstants.USER],
+        parent_group=[SystemGroup.SHARED_MODEL]
+    )
+    SHARED_MODEL_READ = Permission(
+        group=Group.MODEL, operate=Operate.READ, role_list=[RoleConstants.ADMIN, RoleConstants.USER],
+        parent_group=[SystemGroup.SHARED_MODEL]
+    )
+    SHARED_MODEL_EDIT = Permission(
+        group=Group.MODEL, operate=Operate.EDIT, role_list=[RoleConstants.ADMIN, RoleConstants.USER],
+        parent_group=[SystemGroup.SHARED_MODEL]
+    )
+    SHARED_MODEL_DELETE = Permission(
+        group=Group.MODEL, operate=Operate.DELETE, role_list=[RoleConstants.ADMIN, RoleConstants.USER],
+        parent_group=[SystemGroup.SHARED_MODEL]
+    )
+
     def get_workspace_application_permission(self):
         return lambda r, kwargs: Permission(group=self.value.group, operate=self.value.operate,
                                             resource_path=
diff --git a/apps/tools/serializers/tool.py b/apps/tools/serializers/tool.py
index 1b6420a42..64b951801 100644
--- a/apps/tools/serializers/tool.py
+++ b/apps/tools/serializers/tool.py
@@ -208,6 +208,7 @@ class ToolSerializer(serializers.Serializer):
                         desc=instance.get('desc'),
                         code=instance.get('code'),
                         user_id=self.data.get('user_id'),
+                        workspace_id=self.data.get('workspace_id'),
                         input_field_list=instance.get('input_field_list', []),
                         init_field_list=instance.get('init_field_list', []),
                         scope=instance.get('scope', ToolScope.WORKSPACE),
@@ -387,6 +388,7 @@ class ToolSerializer(serializers.Serializer):
                 desc=tool.get('desc'),
                 code=tool.get('code'),
                 user_id=user_id,
+                workspace_id=self.data.get('workspace_id'),
                 input_field_list=tool.get('input_field_list'),
                 init_field_list=tool.get('init_field_list', []),
                 scope=scope,