@ai-sdk/provider-utils 4.0.13 → 4.0.15

package/dist/index.mjs

@@ -228,10 +228,68 @@ var DownloadError = class extends (_b = AISDKError, _a = symbol, _b) {
   }
 };
 
+// src/read-response-with-size-limit.ts
+var DEFAULT_MAX_DOWNLOAD_SIZE = 2 * 1024 * 1024 * 1024;
+async function readResponseWithSizeLimit({
+  response,
+  url,
+  maxBytes = DEFAULT_MAX_DOWNLOAD_SIZE
+}) {
+  const contentLength = response.headers.get("content-length");
+  if (contentLength != null) {
+    const length = parseInt(contentLength, 10);
+    if (!isNaN(length) && length > maxBytes) {
+      throw new DownloadError({
+        url,
+        message: `Download of ${url} exceeded maximum size of ${maxBytes} bytes (Content-Length: ${length}).`
+      });
+    }
+  }
+  const body = response.body;
+  if (body == null) {
+    return new Uint8Array(0);
+  }
+  const reader = body.getReader();
+  const chunks = [];
+  let totalBytes = 0;
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) {
+        break;
+      }
+      totalBytes += value.length;
+      if (totalBytes > maxBytes) {
+        throw new DownloadError({
+          url,
+          message: `Download of ${url} exceeded maximum size of ${maxBytes} bytes.`
+        });
+      }
+      chunks.push(value);
+    }
+  } finally {
+    try {
+      await reader.cancel();
+    } finally {
+      reader.releaseLock();
+    }
+  }
+  const result = new Uint8Array(totalBytes);
+  let offset = 0;
+  for (const chunk of chunks) {
+    result.set(chunk, offset);
+    offset += chunk.length;
+  }
+  return result;
+}
+
 // src/download-blob.ts
-async function downloadBlob(url) {
+async function downloadBlob(url, options) {
+  var _a2, _b2;
   try {
-    const response = await fetch(url);
+    const response = await fetch(url, {
+      signal: options == null ? void 0 : options.abortSignal
+    });
     if (!response.ok) {
       throw new DownloadError({
         url,
@@ -239,7 +297,13 @@ async function downloadBlob(url) {
         statusText: response.statusText
       });
     }
-    return await response.blob();
+    const data = await readResponseWithSizeLimit({
+      response,
+      url,
+      maxBytes: (_a2 = options == null ? void 0 : options.maxBytes) != null ? _a2 : DEFAULT_MAX_DOWNLOAD_SIZE
+    });
+    const contentType = (_b2 = response.headers.get("content-type")) != null ? _b2 : void 0;
+    return new Blob([data], contentType ? { type: contentType } : void 0);
   } catch (error) {
     if (DownloadError.isInstance(error)) {
       throw error;
@@ -410,7 +474,7 @@ function withUserAgentSuffix(headers, ...userAgentSuffixParts) {
 }
 
 // src/version.ts
-var VERSION = true ? "4.0.13" : "0.0.0-test";
+var VERSION = true ? "4.0.15" : "0.0.0-test";
 
 // src/get-from-api.ts
 var getOriginalFetch = () => globalThis.fetch;
@@ -2519,6 +2583,7 @@ import {
   EventSourceParserStream as EventSourceParserStream2
 } from "eventsource-parser/stream";
 export {
+  DEFAULT_MAX_DOWNLOAD_SIZE,
   DelayedPromise,
   DownloadError,
   EventSourceParserStream2 as EventSourceParserStream,
@@ -2567,6 +2632,7 @@ export {
   postFormDataToApi,
   postJsonToApi,
   postToApi,
+  readResponseWithSizeLimit,
   removeUndefinedEntries,
   resolve,
   safeParseJSON,