opencode-token-tracker 1.5.4 → 1.5.5

package/dist/index.d.ts

@@ -1,3 +1,6 @@
 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

@@ -76,14 +76,59 @@ function getModelPricing(model, provider) {
     // 5. Fallback to default
     return BUILTIN_PRICING["_default"];
 }
-function calculateCost(model, provider, input, output, cacheRead = 0, cacheWrite = 0) {
+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) * (pricing.cacheRead ?? pricing.input * 0.1);
-    const cacheWriteCost = (cacheWrite / 1_000_000) * (pricing.cacheWrite ?? pricing.input * 1.25);
+    const cacheReadCost = (cacheRead / 1_000_000) * finalCacheReadPrice;
+    const cacheWriteCost = (cacheWrite / 1_000_000) * finalCacheWritePrice;
     return inputCost + outputCost + cacheReadCost + cacheWriteCost;
 }
 const sessionStats = new Map();

package/dist/test/shared.test.js

@@ -1,6 +1,7 @@
 import { describe, it } from "node:test";
 import { strict as assert } from "node:assert";
 import { BUILTIN_PRICING, DEFAULT_CONFIG, findModelConfigPricing, formatCost, formatTokens, getStartOfDay, getStartOfWeek, getStartOfMonth, validateConfig, } from "../lib/shared.js";
+import { getProviderFamily, calculateCost } from "../index.js";
 // ============================================================================
 // formatCost
 // ============================================================================
@@ -376,3 +377,81 @@ describe("findModelConfigPricing", () => {
         assert.deepEqual(findModelConfigPricing(result.config.models, "prefix/my-model", "any-provider"), { input: 1, output: 2 });
     });
 });
+// ============================================================================
+// getProviderFamily
+// ============================================================================
+describe("getProviderFamily", () => {
+    it("should detect anthropic for various combinations", () => {
+        assert.equal(getProviderFamily("claude-opus-4.6", "openai"), "anthropic");
+        assert.equal(getProviderFamily("some-model", "anthropic"), "anthropic");
+        assert.equal(getProviderFamily("claude-3-5-sonnet", "openrouter"), "anthropic");
+    });
+    it("should detect openai for various combinations", () => {
+        assert.equal(getProviderFamily("gpt-4o", "unknown"), "openai");
+        assert.equal(getProviderFamily("o1-mini", "openrouter"), "openai");
+        assert.equal(getProviderFamily("o3", "together"), "openai");
+        assert.equal(getProviderFamily("some-model", "openai"), "openai");
+    });
+    it("should detect deepseek for various combinations", () => {
+        assert.equal(getProviderFamily("deepseek-chat", "siliconflow"), "deepseek");
+        assert.equal(getProviderFamily("deepseek-reasoner", "deepseek"), "deepseek");
+        assert.equal(getProviderFamily("deepseek/deepseek-r1", "openrouter"), "deepseek");
+    });
+    it("should detect google for various combinations", () => {
+        assert.equal(getProviderFamily("gemini-2.5-pro", "openrouter"), "google");
+        assert.equal(getProviderFamily("some-model", "google"), "google");
+        assert.equal(getProviderFamily("gemini-2.0-flash", "vertex"), "google");
+    });
+    it("should fallback to other for unknown providers/models", () => {
+        assert.equal(getProviderFamily("meta-llama-3-8b", "together"), "other");
+        assert.equal(getProviderFamily("unknown", "unknown"), "other");
+    });
+});
+// ============================================================================
+// calculateCost (provider-specific defaults verification)
+// ============================================================================
+describe("calculateCost", () => {
+    it("should apply Anthropic defaults correctly (cacheRead = 10%, cacheWrite = 125%)", () => {
+        // claude-sonnet-4 base: input = 3, output = 15.
+        // 1M inputs, 1M outputs, 100K cacheRead, 100K cacheWrite
+        const cost = calculateCost("claude-sonnet-4", "anthropic", 1_000_000, 1_000_000, 100_000, 100_000);
+        // Expected details:
+        // billable input = 1M - 100K = 900K tokens = 0.9 * 3.00 = $2.70
+        // output = 1M tokens = 1.0 * 15.00 = $15.00
+        // cacheRead = 100K tokens = 0.1 * 3.00 * 0.1 = $0.03 (10% discount rate)
+        // cacheWrite = 100K tokens = 0.1 * 3.00 * 1.25 = $0.375 (125% rate)
+        // total = 2.70 + 15.00 + 0.03 + 0.375 = $18.105
+        assert.ok(Math.abs(cost - 18.105) < 0.0001, `expected 18.105, got ${cost}`);
+    });
+    it("should apply OpenAI defaults correctly (cacheRead = 50%, cacheWrite = 0%)", () => {
+        // Built-in gpt-4o has input: 2.5, output: 10.
+        // 1M inputs, 1M outputs, 100K cacheRead, 100K cacheWrite
+        const cost = calculateCost("gpt-4o", "openai", 1_000_000, 1_000_000, 100_000, 100_000);
+        // Expected details:
+        // billable input = 1M - 100K = 900K tokens = 0.9 * 2.50 = $2.25
+        // output = 1M tokens = 1.0 * 10.00 = $10.00
+        // cacheRead = 100K tokens = 0.1 * 2.50 * 0.5 = $0.125 (50% discount rate)
+        // cacheWrite = 100K tokens = 0.1 * 2.50 * 0 = $0 (free)
+        // total = 2.25 + 10.00 + 0.125 + 0 = $12.375
+        assert.ok(Math.abs(cost - 12.375) < 0.0001, `expected 12.375, got ${cost}`);
+    });
+    it("should apply DeepSeek defaults correctly (cacheRead = 10%, cacheWrite = 0%)", () => {
+        // deepseek-chat has input: 0.28, output: 0.42. cacheRead: 0.028 (explicit in table)
+        const costExplicit = calculateCost("deepseek-chat", "deepseek", 1_000_000, 1_000_000, 100_000, 100_000);
+        // billable input = 900K = 0.9 * 0.28 = $0.252
+        // output = 1M = 1.0 * 0.42 = $0.42
+        // cacheRead = 100K = 0.1 * 0.028 = $0.0028
+        // cacheWrite = 100K = 0.1 * 0 = $0
+        // total = 0.252 + 0.42 + 0.0028 = $0.6748
+        assert.ok(Math.abs(costExplicit - 0.6748) < 0.0001, `expected 0.6748, got ${costExplicit}`);
+        // Verify fallback using non-builtin model under deepseek family:
+        const costFallback = calculateCost("deepseek-custom", "deepseek", 1_000_000, 1_000_000, 100_000, 100_000);
+        // Default base: input = 1, output = 4.
+        // billable input = 900K = 0.9 * 1.00 = $0.90
+        // output = 1M = 1.0 * 4.00 = $4.00
+        // cacheRead = 100K = 0.1 * 1.00 * 0.1 = $0.01 (10% rate fallback)
+        // cacheWrite = 100K = 0.1 * 1.00 * 0 = $0 (free cache write)
+        // total = 0.90 + 4.00 + 0.01 = $4.91
+        assert.ok(Math.abs(costFallback - 4.91) < 0.0001, `expected 4.91, got ${costFallback}`);
+    });
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-token-tracker",
-  "version": "1.5.4",
+  "version": "1.5.5",
   "description": "Real-time token usage and cost tracking plugin for OpenCode with Toast notifications and CLI stats",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",