opencode-token-tracker 1.6.5 → 1.7.0

package/README.md

@@ -23,6 +23,14 @@ This project uses the [AI Engineering Framework (AIEF)](https://github.com/tongs
 
 If you are building AI-assisted engineering workflows, we strongly recommend adopting AIEF in your own repositories for clearer context management and more consistent agent outputs.
 
+
+## Accuracy & Limitations
+
+- **Costs are estimates**, computed locally from your token logs and the built-in (or user-configured) pricing table. They may differ from your provider's official invoice — for example, when promotional credits, discounts, or enterprise pricing structures apply.
+- **Budgets are warnings, not enforcement.** This plugin does not block API calls, throttle requests, or interrupt active sessions. It is designed purely as an observability and tracking tool.
+- **Subscription or bundled providers** (such as GitHub Copilot, Cursor, etc.) or free local models should be configured with zero-cost overrides in your configuration file (see [Configuration](#configuration)).
+- **Pricing freshness**: The built-in pricing table is manually maintained. Please run `opencode-tokens models` to inspect which of your used models currently fall back to the default pricing, and configure overrides if necessary.
+
 ## Installation
 
 Add to your OpenCode config file (`~/.config/opencode/opencode.json`):
@@ -38,6 +46,8 @@ Restart OpenCode and the plugin will be automatically installed.
 
 ## Usage
 
+For an end-to-end setup and verification path, see [walkthrough.md](./walkthrough.md).
+
 ### Toast Notifications
 
 Once installed, you'll see Toast notifications after each AI response:
@@ -237,8 +247,12 @@ opencode-tokens models
 # Show current config
 opencode-tokens config
 
-# Generate example config based on your usage
+# Print clean example JSON to stdout without writing a file
 opencode-tokens config init
+
+# Write example config to ~/.config/opencode/token-tracker.json
+# Existing config is backed up to token-tracker.json.bak
+opencode-tokens config generate
 ```
 
 Example `models` output:
@@ -255,6 +269,8 @@ This helps you understand:
 - Whether pricing is from built-in table, your config, or default fallback
 - What to add to your config file
 
+`config init` is safe for piping because stdout contains only valid JSON and no file is written. `config generate` is the file-writing path: stdout stays empty, guides and status messages go to stderr, the parent directory is created when needed, and an existing config is backed up before overwrite.
+
 ## Log Files
 
 Token usage is logged to:
@@ -367,11 +383,14 @@ All prices are in **USD per 1 million tokens**:
 
 Pricing is resolved in this order (first match wins):
 
-1. **Provider-level** - Override all models for a provider
-2. **Provider-specific model config** - Custom pricing for the same model under different providers
-3. **User model config** - Generic custom model pricing in config file
-4. **Built-in pricing** - Default pricing table
-5. **Fallback** - $1/M input, $4/M output
+1. **Provider-level override** - Override all models for a provider
+2. **Exact user model config** - Custom pricing for a specific model or provider-specific model entry
+3. **Built-in exact match** - Exact key in the built-in pricing table
+4. **Built-in partial match** - Longest matching built-in key for variant model names
+5. **User model partial match** - Longest matching user config key
+6. **Fallback** - $1/M input, $4/M output
+
+Exact user config is intentionally checked before built-ins, while broad partial user keys are checked after built-ins so a generic key like `"claude"` does not accidentally override a precise built-in model price.
 
 #### Example: Free providers
 
@@ -437,12 +456,15 @@ npm install
 # Build
 npm run build
 
-# Link for local testing
-npm link
-cd ~/.config/opencode
-npm link opencode-token-tracker
+# Unit and CLI tests
+npm test
+
+# Real local OpenCode CLI dogfood
+node scripts/real-opencode-cli-smoke.mjs --use-temporary-link --model deepseek/deepseek-chat
 ```
 
+The dogfood script is repo-only and is not published as an npm command. It verifies the real local `opencode run` path, including OpenCode's cache package directory, and restores any temporary package links after the run.
+
 ## License
 
 MIT © [tongsh6](https://github.com/tongsh6)

package/README.zh-CN.md

@@ -23,6 +23,14 @@
 
 如果你在构建 AI-assisted engineering 工作流，推荐在你的仓库中采用 AIEF，以获得更清晰的上下文管理与更稳定的 agent 输出。
 
+
+## 准确性与限制
+
+- **成本均为估算值**，由本地 Token 日志及内置（或用户配置）的定价表计算得出。这可能与您的 Provider 官方账单存在差异 —— 例如，当您使用促销额度、企业折扣或特定定价优惠时。
+- **预算提醒仅为警告，非强制阻断**。本插件不会拦截 API 调用、节流请求或中断您的会话。其设计初衷纯粹是为了可观测性与用量追踪。
+- **订阅制或打包服务商**（如 GitHub Copilot、Cursor 等）或免费的本地模型，应在您的配置文件中将其价格覆写为 0（参见 [配置说明](#配置说明)）。
+- **定价数据时效性**：内置的定价表为手动维护。请运行 `opencode-tokens models` 来检查您当前使用的哪些模型退化（fallback）到了默认定价，并在需要时手动配置覆写。
+
 ## 安装
 
 在 OpenCode 配置文件 `~/.config/opencode/opencode.json` 中添加插件：
@@ -38,6 +46,8 @@
 
 ## 使用
 
+端到端安装、验证与 dogfood 路径见 [walkthrough.md](./walkthrough.md)。
+
 ### Toast 提示
 
 安装后，每次 AI 响应会看到类似提示：
@@ -152,6 +162,81 @@ opencode-tokens --by daily
   gpt-5.2               86.9K     $0.0000      18
 ```
 
+`--by` 可选：
+- `model`：按模型分组
+- `agent`：按 agent 分组
+- `provider`：按 provider 分组
+- `daily`：按天分组
+- `session`：按 session ID 分组
+- `all`：显示全部分组
+
+### 趋势图
+
+查看每日 token、成本或消息数趋势：
+
+```bash
+# 30 天成本趋势（默认）
+opencode-tokens trend
+
+# 7 天 token 趋势
+opencode-tokens trend --days 7 --metric tokens
+
+# 更紧凑的图表
+opencode-tokens trend --width 40
+```
+
+可选参数：
+- `--days N`：统计天数，默认 30
+- `--metric`：`cost`、`tokens` 或 `messages`，默认 `cost`
+- `--width W`：图表宽度，默认 60
+
+### 数据导出
+
+导出 JSONL 聚合后的用量数据：
+
+```bash
+# 导出全部数据为 CSV
+opencode-tokens export
+
+# 导出本月数据为 JSON
+opencode-tokens export --format json --period month
+
+# 写入文件
+opencode-tokens export --format csv --output usage.csv
+```
+
+可选参数：
+- `--format`：`csv` 或 `json`，默认 `csv`
+- `--period`：`today`、`week`、`month`、`all`，默认 `all`
+- `--output FILE`：写入文件，而不是 stdout
+
+### 配置管理
+
+可直接通过 CLI 管理预算与 Toast 配置：
+
+```bash
+# 查看当前配置
+opencode-tokens config
+
+# 设置每日预算为 $10
+opencode-tokens config set budget.daily 10
+
+# 关闭 Toast
+opencode-tokens config set toast.enabled false
+
+# 查看某个值
+opencode-tokens config get budget.warnAt
+
+# 恢复默认
+opencode-tokens config unset budget.daily
+```
+
+可设置字段：
+- `budget.daily`、`budget.weekly`、`budget.monthly`、`budget.warnAt`
+- `toast.enabled`、`toast.duration`、`toast.showOnIdle`
+
+配置变更会自动备份到 `token-tracker.json.bak`。
+
 ### 定价与配置命令
 
 ```bash
@@ -167,8 +252,12 @@ opencode-tokens models
 # 查看当前配置
 opencode-tokens config
 
-# 基于当前使用情况生成示例配置
+# 输出干净 JSON 到 stdout，不写文件
 opencode-tokens config init
+
+# 写入 ~/.config/opencode/token-tracker.json
+# 已有配置会备份到 token-tracker.json.bak
+opencode-tokens config generate
 ```
 
 `models` 示例输出：
@@ -186,6 +275,8 @@ opencode-tokens config init
 - 定价来源是内置表、用户配置还是默认回退
 - 需要在配置文件中补充哪些模型定价
 
+`config init` 适合管道重定向，因为 stdout 只包含合法 JSON，且不会写文件。`config generate` 是写文件路径：stdout 保持为空，说明和状态信息输出到 stderr；父目录不存在时会自动创建，覆盖已有配置前会先备份。
+
 ## 日志文件
 
 token 记录保存在：
@@ -285,10 +376,13 @@ token 记录保存在：
 定价解析顺序（命中即止）：
 
 1. **Provider 覆盖** — 为某个 provider 的所有模型统一设置
-2. **按 provider 的 model 配置** — 同一模型在不同 provider 下使用不同定价
-3. **用户 model 配置** — 为特定模型自定义通用定价
-4. **内置定价** — 默认定价表
-5. **默认回退** — $1/M input，$4/M output
+2. **用户 model 精确匹配** — 为特定模型或 provider-specific model entry 自定义定价
+3. **内置定价精确匹配** — 命中内置定价表中的精确 key
+4. **内置定价部分匹配** — 对变体模型名使用最长的内置 key 匹配
+5. **用户 model 部分匹配** — 使用最长的用户配置 key 匹配
+6. **默认回退** — $1/M input，$4/M output
+
+精确用户配置会优先于内置定价；但宽泛的用户部分匹配会排在内置匹配之后，避免 `"claude"` 这类泛 key 意外覆盖精确的内置模型价格。
 
 #### 示例：免费 provider
 
@@ -368,12 +462,15 @@ npm install
 # 构建
 npm run build
 
-# 本地联调
-npm link
-cd ~/.config/opencode
-npm link opencode-token-tracker
+# 单元与 CLI 测试
+npm test
+
+# 真实本机 OpenCode CLI dogfood
+node scripts/real-opencode-cli-smoke.mjs --use-temporary-link --model deepseek/deepseek-chat
 ```
 
+dogfood 脚本只作为仓库内开发工具，不作为 npm 包命令发布。它验证真实本机 `opencode run` 路径，包括 OpenCode 的 cache package 目录；运行结束后会恢复临时 package link。
+
 ## License
 
 MIT © [tongsh6](https://github.com/tongsh6)

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,6 +941,8 @@ 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.
 `);
 }
 function getTrendValue(point, metric) {

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;