anantham-v2 2.0.2 → 2.0.4

package/dist/persistence/model-catalog-cache.js

@@ -0,0 +1,398 @@
+import * as fs from "node:fs";
+import * as path from "node:path";
+import * as os from "node:os";
+import { UserConfigManager } from "./user-config-manager.js";
+/**
+ * Normalizes an OpenRouter model ID into a standardized provider string.
+ */
+export function normalizeProvider(rawId) {
+    const cleanId = rawId.startsWith("openrouter/") ? rawId.slice("openrouter/".length) : rawId;
+    const parts = cleanId.split("/");
+    if (parts.length >= 2 && parts[0]) {
+        const p = parts[0].toLowerCase();
+        if (p === "openai")
+            return "openai";
+        if (p === "anthropic")
+            return "anthropic";
+        if (p === "google")
+            return "google";
+        if (p === "deepseek")
+            return "deepseek";
+        if (p === "meta-llama" || p === "meta" || p === "llama")
+            return "meta-llama";
+        if (p === "virtuals")
+            return "virtuals";
+        return p;
+    }
+    return "other";
+}
+/**
+ * Parses per-token pricing to price per 1 Million tokens.
+ */
+export function parsePricePerM(rawPrice) {
+    if (rawPrice === undefined || rawPrice === null)
+        return 0;
+    const num = typeof rawPrice === "number" ? rawPrice : parseFloat(String(rawPrice));
+    if (isNaN(num) || num <= 0)
+        return 0;
+    const perM = num * 1_000_000;
+    return Number(perM.toFixed(4));
+}
+/**
+ * Persistent cache manager for OpenRouter live catalog models.
+ * Saves to ~/.anantham/models_cache.json with 1-hour TTL and atomic writes.
+ */
+export class ModelCatalogCache {
+    static instance;
+    storageDir;
+    storagePath;
+    ttlMs;
+    memoryCache = null;
+    static DEFAULT_TTL_MS = 3_600_000; // 1 hour
+    /**
+     * Curated high-performance fallback models across the 6 major providers.
+     */
+    static CURATED_MODELS = [
+        // Anthropic
+        {
+            id: "anthropic/claude-3.5-sonnet",
+            name: "Claude 3.5 Sonnet",
+            provider: "anthropic",
+            contextLength: 200_000,
+            promptPricePerM: 3.0,
+            completionPricePerM: 15.0,
+            description: "Anthropic's flagship intelligent and balanced reasoning model",
+        },
+        {
+            id: "anthropic/claude-3.7-sonnet",
+            name: "Claude 3.7 Sonnet",
+            provider: "anthropic",
+            contextLength: 200_000,
+            promptPricePerM: 3.0,
+            completionPricePerM: 15.0,
+            description: "Anthropic's premier hybrid reasoning and coding model",
+        },
+        {
+            id: "anthropic/claude-3.5-haiku",
+            name: "Claude 3.5 Haiku",
+            provider: "anthropic",
+            contextLength: 200_000,
+            promptPricePerM: 0.8,
+            completionPricePerM: 4.0,
+            description: "Ultra-fast lightweight model with near-Sonnet performance",
+        },
+        {
+            id: "anthropic/claude-3-opus",
+            name: "Claude 3 Opus",
+            provider: "anthropic",
+            contextLength: 200_000,
+            promptPricePerM: 15.0,
+            completionPricePerM: 75.0,
+            description: "Anthropic's highest-power model for complex creative and scientific work",
+        },
+        // OpenAI
+        {
+            id: "openai/gpt-4o",
+            name: "GPT-4o",
+            provider: "openai",
+            contextLength: 128_000,
+            promptPricePerM: 2.5,
+            completionPricePerM: 10.0,
+            description: "OpenAI versatile high-intelligence multimodal flagship",
+        },
+        {
+            id: "openai/gpt-4o-mini",
+            name: "GPT-4o Mini",
+            provider: "openai",
+            contextLength: 128_000,
+            promptPricePerM: 0.15,
+            completionPricePerM: 0.6,
+            description: "Affordable, fast, lightweight model for high-frequency tasks",
+        },
+        {
+            id: "openai/o1",
+            name: "o1 Reasoning",
+            provider: "openai",
+            contextLength: 200_000,
+            promptPricePerM: 15.0,
+            completionPricePerM: 60.0,
+            description: "Deep chain-of-thought model designed for math, coding, and science",
+        },
+        {
+            id: "openai/o3-mini",
+            name: "o3 Mini",
+            provider: "openai",
+            contextLength: 200_000,
+            promptPricePerM: 1.1,
+            completionPricePerM: 4.4,
+            description: "Fast reasoning model optimized for STEM and coding workflows",
+        },
+        // Google
+        {
+            id: "google/gemini-2.5-pro",
+            name: "Gemini 2.5 Pro",
+            provider: "google",
+            contextLength: 1_000_000,
+            promptPricePerM: 1.25,
+            completionPricePerM: 5.0,
+            description: "Google's ultra-long context reasoning and multimodal powerhouse",
+        },
+        {
+            id: "google/gemini-2.0-flash",
+            name: "Gemini 2.0 Flash",
+            provider: "google",
+            contextLength: 1_000_000,
+            promptPricePerM: 0.1,
+            completionPricePerM: 0.4,
+            description: "Low-latency multimodal model with 1M context window",
+        },
+        {
+            id: "google/gemini-1.5-pro",
+            name: "Gemini 1.5 Pro",
+            provider: "google",
+            contextLength: 2_000_000,
+            promptPricePerM: 1.25,
+            completionPricePerM: 5.0,
+            description: "2M token context window for massive codebase ingestion",
+        },
+        // DeepSeek
+        {
+            id: "deepseek/deepseek-r1",
+            name: "DeepSeek R1",
+            provider: "deepseek",
+            contextLength: 64_000,
+            promptPricePerM: 0.55,
+            completionPricePerM: 2.19,
+            description: "Open-weights reasoning model with exceptional STEM performance",
+        },
+        {
+            id: "deepseek/deepseek-chat",
+            name: "DeepSeek V3",
+            provider: "deepseek",
+            contextLength: 64_000,
+            promptPricePerM: 0.14,
+            completionPricePerM: 0.28,
+            description: "High-efficiency MoE language model with top benchmark scores",
+        },
+        // Meta / Llama
+        {
+            id: "meta-llama/llama-3.3-70b-instruct",
+            name: "Llama 3.3 70B Instruct",
+            provider: "meta-llama",
+            contextLength: 128_000,
+            promptPricePerM: 0.12,
+            completionPricePerM: 0.3,
+            description: "Industry-standard open model with state-of-the-art 70B weights",
+        },
+        {
+            id: "meta-llama/llama-3.1-405b-instruct",
+            name: "Llama 3.1 405B Instruct",
+            provider: "meta-llama",
+            contextLength: 128_000,
+            promptPricePerM: 0.8,
+            completionPricePerM: 0.8,
+            description: "Frontier open-weights 405B parameter instruction model",
+        },
+        {
+            id: "meta-llama/llama-3.1-8b-instruct",
+            name: "Llama 3.1 8B Instruct",
+            provider: "meta-llama",
+            contextLength: 128_000,
+            promptPricePerM: 0.05,
+            completionPricePerM: 0.05,
+            description: "Compact, extremely fast open model for low-latency agent loops",
+        },
+        // Virtuals
+        {
+            id: "virtuals/game-agent",
+            name: "Virtuals GAME Agent",
+            provider: "virtuals",
+            contextLength: 32_000,
+            promptPricePerM: 0.5,
+            completionPricePerM: 1.5,
+            description: "On-chain autonomous agent execution model",
+        },
+        {
+            id: "virtuals/sentient-ai",
+            name: "Virtuals Sentient AI",
+            provider: "virtuals",
+            contextLength: 32_000,
+            promptPricePerM: 1.0,
+            completionPricePerM: 3.0,
+            description: "Interactive autonomous agent persona and simulation model",
+        },
+    ];
+    constructor(customStorageDir, ttlMs = ModelCatalogCache.DEFAULT_TTL_MS) {
+        this.storageDir = customStorageDir || path.join(os.homedir(), ".anantham");
+        this.storagePath = path.join(this.storageDir, "models_cache.json");
+        this.ttlMs = ttlMs;
+        this.loadFromDisk();
+    }
+    static getInstance(customStorageDir, ttlMs) {
+        if (!ModelCatalogCache.instance || customStorageDir !== undefined) {
+            ModelCatalogCache.instance = new ModelCatalogCache(customStorageDir, ttlMs);
+        }
+        return ModelCatalogCache.instance;
+    }
+    static resetInstance() {
+        ModelCatalogCache.instance = undefined;
+    }
+    /**
+     * Reads cached models from disk if present and valid.
+     */
+    loadFromDisk() {
+        try {
+            if (fs.existsSync(this.storagePath)) {
+                const raw = fs.readFileSync(this.storagePath, "utf-8");
+                const parsed = JSON.parse(raw);
+                if (parsed &&
+                    typeof parsed.fetchedAt === "number" &&
+                    Array.isArray(parsed.models) &&
+                    parsed.models.length > 0) {
+                    this.memoryCache = parsed;
+                    return parsed;
+                }
+            }
+        }
+        catch {
+            // Fallback on read or JSON parse failure
+        }
+        return null;
+    }
+    /**
+     * Synchronously returns cached models from memory or disk if available.
+     */
+    getCachedModels() {
+        if (this.memoryCache && Array.isArray(this.memoryCache.models)) {
+            return this.memoryCache.models;
+        }
+        const loaded = this.loadFromDisk();
+        return loaded ? loaded.models : null;
+    }
+    /**
+     * Checks whether the current cache is within its valid TTL window.
+     */
+    isCacheFresh() {
+        if (!this.memoryCache) {
+            this.loadFromDisk();
+        }
+        if (!this.memoryCache) {
+            return false;
+        }
+        const age = Date.now() - this.memoryCache.fetchedAt;
+        return age >= 0 && age < (this.memoryCache.ttlMs || this.ttlMs);
+    }
+    /**
+     * Saves models atomically to memory and disk using a temporary file and atomic rename.
+     */
+    saveModels(models) {
+        const data = {
+            fetchedAt: Date.now(),
+            ttlMs: this.ttlMs,
+            models,
+        };
+        this.memoryCache = data;
+        try {
+            if (!fs.existsSync(this.storageDir)) {
+                fs.mkdirSync(this.storageDir, { recursive: true });
+            }
+            const tmpPath = path.join(this.storageDir, `models_cache.json.tmp.${Date.now()}.${Math.random().toString(36).slice(2, 8)}`);
+            fs.writeFileSync(tmpPath, JSON.stringify(data, null, 2), "utf-8");
+            try {
+                fs.renameSync(tmpPath, this.storagePath);
+            }
+            catch {
+                // Windows fallback: if renameSync encounters locking, copy and unlink
+                fs.copyFileSync(tmpPath, this.storagePath);
+                try {
+                    fs.unlinkSync(tmpPath);
+                }
+                catch {
+                    // Ignore temp cleanup errors
+                }
+            }
+        }
+        catch {
+            // Non-critical cache write failure
+        }
+    }
+    /**
+     * Clears in-memory and on-disk cache.
+     */
+    clearCache() {
+        this.memoryCache = null;
+        try {
+            if (fs.existsSync(this.storagePath)) {
+                fs.unlinkSync(this.storagePath);
+            }
+        }
+        catch {
+            // Ignore cleanup error
+        }
+    }
+    /**
+     * Fetches the model catalog:
+     * 1. Returns fresh cache if not forceRefresh.
+     * 2. Tries live OpenRouter API fetch with configured key.
+     * 3. Falls back to disk cache if available.
+     * 4. Falls back to built-in curated models across all 6 providers.
+     */
+    async getModels(forceRefresh = false) {
+        if (!forceRefresh && this.isCacheFresh()) {
+            const cached = this.getCachedModels();
+            if (cached && cached.length > 0) {
+                return cached;
+            }
+        }
+        const apiKey = UserConfigManager.getInstance().getApiKey("openrouter") || process.env.OPENROUTER_API_KEY;
+        try {
+            const headers = {
+                "HTTP-Referer": "https://anantham.ai",
+                "X-Title": "Anantham V2",
+            };
+            if (apiKey) {
+                headers["Authorization"] = `Bearer ${apiKey}`;
+            }
+            const response = await fetch("https://openrouter.ai/api/v1/models", {
+                method: "GET",
+                headers,
+                signal: AbortSignal.timeout(8000),
+            });
+            if (!response.ok) {
+                throw new Error(`OpenRouter models API returned HTTP ${response.status}: ${response.statusText}`);
+            }
+            const json = (await response.json());
+            if (json && Array.isArray(json.data) && json.data.length > 0) {
+                const normalized = json.data.map((m) => {
+                    const provider = normalizeProvider(m.id);
+                    const promptPricePerM = parsePricePerM(m.pricing?.prompt);
+                    const completionPricePerM = parsePricePerM(m.pricing?.completion);
+                    const contextLength = typeof m.context_length === "number" ? m.context_length : 4096;
+                    return {
+                        id: m.id,
+                        name: m.name && m.name.trim().length > 0 ? m.name : m.id,
+                        provider,
+                        contextLength,
+                        promptPricePerM,
+                        completionPricePerM,
+                        description: m.description,
+                    };
+                });
+                this.saveModels(normalized);
+                return normalized;
+            }
+        }
+        catch {
+            // Network fetch failure or timeout: proceed to fallbacks
+        }
+        // Fallback 1: Disk or memory cache (even if expired)
+        const existing = this.getCachedModels();
+        if (existing && existing.length > 0) {
+            return existing;
+        }
+        // Fallback 2: Comprehensive curated models across all 6 providers
+        this.saveModels(ModelCatalogCache.CURATED_MODELS);
+        return ModelCatalogCache.CURATED_MODELS;
+    }
+}
+//# sourceMappingURL=model-catalog-cache.js.map

