ai 6.0.83 → 6.0.85

package/dist/index.mjs

@@ -1093,17 +1093,25 @@ function detectMediaType({
 }
 
 // src/util/download/download.ts
-import { DownloadError as DownloadError2 } from "@ai-sdk/provider-utils";
+import {
+  DownloadError as DownloadError2,
+  readResponseWithSizeLimit,
+  DEFAULT_MAX_DOWNLOAD_SIZE
+} from "@ai-sdk/provider-utils";
 import {
   withUserAgentSuffix,
   getRuntimeEnvironmentUserAgent
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.83" : "0.0.0-test";
+var VERSION = true ? "6.0.85" : "0.0.0-test";
 
 // src/util/download/download.ts
-var download = async ({ url }) => {
+var download = async ({
+  url,
+  maxBytes,
+  abortSignal
+}) => {
   var _a21;
   const urlText = url.toString();
   try {
@@ -1112,7 +1120,8 @@ var download = async ({ url }) => {
         {},
         `ai-sdk/${VERSION}`,
         getRuntimeEnvironmentUserAgent()
-      )
+      ),
+      signal: abortSignal
     });
     if (!response.ok) {
       throw new DownloadError2({
@@ -1121,8 +1130,13 @@ var download = async ({ url }) => {
         statusText: response.statusText
       });
     }
+    const data = await readResponseWithSizeLimit({
+      response,
+      url: urlText,
+      maxBytes: maxBytes != null ? maxBytes : DEFAULT_MAX_DOWNLOAD_SIZE
+    });
     return {
-      data: new Uint8Array(await response.arrayBuffer()),
+      data,
       mediaType: (_a21 = response.headers.get("content-type")) != null ? _a21 : void 0
     };
   } catch (error) {
@@ -9886,6 +9900,11 @@ function cosineSimilarity(vector1, vector2) {
   return magnitudeSquared1 === 0 || magnitudeSquared2 === 0 ? 0 : dotProduct / (Math.sqrt(magnitudeSquared1) * Math.sqrt(magnitudeSquared2));
 }
 
+// src/util/download/create-download.ts
+function createDownload(options) {
+  return ({ url, abortSignal }) => download({ url, maxBytes: options == null ? void 0 : options.maxBytes, abortSignal });
+}
+
 // src/util/data-url.ts
 function getTextFromDataUrl(dataUrl) {
   const [header, base64Content] = dataUrl.split(",");
@@ -10795,6 +10814,7 @@ import {
   convertBase64ToUint8Array as convertBase64ToUint8Array5,
   withUserAgentSuffix as withUserAgentSuffix8
 } from "@ai-sdk/provider-utils";
+var defaultDownload = createDownload();
 async function experimental_generateVideo({
   model: modelArg,
   prompt: promptArg,
@@ -10808,7 +10828,8 @@ async function experimental_generateVideo({
   providerOptions,
   maxRetries: maxRetriesArg,
   abortSignal,
-  headers
+  headers,
+  download: downloadFn = defaultDownload
 }) {
   var _a21;
   const model = resolveVideoModel(modelArg);
@@ -10854,8 +10875,9 @@ async function experimental_generateVideo({
     for (const videoData of result.videos) {
       switch (videoData.type) {
         case "url": {
-          const { data, mediaType: downloadedMediaType } = await download({
-            url: new URL(videoData.url)
+          const { data, mediaType: downloadedMediaType } = await downloadFn({
+            url: new URL(videoData.url),
+            abortSignal
           });
           const isUsableMediaType = (type) => !!type && type !== "application/octet-stream";
           const mediaType = isUsableMediaType(videoData.mediaType) && videoData.mediaType || isUsableMediaType(downloadedMediaType) && downloadedMediaType || detectMediaType({
@@ -12017,13 +12039,15 @@ var DefaultRerankResult = class {
 
 // src/transcribe/transcribe.ts
 import { withUserAgentSuffix as withUserAgentSuffix9 } from "@ai-sdk/provider-utils";
+var defaultDownload2 = createDownload();
 async function transcribe({
   model,
   audio,
   providerOptions = {},
   maxRetries: maxRetriesArg,
   abortSignal,
-  headers
+  headers,
+  download: downloadFn = defaultDownload2
 }) {
   const resolvedModel = resolveTranscriptionModel(model);
   if (!resolvedModel) {
@@ -12037,7 +12061,7 @@ async function transcribe({
     headers != null ? headers : {},
     `ai/${VERSION}`
   );
-  const audioData = audio instanceof URL ? (await download({ url: audio })).data : convertDataContentToUint8Array(audio);
+  const audioData = audio instanceof URL ? (await downloadFn({ url: audio, abortSignal })).data : convertDataContentToUint8Array(audio);
   const result = await retry(
     () => {
       var _a21;
@@ -12960,6 +12984,7 @@ export {
   cosineSimilarity,
   createAgentUIStream,
   createAgentUIStreamResponse,
+  createDownload,
   createGateway,
   createIdGenerator5 as createIdGenerator,
   createProviderRegistry,