@hoangsonw/forge 0.1.0

package/dist/models/cache.js

@@ -0,0 +1,135 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.clear = exports.store = exports.lookup = void 0;
+/**
+ * Prompt-level response cache. Keyed by the deterministic prompt hash from
+ * the assembler plus call options. Only caches when temperature=0 (or
+ * deterministic=true). Keeps a small LRU in-process + an optional SQLite
+ * mirror so repeated runs benefit from prior output without hitting the
+ * model again.
+ */
+const crypto = __importStar(require("crypto"));
+const index_db_1 = require("../persistence/index-db");
+const logger_1 = require("../logging/logger");
+const migrate = () => {
+    (0, index_db_1.getDb)().exec(`
+    CREATE TABLE IF NOT EXISTS prompt_cache (
+      key TEXT PRIMARY KEY,
+      provider TEXT NOT NULL,
+      model TEXT NOT NULL,
+      content TEXT NOT NULL,
+      input_tokens INTEGER,
+      output_tokens INTEGER,
+      finish_reason TEXT,
+      created_at TEXT NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_prompt_cache_created ON prompt_cache(created_at);
+  `);
+};
+const MAX_LIVE = 128;
+const live = [];
+const makeKey = (provider, model, messages, options) => {
+    const hasher = crypto.createHash('sha256');
+    hasher.update(`${provider}::${model}::`);
+    for (const m of messages)
+        hasher.update(`${m.role}::${m.content}\n`);
+    hasher.update(JSON.stringify({ ...options, timeoutMs: undefined }));
+    return hasher.digest('hex');
+};
+const isCacheable = (options) => options.deterministic === true || (options.temperature ?? 0.3) === 0;
+const lookup = (provider, model, messages, options) => {
+    if (!isCacheable(options))
+        return null;
+    const key = makeKey(provider, model, messages, options);
+    const fromLive = live.find((e) => e.key === key);
+    if (fromLive)
+        return fromLive.value;
+    try {
+        migrate();
+        const row = (0, index_db_1.getDb)()
+            .prepare('SELECT content, provider, model, input_tokens, output_tokens, finish_reason FROM prompt_cache WHERE key = ?')
+            .get(key);
+        if (!row)
+            return null;
+        return {
+            content: row.content,
+            provider: row.provider,
+            model: row.model,
+            inputTokens: row.input_tokens ?? undefined,
+            outputTokens: row.output_tokens ?? undefined,
+            finishReason: row.finish_reason ?? 'stop',
+            durationMs: 0,
+        };
+    }
+    catch (err) {
+        logger_1.log.debug('prompt cache lookup failed', { err: String(err) });
+        return null;
+    }
+};
+exports.lookup = lookup;
+const store = (provider, model, messages, options, response) => {
+    if (!isCacheable(options))
+        return;
+    const key = makeKey(provider, model, messages, options);
+    live.push({ key, value: response, addedAt: Date.now() });
+    if (live.length > MAX_LIVE)
+        live.shift();
+    try {
+        migrate();
+        (0, index_db_1.getDb)()
+            .prepare(`INSERT INTO prompt_cache (key, provider, model, content, input_tokens, output_tokens, finish_reason, created_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+         ON CONFLICT(key) DO UPDATE SET content = excluded.content, created_at = excluded.created_at`)
+            .run(key, provider, model, response.content, response.inputTokens ?? null, response.outputTokens ?? null, response.finishReason ?? null, new Date().toISOString());
+    }
+    catch (err) {
+        logger_1.log.debug('prompt cache store failed', { err: String(err) });
+    }
+};
+exports.store = store;
+const clear = () => {
+    live.length = 0;
+    try {
+        migrate();
+        const r = (0, index_db_1.getDb)().prepare('DELETE FROM prompt_cache').run();
+        return r.changes;
+    }
+    catch {
+        return 0;
+    }
+};
+exports.clear = clear;
+//# sourceMappingURL=cache.js.map

