@shop-os/modelkit 0.4.2 → 0.4.4

package/dist/index.cjs

@@ -2617,16 +2617,26 @@ var ArraysSchema = import_zod2.z.record(import_zod2.z.string().min(1), import_zo
     }
   }
 });
+var DimensionsSchema = import_zod2.z.strictObject({
+  field: import_zod2.z.string().min(1),
+  areas: import_zod2.z.record(import_zod2.z.string(), import_zod2.z.number().int().positive()).refine((a) => Object.keys(a).length > 0, {
+    message: "inputSpec.dimensions.areas must name at least one size"
+  }),
+  maxSide: import_zod2.z.number().int().positive().optional(),
+  multipleOf: import_zod2.z.number().int().positive().optional()
+});
 var InputSpecSchema = import_zod2.z.strictObject({
   fields: FieldsSchema,
   values: ValuesSchema.optional(),
   static: StaticSchema.optional(),
   arrays: ArraysSchema.optional(),
-  editEndpoint: import_zod2.z.string().min(1).optional()
+  editEndpoint: import_zod2.z.string().min(1).optional(),
+  dimensions: DimensionsSchema.optional()
 }).superRefine((spec, ctx) => {
   const mappedNames = /* @__PURE__ */ new Set([
     ...Object.values(spec.fields),
-    ...Object.keys(spec.arrays ?? {})
+    ...Object.keys(spec.arrays ?? {}),
+    ...spec.dimensions ? [spec.dimensions.field] : []
   ]);
   for (const key of Object.keys(spec.static ?? {})) {
     if (mappedNames.has(key)) {
@@ -2661,7 +2671,10 @@ function validateInputSpec(raw, row) {
     return {
       ok: false,
       issues: [
-        { path: [], message: "inputSpec is only supported on replicate and fal rows" }
+        {
+          path: [],
+          message: "inputSpec is only supported on replicate and fal rows"
+        }
       ]
     };
   }
@@ -2754,6 +2767,29 @@ function validateInputSpec(raw, row) {
       }
     }
   }
+  if (spec.dimensions !== void 0) {
+    if (row.type !== "image") {
+      issues.push({
+        path: ["dimensions"],
+        message: "inputSpec.dimensions is only supported on image rows"
+      });
+    }
+    for (const key of ["aspectRatio", "size"]) {
+      if (key in spec.fields) {
+        issues.push({
+          path: ["fields", key],
+          message: `inputSpec.fields "${key}" cannot be set with inputSpec.dimensions, which sends it`
+        });
+      }
+    }
+    const owner = seenNames.get(spec.dimensions.field);
+    if (owner !== void 0 || spec.dimensions.field in (spec.arrays ?? {})) {
+      issues.push({
+        path: ["dimensions", "field"],
+        message: `inputSpec.dimensions.field "${spec.dimensions.field}" is already a mapped provider field`
+      });
+    }
+  }
   const supports = row.constraints.supports ?? [];
   for (const key of INPUT_SPEC_SUPPORTS_KEYS) {
     if (supports.includes(key)) continue;
@@ -4570,11 +4606,37 @@ function applyInputSpec(spec, req, refToUrl2, opts) {
     }
     if (urls.length > 0) out[providerField] = urls;
   }
+  if (spec.dimensions !== void 0) {
+    const dims = dimensionsFor(spec.dimensions, req);
+    if (dims !== void 0) out[spec.dimensions.field] = dims;
+  }
   for (const [key, value] of Object.entries(spec.static ?? {})) {
     out[key] = value;
   }
   return out;
 }
+function dimensionsFor(dims, req) {
+  const ratio = req.aspectRatio;
+  const size = req.size;
+  if (ratio === void 0 || ratio === "adaptive" || size === void 0) return void 0;
+  const area = dims.areas[size];
+  const match = /^(\d+):(\d+)$/.exec(ratio);
+  if (area === void 0 || match === null) return void 0;
+  const r = Number(match[1]) / Number(match[2]);
+  let width = Math.sqrt(area * r);
+  let height = width / r;
+  const longest = Math.max(width, height);
+  if (dims.maxSide !== void 0 && longest > dims.maxSide) {
+    const scale = dims.maxSide / longest;
+    width *= scale;
+    height *= scale;
+  }
+  const step = dims.multipleOf ?? 1;
+  return {
+    width: Math.max(step, Math.floor(width / step) * step),
+    height: Math.max(step, Math.floor(height / step) * step)
+  };
+}
 function resolveRowInputSpec(req, ctx) {
   const raw = ctx.model.providerOptions?.["modelkit"]?.["inputSpec"];
   if (raw === void 0) return { kind: "absent" };
@@ -5757,8 +5819,10 @@ async function generateOmniVideo(req, ctx, createGenAIClient) {
   const apiKey = googleKey(ctx);
   const modelId = ctx.model.id;
   const providerModelId = ctx.model.providerModelId;
-  const duration = req.durationSeconds ?? 10;
   const extend = req.taskType === "extend";
+  const sourceSeconds = extend ? void 0 : req.sourceVideo?.durationSeconds;
+  const editSeconds = sourceSeconds !== void 0 && Number.isFinite(sourceSeconds) && sourceSeconds > 0 ? sourceSeconds : void 0;
+  const duration = req.durationSeconds ?? editSeconds ?? 10;
   const previousId = extend ? req.previousInteractionId : void 0;
   const chained = previousId !== void 0;
   for (const field of [