feat: add scope parameter to ToolTreeSerializer and update get_tools method

Author: liuruibin

apps/tools/serializers/tool.py

@@ -538,6 +538,7 @@ def add(self, instance, with_valid=True):
 
 class ToolTreeSerializer(serializers.Serializer):
     workspace_id = serializers.CharField(required=True, label=_('workspace id'))
+    scope = serializers.CharField(required=True, label=_('scope'))
 
     def get_tools(self, folder_id):
         self.is_valid(raise_exception=True)
@@ -553,8 +554,11 @@ def get_tools(self, folder_id):
         folders_data = ToolFolderFlatSerializer(child_folders, many=True).data
 
         # 获取当前文件夹下的工具
-        tools = QuerySet(Tool).filter(Q(workspace_id=self.data.get('workspace_id')) &
-                                      Q(folder_id=folder_id))
+        tools = QuerySet(Tool).filter(
+            Q(workspace_id=self.data.get('workspace_id')) &
+            Q(scope=self.data.get('scope')) &
+            Q(folder_id=folder_id)
+        )
         tools_data = ToolModelSerializer(tools, many=True).data
 
         # 返回包含文件夹和工具的结构

apps/tools/views/tool.py

@@ -68,7 +68,7 @@ def post(self, request: Request, workspace_id: str):
     )
     def get(self, request: Request, workspace_id: str):
         return result.success(ToolTreeSerializer(
-            data={'workspace_id': workspace_id}
+            data={'workspace_id': workspace_id, 'scope': ToolScope.WORKSPACE, 'user_id': request.user.id}
         ).get_tools(request.query_params.get('folder_id')))
 
     class Debug(APIView):