@shop-os/modelkit 0.2.13 → 0.2.14

package/dist/index.cjs

@@ -246,7 +246,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));
     }
@@ -2787,6 +2789,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;
@@ -2794,6 +2809,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) {
@@ -2906,11 +2922,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(
@@ -3047,11 +3063,6 @@ var SeedreamResponseSchema = import_zod6.z.object({
   data: import_zod6.z.array(SeedreamImageSchema).optional(),
   error: import_zod6.z.object({ message: import_zod6.z.string().optional(), code: import_zod6.z.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;
@@ -3089,20 +3100,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) {
@@ -3238,20 +3252,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);
@@ -3260,7 +3277,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);
     }
@@ -3326,16 +3344,109 @@ async function generateSeedanceVideo(req, ctx) {
 }
 
 // src/adapters/fal.ts
-var import_zod7 = require("zod");
+var import_zod8 = require("zod");
 var import_client = require("@fal-ai/client");
+
+// src/adapters/fal-http.ts
+var import_zod7 = require("zod");
+var queueUrl = import_zod7.z.url().refine((value) => new URL(value).origin === "https://queue.fal.run");
+var submissionSchema = import_zod7.z.object({
+  request_id: import_zod7.z.string(),
+  status_url: queueUrl,
+  response_url: queueUrl
+});
+var statusSchema = import_zod7.z.object({ status: import_zod7.z.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 = import_zod7.z.object({ data: JsonValueSchema });
+var falSubscribePayloadSchema = import_zod8.z.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 = (0, import_client.createFalClient)(opts);
   return {
     async subscribe(endpoint, options) {
@@ -3384,7 +3495,7 @@ function createFalAdapter(deps = defaultDeps) {
   };
 }
 var falAdapter = createFalAdapter();
-var falDetailSchema = import_zod7.z.object({ detail: JsonValueSchema });
+var falDetailSchema = import_zod8.z.object({ detail: JsonValueSchema });
 function liftDetail(body) {
   const parsed = falDetailSchema.safeParse(body);
   const detail = parsed.success ? parsed.data.detail : void 0;
@@ -3393,11 +3504,11 @@ function liftDetail(body) {
 function falError(status, body, model) {
   return classifyHttpError(status, liftDetail(body), "fal", model);
 }
-var falSdkThrownSchema = import_zod7.z.object({
-  status: import_zod7.z.number().optional(),
-  statusCode: import_zod7.z.number().optional(),
+var falSdkThrownSchema = import_zod8.z.object({
+  status: import_zod8.z.number().optional(),
+  statusCode: import_zod8.z.number().optional(),
   body: JsonValueSchema.optional(),
-  message: import_zod7.z.string().optional()
+  message: import_zod8.z.string().optional()
 });
 function falSdkError(cause, model) {
   const parsed = falSdkThrownSchema.safeParse(cause);
@@ -3430,6 +3541,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) {
@@ -3437,16 +3550,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 = import_zod7.z.object({
-  request_id: import_zod7.z.string(),
-  status_url: import_zod7.z.string(),
-  response_url: import_zod7.z.string()
+var falQueueSubmitSchema = import_zod8.z.object({
+  request_id: import_zod8.z.string(),
+  status_url: import_zod8.z.string(),
+  response_url: import_zod8.z.string()
 });
-var falQueueStatusSchema = import_zod7.z.object({ status: import_zod7.z.string() });
+var falQueueStatusSchema = import_zod8.z.object({ status: import_zod8.z.string() });
 async function passthroughRequest(ctx, endpoint, input) {
   const cfg = gatewayConfig(ctx);
   const submitRes = await gatewayFetch(cfg, ctx.fetch, `${FAL_QUEUE_HOST}/${endpoint}`, {
@@ -3503,7 +3618,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);
@@ -3520,7 +3639,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);
@@ -3572,11 +3694,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 = import_zod7.z.object({
-  images: import_zod7.z.array(
-    import_zod7.z.object({
-      url: import_zod7.z.string().optional(),
-      content_type: import_zod7.z.string().nullable().optional()
+var museImageResultSchema = import_zod8.z.object({
+  images: import_zod8.z.array(
+    import_zod8.z.object({
+      url: import_zod8.z.string().optional(),
+      content_type: import_zod8.z.string().nullable().optional()
     })
   ).optional()
 });
@@ -3680,8 +3802,8 @@ function klingO1Input(req) {
   if (refs.length > 0) input.image_urls = refs;
   return input;
 }
-var falVideoResultSchema = import_zod7.z.object({
-  video: import_zod7.z.object({ url: import_zod7.z.string().optional() }).optional()
+var falVideoResultSchema = import_zod8.z.object({
+  video: import_zod8.z.object({ url: import_zod8.z.string().optional() }).optional()
 });
 function extractVideoUrl(result, endpoint, model) {
   const parsed = falVideoResultSchema.safeParse(result);
@@ -3761,7 +3883,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);
@@ -3790,6 +3912,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" };
@@ -3813,7 +3936,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);
@@ -3861,6 +3984,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" };
@@ -3880,11 +4004,11 @@ async function passthroughVideoJob(ctx, req, endpoint, input) {
 }
 
 // src/adapters/google-genai.ts
-var import_zod9 = require("zod");
+var import_zod10 = require("zod");
 var import_genai = require("@google/genai");
 
 // src/adapters/google-files.ts
-var import_zod8 = require("zod");
+var import_zod9 = require("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;
@@ -3894,17 +4018,17 @@ var GENERIC_MEDIA_TYPES = /* @__PURE__ */ new Set([
   "binary/octet-stream",
   "application/binary"
 ]);
-var uploadedFileSchema = import_zod8.z.object({
-  file: import_zod8.z.object({
-    uri: import_zod8.z.string().optional(),
-    name: import_zod8.z.string().optional(),
-    state: import_zod8.z.string().optional(),
-    mimeType: import_zod8.z.string().optional()
+var uploadedFileSchema = import_zod9.z.object({
+  file: import_zod9.z.object({
+    uri: import_zod9.z.string().optional(),
+    name: import_zod9.z.string().optional(),
+    state: import_zod9.z.string().optional(),
+    mimeType: import_zod9.z.string().optional()
   }).optional()
 });
-var fileStateSchema = import_zod8.z.object({
-  state: import_zod8.z.string().optional(),
-  error: import_zod8.z.object({ message: import_zod8.z.string().optional() }).optional()
+var fileStateSchema = import_zod9.z.object({
+  state: import_zod9.z.string().optional(),
+  error: import_zod9.z.object({ message: import_zod9.z.string().optional() }).optional()
 });
 function resolveMime(declared, fallback) {
   const clean = declared?.split(";")[0]?.trim().toLowerCase();
@@ -4038,7 +4162,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);
@@ -4062,8 +4186,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 {
@@ -4078,7 +4201,12 @@ async function uploadGoogleFile(source, opts) {
 
 // src/adapters/google-genai.ts
 function defaultCreateGenAIClient(opts) {
-  const sdk = new import_genai.GoogleGenAI({ ...opts, vertexai: false });
+  const sdk = new import_genai.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) => {
@@ -4099,11 +4227,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;
       }
     }
@@ -4117,7 +4245,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 = import_zod9.z.string();
+var errorTextSchema = import_zod10.z.string();
 function isGoogleGenaiVideoKnown(providerModelId) {
   return providerModelId.startsWith("veo-") || providerModelId.startsWith("gemini-omni");
 }
@@ -4272,7 +4400,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;
@@ -4285,8 +4417,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 () => {
@@ -4399,7 +4530,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) => {
@@ -4410,6 +4545,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);
@@ -4425,8 +4561,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 = [];
@@ -4496,7 +4631,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;
@@ -4606,9 +4741,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 () => {
@@ -4670,6 +4808,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",
@@ -4682,15 +4821,15 @@ async function prepareGoogleMediaPart(ref, mediaType, ctx) {
 // src/adapters/google-image.ts
 var import_google2 = require("@ai-sdk/google");
 var import_ai2 = require("ai");
-var import_zod10 = require("zod");
+var import_zod11 = require("zod");
 var GEMINI_IMAGE_TIMEOUT_MS = 6e5;
 var defaultGoogleImageDeps = {
   createGoogleClient: (opts) => (0, import_google2.createGoogleGenerativeAI)(opts),
   runTextGeneration: (opts) => (0, import_ai2.generateText)(opts)
 };
-var causeChainNodeSchema = import_zod10.z.object({
-  message: import_zod10.z.string().optional(),
-  cause: import_zod10.z.unknown().optional()
+var causeChainNodeSchema = import_zod11.z.object({
+  message: import_zod11.z.string().optional(),
+  cause: import_zod11.z.unknown().optional()
 });
 function extractBlockReason(cause) {
   let current = cause;
@@ -4758,8 +4897,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;
@@ -4811,7 +4949,7 @@ function createGoogleImageAdapter(deps = defaultGoogleImageDeps) {
 var googleImageAdapter = createGoogleImageAdapter();
 
 // src/adapters/higgsfield.ts
-var import_zod11 = require("zod");
+var import_zod12 = require("zod");
 
 // src/adapters/higgsfield-motions.ts
 var MOTION_MAP = {
@@ -5010,14 +5148,14 @@ function videoUsage3(ctx, req, duration) {
   if (req.durationSeconds !== void 0) event.durationSeconds = req.durationSeconds;
   return event;
 }
-var klingCreateSchema = import_zod11.z.object({ status_url: import_zod11.z.string() });
-var klingStatusSchema = import_zod11.z.object({
-  status: import_zod11.z.string().optional(),
-  video: import_zod11.z.object({ url: import_zod11.z.string().optional() }).optional(),
+var klingCreateSchema = import_zod12.z.object({ status_url: import_zod12.z.string() });
+var klingStatusSchema = import_zod12.z.object({
+  status: import_zod12.z.string().optional(),
+  video: import_zod12.z.object({ url: import_zod12.z.string().optional() }).optional(),
   error: JsonValueSchema.optional()
 });
-var errorMessageSchema = import_zod11.z.object({ message: import_zod11.z.string().optional() });
-var textValueSchema = import_zod11.z.string();
+var errorMessageSchema = import_zod12.z.object({ message: import_zod12.z.string().optional() });
+var textValueSchema = import_zod12.z.string();
 function klingErrorDetail(error) {
   if (error === void 0) return "unknown_error";
   const asText = textValueSchema.safeParse(error);
@@ -5066,8 +5204,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
@@ -5087,10 +5224,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);
     }
@@ -5132,6 +5271,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;
       }
@@ -5151,8 +5291,8 @@ async function generateKlingVideo(req, ctx) {
     }
   };
 }
-var dopMotionOptionsSchema = import_zod11.z.object({
-  modelkit: import_zod11.z.object({ motion: import_zod11.z.string().optional() }).optional()
+var dopMotionOptionsSchema = import_zod12.z.object({
+  modelkit: import_zod12.z.object({ motion: import_zod12.z.string().optional() }).optional()
 });
 function dopMotionId(ctx) {
   const parsed = dopMotionOptionsSchema.safeParse(ctx.model.providerOptions);
@@ -5168,7 +5308,7 @@ function dopMotionId(ctx) {
     });
   return id;
 }
-var jsonRecordSchema = import_zod11.z.record(import_zod11.z.string(), JsonValueSchema);
+var jsonRecordSchema = import_zod12.z.record(import_zod12.z.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) {
@@ -5196,8 +5336,8 @@ function dopUrl(body) {
   }
   return void 0;
 }
-var dopCreateSchema = import_zod11.z.object({ id: import_zod11.z.string() });
-var dopStatusSchema = import_zod11.z.object({ status: import_zod11.z.string().optional() });
+var dopCreateSchema = import_zod12.z.object({ id: import_zod12.z.string() });
+var dopStatusSchema = import_zod12.z.object({ status: import_zod12.z.string().optional() });
 async function generateDopVideo(req, ctx) {
   const firstFrame = req.firstFrame ? refToUrl(req.firstFrame) : void 0;
   if (!firstFrame) {
@@ -5222,8 +5362,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",
@@ -5247,10 +5386,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());
@@ -5287,6 +5434,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;
       }
@@ -5317,7 +5465,7 @@ var higgsfieldAdapter = {
 
 // src/adapters/openai-image.ts
 var import_openai2 = __toESM(require("openai"), 1);
-var import_zod12 = require("zod");
+var import_zod13 = require("zod");
 var MAX_EDIT_REFERENCES = 16;
 function aspectRatioToSize(ratio, tier) {
   const TIER_LONG_EDGE = {
@@ -5400,9 +5548,9 @@ function toProviderError2(cause, ctx) {
     ctx.model.id
   );
 }
-var openaiUsageSchema = import_zod12.z.object({
-  input_tokens: import_zod12.z.number().optional(),
-  output_tokens: import_zod12.z.number().optional()
+var openaiUsageSchema = import_zod13.z.object({
+  input_tokens: import_zod13.z.number().optional(),
+  output_tokens: import_zod13.z.number().optional()
 });
 var openaiImageAdapter = {
   id: "openai-image",
@@ -5444,8 +5592,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) {
@@ -5765,13 +5912,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 () => {
@@ -5818,7 +5959,66 @@ var prodiaAdapter = {
 
 // src/adapters/replicate.ts
 var import_replicate = __toESM(require("replicate"), 1);
-var import_zod13 = require("zod");
+
+// src/adapters/replicate-http.ts
+var import_zod14 = require("zod");
+var predictionSchema = import_zod14.z.object({
+  id: import_zod14.z.string(),
+  status: import_zod14.z.string(),
+  output: JsonValueSchema.optional(),
+  error: JsonValueSchema.optional(),
+  metrics: import_zod14.z.object({ predict_time: import_zod14.z.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
+var import_zod15 = require("zod");
 
 // src/adapters/input-spec-mapper.ts
 function canonicalValue(req, key) {
@@ -5857,7 +6057,7 @@ function applyInputSpec(spec, req, refToUrl2, opts) {
 var POLL_INTERVAL_MS = 1e3;
 var POLL_TIMEOUT_MS = 15 * 6e4;
 var defaultReplicateDeps = {
-  createReplicateClient: (opts) => new import_replicate.default(opts)
+  createReplicateClient: (opts) => opts.maxRetries === void 0 ? new import_replicate.default(opts) : createReplicateHttpClient(opts)
 };
 function createReplicateAdapter(deps = defaultReplicateDeps) {
   return {
@@ -5879,7 +6079,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(":")) {
@@ -5889,8 +6093,8 @@ function route(providerModelId) {
   if (!providerModelId.includes("/")) return { version: providerModelId };
   return { model: providerModelId };
 }
-var replicateResponseStatusSchema = import_zod13.z.object({
-  response: import_zod13.z.object({ status: import_zod13.z.number() }).partial()
+var replicateResponseStatusSchema = import_zod15.z.object({
+  response: import_zod15.z.object({ status: import_zod15.z.number() }).partial()
 });
 function replicateError(cause, ctx) {
   if (isModelKitError(cause)) return cause;
@@ -5917,7 +6121,7 @@ function isUrlFn(value) {
 function extractOutputUrl(output) {
   const root = Array.isArray(output) ? output[0] : output;
   if (root === void 0) throw noOutput();
-  const asString = import_zod13.z.string().safeParse(root);
+  const asString = import_zod15.z.string().safeParse(root);
   if (asString.success && asString.data) return asString.data;
   if (root instanceof URL) return root.href;
   if (isObjectOutput(root)) {
@@ -5926,15 +6130,15 @@ function extractOutputUrl(output) {
       if (urlValue !== void 0 && isUrlFn(urlValue)) {
         const called = urlValue();
         if (called instanceof URL) return called.href;
-        const calledString = import_zod13.z.string().safeParse(called);
+        const calledString = import_zod15.z.string().safeParse(called);
         if (calledString.success && calledString.data) return calledString.data;
       } else {
-        const direct = import_zod13.z.string().safeParse(urlValue);
+        const direct = import_zod15.z.string().safeParse(urlValue);
         if (direct.success && direct.data) return direct.data;
       }
     }
     if ("default" in root) {
-      const fallback = import_zod13.z.string().safeParse(root.default);
+      const fallback = import_zod15.z.string().safeParse(root.default);
       if (fallback.success && fallback.data) return fallback.data;
     }
   }
@@ -5964,7 +6168,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);
@@ -5995,7 +6199,7 @@ async function runPrediction(ctx, input, deps) {
     }
     return void 0;
   }, pollOpts);
-  const predictTime = import_zod13.z.number().safeParse(done.metrics?.predict_time);
+  const predictTime = import_zod15.z.number().safeParse(done.metrics?.predict_time);
   if (done.output === void 0) throw noOutput();
   return {
     url: extractOutputUrl(done.output),
@@ -6363,7 +6567,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);
@@ -6380,7 +6584,7 @@ async function generateReplicateVideo(req, ctx, deps) {
       throw replicateError(cause, ctx);
     }
     if (current.status === "succeeded") {
-      const predictTime = import_zod13.z.number().safeParse(current.metrics?.predict_time);
+      const predictTime = import_zod15.z.number().safeParse(current.metrics?.predict_time);
       if (current.output === void 0) throw noOutput();
       return {
         url: extractOutputUrl(current.output),
@@ -6521,7 +6725,7 @@ async function generateReplicateAudio(req, ctx, deps) {
 // src/adapters/xai-video.ts
 var import_ai3 = require("ai");
 var import_xai2 = require("@ai-sdk/xai");
-var import_zod14 = require("zod");
+var import_zod16 = require("zod");
 var POLL_INTERVAL_MS2 = 1e4;
 var POLL_TIMEOUT_MS2 = 15 * 6e4;
 var defaultXaiDeps = {
@@ -6534,8 +6738,8 @@ function createXaiVideoAdapter(deps = defaultXaiDeps) {
   };
 }
 var xaiVideoAdapter = createXaiVideoAdapter();
-var xaiResponseStatusSchema = import_zod14.z.object({
-  response: import_zod14.z.object({ status: import_zod14.z.number() }).partial()
+var xaiResponseStatusSchema = import_zod16.z.object({
+  response: import_zod16.z.object({ status: import_zod16.z.number() }).partial()
 });
 function xaiError(cause, ctx) {
   const info = describeThrown(cause);
@@ -6543,12 +6747,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 = import_zod14.z.object({
+var xaiMetaSchema = import_zod16.z.object({
   videoUrl: JsonValueSchema.optional(),
   costInUsdTicks: JsonValueSchema.optional(),
   duration: JsonValueSchema.optional()
 });
-var xaiProviderMetadataSchema = import_zod14.z.object({ xai: xaiMetaSchema });
+var xaiProviderMetadataSchema = import_zod16.z.object({ xai: xaiMetaSchema });
 function metaOf(result) {
   const parsed = xaiProviderMetadataSchema.partial().safeParse(result.providerMetadata);
   if (!parsed.success) return {};
@@ -6599,6 +6803,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);
@@ -6655,7 +6860,7 @@ async function mapResult(ctx, req, result) {
     };
   }
   const meta = metaOf(result);
-  const videoUrl = import_zod14.z.string().safeParse(meta.videoUrl);
+  const videoUrl = import_zod16.z.string().safeParse(meta.videoUrl);
   if (videoUrl.success && videoUrl.data) {
     const res = await ctx.fetch(videoUrl.data);
     if (!res.ok) {
@@ -6678,7 +6883,7 @@ async function mapResult(ctx, req, result) {
   });
 }
 function usage(ctx, req, meta) {
-  const costTicks = import_zod14.z.number().safeParse(meta.costInUsdTicks);
+  const costTicks = import_zod16.z.number().safeParse(meta.costInUsdTicks);
   const cost = costTicks.success ? costTicks.data / 1e10 : void 0;
   const event = {
     model: ctx.model.id,
@@ -6811,22 +7016,22 @@ var liveState = store[KEY] ??= {
 var liveCatalog = liveState.catalog;
 
 // src/catalog/load.ts
-var import_zod15 = require("zod");
+var import_zod17 = require("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 = import_zod15.z.object({
-  rows: import_zod15.z.array(JsonValueSchema),
-  revision: import_zod15.z.string().optional()
+var EnvelopeSchema = import_zod17.z.object({
+  rows: import_zod17.z.array(JsonValueSchema),
+  revision: import_zod17.z.string().optional()
 });
 function dropReason(path, message) {
   return path.length === 0 ? message : `${path.map((part) => String(part)).join(".")} ${message}`;
 }
 function rowId(raw) {
-  const parsed = import_zod15.z.object({ id: import_zod15.z.string().min(1) }).loose().safeParse(raw);
+  const parsed = import_zod17.z.object({ id: import_zod17.z.string().min(1) }).loose().safeParse(raw);
   if (!parsed.success) return "?";
   return parsed.data.id;
 }
@@ -7579,7 +7784,7 @@ function createTextSteps(initial, resolve, callerProviderOptions) {
 
 // src/client.ts
 var import_ai5 = require("ai");
-var import_zod16 = require("zod");
+var import_zod18 = require("zod");
 
 // src/generated-image.ts
 function toGeneratedImage(image) {
@@ -7654,10 +7859,10 @@ function textAbortOptions(sdk, context) {
   if (abortSignal) return { abortSignal };
   return {};
 }
-var catalogId = import_zod16.z.string();
-var exactSize = import_zod16.z.templateLiteral([import_zod16.z.number(), "x", import_zod16.z.number()]);
-var sdkRatio = import_zod16.z.templateLiteral([import_zod16.z.number(), ":", import_zod16.z.number()]);
-var sizeTier = import_zod16.z.enum(["512", "1K", "2K", "4K"]);
+var catalogId = import_zod18.z.string();
+var exactSize = import_zod18.z.templateLiteral([import_zod18.z.number(), "x", import_zod18.z.number()]);
+var sdkRatio = import_zod18.z.templateLiteral([import_zod18.z.number(), ":", import_zod18.z.number()]);
+var sizeTier = import_zod18.z.enum(["512", "1K", "2K", "4K"]);
 function isCatalogId(model) {
   return catalogId.safeParse(model).success;
 }
@@ -7848,6 +8053,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 (0, import_ai5.generateImage)(options);
     const images = result.images.map(
       (file) => toGeneratedImage({ bytes: file.uint8Array, mime: file.mediaType })
@@ -7887,6 +8093,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;
@@ -7968,6 +8175,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);
       }
@@ -8011,6 +8219,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);
       }
@@ -8065,6 +8274,7 @@ function createModelKit(config, adapters) {
       };
     },
     async generateAudio(req, call = {}) {
+      mediaRetryOptions(call);
       const model = catalog.require(req.model);
       if (model.type !== "audio") {
         throw new ModelKitError({
@@ -8094,6 +8304,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;