package/dist/models/cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.js","sourceRoot":"","sources":["../../src/models/cache.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;;;;;;GAMG;AACH,+CAAiC;AAEjC,sDAAgD;AAChD,8CAAwC;AAExC,MAAM,OAAO,GAAG,GAAS,EAAE;IACzB,IAAA,gBAAK,GAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;GAYZ,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,QAAQ,GAAG,GAAG,CAAC;AAMrB,MAAM,IAAI,GAAY,EAAE,CAAC;AAEzB,MAAM,OAAO,GAAG,CACd,QAAgB,EAChB,KAAa,EACb,QAAwB,EACxB,OAAyB,EACjB,EAAE;IACV,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IAC3C,MAAM,CAAC,MAAM,CAAC,GAAG,QAAQ,KAAK,KAAK,IAAI,CAAC,CAAC;IACzC,KAAK,MAAM,CAAC,IAAI,QAAQ;QAAE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC;IACrE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IACpE,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,CAAC,OAAyB,EAAW,EAAE,CACzD,OAAO,CAAC,aAAa,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;AAEhE,MAAM,MAAM,GAAG,CACpB,QAAgB,EAChB,KAAa,EACb,QAAwB,EACxB,OAAyB,EACH,EAAE;IACxB,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;QAAE,OAAO,IAAI,CAAC;IACvC,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC;IACjD,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC,KAAK,CAAC;IACpC,IAAI,CAAC;QACH,OAAO,EAAE,CAAC;QACV,MAAM,GAAG,GAAG,IAAA,gBAAK,GAAE;aAChB,OAAO,CACN,6GAA6G,CAC9G;aACA,GAAG,CAAC,GAAG,CASG,CAAC;QACd,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,OAAO;YACL,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,WAAW,EAAE,GAAG,CAAC,YAAY,IAAI,SAAS;YAC1C,YAAY,EAAE,GAAG,CAAC,aAAa,IAAI,SAAS;YAC5C,YAAY,EAAG,GAAG,CAAC,aAA+C,IAAI,MAAM;YAC5E,UAAU,EAAE,CAAC;SACd,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,YAAG,CAAC,KAAK,CAAC,4BAA4B,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC,CAAC;AAxCW,QAAA,MAAM,UAwCjB;AAEK,MAAM,KAAK,GAAG,CACnB,QAAgB,EAChB,KAAa,EACb,QAAwB,EACxB,OAAyB,EACzB,QAAuB,EACjB,EAAE;IACR,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;QAAE,OAAO;IAClC,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IACxD,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACzD,IAAI,IAAI,CAAC,MAAM,GAAG,QAAQ;QAAE,IAAI,CAAC,KAAK,EAAE,CAAC;IACzC,IAAI,CAAC;QACH,OAAO,EAAE,CAAC;QACV,IAAA,gBAAK,GAAE;aACJ,OAAO,CACN;;qGAE6F,CAC9F;aACA,GAAG,CACF,GAAG,EACH,QAAQ,EACR,KAAK,EACL,QAAQ,CAAC,OAAO,EAChB,QAAQ,CAAC,WAAW,IAAI,IAAI,EAC5B,QAAQ,CAAC,YAAY,IAAI,IAAI,EAC7B,QAAQ,CAAC,YAAY,IAAI,IAAI,EAC7B,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CACzB,CAAC;IACN,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,YAAG,CAAC,KAAK,CAAC,2BAA2B,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/D,CAAC;AACH,CAAC,CAAC;AAhCW,QAAA,KAAK,SAgChB;AAEK,MAAM,KAAK,GAAG,GAAW,EAAE;IAChC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IAChB,IAAI,CAAC;QACH,OAAO,EAAE,CAAC;QACV,MAAM,CAAC,GAAG,IAAA,gBAAK,GAAE,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC,GAAG,EAAE,CAAC;QAC5D,OAAO,CAAC,CAAC,OAAO,CAAC;IACnB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC,CAAC;AATW,QAAA,KAAK,SAShB"}

package/dist/models/circuit-breaker.d.ts

