oh-my-claude-sisyphus 3.5.5 → 3.5.7

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (63) hide show
  1. package/README.md +58 -343
  2. package/commands/mcp-setup.md +75 -64
  3. package/commands/omc-setup.md +51 -0
  4. package/dist/__tests__/analytics/tokscale-adapter.test.d.ts +2 -0
  5. package/dist/__tests__/analytics/tokscale-adapter.test.d.ts.map +1 -0
  6. package/dist/__tests__/analytics/tokscale-adapter.test.js +79 -0
  7. package/dist/__tests__/analytics/tokscale-adapter.test.js.map +1 -0
  8. package/dist/analytics/cost-estimator.d.ts +14 -0
  9. package/dist/analytics/cost-estimator.d.ts.map +1 -1
  10. package/dist/analytics/cost-estimator.js +65 -0
  11. package/dist/analytics/cost-estimator.js.map +1 -1
  12. package/dist/analytics/index.d.ts +1 -0
  13. package/dist/analytics/index.d.ts.map +1 -1
  14. package/dist/analytics/index.js +4 -0
  15. package/dist/analytics/index.js.map +1 -1
  16. package/dist/analytics/query-engine.d.ts +3 -0
  17. package/dist/analytics/query-engine.d.ts.map +1 -1
  18. package/dist/analytics/query-engine.js +87 -0
  19. package/dist/analytics/query-engine.js.map +1 -1
  20. package/dist/analytics/token-tracker.d.ts +3 -0
  21. package/dist/analytics/token-tracker.d.ts.map +1 -1
  22. package/dist/analytics/token-tracker.js +89 -0
  23. package/dist/analytics/token-tracker.js.map +1 -1
  24. package/dist/analytics/tokscale-adapter.d.ts +71 -0
  25. package/dist/analytics/tokscale-adapter.d.ts.map +1 -0
  26. package/dist/analytics/tokscale-adapter.js +223 -0
  27. package/dist/analytics/tokscale-adapter.js.map +1 -0
  28. package/dist/analytics/types.d.ts +17 -0
  29. package/dist/analytics/types.d.ts.map +1 -1
  30. package/dist/analytics/types.js +5 -0
  31. package/dist/analytics/types.js.map +1 -1
  32. package/dist/cli/analytics.js +35 -0
  33. package/dist/cli/analytics.js.map +1 -1
  34. package/dist/cli/commands/agents.d.ts.map +1 -1
  35. package/dist/cli/commands/agents.js +4 -2
  36. package/dist/cli/commands/agents.js.map +1 -1
  37. package/dist/cli/commands/stats.d.ts.map +1 -1
  38. package/dist/cli/commands/stats.js +6 -1
  39. package/dist/cli/commands/stats.js.map +1 -1
  40. package/dist/cli/index.js +95 -35
  41. package/dist/cli/index.js.map +1 -1
  42. package/dist/cli/utils/tokscale-launcher.d.ts +18 -0
  43. package/dist/cli/utils/tokscale-launcher.d.ts.map +1 -0
  44. package/dist/cli/utils/tokscale-launcher.js +64 -0
  45. package/dist/cli/utils/tokscale-launcher.js.map +1 -0
  46. package/dist/features/auto-update.js +5 -5
  47. package/dist/features/auto-update.js.map +1 -1
  48. package/dist/hud/types.d.ts +6 -2
  49. package/dist/hud/types.d.ts.map +1 -1
  50. package/dist/hud/types.js.map +1 -1
  51. package/dist/hud/usage-api.d.ts.map +1 -1
  52. package/dist/hud/usage-api.js +13 -1
  53. package/dist/hud/usage-api.js.map +1 -1
  54. package/docs/ARCHITECTURE.md +80 -397
  55. package/docs/FEATURES.md +396 -1981
  56. package/docs/MIGRATION.md +4 -4
  57. package/docs/REFERENCE.md +545 -0
  58. package/docs/SYNC-SYSTEM.md +6 -6
  59. package/package.json +2 -1
  60. package/scripts/sync-metadata.ts +2 -2
  61. package/skills/mcp-setup/SKILL.md +63 -75
  62. package/skills/omc-setup/SKILL.md +51 -0
  63. package/docs/FULL-README.md +0 -1001
@@ -82,6 +82,45 @@ fi
82
82
 
83
83
  **Note:** The `npm install` command triggers the `prepare` script which runs `npm run build`, creating the dist/ directory with all compiled HUD files.
84
84
 
85
+ ## Step 3.6: Install CLI Analytics Tools (Optional)
86
+
87
+ The OMC CLI provides standalone token analytics commands (`omc stats`, `omc agents`, `omc backfill`, `omc tui`).
88
+
89
+ Ask user: "Would you like to install the OMC CLI for standalone analytics? (Recommended for tracking token usage and costs)"
90
+
91
+ **Options:**
92
+ 1. **Yes (Recommended)** - Install CLI tools globally for `omc stats`, `omc agents`, etc.
93
+ 2. **No** - Skip CLI installation, use only plugin skills
94
+
95
+ ### If User Chooses YES:
96
+
97
+ ```bash
98
+ # Check for bun (preferred) or npm
99
+ if command -v bun &> /dev/null; then
100
+ echo "Installing OMC CLI via bun..."
101
+ bun install -g oh-my-claudecode
102
+ elif command -v npm &> /dev/null; then
103
+ echo "Installing OMC CLI via npm..."
104
+ npm install -g oh-my-claudecode
105
+ else
106
+ echo "ERROR: Neither bun nor npm found. Please install Node.js or Bun first."
107
+ exit 1
108
+ fi
109
+
110
+ # Verify installation
111
+ if command -v omc &> /dev/null; then
112
+ echo "✓ OMC CLI installed successfully!"
113
+ echo " Try: omc stats, omc agents, omc backfill"
114
+ else
115
+ echo "⚠ CLI installed but 'omc' not in PATH."
116
+ echo " You may need to restart your terminal or add npm/bun global bin to PATH."
117
+ fi
118
+ ```
119
+
120
+ ### If User Chooses NO:
121
+
122
+ Skip this step. User can install later with `npm install -g oh-my-claudecode`.
123
+
85
124
  ## Step 4: Verify Plugin Installation
86
125
 
87
126
  ```bash
@@ -144,6 +183,12 @@ Run /oh-my-claudecode:mcp-setup to add tools like web search, GitHub, etc.
144
183
  HUD STATUSLINE:
145
184
  The status bar now shows OMC state. Restart Claude Code to see it.
146
185
 
186
+ CLI ANALYTICS (if installed):
187
+ - omc - Full dashboard (stats + agents + cost)
188
+ - omc stats - View token usage and costs
189
+ - omc agents - See agent breakdown by cost
190
+ - omc tui - Launch interactive TUI dashboard
191
+
147
192
  That's it! Just use Claude Code normally.
148
193
  ```
