@bratsos/workflow-engine 0.0.11 → 0.2.0

package/dist/cli/sync-models.js

@@ -0,0 +1,210 @@
+#!/usr/bin/env node
+import { existsSync, mkdirSync, writeFileSync } from 'fs';
+import { join, resolve, dirname } from 'path';
+import { pathToFileURL } from 'url';
+
+async function main() {
+  const apiKey = process.env.OPENROUTER_API_KEY;
+  if (!apiKey) {
+    console.error("\u274C OPENROUTER_API_KEY environment variable is required");
+    process.exit(1);
+  }
+  const cwd = process.cwd();
+  console.log(`\u{1F4C1} Working directory: ${cwd}`);
+  let config = {};
+  const configPath = join(cwd, "workflow-engine.models.ts");
+  if (existsSync(configPath)) {
+    try {
+      const configModule = await import(pathToFileURL(configPath).href);
+      config = configModule.default || configModule;
+      console.log("\u{1F4CB} Loaded config from workflow-engine.models.ts");
+    } catch (err) {
+      console.warn(`\u26A0\uFE0F  Could not load config: ${err}`);
+    }
+  }
+  const outputPath = resolve(
+    cwd,
+    config.outputPath || "src/generated/models.ts"
+  );
+  const includePatterns = config.include || [];
+  const excludePatterns = config.exclude || [];
+  const customModels = config.customModels || {};
+  console.log("\u{1F504} Fetching models from OpenRouter API...");
+  const response = await fetch("https://openrouter.ai/api/v1/models", {
+    headers: {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json"
+    }
+  });
+  if (!response.ok) {
+    console.error(
+      `\u274C OpenRouter API error: ${response.status} ${response.statusText}`
+    );
+    process.exit(1);
+  }
+  const data = await response.json();
+  console.log(`\u2705 Fetched ${data.data.length} models from OpenRouter`);
+  console.log("\u{1F504} Fetching embedding models from OpenRouter API...");
+  const embeddingResponse = await fetch(
+    "https://openrouter.ai/api/v1/embeddings/models",
+    {
+      headers: {
+        Authorization: `Bearer ${apiKey}`,
+        "Content-Type": "application/json"
+      }
+    }
+  );
+  let embeddingModelIds = /* @__PURE__ */ new Set();
+  if (embeddingResponse.ok) {
+    const embeddingData = await embeddingResponse.json();
+    console.log(
+      `\u2705 Fetched ${embeddingData.data.length} embedding models from OpenRouter`
+    );
+    embeddingModelIds = new Set(embeddingData.data.map((m) => m.id));
+    for (const model of embeddingData.data) {
+      if (!data.data.find((m) => m.id === model.id)) {
+        data.data.push(model);
+      }
+    }
+  } else {
+    console.warn(
+      `\u26A0\uFE0F  Could not fetch embedding models: ${embeddingResponse.status}`
+    );
+  }
+  const filteredModels = data.data.filter((model) => {
+    if (includePatterns.length > 0) {
+      let matchesInclude = false;
+      for (const pattern of includePatterns) {
+        if (typeof pattern === "string") {
+          if (model.id.includes(pattern)) {
+            matchesInclude = true;
+            break;
+          }
+        } else if (pattern instanceof RegExp) {
+          if (pattern.test(model.id)) {
+            matchesInclude = true;
+            break;
+          }
+        }
+      }
+      if (!matchesInclude) return false;
+    }
+    for (const pattern of excludePatterns) {
+      if (typeof pattern === "string") {
+        if (model.id === pattern) return false;
+      } else if (pattern instanceof RegExp) {
+        if (pattern.test(model.id)) return false;
+      }
+    }
+    return true;
+  });
+  console.log(
+    `\u{1F4CA} After filtering: ${filteredModels.length} models (excluded ${data.data.length - filteredModels.length})`
+  );
+  const models = {};
+  for (const model of filteredModels) {
+    const promptPrice = parseFloat(model.pricing?.prompt || "0");
+    const completionPrice = parseFloat(model.pricing?.completion || "0");
+    const supportsBatch = model.id.startsWith("anthropic/") || model.id.startsWith("openai/") && !model.id.includes("gpt-oss") || model.id.startsWith("google/") && model.id.includes("gemini");
+    const isEmbedding = embeddingModelIds.has(model.id);
+    const supportsTools = model.supported_parameters?.includes("tools") || false;
+    const supportsStructuredOutputs = model.supported_parameters?.includes("structured_outputs") || false;
+    models[model.id] = {
+      id: model.id,
+      name: model.name,
+      inputCostPerMillion: Math.round(promptPrice * 1e6 * 1e4) / 1e4,
+      outputCostPerMillion: Math.round(completionPrice * 1e6 * 1e4) / 1e4,
+      provider: "openrouter",
+      description: model.description,
+      contextLength: model.top_provider?.context_length || model.context_length,
+      maxCompletionTokens: model.top_provider?.max_completion_tokens,
+      ...isEmbedding && { isEmbeddingModel: true },
+      ...supportsTools && { supportsTools: true },
+      ...supportsStructuredOutputs && { supportsStructuredOutputs: true },
+      ...supportsBatch && !isEmbedding && {
+        supportsAsyncBatch: true,
+        batchDiscountPercent: 50
+      }
+    };
+  }
+  for (const [key, modelConfig] of Object.entries(customModels)) {
+    models[key] = modelConfig;
+  }
+  const allModelIds = Object.keys(models);
+  console.log(`\u{1F4E6} Total models to generate: ${allModelIds.length}`);
+  const generatedContent = generateTypeScript(models);
+  const dir = dirname(outputPath);
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true });
+  }
+  writeFileSync(outputPath, generatedContent, "utf-8");
+  console.log(`\u2705 Generated ${outputPath}`);
+  console.log(
+    "\n\u{1F4A1} Import this file in your app entry to enable autocomplete:"
+  );
+  console.log(`   import "${outputPath.replace(cwd, ".")}";`);
+}
+function generateTypeScript(models) {
+  const modelIds = Object.keys(models);
+  const modelEntries = Object.entries(models).map(([key, config]) => {
+    const lines = [
+      `  "${key}": {`,
+      `    id: "${config.id}",`,
+      `    name: "${config.name.replace(/"/g, '\\"')}",`,
+      `    description: "${config.description?.replace(/"/g, '\\"').replace(/\n/g, "\\n")}",`,
+      `    inputCostPerMillion: ${config.inputCostPerMillion},`,
+      `    outputCostPerMillion: ${config.outputCostPerMillion},`,
+      `    provider: "${config.provider}",`,
+      `    isEmbeddingModel: ${config.isEmbeddingModel || false},`,
+      `    supportsTools: ${config.supportsTools || false},`,
+      `    supportsStructuredOutputs: ${config.supportsStructuredOutputs || false},`,
+      `    supportsAsyncBatch: ${config.supportsAsyncBatch || false},`,
+      `    batchDiscountPercent: ${config.batchDiscountPercent || 0},`,
+      `    contextLength: ${config.contextLength || 0},`,
+      `    maxCompletionTokens: ${config.maxCompletionTokens},`,
+      `  },`
+    ];
+    return lines.join("\n");
+  }).join("\n");
+  const registryEntries = modelIds.map((id) => `    "${id}": true;`).join("\n");
+  const enumEntries = modelIds.map((id) => `  "${id}",`).join("\n");
+  return `// AUTO-GENERATED by workflow-engine-sync
+// Run \`npx workflow-engine-sync\` to regenerate
+// Generated at: ${(/* @__PURE__ */ new Date()).toISOString()}
+
+import { z } from "zod";
+import { registerModels, type ModelConfig } from "@bratsos/workflow-engine/client";
+
+// Register all models (OpenRouter + custom)
+const MODELS: Record<string, ModelConfig> = {
+${modelEntries}
+};
+
+registerModels(MODELS);
+
+/**
+ * Zod enum of all generated model IDs
+ * Use GeneratedModelKey.enum["model-id"] for type-safe model selection
+ */
+export const GeneratedModelKey = z.enum([
+${enumEntries}
+]);
+
+export type GeneratedModelKey = z.infer<typeof GeneratedModelKey>;
+
+// TypeScript module augmentation for autocomplete
+declare module "@bratsos/workflow-engine/client" {
+  interface ModelRegistry {
+${registryEntries}
+  }
+}
+
+export { MODELS };
+`;
+}
+main().catch((err) => {
+  console.error("\u274C Error:", err);
+  process.exit(1);
+});
+//# sourceMappingURL=sync-models.js.map
+//# sourceMappingURL=sync-models.js.map