@@ -0,0 +1,18 @@
+/**
+ * Minimal circuit breaker per provider. Opens after `failureThreshold`
+ * consecutive failures, waits `resetMs` before allowing a single probe, and
+ * closes on a successful probe. Half-open state is tracked but the public
+ * surface only distinguishes "can try" from "blocked".
+ */
+export type BreakerState = 'closed' | 'open' | 'half_open';
+export interface BreakerConfig {
+    failureThreshold: number;
+    resetMs: number;
+}
+export declare const configure: (provider: string, cfg: Partial<BreakerConfig>) => void;
+export declare const canTry: (provider: string) => boolean;
+export declare const reportSuccess: (provider: string) => void;
+export declare const reportFailure: (provider: string) => void;
+export declare const status: () => Record<string, BreakerState>;
+export declare const reset: (provider?: string) => void;
+//# sourceMappingURL=circuit-breaker.d.ts.map

package/dist/models/circuit-breaker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"circuit-breaker.d.ts","sourceRoot":"","sources":["../../src/models/circuit-breaker.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;AAU3D,MAAM,WAAW,aAAa;IAC5B,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;CACjB;AAeD,eAAO,MAAM,SAAS,GAAI,UAAU,MAAM,EAAE,KAAK,OAAO,CAAC,aAAa,CAAC,KAAG,IAGzE,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,UAAU,MAAM,KAAG,OAWzC,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,UAAU,MAAM,KAAG,IAIhD,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,UAAU,MAAM,KAAG,IAQhD,CAAC;AAEF,eAAO,MAAM,MAAM,QAAO,MAAM,CAAC,MAAM,EAAE,YAAY,CAIpD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,WAAW,MAAM,KAAG,IAGzC,CAAC"}

package/dist/models/circuit-breaker.js

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.reset = exports.status = exports.reportFailure = exports.reportSuccess = exports.canTry = exports.configure = void 0;
+const DEFAULTS = { failureThreshold: 5, resetMs: 60_000 };
+const state = new Map();
+const get = (provider) => {
+    let s = state.get(provider);
+    if (!s) {
+        s = { state: 'closed', failures: 0, opened: 0, lastFailure: 0, cfg: DEFAULTS };
+        state.set(provider, s);
+    }
+    return s;
+};
+const configure = (provider, cfg) => {
+    const s = get(provider);
+    s.cfg = { ...s.cfg, ...cfg };
+};
+exports.configure = configure;
+const canTry = (provider) => {
+    const s = get(provider);
+    if (s.state === 'closed')
+        return true;
+    if (s.state === 'open') {
+        if (Date.now() - s.opened >= s.cfg.resetMs) {
+            s.state = 'half_open';
+            return true;
+        }
+        return false;
+    }
+    return true; // half_open: allow the probe
+};
+exports.canTry = canTry;
+const reportSuccess = (provider) => {
+    const s = get(provider);
+    s.failures = 0;
+    s.state = 'closed';
+};
+exports.reportSuccess = reportSuccess;
+const reportFailure = (provider) => {
+    const s = get(provider);
+    s.failures++;
+    s.lastFailure = Date.now();
+    if (s.failures >= s.cfg.failureThreshold) {
+        s.state = 'open';
+        s.opened = Date.now();
+    }
+};
+exports.reportFailure = reportFailure;
+const status = () => {
+    const out = {};
+    for (const [k, v] of state)
+        out[k] = v.state;
+    return out;
+};
+exports.status = status;
+const reset = (provider) => {
+    if (provider)
+        state.delete(provider);
+    else
+        state.clear();
+};
+exports.reset = reset;
+//# sourceMappingURL=circuit-breaker.js.map

