modelrig 0.0.1 → 0.1.0

package/dist/watch/state.d.ts

@@ -0,0 +1,21 @@
+/**
+ * Watch state — Phase 4 WS1. The cycle's memory: a JSON snapshot of the
+ * last-seen pricing universe (provider/model → token rates), diffed against
+ * the current pricing map each run. Seed runs emit no events by design.
+ *
+ * The universe is restricted to providers the loaded routes actually
+ * declare — a price move on a provider nobody routes to is not news.
+ */
+import type { TokenRates, WatchEvent } from "./evaluate";
+/** provider/model → rates. Keys use OUR provider ids, not LiteLLM's. */
+export type PriceUniverse = Record<string, TokenRates>;
+/** Build the current universe from the vendored pricing map, filtered to
+ * the providers in scope. Model ids are normalized to the bare form route
+ * candidates use ("gemini/gemini-2.5-pro" → "gemini-2.5-pro"). */
+export declare function buildPriceUniverse(providersInScope: ReadonlySet<string>, pricingMapPath?: string): PriceUniverse;
+/** Pure diff: current vs last-seen → events. Null previous = seed run,
+ * nothing fires. Delisted models are silence, not events. */
+export declare function computeWatchEvents(current: PriceUniverse, previous: PriceUniverse | null): WatchEvent[];
+export declare function loadWatchState(path: string): PriceUniverse | null;
+export declare function saveWatchState(path: string, universe: PriceUniverse): void;
+//# sourceMappingURL=state.d.ts.map

package/dist/watch/state.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"state.d.ts","sourceRoot":"","sources":["../../src/watch/state.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAIH,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAEzD,wEAAwE;AACxE,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;AAkCvD;;kEAEkE;AAClE,wBAAgB,kBAAkB,CAChC,gBAAgB,EAAE,WAAW,CAAC,MAAM,CAAC,EACrC,cAAc,CAAC,EAAE,MAAM,GACtB,aAAa,CAoBf;AAED;6DAC6D;AAC7D,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,aAAa,EACtB,QAAQ,EAAE,aAAa,GAAG,IAAI,GAC7B,UAAU,EAAE,CAkBd;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI,CAGjE;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,aAAa,GAAG,IAAI,CAG1E"}

package/dist/watch/state.js

@@ -0,0 +1,93 @@
+"use strict";
+/**
+ * Watch state — Phase 4 WS1. The cycle's memory: a JSON snapshot of the
+ * last-seen pricing universe (provider/model → token rates), diffed against
+ * the current pricing map each run. Seed runs emit no events by design.
+ *
+ * The universe is restricted to providers the loaded routes actually
+ * declare — a price move on a provider nobody routes to is not news.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildPriceUniverse = buildPriceUniverse;
+exports.computeWatchEvents = computeWatchEvents;
+exports.loadWatchState = loadWatchState;
+exports.saveWatchState = saveWatchState;
+const node_fs_1 = require("node:fs");
+const node_path_1 = require("node:path");
+/** LiteLLM's litellm_provider values → our ProviderId space (watch scope:
+ * the launch providers; extending = adding a line). */
+const LITELLM_PROVIDER_MAP = {
+    gemini: "gemini",
+    openai: "openai",
+    deepseek: "deepseek",
+    anthropic: "anthropic",
+    xai: "grok",
+    // A1 hosts (openai-compatible generic adapter).
+    deepinfra: "deepinfra",
+    fireworks_ai: "fireworks",
+};
+/** Normalize a pricing-map key to the bare model id route candidates use:
+ * strip the leading litellm-provider segment, then Fireworks' wire prefix.
+ * DeepInfra org paths ("Qwen/QwQ-32B") are PART of the model id — kept. */
+function normalizeModelId(rawKey, litellmProvider) {
+    let model = rawKey.startsWith(`${litellmProvider}/`)
+        ? rawKey.slice(litellmProvider.length + 1)
+        : rawKey;
+    if (model.startsWith("accounts/fireworks/models/")) {
+        model = model.slice("accounts/fireworks/models/".length);
+    }
+    return model;
+}
+/** Build the current universe from the vendored pricing map, filtered to
+ * the providers in scope. Model ids are normalized to the bare form route
+ * candidates use ("gemini/gemini-2.5-pro" → "gemini-2.5-pro"). */
+function buildPriceUniverse(providersInScope, pricingMapPath) {
+    const path = pricingMapPath ?? (0, node_path_1.join)(__dirname, "..", "..", "data", "pricing-map.json");
+    const map = JSON.parse((0, node_fs_1.readFileSync)(path, "utf8"));
+    const universe = {};
+    for (const [rawKey, entry] of Object.entries(map)) {
+        const provider = LITELLM_PROVIDER_MAP[entry.litellm_provider ?? ""];
+        if (provider === undefined || !providersInScope.has(provider))
+            continue;
+        if (typeof entry.input_cost_per_token !== "number" ||
+            typeof entry.output_cost_per_token !== "number") {
+            continue;
+        }
+        const model = normalizeModelId(rawKey, entry.litellm_provider ?? "");
+        universe[`${provider}/${model}`] = {
+            inputUsdPerTok: entry.input_cost_per_token,
+            outputUsdPerTok: entry.output_cost_per_token,
+        };
+    }
+    return universe;
+}
+/** Pure diff: current vs last-seen → events. Null previous = seed run,
+ * nothing fires. Delisted models are silence, not events. */
+function computeWatchEvents(current, previous) {
+    if (previous === null)
+        return [];
+    const events = [];
+    for (const [key, rates] of Object.entries(current)) {
+        const slash = key.indexOf("/");
+        const provider = key.slice(0, slash);
+        const model = key.slice(slash + 1);
+        const before = previous[key];
+        if (before === undefined) {
+            events.push({ type: "new_model", provider, model, pricing: rates });
+        }
+        else if (before.inputUsdPerTok !== rates.inputUsdPerTok ||
+            before.outputUsdPerTok !== rates.outputUsdPerTok) {
+            events.push({ type: "price_change", provider, model, old: before, new: rates });
+        }
+    }
+    return events;
+}
+function loadWatchState(path) {
+    if (!(0, node_fs_1.existsSync)(path))
+        return null;
+    return JSON.parse((0, node_fs_1.readFileSync)(path, "utf8"));
+}
+function saveWatchState(path, universe) {
+    (0, node_fs_1.mkdirSync)((0, node_path_1.dirname)(path), { recursive: true });
+    (0, node_fs_1.writeFileSync)(path, `${JSON.stringify(universe, null, 2)}\n`);
+}

