@shop-os/modelkit 0.2.24 → 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,
@@ -81,6 +82,7 @@ __export(index_exports, {
   liveCatalog: () => liveCatalog,
   liveState: () => liveState,
   mergeTextProviderOptions: () => mergeTextProviderOptions,
+  missingTierKeys: () => missingTierKeys,
   nearestAspectRatio: () => nearestAspectRatio,
   normalizeExposure: () => normalizeExposure,
   openaiImageAdapter: () => openaiImageAdapter,
@@ -2203,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",
@@ -2219,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);
@@ -2295,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({
@@ -6843,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);
@@ -8095,10 +8119,22 @@ function hasAll(keys, names) {
 function hasKey(keys, name) {
   return hasAll(keys, [name]);
 }
+function missingTierKeys(config, tier, names) {
+  return names.filter((name) => !hasAll(config.keys[tier], [name]));
+}
 function missingEnterpriseKeys(config, names) {
-  return names.filter((name) => !hasAll(config.keys.enterprise, [name]));
+  return missingTierKeys(config, "enterprise", names);
 }
 function resolveKeys(config, names, tier = "pro") {
+  if (tier === "sloosh") {
+    if (names.length === 0) return config.keys.sloosh ?? {};
+    if (hasAll(config.keys.sloosh, names)) return config.keys.sloosh;
+    throw new ModelKitError({
+      code: "auth",
+      message: `Missing Sloosh API key for ${names.join(", ")}`,
+      retryable: false
+    });
+  }
   if (tier === "enterprise" && hasAll(config.keys.enterprise, names))
     return config.keys.enterprise;
   if (names.length > 0 && !hasAll(config.keys.pro, names)) {
@@ -8287,6 +8323,7 @@ function resolveAnthropicTransport(model, config, tier, warn) {
   const preferred = preferredAnthropicTransport(model, config.routing ?? {});
   const usable = (transport) => {
     const name = transport === "bedrock" ? "aws" : "anthropic";
+    if (tier === "sloosh") return hasKey(config.keys.sloosh, name);
     return hasKey(config.keys.pro, name) || tier === "enterprise" && hasKey(config.keys.enterprise, name);
   };
   if (usable(preferred)) return preferred;
@@ -8868,11 +8905,19 @@ function createModelKit(config, adapters) {
 }
 
 // src/default.ts
+function withSharedAws(keys) {
+  const { pro, sloosh } = keys;
+  if (!sloosh || sloosh.aws || !pro.aws) return keys;
+  return { ...keys, sloosh: { ...sloosh, aws: pro.aws } };
+}
 function createDefaultModelKit(config = {}) {
-  const keys = config.keys ?? {
-    pro: keysFromEnv(process.env),
-    enterprise: keysFromEnv(process.env, "ENTERPRISE_")
-  };
+  const keys = withSharedAws(
+    config.keys ?? {
+      pro: keysFromEnv(process.env),
+      enterprise: keysFromEnv(process.env, "ENTERPRISE_"),
+      sloosh: keysFromEnv(process.env, "SLOOSH_")
+    }
+  );
   return createModelKit({ ...config, keys }, DEFAULT_ADAPTERS);
 }
 var generateText2 = (options, call) => createDefaultModelKit().generateText(options, call);
@@ -8896,6 +8941,7 @@ var describe = (modelId) => createDefaultModelKit().describe(modelId);
   CatalogSchema,
   DEFAULT_ADAPTERS,
   ExposureSchema,
+  IMAGE_INPUT_SPEC_KEYS,
   INPUT_SPEC_SUPPORTS_KEYS,
   InputSpecSchema,
   JsonObjectSchema,
@@ -8937,6 +8983,7 @@ var describe = (modelId) => createDefaultModelKit().describe(modelId);
   liveCatalog,
   liveState,
   mergeTextProviderOptions,
+  missingTierKeys,
   nearestAspectRatio,
   normalizeExposure,
   openaiImageAdapter,