package/dist/models/circuit-breaker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"circuit-breaker.js","sourceRoot":"","sources":["../../src/models/circuit-breaker.ts"],"names":[],"mappings":";;;AAqBA,MAAM,QAAQ,GAAkB,EAAE,gBAAgB,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAEzE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAiB,CAAC;AAEvC,MAAM,GAAG,GAAG,CAAC,QAAgB,EAAS,EAAE;IACtC,IAAI,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC5B,IAAI,CAAC,CAAC,EAAE,CAAC;QACP,CAAC,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC;QAC/E,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;IACzB,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC,CAAC;AAEK,MAAM,SAAS,GAAG,CAAC,QAAgB,EAAE,GAA2B,EAAQ,EAAE;IAC/E,MAAM,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC,CAAC,GAAG,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;AAC/B,CAAC,CAAC;AAHW,QAAA,SAAS,aAGpB;AAEK,MAAM,MAAM,GAAG,CAAC,QAAgB,EAAW,EAAE;IAClD,MAAM,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxB,IAAI,CAAC,CAAC,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACtC,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;QACvB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YAC3C,CAAC,CAAC,KAAK,GAAG,WAAW,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC,CAAC,6BAA6B;AAC5C,CAAC,CAAC;AAXW,QAAA,MAAM,UAWjB;AAEK,MAAM,aAAa,GAAG,CAAC,QAAgB,EAAQ,EAAE;IACtD,MAAM,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC;IACf,CAAC,CAAC,KAAK,GAAG,QAAQ,CAAC;AACrB,CAAC,CAAC;AAJW,QAAA,aAAa,iBAIxB;AAEK,MAAM,aAAa,GAAG,CAAC,QAAgB,EAAQ,EAAE;IACtD,MAAM,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC,CAAC,QAAQ,EAAE,CAAC;IACb,CAAC,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3B,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,KAAK,GAAG,MAAM,CAAC;QACjB,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACxB,CAAC;AACH,CAAC,CAAC;AARW,QAAA,aAAa,iBAQxB;AAEK,MAAM,MAAM,GAAG,GAAiC,EAAE;IACvD,MAAM,GAAG,GAAiC,EAAE,CAAC;IAC7C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,KAAK;QAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;IAC7C,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAJW,QAAA,MAAM,UAIjB;AAEK,MAAM,KAAK,GAAG,CAAC,QAAiB,EAAQ,EAAE;IAC/C,IAAI,QAAQ;QAAE,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;;QAChC,KAAK,CAAC,KAAK,EAAE,CAAC;AACrB,CAAC,CAAC;AAHW,QAAA,KAAK,SAGhB"}

package/dist/models/cost.d.ts

@@ -0,0 +1,13 @@
+import { ModelResponse } from '../types';
+export declare const estimateCostUsd: (provider: string, model: string, input: number, output: number) => number;
+export declare const record: (ctx: {
+    projectId?: string;
+    taskId?: string;
+}, response: ModelResponse) => number;
+export declare const totals: (projectId?: string) => {
+    calls: number;
+    tokens: number;
+    usd: number;
+};
+export declare const recent: (limit?: number) => unknown[];
+//# sourceMappingURL=cost.d.ts.map

package/dist/models/cost.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost.d.ts","sourceRoot":"","sources":["../../src/models/cost.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAyCzC,eAAO,MAAM,eAAe,GAC1B,UAAU,MAAM,EAChB,OAAO,MAAM,EACb,OAAO,MAAM,EACb,QAAQ,MAAM,KACb,MAKF,CAAC;AAEF,eAAO,MAAM,MAAM,GACjB,KAAK;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,EAC5C,UAAU,aAAa,KACtB,MA4BF,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,YAAY,MAAM,KAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAevF,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,cAAU,cAQhC,CAAC"}