149
194
 
@@ -173,6 +218,12 @@ MAGIC KEYWORDS (power-user shortcuts):
173
218
  HUD STATUSLINE:
174
219
  The status bar now shows OMC state. Restart Claude Code to see it.
175
220
 
221
+ CLI ANALYTICS (if installed):
222
+ - omc - Full dashboard (stats + agents + cost)
223
+ - omc stats - View token usage and costs
224
+ - omc agents - See agent breakdown by cost
225
+ - omc tui - Launch interactive TUI dashboard
226
+
176
227
  Your workflow won't break - it just got easier!
177
228
  ```
178
229
 
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=tokscale-adapter.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tokscale-adapter.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/analytics/tokscale-adapter.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,79 @@
1
+ import { describe, it, expect, beforeEach } from 'vitest';
2
+ import { getTokscaleAdapter, lookupPricingWithFallback, isTokscaleAvailable, resetAdapterCache } from '../../analytics/tokscale-adapter.js';
3
+ describe('tokscale-adapter', () => {
4
+ beforeEach(() => {
5
+ // Reset the cached adapter before each test
6
+ resetAdapterCache();
7
+ });
8
+ describe('getTokscaleAdapter', () => {
9
+ it('returns adapter with isAvailable property', async () => {
10
+ const adapter = await getTokscaleAdapter();
11
+ expect(adapter).toHaveProperty('isAvailable');
12
+ expect(typeof adapter.isAvailable).toBe('boolean');
13
+ });
14
+ it('caches adapter instance', async () => {
15
+ const adapter1 = await getTokscaleAdapter();
16
+ const adapter2 = await getTokscaleAdapter();
17
+ expect(adapter1).toBe(adapter2);
18
+ });
19
+ it('returns adapter with expected properties when available', async () => {
20
+ const adapter = await getTokscaleAdapter();
21
+ // Even when unavailable, should have isAvailable property
22
+ expect(adapter).toHaveProperty('isAvailable');
23
+ });
24
+ });
25
+ describe('lookupPricingWithFallback', () => {
26
+ it('returns pricing for known models', async () => {
27
+ const pricing = await lookupPricingWithFallback('claude-sonnet-4.5');
28
+ expect(pricing).toHaveProperty('inputPerMillion');
29
+ expect(pricing).toHaveProperty('outputPerMillion');
30
+ expect(pricing.inputPerMillion).toBeGreaterThan(0);
31
+ expect(pricing.outputPerMillion).toBeGreaterThan(0);
32
+ });
33
+ it('returns pricing for haiku model', async () => {
34
+ const pricing = await lookupPricingWithFallback('claude-haiku-4');
35
+ // Tokscale returns live pricing from LiteLLM database
36
+ expect(pricing.inputPerMillion).toBeGreaterThan(0);
37
+ expect(pricing.outputPerMillion).toBeGreaterThan(0);
38
+ expect(pricing.outputPerMillion).toBeGreaterThan(pricing.inputPerMillion);
39
+ });
40
+ it('returns pricing for opus model', async () => {
41
+ const pricing = await lookupPricingWithFallback('claude-opus-4.5');
42
+ // Tokscale returns live pricing from LiteLLM database
43
+ expect(pricing.inputPerMillion).toBeGreaterThan(0);
44
+ expect(pricing.outputPerMillion).toBeGreaterThan(0);
45
+ expect(pricing.outputPerMillion).toBeGreaterThan(pricing.inputPerMillion);
46
+ });
47
+ it('returns default pricing for unknown models', async () => {
48
+ const pricing = await lookupPricingWithFallback('unknown-model-xyz');
49
+ expect(pricing).toBeDefined();
50
+ expect(pricing).toHaveProperty('inputPerMillion');
51
+ expect(pricing).toHaveProperty('outputPerMillion');
52
+ });
53
+ it('includes cache pricing fields', async () => {
54
+ const pricing = await lookupPricingWithFallback('claude-sonnet-4.5');
55
+ expect(pricing).toHaveProperty('cacheWriteMarkup');
56
+ expect(pricing).toHaveProperty('cacheReadDiscount');
57
+ });
58
+ });
59
+ describe('isTokscaleAvailable', () => {
60
+ it('returns a boolean', async () => {
61
+ const available = await isTokscaleAvailable();
62
+ expect(typeof available).toBe('boolean');
63
+ });
64
+ });
65
+ describe('resetAdapterCache', () => {
66
+ it('clears the cached adapter', async () => {
67
+ // Get adapter to populate cache
68
+ const adapter1 = await getTokscaleAdapter();
69
+ // Reset cache
70
+ resetAdapterCache();
71
+ // Get adapter again - should create new instance
72
+ const adapter2 = await getTokscaleAdapter();
73
+ // Both should have same structure but might be different instances
74
+ // depending on whether tokscale is available
75
+ expect(adapter2).toHaveProperty('isAvailable');
76
+ });
77
+ });
78
+ });
79
+ //# sourceMappingURL=tokscale-adapter.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tokscale-adapter.test.js","sourceRoot":"","sources":["../../../src/__tests__/analytics/tokscale-adapter.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EACL,kBAAkB,EAClB,yBAAyB,EACzB,mBAAmB,EACnB,iBAAiB,EAClB,MAAM,qCAAqC,CAAC;AAE7C,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,4CAA4C;QAC5C,iBAAiB,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,OAAO,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,QAAQ,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAC5C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,OAAO,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAC3C,0DAA0D;YAC1D,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,OAAO,GAAG,MAAM,yBAAyB,CAAC,mBAAmB,CAAC,CAAC;YACrE,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;YAClD,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,OAAO,GAAG,MAAM,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;YAClE,sDAAsD;YACtD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACpD,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,OAAO,GAAG,MAAM,yBAAyB,CAAC,iBAAiB,CAAC,CAAC;YACnE,sDAAsD;YACtD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACpD,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,OAAO,GAAG,MAAM,yBAAyB,CAAC,mBAAmB,CAAC,CAAC;YACrE,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;YAClD,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,OAAO,GAAG,MAAM,yBAAyB,CAAC,mBAAmB,CAAC,CAAC;YACrE,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;YACjC,MAAM,SAAS,GAAG,MAAM,mBAAmB,EAAE,CAAC;YAC9C,MAAM,CAAC,OAAO,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,gCAAgC;YAChC,MAAM,QAAQ,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAE5C,cAAc;YACd,iBAAiB,EAAE,CAAC;YAEpB,iDAAiD;YACjD,MAAM,QAAQ,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAE5C,mEAAmE;YACnE,6CAA6C;YAC7C,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -6,7 +6,21 @@ export interface CostInput {
6
6
  cacheCreationTokens: number;
7
7
  cacheReadTokens: number;
8
8
  }
9
+ /**
10
+ * Synchronous cost calculation using hardcoded pricing.
11
+ * Fast path for HUD and sync operations that need instant response.
12
+ */
9
13
  export declare function calculateCost(input: CostInput): CostBreakdown;
14
+ /**
15
+ * Asynchronous cost calculation using live TokScale pricing.
16
+ * Provides up-to-date pricing with fallback to hardcoded values.
17
+ */
18
+ export declare function calculateCostAsync(input: CostInput): Promise<CostBreakdown>;
19
+ /**
20
+ * Batch cost calculation with efficient pricing lookup.
21
+ * Fetches pricing once per unique model for better performance.
22
+ */
23
+ export declare function batchCalculateCost(inputs: CostInput[]): Promise<CostBreakdown[]>;
10
24
  export declare function formatCost(cost: number): string;
11
25
  export declare function getCostColor(cost: number): 'green' | 'yellow' | 'red';
12
26
  export declare function estimateDailyCost(tokensPerHour: number, modelName: string): number;
@@ -1 +1 @@
1
- {"version":3,"file":"cost-estimator.d.ts","sourceRoot":"","sources":["../../src/analytics/cost-estimator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyB,aAAa,EAAE,MAAM,YAAY,CAAC;AAElE,MAAM,WAAW,SAAS;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,wBAAgB,aAAa,CAAC,KAAK,EAAE,SAAS,GAAG,aAAa,CA0B7D;AA8BD,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAK/C;AAED,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,GAAG,QAAQ,GAAG,KAAK,CAIrE;AAED,wBAAgB,iBAAiB,CAAC,aAAa,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAKlF;AAED,wBAAgB,mBAAmB,CAAC,aAAa,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAEpF"}
1
+ {"version":3,"file":"cost-estimator.d.ts","sourceRoot":"","sources":["../../src/analytics/cost-estimator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyB,aAAa,EAAE,MAAM,YAAY,CAAC;AAGlE,MAAM,WAAW,SAAS;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,eAAe,EAAE,MAAM,CAAC;CACzB;AAED;;;GAGG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,SAAS,GAAG,aAAa,CA0B7D;AAED;;;GAGG;AACH,wBAAsB,kBAAkB,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,CA0BjF;AAED;;;GAGG;AACH,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC,CAsCtF;AA8BD,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAK/C;AAED,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,GAAG,QAAQ,GAAG,KAAK,CAIrE;AAED,wBAAgB,iBAAiB,CAAC,aAAa,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAKlF;AAED,wBAAgB,mBAAmB,CAAC,aAAa,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAEpF"}
@@ -1,4 +1,9 @@
1
1
  import { PRICING } from './types.js';
2
+ import { lookupPricingWithFallback } from './tokscale-adapter.js';
3
+ /**
4
+ * Synchronous cost calculation using hardcoded pricing.
5
+ * Fast path for HUD and sync operations that need instant response.
6
+ */
2
7
  export function calculateCost(input) {
3
8
  const pricing = getPricingForModel(input.modelName);
4
9
  // Base input cost
@@ -20,6 +25,66 @@ export function calculateCost(input) {
20
25
  totalCost
21
26
  };
22
27
  }
28
+ /**
29
+ * Asynchronous cost calculation using live TokScale pricing.
30
+ * Provides up-to-date pricing with fallback to hardcoded values.
31
+ */
32
+ export async function calculateCostAsync(input) {
33
+ const pricing = await lookupPricingWithFallback(input.modelName);
34
+ // Base input cost
35
+ const inputCost = (input.inputTokens / 1_000_000) * pricing.inputPerMillion;
36
+ // Output cost
37
+ const outputCost = (input.outputTokens / 1_000_000) * pricing.outputPerMillion;
38
+ // Cache write cost (25% markup on input price)
39
+ const cacheWriteCost = (input.cacheCreationTokens / 1_000_000) *
40
+ pricing.inputPerMillion * (1 + pricing.cacheWriteMarkup);
41
+ // Cache read cost (90% discount on input price)
42
+ const cacheReadCost = (input.cacheReadTokens / 1_000_000) *
43
+ pricing.inputPerMillion * (1 - pricing.cacheReadDiscount);
44
+ const totalCost = inputCost + outputCost + cacheWriteCost + cacheReadCost;
45
+ return {
46
+ inputCost,
47
+ outputCost,
48
+ cacheWriteCost,
49
+ cacheReadCost,
50
+ totalCost
51
+ };
52
+ }
53
+ /**
54
+ * Batch cost calculation with efficient pricing lookup.
55
+ * Fetches pricing once per unique model for better performance.
56
+ */
57
+ export async function batchCalculateCost(inputs) {
58
+ // Get all unique models first
59
+ const models = [...new Set(inputs.map(i => i.modelName))];
60
+ const pricingMap = new Map();
61
+ // Fetch pricing for all unique models
62
+ for (const model of models) {
63
+ pricingMap.set(model, await lookupPricingWithFallback(model));
64
+ }
65
+ // Calculate costs using cached pricing
66
+ return inputs.map(input => {
67
+ const pricing = pricingMap.get(input.modelName);
68
+ // Base input cost
69
+ const inputCost = (input.inputTokens / 1_000_000) * pricing.inputPerMillion;
70
+ // Output cost
71
+ const outputCost = (input.outputTokens / 1_000_000) * pricing.outputPerMillion;
72
+ // Cache write cost (25% markup on input price)
73
+ const cacheWriteCost = (input.cacheCreationTokens / 1_000_000) *
74
+ pricing.inputPerMillion * (1 + pricing.cacheWriteMarkup);
75
+ // Cache read cost (90% discount on input price)
76
+ const cacheReadCost = (input.cacheReadTokens / 1_000_000) *
77
+ pricing.inputPerMillion * (1 - pricing.cacheReadDiscount);
78
+ const totalCost = inputCost + outputCost + cacheWriteCost + cacheReadCost;
79
+ return {
80
+ inputCost,
81
+ outputCost,
82
+ cacheWriteCost,
83
+ cacheReadCost,
84
+ totalCost
85
+ };
86
+ });
87
+ }
23
88
  function getPricingForModel(modelName) {
24
89
  // Normalize model name
25
90
  const normalized = normalizeModelName(modelName);
@@ -1 +1 @@
1
- {"version":3,"file":"cost-estimator.js","sourceRoot":"","sources":["../../src/analytics/cost-estimator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAA+B,MAAM,YAAY,CAAC;AAUlE,MAAM,UAAU,aAAa,CAAC,KAAgB;IAC5C,MAAM,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAEpD,kBAAkB;IAClB,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;IAE5E,cAAc;IACd,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAE/E,+CAA+C;IAC/C,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,mBAAmB,GAAG,SAAS,CAAC;QAC5D,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAE3D,gDAAgD;IAChD,MAAM,aAAa,GAAG,CAAC,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;QACvD,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAE5D,MAAM,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,cAAc,GAAG,aAAa,CAAC;IAE1E,OAAO;QACL,SAAS;QACT,UAAU;QACV,cAAc;QACd,aAAa;QACb,SAAS;KACV,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,uBAAuB;IACvB,MAAM,UAAU,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAEjD,IAAI,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACxB,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC;IAC7B,CAAC;IAED,+BAA+B;IAC/B,OAAO,CAAC,IAAI,CAAC,kBAAkB,SAAS,gCAAgC,CAAC,CAAC;IAC1E,OAAO,OAAO,CAAC,mBAAmB,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,oCAAoC;IACpC,MAAM,KAAK,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IAEtC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,gBAAgB,CAAC;IACrD,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,mBAAmB,CAAC;IACzD,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,iBAAiB,CAAC;IAErD,sBAAsB;IACtB,IAAI,OAAO,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IAEzC,UAAU;IACV,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,IAAY;IACrC,IAAI,IAAI,GAAG,IAAI,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACxC,CAAC;IACD,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,IAAY;IACvC,IAAI,IAAI,GAAG,GAAG;QAAE,OAAO,OAAO,CAAC;IAC/B,IAAI,IAAI,GAAG,GAAG;QAAE,OAAO,QAAQ,CAAC;IAChC,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,aAAqB,EAAE,SAAiB;IACxE,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAC9C,MAAM,YAAY,GAAG,aAAa,GAAG,EAAE,CAAC;IACxC,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;IACxE,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,aAAqB,EAAE,SAAiB;IAC1E,OAAO,iBAAiB,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,EAAE,CAAC;AAC1D,CAAC"}
1
+ {"version":3,"file":"cost-estimator.js","sourceRoot":"","sources":["../../src/analytics/cost-estimator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAA+B,MAAM,YAAY,CAAC;AAClE,OAAO,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AAUlE;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,KAAgB;IAC5C,MAAM,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAEpD,kBAAkB;IAClB,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;IAE5E,cAAc;IACd,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAE/E,+CAA+C;IAC/C,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,mBAAmB,GAAG,SAAS,CAAC;QAC5D,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAE3D,gDAAgD;IAChD,MAAM,aAAa,GAAG,CAAC,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;QACvD,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAE5D,MAAM,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,cAAc,GAAG,aAAa,CAAC;IAE1E,OAAO;QACL,SAAS;QACT,UAAU;QACV,cAAc;QACd,aAAa;QACb,SAAS;KACV,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,KAAgB;IACvD,MAAM,OAAO,GAAG,MAAM,yBAAyB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAEjE,kBAAkB;IAClB,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;IAE5E,cAAc;IACd,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAE/E,+CAA+C;IAC/C,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,mBAAmB,GAAG,SAAS,CAAC;QAC5D,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAE3D,gDAAgD;IAChD,MAAM,aAAa,GAAG,CAAC,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;QACvD,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAE5D,MAAM,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,cAAc,GAAG,aAAa,CAAC;IAE1E,OAAO;QACL,SAAS;QACT,UAAU;QACV,cAAc;QACd,aAAa;QACb,SAAS;KACV,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,MAAmB;IAC1D,8BAA8B;IAC9B,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC1D,MAAM,UAAU,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEnD,sCAAsC;IACtC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,uCAAuC;IACvC,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;QACxB,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAE,CAAC;QAEjD,kBAAkB;QAClB,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;QAE5E,cAAc;QACd,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;QAE/E,+CAA+C;QAC/C,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,mBAAmB,GAAG,SAAS,CAAC;YAC5D,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAE3D,gDAAgD;QAChD,MAAM,aAAa,GAAG,CAAC,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;YACvD,OAAO,CAAC,eAAe,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAE5D,MAAM,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,cAAc,GAAG,aAAa,CAAC;QAE1E,OAAO;YACL,SAAS;YACT,UAAU;YACV,cAAc;YACd,aAAa;YACb,SAAS;SACV,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,uBAAuB;IACvB,MAAM,UAAU,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAEjD,IAAI,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACxB,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC;IAC7B,CAAC;IAED,+BAA+B;IAC/B,OAAO,CAAC,IAAI,CAAC,kBAAkB,SAAS,gCAAgC,CAAC,CAAC;IAC1E,OAAO,OAAO,CAAC,mBAAmB,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,oCAAoC;IACpC,MAAM,KAAK,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IAEtC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,gBAAgB,CAAC;IACrD,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,mBAAmB,CAAC;IACzD,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,iBAAiB,CAAC;IAErD,sBAAsB;IACtB,IAAI,OAAO,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IAEzC,UAAU;IACV,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,IAAY;IACrC,IAAI,IAAI,GAAG,IAAI,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACxC,CAAC;IACD,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,IAAY;IACvC,IAAI,IAAI,GAAG,GAAG;QAAE,OAAO,OAAO,CAAC;IAC/B,IAAI,IAAI,GAAG,GAAG;QAAE,OAAO,QAAQ,CAAC;IAChC,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,aAAqB,EAAE,SAAiB;IACxE,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAC9C,MAAM,YAAY,GAAG,aAAa,GAAG,EAAE,CAAC;IACxC,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;IACxE,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,aAAqB,EAAE,SAAiB;IAC1E,OAAO,iBAAiB,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,EAAE,CAAC;AAC1D,CAAC"}
@@ -19,4 +19,5 @@ export * from './transcript-parser.js';
19
19
  export * from './transcript-token-extractor.js';
20
20
  export * from './backfill-dedup.js';
21
21
  export * from './backfill-engine.js';
22
+ export * from './tokscale-adapter.js';
22
23
  //# sourceMappingURL=index.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/analytics/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,cAAc,wBAAwB,CAAC;AACvC,cAAc,mBAAmB,CAAC;AAClC,cAAc,aAAa,CAAC;AAG5B,cAAc,sBAAsB,CAAC;AACrC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AAGvC,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,iCAAiC,CAAC;AAChD,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/analytics/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,cAAc,wBAAwB,CAAC;AACvC,cAAc,mBAAmB,CAAC;AAClC,cAAc,aAAa,CAAC;AAG5B,cAAc,sBAAsB,CAAC;AACrC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AAGvC,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,iCAAiC,CAAC;AAIhD,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AAGrC,cAAc,uBAAuB,CAAC"}
@@ -19,6 +19,10 @@ export * from './analytics-summary.js';
19
19
  export * from './transcript-scanner.js';
20
20
  export * from './transcript-parser.js';
21
21
  export * from './transcript-token-extractor.js';
22
+ // DEPRECATED: Backfill is no longer needed - tokscale reads transcripts directly
23
+ // Kept for backward compatibility but no longer used by CLI
22
24
  export * from './backfill-dedup.js';
23
25
  export * from './backfill-engine.js';
26
+ // Tokscale integration (replaces backfill)
27
+ export * from './tokscale-adapter.js';
24
28
  //# sourceMappingURL=index.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/analytics/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,cAAc,wBAAwB,CAAC;AACvC,cAAc,mBAAmB,CAAC;AAClC,cAAc,aAAa,CAAC;AAE5B,kCAAkC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AAEvC,sCAAsC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,iCAAiC,CAAC;AAChD,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/analytics/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,cAAc,wBAAwB,CAAC;AACvC,cAAc,mBAAmB,CAAC;AAClC,cAAc,aAAa,CAAC;AAE5B,kCAAkC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AAEvC,sCAAsC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,iCAAiC,CAAC;AAEhD,iFAAiF;AACjF,4DAA4D;AAC5D,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AAErC,2CAA2C;AAC3C,cAAc,uBAAuB,CAAC"}
@@ -21,6 +21,9 @@ export interface UsagePattern {
21
21
  }
22
22
  export declare class QueryEngine {
23
23
  getCostReport(period: 'daily' | 'weekly' | 'monthly'): Promise<CostReport>;
24
+ private getCostReportViaTokscale;
25
+ private getCostReportLegacy;
26
+ private getAgentCostFromLocalLog;
24
27
  getUsagePatterns(): Promise<UsagePattern>;
25
28
  cleanupOldData(retentionDays?: number): Promise<{
26
29
  removedTokens: number;
@@ -1 +1 @@
1
- {"version":3,"file":"query-engine.d.ts","sourceRoot":"","sources":["../../src/analytics/query-engine.ts"],"names":[],"mappings":"AAQA,MAAM,WAAW,SAAS;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/B,MAAM,EAAE,OAAO,GAAG,QAAQ,GAAG,SAAS,CAAC;IACvC,KAAK,EAAE,SAAS,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,uBAAuB,EAAE,KAAK,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACpE,qBAAqB,EAAE,MAAM,CAAC;IAC9B,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,WAAW;IAChB,aAAa,CAAC,MAAM,EAAE,OAAO,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC;IAkE1E,gBAAgB,IAAI,OAAO,CAAC,YAAY,CAAC;IAiEzC,cAAc,CAAC,aAAa,GAAE,MAAW,GAAG,OAAO,CAAC;QAAE,aAAa,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,CAAC;IAU5G,OAAO,CAAC,kBAAkB;CAiB3B;AAKD,wBAAgB,cAAc,IAAI,WAAW,CAK5C"}
1
+ {"version":3,"file":"query-engine.d.ts","sourceRoot":"","sources":["../../src/analytics/query-engine.ts"],"names":[],"mappings":"AASA,MAAM,WAAW,SAAS;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/B,MAAM,EAAE,OAAO,GAAG,QAAQ,GAAG,SAAS,CAAC;IACvC,KAAK,EAAE,SAAS,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,uBAAuB,EAAE,KAAK,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACpE,qBAAqB,EAAE,MAAM,CAAC;IAC9B,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,WAAW;IAChB,aAAa,CAAC,MAAM,EAAE,OAAO,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC;YAWlE,wBAAwB;YAuDxB,mBAAmB;YAmEnB,wBAAwB;IAmChC,gBAAgB,IAAI,OAAO,CAAC,YAAY,CAAC;IAiEzC,cAAc,CAAC,aAAa,GAAE,MAAW,GAAG,OAAO,CAAC;QAAE,aAAa,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,CAAC;IAU5G,OAAO,CAAC,kBAAkB;CAiB3B;AAKD,wBAAgB,cAAc,IAAI,WAAW,CAK5C"}
@@ -1,11 +1,67 @@
1
1
  import { getTokenTracker } from './token-tracker.js';
2
2
  import { getSessionManager } from './session-manager.js';
3
3
  import { calculateCost } from './cost-estimator.js';
4
+ import { getTokscaleAdapter } from './tokscale-adapter.js';
4
5
  import * as fs from 'fs/promises';
5
6
  import * as path from 'path';
6
7
  import { homedir } from 'os';
7
8
  export class QueryEngine {
8
9
  async getCostReport(period) {
10
+ const adapter = await getTokscaleAdapter();
11
+ if (adapter.isAvailable && adapter.getReport) {
12
+ return this.getCostReportViaTokscale(adapter, period);
13
+ }
14
+ // Fallback to existing implementation
15
+ return this.getCostReportLegacy(period);
16
+ }
17
+ async getCostReportViaTokscale(adapter, period) {
18
+ const range = this.calculateTimeRange(period);
19
+ try {
20
+ const report = await adapter.getReport();
21
+ // Get agent data from local JSONL (populated by backfill)
22
+ const byAgent = await this.getAgentCostFromLocalLog(range);
23
+ const byModel = {};
24
+ for (const [model, data] of Object.entries(report.byModel)) {
25
+ byModel[model] = data.cost;
26
+ }
27
+ return {
28
+ totalCost: report.totalCost,
29
+ byAgent, // From local JSONL
30
+ byModel,
31
+ byDay: {},
32
+ period,
33
+ range
34
+ };
35
+ }
36
+ catch (error) {
37
+ // If tokscale fails, fall back to legacy
38
+ return this.getCostReportLegacy(period);
39
+ }
40
+ }
41
+ // UNUSED: Kept for reference, but tokscale path no longer mixes local JSONL
42
+ // private async convertParsedToEntries(parsed: any[], range: TimeRange): Promise<any[]> {
43
+ // const tokenLogPath = path.join(homedir(), '.omc', 'state', 'token-tracking.jsonl');
44
+ // const entries: any[] = [];
45
+ //
46
+ // try {
47
+ // const content = await fs.readFile(tokenLogPath, 'utf-8');
48
+ // const lines = content.trim().split('\n').filter(l => l.length > 0);
49
+ //
50
+ // for (const line of lines) {
51
+ // const record = JSON.parse(line);
52
+ //
53
+ // // Filter by time range
54
+ // if (record.timestamp >= range.start && record.timestamp <= range.end) {
55
+ // entries.push(record);
56
+ // }
57
+ // }
58
+ // } catch (error) {
59
+ // // Return empty if error
60
+ // }
61
+ //
62
+ // return entries;
63
+ // }
64
+ async getCostReportLegacy(period) {
9
65
  const range = this.calculateTimeRange(period);
10
66
  const tokenLogPath = path.join(homedir(), '.omc', 'state', 'token-tracking.jsonl');
11
67
  try {
@@ -61,6 +117,37 @@ export class QueryEngine {
61
117
  };
62
118
  }
63
119
  }
120
+ // Hybrid data merging: Read agent attribution from local JSONL
121
+ async getAgentCostFromLocalLog(range) {
122
+ const tokenLogPath = path.join(homedir(), '.omc', 'state', 'token-tracking.jsonl');
123
+ const byAgent = {};
124
+ try {
125
+ const content = await fs.readFile(tokenLogPath, 'utf-8');
126
+ const lines = content.trim().split('\n').filter(l => l.length > 0);
127
+ for (const line of lines) {
128
+ const record = JSON.parse(line);
129
+ // Filter by time range
130
+ if (record.timestamp < range.start || record.timestamp > range.end) {
131
+ continue;
132
+ }
133
+ // Aggregate by agent
134
+ if (record.agentName) {
135
+ const cost = calculateCost({
136
+ modelName: record.modelName,
137
+ inputTokens: record.inputTokens,
138
+ outputTokens: record.outputTokens,
139
+ cacheCreationTokens: record.cacheCreationTokens,
140
+ cacheReadTokens: record.cacheReadTokens
141
+ });
142
+ byAgent[record.agentName] = (byAgent[record.agentName] || 0) + cost.totalCost;
143
+ }
144
+ }
145
+ }
146
+ catch (error) {
147
+ // Return empty if error
148
+ }
149
+ return byAgent;
150
+ }
64
151
  async getUsagePatterns() {
65
152
  const tokenLogPath = path.join(homedir(), '.omc', 'state', 'token-tracking.jsonl');
66
153
  const manager = getSessionManager();
@@ -1 +1 @@
1
- {"version":3,"file":"query-engine.js","sourceRoot":"","sources":["../../src/analytics/query-engine.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAuB7B,MAAM,OAAO,WAAW;IACtB,KAAK,CAAC,aAAa,CAAC,MAAsC;QACxD,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;QAEnF,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACzD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAEnE,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,MAAM,KAAK,GAA2B,EAAE,CAAC;YAEzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEhC,uBAAuB;gBACvB,IAAI,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC;oBACnE,SAAS;gBACX,CAAC;gBAED,iCAAiC;gBACjC,MAAM,IAAI,GAAG,aAAa,CAAC;oBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;oBAC/C,eAAe,EAAE,MAAM,CAAC,eAAe;iBACxC,CAAC,CAAC;gBAEH,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;gBAE5B,qBAAqB;gBACrB,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;gBAChF,CAAC;gBAED,qBAAqB;gBACrB,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;gBAE9E,mBAAmB;gBACnB,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC3C,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;YAClD,CAAC;YAED,OAAO;gBACL,SAAS;gBACT,OAAO;gBACP,OAAO;gBACP,KAAK;gBACL,MAAM;gBACN,KAAK;aACN,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iCAAiC;YACjC,OAAO;gBACL,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE;gBACX,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,EAAE;gBACT,MAAM;gBACN,KAAK;aACN,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,gBAAgB;QACpB,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;QACnF,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;QAE3C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACzD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAEnE,MAAM,UAAU,GAA2B,EAAE,CAAC;YAC9C,MAAM,cAAc,GAA2B,EAAE,CAAC;YAElD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEhC,mBAAmB;gBACnB,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;gBACnD,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gBAE/C,mCAAmC;gBACnC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,MAAM,IAAI,GAAG,aAAa,CAAC;wBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;wBAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;wBAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;wBACjC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;wBAC/C,eAAe,EAAE,MAAM,CAAC,eAAe;qBACxC,CAAC,CAAC;oBAEH,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC9F,CAAC;YACH,CAAC;YAED,0BAA0B;YAC1B,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;iBACzC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;iBAC7B,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;iBACX,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YAEnC,yCAAyC;YACzC,MAAM,uBAAuB,GAAG,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC;iBAC3D,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;iBAC7B,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;iBACX,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAErD,MAAM,qBAAqB,GAAG,OAAO,CAAC,aAAa,GAAG,CAAC;gBACrD,CAAC,CAAC,OAAO,CAAC,SAAS,GAAG,OAAO,CAAC,aAAa;gBAC3C,CAAC,CAAC,CAAC,CAAC;YAEN,OAAO;gBACL,SAAS;gBACT,uBAAuB;gBACvB,qBAAqB;gBACrB,aAAa,EAAE,OAAO,CAAC,aAAa;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,SAAS,EAAE,EAAE;gBACb,uBAAuB,EAAE,EAAE;gBAC3B,qBAAqB,EAAE,CAAC;gBACxB,aAAa,EAAE,CAAC;aACjB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,gBAAwB,EAAE;QAC7C,MAAM,OAAO,GAAG,eAAe,EAAE,CAAC;QAClC,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAElE,iEAAiE;QACjE,MAAM,cAAc,GAAG,CAAC,CAAC;QAEzB,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;IAC3C,CAAC;IAEO,kBAAkB,CAAC,MAAsC;QAC/D,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC;QAEzB,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACvB,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACrC,CAAC;aAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACrC,CAAC;aAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAChC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QACtC,CAAC;QAED,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,WAAW,EAAE;YAC1B,GAAG,EAAE,GAAG,CAAC,WAAW,EAAE;SACvB,CAAC;IACJ,CAAC;CACF;AAED,qBAAqB;AACrB,IAAI,YAAY,GAAuB,IAAI,CAAC;AAE5C,MAAM,UAAU,cAAc;IAC5B,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,IAAI,WAAW,EAAE,CAAC;IACnC,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC"}
1
+ {"version":3,"file":"query-engine.js","sourceRoot":"","sources":["../../src/analytics/query-engine.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAmB,MAAM,uBAAuB,CAAC;AAC5E,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAuB7B,MAAM,OAAO,WAAW;IACtB,KAAK,CAAC,aAAa,CAAC,MAAsC;QACxD,MAAM,OAAO,GAAG,MAAM,kBAAkB,EAAE,CAAC;QAE3C,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YAC7C,OAAO,IAAI,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxD,CAAC;QAED,sCAAsC;QACtC,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC;IAEO,KAAK,CAAC,wBAAwB,CACpC,OAAwB,EACxB,MAAsC;QAEtC,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAE9C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAU,EAAE,CAAC;YAE1C,0DAA0D;YAC1D,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;YAE3D,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC3D,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC;YAC7B,CAAC;YAED,OAAO;gBACL,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,OAAO,EAAE,mBAAmB;gBAC5B,OAAO;gBACP,KAAK,EAAE,EAAE;gBACT,MAAM;gBACN,KAAK;aACN,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yCAAyC;YACzC,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,0FAA0F;IAC1F,wFAAwF;IACxF,+BAA+B;IAC/B,EAAE;IACF,UAAU;IACV,gEAAgE;IAChE,0EAA0E;IAC1E,EAAE;IACF,kCAAkC;IAClC,yCAAyC;IACzC,EAAE;IACF,gCAAgC;IAChC,gFAAgF;IAChF,gCAAgC;IAChC,UAAU;IACV,QAAQ;IACR,sBAAsB;IACtB,+BAA+B;IAC/B,MAAM;IACN,EAAE;IACF,oBAAoB;IACpB,IAAI;IAEI,KAAK,CAAC,mBAAmB,CAAC,MAAsC;QACtE,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;QAEnF,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACzD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAEnE,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,MAAM,KAAK,GAA2B,EAAE,CAAC;YAEzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEhC,uBAAuB;gBACvB,IAAI,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC;oBACnE,SAAS;gBACX,CAAC;gBAED,iCAAiC;gBACjC,MAAM,IAAI,GAAG,aAAa,CAAC;oBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;oBAC/C,eAAe,EAAE,MAAM,CAAC,eAAe;iBACxC,CAAC,CAAC;gBAEH,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;gBAE5B,qBAAqB;gBACrB,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;gBAChF,CAAC;gBAED,qBAAqB;gBACrB,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;gBAE9E,mBAAmB;gBACnB,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC3C,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;YAClD,CAAC;YAED,OAAO;gBACL,SAAS;gBACT,OAAO;gBACP,OAAO;gBACP,KAAK;gBACL,MAAM;gBACN,KAAK;aACN,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iCAAiC;YACjC,OAAO;gBACL,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,EAAE;gBACX,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,EAAE;gBACT,MAAM;gBACN,KAAK;aACN,CAAC;QACJ,CAAC;IACH,CAAC;IAED,+DAA+D;IACvD,KAAK,CAAC,wBAAwB,CAAC,KAAgB;QACrD,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;QACnF,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACzD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAEnE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEhC,uBAAuB;gBACvB,IAAI,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC;oBACnE,SAAS;gBACX,CAAC;gBAED,qBAAqB;gBACrB,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,MAAM,IAAI,GAAG,aAAa,CAAC;wBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;wBAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;wBAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;wBACjC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;wBAC/C,eAAe,EAAE,MAAM,CAAC,eAAe;qBACxC,CAAC,CAAC;oBACH,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;gBAChF,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,wBAAwB;QAC1B,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,gBAAgB;QACpB,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;QACnF,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;QAE3C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACzD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAEnE,MAAM,UAAU,GAA2B,EAAE,CAAC;YAC9C,MAAM,cAAc,GAA2B,EAAE,CAAC;YAElD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEhC,mBAAmB;gBACnB,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;gBACnD,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gBAE/C,mCAAmC;gBACnC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,MAAM,IAAI,GAAG,aAAa,CAAC;wBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;wBAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;wBAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;wBACjC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;wBAC/C,eAAe,EAAE,MAAM,CAAC,eAAe;qBACxC,CAAC,CAAC;oBAEH,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC9F,CAAC;YACH,CAAC;YAED,0BAA0B;YAC1B,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;iBACzC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;iBAC7B,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;iBACX,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YAEnC,yCAAyC;YACzC,MAAM,uBAAuB,GAAG,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC;iBAC3D,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;iBAC7B,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;iBACX,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAErD,MAAM,qBAAqB,GAAG,OAAO,CAAC,aAAa,GAAG,CAAC;gBACrD,CAAC,CAAC,OAAO,CAAC,SAAS,GAAG,OAAO,CAAC,aAAa;gBAC3C,CAAC,CAAC,CAAC,CAAC;YAEN,OAAO;gBACL,SAAS;gBACT,uBAAuB;gBACvB,qBAAqB;gBACrB,aAAa,EAAE,OAAO,CAAC,aAAa;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,SAAS,EAAE,EAAE;gBACb,uBAAuB,EAAE,EAAE;gBAC3B,qBAAqB,EAAE,CAAC;gBACxB,aAAa,EAAE,CAAC;aACjB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,gBAAwB,EAAE;QAC7C,MAAM,OAAO,GAAG,eAAe,EAAE,CAAC;QAClC,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAElE,iEAAiE;QACjE,MAAM,cAAc,GAAG,CAAC,CAAC;QAEzB,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;IAC3C,CAAC;IAEO,kBAAkB,CAAC,MAAsC;QAC/D,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC;QAEzB,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACvB,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACrC,CAAC;aAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACrC,CAAC;aAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAChC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QACtC,CAAC;QAED,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,WAAW,EAAE;YAC1B,GAAG,EAAE,GAAG,CAAC,WAAW,EAAE;SACvB,CAAC;IACJ,CAAC;CACF;AAED,qBAAqB;AACrB,IAAI,YAAY,GAAuB,IAAI,CAAC;AAE5C,MAAM,UAAU,cAAc;IAC5B,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,IAAI,WAAW,EAAE,CAAC;IACnC,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC"}
@@ -13,6 +13,9 @@ export declare class TokenTracker {
13
13
  private rebuildStatsFromLog;
14
14
  getSessionStats(): SessionTokenStats;
15
15
  getAllStats(): Promise<AggregateTokenStats>;
16
+ private getAllStatsViaTokscale;
17
+ private getAgentDataFromLocalLog;
18
+ private getAllStatsLegacy;
16
19
  getTopAgentsAllSessions(limit?: number): Promise<Array<{
17
20
  agent: string;
18
21
  tokens: number;
@@ -1 +1 @@
1
- {"version":3,"file":"token-tracker.d.ts","sourceRoot":"","sources":["../../src/analytics/token-tracker.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAQhF,qBAAa,YAAY;IACvB,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,YAAY,CAAoB;gBAE5B,SAAS,CAAC,EAAE,MAAM;IAK9B,OAAO,CAAC,iBAAiB;IAIzB,OAAO,CAAC,sBAAsB;IAexB,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;YAiB3E,WAAW;IAOzB,OAAO,CAAC,kBAAkB;YAqBZ,gBAAgB;IAIxB,gBAAgB,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;YAe/D,mBAAmB;IAoBjC,eAAe,IAAI,iBAAiB;IAI9B,WAAW,IAAI,OAAO,CAAC,mBAAmB,CAAC;IAkF3C,uBAAuB,CAAC,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAY3G,YAAY,CAAC,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAsBhG,cAAc,CAAC,aAAa,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,CAAC;CA2BlE;AAKD,wBAAgB,eAAe,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,YAAY,CAKhE;AAED,wBAAgB,iBAAiB,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,YAAY,CAGlE"}
1
+ {"version":3,"file":"token-tracker.d.ts","sourceRoot":"","sources":["../../src/analytics/token-tracker.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAShF,qBAAa,YAAY;IACvB,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,YAAY,CAAoB;gBAE5B,SAAS,CAAC,EAAE,MAAM;IAK9B,OAAO,CAAC,iBAAiB;IAIzB,OAAO,CAAC,sBAAsB;IAexB,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;YAiB3E,WAAW;IAOzB,OAAO,CAAC,kBAAkB;YAqBZ,gBAAgB;IAIxB,gBAAgB,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;YAe/D,mBAAmB;IAoBjC,eAAe,IAAI,iBAAiB;IAI9B,WAAW,IAAI,OAAO,CAAC,mBAAmB,CAAC;YAWnC,sBAAsB;YA8BtB,wBAAwB;YAoExB,iBAAiB;IAkFzB,uBAAuB,CAAC,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAY3G,YAAY,CAAC,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAsBhG,cAAc,CAAC,aAAa,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,CAAC;CA2BlE;AAKD,wBAAgB,eAAe,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,YAAY,CAKhE;AAED,wBAAgB,iBAAiB,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,YAAY,CAGlE"}
@@ -1,4 +1,5 @@
1
1
  import { readState, writeState, StateLocation } from '../features/state-manager/index.js';
2
+ import { getTokscaleAdapter } from './tokscale-adapter.js';
2
3
  import * as fs from 'fs/promises';
3
4
  import * as path from 'path';
4
5
  import { homedir } from 'os';
@@ -100,6 +101,94 @@ export class TokenTracker {
100
101
  return { ...this.sessionStats };
101
102
  }
102
103
  async getAllStats() {
104
+ const adapter = await getTokscaleAdapter();
105
+ if (adapter.isAvailable && adapter.getReport) {
106
+ return this.getAllStatsViaTokscale(adapter);
107
+ }
108
+ // Fallback to existing implementation
109
+ return this.getAllStatsLegacy();
110
+ }
111
+ async getAllStatsViaTokscale(adapter) {
112
+ try {
113
+ // Get tokscale report using the new unified API
114
+ const report = await adapter.getReport();
115
+ // Get agent data from local JSONL (populated by backfill)
116
+ const agentData = await this.getAgentDataFromLocalLog();
117
+ // Map tokscale report to our AggregateTokenStats format
118
+ // Merge tokscale model-level data with local agent-level data
119
+ return {
120
+ totalInputTokens: report.totalInputTokens,
121
+ totalOutputTokens: report.totalOutputTokens,
122
+ totalCacheCreation: report.totalCacheCreationTokens,
123
+ totalCacheRead: report.totalCacheReadTokens,
124
+ totalCost: report.totalCost,
125
+ byAgent: agentData.byAgent, // From local JSONL
126
+ byModel: report.byModel || {},
127
+ sessionCount: agentData.sessionCount || 1,
128
+ entryCount: report.totalEntries,
129
+ firstEntry: agentData.firstEntry,
130
+ lastEntry: agentData.lastEntry
131
+ };
132
+ }
133
+ catch (error) {
134
+ // If tokscale fails, fall back to legacy implementation
135
+ return this.getAllStatsLegacy();
136
+ }
137
+ }
138
+ // Hybrid data merging: Read agent attribution from local JSONL
139
+ async getAgentDataFromLocalLog() {
140
+ const { calculateCost } = await import('./cost-estimator.js');
141
+ const result = {
142
+ byAgent: {},
143
+ sessionCount: 0,
144
+ entryCount: 0,
145
+ firstEntry: null,
146
+ lastEntry: null
147
+ };
148
+ try {
149
+ const content = await fs.readFile(TOKEN_LOG_FILE, 'utf-8');
150
+ const lines = content.trim().split('\n').filter(line => line.trim());
151
+ if (lines.length === 0) {
152
+ return result;
153
+ }
154
+ const sessions = new Set();
155
+ for (const line of lines) {
156
+ const record = JSON.parse(line);
157
+ result.entryCount++;
158
+ // Track unique sessions
159
+ sessions.add(record.sessionId);
160
+ // Track timestamps
161
+ if (!result.firstEntry || record.timestamp < result.firstEntry) {
162
+ result.firstEntry = record.timestamp;
163
+ }
164
+ if (!result.lastEntry || record.timestamp > result.lastEntry) {
165
+ result.lastEntry = record.timestamp;
166
+ }
167
+ // Calculate cost for this record
168
+ const cost = calculateCost({
169
+ modelName: record.modelName,
170
+ inputTokens: record.inputTokens,
171
+ outputTokens: record.outputTokens,
172
+ cacheCreationTokens: record.cacheCreationTokens,
173
+ cacheReadTokens: record.cacheReadTokens
174
+ });
175
+ // Aggregate by agent (use "(main session)" for entries without agentName)
176
+ const agentKey = record.agentName || '(main session)';
177
+ if (!result.byAgent[agentKey]) {
178
+ result.byAgent[agentKey] = { tokens: 0, cost: 0 };
179
+ }
180
+ result.byAgent[agentKey].tokens += record.inputTokens + record.outputTokens;
181
+ result.byAgent[agentKey].cost += cost.totalCost;
182
+ }
183
+ result.sessionCount = sessions.size;
184
+ return result;
185
+ }
186
+ catch (error) {
187
+ // If file doesn't exist or is empty, return empty result
188
+ return result;
189
+ }
190
+ }
191
+ async getAllStatsLegacy() {
103
192
  const { calculateCost } = await import('./cost-estimator.js');
104
193
  const stats = {
105
194
  totalInputTokens: 0,