@shop-os/modelkit 0.6.0 → 0.6.1

package/dist/index.cjs

@@ -11052,6 +11052,30 @@ var klingV3Family = {
   }
 };
 
+// src/replicate-video-kinds.ts
+function replicateBaseModelId(providerModelId) {
+  const colon = providerModelId.indexOf(":");
+  return colon === -1 ? providerModelId : providerModelId.slice(0, colon);
+}
+var REPLICATE_VIDEO_KINDS = /* @__PURE__ */ new Map([
+  ["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.get(replicateBaseModelId(providerModelId));
+}
+var KLING_TIERS = [
+  { mode: "standard", resolution: "720p" },
+  { mode: "pro", resolution: "1080p" },
+  { mode: "4k", resolution: "4k" }
+];
+function klingTier(req) {
+  return KLING_TIERS.find((tier) => tier.resolution === req.resolution) ?? KLING_TIERS.find((tier) => tier.mode === req.mode) ?? KLING_TIERS[0];
+}
+
 // src/resolve-video-rules.ts
 var MOTION_PRO2 = "fal-ai/kling-video/v3/pro/motion-control";
 var MOTION_STANDARD2 = "fal-ai/kling-video/v3/standard/motion-control";
@@ -11225,6 +11249,16 @@ function h3EndFrameRatioWarning(model, ratio) {
     message: `${model.id} follows the end frame's aspect ratio`
   };
 }
+function isReplicateKling(model) {
+  if (model.adapter !== "replicate") return false;
+  const kind = videoKind(model.providerModelId);
+  return kind === "kling" || kind === "kling-omni";
+}
+function fillReplicateKlingResolution(model, req) {
+  if (isReplicateKling(model) && req.resolution === void 0) {
+    req.resolution = klingTier(req).resolution;
+  }
+}
 function applyFalVideoRules(model, req, callerRatio, warnings) {
   if (model.adapter !== "fal") return;
   const problem = falVideoProblem(model.providerModelId, withPromptForResolve(req));
@@ -14157,10 +14191,6 @@ function replicateError(cause, ctx) {
   const status = info.status ?? (nested.success ? nested.data.response?.status : void 0);
   return classifyThrown("replicate", cause, ctx.model.id, { status, body: info.body });
 }
-function baseModelId(providerModelId) {
-  const colon = providerModelId.indexOf(":");
-  return colon === -1 ? providerModelId : providerModelId.slice(0, colon);
-}
 function predictionErrorBody(cause) {
   const parsed = JsonValueSchema.safeParse(cause);
   if (parsed.success) return parsed.data;
@@ -14314,17 +14344,7 @@ var REPLICATE_IMAGE_KINDS = {
   "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)];
+  return REPLICATE_IMAGE_KINDS[replicateBaseModelId(providerModelId)] ?? "generic";
 }
 function isReplicateVideoKnown(providerModelId) {
   return videoKind(providerModelId) !== void 0;
@@ -14335,7 +14355,7 @@ var REPLICATE_AUDIO_KINDS = {
   "google/lyria-2": "lyria"
 };
 function audioKind(providerModelId) {
-  return REPLICATE_AUDIO_KINDS[baseModelId(providerModelId)];
+  return REPLICATE_AUDIO_KINDS[replicateBaseModelId(providerModelId)];
 }
 function isReplicateAudioKnown(providerModelId) {
   return audioKind(providerModelId) !== void 0;
@@ -14539,14 +14559,6 @@ function replicateVideoResolution(providerModelId, req) {
   if (kind === "hailuo") return hailuoResolution(req);
   return req.resolution;
 }
-var KLING_TIERS = [
-  { mode: "standard", resolution: "720p" },
-  { mode: "pro", resolution: "1080p" },
-  { mode: "4k", resolution: "4k" }
-];
-function klingTier(req) {
-  return KLING_TIERS.find((tier) => tier.resolution === req.resolution) ?? KLING_TIERS.find((tier) => tier.mode === req.mode) ?? KLING_TIERS[0];
-}
 function klingOmniInput(req) {
   const input = {
     prompt: req.prompt,
@@ -15774,8 +15786,10 @@ function resolveVideoOptions(model, request, onUnsupported = "error") {
     warnings
   );
   if (aspectRatio !== void 0) merged.aspectRatio = aspectRatio;
+  fillReplicateKlingResolution(model, merged);
   const resolution = resolveVideoResolution(merged.resolution, model, onUnsupported, warnings);
   if (resolution !== void 0) merged.resolution = resolution;
+  if (isReplicateKling(model)) delete merged.mode;
   const duration = resolveDuration(
     merged.durationSeconds,
     c.durations,