package/dist/models/cost.js

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.recent = exports.totals = exports.record = exports.estimateCostUsd = void 0;
+/**
+ * Cost ledger. Every model response is recorded with an estimated USD cost
+ * based on static per-model rates. Costs are a rough indicator — the
+ * provider's invoice is the source of truth. Rates live in one table so
+ * corrections are cheap.
+ */
+const index_db_1 = require("../persistence/index-db");
+const logger_1 = require("../logging/logger");
+const migrate = () => {
+    (0, index_db_1.getDb)().exec(`
+    CREATE TABLE IF NOT EXISTS model_cost_ledger (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      project_id TEXT,
+      task_id TEXT,
+      provider TEXT NOT NULL,
+      model TEXT NOT NULL,
+      input_tokens INTEGER NOT NULL,
+      output_tokens INTEGER NOT NULL,
+      duration_ms INTEGER NOT NULL,
+      cost_usd REAL NOT NULL,
+      created_at TEXT NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_cost_project ON model_cost_ledger(project_id);
+    CREATE INDEX IF NOT EXISTS idx_cost_created ON model_cost_ledger(created_at);
+  `);
+};
+// Approximate public rates. Local models (ollama/llamacpp) are free.
+const RATE_TABLE = [
+    { match: /^claude-opus/, rate: { inputUsdPerMTok: 15, outputUsdPerMTok: 75 } },
+    { match: /^claude-sonnet/, rate: { inputUsdPerMTok: 3, outputUsdPerMTok: 15 } },
+    { match: /^claude-haiku/, rate: { inputUsdPerMTok: 0.8, outputUsdPerMTok: 4 } },
+    { match: /^gpt-4o-mini/, rate: { inputUsdPerMTok: 0.15, outputUsdPerMTok: 0.6 } },
+    { match: /^gpt-4o/, rate: { inputUsdPerMTok: 2.5, outputUsdPerMTok: 10 } },
+    { match: /^gpt-4-turbo/, rate: { inputUsdPerMTok: 10, outputUsdPerMTok: 30 } },
+    { match: /^gpt-4/, rate: { inputUsdPerMTok: 30, outputUsdPerMTok: 60 } },
+    { match: /^gpt-3\.5/, rate: { inputUsdPerMTok: 0.5, outputUsdPerMTok: 1.5 } },
+    { match: /^o1-mini/, rate: { inputUsdPerMTok: 3, outputUsdPerMTok: 12 } },
+    { match: /^o1/, rate: { inputUsdPerMTok: 15, outputUsdPerMTok: 60 } },
+];
+const estimateCostUsd = (provider, model, input, output) => {
+    if (provider === 'ollama' || provider === 'llamacpp')
+        return 0;
+    const rate = RATE_TABLE.find((r) => r.match.test(model));
+    if (!rate)
+        return 0;
+    return (input * rate.rate.inputUsdPerMTok + output * rate.rate.outputUsdPerMTok) / 1_000_000;
+};
+exports.estimateCostUsd = estimateCostUsd;
+const record = (ctx, response) => {
+    try {
+        migrate();
+        const input = response.inputTokens ?? 0;
+        const output = response.outputTokens ?? 0;
+        const cost = (0, exports.estimateCostUsd)(response.provider, response.model, input, output);
+        (0, index_db_1.getDb)()
+            .prepare(`INSERT INTO model_cost_ledger
+         (project_id, task_id, provider, model, input_tokens, output_tokens, duration_ms, cost_usd, created_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`)
+            .run(ctx.projectId ?? null, ctx.taskId ?? null, response.provider, response.model, input, output, response.durationMs, cost, new Date().toISOString());
+        return cost;
+    }
+    catch (err) {
+        logger_1.log.debug('cost ledger write failed', { err: String(err) });
+        return 0;
+    }
+};
+exports.record = record;
+const totals = (projectId) => {
+    migrate();
+    const db = (0, index_db_1.getDb)();
+    const row = projectId
+        ? db
+            .prepare('SELECT COUNT(*) as calls, COALESCE(SUM(input_tokens + output_tokens), 0) as tokens, COALESCE(SUM(cost_usd), 0) as usd FROM model_cost_ledger WHERE project_id = ?')
+            .get(projectId)
+        : db
+            .prepare('SELECT COUNT(*) as calls, COALESCE(SUM(input_tokens + output_tokens), 0) as tokens, COALESCE(SUM(cost_usd), 0) as usd FROM model_cost_ledger')
+            .get();
+    return row;
+};
+exports.totals = totals;
+const recent = (limit = 25) => {
+    migrate();
+    return (0, index_db_1.getDb)()
+        .prepare(`SELECT provider, model, input_tokens, output_tokens, duration_ms, cost_usd, created_at
+       FROM model_cost_ledger ORDER BY id DESC LIMIT ?`)
+        .all(limit);
+};
+exports.recent = recent;
+//# sourceMappingURL=cost.js.map

