@mastra/ai-sdk 1.2.1 → 1.3.0

package/dist/index.js

@@ -12357,6 +12357,34 @@ function AgentStreamToAISDKV6Transformer({
     onError
   });
 }
+function ensureAgentRunState(bufferedSteps, runId) {
+  if (!bufferedSteps.has(runId)) {
+    bufferedSteps.set(runId, {
+      id: "",
+      object: null,
+      finishReason: null,
+      usage: null,
+      warnings: [],
+      text: "",
+      reasoning: [],
+      sources: [],
+      files: [],
+      toolCalls: [],
+      toolResults: [],
+      request: {},
+      response: {
+        id: "",
+        timestamp: /* @__PURE__ */ new Date(),
+        modelId: "",
+        messages: []
+      },
+      providerMetadata: void 0,
+      steps: [],
+      status: "running"
+    });
+  }
+  return bufferedSteps.get(runId);
+}
 function transformAgent(payload, bufferedSteps) {
   let hasChanged = false;
   switch (payload.type) {
@@ -12397,7 +12425,7 @@ function transformAgent(payload, bufferedSteps) {
       });
       hasChanged = true;
       break;
-    case "text-delta":
+    case "text-delta": {
       const prevData = bufferedSteps.get(payload.runId);
       bufferedSteps.set(payload.runId, {
         ...prevData,
@@ -12405,6 +12433,7 @@ function transformAgent(payload, bufferedSteps) {
       });
       hasChanged = true;
       break;
+    }
     case "reasoning-delta":
       bufferedSteps.set(payload.runId, {
         ...bufferedSteps.get(payload.runId),
@@ -12433,13 +12462,15 @@ function transformAgent(payload, bufferedSteps) {
       });
       hasChanged = true;
       break;
-    case "tool-result":
+    case "tool-result": {
+      const toolResultRun = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
-        toolResults: [...bufferedSteps.get(payload.runId).toolResults, payload.payload]
+        ...toolResultRun,
+        toolResults: [...toolResultRun.toolResults, payload.payload]
       });
       hasChanged = true;
       break;
+    }
     case "object-result":
       bufferedSteps.set(payload.runId, {
         ...bufferedSteps.get(payload.runId),
@@ -12454,35 +12485,44 @@ function transformAgent(payload, bufferedSteps) {
       });
       hasChanged = true;
       break;
-    case "step-finish":
-      const currentRun = bufferedSteps.get(payload.runId);
+    case "step-finish": {
+      const stepRun = ensureAgentRunState(bufferedSteps, payload.runId);
       const stepResult = {
-        ...bufferedSteps.get(payload.runId),
-        stepType: currentRun.steps.length === 0 ? "initial" : "tool-result",
-        reasoningText: bufferedSteps.get(payload.runId).reasoning.join(""),
-        staticToolCalls: bufferedSteps.get(payload.runId).toolCalls.filter((part) => part.type === "tool-call" && part.payload?.dynamic === false),
-        dynamicToolCalls: bufferedSteps.get(payload.runId).toolCalls.filter((part) => part.type === "tool-call" && part.payload?.dynamic === true),
-        staticToolResults: bufferedSteps.get(payload.runId).toolResults.filter((part) => part.type === "tool-result" && part.payload?.dynamic === false),
-        dynamicToolResults: bufferedSteps.get(payload.runId).toolResults.filter((part) => part.type === "tool-result" && part.payload?.dynamic === true),
+        ...stepRun,
+        stepType: stepRun.steps.length === 0 ? "initial" : "tool-result",
+        reasoningText: stepRun.reasoning.join(""),
+        staticToolCalls: stepRun.toolCalls.filter(
+          (part) => part.type === "tool-call" && part.payload?.dynamic === false
+        ),
+        dynamicToolCalls: stepRun.toolCalls.filter(
+          (part) => part.type === "tool-call" && part.payload?.dynamic === true
+        ),
+        staticToolResults: stepRun.toolResults.filter(
+          (part) => part.type === "tool-result" && part.payload?.dynamic === false
+        ),
+        dynamicToolResults: stepRun.toolResults.filter(
+          (part) => part.type === "tool-result" && part.payload?.dynamic === true
+        ),
         finishReason: payload.payload.stepResult.reason,
         usage: payload.payload.output.usage,
         warnings: payload.payload.stepResult.warnings || [],
         response: {
+          ...stepRun.response,
           id: payload.payload.id || "",
           timestamp: payload.payload.metadata?.timestamp || /* @__PURE__ */ new Date(),
           modelId: payload.payload.metadata?.modelId || payload.payload.metadata?.model || "",
-          ...bufferedSteps.get(payload.runId).response,
-          messages: bufferedSteps.get(payload.runId).response.messages || []
+          messages: stepRun.response.messages || []
         }
       };
       bufferedSteps.set(payload.runId, {
-        ...bufferedSteps.get(payload.runId),
+        ...stepRun,
         usage: payload.payload.output.usage,
         warnings: payload.payload.stepResult.warnings || [],
-        steps: [...bufferedSteps.get(payload.runId).steps, stepResult]
+        steps: [...stepRun.steps, stepResult]
       });
       hasChanged = true;
       break;
+    }
   }
   if (hasChanged) {
     return {
@@ -13097,6 +13137,7 @@ async function handleChatStream({
   sendFinish = true,
   sendReasoning = false,
   sendSources = false,
+  onError,
   messageMetadata
 }) {
   const { messages, resumeData, runId, requestContext, trigger, ...rest } = params;
@@ -13148,6 +13189,7 @@ async function handleChatStream({
           sendFinish,
           sendReasoning,
           sendSources,
+          onError,
           messageMetadata
         })) {
           writer.write(part);
@@ -13165,6 +13207,7 @@ async function handleChatStream({
         sendFinish,
         sendReasoning,
         sendSources,
+        onError,
         messageMetadata
       })) {
         writer.write(part);