package/dist/persistence/model-catalog-cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-catalog-cache.js","sourceRoot":"","sources":["../../src/persistence/model-catalog-cache.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAqC7D;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAa;IAC7C,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAC5F,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QAClC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACjC,IAAI,CAAC,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC;QACpC,IAAI,CAAC,KAAK,WAAW;YAAE,OAAO,WAAW,CAAC;QAC1C,IAAI,CAAC,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC;QACpC,IAAI,CAAC,KAAK,UAAU;YAAE,OAAO,UAAU,CAAC;QACxC,IAAI,CAAC,KAAK,YAAY,IAAI,CAAC,KAAK,MAAM,IAAI,CAAC,KAAK,OAAO;YAAE,OAAO,YAAY,CAAC;QAC7E,IAAI,CAAC,KAAK,UAAU;YAAE,OAAO,UAAU,CAAC;QACxC,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,QAA0B;IACvD,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI;QAAE,OAAO,CAAC,CAAC;IAC1D,MAAM,GAAG,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;IACnF,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;QAAE,OAAO,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,GAAG,GAAG,SAAS,CAAC;IAC7B,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;AACjC,CAAC;AAED;;;GAGG;AACH,MAAM,OAAO,iBAAiB;IACpB,MAAM,CAAC,QAAQ,CAAgC;IACtC,UAAU,CAAS;IACnB,WAAW,CAAS;IACpB,KAAK,CAAS;IACvB,WAAW,GAA0B,IAAI,CAAC;IAE3C,MAAM,CAAU,cAAc,GAAG,SAAS,CAAC,CAAC,SAAS;IAE5D;;OAEG;IACI,MAAM,CAAU,cAAc,GAAkB;QACrD,YAAY;QACZ;YACE,EAAE,EAAE,6BAA6B;YACjC,IAAI,EAAE,mBAAmB;YACzB,QAAQ,EAAE,WAAW;YACrB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,+DAA+D;SAC7E;QACD;YACE,EAAE,EAAE,6BAA6B;YACjC,IAAI,EAAE,mBAAmB;YACzB,QAAQ,EAAE,WAAW;YACrB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,uDAAuD;SACrE;QACD;YACE,EAAE,EAAE,4BAA4B;YAChC,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE,WAAW;YACrB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,2DAA2D;SACzE;QACD;YACE,EAAE,EAAE,yBAAyB;YAC7B,IAAI,EAAE,eAAe;YACrB,QAAQ,EAAE,WAAW;YACrB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,0EAA0E;SACxF;QACD,SAAS;QACT;YACE,EAAE,EAAE,eAAe;YACnB,IAAI,EAAE,QAAQ;YACd,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,wDAAwD;SACtE;QACD;YACE,EAAE,EAAE,oBAAoB;YACxB,IAAI,EAAE,aAAa;YACnB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,8DAA8D;SAC5E;QACD;YACE,EAAE,EAAE,WAAW;YACf,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,oEAAoE;SAClF;QACD;YACE,EAAE,EAAE,gBAAgB;YACpB,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,8DAA8D;SAC5E;QACD,SAAS;QACT;YACE,EAAE,EAAE,uBAAuB;YAC3B,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,SAAS;YACxB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,iEAAiE;SAC/E;QACD;YACE,EAAE,EAAE,yBAAyB;YAC7B,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,SAAS;YACxB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,qDAAqD;SACnE;QACD;YACE,EAAE,EAAE,uBAAuB;YAC3B,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,SAAS;YACxB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,wDAAwD;SACtE;QACD,WAAW;QACX;YACE,EAAE,EAAE,sBAAsB;YAC1B,IAAI,EAAE,aAAa;YACnB,QAAQ,EAAE,UAAU;YACpB,aAAa,EAAE,MAAM;YACrB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,gEAAgE;SAC9E;QACD;YACE,EAAE,EAAE,wBAAwB;YAC5B,IAAI,EAAE,aAAa;YACnB,QAAQ,EAAE,UAAU;YACpB,aAAa,EAAE,MAAM;YACrB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,8DAA8D;SAC5E;QACD,eAAe;QACf;YACE,EAAE,EAAE,mCAAmC;YACvC,IAAI,EAAE,wBAAwB;YAC9B,QAAQ,EAAE,YAAY;YACtB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,gEAAgE;SAC9E;QACD;YACE,EAAE,EAAE,oCAAoC;YACxC,IAAI,EAAE,yBAAyB;YAC/B,QAAQ,EAAE,YAAY;YACtB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,wDAAwD;SACtE;QACD;YACE,EAAE,EAAE,kCAAkC;YACtC,IAAI,EAAE,uBAAuB;YAC7B,QAAQ,EAAE,YAAY;YACtB,aAAa,EAAE,OAAO;YACtB,eAAe,EAAE,IAAI;YACrB,mBAAmB,EAAE,IAAI;YACzB,WAAW,EAAE,gEAAgE;SAC9E;QACD,WAAW;QACX;YACE,EAAE,EAAE,qBAAqB;YACzB,IAAI,EAAE,qBAAqB;YAC3B,QAAQ,EAAE,UAAU;YACpB,aAAa,EAAE,MAAM;YACrB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,2CAA2C;SACzD;QACD;YACE,EAAE,EAAE,sBAAsB;YAC1B,IAAI,EAAE,sBAAsB;YAC5B,QAAQ,EAAE,UAAU;YACpB,aAAa,EAAE,MAAM;YACrB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,GAAG;YACxB,WAAW,EAAE,2DAA2D;SACzE;KACF,CAAC;IAEF,YAAmB,gBAAyB,EAAE,QAAgB,iBAAiB,CAAC,cAAc;QAC5F,IAAI,CAAC,UAAU,GAAG,gBAAgB,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;QAC3E,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;QACnE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAEM,MAAM,CAAC,WAAW,CAAC,gBAAyB,EAAE,KAAc;QACjE,IAAI,CAAC,iBAAiB,CAAC,QAAQ,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YAClE,iBAAiB,CAAC,QAAQ,GAAG,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;QAC9E,CAAC;QACD,OAAO,iBAAiB,CAAC,QAAQ,CAAC;IACpC,CAAC;IAEM,MAAM,CAAC,aAAa;QACzB,iBAAiB,CAAC,QAAQ,GAAG,SAAS,CAAC;IACzC,CAAC;IAED;;OAEG;IACK,YAAY;QAClB,IAAI,CAAC;YACH,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;gBACpC,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBACvD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAmB,CAAC;gBACjD,IACE,MAAM;oBACN,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ;oBACpC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;oBAC5B,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EACxB,CAAC;oBACD,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;oBAC1B,OAAO,MAAM,CAAC;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,yCAAyC;QAC3C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACI,eAAe;QACpB,IAAI,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/D,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACjC,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACnC,OAAO,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;IACvC,CAAC;IAED;;OAEG;IACI,YAAY;QACjB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;QACpD,OAAO,GAAG,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC;IAClE,CAAC;IAED;;OAEG;IACI,UAAU,CAAC,MAAqB;QACrC,MAAM,IAAI,GAAmB;YAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM;SACP,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAExB,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CACvB,IAAI,CAAC,UAAU,EACf,yBAAyB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAChF,CAAC;YACF,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YAElE,IAAI,CAAC;gBACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YAC3C,CAAC;YAAC,MAAM,CAAC;gBACP,sEAAsE;gBACtE,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;gBAC3C,IAAI,CAAC;oBACH,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;gBACzB,CAAC;gBAAC,MAAM,CAAC;oBACP,6BAA6B;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,mCAAmC;QACrC,CAAC;IACH,CAAC;IAED;;OAEG;IACI,UAAU;QACf,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC;YACH,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;gBACpC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uBAAuB;QACzB,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACI,KAAK,CAAC,SAAS,CAAC,YAAY,GAAG,KAAK;QACzC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;YACtC,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GACV,iBAAiB,CAAC,WAAW,EAAE,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QAE5F,IAAI,CAAC;YACH,MAAM,OAAO,GAA2B;gBACtC,cAAc,EAAE,qBAAqB;gBACrC,SAAS,EAAE,aAAa;aACzB,CAAC;YACF,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,EAAE,CAAC;YAChD,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,qCAAqC,EAAE;gBAClE,MAAM,EAAE,KAAK;gBACb,OAAO;gBACP,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;aAClC,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,uCAAuC,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACpG,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA0B,CAAC;YAC9D,IAAI,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7D,MAAM,UAAU,GAAkB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;oBACpD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACzC,MAAM,eAAe,GAAG,cAAc,CAAC,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;oBAC1D,MAAM,mBAAmB,GAAG,cAAc,CAAC,CAAC,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;oBAClE,MAAM,aAAa,GAAG,OAAO,CAAC,CAAC,cAAc,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC;oBAErF,OAAO;wBACL,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;wBACxD,QAAQ;wBACR,aAAa;wBACb,eAAe;wBACf,mBAAmB;wBACnB,WAAW,EAAE,CAAC,CAAC,WAAW;qBAC3B,CAAC;gBACJ,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;gBAC5B,OAAO,UAAU,CAAC;YACpB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,yDAAyD;QAC3D,CAAC;QAED,qDAAqD;QACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACxC,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,kEAAkE;QAClE,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QAClD,OAAO,iBAAiB,CAAC,cAAc,CAAC;IAC1C,CAAC"}

