@shop-os/modelkit 0.1.4 → 0.1.6

package/dist/index.cjs

@@ -68,6 +68,7 @@ __export(index_exports, {
   prodiaAdapter: () => prodiaAdapter,
   readErrorBody: () => readErrorBody,
   replicateAdapter: () => replicateAdapter,
+  resolveAudioOptions: () => resolveAudioOptions,
   resolveImageOptions: () => resolveImageOptions,
   resolveKeys: () => resolveKeys,
   resolveVideoOptions: () => resolveVideoOptions,
@@ -175,7 +176,8 @@ var GEMINI_RATIOS = [
   "3:4",
   "21:9",
   "4:5",
-  "5:4"
+  "5:4",
+  "adaptive"
 ];
 function geminiImage(id, label, aliases, sizes) {
   const model = {
@@ -186,7 +188,11 @@ function geminiImage(id, label, aliases, sizes) {
     adapter: "google-image",
     providerModelId: id,
     lifecycle: "active",
-    constraints: { aspectRatios: [...GEMINI_RATIOS], sizes },
+    constraints: {
+      aspectRatios: [...GEMINI_RATIOS],
+      sizes,
+      supports: ["safetySettings", "temperature"]
+    },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE },
     pricingKind: "image_size"
@@ -283,7 +289,8 @@ var IMAGE_MODELS = [
         "9:16",
         "4:5",
         "5:4",
-        "9:21"
+        "9:21",
+        "adaptive"
       ],
       sizes: ["1K"],
       references: { images: 10 },
@@ -324,8 +331,20 @@ var IMAGE_MODELS = [
     constraints: {
       // 97:30 has no Ark size spelling; the A+ flow pads inputs to 97:30 and
       // omits the ratio instead. Advertise only sizes the adapter can send.
-      aspectRatios: ["1:1", "2:3", "3:2", "4:3", "3:4", "4:5", "5:4", "9:16", "16:9", "21:9"],
-      sizes: ["1K", "2K"],
+      aspectRatios: [
+        "1:1",
+        "2:3",
+        "3:2",
+        "4:3",
+        "3:4",
+        "4:5",
+        "5:4",
+        "9:16",
+        "16:9",
+        "21:9",
+        "9:21"
+      ],
+      sizes: ["1K", "2K", "4K"],
       references: { images: 10 },
       supports: ["seed"]
     },
@@ -344,9 +363,10 @@ var IMAGE_MODELS = [
     constraints: {
       aspectRatios: ["1:1", "2:3", "3:2", "16:9", "9:16", "4:3", "3:4", "4:5", "5:4", "21:9"],
       sizes: ["1K"],
-      supports: ["outputFormat"]
+      // flux-2-pro's job type always returns image/png; only the seed is forwarded.
+      supports: ["seed"]
     },
-    defaults: { size: "1K", aspectRatio: "1:1", outputFormat: "png" },
+    defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE },
     pricingKind: "image_size"
   },
@@ -364,7 +384,8 @@ var IMAGE_MODELS = [
     constraints: {
       aspectRatios: ["1:1", "2:3", "3:2", "4:3", "3:4", "4:5", "5:4", "9:16", "16:9", "21:9"],
       sizes: ["1K", "2K", "4K"],
-      requires: ["references"]
+      requires: ["references"],
+      supports: ["seed"]
     },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE },
@@ -383,7 +404,8 @@ var IMAGE_MODELS = [
     constraints: {
       aspectRatios: ["1:1", "2:3", "3:4", "4:3", "9:16", "16:9", "21:9"],
       sizes: ["1K"],
-      requires: ["references"]
+      requires: ["references"],
+      supports: ["seed"]
     },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE },
@@ -403,7 +425,8 @@ var IMAGE_MODELS = [
     constraints: {
       aspectRatios: ["1:1", "16:9", "9:16", "4:3", "3:4"],
       sizes: ["1K", "2K", "4K"],
-      references: { images: 9 }
+      references: { images: 9 },
+      supports: []
     },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { unit: "compute_seconds", modality: "IMAGE" },
@@ -424,7 +447,8 @@ var IMAGE_MODELS = [
       aspectRatios: ["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3"],
       sizes: [],
       references: { images: 1 },
-      requires: ["references"]
+      requires: ["references"],
+      supports: []
     },
     defaults: { aspectRatio: "1:1" },
     usage: { unit: "compute_seconds", modality: "IMAGE" },
@@ -466,7 +490,8 @@ var OTHER_MODELS = [
     lifecycle: "active",
     constraints: {
       durations: [30],
-      references: { images: 10 }
+      references: { images: 10 },
+      supports: []
     },
     defaults: {},
     usage: { unit: "images", modality: "AUDIO" },
@@ -481,7 +506,8 @@ var OTHER_MODELS = [
     providerModelId: "lyria-3-pro-preview",
     lifecycle: "active",
     constraints: {
-      references: { images: 10 }
+      references: { images: 10 },
+      supports: []
     },
     defaults: {},
     usage: { unit: "images", modality: "AUDIO" },
@@ -548,7 +574,7 @@ var OTHER_MODELS = [
     constraints: {
       references: { images: 1 },
       requires: ["references"],
-      supports: ["scaleFactor", "outputFormat"]
+      supports: ["scaleFactor", "outputFormat", "upscale"]
     },
     defaults: { scaleFactor: 2, outputFormat: "png" },
     usage: { unit: "compute_seconds", modality: "IMAGE" },
@@ -564,7 +590,8 @@ var OTHER_MODELS = [
     lifecycle: "active",
     constraints: {
       references: { images: 1 },
-      requires: ["references"]
+      requires: ["references"],
+      supports: []
     },
     defaults: {},
     usage: { unit: "images", modality: "IMAGE" },
@@ -583,7 +610,7 @@ function text(id, label, providerModelId, extra) {
     adapter: "ai-sdk-text",
     providerModelId,
     lifecycle: "active",
-    constraints: {},
+    constraints: { supports: [] },
     defaults: {},
     usage: { ...TEXT_USAGE },
     pricingKind: "token",
@@ -656,7 +683,7 @@ var TEXT_MODELS = [
 // src/catalog/video.ts
 var VIDEO_USAGE = { unit: "seconds", modality: "VIDEO" };
 var SEEDANCE_RATIOS = ["21:9", "16:9", "4:3", "1:1", "3:4", "9:16", "adaptive"];
-var KLING_RATIOS = ["16:9", "9:16", "1:1"];
+var KLING_RATIOS = ["16:9", "9:16", "1:1", "adaptive"];
 var OMNI_RATIOS = ["16:9", "9:16"];
 var GROK_RATIOS = ["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3"];
 var H3_RATIOS = ["21:9", "16:9", "4:3", "1:1", "3:4", "9:16"];
@@ -748,12 +775,15 @@ var VIDEO_MODELS = [
     providerModelId: "gemini-omni-flash-preview",
     lifecycle: "active",
     constraints: {
+      // 1.0 rejects `resolution`; it renders 720p. No defaults: the adapter
+      // sends ratio and duration only when the caller set them, so a
+      // reference clip keeps its own framing.
       aspectRatios: [...OMNI_RATIOS],
-      resolutions: ["720p"],
       durations: [3, 4, 5, 6, 7, 8, 9, 10],
-      references: { images: 6, videos: 1 }
+      references: { images: 6, videos: 1 },
+      supports: []
     },
-    defaults: { durationSeconds: 8, resolution: "720p", aspectRatio: "16:9" },
+    defaults: {},
     usage: { ...VIDEO_USAGE },
     pricingKind: "video_per_second"
   },
@@ -769,9 +799,10 @@ var VIDEO_MODELS = [
       aspectRatios: [...OMNI_RATIOS],
       resolutions: ["360p", "720p", "1080p", "4k"],
       durations: [3, 4, 5, 6, 7, 8, 9, 10],
-      references: { images: 6, videos: 1 }
+      references: { images: 6, videos: 1 },
+      supports: []
     },
-    defaults: { durationSeconds: 8, resolution: "720p", aspectRatio: "16:9" },
+    defaults: {},
     usage: { ...VIDEO_USAGE },
     pricingKind: "video_per_second"
   },
@@ -790,7 +821,7 @@ var VIDEO_MODELS = [
       durations: [4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15],
       references: { images: 9, videos: 3, audios: 3, audioOnly: false },
       ratioIgnoredWith: ["firstFrame"],
-      supports: ["generateAudio", "seed", "lastFrame"]
+      supports: ["generateAudio", "seed", "lastFrame", "watermark"]
     },
     defaults: {
       durationSeconds: 5,
@@ -815,7 +846,7 @@ var VIDEO_MODELS = [
       durations: [4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15],
       references: { images: 9, videos: 3, audios: 3, audioOnly: false },
       ratioIgnoredWith: ["firstFrame"],
-      supports: ["generateAudio", "seed", "lastFrame"]
+      supports: ["generateAudio", "seed", "lastFrame", "watermark"]
     },
     defaults: {
       durationSeconds: 5,
@@ -841,7 +872,7 @@ var VIDEO_MODELS = [
       durations: { min: 4, max: 30 },
       references: { images: 30, videos: 10, audios: 10 },
       ratioIgnoredWith: ["firstFrame"],
-      supports: ["generateAudio", "seed", "lastFrame"]
+      supports: ["generateAudio", "seed", "lastFrame", "watermark"]
     },
     defaults: {
       durationSeconds: 5,
@@ -865,9 +896,11 @@ var VIDEO_MODELS = [
       aspectRatios: [...GROK_RATIOS],
       resolutions: ["480p", "720p"],
       durations: { min: 1, max: 15 },
-      references: { images: 7 }
+      references: { images: 7 },
+      supports: []
     },
-    defaults: { durationSeconds: 8, resolution: "720p", aspectRatio: "16:9" },
+    // No resolution default: xAI picks its own unless the caller asks.
+    defaults: { durationSeconds: 8, aspectRatio: "16:9" },
     usage: { unit: "reported_cost_usd", modality: "VIDEO" },
     pricingKind: "video_per_second"
   },
@@ -961,7 +994,7 @@ var VIDEO_MODELS = [
       aspectRatios: [...H3_RATIOS],
       resolutions: ["480p", "768p", "2k", "4k"],
       durations: { min: 5, max: 15 },
-      supports: ["seed", "promptExpansion"]
+      supports: ["seed", "promptExpansion", "lastFrame", "safetyChecker"]
     },
     defaults: {
       durationSeconds: 5,
@@ -984,7 +1017,7 @@ var VIDEO_MODELS = [
       aspectRatios: [...H3_RATIOS],
       resolutions: ["480p", "768p"],
       durations: { min: 5, max: 15 },
-      supports: ["seed", "promptExpansion"]
+      supports: ["seed", "promptExpansion", "lastFrame", "safetyChecker"]
     },
     defaults: {
       durationSeconds: 5,
@@ -1008,7 +1041,7 @@ var VIDEO_MODELS = [
       resolutions: ["480p", "768p", "2k", "4k"],
       durations: { min: 5, max: 15 },
       references: { images: 9, videos: 3, audios: 3, total: 12, audioOnly: false },
-      supports: ["seed", "promptExpansion"]
+      supports: ["seed", "promptExpansion", "safetyChecker"]
     },
     defaults: {
       durationSeconds: 5,
@@ -1032,7 +1065,7 @@ var VIDEO_MODELS = [
       resolutions: ["480p", "768p"],
       durations: { min: 5, max: 15 },
       references: { images: 9, videos: 3, audios: 3, total: 12, audioOnly: false },
-      supports: ["seed", "promptExpansion"]
+      supports: ["seed", "promptExpansion", "safetyChecker"]
     },
     defaults: {
       durationSeconds: 5,
@@ -1055,7 +1088,8 @@ var VIDEO_MODELS = [
     constraints: {
       aspectRatios: [...KLING_RATIOS],
       durations: { min: 3, max: 10 },
-      requires: ["sourceVideo"]
+      requires: ["sourceVideo"],
+      supports: []
     },
     defaults: { durationSeconds: 5, aspectRatio: "16:9" },
     usage: { ...VIDEO_USAGE },
@@ -1088,7 +1122,7 @@ var VIDEO_MODELS = [
     lifecycle: "active",
     constraints: {
       requires: ["firstFrame"],
-      supports: ["seed"]
+      supports: ["seed", "lastFrame"]
     },
     defaults: {},
     usage: { ...VIDEO_USAGE },
@@ -1159,19 +1193,22 @@ function resolveTier(field, code, requested, supported, order, model, onUnsuppor
   });
   return to;
 }
-function resolveDuration(requested, durations, model, warnings) {
+function resolveDuration(requested, durations, model, onUnsupported, warnings) {
   if (requested === void 0 || !durations) return requested;
-  let to = requested;
-  if (Array.isArray(durations)) {
-    if (durations.includes(requested)) return requested;
-    to = durations.reduce(
-      (best, d) => Math.abs(d - requested) < Math.abs(best - requested) ? d : best,
-      durations[0] ?? requested
-    );
-  } else {
-    to = Math.min(Math.max(requested, durations.min), durations.max);
-    if (to === requested) return requested;
+  const inSet = Array.isArray(durations) ? durations.includes(requested) : requested >= durations.min && requested <= durations.max;
+  if (inSet) return requested;
+  if (onUnsupported === "error") {
+    const supported = Array.isArray(durations) ? `Supported: ${durations.join(", ")}.` : `Supported range: ${durations.min}-${durations.max}.`;
+    throw new ModelKitError({
+      code: "unsupported_option",
+      message: `${model.id} does not support durationSeconds ${requested}. ${supported}`,
+      model: model.id
+    });
   }
+  const to = Array.isArray(durations) ? durations.reduce(
+    (best, d) => Math.abs(d - requested) < Math.abs(best - requested) ? d : best,
+    durations[0] ?? requested
+  ) : Math.min(Math.max(requested, durations.min), durations.max);
   warnings.push({
     code: "duration_coerced",
     field: "durationSeconds",
@@ -1181,6 +1218,26 @@ function resolveDuration(requested, durations, model, warnings) {
   });
   return to;
 }
+function resolveAspectRatio(requested, supported, model, onUnsupported, warnings) {
+  if (requested === void 0 || requested === "adaptive" || !supported) return requested;
+  if (supported.includes(requested)) return requested;
+  if (onUnsupported === "error") {
+    throw new ModelKitError({
+      code: "unsupported_option",
+      message: `${model.id} does not support aspectRatio "${requested}". Supported: ${supported.join(", ")}.`,
+      model: model.id
+    });
+  }
+  const to = nearestAspectRatio(requested, supported);
+  warnings.push({
+    code: "ratio_coerced",
+    field: "aspectRatio",
+    from: requested,
+    to,
+    message: `aspect ratio ${requested} coerced to ${to} for ${model.id}`
+  });
+  return to;
+}
 function compact(refs) {
   return (refs ?? []).filter((r) => Boolean(r));
 }
@@ -1199,17 +1256,14 @@ function resolveImageOptions(model, request, onUnsupported = "error") {
   const warnings = [];
   const c = model.constraints;
   const merged = { ...model.defaults, ...stripUndefined(request), model: model.id };
-  if (merged.aspectRatio && c.aspectRatios && !c.aspectRatios.includes(merged.aspectRatio)) {
-    const to = nearestAspectRatio(merged.aspectRatio, c.aspectRatios);
-    warnings.push({
-      code: "ratio_coerced",
-      field: "aspectRatio",
-      from: merged.aspectRatio,
-      to,
-      message: `aspect ratio ${merged.aspectRatio} coerced to ${to} for ${model.id}`
-    });
-    merged.aspectRatio = to;
-  }
+  const aspectRatio = resolveAspectRatio(
+    merged.aspectRatio,
+    c.aspectRatios,
+    model,
+    onUnsupported,
+    warnings
+  );
+  if (aspectRatio !== void 0) merged.aspectRatio = aspectRatio;
   const size = resolveTier(
     "size",
     "size_coerced",
@@ -1242,7 +1296,10 @@ function resolveImageOptions(model, request, onUnsupported = "error") {
     "quality",
     "count",
     "outputFormat",
-    "scaleFactor"
+    "scaleFactor",
+    "safetySettings",
+    "temperature",
+    "upscale"
   ]);
   return { request: merged, warnings };
 }
@@ -1272,17 +1329,14 @@ function resolveVideoOptions(model, request, onUnsupported = "error") {
       delete merged.aspectRatio;
     }
   }
-  if (merged.aspectRatio && c.aspectRatios && !c.aspectRatios.includes(merged.aspectRatio)) {
-    const to = nearestAspectRatio(merged.aspectRatio, c.aspectRatios);
-    warnings.push({
-      code: "ratio_coerced",
-      field: "aspectRatio",
-      from: merged.aspectRatio,
-      to,
-      message: `aspect ratio ${merged.aspectRatio} coerced to ${to} for ${model.id}`
-    });
-    merged.aspectRatio = to;
-  }
+  const aspectRatio = resolveAspectRatio(
+    merged.aspectRatio,
+    c.aspectRatios,
+    model,
+    onUnsupported,
+    warnings
+  );
+  if (aspectRatio !== void 0) merged.aspectRatio = aspectRatio;
   const resolution = resolveTier(
     "resolution",
     "resolution_coerced",
@@ -1294,7 +1348,13 @@ function resolveVideoOptions(model, request, onUnsupported = "error") {
     warnings
   );
   if (resolution !== void 0) merged.resolution = resolution;
-  const duration = resolveDuration(merged.durationSeconds, c.durations, model, warnings);
+  const duration = resolveDuration(
+    merged.durationSeconds,
+    c.durations,
+    model,
+    onUnsupported,
+    warnings
+  );
   if (duration !== void 0) merged.durationSeconds = duration;
   const limits = c.references ?? {};
   let images = trim(
@@ -1349,13 +1409,47 @@ function resolveVideoOptions(model, request, onUnsupported = "error") {
     "cfgScale",
     "generateAudio",
     "promptExpansion",
-    "lastFrame"
+    "lastFrame",
+    "watermark",
+    "safetyChecker"
   ]);
   return { request: merged, warnings };
 }
+function resolveAudioOptions(model, request, onUnsupported = "error") {
+  const warnings = [];
+  const c = model.constraints;
+  const merged = { ...model.defaults, ...stripUndefined(request), model: model.id };
+  for (const input of c.requires ?? []) {
+    const present = input === "references" ? compact(merged.references).length > 0 : input === "sourceVideo" && merged.sourceVideo !== void 0;
+    if (!present)
+      throw new ModelKitError({
+        code: "missing_input",
+        message: `${model.id} requires ${input}`,
+        model: model.id
+      });
+  }
+  const duration = resolveDuration(
+    merged.durationSeconds,
+    c.durations,
+    model,
+    onUnsupported,
+    warnings
+  );
+  if (duration !== void 0) merged.durationSeconds = duration;
+  const references = trim(
+    "references",
+    compact(merged.references),
+    c.references?.images ?? c.references?.total,
+    model,
+    warnings
+  );
+  if (references.length > 0) merged.references = references;
+  else delete merged.references;
+  dropUnsupported(merged, model, warnings, ["seed", "negativePrompt", "instrumental"]);
+  return { request: merged, warnings };
+}
 function dropUnsupported(req, model, warnings, fields) {
-  const supports = model.constraints.supports;
-  if (!supports) return;
+  const supports = model.constraints.supports ?? [];
   for (const field of fields) {
     if (req[field] !== void 0 && !supports.includes(field)) {
       warnings.push({
@@ -1395,6 +1489,9 @@ function hasAll(keys, names) {
     (n) => Boolean(keys[n]) && (n !== "higgsfield" || Boolean(keys.higgsfieldSecret))
   );
 }
+function missingEnterpriseKeys(config, names) {
+  return names.filter((name) => !hasAll(config.keys.enterprise, [name]));
+}
 function resolveKeys(config, names, tier = "pro") {
   if (tier === "enterprise" && hasAll(config.keys.enterprise, names))
     return config.keys.enterprise;
@@ -1427,18 +1524,56 @@ function keysFromEnv(env, prefix = "") {
   set("prodia", pick("PRODIA_API_KEY", "PRODIA_TOKEN"));
   set("higgsfield", pick("HIGGSFIELD_API_KEY"));
   set("higgsfieldSecret", pick("HIGGSFIELD_API_SECRET"));
-  set("openrouter", pick("OPENROUTER_API_KEY"));
   set("cloudflare", pick("CLOUDFLARE_AI_GATEWAY_TOKEN", "CF_AIG_TOKEN"));
   return keys;
 }
 
 // src/retry.ts
+var import_zod2 = require("zod");
+
+// src/json.ts
 var import_zod = require("zod");
-var textBody = import_zod.z.string();
+var JsonValueSchema = import_zod.z.lazy(
+  () => import_zod.z.union([
+    import_zod.z.string(),
+    import_zod.z.number(),
+    import_zod.z.boolean(),
+    import_zod.z.null(),
+    import_zod.z.array(JsonValueSchema),
+    import_zod.z.record(import_zod.z.string(), JsonValueSchema)
+  ])
+);
+var JsonObjectSchema = import_zod.z.record(import_zod.z.string(), JsonValueSchema);
+async function readErrorBody(res) {
+  const text2 = await res.text().catch(() => "");
+  try {
+    return JsonValueSchema.parse(JSON.parse(text2));
+  } catch {
+    return text2;
+  }
+}
+var thrownErrorSchema = import_zod.z.object({
+  message: import_zod.z.string().optional(),
+  status: import_zod.z.number().optional(),
+  statusCode: import_zod.z.number().optional(),
+  responseBody: import_zod.z.string().optional()
+});
+function describeThrown(cause) {
+  const parsed = thrownErrorSchema.safeParse(cause);
+  if (!parsed.success) return { message: String(cause), status: void 0, body: String(cause) };
+  const { message, status, statusCode, responseBody } = parsed.data;
+  const body = responseBody ?? message ?? String(cause);
+  return { message: message ?? String(cause), status: status ?? statusCode, body };
+}
+
+// src/retry.ts
+var textBody = import_zod2.z.string();
 function classifyHttpError(status, body, provider, model) {
   const asText = textBody.safeParse(body);
   const text2 = asText.success ? asText.data : JSON.stringify(body);
-  const blocked = /content_policy|safety|blockReason|PROHIBITED_CONTENT|moderation/i.test(text2);
+  const blocked = /content_policy_violation|blockReason|PROHIBITED_CONTENT|moderation_blocked|safety(?![_A-Za-z])/i.test(
+    text2
+  );
   let code = "provider_rejected";
   if (blocked) code = "content_blocked";
   else if (status === 401 || status === 403) code = "auth";
@@ -1459,8 +1594,8 @@ function classifyHttpError(status, body, provider, model) {
 }
 function shouldRetry(err) {
   if (err instanceof ModelKitError) return err.retryable;
-  if (err.name === "AbortError" || /ECONNRESET|ETIMEDOUT|fetch failed|socket hang up/i.test(err.message))
-    return true;
+  if (err.name === "AbortError") return false;
+  if (/ECONNRESET|ETIMEDOUT|fetch failed|socket hang up/i.test(err.message)) return true;
   return false;
 }
 async function withRetry(fn, opts = {}) {
@@ -1517,6 +1652,13 @@ function transportError(provider, cause, model) {
   if (model !== void 0) init.model = model;
   return new ModelKitError(init);
 }
+function classifyThrown(provider, cause, model, override = {}) {
+  if (cause instanceof ModelKitError) return cause;
+  const info = describeThrown(cause);
+  const status = override.status ?? info.status;
+  if (status === void 0) return transportError(provider, cause, model);
+  return classifyHttpError(status, override.body ?? info.body, provider, model);
+}
 async function pollUntil(check, opts) {
   const start = Date.now();
   let interval = opts.intervalMs;
@@ -1632,12 +1774,18 @@ function createTextSteps(initial, resolve, callerProviderOptions) {
 
 // src/client.ts
 var import_ai2 = require("ai");
-var import_zod3 = require("zod");
+var import_zod4 = require("zod");
 
 // src/adapters/media-ref.ts
 function isUrlRef(ref) {
   return "url" in ref;
 }
+function abortedError() {
+  return new ModelKitError({ code: "timeout", message: "aborted", retryable: false });
+}
+function throwIfAborted(signal) {
+  if (signal?.aborted) throw abortedError();
+}
 function mediaRefToBase64(bytes) {
   let binary = "";
   const CHUNK = 32768;
@@ -1652,10 +1800,34 @@ function base64ToBytes(b64) {
   for (let i = 0; i < binary.length; i++) out[i] = binary.charCodeAt(i);
   return out;
 }
-async function mediaRefToBytes(ref, fetch) {
+function mediaRefToDataUrl(ref) {
+  if (isUrlRef(ref)) return ref.url;
+  return `data:${ref.mime};base64,${mediaRefToBase64(ref.bytes)}`;
+}
+var refToUrl = mediaRefToDataUrl;
+async function readBody(res) {
+  const text2 = await res.text().catch(() => "");
+  try {
+    const decoded = JsonValueSchema.safeParse(JSON.parse(text2));
+    return decoded.success ? decoded.data : text2;
+  } catch {
+    return text2;
+  }
+}
+async function mediaRefToBytes(ref, fetch, options = {}) {
   if (!isUrlRef(ref)) return { bytes: ref.bytes, mime: ref.mime };
-  const res = await fetch(ref.url);
-  if (!res.ok) throw new Error(`reference fetch failed (${res.status}): ${ref.url}`);
+  const init = {};
+  if (options.signal !== void 0) init.signal = options.signal;
+  const res = await fetch(ref.url, init);
+  if (!res.ok) {
+    const text2 = await res.text().catch(() => "");
+    throw classifyHttpError(
+      res.status,
+      text2,
+      options.provider ?? "media-reference",
+      options.model
+    );
+  }
   const mime = res.headers.get("content-type")?.split(";")[0]?.trim().toLowerCase() || "image/png";
   return { bytes: new Uint8Array(await res.arrayBuffer()), mime };
 }
@@ -1687,7 +1859,7 @@ var import_perplexity = require("@ai-sdk/perplexity");
 var import_xai = require("@ai-sdk/xai");
 var import_deepseek = require("@ai-sdk/deepseek");
 var import_moonshotai = require("@ai-sdk/moonshotai");
-var import_zod2 = require("zod");
+var import_zod3 = require("zod");
 var TEXT_PROVIDERS = [
   "openai",
   "anthropic",
@@ -1698,7 +1870,7 @@ var TEXT_PROVIDERS = [
   "moonshot",
   "minimax"
 ];
-var textProviderSchema = import_zod2.z.enum(TEXT_PROVIDERS);
+var textProviderSchema = import_zod3.z.enum(TEXT_PROVIDERS);
 function isTextProvider(value) {
   return textProviderSchema.safeParse(value).success;
 }
@@ -1707,22 +1879,12 @@ var OPENAI_COMPATIBLE_BASE_URLS = {
 };
 var GATEWAY_SLUGS = {
   openai: "openai",
-  anthropic: "anthropic",
+  anthropic: "anthropic/v1",
   google: "google-ai-studio/v1beta",
-  xai: "grok",
+  xai: "grok/v1",
   perplexity: "perplexity-ai",
   deepseek: "deepseek"
 };
-var OPENROUTER_VENDORS = {
-  openai: "openai",
-  anthropic: "anthropic",
-  google: "google",
-  xai: "x-ai",
-  perplexity: "perplexity",
-  deepseek: "deepseek",
-  moonshot: "moonshotai",
-  minimax: "minimax"
-};
 function detectTextProvider(modelId) {
   const id = modelId.toLowerCase();
   if (id.startsWith("claude") || id.includes("anthropic.")) return "anthropic";
@@ -1739,14 +1901,17 @@ function detectTextProvider(modelId) {
     message: `ai-sdk-text cannot detect a provider for "${modelId}". Set providerOptions.provider to one of ${TEXT_PROVIDERS.join(", ")}.`
   });
 }
-function textKeyName(model, routing) {
-  if (routing.text === "openrouter") return "openrouter";
+function textKeyName(model, _routing) {
   return textProviderOf(model);
 }
 function textProviderOf(model) {
   const explicit = model.providerOptions?.["modelkit"]?.["provider"];
   return isTextProvider(explicit) ? explicit : detectTextProvider(model.providerModelId);
 }
+function textUsesCloudflareGateway(model, routing) {
+  if (routing.text !== "cloudflare") return false;
+  return GATEWAY_SLUGS[textProviderOf(model)] !== void 0;
+}
 function nativeClientFor(provider) {
   switch (provider) {
     case "openai":
@@ -1779,16 +1944,6 @@ function resolveTextRoute(ctx) {
       retryable: false
     });
   }
-  if (mode === "openrouter") {
-    const modelId = ctx.model.providerModelId.includes("/") ? ctx.model.providerModelId : `${OPENROUTER_VENDORS[provider]}/${ctx.model.providerModelId}`;
-    return {
-      provider,
-      client: "openai-compatible",
-      modelId,
-      keyName: "openrouter",
-      baseURL: "https://openrouter.ai/api/v1"
-    };
-  }
   const compatBase = OPENAI_COMPATIBLE_BASE_URLS[provider];
   const route2 = {
     provider,
@@ -1856,7 +2011,7 @@ var aiSdkTextAdapter = {
       case "openai-compatible":
         return (0, import_openai.createOpenAI)({
           ...settings,
-          name: route2.provider === "openai" ? "openai" : ctx.routing.text === "openrouter" ? "openrouter" : route2.provider
+          name: route2.provider
         }).chat(route2.modelId);
       case "openai":
         return (0, import_openai.createOpenAI)(settings)(route2.modelId);
@@ -1869,8 +2024,9 @@ function requiredKeys(model, config) {
   if (model.adapter === "fal" && config.routing?.fal === "cloudflare-passthrough")
     return ["cloudflare"];
   if (model.adapter === "ai-sdk-text") {
-    const names = [textKeyName(model, config.routing ?? {})];
-    if (config.routing?.text === "cloudflare") names.push("cloudflare");
+    const routing = config.routing ?? {};
+    const names = [textKeyName(model, routing)];
+    if (textUsesCloudflareGateway(model, routing)) names.push("cloudflare");
     return names;
   }
   return ADAPTER_KEYS[model.adapter];
@@ -1884,10 +2040,10 @@ function textAbortOptions(sdk, context) {
   if (abortSignal) return { abortSignal };
   return {};
 }
-var catalogId = import_zod3.z.string();
-var exactSize = import_zod3.z.templateLiteral([import_zod3.z.number(), "x", import_zod3.z.number()]);
-var sdkRatio = import_zod3.z.templateLiteral([import_zod3.z.number(), ":", import_zod3.z.number()]);
-var sizeTier = import_zod3.z.enum(["1K", "2K", "4K"]);
+var catalogId = import_zod4.z.string();
+var exactSize = import_zod4.z.templateLiteral([import_zod4.z.number(), "x", import_zod4.z.number()]);
+var sdkRatio = import_zod4.z.templateLiteral([import_zod4.z.number(), ":", import_zod4.z.number()]);
+var sizeTier = import_zod4.z.enum(["512", "1K", "2K", "4K"]);
 function isCatalogId(model) {
   return catalogId.safeParse(model).success;
 }
@@ -1944,6 +2100,15 @@ function pushIgnored(warnings, field, from, modelId) {
     message: `${modelId} provider instance does not support ${field}; ignored`
   });
 }
+function withoutModelkit(options) {
+  if (!options) return {};
+  const { modelkit: _modelkit, ...rest } = options;
+  return rest;
+}
+function logCoercions(logger, modelId, warnings) {
+  if (warnings.length === 0) return;
+  logger.warn(`[modelkit] ${modelId} request coerced`, { model: modelId, warnings });
+}
 function createModelKit(config, adapters) {
   const catalog = new Catalog(MODELS).withOverrides(config.catalogOverride ?? []);
   const logger = config.logger ?? console;
@@ -1956,7 +2121,16 @@ function createModelKit(config, adapters) {
         message: `No adapter registered for ${model.adapter}`,
         model: model.id
       });
-    const keys = resolveKeys(config, requiredKeys(model, config), call.tier);
+    const names = requiredKeys(model, config);
+    if (call.tier === "enterprise") {
+      const missing = missingEnterpriseKeys(config, names);
+      if (missing.length > 0)
+        logger.warn(
+          `[modelkit] enterprise keys incomplete for ${model.adapter}; using pro`,
+          { missing }
+        );
+    }
+    const keys = resolveKeys(config, names, call.tier);
     const ctx = {
       model,
       keys,
@@ -1977,6 +2151,10 @@ function createModelKit(config, adapters) {
       logger.warn("[modelkit] usage emit failed", err);
     }
   }
+  function withRowUnit(event) {
+    const row = catalog.get(event.model);
+    return row ? { ...event, unit: row.usage.unit } : event;
+  }
   function textTarget(model, call) {
     if (!isCatalogId(model)) {
       return {
@@ -2045,6 +2223,7 @@ function createModelKit(config, adapters) {
       pushIgnored(warnings, "outputFormat", _outputFormat, model.modelId);
     if (scaleFactor !== void 0)
       pushIgnored(warnings, "scaleFactor", scaleFactor, model.modelId);
+    logCoercions(logger, model.modelId, warnings);
     const abortSignal = combineSignals(req.abortSignal, call.signal);
     if (abortSignal) options.abortSignal = abortSignal;
     const result = await (0, import_ai2.generateImage)(options);
@@ -2128,6 +2307,7 @@ function createModelKit(config, adapters) {
     if (_mode !== void 0) pushIgnored(warnings, "mode", _mode, model.modelId);
     if (providerOptions) options.providerOptions = providerOptions;
     if (abortSignal) options.abortSignal = abortSignal;
+    logCoercions(logger, model.modelId, warnings);
     const pending = (0, import_ai2.experimental_generateVideo)(options);
     let settled;
     let failed = false;
@@ -2186,6 +2366,7 @@ function createModelKit(config, adapters) {
         call.onUnsupported ?? "error"
       );
       warnings.push(...inputWarnings);
+      logCoercions(logger, model.id, warnings);
       const signal = combineSignals(req.abortSignal, call.signal);
       const { adapter, ctx } = context(model, signal ? { ...call, signal } : call, warnings);
       if (!adapter.generateImage)
@@ -2195,7 +2376,7 @@ function createModelKit(config, adapters) {
           model: model.id
         });
       const result = await adapter.generateImage(request, ctx);
-      const usage2 = { ...result.usage, model: model.id };
+      const usage2 = { ...result.usage, model: model.id, unit: model.usage.unit };
       await emit(usage2);
       const images = result.images.map(toGeneratedImage);
       const first = images[0];
@@ -2228,6 +2409,7 @@ function createModelKit(config, adapters) {
         call.onUnsupported ?? "error"
       );
       warnings.push(...inputWarnings);
+      logCoercions(logger, model.id, warnings);
       const signal = combineSignals(req.abortSignal, call.signal);
       const { adapter, ctx } = context(model, signal ? { ...call, signal } : call, warnings);
       if (!adapter.generateVideo)
@@ -2245,7 +2427,11 @@ function createModelKit(config, adapters) {
         result(opts) {
           if (settled) return settled;
           const pending = result(opts).then(
-            (out) => emit({ ...out.usage, model: model.id }).then(() => out),
+            (out) => emit({
+              ...out.usage,
+              model: model.id,
+              unit: model.usage.unit
+            }).then(() => out),
             (cause) => {
               settled = void 0;
               throw cause;
@@ -2265,7 +2451,15 @@ function createModelKit(config, adapters) {
           model: model.id
         });
       }
-      const { adapter, ctx } = context(model, call, []);
+      const { abortSignal: _signal, ...request } = req;
+      const { request: resolved, warnings } = resolveAudioOptions(
+        model,
+        request,
+        call.onUnsupported ?? "error"
+      );
+      logCoercions(logger, model.id, warnings);
+      const signal = combineSignals(req.abortSignal, call.signal);
+      const { adapter, ctx } = context(model, signal ? { ...call, signal } : call, warnings);
       if (!adapter.generateAudio) {
         throw new ModelKitError({
           code: "unsupported_option",
@@ -2273,11 +2467,8 @@ function createModelKit(config, adapters) {
           model: model.id
         });
       }
-      const result = await adapter.generateAudio(
-        { ...model.defaults, ...req, model: model.id },
-        ctx
-      );
-      await emit({ ...result.usage, model: model.id });
+      const result = await adapter.generateAudio(resolved, ctx);
+      await emit({ ...result.usage, model: model.id, unit: model.usage.unit });
       return result;
     },
     async prepareMediaPart(modelId, ref, mediaType, call = {}) {
@@ -2297,7 +2488,7 @@ function createModelKit(config, adapters) {
         });
       return {
         model,
-        providerOptions: entry.providerOptions ?? {},
+        providerOptions: withoutModelkit(entry.providerOptions),
         catalog: entry
       };
     },
@@ -2317,7 +2508,7 @@ function createModelKit(config, adapters) {
           options.prepareStep ?? options.experimental_prepareStep
         )
       });
-      for (const event of steps.usage(result.steps)) await emit(event);
+      for (const event of steps.usage(result.steps)) await emit(withRowUnit(event));
       return result;
     },
     streamText(options, call = {}) {
@@ -2333,9 +2524,11 @@ function createModelKit(config, adapters) {
         ...textAbortOptions(options.abortSignal, call.signal),
         model: target.languageModel,
         providerOptions: {},
-        prepareStep: steps.wrap(options.prepareStep),
+        prepareStep: steps.wrap(
+          options.prepareStep ?? options.experimental_prepareStep
+        ),
         onFinish: async (event) => {
-          for (const usage2 of steps.usage(event.steps)) await emit(usage2);
+          for (const usage2 of steps.usage(event.steps)) await emit(withRowUnit(usage2));
           await callerOnFinish?.(event);
         }
       });
@@ -2349,7 +2542,7 @@ function createModelKit(config, adapters) {
         providerOptions: mergeTextProviderOptions(entry, options.providerOptions)
       };
       const result = await (0, import_ai2.generateObject)(sdkOptions);
-      await emit(textUsageEvent(entry, provider, result.usage));
+      await emit(withRowUnit(textUsageEvent(entry, provider, result.usage)));
       return result;
     },
     streamObject(options, call = {}) {
@@ -2361,7 +2554,7 @@ function createModelKit(config, adapters) {
         model: languageModel2,
         providerOptions: mergeTextProviderOptions(entry, options.providerOptions),
         onFinish: async (event) => {
-          await emit(textUsageEvent(entry, provider, event.usage));
+          await emit(withRowUnit(textUsageEvent(entry, provider, event.usage)));
           await callerOnFinish?.(event);
         }
       };
@@ -2371,7 +2564,7 @@ function createModelKit(config, adapters) {
 }
 
 // src/adapters/ark.ts
-var import_zod4 = require("zod");
+var import_zod5 = require("zod");
 var ARK_BASE_URL = "https://ark.ap-southeast.bytepluses.com/api/v3";
 var SUBMIT_PATH = "/contents/generations/tasks";
 var SEEDANCE_POLL_INTERVAL_MS = 15e3;
@@ -2390,6 +2583,7 @@ var SEEDREAM_SIZES = {
     "3:2": "1248x832",
     "2:3": "832x1248",
     "21:9": "1568x672",
+    "9:21": "672x1568",
     "4:5": "864x1080",
     "5:4": "1080x864"
   },
@@ -2402,16 +2596,34 @@ var SEEDREAM_SIZES = {
     "3:2": "2496x1664",
     "2:3": "1664x2496",
     "21:9": "3136x1344",
+    "9:21": "1344x3136",
     "4:5": "1728x2160",
     "5:4": "2160x1728"
+  },
+  // 4K keeps the long edge at Ark's 4096 cap (a doubled 2K entry would exceed
+  // it); sides are multiples of 16.
+  "4K": {
+    "1:1": "4096x4096",
+    "4:3": "4096x3072",
+    "3:4": "3072x4096",
+    "16:9": "4096x2304",
+    "9:16": "2304x4096",
+    "3:2": "4096x2736",
+    "2:3": "2736x4096",
+    "21:9": "4096x1760",
+    "9:21": "1760x4096",
+    "4:5": "3280x4096",
+    "5:4": "4096x3280"
   }
 };
 var arkAdapter = {
   id: "ark",
   async generateImage(req, ctx) {
+    throwIfAborted(ctx.signal);
     return generateSeedreamImage(req, ctx);
   },
   async generateVideo(req, ctx) {
+    throwIfAborted(ctx.signal);
     return generateSeedanceVideo(req, ctx);
   }
 };
@@ -2429,9 +2641,9 @@ function arkKey(ctx) {
   return key;
 }
 function throwArkError(status, body, model) {
-  const parsed = import_zod4.z.string().safeParse(body);
+  const parsed = import_zod5.z.string().safeParse(body);
   const text2 = parsed.success ? parsed.data : JSON.stringify(body ?? "");
-  if (/sensitive.?content|inputimage|human.?face/i.test(text2)) {
+  if (/sensitive.?content|human.?face/i.test(text2)) {
     const init = {
       code: "content_blocked",
       message: `ark rejected the request: ${text2.slice(0, 500)}`,
@@ -2443,10 +2655,20 @@ function throwArkError(status, body, model) {
     if (status !== void 0) init.status = status;
     throw new ModelKitError(init);
   }
+  if (status === void 0) {
+    throw new ModelKitError({
+      code: "provider_rejected",
+      message: `ark rejected the request: ${text2.slice(0, 500)}`,
+      provider: "ark",
+      model,
+      body,
+      retryable: false
+    });
+  }
   throw classifyHttpError(status, body, "ark", model);
 }
 function seedreamSize(aspect, size) {
-  const tier = size === "1K" ? "1K" : "2K";
+  const tier = size === "4K" || size === "1K" ? size : "2K";
   const table = SEEDREAM_SIZES[tier];
   return table[aspect ?? ""] ?? table["1:1"] ?? "2048x2048";
 }
@@ -2457,17 +2679,13 @@ function sniffedImageMime(bytes) {
     return "image/webp";
   return "image/jpeg";
 }
-function arkUrl(ref) {
-  if ("url" in ref) return ref.url;
-  return `data:${ref.mime};base64,${Buffer.from(ref.bytes).toString("base64")}`;
-}
-var SeedreamImageSchema = import_zod4.z.object({
-  url: import_zod4.z.string().optional(),
-  b64_json: import_zod4.z.string().optional()
+var SeedreamImageSchema = import_zod5.z.object({
+  url: import_zod5.z.string().optional(),
+  b64_json: import_zod5.z.string().optional()
 });
-var SeedreamResponseSchema = import_zod4.z.object({
-  data: import_zod4.z.array(SeedreamImageSchema).optional(),
-  error: import_zod4.z.object({ message: import_zod4.z.string().optional(), code: import_zod4.z.string().optional() }).optional()
+var SeedreamResponseSchema = import_zod5.z.object({
+  data: import_zod5.z.array(SeedreamImageSchema).optional(),
+  error: import_zod5.z.object({ message: import_zod5.z.string().optional(), code: import_zod5.z.string().optional() }).optional()
 });
 function retryOptsFor(ctx, attempts = 3) {
   const opts = { attempts };
@@ -2478,11 +2696,16 @@ function fetchInitFor(ctx, init) {
   if (ctx.signal !== void 0) init.signal = ctx.signal;
   return init;
 }
+function arkFetch(ctx, url, init) {
+  return ctx.fetch(url, fetchInitFor(ctx, init)).catch((cause) => {
+    throw classifyThrown("ark", cause, ctx.model.id);
+  });
+}
 async function generateSeedreamImage(req, ctx) {
   const key = arkKey(ctx);
   const modelId = ctx.model.id;
   const size = seedreamSize(req.aspectRatio, req.size);
-  const refs = (req.references ?? []).map(arkUrl);
+  const refs = (req.references ?? []).map(refToUrl);
   const body = {
     model: ctx.model.providerModelId,
     prompt: req.prompt,
@@ -2495,14 +2718,11 @@ async function generateSeedreamImage(req, ctx) {
   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 ctx.fetch(
-      `${arkBaseUrl(ctx)}/images/generations`,
-      fetchInitFor(ctx, {
-        method: "POST",
-        headers: { Authorization: `Bearer ${key}`, "Content-Type": "application/json" },
-        body: JSON.stringify(body)
-      })
-    );
+    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);
@@ -2517,7 +2737,7 @@ async function generateSeedreamImage(req, ctx) {
     bytes = Buffer.from(first.b64_json, "base64");
   } else if (first?.url) {
     const imageUrl = first.url;
-    const res = await ctx.fetch(imageUrl, fetchInitFor(ctx, {}));
+    const res = await arkFetch(ctx, imageUrl, {});
     if (!res.ok) {
       const text2 = await res.text().catch(() => "");
       throwArkError(res.status, text2, modelId);
@@ -2537,13 +2757,21 @@ async function generateSeedreamImage(req, ctx) {
   if (req.quality !== void 0) usage2.imageQuality = req.quality;
   return { images: [{ bytes, mime: sniffedImageMime(bytes) }], usage: usage2, raw: data };
 }
-var SeedanceResponseSchema = import_zod4.z.object({
-  id: import_zod4.z.string().optional(),
-  status: import_zod4.z.string().optional(),
-  content: import_zod4.z.object({ video_url: import_zod4.z.string().optional() }).optional(),
-  error: import_zod4.z.union([import_zod4.z.string(), import_zod4.z.object({ message: import_zod4.z.string().optional() })]).optional()
+var SeedanceUsageSchema = import_zod5.z.object({
+  total_tokens: import_zod5.z.number().optional(),
+  completion_tokens: import_zod5.z.number().optional()
 });
-var SeedanceErrorMessageSchema = import_zod4.z.object({ message: import_zod4.z.string().optional() });
+var SeedanceResponseSchema = import_zod5.z.object({
+  id: import_zod5.z.string().optional(),
+  status: import_zod5.z.string().optional(),
+  content: import_zod5.z.object({ video_url: import_zod5.z.string().optional() }).optional(),
+  usage: SeedanceUsageSchema.optional(),
+  duration: import_zod5.z.number().optional(),
+  framespersecond: import_zod5.z.number().optional(),
+  resolution: import_zod5.z.string().optional(),
+  error: import_zod5.z.union([import_zod5.z.string(), import_zod5.z.object({ message: import_zod5.z.string().optional() })]).optional()
+});
+var SeedanceErrorMessageSchema = import_zod5.z.object({ message: import_zod5.z.string().optional() });
 function seedanceUsage(ctx, req, duration) {
   const usage2 = {
     model: ctx.model.id,
@@ -2557,7 +2785,7 @@ function seedanceUsage(ctx, req, duration) {
   return usage2;
 }
 function seedanceErrorMessage(error, status) {
-  const asString = import_zod4.z.string().safeParse(error);
+  const asString = import_zod5.z.string().safeParse(error);
   if (asString.success) return asString.data;
   const asObject = SeedanceErrorMessageSchema.safeParse(error);
   if (asObject.success && asObject.data.message !== void 0) return asObject.data.message;
@@ -2572,42 +2800,42 @@ async function generateSeedanceVideo(req, ctx) {
   if (req.firstFrame) {
     content.push({
       type: "image_url",
-      image_url: { url: arkUrl(req.firstFrame) },
+      image_url: { url: refToUrl(req.firstFrame) },
       role: "first_frame"
     });
   }
   if (req.lastFrame) {
     content.push({
       type: "image_url",
-      image_url: { url: arkUrl(req.lastFrame) },
+      image_url: { url: refToUrl(req.lastFrame) },
       role: "last_frame"
     });
   }
   for (const ref of req.referenceImages ?? []) {
     content.push({
       type: "image_url",
-      image_url: { url: arkUrl(ref) },
+      image_url: { url: refToUrl(ref) },
       role: "reference_image"
     });
   }
   if (req.sourceVideo) {
     content.push({
       type: "video_url",
-      video_url: { url: arkUrl(req.sourceVideo) },
+      video_url: { url: refToUrl(req.sourceVideo) },
       role: "reference_video"
     });
   }
   for (const ref of req.referenceVideos ?? []) {
     content.push({
       type: "video_url",
-      video_url: { url: arkUrl(ref) },
+      video_url: { url: refToUrl(ref) },
       role: "reference_video"
     });
   }
   for (const ref of req.referenceAudios ?? []) {
     content.push({
       type: "audio_url",
-      audio_url: { url: arkUrl(ref) },
+      audio_url: { url: refToUrl(ref) },
       role: "reference_audio"
     });
   }
@@ -2621,15 +2849,13 @@ async function generateSeedanceVideo(req, ctx) {
   if (!hasFrameAnchor && req.aspectRatio !== void 0) body.ratio = req.aspectRatio;
   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 ctx.fetch(
-      `${arkBaseUrl(ctx)}${SUBMIT_PATH}`,
-      fetchInitFor(ctx, {
-        method: "POST",
-        headers: { Authorization: `Bearer ${key}`, "Content-Type": "application/json" },
-        body: JSON.stringify(body)
-      })
-    );
+    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);
@@ -2642,12 +2868,9 @@ async function generateSeedanceVideo(req, ctx) {
   const jobId = submit.id;
   const usage2 = seedanceUsage(ctx, req, duration);
   const fetchStatus = async () => {
-    const res = await ctx.fetch(
-      `${arkBaseUrl(ctx)}${SUBMIT_PATH}/${jobId}`,
-      fetchInitFor(ctx, {
-        headers: { Authorization: `Bearer ${key}` }
-      })
-    );
+    const res = await arkFetch(ctx, `${arkBaseUrl(ctx)}${SUBMIT_PATH}/${jobId}`, {
+      headers: { Authorization: `Bearer ${key}` }
+    });
     if (!res.ok) {
       if (res.status === 429 || res.status >= 500) return void 0;
       const text2 = await res.text().catch(() => "");
@@ -2690,6 +2913,11 @@ async function generateSeedanceVideo(req, ctx) {
               "Seedance task succeeded but returned no video_url",
               modelId
             );
+          const tokens = polled.usage;
+          if (tokens?.total_tokens !== void 0 && tokens.completion_tokens !== void 0) {
+            usage2.inputTokens = tokens.total_tokens - tokens.completion_tokens;
+            usage2.outputTokens = tokens.completion_tokens;
+          }
           return {
             video: { url, expiresAt: new Date(Date.now() + VIDEO_URL_TTL_MS) },
             usage: usage2,
@@ -2708,43 +2936,6 @@ async function generateSeedanceVideo(req, ctx) {
 // src/adapters/fal.ts
 var import_zod6 = require("zod");
 var import_client = require("@fal-ai/client");
-
-// src/json.ts
-var import_zod5 = require("zod");
-var JsonValueSchema = import_zod5.z.lazy(
-  () => import_zod5.z.union([
-    import_zod5.z.string(),
-    import_zod5.z.number(),
-    import_zod5.z.boolean(),
-    import_zod5.z.null(),
-    import_zod5.z.array(JsonValueSchema),
-    import_zod5.z.record(import_zod5.z.string(), JsonValueSchema)
-  ])
-);
-var JsonObjectSchema = import_zod5.z.record(import_zod5.z.string(), JsonValueSchema);
-async function readErrorBody(res) {
-  const text2 = await res.text().catch(() => "");
-  try {
-    return JsonValueSchema.parse(JSON.parse(text2));
-  } catch {
-    return text2;
-  }
-}
-var thrownErrorSchema = import_zod5.z.object({
-  message: import_zod5.z.string().optional(),
-  status: import_zod5.z.number().optional(),
-  statusCode: import_zod5.z.number().optional(),
-  responseBody: import_zod5.z.string().optional()
-});
-function describeThrown(cause) {
-  const parsed = thrownErrorSchema.safeParse(cause);
-  if (!parsed.success) return { message: String(cause), status: void 0, body: String(cause) };
-  const { message, status, statusCode, responseBody } = parsed.data;
-  const body = responseBody ?? message ?? String(cause);
-  return { message: message ?? String(cause), status: status ?? statusCode, body };
-}
-
-// 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";
@@ -2756,10 +2947,33 @@ function defaultCreateFalClient(opts) {
   const client2 = (0, import_client.createFalClient)(opts);
   return {
     async subscribe(endpoint, options) {
-      const outcome = await client2.subscribe(endpoint, { input: options.input });
+      const outcome = await client2.subscribe(endpoint, options);
       const unwrapped = falSubscribePayloadSchema.safeParse(outcome);
       if (unwrapped.success) return unwrapped.data.data;
       return JsonValueSchema.parse(outcome);
+    },
+    queue: {
+      async submit(endpoint, options) {
+        const submitted = await client2.queue.submit(endpoint, {
+          input: options.input,
+          ...options.abortSignal !== void 0 ? { abortSignal: options.abortSignal } : {}
+        });
+        return { requestId: submitted.request_id };
+      },
+      async status(endpoint, options) {
+        const current = await client2.queue.status(endpoint, {
+          requestId: options.requestId,
+          ...options.abortSignal !== void 0 ? { abortSignal: options.abortSignal } : {}
+        });
+        return { status: current.status };
+      },
+      async result(endpoint, options) {
+        const outcome = await client2.queue.result(endpoint, {
+          requestId: options.requestId,
+          ...options.abortSignal !== void 0 ? { abortSignal: options.abortSignal } : {}
+        });
+        return JsonValueSchema.parse(outcome.data);
+      }
     }
   };
 }
@@ -2767,24 +2981,17 @@ var defaultDeps = { createFalClient: defaultCreateFalClient };
 function createFalAdapter(deps = defaultDeps) {
   return {
     id: "fal",
-    generateImage: (req, ctx) => generateMuseImage(req, ctx, deps),
-    generateVideo: (req, ctx) => generateFalVideo(req, ctx, deps)
+    generateImage: (req, ctx) => {
+      throwIfAborted(ctx.signal);
+      return generateMuseImage(req, ctx, deps);
+    },
+    generateVideo: (req, ctx) => {
+      throwIfAborted(ctx.signal);
+      return generateFalVideo(req, ctx, deps);
+    }
   };
 }
 var falAdapter = createFalAdapter();
-function refToUrl(ref) {
-  if ("url" in ref) return ref.url;
-  return `data:${ref.mime};base64,${Buffer.from(ref.bytes).toString("base64")}`;
-}
-async function readBody(res) {
-  const text2 = await res.text().catch(() => "");
-  try {
-    const decoded = JsonValueSchema.safeParse(JSON.parse(text2));
-    return decoded.success ? decoded.data : text2;
-  } catch {
-    return text2;
-  }
-}
 var falDetailSchema = import_zod6.z.object({ detail: JsonValueSchema });
 function liftDetail(body) {
   const parsed = falDetailSchema.safeParse(body);
@@ -2801,15 +3008,17 @@ var falSdkThrownSchema = import_zod6.z.object({
   message: import_zod6.z.string().optional()
 });
 function falSdkError(cause, model) {
-  if (isModelKitError(cause)) return cause;
   const parsed = falSdkThrownSchema.safeParse(cause);
   if (parsed.success) {
     const status = parsed.data.status ?? parsed.data.statusCode;
     const body = parsed.data.body ?? parsed.data.message ?? String(cause);
-    return falError(status, body, model);
+    return classifyThrown("fal", cause, model, { status, body: liftDetail(body) });
   }
   const info = describeThrown(cause);
-  return falError(info.status, info.body, model);
+  return classifyThrown("fal", cause, model, {
+    status: info.status,
+    body: liftDetail(info.body)
+  });
 }
 function gatewayConfig(ctx) {
   const cf = ctx.routing.cloudflare;
@@ -2891,7 +3100,7 @@ async function passthroughRequest(ctx, endpoint, input) {
   }
   return JsonValueSchema.parse(await resultRes.json());
 }
-async function subscribe(ctx, deps, endpoint, input) {
+function falClientOf(ctx, deps) {
   const falKey = ctx.keys.fal;
   if (!falKey) {
     throw new ModelKitError({
@@ -2902,12 +3111,19 @@ async function subscribe(ctx, deps, endpoint, input) {
       retryable: false
     });
   }
-  const client2 = deps.createFalClient({ credentials: falKey });
+  return deps.createFalClient({ credentials: falKey, fetch: ctx.fetch });
+}
+async function subscribe(ctx, deps, endpoint, input) {
+  const client2 = falClientOf(ctx, deps);
+  const options = {
+    input,
+    ...ctx.signal !== void 0 ? { abortSignal: ctx.signal } : {}
+  };
   try {
     return await withRetry(
       async () => {
         try {
-          return await client2.subscribe(endpoint, { input });
+          return await client2.subscribe(endpoint, options);
         } catch (cause) {
           throw falSdkError(cause, ctx.model.id);
         }
@@ -3012,8 +3228,12 @@ async function generateMuseImage(req, ctx, deps) {
     raw: payload
   };
 }
+var H3_MAX_MODEL_IDS = /* @__PURE__ */ new Set([
+  "minimax/h3-max",
+  "minimax/h3-max/reference-to-video"
+]);
 function minimaxFamily(providerModelId) {
-  return providerModelId.toLowerCase().includes("h3-max") ? "h3-max" : "h3";
+  return H3_MAX_MODEL_IDS.has(providerModelId.toLowerCase()) ? "h3-max" : "h3";
 }
 function minimaxBase(req, family) {
   const resolution = (family === "h3-max" ? req.resolution ?? "768p" : req.resolution ?? "2k").toUpperCase();
@@ -3021,14 +3241,17 @@ function minimaxBase(req, family) {
     prompt: req.prompt.trim(),
     duration: req.durationSeconds ?? 5,
     resolution,
-    prompt_expansion_mode: req.promptExpansion ?? "balanced",
-    enable_safety_checker: true
+    prompt_expansion_mode: req.promptExpansion ?? "balanced"
   };
+  if (req.safetyChecker !== void 0) input.enable_safety_checker = req.safetyChecker;
   if (req.seed !== void 0) input.seed = req.seed;
   return input;
 }
+var KLING_MODEL_IDS = /* @__PURE__ */ new Set([
+  "fal-ai/kling-video/o1/video-to-video/reference"
+]);
 function isKling(providerModelId) {
-  return providerModelId.toLowerCase().includes("kling");
+  return KLING_MODEL_IDS.has(providerModelId.toLowerCase());
 }
 function klingO1Input(req) {
   const source = req.sourceVideo ? refToUrl(req.sourceVideo) : void 0;
@@ -3041,13 +3264,14 @@ function klingO1Input(req) {
     });
   }
   const duration = Math.min(10, Math.max(3, Math.round(req.durationSeconds ?? 5)));
-  const ratio = req.aspectRatio === "16:9" || req.aspectRatio === "9:16" || req.aspectRatio === "1:1" ? req.aspectRatio : "auto";
   const input = {
     prompt: req.prompt,
     video_url: source,
-    duration: String(duration),
-    aspect_ratio: ratio
+    duration: String(duration)
   };
+  if (req.aspectRatio !== "adaptive") {
+    input.aspect_ratio = req.aspectRatio === "16:9" || req.aspectRatio === "9:16" || req.aspectRatio === "1:1" ? req.aspectRatio : "auto";
+  }
   const refs = (req.referenceImages ?? []).map(refToUrl);
   if (refs.length > 0) input.image_urls = refs;
   return input;
@@ -3076,13 +3300,7 @@ async function generateFalVideo(req, ctx, deps) {
     if (ctx.routing.fal === "cloudflare-passthrough") {
       return await passthroughVideoJob(ctx, req, KLING_O1_ENDPOINT, input);
     }
-    const result2 = await runJob(ctx, deps, KLING_O1_ENDPOINT, input);
-    return completedVideoJob(
-      ctx,
-      req,
-      extractVideoUrl(result2, KLING_O1_ENDPOINT, ctx.model.id),
-      result2
-    );
+    return await submitQueueJob(ctx, req, deps, KLING_O1_ENDPOINT, input);
   }
   const family = minimaxFamily(providerModelId);
   const refs = {
@@ -3106,24 +3324,88 @@ async function generateFalVideo(req, ctx, deps) {
     if (lastFrame) base.end_image_url = lastFrame;
   } else {
     endpoint = `minimax/${family}/text-to-video`;
-    base.aspect_ratio = req.aspectRatio ?? "adaptive";
+    if (req.aspectRatio !== void 0 && req.aspectRatio !== "adaptive") {
+      base.aspect_ratio = req.aspectRatio;
+    }
   }
   if (ctx.routing.fal === "cloudflare-passthrough") {
     return await passthroughVideoJob(ctx, req, endpoint, base);
   }
-  const result = await runJob(ctx, deps, endpoint, base);
-  return completedVideoJob(ctx, req, extractVideoUrl(result, endpoint, ctx.model.id), result);
+  return await submitQueueJob(ctx, req, deps, endpoint, base);
+}
+async function submitQueueJob(ctx, req, deps, endpoint, input) {
+  const client2 = falClientOf(ctx, deps);
+  let submitted;
+  try {
+    submitted = await withRetry(
+      async () => {
+        try {
+          return await client2.queue.submit(endpoint, {
+            input,
+            ...ctx.signal !== void 0 ? { abortSignal: ctx.signal } : {}
+          });
+        } catch (cause) {
+          throw falSdkError(cause, ctx.model.id);
+        }
+      },
+      ctx.signal !== void 0 ? { signal: ctx.signal } : {}
+    );
+  } catch (cause) {
+    throw falSdkError(cause, ctx.model.id);
+  }
+  return queuedVideoJob(ctx, req, client2, endpoint, submitted.requestId);
 }
-function completedVideoJob(ctx, req, url, raw) {
+function queuedVideoJob(ctx, req, client2, endpoint, requestId) {
+  const readStatus = async () => {
+    try {
+      const options = {
+        requestId,
+        ...ctx.signal !== void 0 ? { abortSignal: ctx.signal } : {}
+      };
+      return (await client2.queue.status(endpoint, options)).status;
+    } catch (cause) {
+      throw falSdkError(cause, ctx.model.id);
+    }
+  };
   const duration = req.durationSeconds ?? 5;
-  const result = { video: { url }, usage: videoUsage(ctx, req, duration), raw };
   return {
-    jobId: url,
+    jobId: requestId,
     async poll() {
-      return { status: "succeeded", raw };
+      try {
+        const status = await readStatus();
+        if (status === "COMPLETED") return { status: "succeeded" };
+        if (status === "IN_QUEUE") return { status: "pending" };
+        return { status: "running" };
+      } catch (cause) {
+        if (isModelKitError(cause) && !cause.retryable)
+          return { status: "failed" };
+        return { status: "pending" };
+      }
     },
-    async result() {
-      return result;
+    async result(opts) {
+      const signal = opts?.signal ?? ctx.signal;
+      const pollOptions = {
+        intervalMs: opts?.pollIntervalMs ?? PASSTHROUGH_POLL_MS,
+        timeoutMs: opts?.timeoutMs ?? 15 * 6e4
+      };
+      if (signal !== void 0) pollOptions.signal = signal;
+      const queueOptions = {
+        requestId,
+        ...signal !== void 0 ? { abortSignal: signal } : {}
+      };
+      const payload = await pollUntil(async () => {
+        try {
+          const current = await client2.queue.status(endpoint, queueOptions);
+          if (current.status !== "COMPLETED") return void 0;
+          return await client2.queue.result(endpoint, queueOptions);
+        } catch (cause) {
+          const err = falSdkError(cause, ctx.model.id);
+          if (err.retryable) return void 0;
+          throw err;
+        }
+      }, pollOptions);
+      const url = extractVideoUrl(payload, endpoint, ctx.model.id);
+      return { video: { url }, usage: videoUsage(ctx, req, duration), raw: payload };
     }
   };
 }
@@ -3474,9 +3756,7 @@ function sdkStatus(cause) {
   return describeThrown(cause).status;
 }
 function throwSdkError(cause, model) {
-  if (cause instanceof ModelKitError) throw cause;
-  const info = describeThrown(cause);
-  throw classifyHttpError(info.status, info.body, "google-genai", model);
+  throw classifyThrown("google-genai", cause, model);
 }
 function toBase64(bytes) {
   return Buffer.from(bytes).toString("base64");
@@ -3532,6 +3812,14 @@ async function generateVeoVideo(req, ctx, createGenAIClient) {
       model: modelId
     });
   }
+  if (req.lastFrame !== void 0 && req.firstFrame === void 0) {
+    throw new ModelKitError({
+      code: "missing_input",
+      message: `${modelId} requires firstFrame when lastFrame is set`,
+      provider: "google-genai",
+      model: modelId
+    });
+  }
   const duration = req.durationSeconds ?? 8;
   const personGeneration = req.firstFrame ?? refs.length > 0 ? "allow_adult" : "allow_all";
   let image;
@@ -3598,7 +3886,7 @@ async function generateVeoVideo(req, ctx, createGenAIClient) {
       throwSdkError(cause, modelId);
     }
   };
-  const jobId = operation.name ?? "veo-operation";
+  const jobId = operation.name ?? crypto.randomUUID();
   const usage2 = videoUsage2(ctx, req, duration);
   return {
     jobId,
@@ -3680,12 +3968,17 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
   const duration = req.durationSeconds ?? 8;
   const images = req.referenceImages ?? [];
   const videos = req.referenceVideos ?? [];
-  const task = req.sourceVideo ? "edit" : videos.length > 0 || images.length > 1 ? "reference_to_video" : req.firstFrame || images.length === 1 ? "image_to_video" : "text_to_video";
-  const responseFormat = { type: "video" };
-  if (req.sourceVideo === void 0) responseFormat.aspect_ratio = toOmniAspect(req.aspectRatio);
-  const omniResolution = toOmniResolution(req.resolution);
-  if (omniResolution !== void 0) responseFormat.resolution = omniResolution;
-  responseFormat.duration = `${duration}s`;
+  const task = req.sourceVideo ? "edit" : req.firstFrame ? "image_to_video" : images.length > 0 || videos.length > 0 ? "reference_to_video" : "text_to_video";
+  const isOmniV1 = providerModelId === "gemini-omni-flash-preview";
+  const responseFormat = { type: "video", delivery: "uri" };
+  if (req.sourceVideo === void 0 && req.aspectRatio !== void 0 && req.aspectRatio !== "adaptive") {
+    responseFormat.aspect_ratio = toOmniAspect(req.aspectRatio);
+  }
+  if (!isOmniV1) {
+    const omniResolution = toOmniResolution(req.resolution);
+    if (omniResolution !== void 0) responseFormat.resolution = omniResolution;
+  }
+  if (req.durationSeconds !== void 0) responseFormat.duration = `${req.durationSeconds}s`;
   const client2 = createGenAIClient({ apiKey });
   const uploadedByUri = /* @__PURE__ */ new Map();
   const uploadedNames = [];
@@ -3709,7 +4002,8 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
     model: providerModelId,
     input,
     response_format: responseFormat,
-    generation_config: { video_config: { task } }
+    generation_config: { video_config: { task } },
+    store: true
   });
   const createRetryOptions = { attempts: 3 };
   if (ctx.signal !== void 0) createRetryOptions.signal = ctx.signal;
@@ -3731,6 +4025,7 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
       });
     }
     for (const ref of videos) {
+      if (ref.label !== void 0) parts.push({ type: "text", text: ref.label });
       const media = await resolveMedia(ref, ctx.fetch, ctx.signal, "video/mp4", modelId);
       parts.push({
         type: "video",
@@ -3739,6 +4034,8 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
       });
     }
     if (req.firstFrame) {
+      if (req.firstFrame.label !== void 0)
+        parts.push({ type: "text", text: req.firstFrame.label });
       const media = await resolveMedia(
         req.firstFrame,
         ctx.fetch,
@@ -3757,6 +4054,7 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
       }
     }
     for (const ref of images) {
+      if (ref.label !== void 0) parts.push({ type: "text", text: ref.label });
       const media = await resolveMedia(ref, ctx.fetch, ctx.signal, "image/png", modelId);
       if (media.bytes.length > OMNI_INLINE_LIMIT_BYTES) {
         parts.push({
@@ -3791,25 +4089,16 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
         };
         return inline;
       });
-      const inlineRetryOptions = { attempts: 3 };
-      if (ctx.signal !== void 0) inlineRetryOptions.signal = ctx.signal;
       try {
-        interaction = await withRetry(async () => {
-          try {
-            return await callCreate(inlineParts);
-          } catch (inlineCause) {
-            throwSdkError(inlineCause, modelId);
-          }
-        }, inlineRetryOptions);
+        interaction = await callCreate(inlineParts);
       } catch (inlineCause) {
         throwSdkError(inlineCause, modelId);
       }
     }
-  } catch (cause) {
+  } finally {
     await cleanupUploads();
-    throw cause;
   }
-  const jobId = interaction.id ?? "gemini-omni-interaction";
+  const jobId = interaction.id ?? crypto.randomUUID();
   const usage2 = videoUsage2(ctx, req, duration);
   const getInteraction = async () => {
     try {
@@ -3836,47 +4125,43 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
         timeoutMs: opts?.timeoutMs ?? RESULT_TIMEOUT_MS2
       };
       if (signal !== void 0) pollOptions.signal = signal;
-      try {
-        const done = await pollUntil(async () => {
-          const current = await getInteraction();
-          const status = (current.status ?? "").toUpperCase();
-          if (status === "COMPLETED") return current;
-          if (status === "FAILED") {
-            throw classifyHttpError(
-              void 0,
-              `Gemini Omni interaction failed: ${JSON.stringify(current).slice(0, 500)}`,
-              "google-genai",
-              modelId
-            );
-          }
-          return void 0;
-        }, pollOptions);
-        const video = extractOmniVideo(done, modelId);
-        if (video.data) {
-          return {
-            video: {
-              bytes: Buffer.from(video.data, "base64"),
-              mime: video.mime_type ?? "video/mp4"
-            },
-            usage: usage2,
-            raw: done
-          };
+      const done = await pollUntil(async () => {
+        const current = await getInteraction();
+        const status = (current.status ?? "").toUpperCase();
+        if (status === "COMPLETED") return current;
+        if (status === "FAILED") {
+          throw classifyHttpError(
+            void 0,
+            `Gemini Omni interaction failed: ${JSON.stringify(current).slice(0, 500)}`,
+            "google-genai",
+            modelId
+          );
         }
-        if (video.uri)
-          return {
-            video: await downloadVideo(ctx, video.uri, apiKey, signal),
-            usage: usage2,
-            raw: done
-          };
-        throw classifyHttpError(
-          void 0,
-          "No video content in Gemini Omni response",
-          "google-genai",
-          modelId
-        );
-      } finally {
-        await cleanupUploads();
+        return void 0;
+      }, pollOptions);
+      const video = extractOmniVideo(done, modelId);
+      if (video.data) {
+        return {
+          video: {
+            bytes: Buffer.from(video.data, "base64"),
+            mime: video.mime_type ?? "video/mp4"
+          },
+          usage: usage2,
+          raw: done
+        };
       }
+      if (video.uri)
+        return {
+          video: await downloadVideo(ctx, video.uri, apiKey, signal),
+          usage: usage2,
+          raw: done
+        };
+      throw classifyHttpError(
+        void 0,
+        "No video content in Gemini Omni response",
+        "google-genai",
+        modelId
+      );
     }
   };
 }
@@ -4039,44 +4324,37 @@ function createGoogleImageAdapter(deps = defaultGoogleImageDeps) {
       const model = google(ctx.model.providerModelId);
       const content = [];
       for (const ref of req.references ?? []) {
+        if (ref.label !== void 0) content.push({ type: "text", text: ref.label });
         if ("url" in ref) content.push({ type: "image", image: ref.url });
         else content.push({ type: "image", image: ref.bytes, mediaType: ref.mime });
       }
       content.push({ type: "text", text: req.prompt });
-      void ctx.call.postCropToAspect;
       const imageConfig = {};
-      if (req.aspectRatio !== void 0) imageConfig.aspectRatio = req.aspectRatio;
+      if (req.aspectRatio !== void 0 && req.aspectRatio !== "adaptive")
+        imageConfig.aspectRatio = req.aspectRatio;
       if (req.size !== void 0) imageConfig.imageSize = req.size;
+      const googleOptions = {
+        imageConfig,
+        responseModalities: ["IMAGE"]
+      };
+      if (req.safetySettings !== void 0) googleOptions.safetySettings = req.safetySettings;
       const retryOptions = {};
       if (ctx.signal !== void 0) retryOptions.signal = 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;
         try {
-          return await deps.runTextGeneration({
+          const genOptions = {
             model,
             messages: [{ role: "user", content }],
             maxRetries: 0,
             providerOptions: {
-              google: {
-                imageConfig,
-                responseModalities: ["IMAGE"],
-                safetySettings: [
-                  {
-                    category: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-                    threshold: "OFF"
-                  },
-                  { category: "HARM_CATEGORY_HATE_SPEECH", threshold: "OFF" },
-                  {
-                    category: "HARM_CATEGORY_DANGEROUS_CONTENT",
-                    threshold: "OFF"
-                  },
-                  { category: "HARM_CATEGORY_HARASSMENT", threshold: "OFF" }
-                ]
-              }
+              google: googleOptions
             },
             abortSignal: signal
-          });
+          };
+          if (req.temperature !== void 0) genOptions.temperature = req.temperature;
+          return await deps.runTextGeneration(genOptions);
         } catch (cause) {
           throw toProviderError(cause, ctx);
         }
@@ -4260,6 +4538,7 @@ var KLING_MAX_INTERVAL_MS = 1e4;
 var KLING_TIMEOUT_MS = 15 * 6e4;
 var DOP_INTERVAL_MS = 5e3;
 var DOP_TIMEOUT_MS = 5 * 6e4;
+var DOP_FALLBACK_DURATION_SECONDS = 5;
 var DEFAULT_NEGATIVE_PROMPT = "text, watermark, copyright, blur, low quality, distortion, abstract, weird, deformed";
 function credentials(ctx) {
   const key = ctx.keys.higgsfield;
@@ -4279,26 +4558,14 @@ function headers(ctx) {
   const { key, secret } = credentials(ctx);
   return { "Content-Type": "application/json", "hf-api-key": key, "hf-secret": secret };
 }
-function refToUrl2(ref) {
-  if ("url" in ref) return ref.url;
-  return `data:${ref.mime};base64,${Buffer.from(ref.bytes).toString("base64")}`;
-}
-async function readBody2(res) {
-  const text2 = await res.text().catch(() => "");
-  try {
-    const decoded = JsonValueSchema.safeParse(JSON.parse(text2));
-    return decoded.success ? decoded.data : text2;
-  } catch {
-    return text2;
-  }
-}
 function klingEndpoint(providerModelId) {
   if (providerModelId.startsWith("http")) return providerModelId;
   const path = providerModelId.startsWith("/") ? providerModelId.slice(1) : providerModelId;
   return `${PLATFORM}/${path}`;
 }
-function isDop(providerModelId) {
-  return providerModelId.toLowerCase().includes("dop");
+var DOP_MODEL_IDS = /* @__PURE__ */ new Set(["dop", "dop-preview"]);
+function higgsfieldVideoKind(providerModelId) {
+  return DOP_MODEL_IDS.has(providerModelId.toLowerCase()) ? "dop" : "kling";
 }
 function videoUsage3(ctx, req, duration) {
   const event = {
@@ -4334,8 +4601,20 @@ function requestInit(ctx, init) {
   if (ctx.signal !== void 0) request.signal = ctx.signal;
   return request;
 }
+async function createRequest(ctx, url, init) {
+  let res;
+  try {
+    res = await ctx.fetch(url, requestInit(ctx, init));
+  } catch (cause) {
+    throw classifyThrown("higgsfield", cause, ctx.model.id);
+  }
+  if (!res.ok) {
+    throw classifyHttpError(res.status, await readBody(res), "higgsfield", ctx.model.id);
+  }
+  return res;
+}
 async function generateKlingVideo(req, ctx) {
-  const firstFrame = req.firstFrame ? refToUrl2(req.firstFrame) : void 0;
+  const firstFrame = req.firstFrame ? refToUrl(req.firstFrame) : void 0;
   if (!firstFrame) {
     throw new ModelKitError({
       code: "missing_input",
@@ -4352,26 +4631,16 @@ async function generateKlingVideo(req, ctx) {
     negative_prompt: req.negativePrompt ?? DEFAULT_NEGATIVE_PROMPT
   };
   if (req.durationSeconds !== void 0) body.duration = req.durationSeconds;
-  if (req.aspectRatio !== void 0) body.aspect_ratio = req.aspectRatio;
-  if (req.lastFrame) body.last_image_url = refToUrl2(req.lastFrame);
+  if (req.aspectRatio !== void 0 && req.aspectRatio !== "adaptive")
+    body.aspect_ratio = req.aspectRatio;
+  if (req.lastFrame) body.last_image_url = refToUrl(req.lastFrame);
   const endpoint = klingEndpoint(ctx.model.providerModelId);
-  let createRes;
-  try {
-    createRes = await ctx.fetch(
-      endpoint,
-      requestInit(ctx, { method: "POST", body: JSON.stringify(body) })
-    );
-  } catch (cause) {
-    throw networkError(cause, ctx);
-  }
-  if (!createRes.ok) {
-    throw classifyHttpError(
-      createRes.status,
-      await readBody2(createRes),
-      "higgsfield",
-      ctx.model.id
-    );
-  }
+  const retryOptions = {};
+  if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+  const createRes = await withRetry(
+    () => createRequest(ctx, endpoint, { method: "POST", body: JSON.stringify(body) }),
+    retryOptions
+  );
   const created = klingCreateSchema.safeParse(await createRes.json());
   if (!created.success) {
     throw new ModelKitError({
@@ -4390,9 +4659,9 @@ async function generateKlingVideo(req, ctx) {
     } catch {
       return void 0;
     }
-    if (res.status >= 500) return void 0;
+    if (res.status === 429 || res.status >= 500) return void 0;
     if (!res.ok) {
-      throw classifyHttpError(res.status, await readBody2(res), "higgsfield", ctx.model.id);
+      throw classifyHttpError(res.status, await readBody(res), "higgsfield", ctx.model.id);
     }
     const payload = JsonValueSchema.parse(await res.json());
     const parsed = klingStatusSchema.safeParse(payload);
@@ -4451,11 +4720,6 @@ async function generateKlingVideo(req, ctx) {
     }
   };
 }
-function networkError(cause, ctx) {
-  if (isModelKitError(cause)) return cause;
-  const info = describeThrown(cause);
-  return classifyHttpError(info.status, info.body, "higgsfield", ctx.model.id);
-}
 var dopMotionOptionsSchema = import_zod10.z.object({
   modelkit: import_zod10.z.object({ motion: import_zod10.z.string().optional() }).optional()
 });
@@ -4504,7 +4768,7 @@ function dopUrl(body) {
 var dopCreateSchema = import_zod10.z.object({ id: import_zod10.z.string() });
 var dopStatusSchema = import_zod10.z.object({ status: import_zod10.z.string().optional() });
 async function generateDopVideo(req, ctx) {
-  const firstFrame = req.firstFrame ? refToUrl2(req.firstFrame) : void 0;
+  const firstFrame = req.firstFrame ? refToUrl(req.firstFrame) : void 0;
   if (!firstFrame) {
     throw new ModelKitError({
       code: "missing_input",
@@ -4524,29 +4788,18 @@ async function generateDopVideo(req, ctx) {
     webhook: null
   };
   if (req.lastFrame) {
-    params.input_images_end = [{ type: "image_url", image_url: refToUrl2(req.lastFrame) }];
+    params.input_images_end = [{ type: "image_url", image_url: refToUrl(req.lastFrame) }];
   }
   if (req.seed !== void 0) params.seed = req.seed;
-  let createRes;
-  try {
-    createRes = await ctx.fetch(
-      DOP_ENDPOINT,
-      requestInit(ctx, {
-        method: "POST",
-        body: JSON.stringify({ params, webhook: null })
-      })
-    );
-  } catch (cause) {
-    throw networkError(cause, ctx);
-  }
-  if (!createRes.ok) {
-    throw classifyHttpError(
-      createRes.status,
-      await readBody2(createRes),
-      "higgsfield",
-      ctx.model.id
-    );
-  }
+  const retryOptions = {};
+  if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
+  const createRes = await withRetry(
+    () => createRequest(ctx, DOP_ENDPOINT, {
+      method: "POST",
+      body: JSON.stringify({ params, webhook: null })
+    }),
+    retryOptions
+  );
   const created = dopCreateSchema.safeParse(await createRes.json());
   if (!created.success) {
     throw new ModelKitError({
@@ -4615,14 +4868,20 @@ async function generateDopVideo(req, ctx) {
       };
       if (dopSignal !== void 0) pollSettings.signal = dopSignal;
       const url = await pollUntil(check, pollSettings);
-      const result = { video: { url }, usage: videoUsage3(ctx, req) };
+      const result = {
+        video: { url },
+        usage: videoUsage3(ctx, req, req.durationSeconds ?? DOP_FALLBACK_DURATION_SECONDS)
+      };
       return result;
     }
   };
 }
 var higgsfieldAdapter = {
   id: "higgsfield",
-  generateVideo: (req, ctx) => isDop(ctx.model.providerModelId) ? generateDopVideo(req, ctx) : generateKlingVideo(req, ctx)
+  generateVideo: (req, ctx) => {
+    throwIfAborted(ctx.signal);
+    return higgsfieldVideoKind(ctx.model.providerModelId) === "dop" ? generateDopVideo(req, ctx) : generateKlingVideo(req, ctx);
+  }
 };
 
 // src/adapters/openai-image.ts
@@ -4717,6 +4976,7 @@ var openaiUsageSchema = import_zod11.z.object({
 var openaiImageAdapter = {
   id: "openai-image",
   async generateImage(req, ctx) {
+    throwIfAborted(ctx.signal);
     const apiKey = ctx.keys.openai;
     if (!apiKey)
       throw new ModelKitError({
@@ -4726,6 +4986,13 @@ var openaiImageAdapter = {
         model: ctx.model.id,
         retryable: false
       });
+    if ((req.references ?? []).some((ref) => ref.label !== void 0)) {
+      ctx.warnings.push({
+        code: "option_ignored",
+        field: "references.label",
+        message: "openai-image cannot express reference labels; ignoring them"
+      });
+    }
     const client2 = new import_openai2.default({ apiKey, fetch: ctx.fetch, maxRetries: 0 });
     const size = req.exactSize !== void 0 ? req.exactSize : req.aspectRatio === void 0 || req.aspectRatio === "adaptive" ? void 0 : aspectRatioToSize(req.aspectRatio, req.size ?? "1K");
     const refs = req.references ?? [];
@@ -4739,24 +5006,34 @@ var openaiImageAdapter = {
       try {
         if (!isEdit) {
           if (size !== void 0) {
-            return await client2.images.generate({
+            return await client2.images.generate(
+              {
+                model: ctx.model.providerModelId,
+                prompt: req.prompt,
+                size,
+                quality: sdkQuality,
+                ...count
+              },
+              { signal: ctx.signal }
+            );
+          }
+          return await client2.images.generate(
+            {
               model: ctx.model.providerModelId,
               prompt: req.prompt,
-              size,
               quality: sdkQuality,
               ...count
-            });
-          }
-          return await client2.images.generate({
-            model: ctx.model.providerModelId,
-            prompt: req.prompt,
-            quality: sdkQuality,
-            ...count
-          });
+            },
+            { signal: ctx.signal }
+          );
         }
         const files = await Promise.all(
           refs.slice(0, MAX_EDIT_REFERENCES).map(async (ref, i) => {
-            const { bytes, mime } = await mediaRefToBytes(ref, ctx.fetch);
+            const { bytes, mime } = await mediaRefToBytes(ref, ctx.fetch, {
+              signal: ctx.signal,
+              provider: "openai-image",
+              model: ctx.model.id
+            });
             const ext = mime.includes("jpeg") ? "jpg" : mime.includes("webp") ? "webp" : "png";
             return new File([bytes], `ref-${i}.${ext}`, { type: mime });
           })
@@ -4764,22 +5041,28 @@ var openaiImageAdapter = {
         const single = files.length === 1 ? files[0] : void 0;
         const imageInput = single ?? files;
         if (size !== void 0) {
-          return await client2.images.edit({
+          return await client2.images.edit(
+            {
+              model: ctx.model.providerModelId,
+              image: imageInput,
+              prompt: req.prompt,
+              size,
+              quality: sdkQuality,
+              ...count
+            },
+            { signal: ctx.signal }
+          );
+        }
+        return await client2.images.edit(
+          {
             model: ctx.model.providerModelId,
             image: imageInput,
             prompt: req.prompt,
-            size,
             quality: sdkQuality,
             ...count
-          });
-        }
-        return await client2.images.edit({
-          model: ctx.model.providerModelId,
-          image: imageInput,
-          prompt: req.prompt,
-          quality: sdkQuality,
-          ...count
-        });
+          },
+          { signal: ctx.signal }
+        );
       } catch (cause) {
         throw toProviderError2(cause, ctx);
       }
@@ -4797,28 +5080,16 @@ var openaiImageAdapter = {
     const usageParsed = openaiUsageSchema.safeParse(response.usage);
     const inputTokens = usageParsed.success ? usageParsed.data.input_tokens : void 0;
     const outputTokens = usageParsed.success ? usageParsed.data.output_tokens : void 0;
-    let usage2;
-    if (inputTokens !== void 0 || outputTokens !== void 0) {
-      usage2 = {
-        model: ctx.model.id,
-        provider: "openai-image",
-        modality: "IMAGE",
-        unit: "tokens",
-        quantity: (inputTokens ?? 0) + (outputTokens ?? 0),
-        imageQuality: quality
-      };
-      if (inputTokens !== void 0) usage2.inputTokens = inputTokens;
-      if (outputTokens !== void 0) usage2.outputTokens = outputTokens;
-    } else {
-      usage2 = {
-        model: ctx.model.id,
-        provider: "openai-image",
-        modality: "IMAGE",
-        unit: "images",
-        quantity: images.length,
-        imageQuality: quality
-      };
-    }
+    const usage2 = {
+      model: ctx.model.id,
+      provider: "openai-image",
+      modality: "IMAGE",
+      unit: "images",
+      quantity: images.length,
+      imageQuality: quality
+    };
+    if (inputTokens !== void 0) usage2.inputTokens = inputTokens;
+    if (outputTokens !== void 0) usage2.outputTokens = outputTokens;
     if (req.size !== void 0) usage2.imageSize = req.size;
     return { images, usage: usage2, raw: response };
   }
@@ -4871,7 +5142,12 @@ var NANO_BANANA_RATIOS = /* @__PURE__ */ new Set([
 ]);
 async function refToBytes(ctx, ref) {
   if ("bytes" in ref) return { bytes: ref.bytes, mime: ref.mime };
-  const res = await ctx.fetch(ref.url);
+  let res;
+  try {
+    res = await ctx.fetch(ref.url, ctx.signal ? { signal: ctx.signal } : {});
+  } catch (cause) {
+    throw classifyThrown("prodia", cause, ctx.model.id);
+  }
   if (!res.ok) {
     const text2 = await res.text().catch(() => "");
     throw classifyHttpError(res.status, text2, "prodia", ctx.model.id);
@@ -4914,6 +5190,12 @@ function emptyImage() {
     retryable: false
   });
 }
+function inputExtension(mime) {
+  if (mime.includes("jpeg") || mime.includes("jpg")) return "jpg";
+  if (mime.includes("webp")) return "webp";
+  if (mime.includes("gif")) return "gif";
+  return "png";
+}
 function isBgRemove(providerModelId) {
   const id = providerModelId.toLowerCase();
   return id.includes("remove-background") || id.includes("bg-remove");
@@ -4922,6 +5204,7 @@ function isKlein(providerModelId) {
   return providerModelId.toLowerCase().includes("klein");
 }
 async function generateProdiaImage(req, ctx) {
+  throwIfAborted(ctx.signal);
   const key = ctx.keys.prodia;
   if (!key) {
     throw new ModelKitError({
@@ -5020,7 +5303,9 @@ async function generateProdiaImage(req, ctx) {
       "input",
       // Uint8Array<ArrayBufferLike> is not a valid BlobPart under the
       // strict lib types; copy into a fresh ArrayBuffer-backed view.
-      new File([new Uint8Array(input.bytes)], `input-${i}.jpg`, { type: "image/jpeg" })
+      new File([new Uint8Array(input.bytes)], `input-${i}.${inputExtension(input.mime)}`, {
+        type: input.mime
+      })
     );
   });
   const retryOpts = {
@@ -5035,14 +5320,15 @@ async function generateProdiaImage(req, ctx) {
     res = await withRetry(async () => {
       let attemptRes;
       try {
-        attemptRes = await ctx.fetch(PRODIA_JOB_URL, {
+        const init = {
           method: "POST",
           headers: { Authorization: `Bearer ${key}`, Accept: accept },
           body: form
-        });
+        };
+        if (ctx.signal !== void 0) init.signal = ctx.signal;
+        attemptRes = await ctx.fetch(PRODIA_JOB_URL, init);
       } catch (err) {
-        if (isModelKitError(err)) throw err;
-        throw transportError("prodia", err, ctx.model.id);
+        throw classifyThrown("prodia", err, ctx.model.id);
       }
       if (!attemptRes.ok) {
         const text2 = await attemptRes.text().catch(() => "");
@@ -5051,8 +5337,7 @@ async function generateProdiaImage(req, ctx) {
       return attemptRes;
     }, retryOpts);
   } catch (err) {
-    if (isModelKitError(err)) throw err;
-    throw transportError("prodia", err, ctx.model.id);
+    throw classifyThrown("prodia", err, ctx.model.id);
   }
   const bytes = parseImageResponse(
     new Uint8Array(await res.arrayBuffer()),
@@ -5102,11 +5387,7 @@ function client(ctx, deps) {
       retryable: false
     });
   }
-  return deps.createReplicateClient({ auth: token });
-}
-function refToUrl3(ref) {
-  if ("url" in ref) return ref.url;
-  return `data:${ref.mime};base64,${Buffer.from(ref.bytes).toString("base64")}`;
+  return deps.createReplicateClient({ auth: token, fetch: ctx.fetch });
 }
 function route(providerModelId) {
   if (providerModelId.includes(":")) {
@@ -5124,10 +5405,11 @@ function replicateError(cause, ctx) {
   const info = describeThrown(cause);
   const nested = replicateResponseStatusSchema.safeParse(cause);
   const status = info.status ?? (nested.success ? nested.data.response?.status : void 0);
-  return classifyHttpError(status, info.body, "replicate", ctx.model.id);
+  return classifyThrown("replicate", cause, ctx.model.id, { status, body: info.body });
 }
-function isLower(id) {
-  return id.toLowerCase();
+function baseModelId(providerModelId) {
+  const colon = providerModelId.indexOf(":");
+  return colon === -1 ? providerModelId : providerModelId.slice(0, colon);
 }
 function predictionErrorBody(cause) {
   const parsed = JsonValueSchema.safeParse(cause);
@@ -5175,6 +5457,7 @@ function noOutput() {
   });
 }
 async function runPrediction(ctx, input, deps) {
+  throwIfAborted(ctx.signal);
   const api = client(ctx, deps);
   const createOptions = { ...route(ctx.model.providerModelId), input };
   let prediction;
@@ -5182,12 +5465,14 @@ async function runPrediction(ctx, input, deps) {
     prediction = await withRetry(
       async () => {
         try {
-          return await api.predictions.create(createOptions);
+          return await api.predictions.create(
+            ctx.signal ? { ...createOptions, signal: ctx.signal } : createOptions
+          );
         } catch (cause) {
           throw replicateError(cause, ctx);
         }
       },
-      ctx.signal ? { signal: ctx.signal } : {}
+      { attempts: 1 }
     );
   } catch (cause) {
     throw replicateError(cause, ctx);
@@ -5200,7 +5485,10 @@ async function runPrediction(ctx, input, deps) {
   const done = await pollUntil(async () => {
     let current;
     try {
-      current = await api.predictions.get(prediction.id);
+      current = await api.predictions.get(
+        prediction.id,
+        ctx.signal ? { signal: ctx.signal } : void 0
+      );
     } catch (cause) {
       throw replicateError(cause, ctx);
     }
@@ -5267,21 +5555,32 @@ async function download(ctx, url) {
   const mime = res.headers.get("content-type") ?? "image/png";
   return { bytes: new Uint8Array(await res.arrayBuffer()), mime };
 }
-function isUpscaler(providerModelId) {
-  const id = isLower(providerModelId);
-  return id.includes("upscal") || id.includes("crystal") || id.includes("topaz");
-}
-function isTopazUpscaler(providerModelId) {
-  return isLower(providerModelId).includes("topazlabs/image-upscale");
-}
-function isQwen(providerModelId) {
-  return isLower(providerModelId).includes("qwen");
-}
-function isWanImage(providerModelId) {
-  return isLower(providerModelId).includes("wan-2.7");
+var REPLICATE_IMAGE_KINDS = {
+  "topazlabs/image-upscale": "topaz",
+  "philz1337x/crystal-upscaler": "upscaler",
+  "wan-video/wan-2.7-image-pro": "wan",
+  "qwen/qwen-image-2-pro": "qwen-image",
+  "qwen-edit-apps/qwen-image-edit-plus-lora-skin": "qwen-edit"
+};
+function imageKind(providerModelId) {
+  return REPLICATE_IMAGE_KINDS[baseModelId(providerModelId)] ?? "generic";
+}
+var REPLICATE_VIDEO_KINDS = {
+  "kling-video/v2.5-turbo/pro/image-to-video": "kling",
+  "kling-video/v3.0/pro/image-to-video": "kling",
+  "kwaivgi/kling-v3-video": "kling",
+  "kwaivgi/kling-v3-omni-video": "kling-omni",
+  "minimax/hailuo-02": "hailuo"
+};
+function videoKind(providerModelId) {
+  return REPLICATE_VIDEO_KINDS[baseModelId(providerModelId)] ?? "wan";
 }
-function isQwenEditPlus(providerModelId) {
-  return isLower(providerModelId).includes("edit");
+var REPLICATE_AUDIO_KINDS = {
+  "zsxkib/mmaudio": "mmaudio",
+  "elevenlabs/music": "elevenlabs"
+};
+function audioKind(providerModelId) {
+  return REPLICATE_AUDIO_KINDS[baseModelId(providerModelId)] ?? "lyria";
 }
 var QWEN_IMAGE_RATIOS = /* @__PURE__ */ new Set([
   "1:1",
@@ -5324,87 +5623,113 @@ function wanImageSize(aspectRatio, size) {
 async function generateReplicateImage(req, ctx, deps) {
   const id = ctx.model.providerModelId;
   let input;
-  if (isTopazUpscaler(id)) {
-    const first = (req.references ?? [])[0];
-    if (!first) {
-      throw new ModelKitError({
-        code: "missing_input",
-        message: `${ctx.model.id} requires a reference image to upscale`,
-        provider: "replicate",
-        model: ctx.model.id,
-        retryable: false
-      });
+  switch (imageKind(id)) {
+    case "topaz": {
+      const first = (req.references ?? [])[0];
+      if (!first) {
+        throw new ModelKitError({
+          code: "missing_input",
+          message: `${ctx.model.id} requires a reference image to upscale`,
+          provider: "replicate",
+          model: ctx.model.id,
+          retryable: false
+        });
+      }
+      let format = req.outputFormat === "jpeg" ? "jpg" : req.outputFormat ?? "png";
+      if (format !== "png" && format !== "jpg") {
+        ctx.warnings.push({
+          code: "option_ignored",
+          field: "outputFormat",
+          from: format,
+          message: "topazlabs/image-upscale supports only png and jpg; using png"
+        });
+        format = "png";
+      }
+      const upscale = req.upscale;
+      const topaz = {
+        image: refToUrl(first),
+        enhance_model: upscale?.enhanceModel ?? "Standard V2",
+        output_format: format,
+        upscale_factor: `${req.scaleFactor ?? 2}x`,
+        face_enhancement: upscale?.faceEnhancement ?? false
+      };
+      if (upscale?.subjectDetection !== void 0)
+        topaz.subject_detection = upscale.subjectDetection;
+      if (upscale?.faceEnhancementCreativity !== void 0)
+        topaz.face_enhancement_creativity = upscale.faceEnhancementCreativity;
+      if (upscale?.faceEnhancementStrength !== void 0)
+        topaz.face_enhancement_strength = upscale.faceEnhancementStrength;
+      input = topaz;
+      break;
     }
-    let format = req.outputFormat === "jpeg" ? "jpg" : req.outputFormat ?? "png";
-    if (format !== "png" && format !== "jpg") {
-      ctx.warnings.push({
-        code: "option_ignored",
-        field: "outputFormat",
-        from: format,
-        message: "topazlabs/image-upscale supports only png and jpg; using png"
-      });
-      format = "png";
+    case "upscaler": {
+      const first = (req.references ?? [])[0];
+      if (!first) {
+        throw new ModelKitError({
+          code: "missing_input",
+          message: `${ctx.model.id} requires a reference image to upscale`,
+          provider: "replicate",
+          model: ctx.model.id,
+          retryable: false
+        });
+      }
+      const imageInput = { prompt: req.prompt };
+      imageInput.image = refToUrl(first);
+      imageInput.creativity = 0;
+      imageInput.scale_factor = 2;
+      imageInput.output_format = "png";
+      input = imageInput;
+      break;
     }
-    input = {
-      image: refToUrl3(first),
-      enhance_model: "Standard V2",
-      output_format: format,
-      upscale_factor: `${req.scaleFactor ?? 2}x`,
-      face_enhancement: true
-    };
-  } else if (isUpscaler(id)) {
-    const refs = req.references ?? [];
-    const first = refs[0];
-    if (!first) {
-      throw new ModelKitError({
-        code: "missing_input",
-        message: `${ctx.model.id} requires a reference image to upscale`,
-        provider: "replicate",
-        model: ctx.model.id,
-        retryable: false
-      });
+    case "wan": {
+      const refs = (req.references ?? []).map(refToUrl);
+      const imageInput = { prompt: req.prompt };
+      if (refs.length > 0) imageInput.images = refs;
+      imageInput.size = wanImageSize(req.aspectRatio, req.size);
+      input = imageInput;
+      break;
     }
-    const imageInput = { prompt: req.prompt };
-    imageInput.image = refToUrl3(first);
-    imageInput.creativity = 0;
-    imageInput.scale_factor = 2;
-    imageInput.output_format = "png";
-    input = imageInput;
-  } else if (isWanImage(id)) {
-    const refs = (req.references ?? []).map(refToUrl3);
-    const imageInput = { prompt: req.prompt };
-    if (refs.length > 0) imageInput.images = refs;
-    imageInput.size = wanImageSize(req.aspectRatio, req.size);
-    input = imageInput;
-  } else if (isQwen(id)) {
-    const refs = (req.references ?? []).map(refToUrl3);
-    const firstUrl = refs[0];
-    const imageInput = { prompt: req.prompt };
-    if (firstUrl !== void 0) {
-      imageInput.image = firstUrl;
-      if (!isQwenEditPlus(id)) imageInput.match_input_image = true;
+    case "qwen-image": {
+      const refs = (req.references ?? []).map(refToUrl);
+      const firstUrl = refs[0];
+      const imageInput = { prompt: req.prompt };
+      if (firstUrl !== void 0) {
+        imageInput.image = firstUrl;
+        imageInput.match_input_image = true;
+      } else if (req.aspectRatio !== void 0 && QWEN_IMAGE_RATIOS.has(req.aspectRatio)) {
+        imageInput.aspect_ratio = req.aspectRatio;
+      }
+      input = imageInput;
+      break;
     }
-    const ratios = isQwenEditPlus(id) ? QWEN_EDIT_RATIOS : QWEN_IMAGE_RATIOS;
-    if (req.aspectRatio !== void 0) {
-      if (ratios.has(req.aspectRatio)) imageInput.aspect_ratio = req.aspectRatio;
-      else
-        ctx.warnings.push({
-          code: "option_ignored",
-          field: "aspectRatio",
-          from: req.aspectRatio,
-          message: `${id} has no ${req.aspectRatio} aspect_ratio; using the provider default`
-        });
+    case "qwen-edit": {
+      const refs = (req.references ?? []).map(refToUrl);
+      const firstUrl = refs[0];
+      const imageInput = { prompt: req.prompt };
+      if (firstUrl !== void 0) imageInput.image = firstUrl;
+      if (req.aspectRatio !== void 0) {
+        if (QWEN_EDIT_RATIOS.has(req.aspectRatio))
+          imageInput.aspect_ratio = req.aspectRatio;
+        else
+          ctx.warnings.push({
+            code: "option_ignored",
+            field: "aspectRatio",
+            from: req.aspectRatio,
+            message: `${id} has no ${req.aspectRatio} aspect_ratio; using the provider default`
+          });
+      }
+      if (req.outputFormat !== void 0) {
+        imageInput.output_format = req.outputFormat === "jpeg" ? "jpg" : req.outputFormat;
+      }
+      input = imageInput;
+      break;
     }
-    if (isQwenEditPlus(id) && req.outputFormat !== void 0) {
-      imageInput.output_format = req.outputFormat === "jpeg" ? "jpg" : req.outputFormat;
+    default: {
+      const first = (req.references ?? [])[0];
+      const imageInput = { prompt: req.prompt };
+      if (first) imageInput.image = refToUrl(first);
+      input = imageInput;
     }
-    input = imageInput;
-  } else {
-    const refs = req.references ?? [];
-    const first = refs[0];
-    const imageInput = { prompt: req.prompt };
-    if (first) imageInput.image = refToUrl3(first);
-    input = imageInput;
   }
   const { url, predictTime, raw } = await runPrediction(ctx, input, deps);
   const file = await download(ctx, url);
@@ -5421,8 +5746,8 @@ function wanSize(aspectRatio) {
   return w > h ? "2048*1152" : "1536*2048";
 }
 function klingInput(req) {
-  const images = (req.firstFrame ? [req.firstFrame] : []).map(refToUrl3);
-  if (req.lastFrame) images.push(refToUrl3(req.lastFrame));
+  const images = (req.firstFrame ? [req.firstFrame] : []).map(refToUrl);
+  if (req.lastFrame) images.push(refToUrl(req.lastFrame));
   const input = {
     prompt: req.prompt,
     duration: req.durationSeconds ?? 10,
@@ -5434,19 +5759,24 @@ function klingInput(req) {
     input.start_image = startImage;
     const endImage = images[1];
     if (endImage !== void 0) input.end_image = endImage;
-  } else {
+  } else if (req.aspectRatio !== "adaptive") {
     input.aspect_ratio = req.aspectRatio ?? "16:9";
   }
   if (req.negativePrompt !== void 0) input.negative_prompt = req.negativePrompt;
   return input;
 }
+function klingOmniMode(req) {
+  if (req.resolution === "1080p") return "pro";
+  if (req.resolution === "720p") return "standard";
+  return req.mode ?? "standard";
+}
 function klingOmniInput(req) {
-  const images = (req.firstFrame ? [req.firstFrame] : []).map(refToUrl3);
-  if (req.lastFrame) images.push(refToUrl3(req.lastFrame));
+  const images = (req.firstFrame ? [req.firstFrame] : []).map(refToUrl);
+  if (req.lastFrame) images.push(refToUrl(req.lastFrame));
   const input = {
     prompt: req.prompt,
     duration: req.durationSeconds ?? 10,
-    mode: req.mode ?? "standard",
+    mode: klingOmniMode(req),
     generate_audio: req.generateAudio ?? false
   };
   const startImage = images[0];
@@ -5454,10 +5784,10 @@ function klingOmniInput(req) {
     input.start_image = startImage;
     const endImage = images[1];
     if (endImage !== void 0) input.end_image = endImage;
-  } else {
+  } else if (req.aspectRatio !== "adaptive") {
     input.aspect_ratio = req.aspectRatio ?? "16:9";
   }
-  const refs = (req.referenceImages ?? []).map(refToUrl3);
+  const refs = (req.referenceImages ?? []).map(refToUrl);
   if (refs.length > 0) input.reference_images = refs;
   return input;
 }
@@ -5468,8 +5798,8 @@ function hailuoInput(req) {
     resolution: req.resolution ?? "768p",
     prompt_optimizer: true
   };
-  if (req.firstFrame) input.first_frame_image = refToUrl3(req.firstFrame);
-  if (req.lastFrame) input.last_frame_image = refToUrl3(req.lastFrame);
+  if (req.firstFrame) input.first_frame_image = refToUrl(req.firstFrame);
+  if (req.lastFrame) input.last_frame_image = refToUrl(req.lastFrame);
   return input;
 }
 function wanInput(req) {
@@ -5478,12 +5808,13 @@ function wanInput(req) {
     duration: req.durationSeconds ?? 5,
     size: wanSize(req.aspectRatio)
   };
-  if (req.firstFrame) input.image = refToUrl3(req.firstFrame);
+  if (req.firstFrame) input.image = refToUrl(req.firstFrame);
   return input;
 }
 async function generateReplicateVideo(req, ctx, deps) {
-  const id = isLower(ctx.model.providerModelId);
-  const input = id.includes("kling") ? id.includes("omni") ? klingOmniInput(req) : klingInput(req) : id.includes("hailuo") ? hailuoInput(req) : wanInput(req);
+  throwIfAborted(ctx.signal);
+  const kind = videoKind(ctx.model.providerModelId);
+  const input = kind === "kling" ? klingInput(req) : kind === "kling-omni" ? klingOmniInput(req) : kind === "hailuo" ? hailuoInput(req) : wanInput(req);
   const api = client(ctx, deps);
   const createOptions = { ...route(ctx.model.providerModelId), input };
   let prediction;
@@ -5491,12 +5822,14 @@ async function generateReplicateVideo(req, ctx, deps) {
     prediction = await withRetry(
       async () => {
         try {
-          return await api.predictions.create(createOptions);
+          return await api.predictions.create(
+            ctx.signal ? { ...createOptions, signal: ctx.signal } : createOptions
+          );
         } catch (cause) {
           throw replicateError(cause, ctx);
         }
       },
-      ctx.signal ? { signal: ctx.signal } : {}
+      { attempts: 1 }
     );
   } catch (cause) {
     throw replicateError(cause, ctx);
@@ -5505,7 +5838,10 @@ async function generateReplicateVideo(req, ctx, deps) {
   const check = async () => {
     let current;
     try {
-      current = await api.predictions.get(jobId);
+      current = await api.predictions.get(
+        jobId,
+        ctx.signal ? { signal: ctx.signal } : void 0
+      );
     } catch (cause) {
       throw replicateError(cause, ctx);
     }
@@ -5606,7 +5942,7 @@ function mmaudioInput(req, ctx) {
     });
   }
   const input = {
-    video: refToUrl3(video),
+    video: refToUrl(video),
     duration: req.durationSeconds ?? 8,
     seed: req.seed ?? -1
   };
@@ -5625,8 +5961,8 @@ function predictionRaw(prediction) {
   return raw;
 }
 async function generateReplicateAudio(req, ctx, deps) {
-  const id = isLower(ctx.model.providerModelId);
-  const input = id.includes("mmaudio") ? mmaudioInput(req, ctx) : id.includes("elevenlabs") ? elevenLabsMusicInput(req, ctx) : lyriaAudioInput(req, ctx);
+  const kind = audioKind(ctx.model.providerModelId);
+  const input = kind === "mmaudio" ? mmaudioInput(req, ctx) : kind === "elevenlabs" ? elevenLabsMusicInput(req, ctx) : lyriaAudioInput(req, ctx);
   const { url, predictTime, raw } = await runPrediction(ctx, input, deps);
   const usage2 = audioUsage(ctx, req);
   if (ctx.model.usage.unit === "seconds") {
@@ -5655,19 +5991,14 @@ function createXaiVideoAdapter(deps = defaultXaiDeps) {
   };
 }
 var xaiVideoAdapter = createXaiVideoAdapter();
-function refToUrl4(ref) {
-  if ("url" in ref) return ref.url;
-  return `data:${ref.mime};base64,${Buffer.from(ref.bytes).toString("base64")}`;
-}
 var xaiResponseStatusSchema = import_zod13.z.object({
   response: import_zod13.z.object({ status: import_zod13.z.number() }).partial()
 });
 function xaiError(cause, ctx) {
-  if (isModelKitError(cause)) return cause;
   const info = describeThrown(cause);
   const nested = xaiResponseStatusSchema.safeParse(cause);
   const status = info.status ?? (nested.success ? nested.data.response?.status : void 0);
-  return classifyHttpError(status, info.body, "xai-video", ctx.model.id);
+  return classifyThrown("xai-video", cause, ctx.model.id, { status, body: info.body });
 }
 var xaiMetaSchema = import_zod13.z.object({
   videoUrl: JsonValueSchema.optional(),
@@ -5691,27 +6022,35 @@ async function generateXaiVideo(req, ctx, deps) {
       retryable: false
     });
   }
-  const frames = [req.firstFrame, req.lastFrame, ...req.referenceImages ?? []].filter(
-    (r) => r !== void 0
-  );
-  const refUrls = frames.map(refToUrl4).slice(0, 5);
-  const resolution = req.resolution === "480p" ? "480p" : "720p";
+  const refUrls = (req.referenceImages ?? []).map(refToUrl).slice(0, 7);
   const xaiOptions = {
-    resolution,
     pollIntervalMs: POLL_INTERVAL_MS2,
     pollTimeoutMs: POLL_TIMEOUT_MS2
   };
+  if (req.resolution === "480p" || req.resolution === "720p" || req.resolution === "1080p") {
+    xaiOptions.resolution = req.resolution;
+  }
   if (refUrls.length > 0) {
     xaiOptions.mode = "reference-to-video";
     xaiOptions.referenceImageUrls = refUrls;
   }
+  if (req.lastFrame !== void 0) {
+    ctx.warnings.push({
+      code: "option_ignored",
+      field: "lastFrame",
+      message: "xai-video does not support lastFrame; ignored"
+    });
+  }
   const client2 = deps.createXaiClient({ apiKey });
-  const model = (0, import_xai2.createXai)({ apiKey }).video(ctx.model.providerModelId);
+  const model = (0, import_xai2.createXai)({ apiKey, fetch: ctx.fetch }).video(ctx.model.providerModelId);
   const callBody = {
     model,
     prompt: req.prompt,
     providerOptions: { xai: xaiOptions }
   };
+  if (req.firstFrame) {
+    callBody.frameImages = [{ image: refToUrl(req.firstFrame), frameType: "first_frame" }];
+  }
   if (req.aspectRatio !== void 0 && req.aspectRatio !== "adaptive") {
     callBody.aspectRatio = req.aspectRatio;
   }
@@ -5746,7 +6085,8 @@ async function generateXaiVideo(req, ctx, deps) {
     return settled;
   };
   return {
-    jobId: `${ctx.model.id}-grok`,
+    // xAI exposes no request id on the synchronous handle, so each job gets a fresh one.
+    jobId: crypto.randomUUID(),
     async poll() {
       if (failed !== void 0) return { status: "failed" };
       return settled ? { status: "succeeded" } : { status: "pending" };
@@ -5797,26 +6137,14 @@ async function mapResult(ctx, req, result) {
 function usage(ctx, req, meta) {
   const costTicks = import_zod13.z.number().safeParse(meta.costInUsdTicks);
   const cost = costTicks.success ? costTicks.data / 1e10 : void 0;
-  if (cost !== void 0) {
-    const event2 = {
-      model: ctx.model.id,
-      provider: "xai-video",
-      modality: "VIDEO",
-      unit: "reported_cost_usd",
-      quantity: cost,
-      reportedCostUsd: cost
-    };
-    if (req.durationSeconds !== void 0) event2.durationSeconds = req.durationSeconds;
-    if (req.resolution !== void 0) event2.resolution = req.resolution;
-    return event2;
-  }
   const event = {
     model: ctx.model.id,
     provider: "xai-video",
     modality: "VIDEO",
-    unit: "seconds",
-    quantity: req.durationSeconds ?? 0
+    unit: "reported_cost_usd",
+    quantity: cost ?? 0
   };
+  if (cost !== void 0) event.reportedCostUsd = cost;
   if (req.durationSeconds !== void 0) event.durationSeconds = req.durationSeconds;
   if (req.resolution !== void 0) event.resolution = req.resolution;
   return event;
@@ -5893,6 +6221,7 @@ var describe = (modelId) => createDefaultModelKit().describe(modelId);
   prodiaAdapter,
   readErrorBody,
   replicateAdapter,
+  resolveAudioOptions,
   resolveImageOptions,
   resolveKeys,
   resolveVideoOptions,