@mastra/ai-sdk 0.3.2 → 0.3.3-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @mastra/ai-sdk
 
+## 0.3.3-alpha.0
+
+### Patch Changes
+
+- Return NetworkDataPart on each agent-execution-event and workflow-execution-event in network streams ([#10979](https://github.com/mastra-ai/mastra/pull/10979))
+
+- Fixed tool-call-suspended chunks being dropped in workflow-step-output when using AI SDK. Previously, when an agent inside a workflow step called a tool that got suspended, the tool-call-suspended chunk was not received on the frontend even though tool-input-available chunks were correctly received. ([#10988](https://github.com/mastra-ai/mastra/pull/10988))
+
+  The issue occurred because tool-call-suspended was not included in the isMastraTextStreamChunk list, causing it to be filtered out in transformWorkflow. Now tool-call-suspended, tool-call-approval, object, and tripwire chunks are properly included in the text stream chunk list and will be transformed and passed through correctly.
+
+  Fixes #10978
+
+- Updated dependencies []:
+  - @mastra/core@0.24.7-alpha.3
+
 ## 0.3.2
 
 ### Patch Changes

package/dist/index.cjs

@@ -22,6 +22,8 @@ var isMastraTextStreamChunk = (chunk) => {
     "source",
     "tool-input-start",
     "tool-input-delta",
+    "tool-call-approval",
+    "tool-call-suspended",
     "tool-call",
     "tool-result",
     "tool-error",
@@ -32,6 +34,8 @@ var isMastraTextStreamChunk = (chunk) => {
     "finish",
     "abort",
     "tool-input-end",
+    "object",
+    "tripwire",
     "raw"
   ].includes(chunk.type);
 };
@@ -1134,6 +1138,26 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
       };
     }
     default: {
+      if (isAgentExecutionDataChunkType(payload)) {
+        if (!("data" in payload.payload)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(payload)}`
+          );
+        }
+        const { type, data } = payload.payload;
+        return { type, data };
+      }
+      if (isWorkflowExecutionDataChunkType(payload)) {
+        if (!("data" in payload.payload)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(payload)}`
+          );
+        }
+        const { type, data } = payload.payload;
+        return { type, data };
+      }
       if (payload.type.startsWith("agent-execution-event-")) {
         const stepId = payload.payload.runId;
         const current = bufferedNetworks.get(payload.runId);
@@ -1148,6 +1172,15 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
           const { request, response, ...data } = result.data;
           step.task = data;
         }
+        bufferedNetworks.set(payload.runId, current);
+        return {
+          type: isNested ? "data-tool-network" : "data-network",
+          id: payload.runId,
+          data: {
+            ...current,
+            status: "running"
+          }
+        };
       }
       if (payload.type.startsWith("workflow-execution-event-")) {
         const stepId = payload.payload.runId;
@@ -1166,6 +1199,15 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
             step.task.id = data.name;
           }
         }
+        bufferedNetworks.set(payload.runId, current);
+        return {
+          type: isNested ? "data-tool-network" : "data-network",
+          id: payload.runId,
+          data: {
+            ...current,
+            status: "running"
+          }
+        };
       }
       if (isDataChunkType(payload)) {
         if (!("data" in payload)) {
@@ -1177,26 +1219,6 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         const { type, data } = payload;
         return { type, data };
       }
-      if (isAgentExecutionDataChunkType(payload)) {
-        if (!("data" in payload.payload)) {
-          throw new Error(
-            `UI Messages require a data property when using data- prefixed chunks 
- ${JSON.stringify(payload)}`
-          );
-        }
-        const { type, data } = payload.payload;
-        return { type, data };
-      }
-      if (isWorkflowExecutionDataChunkType(payload)) {
-        if (!("data" in payload.payload)) {
-          throw new Error(
-            `UI Messages require a data property when using data- prefixed chunks 
- ${JSON.stringify(payload)}`
-          );
-        }
-        const { type, data } = payload.payload;
-        return { type, data };
-      }
       return null;
     }
   }