@shop-os/modelkit 0.2.25 → 0.2.27

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);
@@ -8315,11 +8338,17 @@ function textUsageProvider(model, transport) {
 }
 
 // src/client.ts
-function requiredKeys(model, config, transport) {
-  if (model.adapter === "fal" && config.routing?.fal === "cloudflare-passthrough")
-    return ["cloudflare"];
+function routingFor(config, tier) {
+  const routing = config.routing ?? {};
+  if (tier !== "sloosh") return routing;
+  const direct = { ...routing };
+  if (direct.text === "cloudflare") direct.text = "direct";
+  if (direct.fal === "cloudflare-passthrough") direct.fal = "direct";
+  return direct;
+}
+function requiredKeys(model, routing, transport) {
+  if (model.adapter === "fal" && routing.fal === "cloudflare-passthrough") return ["cloudflare"];
   if (model.adapter === "ai-sdk-text") {
-    const routing = config.routing ?? {};
     const names = [textKeyName(model, transport)];
     if (textUsesCloudflareGateway(model, routing, transport)) names.push("cloudflare");
     return names;
@@ -8423,7 +8452,8 @@ function createModelKit(config, adapters) {
         model: model.id
       });
     const transport = resolveAnthropicTransport(model, config, call.tier, warnOnce);
-    const names = requiredKeys(model, config, transport);
+    const routing = routingFor(config, call.tier);
+    const names = requiredKeys(model, routing, transport);
     if (call.tier === "enterprise") {
       const missing = missingEnterpriseKeys(config, names);
       if (missing.length > 0)
@@ -8436,7 +8466,7 @@ function createModelKit(config, adapters) {
     const ctx = {
       model,
       keys,
-      routing: config.routing ?? {},
+      routing,
       fetch: fetchImpl,
       logger,
       call,
@@ -8918,6 +8948,7 @@ var describe = (modelId) => createDefaultModelKit().describe(modelId);
   CatalogSchema,
   DEFAULT_ADAPTERS,
   ExposureSchema,
+  IMAGE_INPUT_SPEC_KEYS,
   INPUT_SPEC_SUPPORTS_KEYS,
   InputSpecSchema,
   JsonObjectSchema,