feat: add workflow template handling in tool.py and ToolStoreDialog.vue

Author: liuruibin

apps/tools/serializers/tool.py

@@ -33,7 +33,7 @@
 from common.exception.app_exception import AppApiException
 from common.field.common import UploadedImageField
 from common.result import result
-from common.utils.common import get_file_content, generate_uuid
+from common.utils.common import get_file_content, generate_uuid, bytes_to_uploaded_file
 from common.utils.logger import maxkb_logger
 from common.utils.rsa_util import rsa_long_decrypt, rsa_long_encrypt
 from common.utils.tool_code import ToolExecutor
@@ -400,6 +400,25 @@ def insert(self, instance, with_valid=True):
                 if instance.get('tool_type') == ToolType.MCP:
                     ToolExecutor().validate_mcp_transport(instance.get('code', ''))
 
+            # 处理 work_flow_template
+            if instance.get('work_flow_template') is not None:
+                template_instance = instance.get('work_flow_template')
+                download_url = template_instance.get('downloadUrl')
+                # 查找匹配的版本名称
+                res = requests.get(download_url, timeout=5)
+                tool = ToolSerializer.Import(data={
+                    'file': bytes_to_uploaded_file(res.content, 'file.tool'),
+                    'user_id': self.data.get('user_id'),
+                    'workspace_id': self.data.get('workspace_id'),
+                    'folder_id': str(instance.get('folder_id', self.data.get('workspace_id'))),
+                }).import_()
+
+                try:
+                    requests.get(template_instance.get('downloadCallbackUrl'), timeout=5)
+                except Exception as e:
+                    maxkb_logger.error(f"callback appstore tool download error: {e}")
+                return tool
+
             tool_id = uuid.uuid7()
             Tool(
                 id=tool_id,
@@ -860,7 +879,9 @@ def import_(self, scope=ToolScope.WORKSPACE):
                 'auth_target_type': AuthTargetType.TOOL.value
             }).auth_resource(str(tool_id))
 
-            return True
+            return ToolSerializer.Operate(data={
+                'id': tool_id, 'workspace_id': self.data.get('workspace_id')
+            }).one()
 
     class IconOperate(serializers.Serializer):
         id = serializers.UUIDField(required=True, label=_("function ID"))

ui/src/views/tool/tool-store/ToolStoreDialog.vue

@@ -243,6 +243,8 @@ const addLoading = ref(false)
 async function handleAdd(tool: any) {
   if (tool.tool_type === 'INTERNAL') {
     await handleInternalAdd(tool)
+  } else if (tool.label === 'workflow_template') {
+    await handleTemplateAdd(tool)
   } else {
     await handleStoreAdd(tool)
   }
@@ -287,6 +289,31 @@ async function handleStoreAdd(tool: any) {
   }
 }
 
+async function handleTemplateAdd(tool: any) {
+  try {
+    const obj = {
+      name: tool.name,
+      folder_id: folderId.value,
+      code: '{}',
+      work_flow_template: tool
+    }
+    await loadSharedApi({ type: 'tool', systemType: props.apiType })
+      .postTool(obj)
+      .then((res: any) => {
+        MsgSuccess(t('common.addSuccess'))
+        emit('refresh')
+        return user.profile().then(() => {
+          dialogVisible.value = false
+        })
+      })
+      .finally(() => {
+        loading.value = false
+      })
+  } catch (error) {
+    console.error(error)
+  }
+}
+
 
 defineExpose({ open })
 </script>