@mastra/ai-sdk 1.4.1-alpha.0 → 1.4.1-alpha.1

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @mastra/ai-sdk
 
+## 1.4.1-alpha.1
+
+### Patch Changes
+
+- Fixed workflow streaming in @mastra/ai-sdk so intermediate `data-workflow` parts stop repeating every completed step output. Added `data-workflow-step` parts with the full payload for the step that just changed, which reduces stream size for long-running workflows while preserving final workflow outputs. ([#15218](https://github.com/mastra-ai/mastra/pull/15218))
+
+  If your UI reads live step outputs during workflow execution, it should now consume `data-workflow-step` parts in addition to `data-workflow`. Final workflow snapshots still include the full step outputs.
+
+- Updated dependencies [[`3d83d06`](https://github.com/mastra-ai/mastra/commit/3d83d06f776f00fb5f4163dddd32a030c5c20844), [`7e0e63e`](https://github.com/mastra-ai/mastra/commit/7e0e63e2e485e84442351f4c7a79a424c83539dc), [`9467ea8`](https://github.com/mastra-ai/mastra/commit/9467ea87695749a53dfc041576410ebf9ee7bb67), [`7338d94`](https://github.com/mastra-ai/mastra/commit/7338d949380cf68b095342e8e42610dc51d557c1)]:
+  - @mastra/core@1.26.0-alpha.2
+
 ## 1.4.1-alpha.0
 
 ### Patch Changes

package/dist/index.cjs

@@ -12219,6 +12219,45 @@ function convertFullStreamChunkToUIMessageStream({
 
 // src/transformers.ts
 var PRIMITIVE_CACHE_SYMBOL = /* @__PURE__ */ Symbol("primitive-cache");
+function cloneWorkflowStep(step, includeOutput) {
+  return {
+    name: step.name,
+    status: step.status,
+    input: step.input,
+    output: includeOutput ? step.output : null,
+    suspendPayload: step.suspendPayload,
+    resumePayload: step.resumePayload
+  };
+}
+function serializeWorkflowSteps(steps, { includeOutputs }) {
+  return Object.fromEntries(Object.entries(steps).map(([id, step]) => [id, cloneWorkflowStep(step, includeOutputs)]));
+}
+function createWorkflowDataPart(args) {
+  const { current, isNested, runId, status, includeOutputs = false, output = null } = args;
+  return {
+    type: isNested ? "data-tool-workflow" : "data-workflow",
+    id: runId,
+    data: {
+      name: current.name,
+      status,
+      steps: serializeWorkflowSteps(current.steps, { includeOutputs }),
+      output
+    }
+  };
+}
+function createWorkflowStepDataPart(args) {
+  const { current, isNested, runId, status, stepId } = args;
+  return {
+    type: isNested ? "data-tool-workflow-step" : "data-workflow-step",
+    id: `${runId}:${stepId}`,
+    data: {
+      name: current.name,
+      status,
+      stepId,
+      step: cloneWorkflowStep(current.steps[stepId], true)
+    }
+  };
+}
 function createWorkflowStreamToAISDKTransformer(convertMastraChunkToAISDK, {
   includeTextStreamParts,
   sendReasoning,
@@ -12248,12 +12287,14 @@ function createWorkflowStreamToAISDKTransformer(convertMastraChunkToAISDK, {
         },
         convertMastraChunkToAISDK
       );
-      if (Array.isArray(transformed)) {
-        for (const c of transformed) {
-          if (c) controller.enqueue(c);
+      if (transformed) {
+        if (Array.isArray(transformed)) {
+          for (const item of transformed) {
+            controller.enqueue(item);
+          }
+        } else {
+          controller.enqueue(transformed);
         }
-      } else if (transformed) {
-        controller.enqueue(transformed);
       }
     }
   });
@@ -12370,12 +12411,14 @@ function createAgentStreamToAISDKTransformer(convertMastraChunkToAISDK, {
               void 0,
               convertMastraChunkToAISDK
             );
-            if (Array.isArray(workflowChunk)) {
-              for (const c of workflowChunk) {
-                if (c) controller.enqueue(c);
+            if (workflowChunk) {
+              if (Array.isArray(workflowChunk)) {
+                for (const item of workflowChunk) {
+                  controller.enqueue(item);
+                }
+              } else {
+                controller.enqueue(workflowChunk);
               }
-            } else if (workflowChunk) {
-              controller.enqueue(workflowChunk);
             }
           } else if (transformedChunk.type === "tool-network") {
             const payload = transformedChunk.payload;
@@ -12650,16 +12693,12 @@ function transformWorkflow(payload, bufferedWorkflows, isNested, includeTextStre
         name: payload.payload.workflowId,
         steps: {}
       });
-      return {
-        type: isNested ? "data-tool-workflow" : "data-workflow",
-        id: payload.runId,
-        data: {
-          name: bufferedWorkflows.get(payload.runId).name,
-          status: "running",
-          steps: bufferedWorkflows.get(payload.runId).steps,
-          output: null
-        }
-      };
+      return createWorkflowDataPart({
+        current: bufferedWorkflows.get(payload.runId),
+        isNested,
+        runId: payload.runId,
+        status: "running"
+      });
     case "workflow-step-start": {
       const current = bufferedWorkflows.get(payload.runId) || { name: "", steps: {} };
       current.steps[payload.payload.id] = {
@@ -12671,16 +12710,12 @@ function transformWorkflow(payload, bufferedWorkflows, isNested, includeTextStre
         resumePayload: null
       };
       bufferedWorkflows.set(payload.runId, current);
-      return {
-        type: isNested ? "data-tool-workflow" : "data-workflow",
-        id: payload.runId,
-        data: {
-          name: current.name,
-          status: "running",
-          steps: current.steps,
-          output: null
-        }
-      };
+      return createWorkflowDataPart({
+        current,
+        isNested,
+        runId: payload.runId,
+        status: "running"
+      });
     }
     case "workflow-step-result": {
       const current = bufferedWorkflows.get(payload.runId);
@@ -12690,16 +12725,21 @@ function transformWorkflow(payload, bufferedWorkflows, isNested, includeTextStre
         status: payload.payload.status,
         output: payload.payload.output ?? null
       };
-      return {
-        type: isNested ? "data-tool-workflow" : "data-workflow",
-        id: payload.runId,
-        data: {
-          name: current.name,
+      return [
+        createWorkflowDataPart({
+          current,
+          isNested,
+          runId: payload.runId,
+          status: "running"
+        }),
+        createWorkflowStepDataPart({
+          current,
+          isNested,
+          runId: payload.runId,
           status: "running",
-          steps: current.steps,
-          output: null
-        }
-      };
+          stepId: payload.payload.id
+        })
+      ];
     }
     case "workflow-step-suspended": {
       const current = bufferedWorkflows.get(payload.runId);
@@ -12711,30 +12751,33 @@ function transformWorkflow(payload, bufferedWorkflows, isNested, includeTextStre
         resumePayload: payload.payload.resumePayload ?? null,
         output: null
       };
-      return {
-        type: isNested ? "data-tool-workflow" : "data-workflow",
-        id: payload.runId,
-        data: {
-          name: current.name,
+      return [
+        createWorkflowDataPart({
+          current,
+          isNested,
+          runId: payload.runId,
+          status: "suspended"
+        }),
+        createWorkflowStepDataPart({
+          current,
+          isNested,
+          runId: payload.runId,
           status: "suspended",
-          steps: current.steps,
-          output: null
-        }
-      };
+          stepId: payload.payload.id
+        })
+      ];
     }
     case "workflow-finish": {
       const current = bufferedWorkflows.get(payload.runId);
       if (!current) return null;
-      return {
-        type: isNested ? "data-tool-workflow" : "data-workflow",
-        id: payload.runId,
-        data: {
-          name: current.name,
-          steps: current.steps,
-          output: payload.payload.output ?? null,
-          status: payload.payload.workflowStatus
-        }
-      };
+      return createWorkflowDataPart({
+        current,
+        isNested,
+        runId: payload.runId,
+        status: payload.payload.workflowStatus,
+        includeOutputs: true,
+        output: payload.payload.output ?? null
+      });
     }
     case "workflow-step-output": {
       const output = payload.payload.output;
@@ -13119,15 +13162,16 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         }
         step[PRIMITIVE_CACHE_SYMBOL] = step[PRIMITIVE_CACHE_SYMBOL] || /* @__PURE__ */ new Map();
         const result = transformWorkflow(payload.payload, step[PRIMITIVE_CACHE_SYMBOL]);
-        if (result && "data" in result) {
-          const data = result.data;
+        const workflowResult = Array.isArray(result) ? result.find((item) => item?.type === "data-workflow" || item?.type === "data-tool-workflow") : result;
+        if (workflowResult && "data" in workflowResult) {
+          const data = workflowResult.data;
           step.task = data;
           if (data.name && step.task) {
             step.task.id = data.name;
           }
         }
         bufferedNetworks.set(payload.runId, current);
-        return {
+        const networkChunk = {
           type: isNested ? "data-tool-network" : "data-network",
           id: payload.runId,
           data: {
@@ -13135,6 +13179,10 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
             status: "running"
           }
         };
+        if (Array.isArray(result)) {
+          return [networkChunk, ...result.filter((r) => r != null)];
+        }
+        return networkChunk;
       }
       if (isDataChunkType(payload)) {
         if (!("data" in payload)) {