package/dist/models/cost.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost.js","sourceRoot":"","sources":["../../src/models/cost.ts"],"names":[],"mappings":";;;AAAA;;;;;GAKG;AACH,sDAAgD;AAEhD,8CAAwC;AAExC,MAAM,OAAO,GAAG,GAAS,EAAE;IACzB,IAAA,gBAAK,GAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;GAeZ,CAAC,CAAC;AACL,CAAC,CAAC;AAOF,qEAAqE;AACrE,MAAM,UAAU,GAAyC;IACvD,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE;IAC9E,EAAE,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,CAAC,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE;IAC/E,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,GAAG,EAAE,gBAAgB,EAAE,CAAC,EAAE,EAAE;IAC/E,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE,EAAE;IACjF,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,GAAG,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE;IAC1E,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE;IAC9E,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE;IACxE,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,GAAG,EAAE,gBAAgB,EAAE,GAAG,EAAE,EAAE;IAC7E,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,CAAC,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE;IACzE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,EAAE;CACtE,CAAC;AAEK,MAAM,eAAe,GAAG,CAC7B,QAAgB,EAChB,KAAa,EACb,KAAa,EACb,MAAc,EACN,EAAE;IACV,IAAI,QAAQ,KAAK,QAAQ,IAAI,QAAQ,KAAK,UAAU;QAAE,OAAO,CAAC,CAAC;IAC/D,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACzD,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,CAAC;IACpB,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,GAAG,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,SAAS,CAAC;AAC/F,CAAC,CAAC;AAVW,QAAA,eAAe,mBAU1B;AAEK,MAAM,MAAM,GAAG,CACpB,GAA4C,EAC5C,QAAuB,EACf,EAAE;IACV,IAAI,CAAC;QACH,OAAO,EAAE,CAAC;QACV,MAAM,KAAK,GAAG,QAAQ,CAAC,WAAW,IAAI,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC;QAC1C,MAAM,IAAI,GAAG,IAAA,uBAAe,EAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/E,IAAA,gBAAK,GAAE;aACJ,OAAO,CACN;;4CAEoC,CACrC;aACA,GAAG,CACF,GAAG,CAAC,SAAS,IAAI,IAAI,EACrB,GAAG,CAAC,MAAM,IAAI,IAAI,EAClB,QAAQ,CAAC,QAAQ,EACjB,QAAQ,CAAC,KAAK,EACd,KAAK,EACL,MAAM,EACN,QAAQ,CAAC,UAAU,EACnB,IAAI,EACJ,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CACzB,CAAC;QACJ,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,YAAG,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC5D,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC,CAAC;AA/BW,QAAA,MAAM,UA+BjB;AAEK,MAAM,MAAM,GAAG,CAAC,SAAkB,EAAkD,EAAE;IAC3F,OAAO,EAAE,CAAC;IACV,MAAM,EAAE,GAAG,IAAA,gBAAK,GAAE,CAAC;IACnB,MAAM,GAAG,GAAG,SAAS;QACnB,CAAC,CAAC,EAAE;aACC,OAAO,CACN,mKAAmK,CACpK;aACA,GAAG,CAAC,SAAS,CAAC;QACnB,CAAC,CAAC,EAAE;aACC,OAAO,CACN,8IAA8I,CAC/I;aACA,GAAG,EAAE,CAAC;IACb,OAAO,GAAqD,CAAC;AAC/D,CAAC,CAAC;AAfW,QAAA,MAAM,UAejB;AAEK,MAAM,MAAM,GAAG,CAAC,KAAK,GAAG,EAAE,EAAE,EAAE;IACnC,OAAO,EAAE,CAAC;IACV,OAAO,IAAA,gBAAK,GAAE;SACX,OAAO,CACN;uDACiD,CAClD;SACA,GAAG,CAAC,KAAK,CAAC,CAAC;AAChB,CAAC,CAAC;AARW,QAAA,MAAM,UAQjB"}

package/dist/models/llamacpp.d.ts

@@ -0,0 +1,9 @@
+import { OpenAIProvider } from './openai';
+/**
+ * llama.cpp via its OpenAI-compatible `server` endpoint. Same wire protocol,
+ * just a different default base URL and no API key required.
+ */
+export declare class LlamaCppProvider extends OpenAIProvider {
+    constructor(endpoint?: string);
+}
+//# sourceMappingURL=llamacpp.d.ts.map

package/dist/models/llamacpp.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamacpp.d.ts","sourceRoot":"","sources":["../../src/models/llamacpp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C;;;GAGG;AACH,qBAAa,gBAAiB,SAAQ,cAAc;gBACtC,QAAQ,GAAE,MAAoE;CAG3F"}

