ai 6.0.232 → 6.0.233

package/dist/index.mjs

@@ -1081,7 +1081,17 @@ var audioMediaTypeSignatures = [
   },
   {
     mediaType: "audio/mp4",
-    bytesPrefix: [102, 116, 121, 112]
+    bytesPrefix: [
+      0,
+      0,
+      0,
+      null,
+      102,
+      116,
+      121,
+      112
+      // ftyp
+    ]
   },
   {
     mediaType: "audio/webm",
@@ -1171,7 +1181,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.232" : "0.0.0-test";
+var VERSION = true ? "6.0.233" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5712,11 +5722,32 @@ function processUIMessageStream({
       async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
           var _a22, _b, _c, _d;
+          function getCurrentStepParts() {
+            const parts = state.message.parts;
+            let currentStepStartIndex = parts.length - 1;
+            while (currentStepStartIndex >= 0 && parts[currentStepStartIndex].type !== "step-start") {
+              currentStepStartIndex--;
+            }
+            return parts.slice(currentStepStartIndex + 1);
+          }
+          function getCurrentStepToolInvocations() {
+            return getCurrentStepParts().filter(isToolUIPart);
+          }
           function getToolInvocation(toolCallId) {
-            const toolInvocations = state.message.parts.filter(isToolUIPart);
-            const toolInvocation = toolInvocations.find(
+            const toolInvocations = getCurrentStepToolInvocations();
+            let toolInvocation = toolInvocations.find(
               (invocation) => invocation.toolCallId === toolCallId
             );
+            if (toolInvocation == null) {
+              const parts = state.message.parts;
+              for (let i = parts.length - 1; i >= 0; i--) {
+                const part = parts[i];
+                if (isToolUIPart(part) && part.toolCallId === toolCallId) {
+                  toolInvocation = part;
+                  break;
+                }
+              }
+            }
             if (toolInvocation == null) {
               throw new UIMessageStreamError({
                 chunkType: "tool-invocation",
@@ -5726,9 +5757,9 @@ function processUIMessageStream({
             }
             return toolInvocation;
           }
-          function updateToolPart(options) {
+          function updateToolPart(options, existingPart) {
             var _a23;
-            const part = state.message.parts.find(
+            const part = existingPart != null ? existingPart : getCurrentStepParts().find(
               (part2) => isStaticToolUIPart(part2) && part2.toolCallId === options.toolCallId
             );
             const anyOptions = options;
@@ -5774,9 +5805,9 @@ function processUIMessageStream({
               });
             }
           }
-          function updateDynamicToolPart(options) {
+          function updateDynamicToolPart(options, existingPart) {
             var _a23, _b2;
-            const part = state.message.parts.find(
+            const part = existingPart != null ? existingPart : getCurrentStepParts().find(
               (part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId
             );
             const anyOptions = options;
@@ -5956,7 +5987,7 @@ function processUIMessageStream({
               break;
             }
             case "tool-input-start": {
-              const toolInvocations = state.message.parts.filter(isStaticToolUIPart);
+              const toolInvocations = getCurrentStepParts().filter(isStaticToolUIPart);
               state.partialToolCalls[chunk.toolCallId] = {
                 text: "",
                 toolName: chunk.toolName,
@@ -6059,7 +6090,7 @@ function processUIMessageStream({
               break;
             }
             case "tool-input-error": {
-              const existingPart = state.message.parts.filter(isToolUIPart).find((p) => p.toolCallId === chunk.toolCallId);
+              const existingPart = getCurrentStepParts().filter(isToolUIPart).find((p) => p.toolCallId === chunk.toolCallId);
               const isDynamic = existingPart != null ? existingPart.type === "dynamic-tool" : !!chunk.dynamic;
               if (isDynamic) {
                 updateDynamicToolPart({
@@ -6107,31 +6138,37 @@ function processUIMessageStream({
             case "tool-output-available": {
               const toolInvocation = getToolInvocation(chunk.toolCallId);
               if (toolInvocation.type === "dynamic-tool") {
-                updateDynamicToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: toolInvocation.toolName,
-                  state: "output-available",
-                  input: toolInvocation.input,
-                  output: chunk.output,
-                  preliminary: chunk.preliminary,
-                  providerExecuted: chunk.providerExecuted,
-                  providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title,
-                  toolMetadata: toolInvocation.toolMetadata
-                });
+                updateDynamicToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: toolInvocation.toolName,
+                    state: "output-available",
+                    input: toolInvocation.input,
+                    output: chunk.output,
+                    preliminary: chunk.preliminary,
+                    providerExecuted: chunk.providerExecuted,
+                    providerMetadata: chunk.providerMetadata,
+                    title: toolInvocation.title,
+                    toolMetadata: toolInvocation.toolMetadata
+                  },
+                  toolInvocation
+                );
               } else {
-                updateToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: getStaticToolName(toolInvocation),
-                  state: "output-available",
-                  input: toolInvocation.input,
-                  output: chunk.output,
-                  providerExecuted: chunk.providerExecuted,
-                  preliminary: chunk.preliminary,
-                  providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title,
-                  toolMetadata: toolInvocation.toolMetadata
-                });
+                updateToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: getStaticToolName(toolInvocation),
+                    state: "output-available",
+                    input: toolInvocation.input,
+                    output: chunk.output,
+                    providerExecuted: chunk.providerExecuted,
+                    preliminary: chunk.preliminary,
+                    providerMetadata: chunk.providerMetadata,
+                    title: toolInvocation.title,
+                    toolMetadata: toolInvocation.toolMetadata
+                  },
+                  toolInvocation
+                );
               }
               write();
               break;
@@ -6139,30 +6176,36 @@ function processUIMessageStream({
             case "tool-output-error": {
               const toolInvocation = getToolInvocation(chunk.toolCallId);
               if (toolInvocation.type === "dynamic-tool") {
-                updateDynamicToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: toolInvocation.toolName,
-                  state: "output-error",
-                  input: toolInvocation.input,
-                  errorText: chunk.errorText,
-                  providerExecuted: chunk.providerExecuted,
-                  providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title,
-                  toolMetadata: toolInvocation.toolMetadata
-                });
+                updateDynamicToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: toolInvocation.toolName,
+                    state: "output-error",
+                    input: toolInvocation.input,
+                    errorText: chunk.errorText,
+                    providerExecuted: chunk.providerExecuted,
+                    providerMetadata: chunk.providerMetadata,
+                    title: toolInvocation.title,
+                    toolMetadata: toolInvocation.toolMetadata
+                  },
+                  toolInvocation
+                );
               } else {
-                updateToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: getStaticToolName(toolInvocation),
-                  state: "output-error",
-                  input: toolInvocation.input,
-                  rawInput: toolInvocation.rawInput,
-                  errorText: chunk.errorText,
-                  providerExecuted: chunk.providerExecuted,
-                  providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title,
-                  toolMetadata: toolInvocation.toolMetadata
-                });
+                updateToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: getStaticToolName(toolInvocation),
+                    state: "output-error",
+                    input: toolInvocation.input,
+                    rawInput: toolInvocation.rawInput,
+                    errorText: chunk.errorText,
+                    providerExecuted: chunk.providerExecuted,
+                    providerMetadata: chunk.providerMetadata,
+                    title: toolInvocation.title,
+                    toolMetadata: toolInvocation.toolMetadata
+                  },
+                  toolInvocation
+                );
               }
               write();
               break;