ai 6.0.279 → 6.0.280

package/dist/index.mjs

@@ -1224,7 +1224,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.279" : "0.0.0-test";
+var VERSION = true ? "6.0.280" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -2775,13 +2775,14 @@ var retryWithExponentialBackoffRespectingRetryHeaders = ({
   maxRetries = 2,
   initialDelayInMs = 2e3,
   backoffFactor = 2,
-  abortSignal
+  abortSignal,
+  additionalRetryableError
 } = {}) => retryWithExponentialBackoff({
   maxRetries,
   initialDelayInMs,
   backoffFactor,
   abortSignal,
-  shouldRetry: (error) => error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true),
+  shouldRetry: async (error) => error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true) || additionalRetryableError != null && await additionalRetryableError(error),
   getDelayInMs: ({ error, exponentialBackoffDelay }) => getRetryDelayInMs({
     error,
     exponentialBackoffDelay
@@ -2792,7 +2793,8 @@ var retryWithExponentialBackoffRespectingRetryHeaders = ({
 // src/util/prepare-retries.ts
 function prepareRetries({
   maxRetries,
-  abortSignal
+  abortSignal,
+  additionalRetryableError
 }) {
   if (maxRetries != null) {
     if (!Number.isInteger(maxRetries)) {
@@ -2815,7 +2817,8 @@ function prepareRetries({
     maxRetries: maxRetriesResult,
     retry: retryWithExponentialBackoffRespectingRetryHeaders({
       maxRetries: maxRetriesResult,
-      abortSignal
+      abortSignal,
+      additionalRetryableError
     })
   };
 }
@@ -10652,6 +10655,12 @@ import {
   convertBase64ToUint8Array as convertBase64ToUint8Array5,
   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: promptArg,
@@ -10673,7 +10682,8 @@ async function generateImage({
   );
   const { retry } = prepareRetries({
     maxRetries: maxRetriesArg,
-    abortSignal
+    abortSignal,
+    additionalRetryableError: (error) => error instanceof RetryableNoImageResultError
   });
   const maxImagesPerCallWithDefault = (_a23 = maxImagesPerCall != null ? maxImagesPerCall : await invokeModelMaxImagesPerCall(model)) != null ? _a23 : 1;
   const callCount = Math.ceil(n / maxImagesPerCallWithDefault);
@@ -10684,25 +10694,41 @@ async function generateImage({
     const remainder = n % maxImagesPerCallWithDefault;
     return remainder === 0 ? maxImagesPerCallWithDefault : remainder;
   });
-  const results = await Promise.all(
-    callImageCounts.map(
-      async (callImageCount) => retry(() => {
-        const { prompt, files, mask } = normalizePrompt(promptArg);
-        return model.doGenerate({
-          prompt,
-          files,
-          mask,
-          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 { prompt, files, mask } = normalizePrompt(promptArg);
+          const result = await model.doGenerate({
+            prompt,
+            files,
+            mask,
+            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 = [];
@@ -11597,6 +11623,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];
@@ -11604,7 +11631,7 @@ function getTextFromDataUrl(dataUrl) {
     throw new Error("Invalid data URL format");
   }
   try {
-    return globalThis.atob(base64Content);
+    return atob(base64Content);
   } catch (error) {
     throw new Error(`Error decoding data URL`);
   }