package/dist/models/llamacpp.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LlamaCppProvider = void 0;
+const openai_1 = require("./openai");
+/**
+ * llama.cpp via its OpenAI-compatible `server` endpoint. Same wire protocol,
+ * just a different default base URL and no API key required.
+ */
+class LlamaCppProvider extends openai_1.OpenAIProvider {
+    constructor(endpoint = process.env.LLAMACPP_ENDPOINT ?? 'http://127.0.0.1:8080/v1') {
+        super(process.env.LLAMACPP_API_KEY, endpoint, 'llamacpp');
+    }
+}
+exports.LlamaCppProvider = LlamaCppProvider;
+//# sourceMappingURL=llamacpp.js.map

package/dist/models/llamacpp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamacpp.js","sourceRoot":"","sources":["../../src/models/llamacpp.ts"],"names":[],"mappings":";;;AAAA,qCAA0C;AAE1C;;;GAGG;AACH,MAAa,gBAAiB,SAAQ,uBAAc;IAClD,YAAY,WAAmB,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,0BAA0B;QACxF,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC5D,CAAC;CACF;AAJD,4CAIC"}

package/dist/models/lmstudio.d.ts

@@ -0,0 +1,11 @@
+import { OpenAIProvider } from './openai';
+/**
+ * LM Studio's built-in server (Local Server → "Start Server"). Defaults to
+ * http://127.0.0.1:1234/v1 and speaks the OpenAI chat-completions API.
+ *
+ * Override with `LMSTUDIO_ENDPOINT`.
+ */
+export declare class LmStudioProvider extends OpenAIProvider {
+    constructor(endpoint?: string);
+}
+//# sourceMappingURL=lmstudio.d.ts.map

package/dist/models/lmstudio.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lmstudio.d.ts","sourceRoot":"","sources":["../../src/models/lmstudio.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C;;;;;GAKG;AACH,qBAAa,gBAAiB,SAAQ,cAAc;gBACtC,QAAQ,GAAE,MAAoE;CAI3F"}

package/dist/models/lmstudio.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LmStudioProvider = void 0;
+const openai_1 = require("./openai");
+/**
+ * LM Studio's built-in server (Local Server → "Start Server"). Defaults to
+ * http://127.0.0.1:1234/v1 and speaks the OpenAI chat-completions API.
+ *
+ * Override with `LMSTUDIO_ENDPOINT`.
+ */
+class LmStudioProvider extends openai_1.OpenAIProvider {
+    constructor(endpoint = process.env.LMSTUDIO_ENDPOINT ?? 'http://127.0.0.1:1234/v1') {
+        // LM Studio does not require an API key but will ignore one if sent.
+        super(process.env.LMSTUDIO_API_KEY ?? 'lm-studio', endpoint, 'lmstudio');
+    }
+}
+exports.LmStudioProvider = LmStudioProvider;
+//# sourceMappingURL=lmstudio.js.map

package/dist/models/lmstudio.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lmstudio.js","sourceRoot":"","sources":["../../src/models/lmstudio.ts"],"names":[],"mappings":";;;AAAA,qCAA0C;AAE1C;;;;;GAKG;AACH,MAAa,gBAAiB,SAAQ,uBAAc;IAClD,YAAY,WAAmB,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,0BAA0B;QACxF,qEAAqE;QACrE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,WAAW,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC3E,CAAC;CACF;AALD,4CAKC"}

package/dist/models/local-catalog.d.ts

