ai 5.0.253 → 5.0.255

package/dist/index.mjs

@@ -727,7 +727,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.253" : "0.0.0-test";
+var VERSION = true ? "5.0.255" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1779,18 +1779,21 @@ var retryWithExponentialBackoffRespectingRetryHeaders = ({
   maxRetries = 2,
   initialDelayInMs = 2e3,
   backoffFactor = 2,
-  abortSignal
+  abortSignal,
+  additionalRetryableError
 } = {}) => async (f) => _retryWithExponentialBackoff(f, {
   maxRetries,
   delayInMs: initialDelayInMs,
   backoffFactor,
-  abortSignal
+  abortSignal,
+  additionalRetryableError
 });
 async function _retryWithExponentialBackoff(f, {
   maxRetries,
   delayInMs,
   backoffFactor,
-  abortSignal
+  abortSignal,
+  additionalRetryableError
 }, errors = []) {
   try {
     return await f();
@@ -1811,7 +1814,7 @@ async function _retryWithExponentialBackoff(f, {
         errors: newErrors
       });
     }
-    if (error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true) && tryNumber <= maxRetries) {
+    if ((error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true) || (additionalRetryableError == null ? void 0 : additionalRetryableError(error)) === true) && tryNumber <= maxRetries) {
       await delay(
         getRetryDelayInMs({
           error,
@@ -1825,7 +1828,8 @@ async function _retryWithExponentialBackoff(f, {
           maxRetries,
           delayInMs: backoffFactor * delayInMs,
           backoffFactor,
-          abortSignal
+          abortSignal,
+          additionalRetryableError
         },
         newErrors
       );
@@ -1844,7 +1848,8 @@ async function _retryWithExponentialBackoff(f, {
 // src/util/prepare-retries.ts
 function prepareRetries({
   maxRetries,
-  abortSignal
+  abortSignal,
+  additionalRetryableError
 }) {
   if (maxRetries != null) {
     if (!Number.isInteger(maxRetries)) {
@@ -1867,7 +1872,8 @@ function prepareRetries({
     maxRetries: maxRetriesResult,
     retry: retryWithExponentialBackoffRespectingRetryHeaders({
       maxRetries: maxRetriesResult,
-      abortSignal
+      abortSignal,
+      additionalRetryableError
     })
   };
 }
@@ -6802,6 +6808,12 @@ var DefaultEmbedManyResult = class {
 import {
   withUserAgentSuffix as withUserAgentSuffix5
 } from "@ai-sdk/provider-utils";
+var RetryableNoImageResultError = class extends Error {
+  constructor() {
+    super("No image generated.");
+    this.name = "RetryableNoImageResultError";
+  }
+};
 async function generateImage({
   model: modelArg,
   prompt,
@@ -6823,7 +6835,8 @@ async function generateImage({
   );
   const { retry } = prepareRetries({
     maxRetries: maxRetriesArg,
-    abortSignal
+    abortSignal,
+    additionalRetryableError: (error) => error instanceof RetryableNoImageResultError
   });
   const maxImagesPerCallWithDefault = (_a16 = maxImagesPerCall != null ? maxImagesPerCall : await invokeModelMaxImagesPerCall(model)) != null ? _a16 : 1;
   const callCount = Math.ceil(n / maxImagesPerCallWithDefault);
@@ -6834,22 +6847,38 @@ async function generateImage({
     const remainder = n % maxImagesPerCallWithDefault;
     return remainder === 0 ? maxImagesPerCallWithDefault : remainder;
   });
-  const results = await Promise.all(
-    callImageCounts.map(
-      async (callImageCount) => retry(
-        () => model.doGenerate({
-          prompt,
-          n: callImageCount,
-          abortSignal,
-          headers: headersWithUserAgent,
-          size,
-          aspectRatio,
-          seed,
-          providerOptions: providerOptions != null ? providerOptions : {}
-        })
-      )
-    )
+  const resultGroups = await Promise.all(
+    callImageCounts.map(async (callImageCount) => {
+      const callResults = [];
+      try {
+        await retry(async () => {
+          const result = await model.doGenerate({
+            prompt,
+            n: callImageCount,
+            abortSignal,
+            headers: headersWithUserAgent,
+            size,
+            aspectRatio,
+            seed,
+            providerOptions: providerOptions != null ? providerOptions : {}
+          });
+          callResults.push(result);
+          if (result.images.length === 0 && result.isRetryable !== false) {
+            throw new RetryableNoImageResultError();
+          }
+          return result;
+        });
+        return callResults;
+      } catch (error) {
+        const noImageResultError = error instanceof RetryableNoImageResultError ? error : RetryError.isInstance(error) && error.lastError instanceof RetryableNoImageResultError ? error.lastError : void 0;
+        if (noImageResultError != null) {
+          return callResults;
+        }
+        throw error;
+      }
+    })
   );
+  const results = resultGroups.flat();
   const images = [];
   const warnings = [];
   const responses = [];
@@ -7689,6 +7718,7 @@ function createDownload(options) {
 }
 
 // src/util/data-url.ts
+var { atob } = globalThis;
 function getTextFromDataUrl(dataUrl) {
   const [header, base64Content] = dataUrl.split(",");
   const mediaType = header.split(";")[0].split(":")[1];
@@ -7696,7 +7726,7 @@ function getTextFromDataUrl(dataUrl) {
     throw new Error("Invalid data URL format");
   }
   try {
-    return window.atob(base64Content);
+    return atob(base64Content);
   } catch (error) {
     throw new Error(`Error decoding data URL`);
   }