ai 6.0.232 → 6.0.234

package/CHANGELOG.md

@@ -1,5 +1,26 @@
 # ai
 
+## 6.0.234
+
+### Patch Changes
+
+- 1efdef8: fix(ai): bound media-type sniffing decode for ID3-prefixed input
+
+  Media-type detection stripped ID3 tags before the ~18-byte prefix cap, decoding the entire base64 attachment (plus a full-size copy) whenever the data began with `ID3`/`SUQz`. This turned the intended O(1) sniff into an O(N) decode of the whole attachment. Detection now decodes at most a bounded prefix and skips the ID3 tag within that bound, keeping cost O(1) in input size on all paths (image, audio, and combined).
+
+- 49414cf: Return response piping promises so callers can catch stream read and write errors.
+- Updated dependencies [e7d064f]
+  - @ai-sdk/gateway@3.0.156
+
+## 6.0.233
+
+### Patch Changes
+
+- fe410e7: fix: detect MP4 audio from its ftyp box during transcription
+- af7188c: fix(ai): preserve tool parts when tool call IDs repeat across steps
+- Updated dependencies [a09f944]
+  - @ai-sdk/gateway@3.0.155
+
 ## 6.0.232
 
 ### Patch Changes

package/dist/index.d.mts

@@ -2395,7 +2395,7 @@ interface StreamTextResult<TOOLS extends ToolSet, OUTPUT extends Output> {
     /**
      * Writes UI message stream output to a Node.js response-like object.
      */
-    pipeUIMessageStreamToResponse<UI_MESSAGE extends UIMessage>(response: ServerResponse, options?: UIMessageStreamResponseInit & UIMessageStreamOptions<UI_MESSAGE>): void;
+    pipeUIMessageStreamToResponse<UI_MESSAGE extends UIMessage>(response: ServerResponse, options?: UIMessageStreamResponseInit & UIMessageStreamOptions<UI_MESSAGE>): Promise<void>;
     /**
      * Writes text delta output to a Node.js response-like object.
      * It sets a `Content-Type` header to `text/plain; charset=utf-8` and
@@ -2404,7 +2404,7 @@ interface StreamTextResult<TOOLS extends ToolSet, OUTPUT extends Output> {
      * @param response A Node.js response-like object (ServerResponse).
      * @param init Optional headers, status code, and status text.
      */
-    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): Promise<void>;
     /**
      * Converts the result to a streamed response object with a stream data part stream.
      *
@@ -4245,11 +4245,12 @@ declare class JsonToSseTransformStream extends TransformStream<unknown, string>
  * @param options.headers - Additional HTTP headers to include in the response.
  * @param options.stream - The UI message chunk stream to send.
  * @param options.consumeSseStream - Optional callback to consume a copy of the SSE stream independently.
+ * @returns A promise that resolves when the stream has been written.
  */
 declare function pipeUIMessageStreamToResponse({ response, status, statusText, headers, stream, consumeSseStream, }: {
     response: ServerResponse;
     stream: ReadableStream<UIMessageChunk>;
-} & UIMessageStreamResponseInit): void;
+} & UIMessageStreamResponseInit): Promise<void>;
 
 /**
  * Transforms a stream of `UIMessageChunk`s into an `AsyncIterableStream` of `UIMessage`s.
@@ -5398,7 +5399,7 @@ interface StreamObjectResult<PARTIAL, RESULT, ELEMENT_STREAM> {
      * @param response A Node.js response-like object (ServerResponse).
      * @param init Optional headers, status code, and status text.
      */
-    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): Promise<void>;
     /**
      * Creates a simple text stream response.
      * The response has a `Content-Type` header set to `text/plain; charset=utf-8`.
@@ -6283,11 +6284,12 @@ declare function createTextStreamResponse({ status, statusText, headers, textStr
  * @param options.statusText - Optional HTTP status text.
  * @param options.headers - Optional response headers.
  * @param options.textStream - The text stream to pipe.
+ * @returns A promise that resolves when the stream has been written.
  */
 declare function pipeTextStreamToResponse({ response, status, statusText, headers, textStream, }: {
     response: ServerResponse;
     textStream: ReadableStream<string>;
-} & ResponseInit): void;
+} & ResponseInit): Promise<void>;
 
 /**
  * The result of a `transcribe` call.

package/dist/index.d.ts

@@ -2395,7 +2395,7 @@ interface StreamTextResult<TOOLS extends ToolSet, OUTPUT extends Output> {
     /**
      * Writes UI message stream output to a Node.js response-like object.
      */
