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

package/dist/index.js

@@ -20,6 +20,8 @@ var isMastraTextStreamChunk = (chunk) => {
     "source",
     "tool-input-start",
     "tool-input-delta",
+    "tool-call-approval",
+    "tool-call-suspended",
     "tool-call",
     "tool-result",
     "tool-error",
@@ -30,6 +32,8 @@ var isMastraTextStreamChunk = (chunk) => {
     "finish",
     "abort",
     "tool-input-end",
+    "object",
+    "tripwire",
     "raw"
   ].includes(chunk.type);
 };
@@ -1132,6 +1136,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);
@@ -1146,6 +1170,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;
@@ -1164,6 +1197,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)) {
@@ -1175,26 +1217,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;
     }
   }