@@ -0,0 +1,45 @@
+/**
+ * Local-model catalogue: one place to map a model id → Forge's routing
+ * metadata. Used by every local-runtime provider (Ollama, llama.cpp, vLLM,
+ * LM Studio, LocalAI, OpenAI-compatible generic) so they stay consistent.
+ *
+ * The matcher intentionally handles both Ollama-style ids (`llama3.1:8b`,
+ * `qwen2.5-coder:14b`) and HF-style ids (`meta-llama/Llama-3.1-8B-Instruct`)
+ * without a hard dependency on either.
+ */
+import { ModelDescriptor, ModelRole } from '../types';
+export type ModelClass = ModelDescriptor['class'];
+export interface LocalModelMeta {
+    class: ModelClass;
+    roles: ModelRole[];
+    contextTokens: number;
+}
+/**
+ * Family detection — ordered by specificity so the most accurate hit wins.
+ * Every entry returns the *canonical* family tag which the metadata table
+ * then resolves into class/roles/context.
+ */
+type Family = 'llama4' | 'llama3' | 'llama2' | 'codellama' | 'deepseek-r1' | 'deepseek-v3' | 'deepseek-coder' | 'deepseek' | 'qwen3' | 'qwen25-coder' | 'qwen25' | 'qwen2' | 'qwen' | 'gemma3' | 'gemma2' | 'gemma' | 'phi4' | 'phi3' | 'phi' | 'mixtral' | 'mistral-nemo' | 'mistral-small' | 'mistral-large' | 'mistral' | 'command-r-plus' | 'command-r' | 'starcoder' | 'codegemma' | 'codestral' | 'granite-code' | 'granite' | 'yi' | 'solar' | 'zephyr' | 'minicpm' | 'llava' | 'openchat' | 'wizardcoder' | 'aya' | 'nemotron' | 'smollm' | 'orca' | 'tinyllama' | 'gpt4' | 'gpt35' | 'o-series' | 'claude-opus' | 'claude-sonnet' | 'claude-haiku' | 'unknown';
+/**
+ * Classify any local/hosted model id.
+ *
+ * The function is deterministic and side-effect free so it's safe to call
+ * from anywhere (registry, router, doctor, UI).
+ */
+export declare const classifyModel: (id: string) => LocalModelMeta;
+/**
+ * Given a list of models available on a provider, pick the best match
+ * for a role. Preference order:
+ *   1. models whose classifier roles explicitly include the desired role
+ *   2. specialized (for 'executor') or heavy (for architect/reviewer)
+ *   3. largest parameter count as a tiebreaker
+ *   4. any model at all as a last resort
+ */
+export declare const pickModelForRole: (installed: Array<{
+    id: string;
+    meta?: LocalModelMeta;
+}>, role: ModelRole) => string | null;
+export declare const _sizeBForTest: (id: string) => number | null;
+export declare const _detectFamilyForTest: (rawId: string) => Family;
+export {};
+//# sourceMappingURL=local-catalog.d.ts.map

package/dist/models/local-catalog.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"local-catalog.d.ts","sourceRoot":"","sources":["../../src/models/local-catalog.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAEtD,MAAM,MAAM,UAAU,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;AAElD,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,UAAU,CAAC;IAClB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB;AAoBD;;;;GAIG;AACH,KAAK,MAAM,GACP,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,WAAW,GACX,aAAa,GACb,aAAa,GACb,gBAAgB,GAChB,UAAU,GACV,OAAO,GACP,cAAc,GACd,QAAQ,GACR,OAAO,GACP,MAAM,GACN,QAAQ,GACR,QAAQ,GACR,OAAO,GACP,MAAM,GACN,MAAM,GACN,KAAK,GACL,SAAS,GACT,cAAc,GACd,eAAe,GACf,eAAe,GACf,SAAS,GACT,gBAAgB,GAChB,WAAW,GACX,WAAW,GACX,WAAW,GACX,WAAW,GACX,cAAc,GACd,SAAS,GACT,IAAI,GACJ,OAAO,GACP,QAAQ,GACR,SAAS,GACT,OAAO,GACP,UAAU,GACV,aAAa,GACb,KAAK,GACL,UAAU,GACV,QAAQ,GACR,MAAM,GACN,WAAW,GACX,MAAM,GACN,OAAO,GACP,UAAU,GACV,aAAa,GACb,eAAe,GACf,cAAc,GACd,SAAS,CAAC;AAiNd;;;;;GAKG;AACH,eAAO,MAAM,aAAa,GAAI,IAAI,MAAM,KAAG,cAQ1C,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,gBAAgB,GAC3B,WAAW,KAAK,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,cAAc,CAAA;CAAE,CAAC,EACvD,MAAM,SAAS,KACd,MAAM,GAAG,IA6BX,CAAC;AAEF,eAAO,MAAM,aAAa,OApUP,MAAM,KAAG,MAAM,GAAG,IAoUH,CAAC;AACnC,eAAO,MAAM,oBAAoB,UA1QV,MAAM,KAAG,MA0QU,CAAC"}