package/package.json

@@ -1,20 +1,56 @@
 {
   "name": "modelrig",
-  "version": "0.0.1",
-  "description": "ModelRig — the capability-aware AI gateway. Rig it. Prove it. Sell it. Full release coming soon.",
-  "homepage": "https://modelrig.ai",
+  "version": "0.1.0",
+  "private": false,
+  "description": "ModelRig — capability-aware AI gateway: routes-as-code, provider adapters, typed failures, budget envelopes, and local-first telemetry.",
   "license": "Apache-2.0",
-  "main": "index.js",
-  "files": [
-    "index.js",
-    "README.md"
-  ],
+  "homepage": "https://modelrig.dev",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/modelrig/modelrig.git"
+  },
   "keywords": [
-    "ai",
     "llm",
-    "gateway",
-    "routing",
-    "evals",
-    "structured-output"
-  ]
+    "ai-gateway",
+    "model-routing",
+    "openai",
+    "anthropic",
+    "gemini",
+    "telemetry",
+    "failover"
+  ],
+  "engines": {
+    "node": ">=20"
+  },
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "//files": "ALLOWLIST, not a denylist, and the difference is not academic. Without this field npm falls back to .npmignore, then to THIS DIRECTORY's .gitignore — never the repo root's. The root ignores `.env.*`, which is why git has never committed .env.local; the package-level .gitignore says nothing about it, so `npm pack` happily included it. A publish would have shipped 13 live secrets, the Supabase service-role key among them, into an immutable public tarball. Anything added here must be something we would hand a stranger. See __tests__/publish-safety.test.ts.",
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "type-check": "tsc --noEmit && tsc -p tsconfig.test.json",
+    "test": "vitest run",
+    "test:live": "MODELRIG_LIVE=1 vitest run"
+  },
+  "dependencies": {
+    "@anthropic-ai/sdk": "^0.71.2",
+    "@google/genai": "^2.4.0",
+    "ajv": "^8.17.0",
+    "better-sqlite3": "^11.10.0",
+    "openai": "^4.77.0",
+    "yaml": "^2.6.0"
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.12",
+    "tsx": "^4.21.0",
+    "typescript": "^5.2.0",
+    "vitest": "^2.1.0"
+  },
+  "bin": {
+    "modelrig": "dist/cli.js"
+  }
 }

package/index.js

@@ -1,3 +0,0 @@
-module.exports = {
-  message: "ModelRig is coming soon. See https://modelrig.ai",
-};