@ai-sdk/provider-utils 4.0.14 → 4.0.16

package/dist/index.mjs

@@ -50,25 +50,30 @@ function convertAsyncIteratorToReadableStream(iterator) {
 // src/create-tool-name-mapping.ts
 function createToolNameMapping({
   tools = [],
-  providerToolNames
+  providerToolNames,
+  resolveProviderToolName
 }) {
+  var _a2;
   const customToolNameToProviderToolName = {};
   const providerToolNameToCustomToolName = {};
   for (const tool2 of tools) {
-    if (tool2.type === "provider" && tool2.id in providerToolNames) {
-      const providerToolName = providerToolNames[tool2.id];
+    if (tool2.type === "provider") {
+      const providerToolName = (_a2 = resolveProviderToolName == null ? void 0 : resolveProviderToolName(tool2)) != null ? _a2 : tool2.id in providerToolNames ? providerToolNames[tool2.id] : void 0;
+      if (providerToolName == null) {
+        continue;
+      }
       customToolNameToProviderToolName[tool2.name] = providerToolName;
       providerToolNameToCustomToolName[providerToolName] = tool2.name;
     }
   }
   return {
     toProviderToolName: (customToolName) => {
-      var _a2;
-      return (_a2 = customToolNameToProviderToolName[customToolName]) != null ? _a2 : customToolName;
+      var _a3;
+      return (_a3 = customToolNameToProviderToolName[customToolName]) != null ? _a3 : customToolName;
     },
     toCustomToolName: (providerToolName) => {
-      var _a2;
-      return (_a2 = providerToolNameToCustomToolName[providerToolName]) != null ? _a2 : providerToolName;
+      var _a3;
+      return (_a3 = providerToolNameToCustomToolName[providerToolName]) != null ? _a3 : providerToolName;
     }
   };
 }
@@ -228,10 +233,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 +302,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 +479,7 @@ function withUserAgentSuffix(headers, ...userAgentSuffixParts) {
 }
 
 // src/version.ts
-var VERSION = true ? "4.0.14" : "0.0.0-test";
+var VERSION = true ? "4.0.16" : "0.0.0-test";
 
 // src/get-from-api.ts
 var getOriginalFetch = () => globalThis.fetch;
@@ -2519,6 +2588,7 @@ import {
   EventSourceParserStream as EventSourceParserStream2
 } from "eventsource-parser/stream";
 export {
+  DEFAULT_MAX_DOWNLOAD_SIZE,
   DelayedPromise,
   DownloadError,
   EventSourceParserStream2 as EventSourceParserStream,
@@ -2567,6 +2637,7 @@ export {
   postFormDataToApi,
   postJsonToApi,
   postToApi,
+  readResponseWithSizeLimit,
   removeUndefinedEntries,
   resolve,
   safeParseJSON,