opencode-token-tracker 1.6.4 → 1.7.0

package/dist/bin/opencode-tokens.js

@@ -2,7 +2,7 @@
 import { closeSync, copyFileSync, existsSync, mkdirSync, openSync, readFileSync, readSync, statSync, writeFileSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
-import { BUILTIN_PRICING, DEFAULT_CONFIG, findModelConfigPricing, formatCost, formatTokens, getStartOfDay, getStartOfMonth, getStartOfWeek, validateConfig } from "../lib/shared.js";
+import { BUILTIN_PRICING, DEFAULT_CONFIG, formatCost, formatTokens, getStartOfDay, getStartOfMonth, getStartOfWeek, validateConfig, BUILTIN_PRICING_META, resolvePricingStatus, round2 } from "../lib/shared.js";
 const CONFIG_DIR = join(homedir(), ".config", "opencode");
 const CONFIG_FILE = join(CONFIG_DIR, "token-tracker.json");
 const LOG_FILE = join(CONFIG_DIR, "logs", "token-tracker", "tokens.jsonl");
@@ -314,8 +314,11 @@ function cmdStats(period, breakdown) {
 function cmdPricing() {
     const config = loadConfig();
     console.log(`
-  Built-in Pricing Table (USD per 1M tokens) - Updated 2026-02-11
+  Built-in Pricing Table (USD per 1M tokens)
   ══════════════════════════════════════════════════════════════════
+  Pricing last updated:  ${BUILTIN_PRICING_META.pricingLastUpdated}
+  Metadata last updated: ${BUILTIN_PRICING_META.metadataLastUpdated}
+  Source:                ${BUILTIN_PRICING_META.source}
 `);
     // Group by provider
     const groups = {
@@ -347,6 +350,12 @@ function cmdPricing() {
     if (Object.keys(config.models || {}).length > 0) {
         console.log(`  * = overridden in config`);
     }
+    console.log(`  Fallback Pricing Notice:`);
+    console.log(`    When a model is not matched in the built-in pricing table or user configuration,`);
+    console.log(`    it falls back to the default rate ($1.0 / $4.0 per 1M tokens).`);
+    console.log(`    You can easily override it in your configuration.`);
+    console.log(`    ${BUILTIN_PRICING_META.notes}`);
+    console.log();
 }
 function cmdModels() {
     const entries = loadEntries();
@@ -384,30 +393,7 @@ function cmdModels() {
     console.log(`  ${padRight("Model", modelWidth)}  ${padRight("Provider", providerWidth)}  ${padLeft("Msgs", countWidth)}  ${padRight("Pricing", statusWidth)}`);
     console.log(`  ${"-".repeat(modelWidth)}  ${"-".repeat(providerWidth)}  ${"-".repeat(countWidth)}  ${"-".repeat(statusWidth)}`);
     for (const { model, provider, count } of sorted) {
-        let status;
-        // Mirror the runtime pricing resolution order (getModelPricing in index.ts)
-        if (config.providers?.[provider]) {
-            status = "provider cfg";
-        }
-        else if (findModelConfigPricing(config.models, model, provider, false)) {
-            status = "model cfg";
-        }
-        else if (BUILTIN_PRICING[model]) {
-            status = "built-in";
-        }
-        else {
-            const modelLower = model.toLowerCase();
-            const hasBuiltinPartial = Object.keys(BUILTIN_PRICING).some(k => k !== "_default" && modelLower.includes(k.toLowerCase()));
-            if (hasBuiltinPartial) {
-                status = "built-in";
-            }
-            else if (findModelConfigPricing(config.models, model, provider, true)) {
-                status = "model cfg";
-            }
-            else {
-                status = "default";
-            }
-        }
+        const status = resolvePricingStatus(config, model, provider);
         console.log(`  ${padRight(model, modelWidth)}  ${padRight(provider, providerWidth)}  ${padLeft(count.toString(), countWidth)}  ${padRight(status, statusWidth)}`);
     }
     console.log();
@@ -423,22 +409,25 @@ function cmdConfig(positional) {
     const config = loadConfig();
     const entries = loadEntries();
     if (action === "init" || action === "generate") {
-        // Get unique providers from logs
+        // Get unique providers and model+provider combinations from logs
         const providers = new Set();
-        const models = new Set();
+        const modelProviders = new Set();
         for (const e of entries) {
             if (e.provider)
                 providers.add(e.provider);
-            if (e.model)
-                models.add(e.model);
+            const model = e.model ?? "unknown";
+            const provider = e.provider ?? "unknown";
+            modelProviders.add(`${model}|${provider}`);
         }
-        // Find providers/models without built-in pricing
-        const unknownModels = Array.from(models).filter(m => {
-            if (BUILTIN_PRICING[m])
-                return false;
-            const hasMatch = Object.keys(BUILTIN_PRICING).some(k => k !== "_default" && m.toLowerCase().includes(k.toLowerCase()));
-            return !hasMatch;
-        });
+        // Estimate daily avg from last 7 days
+        const now = Date.now();
+        const sevenDaysAgo = now - 7 * 24 * 60 * 60 * 1000;
+        const recentEntries = entries.filter(e => e._ts >= sevenDaysAgo);
+        const totalSpent = recentEntries.reduce((sum, e) => sum + (e.cost ?? 0), 0);
+        const dailyAvg = totalSpent / 7;
+        const dailyLimit = dailyAvg > 0 ? Math.max(0.5, round2(dailyAvg * 1.5)) : 5;
+        const weeklyLimit = dailyAvg > 0 ? Math.max(0.5, round2(dailyAvg * 7 * 1.3)) : 25;
+        const monthlyLimit = dailyAvg > 0 ? Math.max(0.5, round2(dailyAvg * 30 * 1.2)) : 100;
         const exampleConfig = {
             providers: {},
             models: {},
@@ -448,25 +437,74 @@ function cmdConfig(positional) {
                 showOnIdle: true,
             },
             budget: {
-                daily: 5,
-                weekly: 25,
-                monthly: 100,
+                daily: dailyLimit,
+                weekly: weeklyLimit,
+                monthly: monthlyLimit,
                 warnAt: 0.8,
             },
         };
-        // Add providers as comments/examples
+        const suggestedProviders = [];
+        // Add providers as comments/examples (Common free providers)
         for (const provider of providers) {
-            // Common free providers
-            if (provider.includes("copilot") || provider.includes("cursor") || provider.includes("free")) {
+            const pLower = provider.toLowerCase();
+            if (pLower.includes("copilot") || pLower.includes("cursor") || pLower.includes("free")) {
                 exampleConfig.providers[provider] = { input: 0, output: 0 };
+                suggestedProviders.push(provider);
+            }
+        }
+        const suggestedModels = [];
+        // Add unknown/fallback models where status is "default"
+        for (const mp of modelProviders) {
+            const [model, provider] = mp.split("|");
+            if (model === "unknown" || provider === "unknown")
+                continue;
+            const status = resolvePricingStatus(config, model, provider);
+            if (status === "default") {
+                exampleConfig.models[model] = { input: 1, output: 4 };
+                suggestedModels.push(`${model} (${provider})`);
             }
         }
-        // Add unknown models
-        for (const model of unknownModels) {
-            exampleConfig.models[model] = { input: 1, output: 4 };
+        // Construct Dynamic Usage-Aware Suggestions Summary
+        let suggestionsSummary = `
+  📢 Usage-Aware Suggestions Summary
+  ══════════════════════════════════════════════════════════════════
+  - Analyzed timeframe: Last 7 days
+  - Found ${entries.length} historical usage entries in total.
+  - Calculated 7-day average daily cost: $${round2(dailyAvg).toFixed(4)}
+`;
+        if (dailyAvg > 0) {
+            suggestionsSummary += `
+  Recommended Budget Limits (derived from daily avg $${round2(dailyAvg).toFixed(4)}):
+    - Daily Limit   : $${dailyLimit.toFixed(2)} (Avg * 1.5 buffer, clamped min $0.50)
+    - Weekly Limit  : $${weeklyLimit.toFixed(2)} (Avg * 7 * 1.3 buffer, clamped min $0.50)
+    - Monthly Limit : $${monthlyLimit.toFixed(2)} (Avg * 30 * 1.2 buffer, clamped min $0.50)
+`;
         }
-        // Print explanation first
-        console.log(`
+        else {
+            suggestionsSummary += `
+  No active usage logs detected in the last 7 days.
+  - Applying fallback default budgets: Daily $5.00, Weekly $25.00, Monthly $100.00.
+`;
+        }
+        if (suggestedProviders.length > 0) {
+            suggestionsSummary += `
+  Detected zero-cost/subscription providers:
+`;
+            for (const p of suggestedProviders) {
+                suggestionsSummary += `    • ${p} (automatically pre-configured to $0.00)\n`;
+            }
+        }
+        if (suggestedModels.length > 0) {
+            suggestionsSummary += `
+  Detected unrecognized fallback models:
+`;
+            for (const m of suggestedModels) {
+                suggestionsSummary += `    • ${m} (automatically pre-configured with fallback $1/$4 rate)\n`;
+            }
+        }
+        suggestionsSummary += `  ────────────────────────────────────────────────────────────────\n`;
+        // Print explanation first to stderr
+        const guideText = `
   Configuration Guide
   ══════════════════════════════════════════════════════════════════
 
@@ -507,22 +545,25 @@ function cmdConfig(positional) {
 
   ────────────────────────────────────────────────────────────────
   Example config based on your usage:
-`);
-        console.log(JSON.stringify(exampleConfig, null, 2));
-        if (action === "generate") {
-            const json = JSON.stringify(exampleConfig, null, 2);
-            writeFileSync(CONFIG_FILE, json);
-            console.log(`
-  Config file created: ${CONFIG_FILE}
-`);
-        }
-        else {
-            console.log(`
+`;
+        process.stderr.write(suggestionsSummary + "\n");
+        process.stderr.write(guideText + "\n");
+        if (action === "init") {
+            // Print clean config to stdout ONLY on init
+            process.stdout.write(JSON.stringify(exampleConfig, null, 2) + "\n");
+            process.stderr.write(`
   To create this config file, run:
     opencode-tokens config generate
   
   Or manually create: ${CONFIG_FILE}
-`);
+\n`);
+        }
+        else if (action === "generate") {
+            // Generate has completely empty stdout! Writes to config file, alerts to stderr
+            saveConfig(exampleConfig);
+            process.stderr.write(`
+  Config file created: ${CONFIG_FILE}
+\n`);
         }
         return;
     }
@@ -777,6 +818,8 @@ function cmdBudget() {
   }
 
   Run: opencode-tokens config init  for more details.
+
+  Costs are estimates from local logs; budgets are warnings, not enforcement.
 `);
         return;
     }
@@ -841,6 +884,7 @@ function cmdBudget() {
         console.log();
     }
     console.log(`  Legend: 🟢 OK  🟡 Warning (>${Math.round(warnAt * 100)}%)  🔴 Exceeded`);
+    console.log(`  Costs are estimates from local logs; budgets are warnings, not enforcement.`);
     console.log();
 }
 function cmdHelp() {
@@ -897,9 +941,10 @@ function cmdHelp() {
     opencode-tokens export --format csv   # Export all data as CSV
     opencode-tokens config set budget.daily 10  # Set daily budget to $10
     opencode-tokens config get toast.enabled   # Check if toast is enabled
+
+  Costs are estimates from local logs; budgets are warnings, not enforcement.
 `);
 }
-const BRAILLE_DOTS = [0x01, 0x02, 0x04, 0x40, 0x08, 0x10, 0x20, 0x80];
 function getTrendValue(point, metric) {
     return metric === "tokens" ? point.tokens : metric === "messages" ? point.messages : point.cost;
 }
@@ -913,43 +958,110 @@ function formatSignedTrendValue(value, metric) {
 function metricLabel(metric) {
     return metric === "tokens" ? "Token Trend" : metric === "messages" ? "Message Trend" : "Cost Trend";
 }
-function addBraillePoint(cells, x, y) {
-    const cellX = Math.floor(x / 2);
-    const cellY = Math.floor(y / 4);
-    const dotX = x % 2;
-    const dotY = y % 4;
-    cells[cellY][cellX] |= BRAILLE_DOTS[dotX * 4 + dotY];
-}
-function buildBrailleRows(points, width, height) {
-    const cells = Array.from({ length: height }, () => Array.from({ length: width }, () => 0));
-    for (let i = 1; i < points.length; i++) {
-        const from = points[i - 1];
-        const to = points[i];
-        const steps = Math.max(Math.abs(to.x - from.x), Math.abs(to.y - from.y), 1);
-        for (let step = 0; step <= steps; step++) {
-            const t = step / steps;
-            addBraillePoint(cells, Math.round(from.x + (to.x - from.x) * t), Math.round(from.y + (to.y - from.y) * t));
+function buildLineRows(points, width, height) {
+    const grid = Array.from({ length: height }, () => Array.from({ length: width }, () => " "));
+    // 1. Calculate the exact integer row r[x] for every column x from 0 to width - 1
+    const r = Array.from({ length: width }, () => 0);
+    for (let x = 0; x < width; x++) {
+        // Find which segment p[i] -> p[i+1] contains x
+        let i = 0;
+        for (; i < points.length - 1; i++) {
+            if (x >= points[i].x && x <= points[i + 1].x) {
+                break;
+            }
+        }
+        // Clamp to valid index
+        if (i >= points.length - 1) {
+            i = points.length - 2;
+        }
+        const p1 = points[i];
+        const p2 = points[i + 1];
+        if (p1.x === p2.x) {
+            r[x] = p1.y;
+        }
+        else {
+            const t = (x - p1.x) / (p2.x - p1.x);
+            const y = p1.y + (p2.y - p1.y) * t;
+            r[x] = Math.round(y);
+        }
+    }
+    // 2. Plot the line characters based on r[x] and r[x+1]
+    for (let x = 0; x < width; x++) {
+        const rCurr = r[x];
+        if (x === width - 1) {
+            // Last column has no next column, draw horizontal line
+            if (grid[rCurr][x] === " ") {
+                grid[rCurr][x] = "─";
+            }
+            break;
+        }
+        const rNext = r[x + 1];
+        if (rCurr === rNext) {
+            if (grid[rCurr][x] === " ") {
+                grid[rCurr][x] = "─";
+            }
+        }
+        else if (rCurr < rNext) {
+            // Going down (larger row index)
+            if (grid[rCurr][x] === " ") {
+                grid[rCurr][x] = "┐";
+            }
+            for (let y = rCurr + 1; y < rNext; y++) {
+                if (grid[y][x] === " ") {
+                    grid[y][x] = "│";
+                }
+            }
+            if (grid[rNext][x] === " ") {
+                grid[rNext][x] = "└";
+            }
+        }
+        else {
+            // Going up (smaller row index)
+            if (grid[rCurr][x] === " ") {
+                grid[rCurr][x] = "┘";
+            }
+            for (let y = rNext + 1; y < rCurr; y++) {
+                if (grid[y][x] === " ") {
+                    grid[y][x] = "│";
+                }
+            }
+            if (grid[rNext][x] === " ") {
+                grid[rNext][x] = "┌";
+            }
         }
     }
-    return cells.map(row => row.map(mask => mask === 0 ? " " : String.fromCharCode(0x2800 + mask)).join(""));
+    return grid;
 }
+const ESC = "\x1b";
+const RESET = `${ESC}[0m`;
+const BOLD = `${ESC}[1m`;
+const DIM = `${ESC}[2m`;
+const GRAY = `${ESC}[90m`;
 function buildTrendXAxis(points, chartPoints, chartWidth) {
     const labelChars = Array.from({ length: chartWidth }, () => " ");
+    const tickChars = Array.from({ length: chartWidth }, () => "─");
     const labelStep = Math.max(1, Math.ceil(points.length / 6));
     for (let i = 0; i < chartPoints.length; i++) {
         if (i % labelStep !== 0 && i !== chartPoints.length - 1)
             continue;
         const date = new Date(points[i][0]);
         const label = `${date.getMonth() + 1}/${date.getDate()}`;
-        const start = Math.min(Math.max(0, chartPoints[i].x - Math.floor(label.length / 2)), Math.max(0, chartWidth - label.length));
+        const centerOfLabel = chartPoints[i].x;
+        const start = Math.min(Math.max(0, centerOfLabel - Math.floor(label.length / 2)), Math.max(0, chartWidth - label.length));
         const hasSpace = labelChars.slice(Math.max(0, start - 1), Math.min(chartWidth, start + label.length + 1)).every((c) => c === " ");
         if (!hasSpace)
             continue;
         for (let j = 0; j < label.length; j++) {
             labelChars[start + j] = label[j];
         }
+        if (centerOfLabel < chartWidth) {
+            tickChars[centerOfLabel] = "┬";
+        }
     }
-    return labelChars.join("");
+    return {
+        axisTicks: tickChars.join(""),
+        axisLabels: labelChars.join("")
+    };
 }
 function cmdTrend(flags) {
     const days = parseInt(String(flagValue(flags, "days") ?? "30"), 10);
@@ -985,6 +1097,12 @@ function cmdTrend(flags) {
     const values = sorted.map(([, d]) => getTrendValue(d, metric));
     const maxVal = Math.max(...values, 1);
     const minVal = Math.min(...values);
+    const valRange = maxVal - minVal;
+    // Adaptive scaling padding (e.g. 5% of range) to show beautiful ups & downs
+    const pad = valRange === 0 ? maxVal * 0.1 : valRange * 0.05;
+    const chartMax = maxVal + pad;
+    const chartMin = Math.max(0, minVal - pad);
+    const chartRange = chartMax - chartMin || 1;
     const totalVal = values.reduce((sum, value) => sum + value, 0);
     const avgVal = totalVal / values.length;
     const deltaVal = values[values.length - 1] - values[0];
@@ -998,29 +1116,47 @@ function cmdTrend(flags) {
     }
     const chartWidth = Math.max(width - 12, 20);
     const yLabelStep = Math.max(1, Math.floor(chartHeight / 4));
-    const dotWidth = chartWidth * 2;
-    const dotHeight = chartHeight * 4;
     const chartPoints = values.map((value, i) => ({
-        x: values.length === 1 ? Math.floor(dotWidth / 2) : Math.round((i / (values.length - 1)) * (dotWidth - 1)),
-        y: Math.max(0, dotHeight - 1 - Math.round((value / maxVal) * (dotHeight - 1))),
+        x: values.length === 1 ? Math.floor(chartWidth / 2) : Math.round((i / (values.length - 1)) * (chartWidth - 1)),
+        y: Math.max(0, chartHeight - 1 - Math.round(((value - chartMin) / chartRange) * (chartHeight - 1))),
     }));
-    const rows = buildBrailleRows(chartPoints, chartWidth, chartHeight);
-    const labelPoints = chartPoints.map(point => ({ x: Math.floor(point.x / 2), y: point.y }));
+    const grid = buildLineRows(chartPoints, chartWidth, chartHeight);
+    // Metric premium color themes
+    let metricColor = "\x1b[36m"; // default Cyan
+    if (metric === "cost") {
+        metricColor = "\x1b[32m"; // Green
+    }
+    else if (metric === "messages") {
+        metricColor = "\x1b[33m"; // Yellow
+    }
+    const peakStr = `${metricColor}${BOLD}${formatTrendValue(maxVal, metric)}${RESET}`;
+    const avgStr = `${metricColor}${BOLD}${formatTrendValue(avgVal, metric)}${RESET}`;
+    let deltaColor = "\x1b[32m"; // Green
+    if (metric === "cost") {
+        deltaColor = deltaVal > 0 ? "\x1b[31m" : "\x1b[32m"; // Red for cost increase, Green for decrease
+    }
+    else {
+        deltaColor = deltaVal > 0 ? "\x1b[32m" : "\x1b[31m"; // Green for token/msg increase, Red for decrease
+    }
+    const deltaStr = `${deltaColor}${BOLD}${formatSignedTrendValue(deltaVal, metric)}${RESET}`;
     const lines = [];
-    lines.push(`${metricLabel(metric)} · ${sorted.length} days · peak ${formatTrendValue(maxVal, metric)} · avg ${formatTrendValue(avgVal, metric)} · Δ ${formatSignedTrendValue(deltaVal, metric)}`);
-    lines.push(`range ${formatTrendValue(minVal, metric)} → ${formatTrendValue(maxVal, metric)}`);
+    lines.push(`${metricColor}${BOLD}${metricLabel(metric)}${RESET} · ${sorted.length} days · peak ${peakStr} · avg ${avgStr} · Δ ${deltaStr}`);
+    lines.push(`${GRAY}range ${formatTrendValue(minVal, metric)} → ${formatTrendValue(maxVal, metric)}${RESET}`);
+    // Colorize the line chart points and connection characters
+    const colorRows = grid.map(row => row.map(char => char === " " ? " " : `${metricColor}${char}${RESET}`).join(""));
+    const { axisTicks, axisLabels } = buildTrendXAxis(sorted, chartPoints, chartWidth);
     for (let row = 0; row < chartHeight; row++) {
         const valueRatio = 1 - row / (chartHeight - 1);
-        const valAtRow = valueRatio * maxVal;
-        const label = row === 0 || row === chartHeight - 1 || row % yLabelStep === 0
-            ? formatTrendValue(valAtRow, metric)
-            : "";
-        const line = `${padLeft(label, 9)} ┤${rows[row]}`;
+        const valAtRow = chartMin + valueRatio * chartRange;
+        const hasLabel = row === 0 || row === chartHeight - 1 || row % yLabelStep === 0;
+        const label = hasLabel ? formatTrendValue(valAtRow, metric) : "";
+        const tick = hasLabel ? "┤" : "│";
+        const line = `${padLeft(label, 9)} ${GRAY}${tick}${RESET}${colorRows[row]}`;
         lines.push(line);
     }
-    const axis = `${" ".repeat(9)} └${"─".repeat(chartWidth)}`;
+    const axis = `${" ".repeat(9)} ${GRAY}└${axisTicks}${RESET}`;
     lines.push(axis);
-    lines.push(`${" ".repeat(11)}${buildTrendXAxis(sorted, labelPoints, chartWidth)}`);
+    lines.push(`${" ".repeat(11)}${GRAY}${axisLabels}${RESET}`);
     console.log();
     for (const l of lines)
         console.log(`  ${l}`);

package/dist/index.d.ts

@@ -1,6 +1,3 @@
 import type { Plugin } from "@opencode-ai/plugin";
-type ProviderFamily = "anthropic" | "openai" | "deepseek" | "google" | "other";
-export declare function getProviderFamily(model: string, provider: string): ProviderFamily;
-export declare function calculateCost(model: string, provider: string, input: number, output: number, cacheRead?: number, cacheWrite?: number): number;
 export declare const TokenTrackerPlugin: Plugin;
 export default TokenTrackerPlugin;

package/dist/index.js

@@ -1,4 +1,4 @@
-import { BUILTIN_PRICING, DEFAULT_CONFIG, findModelConfigPricing, formatCost, formatTokens, getStartOfDay, getStartOfWeek, getStartOfMonth, validateConfig } from "./lib/shared.js";
+import { DEFAULT_CONFIG, formatCost, formatTokens, getStartOfDay, getStartOfWeek, getStartOfMonth, validateConfig, evaluateBudgetStatus, calculateCost } from "./lib/shared.js";
 import { appendFileSync, existsSync, mkdirSync, readFileSync, statSync, openSync, readSync, closeSync } from "fs";
 import { open } from "fs/promises";
 import { join } from "path";
@@ -50,93 +50,6 @@ function ensureLatestConfig() {
         // Keep current config on error
     }
 }
-// ============================================================================
-// Pricing
-// ============================================================================
-function getModelPricing(model, provider) {
-    // 1. Check provider-level override first (highest priority)
-    if (config.providers[provider]) {
-        return config.providers[provider];
-    }
-    // 2. Check user-defined model pricing (exact match only)
-    const configuredPricing = findModelConfigPricing(config.models, model, provider, false);
-    if (configuredPricing) {
-        return configuredPricing;
-    }
-    // 3. Check built-in exact match
-    if (BUILTIN_PRICING[model]) {
-        return BUILTIN_PRICING[model];
-    }
-    // 4. Try partial match in built-in pricing
-    const modelLower = model.toLowerCase();
-    for (const [key, pricing] of Object.entries(BUILTIN_PRICING).sort(([a], [b]) => b.length - a.length)) {
-        if (key !== "_default" && modelLower.includes(key.toLowerCase())) {
-            return pricing;
-        }
-    }
-    // 5. Try partial match in user config
-    const partialUserPricing = findModelConfigPricing(config.models, model, provider, true);
-    if (partialUserPricing) {
-        return partialUserPricing;
-    }
-    // 6. Fallback to default
-    return BUILTIN_PRICING["_default"];
-}
-export function getProviderFamily(model, provider) {
-    const p = provider.toLowerCase();
-    const m = model.toLowerCase();
-    if (p.includes("anthropic") || m.startsWith("claude-")) {
-        return "anthropic";
-    }
-    if (p.includes("openai") ||
-        m.startsWith("gpt-") ||
-        m.startsWith("o1-") ||
-        m.startsWith("o3-") ||
-        m.startsWith("o4-") ||
-        m === "o3" ||
-        m === "o1") {
-        return "openai";
-    }
-    if (p.includes("deepseek") || m.includes("deepseek")) {
-        return "deepseek";
-    }
-    if (p.includes("google") || p.includes("vertex") || m.startsWith("gemini-")) {
-        return "google";
-    }
-    return "other";
-}
-export function calculateCost(model, provider, input, output, cacheRead = 0, cacheWrite = 0) {
-    const pricing = getModelPricing(model, provider);
-    const family = getProviderFamily(model, provider);
-    let defaultCacheReadRate = 0.5; // Default 50% discount (OpenAI style)
-    let defaultCacheWriteRate = 0; // Default free cache writing
-    if (family === "anthropic") {
-        defaultCacheReadRate = 0.1;
-        defaultCacheWriteRate = 1.25;
-    }
-    else if (family === "deepseek" || family === "google") {
-        defaultCacheReadRate = 0.1;
-        defaultCacheWriteRate = 0;
-    }
-    else if (family === "openai") {
-        defaultCacheReadRate = 0.5;
-        defaultCacheWriteRate = 0;
-    }
-    else {
-        // "other" / general default
-        defaultCacheReadRate = 0.5;
-        defaultCacheWriteRate = 0;
-    }
-    const finalCacheReadPrice = pricing.cacheRead ?? (pricing.input * defaultCacheReadRate);
-    const finalCacheWritePrice = pricing.cacheWrite ?? (pricing.input * defaultCacheWriteRate);
-    // Billable input = total input - cache read (cached tokens are charged at cache rate)
-    const billableInput = Math.max(0, input - cacheRead);
-    const inputCost = (billableInput / 1_000_000) * pricing.input;
-    const outputCost = (output / 1_000_000) * pricing.output;
-    const cacheReadCost = (cacheRead / 1_000_000) * finalCacheReadPrice;
-    const cacheWriteCost = (cacheWrite / 1_000_000) * finalCacheWritePrice;
-    return inputCost + outputCost + cacheReadCost + cacheWriteCost;
-}
 const sessionStats = new Map();
 function getOrCreateSessionStats(sessionId) {
     if (!sessionStats.has(sessionId)) {
@@ -387,48 +300,12 @@ function accumulateBudget(cost) {
     budgetTracker.monthlySpent += cost;
 }
 function checkBudgetStatus() {
-    const budget = config.budget;
-    if (!budget.daily && !budget.weekly && !budget.monthly) {
-        return null;
-    }
-    if (!budgetTracker.initialized)
-        return null;
-    const warnAt = budget.warnAt ?? 0.8;
-    // Check in order: daily -> weekly -> monthly (most restrictive first)
-    if (budget.daily) {
-        const percentage = budgetTracker.dailySpent / budget.daily;
-        return {
-            period: "daily",
-            spent: budgetTracker.dailySpent,
-            limit: budget.daily,
-            percentage,
-            exceeded: percentage >= 1,
-            warning: percentage >= warnAt && percentage < 1,
-        };
-    }
-    if (budget.weekly) {
-        const percentage = budgetTracker.weeklySpent / budget.weekly;
-        return {
-            period: "weekly",
-            spent: budgetTracker.weeklySpent,
-            limit: budget.weekly,
-            percentage,
-            exceeded: percentage >= 1,
-            warning: percentage >= warnAt && percentage < 1,
-        };
-    }
-    if (budget.monthly) {
-        const percentage = budgetTracker.monthlySpent / budget.monthly;
-        return {
-            period: "monthly",
-            spent: budgetTracker.monthlySpent,
-            limit: budget.monthly,
-            percentage,
-            exceeded: percentage >= 1,
-            warning: percentage >= warnAt && percentage < 1,
-        };
-    }
-    return null;
+    const snapshot = {
+        dailySpent: budgetTracker.dailySpent,
+        weeklySpent: budgetTracker.weeklySpent,
+        monthlySpent: budgetTracker.monthlySpent,
+    };
+    return evaluateBudgetStatus(config.budget, snapshot, budgetTracker.initialized);
 }
 function formatBudgetMessage(status) {
     const pct = Math.round(status.percentage * 100);
@@ -497,7 +374,7 @@ export const TokenTrackerPlugin = async ({ directory, client }) => {
                             return;
                         const model = info.model?.modelID ?? info.modelID ?? "unknown";
                         const provider = info.model?.providerID ?? info.providerID ?? "unknown";
-                        const cost = calculateCost(model, provider, input, output, cacheRead, cacheWrite);
+                        const cost = calculateCost(model, provider, input, output, cacheRead, cacheWrite, config);
                         // Update session stats
                         const stats = getOrCreateSessionStats(sessionId);
                         stats.totalInput += input;