awarts 0.2.2 → 0.2.3

package/dist/index.js

@@ -5880,6 +5880,27 @@ import path4 from "node:path";
 import os5 from "node:os";
 var CLAUDE_DIR = path4.join(os5.homedir(), ".claude");
 var STATS_CACHE = path4.join(CLAUDE_DIR, "stats-cache.json");
+var MODEL_PRICING = {
+  "claude-opus-4-6": { input: 15, output: 75, cacheRead: 1.875, cacheWrite: 18.75 },
+  "claude-opus-4": { input: 15, output: 75, cacheRead: 1.875, cacheWrite: 18.75 },
+  "claude-sonnet-4": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  "claude-sonnet-3-5": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  "claude-haiku-3-5": { input: 0.8, output: 4, cacheRead: 0.08, cacheWrite: 1 }
+};
+var DEFAULT_PRICING = { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 };
+function getPricing(model) {
+  if (MODEL_PRICING[model])
+    return MODEL_PRICING[model];
+  for (const [key, pricing] of Object.entries(MODEL_PRICING)) {
+    if (model.startsWith(key))
+      return pricing;
+  }
+  return DEFAULT_PRICING;
+}
+function estimateCost(model, inputTokens, outputTokens, cacheReadTokens, cacheCreationTokens) {
+  const p = getPricing(model);
+  return inputTokens / 1e6 * p.input + outputTokens / 1e6 * p.output + cacheReadTokens / 1e6 * p.cacheRead + cacheCreationTokens / 1e6 * p.cacheWrite;
+}
 async function fileExists(filePath) {
   try {
     const stat = await fs8.stat(filePath);
@@ -5908,12 +5929,16 @@ async function readStatsCache() {
   let totalCacheCreation = 0;
   let totalCost = 0;
   if (cache.modelUsage) {
-    for (const usage of Object.values(cache.modelUsage)) {
+    for (const [model, usage] of Object.entries(cache.modelUsage)) {
       totalInputTokens += usage.inputTokens || 0;
       totalOutputTokens += usage.outputTokens || 0;
       totalCacheRead += usage.cacheReadInputTokens || 0;
       totalCacheCreation += usage.cacheCreationInputTokens || 0;
-      totalCost += usage.costUSD || 0;
+      if (usage.costUSD > 0) {
+        totalCost += usage.costUSD;
+      } else {
+        totalCost += estimateCost(model, usage.inputTokens || 0, usage.outputTokens || 0, usage.cacheReadInputTokens || 0, usage.cacheCreationInputTokens || 0);
+      }
     }
   }
   const dailyOutputSum = dailyTokens.reduce((sum, day) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "awarts",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Track your AI coding across Claude, Codex, Gemini & Antigravity",
   "type": "module",
   "bin": {