-    pipeUIMessageStreamToResponse<UI_MESSAGE extends UIMessage>(response: ServerResponse, options?: UIMessageStreamResponseInit & UIMessageStreamOptions<UI_MESSAGE>): void;
+    pipeUIMessageStreamToResponse<UI_MESSAGE extends UIMessage>(response: ServerResponse, options?: UIMessageStreamResponseInit & UIMessageStreamOptions<UI_MESSAGE>): Promise<void>;
     /**
      * Writes text delta output to a Node.js response-like object.
      * It sets a `Content-Type` header to `text/plain; charset=utf-8` and
@@ -2404,7 +2404,7 @@ interface StreamTextResult<TOOLS extends ToolSet, OUTPUT extends Output> {
      * @param response A Node.js response-like object (ServerResponse).
      * @param init Optional headers, status code, and status text.
      */
-    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): Promise<void>;
     /**
      * Converts the result to a streamed response object with a stream data part stream.
      *
@@ -4245,11 +4245,12 @@ declare class JsonToSseTransformStream extends TransformStream<unknown, string>
  * @param options.headers - Additional HTTP headers to include in the response.
  * @param options.stream - The UI message chunk stream to send.
  * @param options.consumeSseStream - Optional callback to consume a copy of the SSE stream independently.
+ * @returns A promise that resolves when the stream has been written.
  */
 declare function pipeUIMessageStreamToResponse({ response, status, statusText, headers, stream, consumeSseStream, }: {
     response: ServerResponse;
     stream: ReadableStream<UIMessageChunk>;
-} & UIMessageStreamResponseInit): void;
+} & UIMessageStreamResponseInit): Promise<void>;
 
 /**
  * Transforms a stream of `UIMessageChunk`s into an `AsyncIterableStream` of `UIMessage`s.
@@ -5398,7 +5399,7 @@ interface StreamObjectResult<PARTIAL, RESULT, ELEMENT_STREAM> {
      * @param response A Node.js response-like object (ServerResponse).
      * @param init Optional headers, status code, and status text.
      */
