@mastra/ai-sdk 1.4.2 → 1.4.3-alpha.0

package/dist/index.js

@@ -12905,6 +12905,7 @@ function ensureAgentRunState(bufferedSteps, runId) {
       sources: [],
       files: [],
       toolCalls: [],
+      pendingToolCalls: [],
       toolResults: [],
       request: {},
       response: {
@@ -12920,6 +12921,44 @@ function ensureAgentRunState(bufferedSteps, runId) {
   }
   return bufferedSteps.get(runId);
 }
+function upsertPendingToolCall(pendingToolCalls = [], toolCallId, updates) {
+  const existingIndex = pendingToolCalls.findIndex((call) => call.toolCallId === toolCallId);
+  if (existingIndex === -1) {
+    return [
+      ...pendingToolCalls,
+      {
+        toolCallId,
+        toolName: updates.toolName || "",
+        argsText: updates.argsText || "",
+        state: updates.state || "input-streaming",
+        ...updates.providerExecuted != null ? { providerExecuted: updates.providerExecuted } : {},
+        ...updates.providerMetadata != null ? { providerMetadata: updates.providerMetadata } : {},
+        ...updates.dynamic != null ? { dynamic: updates.dynamic } : {}
+      }
+    ];
+  }
+  return pendingToolCalls.map((call, index) => {
+    if (index !== existingIndex) return call;
+    return {
+      ...call,
+      ...updates,
+      toolName: updates.toolName || call.toolName,
+      argsText: updates.argsText ?? call.argsText
+    };
+  });
+}
+function appendPendingToolCallArgs(pendingToolCalls = [], payload) {
+  const existing = pendingToolCalls.find((call) => call.toolCallId === payload.toolCallId);
+  return upsertPendingToolCall(pendingToolCalls, payload.toolCallId, {
+    toolName: payload.toolName || existing?.toolName || "",
+    argsText: `${existing?.argsText || ""}${payload.argsTextDelta || ""}`,
+    state: "input-streaming",
+    providerMetadata: payload.providerMetadata ?? existing?.providerMetadata
+  });
+}
+function removePendingToolCall(pendingToolCalls = [], toolCallId) {
+  return pendingToolCalls.filter((call) => call.toolCallId !== toolCallId);
+}
 function transformAgent(payload, bufferedSteps) {
   let hasChanged = false;
   switch (payload.type) {
@@ -12935,6 +12974,7 @@ function transformAgent(payload, bufferedSteps) {
         sources: [],
         files: [],
         toolCalls: [],
+        pendingToolCalls: [],
         toolResults: [],
         request: {},
         response: {
@@ -12949,6 +12989,50 @@ function transformAgent(payload, bufferedSteps) {
       });
       hasChanged = true;
       break;
+    case "tool-call-input-streaming-start": {
+      const toolInputStartRun = ensureAgentRunState(bufferedSteps, payload.runId);
+      const existing = toolInputStartRun.pendingToolCalls?.find(
+        (call) => call.toolCallId === payload.payload.toolCallId
+      );
+      bufferedSteps.set(payload.runId, {
+        ...toolInputStartRun,
+        pendingToolCalls: upsertPendingToolCall(toolInputStartRun.pendingToolCalls, payload.payload.toolCallId, {
+          toolName: payload.payload.toolName,
+          argsText: existing?.argsText ?? "",
+          state: "input-streaming",
+          providerExecuted: payload.payload.providerExecuted,
+          providerMetadata: payload.payload.providerMetadata,
+          dynamic: payload.payload.dynamic
+        })
+      });
+      hasChanged = true;
+      break;
+    }
+    case "tool-call-delta": {
+      const toolCallDeltaRun = ensureAgentRunState(bufferedSteps, payload.runId);
+      bufferedSteps.set(payload.runId, {
+        ...toolCallDeltaRun,
+        pendingToolCalls: appendPendingToolCallArgs(toolCallDeltaRun.pendingToolCalls, payload.payload)
+      });
+      hasChanged = true;
+      break;
+    }
+    case "tool-call-input-streaming-end": {
+      const toolInputEndRun = ensureAgentRunState(bufferedSteps, payload.runId);
+      const existing = toolInputEndRun.pendingToolCalls?.find(
+        (call) => call.toolCallId === payload.payload.toolCallId
+      );
+      bufferedSteps.set(payload.runId, {
+        ...toolInputEndRun,
+        pendingToolCalls: upsertPendingToolCall(toolInputEndRun.pendingToolCalls, payload.payload.toolCallId, {
+          toolName: existing?.toolName || "",
+          state: "input-available",
+          providerMetadata: payload.payload.providerMetadata ?? existing?.providerMetadata
+        })
+      });
+      hasChanged = true;
+      break;
+    }
     case "finish":
       bufferedSteps.set(payload.runId, {
         ...bufferedSteps.get(payload.runId),
@@ -12997,6 +13081,10 @@ function transformAgent(payload, bufferedSteps) {
     case "tool-call":
       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]
       });
       hasChanged = true;
@@ -13005,6 +13093,7 @@ function transformAgent(payload, bufferedSteps) {
       const toolResultRun = ensureAgentRunState(bufferedSteps, payload.runId);
       bufferedSteps.set(payload.runId, {
         ...toolResultRun,
+        pendingToolCalls: removePendingToolCall(toolResultRun.pendingToolCalls, payload.payload.toolCallId),
         toolResults: [...toolResultRun.toolResults, payload.payload]
       });
       hasChanged = true;
@@ -13035,6 +13124,7 @@ function transformAgent(payload, bufferedSteps) {
         ...stepRunWithoutSteps,
         text: stepText,
         reasoning: stepReasoning,
+        pendingToolCalls: [],
         stepType: stepRun.steps.length === 0 ? "initial" : "tool-result",
         reasoningText: stepReasoning.join(""),
         staticToolCalls: stepRun.toolCalls.filter(
@@ -13066,6 +13156,7 @@ function transformAgent(payload, bufferedSteps) {
         sources: [],
         files: [],
         toolCalls: [],
+        pendingToolCalls: [],
         toolResults: [],
         usage: payload.payload.output.usage,
         warnings: payload.payload.stepResult.warnings || [],