@shop-os/modelkit 0.2.25 → 0.2.26

package/dist/index.cjs

@@ -40,6 +40,7 @@ __export(index_exports, {
   CatalogSchema: () => CatalogSchema,
   DEFAULT_ADAPTERS: () => DEFAULT_ADAPTERS,
   ExposureSchema: () => ExposureSchema,
+  IMAGE_INPUT_SPEC_KEYS: () => IMAGE_INPUT_SPEC_KEYS,
   INPUT_SPEC_SUPPORTS_KEYS: () => INPUT_SPEC_SUPPORTS_KEYS,
   InputSpecSchema: () => InputSpecSchema,
   JsonObjectSchema: () => JsonObjectSchema,
@@ -2204,6 +2205,14 @@ var VIDEO_INPUT_SPEC_KEYS = [
   "promptExpansion",
   "referenceImages"
 ];
+var IMAGE_INPUT_SPEC_KEYS = [
+  "prompt",
+  "references",
+  "aspectRatio",
+  "size",
+  "outputFormat",
+  "seed"
+];
 var AUDIO_INPUT_SPEC_KEYS = [
   "prompt",
   "durationSeconds",
@@ -2220,14 +2229,22 @@ var INPUT_SPEC_SUPPORTS_KEYS = [
   "generateAudio",
   "cfgScale",
   "promptExpansion",
+  "outputFormat",
   "instrumental"
 ];
 var VIDEO_KEY_SET = new Set(VIDEO_INPUT_SPEC_KEYS);
+var IMAGE_KEY_SET = new Set(IMAGE_INPUT_SPEC_KEYS);
 var AUDIO_KEY_SET = new Set(AUDIO_INPUT_SPEC_KEYS);
 var ALL_KEY_SET = /* @__PURE__ */ new Set([
   ...VIDEO_INPUT_SPEC_KEYS,
+  ...IMAGE_INPUT_SPEC_KEYS,
   ...AUDIO_INPUT_SPEC_KEYS
 ]);
+var ALLOWED_KEYS_BY_TYPE = /* @__PURE__ */ new Map([
+  ["video", VIDEO_KEY_SET],
+  ["image", IMAGE_KEY_SET],
+  ["audio", AUDIO_KEY_SET]
+]);
 var ProviderScalarSchema = import_zod2.z.union([import_zod2.z.string(), import_zod2.z.number(), import_zod2.z.boolean()]);
 var FieldsSchema = import_zod2.z.record(import_zod2.z.string(), import_zod2.z.string().min(1)).superRefine((fields, ctx) => {
   const keys = Object.keys(fields);
@@ -2296,13 +2313,13 @@ function validateInputSpec(raw, row) {
     };
   }
   const issues = [];
-  if (row.type !== "video" && row.type !== "audio") {
+  const allowed = ALLOWED_KEYS_BY_TYPE.get(row.type);
+  if (allowed === void 0) {
     issues.push({
       path: [],
       message: `inputSpec is not supported for "${row.type}" rows`
     });
   } else {
-    const allowed = row.type === "video" ? VIDEO_KEY_SET : AUDIO_KEY_SET;
     for (const key of Object.keys(spec.fields)) {
       if (!allowed.has(key)) {
         issues.push({
@@ -6844,6 +6861,12 @@ async function generateReplicateImage(req, ctx, deps) {
       break;
     }
     default: {
+      const specResult = replicateInputSpec(req, ctx);
+      if (specResult.kind === "invalid") throw replicateInvalidSpec(ctx, specResult.issues);
+      if (specResult.kind === "ok") {
+        input = specResult.input;
+        break;
+      }
       const first = (req.references ?? [])[0];
       const imageInput = { prompt: req.prompt };
       if (first) imageInput.image = refToUrl(first);
@@ -8918,6 +8941,7 @@ var describe = (modelId) => createDefaultModelKit().describe(modelId);
   CatalogSchema,
   DEFAULT_ADAPTERS,
   ExposureSchema,
+  IMAGE_INPUT_SPEC_KEYS,
   INPUT_SPEC_SUPPORTS_KEYS,
   InputSpecSchema,
   JsonObjectSchema,