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

package/dist/index.js

@@ -10825,23 +10825,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}`
@@ -10849,38 +10852,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, {
@@ -10893,14 +10901,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;