ai 6.0.178 → 6.0.182

package/dist/index.mjs

@@ -1144,7 +1144,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.178" : "0.0.0-test";
+var VERSION = true ? "6.0.182" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1485,7 +1485,10 @@ function convertToLanguageModelMessage({
                 type: "tool-result",
                 toolCallId: part.toolCallId,
                 toolName: part.toolName,
-                output: mapToolResultOutput(part.output),
+                output: mapToolResultOutput({
+                  output: part.output,
+                  downloadedAssets
+                }),
                 providerOptions
               };
             }
@@ -1507,7 +1510,10 @@ function convertToLanguageModelMessage({
                 type: "tool-result",
                 toolCallId: part.toolCallId,
                 toolName: part.toolName,
-                output: mapToolResultOutput(part.output),
+                output: mapToolResultOutput({
+                  output: part.output,
+                  downloadedAssets
+                }),
                 providerOptions: part.providerOptions
               };
             }
@@ -1531,20 +1537,44 @@ function convertToLanguageModelMessage({
   }
 }
 async function downloadAssets(messages, download2, supportedUrls) {
-  const plannedDownloads = messages.filter((message) => message.role === "user").map((message) => message.content).filter(
-    (content) => Array.isArray(content)
-  ).flat().filter(
-    (part) => part.type === "image" || part.type === "file"
-  ).map((part) => {
-    var _a21;
-    const mediaType = (_a21 = part.mediaType) != null ? _a21 : part.type === "image" ? "image/*" : void 0;
-    let data = part.type === "image" ? part.image : part.data;
-    if (typeof data === "string") {
-      try {
-        data = new URL(data);
-      } catch (ignored) {
+  var _a21;
+  const downloadableFiles = [];
+  for (const message of messages) {
+    if (message.role === "user" && Array.isArray(message.content)) {
+      for (const part of message.content) {
+        if (part.type === "image" || part.type === "file") {
+          downloadableFiles.push({
+            data: part.type === "image" ? part.image : part.data,
+            mediaType: (_a21 = part.mediaType) != null ? _a21 : part.type === "image" ? "image/*" : void 0
+          });
+        }
       }
     }
+    if (message.role === "tool" || message.role === "assistant") {
+      if (!Array.isArray(message.content)) {
+        continue;
+      }
+      for (const part of message.content) {
+        if (part.type !== "tool-result") {
+          continue;
+        }
+        if (part.output.type !== "content") {
+          continue;
+        }
+        for (const contentPart of part.output.value) {
+          if (contentPart.type === "image-url" || contentPart.type === "file-url") {
+            downloadableFiles.push({
+              data: new URL(contentPart.url),
+              mediaType: contentPart.type === "image-url" ? "image/*" : void 0
+            });
+          }
+        }
+      }
+    }
+  }
+  const plannedDownloads = downloadableFiles.map((part) => {
+    const mediaType = part.mediaType;
+    const { data } = convertToLanguageModelV3DataContent(part.data);
     return { mediaType, data };
   }).filter(
     (part) => part.data instanceof URL
@@ -1625,13 +1655,41 @@ function convertPartToLanguageModelPart(part, downloadedAssets) {
     }
   }
 }
-function mapToolResultOutput(output) {
+function mapToolResultOutput({
+  output,
+  downloadedAssets
+}) {
   if (output.type !== "content") {
     return output;
   }
   return {
     type: "content",
     value: output.value.map((item) => {
+      var _a21, _b;
+      if (item.type === "image-url") {
+        const downloadedFile = downloadedAssets[new URL(item.url).toString()];
+        if (downloadedFile) {
+          return {
+            type: "image-data",
+            data: convertDataContentToBase64String(downloadedFile.data),
+            mediaType: (_a21 = downloadedFile.mediaType) != null ? _a21 : "image/*",
+            providerOptions: item.providerOptions
+          };
+        }
+        return item;
+      }
+      if (item.type === "file-url") {
+        const downloadedFile = downloadedAssets[new URL(item.url).toString()];
+        if (downloadedFile) {
+          return {
+            type: "file-data",
+            data: convertDataContentToBase64String(downloadedFile.data),
+            mediaType: (_b = downloadedFile.mediaType) != null ? _b : "application/octet-stream",
+            providerOptions: item.providerOptions
+          };
+        }
+        return item;
+      }
       if (item.type !== "media") {
         return item;
       }
@@ -2560,12 +2618,13 @@ function mergeObjects(base, overrides) {
 
 // src/util/retry-with-exponential-backoff.ts
 import { APICallError as APICallError2 } from "@ai-sdk/provider";
+import { GatewayError } from "@ai-sdk/gateway";
 import { delay, getErrorMessage as getErrorMessage4, isAbortError } from "@ai-sdk/provider-utils";
 function getRetryDelayInMs({
   error,
   exponentialBackoffDelay
 }) {
-  const headers = error.responseHeaders;
+  const headers = APICallError2.isInstance(error) ? error.responseHeaders : APICallError2.isInstance(error.cause) ? error.cause.responseHeaders : void 0;
   if (!headers)
     return exponentialBackoffDelay;
   let ms;
@@ -2626,7 +2685,7 @@ async function _retryWithExponentialBackoff(f, {
         errors: newErrors
       });
     }
-    if (error instanceof Error && APICallError2.isInstance(error) && error.isRetryable === true && tryNumber <= maxRetries) {
+    if (error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true) && tryNumber <= maxRetries) {
       await delay(
         getRetryDelayInMs({
           error,
@@ -8766,7 +8825,7 @@ var uiMessagesSchema = lazySchema2(
               toolCallId: z8.string(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z8.literal("output-error"),
-              input: z8.unknown(),
+              input: z8.unknown().optional(),
               rawInput: z8.unknown().optional(),
               providerExecuted: z8.boolean().optional(),
               output: z8.never().optional(),
@@ -8876,7 +8935,7 @@ var uiMessagesSchema = lazySchema2(
               toolMetadata: toolMetadataSchema2.optional(),
               state: z8.literal("output-error"),
               providerExecuted: z8.boolean().optional(),
-              input: z8.unknown(),
+              input: z8.unknown().optional(),
               rawInput: z8.unknown().optional(),
               output: z8.never().optional(),
               errorText: z8.string(),