lean-claudient-core 0.1.0

package/src/telemetry/tokenCounter.ts

@@ -0,0 +1,81 @@
+interface Turn {
+  inputTokens: number;
+  outputTokens: number;
+  timestamp: number;
+}
+
+interface Stats {
+  totalInputTokens: number;
+  totalOutputTokens: number;
+  turnCount: number;
+  estimatedCost: number;
+}
+
+interface CostSavings {
+  savedTokens: number;
+  savedDollars: number;
+  savingsPercent: number;
+}
+
+const INPUT_PRICE_PER_M = 3;
+const OUTPUT_PRICE_PER_M = 15;
+
+export class TokenCounter {
+  private turns: Turn[] = [];
+
+  constructor() {}
+
+  recordTurn(inputTokens: number, outputTokens: number): void {
+    this.turns.push({
+      inputTokens,
+      outputTokens,
+      timestamp: Date.now(),
+    });
+  }
+
+  getStats(): Stats {
+    let totalInputTokens = 0;
+    let totalOutputTokens = 0;
+
+    for (const turn of this.turns) {
+      totalInputTokens += turn.inputTokens;
+      totalOutputTokens += turn.outputTokens;
+    }
+
+    const estimatedCost =
+      (totalInputTokens * INPUT_PRICE_PER_M +
+        totalOutputTokens * OUTPUT_PRICE_PER_M) /
+      1000000;
+
+    return {
+      totalInputTokens,
+      totalOutputTokens,
+      turnCount: this.turns.length,
+      estimatedCost,
+    };
+  }
+
+  getCompressionRatio(baselineTokens: number): number {
+    const stats = this.getStats();
+    const totalTokens = stats.totalInputTokens + stats.totalOutputTokens;
+    return totalTokens / baselineTokens;
+  }
+
+  getCostSavings(
+    baselineTokens: number,
+    baselineCost: number
+  ): CostSavings {
+    const stats = this.getStats();
+    const totalTokens = stats.totalInputTokens + stats.totalOutputTokens;
+    const savedTokens = baselineTokens - totalTokens;
+    const savedDollars = baselineCost - stats.estimatedCost;
+    const savingsPercent =
+      baselineCost > 0 ? (savedDollars / baselineCost) * 100 : 0;
+
+    return {
+      savedTokens,
+      savedDollars,
+      savingsPercent,
+    };
+  }
+}