-    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): Promise<void>;
     /**
      * Creates a simple text stream response.
      * The response has a `Content-Type` header set to `text/plain; charset=utf-8`.
@@ -6283,11 +6284,12 @@ declare function createTextStreamResponse({ status, statusText, headers, textStr
  * @param options.statusText - Optional HTTP status text.
  * @param options.headers - Optional response headers.
  * @param options.textStream - The text stream to pipe.
+ * @returns A promise that resolves when the stream has been written.
  */
 declare function pipeTextStreamToResponse({ response, status, statusText, headers, textStream, }: {
     response: ServerResponse;
     textStream: ReadableStream<string>;
-} & ResponseInit): void;
+} & ResponseInit): Promise<void>;
 
 /**
  * The result of a `transcribe` call.

package/dist/index.js

@@ -1200,7 +1200,17 @@ var audioMediaTypeSignatures = [
   },
   {
     mediaType: "audio/mp4",
-    bytesPrefix: [102, 116, 121, 112]
+    bytesPrefix: [
+      0,
+      0,
+      0,
+      null,
+      102,
+      116,
+      121,
+      112
+      // ftyp
+    ]
   },
   {
     mediaType: "audio/webm",
@@ -1249,25 +1259,37 @@ var videoMediaTypeSignatures = [
     // RIFF (AVI)
   }
 ];
-var stripID3 = (data) => {
-  const bytes = typeof data === "string" ? (0, import_provider_utils2.convertBase64ToUint8Array)(data) : data;
+var DEFAULT_SNIFF_BYTES = 18;
+var MAX_SIGNATURE_BYTES = 12;
+var MAX_ID3_TAG_BYTES = 128 * 1024;
+var ID3_SCAN_BYTES = MAX_ID3_TAG_BYTES + MAX_SIGNATURE_BYTES;
+function decodePrefix(data, maxBytes) {
+  if (typeof data !== "string") {
+    return data.length > maxBytes ? data.subarray(0, maxBytes) : data;
+  }
+  const maxChars = Math.ceil(maxBytes / 3) * 4;
+  const bytes = (0, import_provider_utils2.convertBase64ToUint8Array)(
+    data.substring(0, Math.min(data.length, maxChars))
+  );
+  return bytes.length > maxBytes ? bytes.subarray(0, maxBytes) : bytes;
+}
+function hasID3(bytes) {
+  return bytes.length > 10 && bytes[0] === 73 && // 'I'
+  bytes[1] === 68 && // 'D'
+  bytes[2] === 51;
+}
+var stripID3 = (bytes) => {
   const id3Size = (bytes[6] & 127) << 21 | (bytes[7] & 127) << 14 | (bytes[8] & 127) << 7 | bytes[9] & 127;
-  return bytes.slice(id3Size + 10);
+  return bytes.subarray(id3Size + 10);
 };
-function stripID3TagsIfPresent(data) {
-  const hasId3 = typeof data === "string" && data.startsWith("SUQz") || typeof data !== "string" && data.length > 10 && data[0] === 73 && // 'I'
-  data[1] === 68 && // 'D'
-  data[2] === 51;
-  return hasId3 ? stripID3(data) : data;
-}
 function detectMediaType({
   data,
   signatures
 }) {
-  const processedData = stripID3TagsIfPresent(data);
-  const bytes = typeof processedData === "string" ? (0, import_provider_utils2.convertBase64ToUint8Array)(
-    processedData.substring(0, Math.min(processedData.length, 24))
-  ) : processedData;
+  let bytes = decodePrefix(data, DEFAULT_SNIFF_BYTES);
+  if (hasID3(bytes)) {
+    bytes = stripID3(decodePrefix(data, ID3_SCAN_BYTES));
+  }
   for (const signature of signatures) {
     if (bytes.length >= signature.bytesPrefix.length && signature.bytesPrefix.every(
       (byte, index) => byte === null || bytes[index] === byte
@@ -1282,7 +1304,7 @@ function detectMediaType({
 var import_provider_utils3 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.232" : "0.0.0-test";
+var VERSION = true ? "6.0.234" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5452,7 +5474,7 @@ function writeToServerResponse({
       response.end();
     }
   };
-  read();
+  return read();
 }
 
 // src/text-stream/pipe-text-stream-to-response.ts
@@ -5463,7 +5485,7 @@ function pipeTextStreamToResponse({
   headers,
   textStream
 }) {
-  writeToServerResponse({
+  return writeToServerResponse({
     response,
     status,
     statusText,
@@ -5786,11 +5808,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",
@@ -5800,9 +5843,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;
@@ -5848,9 +5891,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;
@@ -6030,7 +6073,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,
@@ -6133,7 +6176,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({
@@ -6181,31 +6224,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;
@@ -6213,30 +6262,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;
@@ -6441,7 +6496,7 @@ function pipeUIMessageStreamToResponse({
     sseStream = stream1;
     consumeSseStream({ stream: stream2 });
   }
-  writeToServerResponse({
+  return writeToServerResponse({
     response,
     status,
     statusText,
@@ -8678,7 +8733,7 @@ var DefaultStreamTextResult = class {
     onError,
     ...init
   } = {}) {
-    pipeUIMessageStreamToResponse({
+    return pipeUIMessageStreamToResponse({
       response,
       stream: this.toUIMessageStream({
         originalMessages,
@@ -8695,7 +8750,7 @@ var DefaultStreamTextResult = class {
     });
   }
   pipeTextStreamToResponse(response, init) {
-    pipeTextStreamToResponse({
+    return pipeTextStreamToResponse({
       response,
       textStream: this.textStream,
       ...init
@@ -9722,7 +9777,7 @@ async function pipeAgentUIStreamToResponse({
   consumeSseStream,
   ...options
 }) {
-  pipeUIMessageStreamToResponse({
+  return pipeUIMessageStreamToResponse({
     response,
     headers,
     status,
@@ -11669,7 +11724,7 @@ var DefaultStreamObjectResult = class {
     return createAsyncIterableStream(this.baseStream);
   }
   pipeTextStreamToResponse(response, init) {
-    pipeTextStreamToResponse({
+    return pipeTextStreamToResponse({
       response,
       textStream: this.textStream,
       ...init