package/dist/persistence/token-metrics-manager.d.ts

@@ -0,0 +1,88 @@
+export interface TokenRecord {
+    id: string;
+    timestamp: number;
+    date: string;
+    modelId: string;
+    provider: string;
+    inputTokens: number;
+    outputTokens: number;
+    cachedTokens: number;
+    estimatedCostUsd: number;
+    command?: string;
+}
+export interface ModelUsageStats {
+    inputTokens: number;
+    outputTokens: number;
+    cachedTokens: number;
+    totalTokens: number;
+    costUsd: number;
+    count: number;
+}
+export interface DailyTokenSummary {
+    date: string;
+    totalInputTokens: number;
+    totalOutputTokens: number;
+    totalCachedTokens: number;
+    totalTokens: number;
+    totalCostUsd: number;
+    requestCount: number;
+    models: Record<string, ModelUsageStats>;
+}
+export interface ModelPricing {
+    inputPerM: number;
+    outputPerM: number;
+    cachedPerM: number;
+}
+export interface TokenMetricsData {
+    records: TokenRecord[];
+    monthlyBudgetUsd?: number;
+}
+export declare class TokenMetricsManager {
+    private static instance;
+    private readonly storageDir;
+    private readonly storagePath;
+    private data;
+    static readonly PRICING: Record<string, ModelPricing>;
+    static readonly DEFAULT_PRICING: ModelPricing;
+    constructor(customStorageDir?: string);
+    static getInstance(customStorageDir?: string): TokenMetricsManager;
+    static resetInstance(): void;
+    private readFromDisk;
+    save(): void;
+    static calculateCost(modelId: string, inputTokens: number, outputTokens: number, cachedTokens?: number): number;
+    recordUsage(params: {
+        modelId: string;
+        provider?: string;
+        inputTokens: number;
+        outputTokens: number;
+        cachedTokens?: number;
+        command?: string;
+        estimatedCostUsd?: number;
+        timestamp?: number;
+    }): TokenRecord;
+    getRecords(): TokenRecord[];
+    getMonthlyBudget(): number;
+    setMonthlyBudget(usd: number): void;
+    getDailySummary(targetDate?: string): DailyTokenSummary;
+    getMtdSummary(): {
+        totalInputTokens: number;
+        totalOutputTokens: number;
+        totalCachedTokens: number;
+        totalTokens: number;
+        totalCostUsd: number;
+        requestCount: number;
+    };
+    getSevenDayTrend(): {
+        date: string;
+        tokens: number;
+        costUsd: number;
+    }[];
+    getTopModels(limit?: number): {
+        modelId: string;
+        totalTokens: number;
+        costUsd: number;
+        percentage: number;
+    }[];
+    private seedRealisticMetrics;
+}
+//# sourceMappingURL=token-metrics-manager.d.ts.map