package/dist/cli/sync-models.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/cli/sync-models.ts"],"names":[],"mappings":";;;;;AAyCA,eAAe,IAAA,GAAO;AACpB,EAAA,MAAM,MAAA,GAAS,QAAQ,GAAA,CAAI,kBAAA;AAC3B,EAAA,IAAI,CAAC,MAAA,EAAQ;AACX,IAAA,OAAA,CAAQ,MAAM,4DAAuD,CAAA;AACrE,IAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAAA,EAChB;AAEA,EAAA,MAAM,GAAA,GAAM,QAAQ,GAAA,EAAI;AACxB,EAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6BAAA,EAAyB,GAAG,CAAA,CAAE,CAAA;AAG1C,EAAA,IAAI,SAA0B,EAAC;AAC/B,EAAA,MAAM,UAAA,GAAa,IAAA,CAAK,GAAA,EAAK,2BAA2B,CAAA;AACxD,EAAA,IAAI,UAAA,CAAW,UAAU,CAAA,EAAG;AAC1B,IAAA,IAAI;AACF,MAAA,MAAM,YAAA,GAAe,MAAM,OAAO,aAAA,CAAc,UAAU,CAAA,CAAE,IAAA,CAAA;AAC5D,MAAA,MAAA,GAAS,aAAa,OAAA,IAAW,YAAA;AACjC,MAAA,OAAA,CAAQ,IAAI,wDAAiD,CAAA;AAAA,IAC/D,SAAS,GAAA,EAAK;AACZ,MAAA,OAAA,CAAQ,IAAA,CAAK,CAAA,qCAAA,EAA8B,GAAG,CAAA,CAAE,CAAA;AAAA,IAClD;AAAA,EACF;AAEA,EAAA,MAAM,UAAA,GAAa,OAAA;AAAA,IACjB,GAAA;AAAA,IACA,OAAO,UAAA,IAAc;AAAA,GACvB;AACA,EAAA,MAAM,eAAA,GAAkB,MAAA,CAAO,OAAA,IAAW,EAAC;AAC3C,EAAA,MAAM,eAAA,GAAkB,MAAA,CAAO,OAAA,IAAW,EAAC;AAC3C,EAAA,MAAM,YAAA,GAAe,MAAA,CAAO,YAAA,IAAgB,EAAC;AAG7C,EAAA,OAAA,CAAQ,IAAI,kDAA2C,CAAA;AACvD,EAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,qCAAA,EAAuC;AAAA,IAClE,OAAA,EAAS;AAAA,MACP,aAAA,EAAe,UAAU,MAAM,CAAA,CAAA;AAAA,MAC/B,cAAA,EAAgB;AAAA;AAClB,GACD,CAAA;AAED,EAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,IAAA,OAAA,CAAQ,KAAA;AAAA,MACN,CAAA,6BAAA,EAA2B,QAAA,CAAS,MAAM,CAAA,CAAA,EAAI,SAAS,UAAU,CAAA;AAAA,KACnE;AACA,IAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAAA,EAChB;AAEA,EAAA,MAAM,IAAA,GAAQ,MAAM,QAAA,CAAS,IAAA,EAAK;AAClC,EAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,eAAA,EAAa,IAAA,CAAK,IAAA,CAAK,MAAM,CAAA,uBAAA,CAAyB,CAAA;AAGlE,EAAA,OAAA,CAAQ,IAAI,4DAAqD,CAAA;AACjE,EAAA,MAAM,oBAAoB,MAAM,KAAA;AAAA,IAC9B,gDAAA;AAAA,IACA;AAAA,MACE,OAAA,EAAS;AAAA,QACP,aAAA,EAAe,UAAU,MAAM,CAAA,CAAA;AAAA,QAC/B,cAAA,EAAgB;AAAA;AAClB;AACF,GACF;AAEA,EAAA,IAAI,iBAAA,uBAAwB,GAAA,EAAY;AACxC,EAAA,IAAI,kBAAkB,EAAA,EAAI;AACxB,IAAA,MAAM,aAAA,GACH,MAAM,iBAAA,CAAkB,IAAA,EAAK;AAChC,IAAA,OAAA,CAAQ,GAAA;AAAA,MACN,CAAA,eAAA,EAAa,aAAA,CAAc,IAAA,CAAK,MAAM,CAAA,iCAAA;AAAA,KACxC;AAEA,IAAA,iBAAA,GAAoB,IAAI,IAAI,aAAA,CAAc,IAAA,CAAK,IAAI,CAAC,CAAA,KAAM,CAAA,CAAE,EAAE,CAAC,CAAA;AAE/D,IAAA,KAAA,MAAW,KAAA,IAAS,cAAc,IAAA,EAAM;AACtC,MAAA,IAAI,CAAC,IAAA,CAAK,IAAA,CAAK,IAAA,CAAK,CAAC,MAAM,CAAA,CAAE,EAAA,KAAO,KAAA,CAAM,EAAE,CAAA,EAAG;AAC7C,QAAA,IAAA,CAAK,IAAA,CAAK,KAAK,KAAK,CAAA;AAAA,MACtB;AAAA,IACF;AAAA,EACF,CAAA,MAAO;AACL,IAAA,OAAA,CAAQ,IAAA;AAAA,MACN,CAAA,gDAAA,EAAyC,kBAAkB,MAAM,CAAA;AAAA,KACnE;AAAA,EACF;AAGA,EAAA,MAAM,cAAA,GAAiB,IAAA,CAAK,IAAA,CAAK,MAAA,CAAO,CAAC,KAAA,KAAU;AAEjD,IAAA,IAAI,eAAA,CAAgB,SAAS,CAAA,EAAG;AAC9B,MAAA,IAAI,cAAA,GAAiB,KAAA;AACrB,MAAA,KAAA,MAAW,WAAW,eAAA,EAAiB;AACrC,QAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,UAAA,IAAI,KAAA,CAAM,EAAA,CAAG,QAAA,CAAS,OAAO,CAAA,EAAG;AAC9B,YAAA,cAAA,GAAiB,IAAA;AACjB,YAAA;AAAA,UACF;AAAA,QACF,CAAA,MAAA,IAAW,mBAAmB,MAAA,EAAQ;AACpC,UAAA,IAAI,OAAA,CAAQ,IAAA,CAAK,KAAA,CAAM,EAAE,CAAA,EAAG;AAC1B,YAAA,cAAA,GAAiB,IAAA;AACjB,YAAA;AAAA,UACF;AAAA,QACF;AAAA,MACF;AACA,MAAA,IAAI,CAAC,gBAAgB,OAAO,KAAA;AAAA,IAC9B;AAGA,IAAA,KAAA,MAAW,WAAW,eAAA,EAAiB;AACrC,MAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,QAAA,IAAI,KAAA,CAAM,EAAA,KAAO,OAAA,EAAS,OAAO,KAAA;AAAA,MACnC,CAAA,MAAA,IAAW,mBAAmB,MAAA,EAAQ;AACpC,QAAA,IAAI,OAAA,CAAQ,IAAA,CAAK,KAAA,CAAM,EAAE,GAAG,OAAO,KAAA;AAAA,MACrC;AAAA,IACF;AACA,IAAA,OAAO,IAAA;AAAA,EACT,CAAC,CAAA;AAED,EAAA,OAAA,CAAQ,GAAA;AAAA,IACN,CAAA,2BAAA,EAAuB,eAAe,MAAM,CAAA,kBAAA,EAAqB,KAAK,IAAA,CAAK,MAAA,GAAS,eAAe,MAAM,CAAA,CAAA;AAAA,GAC3G;AAGA,EAAA,MAAM,SAAsC,EAAC;AAE7C,EAAA,KAAA,MAAW,SAAS,cAAA,EAAgB;AAElC,IAAA,MAAM,WAAA,GAAc,UAAA,CAAW,KAAA,CAAM,OAAA,EAAS,UAAU,GAAG,CAAA;AAC3D,IAAA,MAAM,eAAA,GAAkB,UAAA,CAAW,KAAA,CAAM,OAAA,EAAS,cAAc,GAAG,CAAA;AAGnE,IAAA,MAAM,aAAA,GACJ,KAAA,CAAM,EAAA,CAAG,UAAA,CAAW,YAAY,KAC/B,KAAA,CAAM,EAAA,CAAG,UAAA,CAAW,SAAS,CAAA,IAAK,CAAC,MAAM,EAAA,CAAG,QAAA,CAAS,SAAS,CAAA,IAC9D,KAAA,CAAM,EAAA,CAAG,UAAA,CAAW,SAAS,CAAA,IAAK,KAAA,CAAM,EAAA,CAAG,QAAA,CAAS,QAAQ,CAAA;AAG/D,IAAA,MAAM,WAAA,GAAc,iBAAA,CAAkB,GAAA,CAAI,KAAA,CAAM,EAAE,CAAA;AAGlD,IAAA,MAAM,aAAA,GACJ,KAAA,CAAM,oBAAA,EAAsB,QAAA,CAAS,OAAO,CAAA,IAAK,KAAA;AACnD,IAAA,MAAM,yBAAA,GACJ,KAAA,CAAM,oBAAA,EAAsB,QAAA,CAAS,oBAAoB,CAAA,IAAK,KAAA;AAEhE,IAAA,MAAA,CAAO,KAAA,CAAM,EAAE,CAAA,GAAI;AAAA,MACjB,IAAI,KAAA,CAAM,EAAA;AAAA,MACV,MAAM,KAAA,CAAM,IAAA;AAAA,MACZ,qBAAqB,IAAA,CAAK,KAAA,CAAM,WAAA,GAAc,GAAA,GAAY,GAAK,CAAA,GAAI,GAAA;AAAA,MACnE,sBACE,IAAA,CAAK,KAAA,CAAM,eAAA,GAAkB,GAAA,GAAY,GAAK,CAAA,GAAI,GAAA;AAAA,MACpD,QAAA,EAAU,YAAA;AAAA,MACV,aAAa,KAAA,CAAM,WAAA;AAAA,MACnB,aAAA,EAAe,KAAA,CAAM,YAAA,EAAc,cAAA,IAAkB,KAAA,CAAM,cAAA;AAAA,MAC3D,mBAAA,EAAqB,MAAM,YAAA,EAAc,qBAAA;AAAA,MACzC,GAAI,WAAA,IAAe,EAAE,gBAAA,EAAkB,IAAA,EAAK;AAAA,MAC5C,GAAI,aAAA,IAAiB,EAAE,aAAA,EAAe,IAAA,EAAK;AAAA,MAC3C,GAAI,yBAAA,IAA6B,EAAE,yBAAA,EAA2B,IAAA,EAAK;AAAA,MACnE,GAAI,aAAA,IACF,CAAC,WAAA,IAAe;AAAA,QACd,kBAAA,EAAoB,IAAA;AAAA,QACpB,oBAAA,EAAsB;AAAA;AACxB,KACJ;AAAA,EACF;AAGA,EAAA,KAAA,MAAW,CAAC,GAAA,EAAK,WAAW,KAAK,MAAA,CAAO,OAAA,CAAQ,YAAY,CAAA,EAAG;AAC7D,IAAA,MAAA,CAAO,GAAG,CAAA,GAAI,WAAA;AAAA,EAChB;AAEA,EAAA,MAAM,WAAA,GAAc,MAAA,CAAO,IAAA,CAAK,MAAM,CAAA;AACtC,EAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,oCAAA,EAAgC,WAAA,CAAY,MAAM,CAAA,CAAE,CAAA;AAGhE,EAAA,MAAM,gBAAA,GAAmB,mBAAmB,MAAM,CAAA;AAGlD,EAAA,MAAM,GAAA,GAAM,QAAQ,UAAU,CAAA;AAC9B,EAAA,IAAI,CAAC,UAAA,CAAW,GAAG,CAAA,EAAG;AACpB,IAAA,SAAA,CAAU,GAAA,EAAK,EAAE,SAAA,EAAW,IAAA,EAAM,CAAA;AAAA,EACpC;AAEA,EAAA,aAAA,CAAc,UAAA,EAAY,kBAAkB,OAAO,CAAA;AACnD,EAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iBAAA,EAAe,UAAU,CAAA,CAAE,CAAA;AACvC,EAAA,OAAA,CAAQ,GAAA;AAAA,IACN;AAAA,GACF;AACA,EAAA,OAAA,CAAQ,IAAI,CAAA,WAAA,EAAc,UAAA,CAAW,QAAQ,GAAA,EAAK,GAAG,CAAC,CAAA,EAAA,CAAI,CAAA;AAC5D;AAEA,SAAS,mBAAmB,MAAA,EAA6C;AACvE,EAAA,MAAM,QAAA,GAAW,MAAA,CAAO,IAAA,CAAK,MAAM,CAAA;AAGnC,EAAA,MAAM,YAAA,GAAe,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,CACvC,IAAI,CAAC,CAAC,GAAA,EAAK,MAAM,CAAA,KAAM;AACtB,IAAA,MAAM,KAAA,GAAQ;AAAA,MACZ,MAAM,GAAG,CAAA,IAAA,CAAA;AAAA,MACT,CAAA,SAAA,EAAY,OAAO,EAAE,CAAA,EAAA,CAAA;AAAA,MACrB,cAAc,MAAA,CAAO,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM,KAAK,CAAC,CAAA,EAAA,CAAA;AAAA,MAC9C,CAAA,kBAAA,EAAqB,MAAA,CAAO,WAAA,EAAa,OAAA,CAAQ,IAAA,EAAM,KAAK,CAAA,CAAE,OAAA,CAAQ,KAAA,EAAO,KAAK,CAAC,CAAA,EAAA,CAAA;AAAA,MACnF,CAAA,yBAAA,EAA4B,OAAO,mBAAmB,CAAA,CAAA,CAAA;AAAA,MACtD,CAAA,0BAAA,EAA6B,OAAO,oBAAoB,CAAA,CAAA,CAAA;AAAA,MACxD,CAAA,eAAA,EAAkB,OAAO,QAAQ,CAAA,EAAA,CAAA;AAAA,MACjC,CAAA,sBAAA,EAAyB,MAAA,CAAO,gBAAA,IAAoB,KAAK,CAAA,CAAA,CAAA;AAAA,MACzD,CAAA,mBAAA,EAAsB,MAAA,CAAO,aAAA,IAAiB,KAAK,CAAA,CAAA,CAAA;AAAA,MACnD,CAAA,+BAAA,EAAkC,MAAA,CAAO,yBAAA,IAA6B,KAAK,CAAA,CAAA,CAAA;AAAA,MAC3E,CAAA,wBAAA,EAA2B,MAAA,CAAO,kBAAA,IAAsB,KAAK,CAAA,CAAA,CAAA;AAAA,MAC7D,CAAA,0BAAA,EAA6B,MAAA,CAAO,oBAAA,IAAwB,CAAC,CAAA,CAAA,CAAA;AAAA,MAC7D,CAAA,mBAAA,EAAsB,MAAA,CAAO,aAAA,IAAiB,CAAC,CAAA,CAAA,CAAA;AAAA,MAC/C,CAAA,yBAAA,EAA4B,OAAO,mBAAmB,CAAA,CAAA,CAAA;AAAA,MACtD,CAAA,IAAA;AAAA,KACF;AAEA,IAAA,OAAO,KAAA,CAAM,KAAK,IAAI,CAAA;AAAA,EACxB,CAAC,CAAA,CACA,IAAA,CAAK,IAAI,CAAA;AAGZ,EAAA,MAAM,eAAA,GAAkB,QAAA,CAAS,GAAA,CAAI,CAAC,EAAA,KAAO,QAAQ,EAAE,CAAA,QAAA,CAAU,CAAA,CAAE,IAAA,CAAK,IAAI,CAAA;AAG5E,EAAA,MAAM,WAAA,GAAc,QAAA,CAAS,GAAA,CAAI,CAAC,EAAA,KAAO,MAAM,EAAE,CAAA,EAAA,CAAI,CAAA,CAAE,IAAA,CAAK,IAAI,CAAA;AAEhE,EAAA,OAAO,CAAA;AAAA;AAAA,iBAAA,EAAA,iBAEU,IAAI,IAAA,EAAK,EAAE,WAAA,EAAa;;AAAA;AAAA;;AAAA;AAAA;AAAA,EAOzC,YAAY;AAAA;;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUZ,WAAW;AAAA;;AAAA;;AAAA;AAAA;AAAA;AAAA,EAQX,eAAe;AAAA;AAAA;;AAAA;AAAA,CAAA;AAMjB;AAEA,IAAA,EAAK,CAAE,KAAA,CAAM,CAAC,GAAA,KAAQ;AACpB,EAAA,OAAA,CAAQ,KAAA,CAAM,iBAAY,GAAG,CAAA;AAC7B,EAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAChB,CAAC,CAAA","file":"sync-models.js","sourcesContent":["#!/usr/bin/env node\n/**\n * workflow-engine-sync - CLI to sync models from OpenRouter API\n *\n * Usage:\n *   npx workflow-engine-sync\n *\n * Reads optional workflow-engine.models.ts config from cwd for:\n * - include: patterns to include models\n * - exclude: patterns to filter out models\n * - customModels: additional models to include\n */\n\n/// <reference types=\"node\" />\n\nimport { existsSync, mkdirSync, writeFileSync } from \"fs\";\nimport { dirname, join, resolve } from \"path\";\nimport { pathToFileURL } from \"url\";\nimport { type ModelConfig, type ModelSyncConfig } from \"../ai/model-helper\";\n\ninterface OpenRouterModel {\n  id: string;\n  name: string;\n  description?: string;\n  context_length?: number;\n  pricing?: {\n    prompt?: string;\n    completion?: string;\n  };\n  supported_parameters: string[];\n  top_provider?: {\n    context_length?: number;\n    max_completion_tokens?: number;\n  };\n}\n\ninterface OpenRouterResponse {\n  data: OpenRouterModel[];\n}\n\n// Main\nasync function main() {\n  const apiKey = process.env.OPENROUTER_API_KEY;\n  if (!apiKey) {\n    console.error(\"❌ OPENROUTER_API_KEY environment variable is required\");\n    process.exit(1);\n  }\n\n  const cwd = process.cwd();\n  console.log(`📁 Working directory: ${cwd}`);\n\n  // Load config if exists\n  let config: ModelSyncConfig = {};\n  const configPath = join(cwd, \"workflow-engine.models.ts\");\n  if (existsSync(configPath)) {\n    try {\n      const configModule = await import(pathToFileURL(configPath).href);\n      config = configModule.default || configModule;\n      console.log(\"📋 Loaded config from workflow-engine.models.ts\");\n    } catch (err) {\n      console.warn(`⚠️  Could not load config: ${err}`);\n    }\n  }\n\n  const outputPath = resolve(\n    cwd,\n    config.outputPath || \"src/generated/models.ts\",\n  );\n  const includePatterns = config.include || [];\n  const excludePatterns = config.exclude || [];\n  const customModels = config.customModels || {};\n\n  // Fetch models from OpenRouter\n  console.log(\"🔄 Fetching models from OpenRouter API...\");\n  const response = await fetch(\"https://openrouter.ai/api/v1/models\", {\n    headers: {\n      Authorization: `Bearer ${apiKey}`,\n      \"Content-Type\": \"application/json\",\n    },\n  });\n\n  if (!response.ok) {\n    console.error(\n      `❌ OpenRouter API error: ${response.status} ${response.statusText}`,\n    );\n    process.exit(1);\n  }\n\n  const data = (await response.json()) as OpenRouterResponse;\n  console.log(`✅ Fetched ${data.data.length} models from OpenRouter`);\n\n  // Fetch embedding models from OpenRouter\n  console.log(\"🔄 Fetching embedding models from OpenRouter API...\");\n  const embeddingResponse = await fetch(\n    \"https://openrouter.ai/api/v1/embeddings/models\",\n    {\n      headers: {\n        Authorization: `Bearer ${apiKey}`,\n        \"Content-Type\": \"application/json\",\n      },\n    },\n  );\n\n  let embeddingModelIds = new Set<string>();\n  if (embeddingResponse.ok) {\n    const embeddingData =\n      (await embeddingResponse.json()) as OpenRouterResponse;\n    console.log(\n      `✅ Fetched ${embeddingData.data.length} embedding models from OpenRouter`,\n    );\n    // Track embedding model IDs for later marking\n    embeddingModelIds = new Set(embeddingData.data.map((m) => m.id));\n    // Add embedding models that might not be in the main list\n    for (const model of embeddingData.data) {\n      if (!data.data.find((m) => m.id === model.id)) {\n        data.data.push(model);\n      }\n    }\n  } else {\n    console.warn(\n      `⚠️  Could not fetch embedding models: ${embeddingResponse.status}`,\n    );\n  }\n\n  // Filter models - include patterns first, then exclude\n  const filteredModels = data.data.filter((model) => {\n    // If include patterns are specified, model must match at least one\n    if (includePatterns.length > 0) {\n      let matchesInclude = false;\n      for (const pattern of includePatterns) {\n        if (typeof pattern === \"string\") {\n          if (model.id.includes(pattern)) {\n            matchesInclude = true;\n            break;\n          }\n        } else if (pattern instanceof RegExp) {\n          if (pattern.test(model.id)) {\n            matchesInclude = true;\n            break;\n          }\n        }\n      }\n      if (!matchesInclude) return false;\n    }\n\n    // Check exclude patterns\n    for (const pattern of excludePatterns) {\n      if (typeof pattern === \"string\") {\n        if (model.id === pattern) return false;\n      } else if (pattern instanceof RegExp) {\n        if (pattern.test(model.id)) return false;\n      }\n    }\n    return true;\n  });\n\n  console.log(\n    `📊 After filtering: ${filteredModels.length} models (excluded ${data.data.length - filteredModels.length})`,\n  );\n\n  // Transform to ModelConfig\n  const models: Record<string, ModelConfig> = {};\n\n  for (const model of filteredModels) {\n    // Convert per-token pricing to per-million\n    const promptPrice = parseFloat(model.pricing?.prompt || \"0\");\n    const completionPrice = parseFloat(model.pricing?.completion || \"0\");\n\n    // Check if model supports async batch (Anthropic, OpenAI, or Google Gemini)\n    const supportsBatch =\n      model.id.startsWith(\"anthropic/\") ||\n      (model.id.startsWith(\"openai/\") && !model.id.includes(\"gpt-oss\")) ||\n      (model.id.startsWith(\"google/\") && model.id.includes(\"gemini\"));\n\n    // Check if this is an embedding model\n    const isEmbedding = embeddingModelIds.has(model.id);\n\n    // Check for tool and structured output support from supported_parameters\n    const supportsTools =\n      model.supported_parameters?.includes(\"tools\") || false;\n    const supportsStructuredOutputs =\n      model.supported_parameters?.includes(\"structured_outputs\") || false;\n\n    models[model.id] = {\n      id: model.id,\n      name: model.name,\n      inputCostPerMillion: Math.round(promptPrice * 1_000_000 * 10000) / 10000,\n      outputCostPerMillion:\n        Math.round(completionPrice * 1_000_000 * 10000) / 10000,\n      provider: \"openrouter\",\n      description: model.description,\n      contextLength: model.top_provider?.context_length || model.context_length,\n      maxCompletionTokens: model.top_provider?.max_completion_tokens,\n      ...(isEmbedding && { isEmbeddingModel: true }),\n      ...(supportsTools && { supportsTools: true }),\n      ...(supportsStructuredOutputs && { supportsStructuredOutputs: true }),\n      ...(supportsBatch &&\n        !isEmbedding && {\n          supportsAsyncBatch: true,\n          batchDiscountPercent: 50,\n        }),\n    };\n  }\n\n  // Merge custom models\n  for (const [key, modelConfig] of Object.entries(customModels)) {\n    models[key] = modelConfig;\n  }\n\n  const allModelIds = Object.keys(models);\n  console.log(`📦 Total models to generate: ${allModelIds.length}`);\n\n  // Generate TypeScript file\n  const generatedContent = generateTypeScript(models);\n\n  // Ensure directory exists\n  const dir = dirname(outputPath);\n  if (!existsSync(dir)) {\n    mkdirSync(dir, { recursive: true });\n  }\n\n  writeFileSync(outputPath, generatedContent, \"utf-8\");\n  console.log(`✅ Generated ${outputPath}`);\n  console.log(\n    \"\\n💡 Import this file in your app entry to enable autocomplete:\",\n  );\n  console.log(`   import \"${outputPath.replace(cwd, \".\")}\";`);\n}\n\nfunction generateTypeScript(models: Record<string, ModelConfig>): string {\n  const modelIds = Object.keys(models);\n\n  // Generate model entries\n  const modelEntries = Object.entries(models)\n    .map(([key, config]) => {\n      const lines = [\n        `  \"${key}\": {`,\n        `    id: \"${config.id}\",`,\n        `    name: \"${config.name.replace(/\"/g, '\\\\\"')}\",`,\n        `    description: \"${config.description?.replace(/\"/g, '\\\\\"').replace(/\\n/g, \"\\\\n\")}\",`,\n        `    inputCostPerMillion: ${config.inputCostPerMillion},`,\n        `    outputCostPerMillion: ${config.outputCostPerMillion},`,\n        `    provider: \"${config.provider}\",`,\n        `    isEmbeddingModel: ${config.isEmbeddingModel || false},`,\n        `    supportsTools: ${config.supportsTools || false},`,\n        `    supportsStructuredOutputs: ${config.supportsStructuredOutputs || false},`,\n        `    supportsAsyncBatch: ${config.supportsAsyncBatch || false},`,\n        `    batchDiscountPercent: ${config.batchDiscountPercent || 0},`,\n        `    contextLength: ${config.contextLength || 0},`,\n        `    maxCompletionTokens: ${config.maxCompletionTokens},`,\n        `  },`,\n      ];\n\n      return lines.join(\"\\n\");\n    })\n    .join(\"\\n\");\n\n  // Generate registry interface entries\n  const registryEntries = modelIds.map((id) => `    \"${id}\": true;`).join(\"\\n\");\n\n  // Generate enum entries for GeneratedModelKey\n  const enumEntries = modelIds.map((id) => `  \"${id}\",`).join(\"\\n\");\n\n  return `// AUTO-GENERATED by workflow-engine-sync\n// Run \\`npx workflow-engine-sync\\` to regenerate\n// Generated at: ${new Date().toISOString()}\n\nimport { z } from \"zod\";\nimport { registerModels, type ModelConfig } from \"@bratsos/workflow-engine/client\";\n\n// Register all models (OpenRouter + custom)\nconst MODELS: Record<string, ModelConfig> = {\n${modelEntries}\n};\n\nregisterModels(MODELS);\n\n/**\n * Zod enum of all generated model IDs\n * Use GeneratedModelKey.enum[\"model-id\"] for type-safe model selection\n */\nexport const GeneratedModelKey = z.enum([\n${enumEntries}\n]);\n\nexport type GeneratedModelKey = z.infer<typeof GeneratedModelKey>;\n\n// TypeScript module augmentation for autocomplete\ndeclare module \"@bratsos/workflow-engine/client\" {\n  interface ModelRegistry {\n${registryEntries}\n  }\n}\n\nexport { MODELS };\n`;\n}\n\nmain().catch((err) => {\n  console.error(\"❌ Error:\", err);\n  process.exit(1);\n});\n"]}