Add support for step functions defined as object properties

.changeset/twenty-gifts-win.md

@@ -0,0 +1,5 @@
+---
+"@workflow/swc-plugin": patch
+---
+
+Add support for step functions defined as object properties

packages/swc-plugin-workflow/transform/tests/fixture/object-property-step/input.js

@@ -0,0 +1,40 @@
+import * as z from 'zod';
+import { tool } from 'ai';
+
+export const weatherTool = tool({
+  description: 'Get the weather in a location',
+  inputSchema: z.object({
+    location: z.string().describe('The location to get the weather for'),
+  }),
+  execute: async ({ location }) => {
+    "use step";
+    return {
+      location,
+      temperature: 72 + Math.floor(Math.random() * 21) - 10,
+    };
+  },
+});
+
+export const timeTool = tool({
+  description: 'Get the current time',
+  execute: async function timeToolImpl () {
+    "use step";
+    return {
+      time: new Date().toISOString(),
+    };
+  },
+});
+
+export const weatherTool2 = tool({
+  description: 'Get the weather in a location',
+  inputSchema: z.object({
+    location: z.string().describe('The location to get the weather for'),
+  }),
+  async execute({ location }) {
+    "use step";
+    return {
+      location,
+      temperature: 72 + Math.floor(Math.random() * 21) - 10,
+    };
+  },
+});

packages/swc-plugin-workflow/transform/tests/fixture/object-property-step/output-client.js

@@ -0,0 +1,34 @@
+import * as z from 'zod';
+import { tool } from 'ai';
+export const weatherTool = tool({
+    description: 'Get the weather in a location',
+    inputSchema: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: async ({ location })=>{
+        return {
+            location,
+            temperature: 72 + Math.floor(Math.random() * 21) - 10
+        };
+    }
+});
+export const timeTool = tool({
+    description: 'Get the current time',
+    execute: async function timeToolImpl() {
+        return {
+            time: new Date().toISOString()
+        };
+    }
+});
+export const weatherTool2 = tool({
+    description: 'Get the weather in a location',
+    inputSchema: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    async execute ({ location }) {
+        return {
+            location,
+            temperature: 72 + Math.floor(Math.random() * 21) - 10
+        };
+    }
+});

packages/swc-plugin-workflow/transform/tests/fixture/object-property-step/output-step.js

@@ -0,0 +1,42 @@
+import { registerStepFunction } from "workflow/internal/private";
+import * as z from 'zod';
+import { tool } from 'ai';
+/**__internal_workflows{"steps":{"input.js":{"timeTool/execute":{"stepId":"step//input.js//timeTool/execute"},"weatherTool/execute":{"stepId":"step//input.js//weatherTool/execute"},"weatherTool2/execute":{"stepId":"step//input.js//weatherTool2/execute"}}}}*/;
+var weatherTool2$execute = async ({ location })=>{
+    return {
+        location,
+        temperature: 72 + Math.floor(Math.random() * 21) - 10
+    };
+};
+var timeTool$execute = async ()=>{
+    return {
+        time: new Date().toISOString()
+    };
+};
+var weatherTool$execute = async ({ location })=>{
+    return {
+        location,
+        temperature: 72 + Math.floor(Math.random() * 21) - 10
+    };
+};
+export const weatherTool = tool({
+    description: 'Get the weather in a location',
+    inputSchema: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: weatherTool$execute
+});
+export const timeTool = tool({
+    description: 'Get the current time',
+    execute: timeTool$execute
+});
+export const weatherTool2 = tool({
+    description: 'Get the weather in a location',
+    inputSchema: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: weatherTool2$execute
+});
+registerStepFunction("step//input.js//weatherTool/execute", weatherTool$execute);
+registerStepFunction("step//input.js//timeTool/execute", timeTool$execute);
+registerStepFunction("step//input.js//weatherTool2/execute", weatherTool2$execute);

packages/swc-plugin-workflow/transform/tests/fixture/object-property-step/output-workflow.js

@@ -0,0 +1,21 @@
+import * as z from 'zod';
+import { tool } from 'ai';
+/**__internal_workflows{"steps":{"input.js":{"timeTool/execute":{"stepId":"step//input.js//timeTool/execute"},"weatherTool/execute":{"stepId":"step//input.js//weatherTool/execute"},"weatherTool2/execute":{"stepId":"step//input.js//weatherTool2/execute"}}}}*/;
+export const weatherTool = tool({
+    description: 'Get the weather in a location',
+    inputSchema: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: globalThis[Symbol.for("WORKFLOW_USE_STEP")]("step//input.js//weatherTool/execute")
+});
+export const timeTool = tool({
+    description: 'Get the current time',
+    execute: globalThis[Symbol.for("WORKFLOW_USE_STEP")]("step//input.js//timeTool/execute")
+});
+export const weatherTool2 = tool({
+    description: 'Get the weather in a location',
+    inputSchema: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: globalThis[Symbol.for("WORKFLOW_USE_STEP")]("step//input.js//weatherTool2/execute")
+});