package/dist/persistence/token-metrics-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-metrics-manager.d.ts","sourceRoot":"","sources":["../../src/persistence/token-metrics-manager.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC9B,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,gBAAgB,EAAE,MAAM,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;CACzC;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAsB;IAC7C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,IAAI,CAAmB;IAE/B,gBAAuB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAU1D;IAEF,gBAAuB,eAAe,EAAE,YAAY,CAIlD;gBAEU,gBAAgB,CAAC,EAAE,MAAM;WASvB,WAAW,CAAC,gBAAgB,CAAC,EAAE,MAAM,GAAG,mBAAmB;WAO3D,aAAa,IAAI,IAAI;IAInC,OAAO,CAAC,YAAY;IAkBb,IAAI,IAAI,IAAI;WAaL,aAAa,CACzB,OAAO,EAAE,MAAM,EACf,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,YAAY,GAAE,MAAU,GACvB,MAAM;IAQF,WAAW,CAAC,MAAM,EAAE;QACzB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,GAAG,WAAW;IA0BR,UAAU,IAAI,WAAW,EAAE;IAI3B,gBAAgB,IAAI,MAAM;IAI1B,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAKnC,eAAe,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,iBAAiB;IAgDvD,aAAa,IAAI;QACtB,gBAAgB,EAAE,MAAM,CAAC;QACzB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,iBAAiB,EAAE,MAAM,CAAC;QAC1B,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;KACtB;IA0BM,gBAAgB,IAAI;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,EAAE;IA0BvE,YAAY,CAAC,KAAK,GAAE,MAAU,GAAG;QACtC,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,MAAM,CAAC;QACpB,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,EAAE,MAAM,CAAC;KACpB,EAAE;IA6BH,OAAO,CAAC,oBAAoB;CAoD7B"}