@mastra/ai-sdk 1.6.2-alpha.1 → 1.6.2-alpha.2

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/ai-sdk
 
+## 1.6.2-alpha.2
+
+### Patch Changes
+
+- Fixed a crash in the AI SDK stream transformer when a supervisor agent delegates to a remote A2A agent (A2AAgent) through chatRoute() or handleChatStream. A2A sub-agent streams do not emit a start chunk and use a flat finish payload, which caused the UI stream to end with "Cannot read properties of undefined" errors. The remote agent's answer now streams to the client correctly. ([#19443](https://github.com/mastra-ai/mastra/pull/19443))
+
+  The same guard now also covers resumed sub-agent streams, which skip the start chunk as well and could crash or corrupt the delegation state on tool calls and structured output.
+
 ## 1.6.2-alpha.1
 
 ### Patch Changes

package/dist/index.cjs

@@ -10846,23 +10846,26 @@ function transformAgent(payload, bufferedSteps) {
       hasChanged = true;
       break;
     }
-    case "finish":
+    case "finish": {
+      const finishRun = ensureAgentRunState(bufferedSteps, payload.runId);
+      const finishPayload = payload.payload;
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
-        finishReason: payload.payload.stepResult.reason,
-        usage: payload.payload?.output?.usage,
-        warnings: payload.payload?.stepResult?.warnings,
-        steps: bufferedSteps.get(payload.runId).steps,
+        ...finishRun,
+        finishReason: finishPayload.stepResult?.reason ?? finishPayload.finishReason ?? null,
+        usage: finishPayload.output?.usage ?? finishPayload.usage,
+        warnings: finishPayload.stepResult?.warnings,
+        steps: finishRun.steps,
         status: "finished",
         response: {
-          ...bufferedSteps.get(payload.runId).response,
-          ...payload.payload.response || {}
+          ...finishRun.response,
+          ...finishPayload.response || {}
         }
       });
       hasChanged = true;
       break;
+    }
     case "text-delta": {
-      const prevData = bufferedSteps.get(payload.runId);
+      const prevData = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
         ...prevData,
         text: `${prevData.text}${payload.payload.text}`
@@ -10870,38 +10873,43 @@ function transformAgent(payload, bufferedSteps) {
       hasChanged = true;
       break;
     }
-    case "reasoning-delta":
+    case "reasoning-delta": {
+      const reasoningRun = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
-        reasoning: [...bufferedSteps.get(payload.runId).reasoning, payload.payload.text]
+        ...reasoningRun,
+        reasoning: [...reasoningRun.reasoning, payload.payload.text]
       });
       hasChanged = true;
       break;
-    case "source":
+    }
+    case "source": {
+      const sourceRun = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
-        sources: [...bufferedSteps.get(payload.runId).sources, payload.payload]
+        ...sourceRun,
+        sources: [...sourceRun.sources, payload.payload]
       });
       hasChanged = true;
       break;
-    case "file":
+    }
+    case "file": {
+      const fileRun = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
-        files: [...bufferedSteps.get(payload.runId).files, payload.payload]
+        ...fileRun,
+        files: [...fileRun.files, payload.payload]
       });
       hasChanged = true;
       break;
-    case "tool-call":
+    }
+    case "tool-call": {
+      const toolCallRun = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
-        pendingToolCalls: removePendingToolCall(
-          bufferedSteps.get(payload.runId).pendingToolCalls,
-          payload.payload.toolCallId
-        ),
-        toolCalls: [...bufferedSteps.get(payload.runId).toolCalls, payload.payload]
+        ...toolCallRun,
+        pendingToolCalls: removePendingToolCall(toolCallRun.pendingToolCalls, payload.payload.toolCallId),
+        toolCalls: [...toolCallRun.toolCalls, payload.payload]
       });
       hasChanged = true;
       break;
+    }
     case "tool-result": {
       const toolResultRun = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
@@ -10914,14 +10922,14 @@ function transformAgent(payload, bufferedSteps) {
     }
     case "object-result":
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
+        ...ensureAgentRunState(bufferedSteps, payload.runId),
         object: payload.object
       });
       hasChanged = true;
       break;
     case "object":
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
+        ...ensureAgentRunState(bufferedSteps, payload.runId),
         object: payload.object
       });
       hasChanged = true;