@shop-os/modelkit 0.2.13 → 0.2.14

package/dist/index.js

@@ -137,7 +137,9 @@ function normalizeExposure(value) {
   if (value === "catalog") return ["cowork", "space_builder"];
   if (value === "internal") return [];
   if (Array.isArray(value)) {
-    if (value.every((entry) => SURFACES.includes(entry))) {
+    if (value.every(
+      (entry) => SURFACES.includes(entry)
+    )) {
       const present = new Set(value);
       return SURFACES.filter((surface) => present.has(surface));
     }
@@ -2678,6 +2680,19 @@ function shouldRetry(err) {
   if (/ECONNRESET|ETIMEDOUT|fetch failed|socket hang up/i.test(err.message)) return true;
   return false;
 }
+function mediaRetryOptions(call, signal, defaultAttempts = 3) {
+  if (call.maxRetries !== void 0 && (!Number.isSafeInteger(call.maxRetries + 1) || call.maxRetries < 0)) {
+    throw new ModelKitError({
+      code: "unsupported_option",
+      message: "maxRetries must be a non-negative safe integer",
+      retryable: false
+    });
+  }
+  return {
+    attempts: call.maxRetries === void 0 ? defaultAttempts : call.maxRetries + 1,
+    ...signal !== void 0 ? { signal } : {}
+  };
+}
 async function withRetry(fn, opts = {}) {
   const attempts = opts.attempts ?? 3;
   const base = opts.baseDelayMs ?? 1e3;
@@ -2685,6 +2700,7 @@ async function withRetry(fn, opts = {}) {
   const factor = opts.factor ?? 2;
   let last;
   for (let attempt = 1; attempt <= attempts; attempt++) {
+    if (opts.signal?.aborted) throw throwAborted();
     try {
       return await fn(attempt);
     } catch (cause) {
@@ -2797,11 +2813,11 @@ async function readBody(res) {
     return text2;
   }
 }
-async function mediaRefToBytes(ref, fetch, options = {}) {
+async function mediaRefToBytes(ref, fetch2, options = {}) {
   if (!isUrlRef(ref)) return { bytes: ref.bytes, mime: ref.mime };
   const init = {};
   if (options.signal !== void 0) init.signal = options.signal;
-  const res = await fetch(ref.url, init);
+  const res = await fetch2(ref.url, init);
   if (!res.ok) {
     const text2 = await res.text().catch(() => "");
     throw classifyHttpError(
@@ -2938,11 +2954,6 @@ var SeedreamResponseSchema = z6.object({
   data: z6.array(SeedreamImageSchema).optional(),
   error: z6.object({ message: z6.string().optional(), code: z6.string().optional() }).optional()
 });
-function retryOptsFor(ctx, attempts = 3) {
-  const opts = { attempts };
-  if (ctx.signal !== void 0) opts.signal = ctx.signal;
-  return opts;
-}
 function fetchInitFor(ctx, init) {
   if (ctx.signal !== void 0) init.signal = ctx.signal;
   return init;
@@ -2980,20 +2991,23 @@ async function generateSeedreamImage(req, ctx) {
   const firstRef = refs[0];
   if (refs.length === 1 && firstRef !== void 0) body.image = firstRef;
   else if (refs.length > 1) body.image = refs;
-  const data = await withRetry(async () => {
-    const res = await arkFetch(ctx, `${arkBaseUrl(ctx)}/images/generations`, {
-      method: "POST",
-      headers: { Authorization: `Bearer ${key}`, "Content-Type": "application/json" },
-      body: JSON.stringify(body)
-    });
-    if (!res.ok) {
-      const text2 = await res.text().catch(() => "");
-      throwArkError(res.status, text2, modelId);
-    }
-    const parsed = SeedreamResponseSchema.safeParse(await res.json());
-    const decoded = parsed.success ? parsed.data : {};
-    return decoded;
-  }, retryOptsFor(ctx));
+  const data = await withRetry(
+    async () => {
+      const res = await arkFetch(ctx, `${arkBaseUrl(ctx)}/images/generations`, {
+        method: "POST",
+        headers: { Authorization: `Bearer ${key}`, "Content-Type": "application/json" },
+        body: JSON.stringify(body)
+      });
+      if (!res.ok) {
+        const text2 = await res.text().catch(() => "");
+        throwArkError(res.status, text2, modelId);
+      }
+      const parsed = SeedreamResponseSchema.safeParse(await res.json());
+      const decoded = parsed.success ? parsed.data : {};
+      return decoded;
+    },
+    mediaRetryOptions(ctx.call, ctx.signal)
+  );
   const first = data.data?.[0];
   let bytes;
   if (first?.b64_json) {
@@ -3129,20 +3143,23 @@ async function generateSeedanceVideo(req, ctx) {
   if (req.generateAudio !== void 0) body.generate_audio = req.generateAudio;
   if (req.seed !== void 0) body.seed = req.seed;
   if (req.watermark !== void 0) body.watermark = req.watermark;
-  const submit = await withRetry(async () => {
-    const res = await arkFetch(ctx, `${arkBaseUrl(ctx)}${SUBMIT_PATH}`, {
-      method: "POST",
-      headers: { Authorization: `Bearer ${key}`, "Content-Type": "application/json" },
-      body: JSON.stringify(body)
-    });
-    if (!res.ok) {
-      const text2 = await res.text().catch(() => "");
-      throwArkError(res.status, text2, modelId);
-    }
-    const parsed = SeedanceResponseSchema.safeParse(await res.json());
-    const decoded = parsed.success ? parsed.data : {};
-    return decoded;
-  }, retryOptsFor(ctx));
+  const submit = await withRetry(
+    async () => {
+      const res = await arkFetch(ctx, `${arkBaseUrl(ctx)}${SUBMIT_PATH}`, {
+        method: "POST",
+        headers: { Authorization: `Bearer ${key}`, "Content-Type": "application/json" },
+        body: JSON.stringify(body)
+      });
+      if (!res.ok) {
+        const text2 = await res.text().catch(() => "");
+        throwArkError(res.status, text2, modelId);
+      }
+      const parsed = SeedanceResponseSchema.safeParse(await res.json());
+      const decoded = parsed.success ? parsed.data : {};
+      return decoded;
+    },
+    mediaRetryOptions(ctx.call, ctx.signal)
+  );
   if (!submit.id) throwArkError(void 0, "Seedance task create returned no id", modelId);
   const jobId = submit.id;
   const usage2 = seedanceUsage(ctx, req, taskType === "edit" ? requestedDuration : duration);
@@ -3151,7 +3168,8 @@ async function generateSeedanceVideo(req, ctx) {
       headers: { Authorization: `Bearer ${key}` }
     });
     if (!res.ok) {
-      if (res.status === 429 || res.status >= 500) return void 0;
+      if (ctx.call.maxRetries !== 0 && (res.status === 429 || res.status >= 500))
+        return void 0;
       const text2 = await res.text().catch(() => "");
       throwArkError(res.status, text2, modelId);
     }
@@ -3217,16 +3235,109 @@ async function generateSeedanceVideo(req, ctx) {
 }
 
 // src/adapters/fal.ts
-import { z as z7 } from "zod";
+import { z as z8 } from "zod";
 import { createFalClient } from "@fal-ai/client";
+
+// src/adapters/fal-http.ts
+import { z as z7 } from "zod";
+var queueUrl = z7.url().refine((value) => new URL(value).origin === "https://queue.fal.run");
+var submissionSchema = z7.object({
+  request_id: z7.string(),
+  status_url: queueUrl,
+  response_url: queueUrl
+});
+var statusSchema = z7.object({ status: z7.string() });
+function createFalHttpClient(opts) {
+  const jobs = /* @__PURE__ */ new Map();
+  async function request(url, input, signal) {
+    let response;
+    try {
+      response = await opts.fetch(url, {
+        method: input === void 0 ? "GET" : "POST",
+        headers: {
+          Authorization: `Key ${opts.credentials}`,
+          "Content-Type": "application/json"
+        },
+        ...input !== void 0 ? { body: JSON.stringify(input) } : {},
+        ...signal !== void 0 ? { signal } : {}
+      });
+    } catch (cause) {
+      throw classifyThrown("fal", cause);
+    }
+    if (!response.ok) throw classifyHttpError(response.status, await response.text(), "fal");
+    return JsonValueSchema.parse(await response.json());
+  }
+  function job(id) {
+    const saved = jobs.get(id);
+    if (!saved)
+      throw new ModelKitError({
+        code: "provider_rejected",
+        provider: "fal",
+        message: "Unknown fal queue request"
+      });
+    return saved;
+  }
+  const queue = {
+    async submit(endpoint, options) {
+      const submitted = submissionSchema.parse(
+        await request(
+          `https://queue.fal.run/${endpoint}`,
+          options.input,
+          options.abortSignal
+        )
+      );
+      jobs.set(submitted.request_id, submitted);
+      return { requestId: submitted.request_id };
+    },
+    async status(_endpoint, options) {
+      return statusSchema.parse(
+        await request(job(options.requestId).status_url, void 0, options.abortSignal)
+      );
+    },
+    result: (_endpoint, options) => request(job(options.requestId).response_url, void 0, options.abortSignal)
+  };
+  return {
+    queue,
+    async subscribe(endpoint, options) {
+      const submitted = await withRetry(
+        () => queue.submit(endpoint, options),
+        mediaRetryOptions(opts, options.abortSignal)
+      );
+      return pollUntil(
+        async () => {
+          const readOptions = {
+            requestId: submitted.requestId,
+            ...options.abortSignal !== void 0 ? { abortSignal: options.abortSignal } : {}
+          };
+          const current = await queue.status(endpoint, readOptions);
+          if (current.status !== "COMPLETED") return void 0;
+          return queue.result(endpoint, readOptions);
+        },
+        {
+          intervalMs: 1e3,
+          timeoutMs: 18e4,
+          ...options.abortSignal !== void 0 ? { signal: options.abortSignal } : {}
+        }
+      );
+    }
+  };
+}
+
+// src/adapters/fal.ts
 var MUSE_T2I_ENDPOINT = "meta/muse-image/text-to-image";
 var MUSE_EDIT_ENDPOINT = "meta/muse-image/edit";
 var KLING_O1_ENDPOINT = "fal-ai/kling-video/o1/video-to-video/reference";
 var FAL_QUEUE_HOST = "https://queue.fal.run";
 var PASSTHROUGH_POLL_MS = 2e3;
 var PASSTHROUGH_MAX_WAIT_MS = 18e4;
-var falSubscribePayloadSchema = z7.object({ data: JsonValueSchema });
+var falSubscribePayloadSchema = z8.object({ data: JsonValueSchema });
 function defaultCreateFalClient(opts) {
+  if (opts.maxRetries !== void 0)
+    return createFalHttpClient({
+      credentials: opts.credentials,
+      fetch: opts.fetch ?? fetch,
+      maxRetries: opts.maxRetries
+    });
   const client2 = createFalClient(opts);
   return {
     async subscribe(endpoint, options) {
@@ -3275,7 +3386,7 @@ function createFalAdapter(deps = defaultDeps) {
   };
 }
 var falAdapter = createFalAdapter();
-var falDetailSchema = z7.object({ detail: JsonValueSchema });
+var falDetailSchema = z8.object({ detail: JsonValueSchema });
 function liftDetail(body) {
   const parsed = falDetailSchema.safeParse(body);
   const detail = parsed.success ? parsed.data.detail : void 0;
@@ -3284,11 +3395,11 @@ function liftDetail(body) {
 function falError(status, body, model) {
   return classifyHttpError(status, liftDetail(body), "fal", model);
 }
-var falSdkThrownSchema = z7.object({
-  status: z7.number().optional(),
-  statusCode: z7.number().optional(),
+var falSdkThrownSchema = z8.object({
+  status: z8.number().optional(),
+  statusCode: z8.number().optional(),
   body: JsonValueSchema.optional(),
-  message: z7.string().optional()
+  message: z8.string().optional()
 });
 function falSdkError(cause, model) {
   const parsed = falSdkThrownSchema.safeParse(cause);
@@ -3321,6 +3432,8 @@ function gatewayConfig(ctx) {
   };
   if (falKey) config.falKey = falKey;
   if (cfToken) config.cfToken = cfToken;
+  if (ctx.call.maxRetries !== void 0) config.singleAttempt = true;
+  if (ctx.signal !== void 0) config.signal = ctx.signal;
   return config;
 }
 async function gatewayFetch(cfg, fetchImpl, targetUrl, init) {
@@ -3328,16 +3441,18 @@ async function gatewayFetch(cfg, fetchImpl, targetUrl, init) {
   if (cfg.falKey) headers2.Authorization = `Bearer ${cfg.falKey}`;
   if (init.body) headers2["Content-Type"] = "application/json";
   if (cfg.cfToken) headers2["cf-aig-authorization"] = `Bearer ${cfg.cfToken}`;
+  if (cfg.singleAttempt) headers2["cf-aig-max-attempts"] = "1";
   const request = { method: init.method, headers: headers2 };
+  if (cfg.signal !== void 0) request.signal = cfg.signal;
   if (init.body !== void 0) request.body = init.body;
   return fetchImpl(cfg.baseUrl, request);
 }
-var falQueueSubmitSchema = z7.object({
-  request_id: z7.string(),
-  status_url: z7.string(),
-  response_url: z7.string()
+var falQueueSubmitSchema = z8.object({
+  request_id: z8.string(),
+  status_url: z8.string(),
+  response_url: z8.string()
 });
-var falQueueStatusSchema = z7.object({ status: z7.string() });
+var falQueueStatusSchema = z8.object({ status: z8.string() });
 async function passthroughRequest(ctx, endpoint, input) {
   const cfg = gatewayConfig(ctx);
   const submitRes = await gatewayFetch(cfg, ctx.fetch, `${FAL_QUEUE_HOST}/${endpoint}`, {
@@ -3394,7 +3509,11 @@ function falClientOf(ctx, deps) {
       retryable: false
     });
   }
-  return deps.createFalClient({ credentials: falKey, fetch: ctx.fetch });
+  return deps.createFalClient({
+    credentials: falKey,
+    fetch: ctx.fetch,
+    ...ctx.call.maxRetries !== void 0 ? { maxRetries: ctx.call.maxRetries } : {}
+  });
 }
 async function subscribe(ctx, deps, endpoint, input) {
   const client2 = falClientOf(ctx, deps);
@@ -3411,7 +3530,10 @@ async function subscribe(ctx, deps, endpoint, input) {
           throw falSdkError(cause, ctx.model.id);
         }
       },
-      ctx.signal !== void 0 ? { signal: ctx.signal } : {}
+      mediaRetryOptions(
+        ctx.call.maxRetries === void 0 ? {} : { maxRetries: 0 },
+        ctx.signal
+      )
     );
   } catch (cause) {
     throw falSdkError(cause, ctx.model.id);
@@ -3463,11 +3585,11 @@ async function decodeImage(ctx, url, contentType) {
   const mime = res.headers.get("content-type") ?? contentType ?? "image/png";
   return { bytes: new Uint8Array(await res.arrayBuffer()), mime };
 }
-var museImageResultSchema = z7.object({
-  images: z7.array(
-    z7.object({
-      url: z7.string().optional(),
-      content_type: z7.string().nullable().optional()
+var museImageResultSchema = z8.object({
+  images: z8.array(
+    z8.object({
+      url: z8.string().optional(),
+      content_type: z8.string().nullable().optional()
     })
   ).optional()
 });
@@ -3571,8 +3693,8 @@ function klingO1Input(req) {
   if (refs.length > 0) input.image_urls = refs;
   return input;
 }
-var falVideoResultSchema = z7.object({
-  video: z7.object({ url: z7.string().optional() }).optional()
+var falVideoResultSchema = z8.object({
+  video: z8.object({ url: z8.string().optional() }).optional()
 });
 function extractVideoUrl(result, endpoint, model) {
   const parsed = falVideoResultSchema.safeParse(result);
@@ -3652,7 +3774,7 @@ async function submitQueueJob(ctx, req, deps, endpoint, input) {
           throw falSdkError(cause, ctx.model.id);
         }
       },
-      ctx.signal !== void 0 ? { signal: ctx.signal } : {}
+      mediaRetryOptions(ctx.call, ctx.signal)
     );
   } catch (cause) {
     throw falSdkError(cause, ctx.model.id);
@@ -3681,6 +3803,7 @@ function queuedVideoJob(ctx, req, client2, endpoint, requestId) {
         if (status === "IN_QUEUE") return { status: "pending" };
         return { status: "running" };
       } catch (cause) {
+        if (ctx.call.maxRetries === 0) throw cause;
         if (isModelKitError(cause) && !cause.retryable)
           return { status: "failed" };
         return { status: "pending" };
@@ -3704,7 +3827,7 @@ function queuedVideoJob(ctx, req, client2, endpoint, requestId) {
           return await client2.queue.result(endpoint, queueOptions);
         } catch (cause) {
           const err = falSdkError(cause, ctx.model.id);
-          if (err.retryable) return void 0;
+          if (ctx.call.maxRetries !== 0 && err.retryable) return void 0;
           throw err;
         }
       }, pollOptions);
@@ -3752,6 +3875,7 @@ async function passthroughVideoJob(ctx, req, endpoint, input) {
         const url = await checkStatus();
         return url ? { status: "succeeded" } : { status: "pending" };
       } catch (cause) {
+        if (ctx.call.maxRetries === 0) throw cause;
         if (isModelKitError(cause) && !cause.retryable)
           return { status: "failed" };
         return { status: "pending" };
@@ -3771,13 +3895,13 @@ async function passthroughVideoJob(ctx, req, endpoint, input) {
 }
 
 // src/adapters/google-genai.ts
-import { z as z9 } from "zod";
+import { z as z10 } from "zod";
 import {
   GoogleGenAI
 } from "@google/genai";
 
 // src/adapters/google-files.ts
-import { z as z8 } from "zod";
+import { z as z9 } from "zod";
 var GOOGLE_FILES_BASE = "https://generativelanguage.googleapis.com";
 var GOOGLE_FILES_POLL_INTERVAL_MS = 2e3;
 var GOOGLE_FILES_POLL_TIMEOUT_MS = 5 * 60 * 1e3;
@@ -3787,17 +3911,17 @@ var GENERIC_MEDIA_TYPES = /* @__PURE__ */ new Set([
   "binary/octet-stream",
   "application/binary"
 ]);
-var uploadedFileSchema = z8.object({
-  file: z8.object({
-    uri: z8.string().optional(),
-    name: z8.string().optional(),
-    state: z8.string().optional(),
-    mimeType: z8.string().optional()
+var uploadedFileSchema = z9.object({
+  file: z9.object({
+    uri: z9.string().optional(),
+    name: z9.string().optional(),
+    state: z9.string().optional(),
+    mimeType: z9.string().optional()
   }).optional()
 });
-var fileStateSchema = z8.object({
-  state: z8.string().optional(),
-  error: z8.object({ message: z8.string().optional() }).optional()
+var fileStateSchema = z9.object({
+  state: z9.string().optional(),
+  error: z9.object({ message: z9.string().optional() }).optional()
 });
 function resolveMime(declared, fallback) {
   const clean = declared?.split(";")[0]?.trim().toLowerCase();
@@ -3931,7 +4055,7 @@ async function waitForActive(fileName, opts) {
       }
       return void 0;
     }
-    if (res.status === 429 || res.status >= 500) return void 0;
+    if (opts.maxRetries !== 0 && (res.status === 429 || res.status >= 500)) return void 0;
     const text2 = await res.text().catch(() => "");
     throw classifyHttpError(res.status, text2, "google-genai", opts.model);
   }, pollOptions);
@@ -3955,8 +4079,7 @@ async function uploadGoogleFile(source, opts) {
     opts.model
   );
   if (bytes.length > GOOGLE_FILES_MAX_BYTES) throw oversizeError(bytes.length, label, opts.model);
-  const retryOptions = { attempts: 3 };
-  if (opts.signal !== void 0) retryOptions.signal = opts.signal;
+  const retryOptions = mediaRetryOptions(opts, opts.signal);
   const uploaded = await withRetry(() => startAndUpload(bytes, mimeType, opts), retryOptions);
   if (uploaded.state !== "ACTIVE") {
     try {
@@ -3971,7 +4094,12 @@ async function uploadGoogleFile(source, opts) {
 
 // src/adapters/google-genai.ts
 function defaultCreateGenAIClient(opts) {
-  const sdk = new GoogleGenAI({ ...opts, vertexai: false });
+  const sdk = new GoogleGenAI({
+    apiKey: opts.apiKey,
+    vertexai: false,
+    ...opts.maxRetries !== void 0 ? { httpOptions: { retryOptions: { attempts: 1 } } } : {}
+  });
+  const interactionOptions = opts.maxRetries === void 0 ? void 0 : { maxRetries: 0, ...opts.signal !== void 0 ? { signal: opts.signal } : {} };
   return {
     models: {
       generateVideos: async (body) => {
@@ -3992,11 +4120,11 @@ function defaultCreateGenAIClient(opts) {
     interactions: {
       create: async (args) => {
         const params = args;
-        const response = await sdk.interactions.create(params);
+        const response = await sdk.interactions.create(params, interactionOptions);
         return response;
       },
       get: async (jobId, params) => {
-        const response = await sdk.interactions.get(jobId, params);
+        const response = await sdk.interactions.get(jobId, params, interactionOptions);
         return response;
       }
     }
@@ -4010,7 +4138,7 @@ var OMNI_POLL_INTERVAL_MS = 1e4;
 var RESULT_TIMEOUT_MS2 = 15 * 60 * 1e3;
 var OMNI_INLINE_LIMIT_BYTES = 4 * 1024 * 1024;
 var MEDIA_PART_INLINE_LIMIT_BYTES = 20 * 1024 * 1024;
-var errorTextSchema = z9.string();
+var errorTextSchema = z10.string();
 function isGoogleGenaiVideoKnown(providerModelId) {
   return providerModelId.startsWith("veo-") || providerModelId.startsWith("gemini-omni");
 }
@@ -4165,7 +4293,11 @@ async function generateVeoVideo(req, ctx, createGenAIClient) {
       referenceType: "ASSET"
     }));
   }
-  const client2 = createGenAIClient({ apiKey });
+  const client2 = createGenAIClient({
+    apiKey,
+    ...ctx.call.maxRetries !== void 0 ? { maxRetries: ctx.call.maxRetries } : {},
+    ...ctx.signal !== void 0 ? { signal: ctx.signal } : {}
+  });
   const config = { durationSeconds: duration, personGeneration };
   if (req.aspectRatio !== void 0) config.aspectRatio = req.aspectRatio;
   if (req.resolution !== void 0) config.resolution = req.resolution;
@@ -4178,8 +4310,7 @@ async function generateVeoVideo(req, ctx, createGenAIClient) {
     config
   };
   if (image !== void 0) requestBody.image = image;
-  const retryOptions = { attempts: 3 };
-  if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+  const retryOptions = mediaRetryOptions(ctx.call, ctx.signal);
   let operation;
   try {
     operation = await withRetry(async () => {
@@ -4292,7 +4423,11 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
     if (omniResolution !== void 0) responseFormat.resolution = omniResolution;
   }
   if (req.durationSeconds !== void 0) responseFormat.duration = `${req.durationSeconds}s`;
-  const client2 = createGenAIClient({ apiKey });
+  const client2 = createGenAIClient({
+    apiKey,
+    ...ctx.call.maxRetries !== void 0 ? { maxRetries: ctx.call.maxRetries } : {},
+    ...ctx.signal !== void 0 ? { signal: ctx.signal } : {}
+  });
   const uploadedByUri = /* @__PURE__ */ new Map();
   const uploadedNames = [];
   const uploadOne = async (kind, bytes, mime) => {
@@ -4303,6 +4438,7 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
       label: "gemini-omni"
     };
     if (ctx.signal !== void 0) uploadOptions.signal = ctx.signal;
+    if (ctx.call.maxRetries !== void 0) uploadOptions.maxRetries = ctx.call.maxRetries;
     const file = await uploadGoogleFile({ bytes, mime }, uploadOptions);
     uploadedByUri.set(file.fileUri, { type: kind, bytes, mime: file.mimeType });
     uploadedNames.push(file.fileName);
@@ -4318,8 +4454,7 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
     generation_config: { video_config: { task } },
     store: true
   });
-  const createRetryOptions = { attempts: 3 };
-  if (ctx.signal !== void 0) createRetryOptions.signal = ctx.signal;
+  const createRetryOptions = mediaRetryOptions(ctx.call, ctx.signal);
   let interaction;
   try {
     const parts = [];
@@ -4389,7 +4524,7 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
         }
       }, createRetryOptions);
     } catch (cause) {
-      if (uploadedByUri.size === 0 || !isFileReferenceRejected(cause))
+      if (ctx.call.maxRetries !== void 0 || uploadedByUri.size === 0 || !isFileReferenceRejected(cause))
         throwSdkError(cause, modelId);
       const inlineParts = parts.map((part) => {
         if (part.uri === void 0) return part;
@@ -4499,9 +4634,12 @@ async function generateLyriaAudio(req, ctx, createGenAIClient) {
     const image = await resolveMedia(ref, ctx.fetch, ctx.signal, "image/jpeg", modelId);
     parts.push({ inlineData: { mimeType: image.mime, data: toBase64(image.bytes) } });
   }
-  const client2 = createGenAIClient({ apiKey });
-  const retryOptions = { attempts: 3 };
-  if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+  const client2 = createGenAIClient({
+    apiKey,
+    ...ctx.call.maxRetries !== void 0 ? { maxRetries: ctx.call.maxRetries } : {},
+    ...ctx.signal !== void 0 ? { signal: ctx.signal } : {}
+  });
+  const retryOptions = mediaRetryOptions(ctx.call, ctx.signal);
   let response;
   try {
     response = await withRetry(async () => {
@@ -4563,6 +4701,7 @@ async function prepareGoogleMediaPart(ref, mediaType, ctx) {
     label: "media-part"
   };
   if (ctx.signal !== void 0) uploadOptions.signal = ctx.signal;
+  if (ctx.call.maxRetries !== void 0) uploadOptions.maxRetries = ctx.call.maxRetries;
   const file = await uploadGoogleFile({ bytes: ref.bytes, mime: mediaType }, uploadOptions);
   return {
     type: "file",
@@ -4575,15 +4714,15 @@ async function prepareGoogleMediaPart(ref, mediaType, ctx) {
 // src/adapters/google-image.ts
 import { createGoogleGenerativeAI as createGoogleGenerativeAI2 } from "@ai-sdk/google";
 import { generateText } from "ai";
-import { z as z10 } from "zod";
+import { z as z11 } from "zod";
 var GEMINI_IMAGE_TIMEOUT_MS = 6e5;
 var defaultGoogleImageDeps = {
   createGoogleClient: (opts) => createGoogleGenerativeAI2(opts),
   runTextGeneration: (opts) => generateText(opts)
 };
-var causeChainNodeSchema = z10.object({
-  message: z10.string().optional(),
-  cause: z10.unknown().optional()
+var causeChainNodeSchema = z11.object({
+  message: z11.string().optional(),
+  cause: z11.unknown().optional()
 });
 function extractBlockReason(cause) {
   let current = cause;
@@ -4651,8 +4790,7 @@ function createGoogleImageAdapter(deps = defaultGoogleImageDeps) {
         responseModalities: ["IMAGE"]
       };
       if (req.safetySettings !== void 0) googleOptions.safetySettings = req.safetySettings;
-      const retryOptions = {};
-      if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+      const retryOptions = mediaRetryOptions(ctx.call, ctx.signal);
       const result = await withRetry(async () => {
         const timeout = AbortSignal.timeout(GEMINI_IMAGE_TIMEOUT_MS);
         const signal = ctx.signal !== void 0 ? AbortSignal.any([ctx.signal, timeout]) : timeout;
@@ -4704,7 +4842,7 @@ function createGoogleImageAdapter(deps = defaultGoogleImageDeps) {
 var googleImageAdapter = createGoogleImageAdapter();
 
 // src/adapters/higgsfield.ts
-import { z as z11 } from "zod";
+import { z as z12 } from "zod";
 
 // src/adapters/higgsfield-motions.ts
 var MOTION_MAP = {
@@ -4903,14 +5041,14 @@ function videoUsage3(ctx, req, duration) {
   if (req.durationSeconds !== void 0) event.durationSeconds = req.durationSeconds;
   return event;
 }
-var klingCreateSchema = z11.object({ status_url: z11.string() });
-var klingStatusSchema = z11.object({
-  status: z11.string().optional(),
-  video: z11.object({ url: z11.string().optional() }).optional(),
+var klingCreateSchema = z12.object({ status_url: z12.string() });
+var klingStatusSchema = z12.object({
+  status: z12.string().optional(),
+  video: z12.object({ url: z12.string().optional() }).optional(),
   error: JsonValueSchema.optional()
 });
-var errorMessageSchema = z11.object({ message: z11.string().optional() });
-var textValueSchema = z11.string();
+var errorMessageSchema = z12.object({ message: z12.string().optional() });
+var textValueSchema = z12.string();
 function klingErrorDetail(error) {
   if (error === void 0) return "unknown_error";
   const asText = textValueSchema.safeParse(error);
@@ -4959,8 +5097,7 @@ async function generateKlingVideo(req, ctx) {
     body.aspect_ratio = req.aspectRatio;
   if (req.lastFrame) body.last_image_url = refToUrl(req.lastFrame);
   const endpoint = klingEndpoint(ctx.model.providerModelId, ctx.model.id);
-  const retryOptions = {};
-  if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+  const retryOptions = mediaRetryOptions(ctx.call, ctx.signal);
   const createRes = await withRetry(
     () => createRequest(ctx, endpoint, { method: "POST", body: JSON.stringify(body) }),
     retryOptions
@@ -4980,10 +5117,12 @@ async function generateKlingVideo(req, ctx) {
     let res;
     try {
       res = await ctx.fetch(statusUrl, requestInit(ctx, { method: "GET" }));
-    } catch {
+    } catch (cause) {
+      if (ctx.call.maxRetries === 0) throw classifyThrown("higgsfield", cause, ctx.model.id);
       return void 0;
     }
-    if (res.status === 429 || res.status >= 500) return void 0;
+    if (ctx.call.maxRetries !== 0 && (res.status === 429 || res.status >= 500))
+      return void 0;
     if (!res.ok) {
       throw classifyHttpError(res.status, await readBody(res), "higgsfield", ctx.model.id);
     }
@@ -5025,6 +5164,7 @@ async function generateKlingVideo(req, ctx) {
         const url = await check();
         return url ? { status: "succeeded" } : { status: "pending" };
       } catch (cause) {
+        if (ctx.call.maxRetries === 0) throw cause;
         if (isModelKitError(cause) && !cause.retryable) return { status: "failed" };
         throw cause;
       }
@@ -5044,8 +5184,8 @@ async function generateKlingVideo(req, ctx) {
     }
   };
 }
-var dopMotionOptionsSchema = z11.object({
-  modelkit: z11.object({ motion: z11.string().optional() }).optional()
+var dopMotionOptionsSchema = z12.object({
+  modelkit: z12.object({ motion: z12.string().optional() }).optional()
 });
 function dopMotionId(ctx) {
   const parsed = dopMotionOptionsSchema.safeParse(ctx.model.providerOptions);
@@ -5061,7 +5201,7 @@ function dopMotionId(ctx) {
     });
   return id;
 }
-var jsonRecordSchema = z11.record(z11.string(), JsonValueSchema);
+var jsonRecordSchema = z12.record(z12.string(), JsonValueSchema);
 var DOP_URL_KEYS = ["url", "video_url", "output_url", "result_url", "download_url"];
 var DOP_NESTED_KEYS = ["video", "result", "results"];
 function dopUrl(body) {
@@ -5089,8 +5229,8 @@ function dopUrl(body) {
   }
   return void 0;
 }
-var dopCreateSchema = z11.object({ id: z11.string() });
-var dopStatusSchema = z11.object({ status: z11.string().optional() });
+var dopCreateSchema = z12.object({ id: z12.string() });
+var dopStatusSchema = z12.object({ status: z12.string().optional() });
 async function generateDopVideo(req, ctx) {
   const firstFrame = req.firstFrame ? refToUrl(req.firstFrame) : void 0;
   if (!firstFrame) {
@@ -5115,8 +5255,7 @@ async function generateDopVideo(req, ctx) {
     params.input_images_end = [{ type: "image_url", image_url: refToUrl(req.lastFrame) }];
   }
   if (req.seed !== void 0) params.seed = req.seed;
-  const retryOptions = {};
-  if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+  const retryOptions = mediaRetryOptions(ctx.call, ctx.signal);
   const createRes = await withRetry(
     () => createRequest(ctx, DOP_ENDPOINT, {
       method: "POST",
@@ -5140,10 +5279,18 @@ async function generateDopVideo(req, ctx) {
     let res;
     try {
       res = await ctx.fetch(statusUrl, requestInit(ctx, { method: "GET" }));
-    } catch {
+    } catch (cause) {
+      if (ctx.call.maxRetries === 0) throw classifyThrown("higgsfield", cause, ctx.model.id);
       return void 0;
     }
     if (!res.ok) {
+      if (ctx.call.maxRetries === 0)
+        throw classifyHttpError(
+          res.status,
+          await readBody(res),
+          "higgsfield",
+          ctx.model.id
+        );
       return void 0;
     }
     const payload = JsonValueSchema.parse(await res.json());
@@ -5180,6 +5327,7 @@ async function generateDopVideo(req, ctx) {
         const url = await check();
         return url ? { status: "succeeded" } : { status: "pending" };
       } catch (cause) {
+        if (ctx.call.maxRetries === 0) throw cause;
         if (isModelKitError(cause) && !cause.retryable) return { status: "failed" };
         throw cause;
       }
@@ -5210,7 +5358,7 @@ var higgsfieldAdapter = {
 
 // src/adapters/openai-image.ts
 import OpenAI from "openai";
-import { z as z12 } from "zod";
+import { z as z13 } from "zod";
 var MAX_EDIT_REFERENCES = 16;
 function aspectRatioToSize(ratio, tier) {
   const TIER_LONG_EDGE = {
@@ -5293,9 +5441,9 @@ function toProviderError2(cause, ctx) {
     ctx.model.id
   );
 }
-var openaiUsageSchema = z12.object({
-  input_tokens: z12.number().optional(),
-  output_tokens: z12.number().optional()
+var openaiUsageSchema = z13.object({
+  input_tokens: z13.number().optional(),
+  output_tokens: z13.number().optional()
 });
 var openaiImageAdapter = {
   id: "openai-image",
@@ -5337,8 +5485,7 @@ var openaiImageAdapter = {
     }
     const format = outputFormat !== void 0 ? { output_format: outputFormat } : {};
     const mime = outputFormat === "jpeg" ? "image/jpeg" : "image/png";
-    const retryOptions = {};
-    if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+    const retryOptions = mediaRetryOptions(ctx.call, ctx.signal);
     const response = await withRetry(async () => {
       try {
         if (!isEdit) {
@@ -5658,13 +5805,7 @@ async function generateProdiaImage(req, ctx) {
       })
     );
   });
-  const retryOpts = {
-    attempts: 3,
-    baseDelayMs: 1e3,
-    factor: 2,
-    maxDelayMs: 1e4
-  };
-  if (ctx.signal !== void 0) retryOpts.signal = ctx.signal;
+  const retryOpts = mediaRetryOptions(ctx.call, ctx.signal);
   let res;
   try {
     res = await withRetry(async () => {
@@ -5711,7 +5852,66 @@ var prodiaAdapter = {
 
 // src/adapters/replicate.ts
 import Replicate from "replicate";
-import { z as z13 } from "zod";
+
+// src/adapters/replicate-http.ts
+import { z as z14 } from "zod";
+var predictionSchema = z14.object({
+  id: z14.string(),
+  status: z14.string(),
+  output: JsonValueSchema.optional(),
+  error: JsonValueSchema.optional(),
+  metrics: z14.object({ predict_time: z14.number().optional() }).optional()
+});
+function createReplicateHttpClient(opts) {
+  async function request(path, body, signal) {
+    let response;
+    try {
+      response = await opts.fetch(`https://api.replicate.com/v1${path}`, {
+        method: body === void 0 ? "GET" : "POST",
+        headers: {
+          Authorization: `Bearer ${opts.auth}`,
+          "Content-Type": "application/json"
+        },
+        ...body !== void 0 ? { body: JSON.stringify(body) } : {},
+        ...signal !== void 0 ? { signal } : {}
+      });
+    } catch (cause) {
+      throw classifyThrown("replicate", cause);
+    }
+    if (!response.ok)
+      throw classifyHttpError(response.status, await response.text(), "replicate");
+    const parsed = predictionSchema.parse(await response.json());
+    return {
+      id: parsed.id,
+      status: parsed.status,
+      ...parsed.output !== void 0 ? { output: parsed.output } : {},
+      ...parsed.error !== void 0 ? { error: parsed.error } : {},
+      ...parsed.metrics?.predict_time !== void 0 ? { metrics: { predict_time: parsed.metrics.predict_time } } : {}
+    };
+  }
+  return {
+    predictions: {
+      create(options) {
+        if (options.version) {
+          return request(
+            "/predictions",
+            { input: options.input, version: options.version },
+            options.signal
+          );
+        }
+        return request(
+          `/models/${options.model}/predictions`,
+          { input: options.input },
+          options.signal
+        );
+      },
+      get: (id, options) => request(`/predictions/${encodeURIComponent(id)}`, void 0, options?.signal)
+    }
+  };
+}
+
+// src/adapters/replicate.ts
+import { z as z15 } from "zod";
 
 // src/adapters/input-spec-mapper.ts
 function canonicalValue(req, key) {
@@ -5750,7 +5950,7 @@ function applyInputSpec(spec, req, refToUrl2, opts) {
 var POLL_INTERVAL_MS = 1e3;
 var POLL_TIMEOUT_MS = 15 * 6e4;
 var defaultReplicateDeps = {
-  createReplicateClient: (opts) => new Replicate(opts)
+  createReplicateClient: (opts) => opts.maxRetries === void 0 ? new Replicate(opts) : createReplicateHttpClient(opts)
 };
 function createReplicateAdapter(deps = defaultReplicateDeps) {
   return {
@@ -5772,7 +5972,11 @@ function client(ctx, deps) {
       retryable: false
     });
   }
-  return deps.createReplicateClient({ auth: token, fetch: ctx.fetch });
+  return deps.createReplicateClient({
+    auth: token,
+    fetch: ctx.fetch,
+    ...ctx.call.maxRetries !== void 0 ? { maxRetries: ctx.call.maxRetries } : {}
+  });
 }
 function route(providerModelId) {
   if (providerModelId.includes(":")) {
@@ -5782,8 +5986,8 @@ function route(providerModelId) {
   if (!providerModelId.includes("/")) return { version: providerModelId };
   return { model: providerModelId };
 }
-var replicateResponseStatusSchema = z13.object({
-  response: z13.object({ status: z13.number() }).partial()
+var replicateResponseStatusSchema = z15.object({
+  response: z15.object({ status: z15.number() }).partial()
 });
 function replicateError(cause, ctx) {
   if (isModelKitError(cause)) return cause;
@@ -5810,7 +6014,7 @@ function isUrlFn(value) {
 function extractOutputUrl(output) {
   const root = Array.isArray(output) ? output[0] : output;
   if (root === void 0) throw noOutput();
-  const asString = z13.string().safeParse(root);
+  const asString = z15.string().safeParse(root);
   if (asString.success && asString.data) return asString.data;
   if (root instanceof URL) return root.href;
   if (isObjectOutput(root)) {
@@ -5819,15 +6023,15 @@ function extractOutputUrl(output) {
       if (urlValue !== void 0 && isUrlFn(urlValue)) {
         const called = urlValue();
         if (called instanceof URL) return called.href;
-        const calledString = z13.string().safeParse(called);
+        const calledString = z15.string().safeParse(called);
         if (calledString.success && calledString.data) return calledString.data;
       } else {
-        const direct = z13.string().safeParse(urlValue);
+        const direct = z15.string().safeParse(urlValue);
         if (direct.success && direct.data) return direct.data;
       }
     }
     if ("default" in root) {
-      const fallback = z13.string().safeParse(root.default);
+      const fallback = z15.string().safeParse(root.default);
       if (fallback.success && fallback.data) return fallback.data;
     }
   }
@@ -5857,7 +6061,7 @@ async function runPrediction(ctx, input, deps) {
           throw replicateError(cause, ctx);
         }
       },
-      { attempts: 1 }
+      mediaRetryOptions(ctx.call, ctx.signal, 1)
     );
   } catch (cause) {
     throw replicateError(cause, ctx);
@@ -5888,7 +6092,7 @@ async function runPrediction(ctx, input, deps) {
     }
     return void 0;
   }, pollOpts);
-  const predictTime = z13.number().safeParse(done.metrics?.predict_time);
+  const predictTime = z15.number().safeParse(done.metrics?.predict_time);
   if (done.output === void 0) throw noOutput();
   return {
     url: extractOutputUrl(done.output),
@@ -6256,7 +6460,7 @@ async function generateReplicateVideo(req, ctx, deps) {
           throw replicateError(cause, ctx);
         }
       },
-      { attempts: 1 }
+      mediaRetryOptions(ctx.call, ctx.signal, 1)
     );
   } catch (cause) {
     throw replicateError(cause, ctx);
@@ -6273,7 +6477,7 @@ async function generateReplicateVideo(req, ctx, deps) {
       throw replicateError(cause, ctx);
     }
     if (current.status === "succeeded") {
-      const predictTime = z13.number().safeParse(current.metrics?.predict_time);
+      const predictTime = z15.number().safeParse(current.metrics?.predict_time);
       if (current.output === void 0) throw noOutput();
       return {
         url: extractOutputUrl(current.output),
@@ -6414,7 +6618,7 @@ async function generateReplicateAudio(req, ctx, deps) {
 // src/adapters/xai-video.ts
 import { experimental_generateVideo } from "ai";
 import { createXai as createXai2 } from "@ai-sdk/xai";
-import { z as z14 } from "zod";
+import { z as z16 } from "zod";
 var POLL_INTERVAL_MS2 = 1e4;
 var POLL_TIMEOUT_MS2 = 15 * 6e4;
 var defaultXaiDeps = {
@@ -6427,8 +6631,8 @@ function createXaiVideoAdapter(deps = defaultXaiDeps) {
   };
 }
 var xaiVideoAdapter = createXaiVideoAdapter();
-var xaiResponseStatusSchema = z14.object({
-  response: z14.object({ status: z14.number() }).partial()
+var xaiResponseStatusSchema = z16.object({
+  response: z16.object({ status: z16.number() }).partial()
 });
 function xaiError(cause, ctx) {
   const info = describeThrown(cause);
@@ -6436,12 +6640,12 @@ function xaiError(cause, ctx) {
   const status = info.status ?? (nested.success ? nested.data.response?.status : void 0);
   return classifyThrown("xai-video", cause, ctx.model.id, { status, body: info.body });
 }
-var xaiMetaSchema = z14.object({
+var xaiMetaSchema = z16.object({
   videoUrl: JsonValueSchema.optional(),
   costInUsdTicks: JsonValueSchema.optional(),
   duration: JsonValueSchema.optional()
 });
-var xaiProviderMetadataSchema = z14.object({ xai: xaiMetaSchema });
+var xaiProviderMetadataSchema = z16.object({ xai: xaiMetaSchema });
 function metaOf(result) {
   const parsed = xaiProviderMetadataSchema.partial().safeParse(result.providerMetadata);
   if (!parsed.success) return {};
@@ -6492,6 +6696,7 @@ async function generateXaiVideo(req, ctx, deps) {
   }
   if (req.durationSeconds !== void 0) callBody.duration = req.durationSeconds;
   if (ctx.signal) callBody.abortSignal = ctx.signal;
+  if (ctx.call.maxRetries !== void 0) callBody.maxRetries = ctx.call.maxRetries;
   const pending = (async () => {
     try {
       return await client2.generateVideo(callBody);
@@ -6548,7 +6753,7 @@ async function mapResult(ctx, req, result) {
     };
   }
   const meta = metaOf(result);
-  const videoUrl = z14.string().safeParse(meta.videoUrl);
+  const videoUrl = z16.string().safeParse(meta.videoUrl);
   if (videoUrl.success && videoUrl.data) {
     const res = await ctx.fetch(videoUrl.data);
     if (!res.ok) {
@@ -6571,7 +6776,7 @@ async function mapResult(ctx, req, result) {
   });
 }
 function usage(ctx, req, meta) {
-  const costTicks = z14.number().safeParse(meta.costInUsdTicks);
+  const costTicks = z16.number().safeParse(meta.costInUsdTicks);
   const cost = costTicks.success ? costTicks.data / 1e10 : void 0;
   const event = {
     model: ctx.model.id,
@@ -6704,22 +6909,22 @@ var liveState = store[KEY] ??= {
 var liveCatalog = liveState.catalog;
 
 // src/catalog/load.ts
-import { z as z15 } from "zod";
+import { z as z17 } from "zod";
 function rowCount() {
   return liveCatalog.list({ lifecycle: ["active", "deprecated", "retired"] }).length;
 }
 function authHeaders(auth) {
   return "bearer" in auth ? { Authorization: `Bearer ${auth.bearer}` } : { "x-api-key": auth.apiKey };
 }
-var EnvelopeSchema = z15.object({
-  rows: z15.array(JsonValueSchema),
-  revision: z15.string().optional()
+var EnvelopeSchema = z17.object({
+  rows: z17.array(JsonValueSchema),
+  revision: z17.string().optional()
 });
 function dropReason(path, message) {
   return path.length === 0 ? message : `${path.map((part) => String(part)).join(".")} ${message}`;
 }
 function rowId(raw) {
-  const parsed = z15.object({ id: z15.string().min(1) }).loose().safeParse(raw);
+  const parsed = z17.object({ id: z17.string().min(1) }).loose().safeParse(raw);
   if (!parsed.success) return "?";
   return parsed.data.id;
 }
@@ -7479,7 +7684,7 @@ import {
   streamObject as aiStreamObject2,
   streamText as aiStreamText2
 } from "ai";
-import { z as z16 } from "zod";
+import { z as z18 } from "zod";
 
 // src/generated-image.ts
 function toGeneratedImage(image) {
@@ -7554,10 +7759,10 @@ function textAbortOptions(sdk, context) {
   if (abortSignal) return { abortSignal };
   return {};
 }
-var catalogId = z16.string();
-var exactSize = z16.templateLiteral([z16.number(), "x", z16.number()]);
-var sdkRatio = z16.templateLiteral([z16.number(), ":", z16.number()]);
-var sizeTier = z16.enum(["512", "1K", "2K", "4K"]);
+var catalogId = z18.string();
+var exactSize = z18.templateLiteral([z18.number(), "x", z18.number()]);
+var sdkRatio = z18.templateLiteral([z18.number(), ":", z18.number()]);
+var sizeTier = z18.enum(["512", "1K", "2K", "4K"]);
 function isCatalogId(model) {
   return catalogId.safeParse(model).success;
 }
@@ -7748,6 +7953,7 @@ function createModelKit(config, adapters) {
     logCoercions(logger, model.modelId, warnings);
     const abortSignal = combineSignals(req.abortSignal, call.signal);
     if (abortSignal) options.abortSignal = abortSignal;
+    if (call.maxRetries !== void 0) options.maxRetries = call.maxRetries;
     const result = await aiGenerateImage(options);
     const images = result.images.map(
       (file) => toGeneratedImage({ bytes: file.uint8Array, mime: file.mediaType })
@@ -7787,6 +7993,7 @@ function createModelKit(config, adapters) {
     const seconds = duration ?? durationSeconds;
     const abortSignal = combineSignals(req.abortSignal, call.signal);
     const options = { model, prompt };
+    if (call.maxRetries !== void 0) options.maxRetries = call.maxRetries;
     const warnings = [];
     const ratio = sdkAspectRatio(aspectRatio) ?? (aspectRatio === "adaptive" ? "adaptive" : void 0);
     if (ratio) options.aspectRatio = ratio;
@@ -7868,6 +8075,7 @@ function createModelKit(config, adapters) {
     catalog,
     describe: (id) => catalog.require(id),
     async generateImage(req, call = {}) {
+      mediaRetryOptions(call);
       if (!isCatalogId(req.model)) {
         return passthroughImage({ ...req, model: req.model }, call);
       }
@@ -7911,6 +8119,7 @@ function createModelKit(config, adapters) {
       return { ...result, usage: usage2, image: first, images, warnings };
     },
     async generateVideo(req, call = {}) {
+      mediaRetryOptions(call);
       if (!isCatalogId(req.model)) {
         return passthroughVideo({ ...req, model: req.model }, call);
       }
@@ -7965,6 +8174,7 @@ function createModelKit(config, adapters) {
       };
     },
     async generateAudio(req, call = {}) {
+      mediaRetryOptions(call);
       const model = catalog.require(req.model);
       if (model.type !== "audio") {
         throw new ModelKitError({
@@ -7994,6 +8204,7 @@ function createModelKit(config, adapters) {
       return result;
     },
     async prepareMediaPart(modelId, ref, mediaType, call = {}) {
+      mediaRetryOptions(call);
       const model = catalog.require(modelId);
       const { adapter, ctx } = context(model, call, []);
       const fileStore = model.adapter === "ai-sdk-text" && textKeyName(model) === "google" ? adapters["google-genai"] : adapter;