ai 6.0.83 → 6.0.85

package/src/transcribe/transcribe.ts

@@ -10,7 +10,7 @@ import {
   audioMediaTypeSignatures,
   detectMediaType,
 } from '../util/detect-media-type';
-import { download } from '../util/download/download';
+import { createDownload } from '../util/download/create-download';
 import { prepareRetries } from '../util/prepare-retries';
 import { TranscriptionResult } from './transcribe-result';
 import { VERSION } from '../version';
@@ -29,6 +29,8 @@ import { Warning } from '../types';
  *
  * @returns A result object that contains the generated transcript.
  */
+const defaultDownload = createDownload();
+
 export async function transcribe({
   model,
   audio,
@@ -36,6 +38,7 @@ export async function transcribe({
   maxRetries: maxRetriesArg,
   abortSignal,
   headers,
+  download: downloadFn = defaultDownload,
 }: {
   /**
    * The transcription model to use.
@@ -80,6 +83,17 @@ export async function transcribe({
    * Only applicable for HTTP-based providers.
    */
   headers?: Record<string, string>;
+
+  /**
+   * Custom download function for fetching audio from URLs.
+   * Use `createDownload()` from `ai` to create a download function with custom size limits.
+   *
+   * @default createDownload() (2 GiB limit)
+   */
+  download?: (options: {
+    url: URL;
+    abortSignal?: AbortSignal;
+  }) => Promise<{ data: Uint8Array; mediaType: string | undefined }>;
 }): Promise<TranscriptionResult> {
   const resolvedModel = resolveTranscriptionModel(model);
   if (!resolvedModel) {
@@ -98,7 +112,7 @@ export async function transcribe({
 
   const audioData =
     audio instanceof URL
-      ? (await download({ url: audio })).data
+      ? (await downloadFn({ url: audio, abortSignal })).data
       : convertDataContentToUint8Array(audio);
 
   const result = await retry(() =>

package/src/util/download/create-download.ts

@@ -0,0 +1,13 @@
+import { download as internalDownload } from './download';
+
+/**
+ * Creates a download function with configurable options.
+ *
+ * @param options - Configuration options for the download function.
+ * @param options.maxBytes - Maximum allowed download size in bytes. Default: 2 GiB.
+ * @returns A download function that can be passed to `transcribe()` or `experimental_generateVideo()`.
+ */
+export function createDownload(options?: { maxBytes?: number }) {
+  return ({ url, abortSignal }: { url: URL; abortSignal?: AbortSignal }) =>
+    internalDownload({ url, maxBytes: options?.maxBytes, abortSignal });
+}

package/src/util/download/download.ts

@@ -1,4 +1,8 @@
-import { DownloadError } from '@ai-sdk/provider-utils';
+import {
+  DownloadError,
+  readResponseWithSizeLimit,
+  DEFAULT_MAX_DOWNLOAD_SIZE,
+} from '@ai-sdk/provider-utils';
 import {
   withUserAgentSuffix,
   getRuntimeEnvironmentUserAgent,
@@ -9,11 +13,21 @@ import { VERSION } from '../../version';
  * Download a file from a URL.
  *
  * @param url - The URL to download from.
+ * @param maxBytes - Maximum allowed download size in bytes. Defaults to 100 MiB.
+ * @param abortSignal - An optional abort signal to cancel the download.
  * @returns The downloaded data and media type.
  *
- * @throws DownloadError if the download fails.
+ * @throws DownloadError if the download fails or exceeds maxBytes.
  */
-export const download = async ({ url }: { url: URL }) => {
+export const download = async ({
+  url,
+  maxBytes,
+  abortSignal,
+}: {
+  url: URL;
+  maxBytes?: number;
+  abortSignal?: AbortSignal;
+}) => {
   const urlText = url.toString();
   try {
     const response = await fetch(urlText, {
@@ -22,6 +36,7 @@ export const download = async ({ url }: { url: URL }) => {
         `ai-sdk/${VERSION}`,
         getRuntimeEnvironmentUserAgent(),
       ),
+      signal: abortSignal,
     });
 
     if (!response.ok) {
@@ -32,8 +47,14 @@ export const download = async ({ url }: { url: URL }) => {
       });
     }
 
+    const data = await readResponseWithSizeLimit({
+      response,
+      url: urlText,
+      maxBytes: maxBytes ?? DEFAULT_MAX_DOWNLOAD_SIZE,
+    });
+
     return {
-      data: new Uint8Array(await response.arrayBuffer()),
+      data,
       mediaType: response.headers.get('content-type') ?? undefined,
     };
   } catch (error) {

package/src/util/index.ts

@@ -1,6 +1,7 @@
 export type { AsyncIterableStream } from './async-iterable-stream';
 export { consumeStream } from './consume-stream';
 export { cosineSimilarity } from './cosine-similarity';
+export { createDownload } from './download/create-download';
 export { getTextFromDataUrl } from './data-url';
 export type { DeepPartial } from './deep-partial';
 export type { DownloadFunction as Experimental_DownloadFunction } from './download/download-function';