@nanocollective/nanocoder 1.15.1 → 1.16.0
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.
- package/README.md +8 -1
- package/dist/ai-sdk-client.d.ts +5 -0
- package/dist/ai-sdk-client.d.ts.map +1 -1
- package/dist/ai-sdk-client.js +21 -3
- package/dist/ai-sdk-client.js.map +1 -1
- package/dist/app/utils/appUtils.d.ts.map +1 -1
- package/dist/app/utils/appUtils.js +5 -3
- package/dist/app/utils/appUtils.js.map +1 -1
- package/dist/commands/index.d.ts +1 -0
- package/dist/commands/index.d.ts.map +1 -1
- package/dist/commands/index.js +1 -0
- package/dist/commands/index.js.map +1 -1
- package/dist/commands/usage.d.ts +7 -0
- package/dist/commands/usage.d.ts.map +1 -0
- package/dist/commands/usage.js +69 -0
- package/dist/commands/usage.js.map +1 -0
- package/dist/commands.d.ts +3 -2
- package/dist/commands.d.ts.map +1 -1
- package/dist/commands.js.map +1 -1
- package/dist/components/usage/progress-bar.d.ts +14 -0
- package/dist/components/usage/progress-bar.d.ts.map +1 -0
- package/dist/components/usage/progress-bar.js +14 -0
- package/dist/components/usage/progress-bar.js.map +1 -0
- package/dist/components/usage/usage-display.d.ts +18 -0
- package/dist/components/usage/usage-display.d.ts.map +1 -0
- package/dist/components/usage/usage-display.js +50 -0
- package/dist/components/usage/usage-display.js.map +1 -0
- package/dist/config/paths.d.ts +0 -4
- package/dist/config/paths.d.ts.map +1 -1
- package/dist/config/paths.js +2 -18
- package/dist/config/paths.js.map +1 -1
- package/dist/hooks/useAppInitialization.d.ts.map +1 -1
- package/dist/hooks/useAppInitialization.js +2 -1
- package/dist/hooks/useAppInitialization.js.map +1 -1
- package/dist/hooks/useAppState.d.ts +2 -0
- package/dist/hooks/useAppState.d.ts.map +1 -1
- package/dist/hooks/useAppState.js +27 -5
- package/dist/hooks/useAppState.js.map +1 -1
- package/dist/hooks/useChatHandler.d.ts.map +1 -1
- package/dist/hooks/useChatHandler.js +45 -9
- package/dist/hooks/useChatHandler.js.map +1 -1
- package/dist/models/index.d.ts +2 -0
- package/dist/models/index.d.ts.map +1 -0
- package/dist/models/index.js +2 -0
- package/dist/models/index.js.map +1 -0
- package/dist/models/models-cache.d.ts +8 -0
- package/dist/models/models-cache.d.ts.map +1 -0
- package/dist/models/models-cache.js +63 -0
- package/dist/models/models-cache.js.map +1 -0
- package/dist/models/models-dev-client.d.ts +10 -0
- package/dist/models/models-dev-client.d.ts.map +1 -0
- package/dist/models/models-dev-client.js +268 -0
- package/dist/models/models-dev-client.js.map +1 -0
- package/dist/models/models-types.d.ts +66 -0
- package/dist/models/models-types.d.ts.map +1 -0
- package/dist/models/models-types.js +5 -0
- package/dist/models/models-types.js.map +1 -0
- package/dist/recommendations/model-database.d.ts.map +1 -1
- package/dist/recommendations/model-database.js +45 -0
- package/dist/recommendations/model-database.js.map +1 -1
- package/dist/tokenization/index.d.ts +2 -0
- package/dist/tokenization/index.d.ts.map +1 -0
- package/dist/tokenization/index.js +2 -0
- package/dist/tokenization/index.js.map +1 -0
- package/dist/tokenization/tokenizer-factory.d.ts +14 -0
- package/dist/tokenization/tokenizer-factory.d.ts.map +1 -0
- package/dist/tokenization/tokenizer-factory.js +90 -0
- package/dist/tokenization/tokenizer-factory.js.map +1 -0
- package/dist/tokenization/tokenizer-factory.spec.d.ts +5 -0
- package/dist/tokenization/tokenizer-factory.spec.d.ts.map +1 -0
- package/dist/tokenization/tokenizer-factory.spec.js +137 -0
- package/dist/tokenization/tokenizer-factory.spec.js.map +1 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.d.ts +17 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.js +35 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.js.map +1 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.spec.d.ts +5 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.spec.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.spec.js +152 -0
- package/dist/tokenization/tokenizers/anthropic-tokenizer.spec.js.map +1 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.d.ts +13 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.js +20 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.js.map +1 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.spec.d.ts +5 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.spec.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.spec.js +183 -0
- package/dist/tokenization/tokenizers/fallback-tokenizer.spec.js.map +1 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.d.ts +14 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.js +33 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.js.map +1 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.spec.d.ts +5 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.spec.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.spec.js +170 -0
- package/dist/tokenization/tokenizers/llama-tokenizer.spec.js.map +1 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.d.ts +22 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.js +48 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.js.map +1 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.spec.d.ts +5 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.spec.d.ts.map +1 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.spec.js +140 -0
- package/dist/tokenization/tokenizers/openai-tokenizer.spec.js.map +1 -0
- package/dist/types/commands.d.ts +1 -0
- package/dist/types/commands.d.ts.map +1 -1
- package/dist/types/tokenization.d.ts +31 -0
- package/dist/types/tokenization.d.ts.map +1 -0
- package/dist/types/tokenization.js +5 -0
- package/dist/types/tokenization.js.map +1 -0
- package/dist/types/usage.d.ts +55 -0
- package/dist/types/usage.d.ts.map +1 -0
- package/dist/types/usage.js +2 -0
- package/dist/types/usage.js.map +1 -0
- package/dist/usage/calculator.d.ts +28 -0
- package/dist/usage/calculator.d.ts.map +1 -0
- package/dist/usage/calculator.js +81 -0
- package/dist/usage/calculator.js.map +1 -0
- package/dist/usage/calculator.spec.d.ts +2 -0
- package/dist/usage/calculator.spec.d.ts.map +1 -0
- package/dist/usage/calculator.spec.js +303 -0
- package/dist/usage/calculator.spec.js.map +1 -0
- package/dist/usage/storage.d.ts +19 -0
- package/dist/usage/storage.d.ts.map +1 -0
- package/dist/usage/storage.js +134 -0
- package/dist/usage/storage.js.map +1 -0
- package/dist/usage/storage.spec.d.ts +2 -0
- package/dist/usage/storage.spec.d.ts.map +1 -0
- package/dist/usage/storage.spec.js +417 -0
- package/dist/usage/storage.spec.js.map +1 -0
- package/dist/usage/tracker.d.ts +28 -0
- package/dist/usage/tracker.d.ts.map +1 -0
- package/dist/usage/tracker.js +75 -0
- package/dist/usage/tracker.js.map +1 -0
- package/dist/usage/tracker.spec.d.ts +2 -0
- package/dist/usage/tracker.spec.d.ts.map +1 -0
- package/dist/usage/tracker.spec.js +347 -0
- package/dist/usage/tracker.spec.js.map +1 -0
- package/dist/utils/paste-roundtrip.spec.d.ts.map +1 -0
- package/dist/utils/paste-roundtrip.spec.js.map +1 -0
- package/package.json +7 -2
- package/dist/integration/paste-roundtrip.spec.d.ts.map +0 -1
- package/dist/integration/paste-roundtrip.spec.js.map +0 -1
- /package/dist/{integration → utils}/paste-roundtrip.spec.d.ts +0 -0
- /package/dist/{integration → utils}/paste-roundtrip.spec.js +0 -0
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Usage calculator
|
|
3
|
+
* Calculates token breakdown by category
|
|
4
|
+
*/
|
|
5
|
+
/**
|
|
6
|
+
* Calculate token breakdown from messages
|
|
7
|
+
* @param messages - Messages to calculate breakdown for
|
|
8
|
+
* @param tokenizer - Tokenizer instance (used as fallback if getTokens not provided)
|
|
9
|
+
* @param getTokens - Optional cached token counting function for performance
|
|
10
|
+
*/
|
|
11
|
+
export function calculateTokenBreakdown(messages, tokenizer, getTokens) {
|
|
12
|
+
const breakdown = {
|
|
13
|
+
system: 0,
|
|
14
|
+
userMessages: 0,
|
|
15
|
+
assistantMessages: 0,
|
|
16
|
+
toolDefinitions: 0,
|
|
17
|
+
toolResults: 0,
|
|
18
|
+
total: 0,
|
|
19
|
+
};
|
|
20
|
+
for (const message of messages) {
|
|
21
|
+
const tokens = getTokens
|
|
22
|
+
? getTokens(message)
|
|
23
|
+
: tokenizer.countTokens(message);
|
|
24
|
+
switch (message.role) {
|
|
25
|
+
case 'system':
|
|
26
|
+
breakdown.system += tokens;
|
|
27
|
+
break;
|
|
28
|
+
case 'user':
|
|
29
|
+
breakdown.userMessages += tokens;
|
|
30
|
+
break;
|
|
31
|
+
case 'assistant':
|
|
32
|
+
breakdown.assistantMessages += tokens;
|
|
33
|
+
break;
|
|
34
|
+
case 'tool':
|
|
35
|
+
breakdown.toolResults += tokens;
|
|
36
|
+
break;
|
|
37
|
+
default:
|
|
38
|
+
// Unknown roles go to assistant messages
|
|
39
|
+
breakdown.assistantMessages += tokens;
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
// Calculate total
|
|
43
|
+
breakdown.total =
|
|
44
|
+
breakdown.system +
|
|
45
|
+
breakdown.userMessages +
|
|
46
|
+
breakdown.assistantMessages +
|
|
47
|
+
breakdown.toolDefinitions +
|
|
48
|
+
breakdown.toolResults;
|
|
49
|
+
return breakdown;
|
|
50
|
+
}
|
|
51
|
+
/**
|
|
52
|
+
* Calculate tool definitions token count
|
|
53
|
+
* This estimates the tokens used by tool definitions sent to the model
|
|
54
|
+
*/
|
|
55
|
+
export function calculateToolDefinitionsTokens(toolCount) {
|
|
56
|
+
// Rough estimate: each tool definition is about 150 tokens
|
|
57
|
+
// This includes name, description, parameters schema
|
|
58
|
+
const TOKENS_PER_TOOL = 150;
|
|
59
|
+
return toolCount * TOKENS_PER_TOOL;
|
|
60
|
+
}
|
|
61
|
+
/**
|
|
62
|
+
* Get status color based on percentage used
|
|
63
|
+
*/
|
|
64
|
+
export function getUsageStatusColor(percentUsed) {
|
|
65
|
+
if (percentUsed < 70) {
|
|
66
|
+
return 'success';
|
|
67
|
+
}
|
|
68
|
+
else if (percentUsed < 90) {
|
|
69
|
+
return 'warning';
|
|
70
|
+
}
|
|
71
|
+
else {
|
|
72
|
+
return 'error';
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
/**
|
|
76
|
+
* Format token count with thousands separator
|
|
77
|
+
*/
|
|
78
|
+
export function formatTokenCount(tokens) {
|
|
79
|
+
return tokens.toLocaleString();
|
|
80
|
+
}
|
|
81
|
+
//# sourceMappingURL=calculator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"calculator.js","sourceRoot":"","sources":["../../source/usage/calculator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAMH;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CACtC,QAAmB,EACnB,SAAoB,EACpB,SAAwC;IAExC,MAAM,SAAS,GAAmB;QACjC,MAAM,EAAE,CAAC;QACT,YAAY,EAAE,CAAC;QACf,iBAAiB,EAAE,CAAC;QACpB,eAAe,EAAE,CAAC;QAClB,WAAW,EAAE,CAAC;QACd,KAAK,EAAE,CAAC;KACR,CAAC;IAEF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,SAAS;YACvB,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC;YACpB,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAElC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACtB,KAAK,QAAQ;gBACZ,SAAS,CAAC,MAAM,IAAI,MAAM,CAAC;gBAC3B,MAAM;YAEP,KAAK,MAAM;gBACV,SAAS,CAAC,YAAY,IAAI,MAAM,CAAC;gBACjC,MAAM;YAEP,KAAK,WAAW;gBACf,SAAS,CAAC,iBAAiB,IAAI,MAAM,CAAC;gBACtC,MAAM;YAEP,KAAK,MAAM;gBACV,SAAS,CAAC,WAAW,IAAI,MAAM,CAAC;gBAChC,MAAM;YAEP;gBACC,yCAAyC;gBACzC,SAAS,CAAC,iBAAiB,IAAI,MAAM,CAAC;QACxC,CAAC;IACF,CAAC;IAED,kBAAkB;IAClB,SAAS,CAAC,KAAK;QACd,SAAS,CAAC,MAAM;YAChB,SAAS,CAAC,YAAY;YACtB,SAAS,CAAC,iBAAiB;YAC3B,SAAS,CAAC,eAAe;YACzB,SAAS,CAAC,WAAW,CAAC;IAEvB,OAAO,SAAS,CAAC;AAClB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,8BAA8B,CAAC,SAAiB;IAC/D,2DAA2D;IAC3D,qDAAqD;IACrD,MAAM,eAAe,GAAG,GAAG,CAAC;IAE5B,OAAO,SAAS,GAAG,eAAe,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAClC,WAAmB;IAEnB,IAAI,WAAW,GAAG,EAAE,EAAE,CAAC;QACtB,OAAO,SAAS,CAAC;IAClB,CAAC;SAAM,IAAI,WAAW,GAAG,EAAE,EAAE,CAAC;QAC7B,OAAO,SAAS,CAAC;IAClB,CAAC;SAAM,CAAC;QACP,OAAO,OAAO,CAAC;IAChB,CAAC;AACF,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,MAAc;IAC9C,OAAO,MAAM,CAAC,cAAc,EAAE,CAAC;AAChC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"calculator.spec.d.ts","sourceRoot":"","sources":["../../source/usage/calculator.spec.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,303 @@
|
|
|
1
|
+
import test from 'ava';
|
|
2
|
+
import { calculateTokenBreakdown, calculateToolDefinitionsTokens, getUsageStatusColor, formatTokenCount, } from './calculator.js';
|
|
3
|
+
console.log('\ncalculator.spec.ts');
|
|
4
|
+
// ============================================================================
|
|
5
|
+
// Mock Tokenizer
|
|
6
|
+
// ============================================================================
|
|
7
|
+
class MockTokenizer {
|
|
8
|
+
getName() {
|
|
9
|
+
return 'mock-tokenizer';
|
|
10
|
+
}
|
|
11
|
+
countTokens(message) {
|
|
12
|
+
// Simple mock: 1 token per 4 characters
|
|
13
|
+
return Math.ceil(message.content.length / 4);
|
|
14
|
+
}
|
|
15
|
+
encode(text) {
|
|
16
|
+
// Return token count (1 token per 4 characters)
|
|
17
|
+
return Math.ceil(text.length / 4);
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
// ============================================================================
|
|
21
|
+
// calculateTokenBreakdown Tests
|
|
22
|
+
// ============================================================================
|
|
23
|
+
test('calculateTokenBreakdown returns empty breakdown for no messages', t => {
|
|
24
|
+
const messages = [];
|
|
25
|
+
const tokenizer = new MockTokenizer();
|
|
26
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
27
|
+
t.is(breakdown.system, 0);
|
|
28
|
+
t.is(breakdown.userMessages, 0);
|
|
29
|
+
t.is(breakdown.assistantMessages, 0);
|
|
30
|
+
t.is(breakdown.toolResults, 0);
|
|
31
|
+
t.is(breakdown.toolDefinitions, 0);
|
|
32
|
+
t.is(breakdown.total, 0);
|
|
33
|
+
});
|
|
34
|
+
test('calculateTokenBreakdown counts system messages', t => {
|
|
35
|
+
const messages = [
|
|
36
|
+
{ role: 'system', content: 'You are a helpful assistant.' }, // 28 chars / 4 = 7 tokens
|
|
37
|
+
];
|
|
38
|
+
const tokenizer = new MockTokenizer();
|
|
39
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
40
|
+
t.is(breakdown.system, 7);
|
|
41
|
+
t.is(breakdown.userMessages, 0);
|
|
42
|
+
t.is(breakdown.assistantMessages, 0);
|
|
43
|
+
t.is(breakdown.toolResults, 0);
|
|
44
|
+
t.is(breakdown.total, 7);
|
|
45
|
+
});
|
|
46
|
+
test('calculateTokenBreakdown counts user messages', t => {
|
|
47
|
+
const messages = [
|
|
48
|
+
{ role: 'user', content: 'Hello' }, // 5 chars / 4 = 2 tokens
|
|
49
|
+
{ role: 'user', content: 'How are you?' }, // 12 chars / 4 = 3 tokens
|
|
50
|
+
];
|
|
51
|
+
const tokenizer = new MockTokenizer();
|
|
52
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
53
|
+
t.is(breakdown.userMessages, 5); // 2 + 3
|
|
54
|
+
t.is(breakdown.system, 0);
|
|
55
|
+
t.is(breakdown.assistantMessages, 0);
|
|
56
|
+
t.is(breakdown.total, 5);
|
|
57
|
+
});
|
|
58
|
+
test('calculateTokenBreakdown counts assistant messages', t => {
|
|
59
|
+
const messages = [
|
|
60
|
+
{ role: 'assistant', content: 'I am well.' }, // 10 chars / 4 = 3 tokens
|
|
61
|
+
{ role: 'assistant', content: 'Thank you!' }, // 10 chars / 4 = 3 tokens
|
|
62
|
+
];
|
|
63
|
+
const tokenizer = new MockTokenizer();
|
|
64
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
65
|
+
t.is(breakdown.assistantMessages, 6); // 3 + 3
|
|
66
|
+
t.is(breakdown.total, 6);
|
|
67
|
+
});
|
|
68
|
+
test('calculateTokenBreakdown counts tool result messages', t => {
|
|
69
|
+
const messages = [
|
|
70
|
+
{ role: 'tool', content: 'File contents here', tool_call_id: 'call_123' },
|
|
71
|
+
];
|
|
72
|
+
const tokenizer = new MockTokenizer();
|
|
73
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
74
|
+
t.is(breakdown.toolResults, 5); // 18 chars / 4 = 5 tokens
|
|
75
|
+
t.is(breakdown.total, 5);
|
|
76
|
+
});
|
|
77
|
+
test('calculateTokenBreakdown counts mixed message types', t => {
|
|
78
|
+
const messages = [
|
|
79
|
+
{ role: 'system', content: 'abcd' }, // 4 chars = 1 token
|
|
80
|
+
{ role: 'user', content: 'abcdefgh' }, // 8 chars = 2 tokens
|
|
81
|
+
{ role: 'assistant', content: 'abcdefghijkl' }, // 12 chars = 3 tokens
|
|
82
|
+
{ role: 'tool', content: 'abcdefghijklmnop', tool_call_id: 'call_1' }, // 16 chars = 4 tokens
|
|
83
|
+
];
|
|
84
|
+
const tokenizer = new MockTokenizer();
|
|
85
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
86
|
+
t.is(breakdown.system, 1);
|
|
87
|
+
t.is(breakdown.userMessages, 2);
|
|
88
|
+
t.is(breakdown.assistantMessages, 3);
|
|
89
|
+
t.is(breakdown.toolResults, 4);
|
|
90
|
+
t.is(breakdown.total, 10); // 1 + 2 + 3 + 4
|
|
91
|
+
});
|
|
92
|
+
test('calculateTokenBreakdown uses custom getTokens function when provided', t => {
|
|
93
|
+
const messages = [
|
|
94
|
+
{ role: 'user', content: 'test' },
|
|
95
|
+
{ role: 'assistant', content: 'response' },
|
|
96
|
+
];
|
|
97
|
+
const tokenizer = new MockTokenizer();
|
|
98
|
+
let callCount = 0;
|
|
99
|
+
const customGetTokens = (_message) => {
|
|
100
|
+
callCount++;
|
|
101
|
+
return 10; // Fixed token count
|
|
102
|
+
};
|
|
103
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer, customGetTokens);
|
|
104
|
+
// Should have called custom function for each message
|
|
105
|
+
t.is(callCount, 2);
|
|
106
|
+
t.is(breakdown.userMessages, 10);
|
|
107
|
+
t.is(breakdown.assistantMessages, 10);
|
|
108
|
+
t.is(breakdown.total, 20);
|
|
109
|
+
});
|
|
110
|
+
test('calculateTokenBreakdown handles empty message content', t => {
|
|
111
|
+
const messages = [
|
|
112
|
+
{ role: 'user', content: '' },
|
|
113
|
+
{ role: 'assistant', content: '' },
|
|
114
|
+
];
|
|
115
|
+
const tokenizer = new MockTokenizer();
|
|
116
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
117
|
+
t.is(breakdown.userMessages, 0);
|
|
118
|
+
t.is(breakdown.assistantMessages, 0);
|
|
119
|
+
t.is(breakdown.total, 0);
|
|
120
|
+
});
|
|
121
|
+
test('calculateTokenBreakdown handles unknown role as assistant', t => {
|
|
122
|
+
const messages = [
|
|
123
|
+
{ role: 'unknown', content: 'abcdefgh' }, // 8 chars = 2 tokens
|
|
124
|
+
];
|
|
125
|
+
const tokenizer = new MockTokenizer();
|
|
126
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
127
|
+
// Unknown roles should be counted as assistant messages
|
|
128
|
+
t.is(breakdown.assistantMessages, 2);
|
|
129
|
+
t.is(breakdown.total, 2);
|
|
130
|
+
});
|
|
131
|
+
test('calculateTokenBreakdown calculates correct total', t => {
|
|
132
|
+
const messages = [
|
|
133
|
+
{ role: 'system', content: 'a'.repeat(100) }, // 25 tokens
|
|
134
|
+
{ role: 'user', content: 'b'.repeat(80) }, // 20 tokens
|
|
135
|
+
{ role: 'assistant', content: 'c'.repeat(40) }, // 10 tokens
|
|
136
|
+
{ role: 'tool', content: 'd'.repeat(60), tool_call_id: 'call_1' }, // 15 tokens
|
|
137
|
+
];
|
|
138
|
+
const tokenizer = new MockTokenizer();
|
|
139
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
140
|
+
t.is(breakdown.total, 70); // 25 + 20 + 10 + 15
|
|
141
|
+
});
|
|
142
|
+
// ============================================================================
|
|
143
|
+
// calculateToolDefinitionsTokens Tests
|
|
144
|
+
// ============================================================================
|
|
145
|
+
test('calculateToolDefinitionsTokens returns 0 for no tools', t => {
|
|
146
|
+
const tokens = calculateToolDefinitionsTokens(0);
|
|
147
|
+
t.is(tokens, 0);
|
|
148
|
+
});
|
|
149
|
+
test('calculateToolDefinitionsTokens returns 150 for one tool', t => {
|
|
150
|
+
const tokens = calculateToolDefinitionsTokens(1);
|
|
151
|
+
t.is(tokens, 150);
|
|
152
|
+
});
|
|
153
|
+
test('calculateToolDefinitionsTokens returns 300 for two tools', t => {
|
|
154
|
+
const tokens = calculateToolDefinitionsTokens(2);
|
|
155
|
+
t.is(tokens, 300);
|
|
156
|
+
});
|
|
157
|
+
test('calculateToolDefinitionsTokens scales linearly', t => {
|
|
158
|
+
const tokens5 = calculateToolDefinitionsTokens(5);
|
|
159
|
+
const tokens10 = calculateToolDefinitionsTokens(10);
|
|
160
|
+
t.is(tokens5, 750);
|
|
161
|
+
t.is(tokens10, 1500);
|
|
162
|
+
t.is(tokens10, tokens5 * 2);
|
|
163
|
+
});
|
|
164
|
+
test('calculateToolDefinitionsTokens handles large number of tools', t => {
|
|
165
|
+
const tokens = calculateToolDefinitionsTokens(100);
|
|
166
|
+
t.is(tokens, 15000);
|
|
167
|
+
});
|
|
168
|
+
// ============================================================================
|
|
169
|
+
// getUsageStatusColor Tests
|
|
170
|
+
// ============================================================================
|
|
171
|
+
test('getUsageStatusColor returns success for low usage', t => {
|
|
172
|
+
t.is(getUsageStatusColor(0), 'success');
|
|
173
|
+
t.is(getUsageStatusColor(10), 'success');
|
|
174
|
+
t.is(getUsageStatusColor(50), 'success');
|
|
175
|
+
t.is(getUsageStatusColor(69), 'success');
|
|
176
|
+
});
|
|
177
|
+
test('getUsageStatusColor returns warning for moderate usage', t => {
|
|
178
|
+
t.is(getUsageStatusColor(70), 'warning');
|
|
179
|
+
t.is(getUsageStatusColor(75), 'warning');
|
|
180
|
+
t.is(getUsageStatusColor(85), 'warning');
|
|
181
|
+
t.is(getUsageStatusColor(89), 'warning');
|
|
182
|
+
});
|
|
183
|
+
test('getUsageStatusColor returns error for high usage', t => {
|
|
184
|
+
t.is(getUsageStatusColor(90), 'error');
|
|
185
|
+
t.is(getUsageStatusColor(95), 'error');
|
|
186
|
+
t.is(getUsageStatusColor(100), 'error');
|
|
187
|
+
t.is(getUsageStatusColor(110), 'error');
|
|
188
|
+
});
|
|
189
|
+
test('getUsageStatusColor handles edge cases', t => {
|
|
190
|
+
t.is(getUsageStatusColor(69.9), 'success');
|
|
191
|
+
t.is(getUsageStatusColor(70.0), 'warning');
|
|
192
|
+
t.is(getUsageStatusColor(89.9), 'warning');
|
|
193
|
+
t.is(getUsageStatusColor(90.0), 'error');
|
|
194
|
+
});
|
|
195
|
+
test('getUsageStatusColor handles negative percentages', t => {
|
|
196
|
+
t.is(getUsageStatusColor(-10), 'success');
|
|
197
|
+
t.is(getUsageStatusColor(-1), 'success');
|
|
198
|
+
});
|
|
199
|
+
test('getUsageStatusColor handles very large percentages', t => {
|
|
200
|
+
t.is(getUsageStatusColor(200), 'error');
|
|
201
|
+
t.is(getUsageStatusColor(1000), 'error');
|
|
202
|
+
});
|
|
203
|
+
// ============================================================================
|
|
204
|
+
// formatTokenCount Tests
|
|
205
|
+
// ============================================================================
|
|
206
|
+
test('formatTokenCount formats small numbers', t => {
|
|
207
|
+
t.is(formatTokenCount(0), '0');
|
|
208
|
+
t.is(formatTokenCount(1), '1');
|
|
209
|
+
t.is(formatTokenCount(10), '10');
|
|
210
|
+
t.is(formatTokenCount(100), '100');
|
|
211
|
+
t.is(formatTokenCount(999), '999');
|
|
212
|
+
});
|
|
213
|
+
test('formatTokenCount formats thousands with comma', t => {
|
|
214
|
+
t.is(formatTokenCount(1000), '1,000');
|
|
215
|
+
t.is(formatTokenCount(1234), '1,234');
|
|
216
|
+
t.is(formatTokenCount(9999), '9,999');
|
|
217
|
+
});
|
|
218
|
+
test('formatTokenCount formats millions with commas', t => {
|
|
219
|
+
t.is(formatTokenCount(1000000), '1,000,000');
|
|
220
|
+
t.is(formatTokenCount(1234567), '1,234,567');
|
|
221
|
+
});
|
|
222
|
+
test('formatTokenCount formats large numbers', t => {
|
|
223
|
+
t.is(formatTokenCount(100000), '100,000');
|
|
224
|
+
t.is(formatTokenCount(200000), '200,000');
|
|
225
|
+
t.is(formatTokenCount(8192000), '8,192,000');
|
|
226
|
+
});
|
|
227
|
+
test('formatTokenCount handles edge cases', t => {
|
|
228
|
+
t.is(formatTokenCount(1001), '1,001');
|
|
229
|
+
t.is(formatTokenCount(10001), '10,001');
|
|
230
|
+
t.is(formatTokenCount(100001), '100,001');
|
|
231
|
+
});
|
|
232
|
+
// ============================================================================
|
|
233
|
+
// Integration Tests
|
|
234
|
+
// ============================================================================
|
|
235
|
+
test('calculateTokenBreakdown with realistic conversation', t => {
|
|
236
|
+
const messages = [
|
|
237
|
+
{
|
|
238
|
+
role: 'system',
|
|
239
|
+
content: 'You are a helpful coding assistant. Help users with their code.',
|
|
240
|
+
},
|
|
241
|
+
{ role: 'user', content: 'How do I create a React component?' },
|
|
242
|
+
{
|
|
243
|
+
role: 'assistant',
|
|
244
|
+
content: 'To create a React component, you can use function components. Here is an example...',
|
|
245
|
+
},
|
|
246
|
+
{ role: 'user', content: 'Can you show me a more complex example?' },
|
|
247
|
+
{
|
|
248
|
+
role: 'assistant',
|
|
249
|
+
content: '',
|
|
250
|
+
tool_calls: [
|
|
251
|
+
{ id: 'call_1', function: { name: 'search_files', arguments: {} } },
|
|
252
|
+
],
|
|
253
|
+
},
|
|
254
|
+
{
|
|
255
|
+
role: 'tool',
|
|
256
|
+
content: 'Found 10 React component files...',
|
|
257
|
+
tool_call_id: 'call_1',
|
|
258
|
+
},
|
|
259
|
+
{
|
|
260
|
+
role: 'assistant',
|
|
261
|
+
content: 'Based on the search results, here is a more complex example...',
|
|
262
|
+
},
|
|
263
|
+
];
|
|
264
|
+
const tokenizer = new MockTokenizer();
|
|
265
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
266
|
+
// Verify all categories are populated
|
|
267
|
+
t.true(breakdown.system > 0);
|
|
268
|
+
t.true(breakdown.userMessages > 0);
|
|
269
|
+
t.true(breakdown.assistantMessages > 0);
|
|
270
|
+
t.true(breakdown.toolResults > 0);
|
|
271
|
+
// Verify total is sum of all categories
|
|
272
|
+
const expectedTotal = breakdown.system +
|
|
273
|
+
breakdown.userMessages +
|
|
274
|
+
breakdown.assistantMessages +
|
|
275
|
+
breakdown.toolResults +
|
|
276
|
+
breakdown.toolDefinitions;
|
|
277
|
+
t.is(breakdown.total, expectedTotal);
|
|
278
|
+
});
|
|
279
|
+
test('getUsageStatusColor with calculateTokenBreakdown', t => {
|
|
280
|
+
const messages = [
|
|
281
|
+
{ role: 'user', content: 'a'.repeat(2800) }, // 700 tokens
|
|
282
|
+
];
|
|
283
|
+
const tokenizer = new MockTokenizer();
|
|
284
|
+
const contextLimit = 1000;
|
|
285
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
286
|
+
const percentUsed = (breakdown.total / contextLimit) * 100;
|
|
287
|
+
const color = getUsageStatusColor(percentUsed);
|
|
288
|
+
t.is(breakdown.total, 700);
|
|
289
|
+
t.is(percentUsed, 70);
|
|
290
|
+
t.is(color, 'warning');
|
|
291
|
+
});
|
|
292
|
+
test('formatTokenCount with calculateTokenBreakdown result', t => {
|
|
293
|
+
const messages = [
|
|
294
|
+
{ role: 'user', content: 'a'.repeat(4000) }, // 1000 tokens
|
|
295
|
+
{ role: 'assistant', content: 'b'.repeat(8000) }, // 2000 tokens
|
|
296
|
+
];
|
|
297
|
+
const tokenizer = new MockTokenizer();
|
|
298
|
+
const breakdown = calculateTokenBreakdown(messages, tokenizer);
|
|
299
|
+
const formatted = formatTokenCount(breakdown.total);
|
|
300
|
+
t.is(breakdown.total, 3000);
|
|
301
|
+
t.is(formatted, '3,000');
|
|
302
|
+
});
|
|
303
|
+
//# sourceMappingURL=calculator.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"calculator.spec.js","sourceRoot":"","sources":["../../source/usage/calculator.spec.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,KAAK,CAAC;AACvB,OAAO,EACN,uBAAuB,EACvB,8BAA8B,EAC9B,mBAAmB,EACnB,gBAAgB,GAChB,MAAM,iBAAiB,CAAC;AAIzB,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;AAEpC,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E,MAAM,aAAa;IAClB,OAAO;QACN,OAAO,gBAAgB,CAAC;IACzB,CAAC;IAED,WAAW,CAAC,OAAgB;QAC3B,wCAAwC;QACxC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,CAAC,IAAY;QAClB,gDAAgD;QAChD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACnC,CAAC;CACD;AAED,+EAA+E;AAC/E,gCAAgC;AAChC,+EAA+E;AAE/E,IAAI,CAAC,iEAAiE,EAAE,CAAC,CAAC,EAAE;IAC3E,MAAM,QAAQ,GAAc,EAAE,CAAC;IAC/B,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC1B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gDAAgD,EAAE,CAAC,CAAC,EAAE;IAC1D,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,8BAA8B,EAAC,EAAE,0BAA0B;KACrF,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC1B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8CAA8C,EAAE,CAAC,CAAC,EAAE;IACxD,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAC,EAAE,yBAAyB;QAC3D,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAC,EAAE,0BAA0B;KACnE,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ;IACzC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC1B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mDAAmD,EAAE,CAAC,CAAC,EAAE;IAC7D,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,EAAC,EAAE,0BAA0B;QACtE,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,EAAC,EAAE,0BAA0B;KACtE,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ;IAC9C,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qDAAqD,EAAE,CAAC,CAAC,EAAE;IAC/D,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,UAAU,EAAC;KACvE,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,0BAA0B;IAC1D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oDAAoD,EAAE,CAAC,CAAC,EAAE;IAC9D,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAC,EAAE,oBAAoB;QACvD,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAC,EAAE,qBAAqB;QAC1D,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc,EAAC,EAAE,sBAAsB;QACpE,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,QAAQ,EAAC,EAAE,sBAAsB;KAC3F,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC1B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,gBAAgB;AAC5C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sEAAsE,EAAE,CAAC,CAAC,EAAE;IAChF,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAC;QAC/B,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,UAAU,EAAC;KACxC,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,MAAM,eAAe,GAAG,CAAC,QAAiB,EAAE,EAAE;QAC7C,SAAS,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC,CAAC,oBAAoB;IAChC,CAAC,CAAC;IAEF,MAAM,SAAS,GAAG,uBAAuB,CACxC,QAAQ,EACR,SAAS,EACT,eAAe,CACf,CAAC;IAEF,sDAAsD;IACtD,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;IACnB,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IACjC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;AAC3B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uDAAuD,EAAE,CAAC,CAAC,EAAE;IACjE,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAC;QAC3B,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAC;KAChC,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2DAA2D,EAAE,CAAC,CAAC,EAAE;IACrE,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,SAAgB,EAAE,OAAO,EAAE,UAAU,EAAC,EAAE,qBAAqB;KACpE,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,wDAAwD;IACxD,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kDAAkD,EAAE,CAAC,CAAC,EAAE;IAC5D,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EAAC,EAAE,YAAY;QACxD,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAC,EAAE,YAAY;QACrD,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAC,EAAE,YAAY;QAC1D,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,YAAY,EAAE,QAAQ,EAAC,EAAE,YAAY;KAC7E,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,oBAAoB;AAChD,CAAC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,uCAAuC;AACvC,+EAA+E;AAE/E,IAAI,CAAC,uDAAuD,EAAE,CAAC,CAAC,EAAE;IACjE,MAAM,MAAM,GAAG,8BAA8B,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yDAAyD,EAAE,CAAC,CAAC,EAAE;IACnE,MAAM,MAAM,GAAG,8BAA8B,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AACnB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0DAA0D,EAAE,CAAC,CAAC,EAAE;IACpE,MAAM,MAAM,GAAG,8BAA8B,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AACnB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gDAAgD,EAAE,CAAC,CAAC,EAAE;IAC1D,MAAM,OAAO,GAAG,8BAA8B,CAAC,CAAC,CAAC,CAAC;IAClD,MAAM,QAAQ,GAAG,8BAA8B,CAAC,EAAE,CAAC,CAAC;IAEpD,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACnB,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACrB,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8DAA8D,EAAE,CAAC,CAAC,EAAE;IACxE,MAAM,MAAM,GAAG,8BAA8B,CAAC,GAAG,CAAC,CAAC;IACnD,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,4BAA4B;AAC5B,+EAA+E;AAE/E,IAAI,CAAC,mDAAmD,EAAE,CAAC,CAAC,EAAE;IAC7D,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IACxC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wDAAwD,EAAE,CAAC,CAAC,EAAE;IAClE,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kDAAkD,EAAE,CAAC,CAAC,EAAE;IAC5D,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IACxC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wCAAwC,EAAE,CAAC,CAAC,EAAE;IAClD,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC,CAAC;IAC3C,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC,CAAC;IAC3C,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC,CAAC;IAC3C,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kDAAkD,EAAE,CAAC,CAAC,EAAE;IAC5D,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IAC1C,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oDAAoD,EAAE,CAAC,CAAC,EAAE;IAC9D,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IACxC,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,yBAAyB;AACzB,+EAA+E;AAE/E,IAAI,CAAC,wCAAwC,EAAE,CAAC,CAAC,EAAE;IAClD,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IACjC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;IACnC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;AACpC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+CAA+C,EAAE,CAAC,CAAC,EAAE;IACzD,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+CAA+C,EAAE,CAAC,CAAC,EAAE;IACzD,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,WAAW,CAAC,CAAC;IAC7C,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,WAAW,CAAC,CAAC;AAC9C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wCAAwC,EAAE,CAAC,CAAC,EAAE;IAClD,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;IAC1C,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;IAC1C,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,WAAW,CAAC,CAAC;AAC9C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qCAAqC,EAAE,CAAC,CAAC,EAAE;IAC/C,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;AAC3C,CAAC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,oBAAoB;AACpB,+EAA+E;AAE/E,IAAI,CAAC,qDAAqD,EAAE,CAAC,CAAC,EAAE;IAC/D,MAAM,QAAQ,GAAc;QAC3B;YACC,IAAI,EAAE,QAAQ;YACd,OAAO,EACN,iEAAiE;SAClE;QACD,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,oCAAoC,EAAC;QAC7D;YACC,IAAI,EAAE,WAAW;YACjB,OAAO,EACN,qFAAqF;SACtF;QACD,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,yCAAyC,EAAC;QAClE;YACC,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE;YACX,UAAU,EAAE;gBACX,EAAC,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAC,IAAI,EAAE,cAAc,EAAE,SAAS,EAAE,EAAE,EAAC,EAAC;aAC/D;SACD;QACD;YACC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,mCAAmC;YAC5C,YAAY,EAAE,QAAQ;SACtB;QACD;YACC,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,gEAAgE;SACzE;KACD,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE/D,sCAAsC;IACtC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC7B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,iBAAiB,GAAG,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;IAElC,wCAAwC;IACxC,MAAM,aAAa,GAClB,SAAS,CAAC,MAAM;QAChB,SAAS,CAAC,YAAY;QACtB,SAAS,CAAC,iBAAiB;QAC3B,SAAS,CAAC,WAAW;QACrB,SAAS,CAAC,eAAe,CAAC;IAE3B,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;AACtC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kDAAkD,EAAE,CAAC,CAAC,EAAE;IAC5D,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAC,EAAE,aAAa;KACxD,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IACtC,MAAM,YAAY,GAAG,IAAI,CAAC;IAE1B,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC/D,MAAM,WAAW,GAAG,CAAC,SAAS,CAAC,KAAK,GAAG,YAAY,CAAC,GAAG,GAAG,CAAC;IAC3D,MAAM,KAAK,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAE/C,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC3B,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IACtB,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;AACxB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sDAAsD,EAAE,CAAC,CAAC,EAAE;IAChE,MAAM,QAAQ,GAAc;QAC3B,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAC,EAAE,cAAc;QACzD,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAC,EAAE,cAAc;KAC9D,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC/D,MAAM,SAAS,GAAG,gBAAgB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAEpD,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AAC1B,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Usage data storage
|
|
3
|
+
* Persists usage statistics to config directory
|
|
4
|
+
*/
|
|
5
|
+
import type { UsageData, SessionUsage, DailyAggregate } from '../types/usage.js';
|
|
6
|
+
export declare function readUsageData(): UsageData;
|
|
7
|
+
export declare function writeUsageData(data: UsageData): void;
|
|
8
|
+
export declare function addSession(session: SessionUsage): void;
|
|
9
|
+
export declare function getTodayAggregate(): DailyAggregate | null;
|
|
10
|
+
export declare function getLastNDaysAggregate(days: number): {
|
|
11
|
+
totalTokens: number;
|
|
12
|
+
totalSessions: number;
|
|
13
|
+
avgTokensPerDay: number;
|
|
14
|
+
};
|
|
15
|
+
/**
|
|
16
|
+
* Clear all usage data
|
|
17
|
+
*/
|
|
18
|
+
export declare function clearUsageData(): void;
|
|
19
|
+
//# sourceMappingURL=storage.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../source/usage/storage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAKH,OAAO,KAAK,EAAC,SAAS,EAAE,YAAY,EAAE,cAAc,EAAC,MAAM,gBAAgB,CAAC;AA2B5E,wBAAgB,aAAa,IAAI,SAAS,CAgBzC;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI,CAWpD;AAED,wBAAgB,UAAU,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAkBtD;AAuCD,wBAAgB,iBAAiB,IAAI,cAAc,GAAG,IAAI,CAMzD;AAED,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG;IACpD,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;CACxB,CAyBA;AAED;;GAEG;AACH,wBAAgB,cAAc,IAAI,IAAI,CASrC"}
|
|
@@ -0,0 +1,134 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Usage data storage
|
|
3
|
+
* Persists usage statistics to config directory
|
|
4
|
+
*/
|
|
5
|
+
import * as fs from 'node:fs';
|
|
6
|
+
import * as path from 'node:path';
|
|
7
|
+
import { getAppDataPath } from '../config/paths.js';
|
|
8
|
+
const USAGE_FILE_NAME = 'usage.json';
|
|
9
|
+
const MAX_SESSIONS = 100;
|
|
10
|
+
const MAX_DAILY_AGGREGATES = 30;
|
|
11
|
+
function getUsageFilePath() {
|
|
12
|
+
const configDir = getAppDataPath();
|
|
13
|
+
return path.join(configDir, USAGE_FILE_NAME);
|
|
14
|
+
}
|
|
15
|
+
function ensureConfigDir() {
|
|
16
|
+
const configDir = getAppDataPath();
|
|
17
|
+
if (!fs.existsSync(configDir)) {
|
|
18
|
+
fs.mkdirSync(configDir, { recursive: true });
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
function createEmptyUsageData() {
|
|
22
|
+
return {
|
|
23
|
+
sessions: [],
|
|
24
|
+
dailyAggregates: [],
|
|
25
|
+
totalLifetime: 0,
|
|
26
|
+
lastUpdated: Date.now(),
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
export function readUsageData() {
|
|
30
|
+
try {
|
|
31
|
+
const filePath = getUsageFilePath();
|
|
32
|
+
if (!fs.existsSync(filePath)) {
|
|
33
|
+
return createEmptyUsageData();
|
|
34
|
+
}
|
|
35
|
+
const content = fs.readFileSync(filePath, 'utf-8');
|
|
36
|
+
const data = JSON.parse(content);
|
|
37
|
+
return data;
|
|
38
|
+
}
|
|
39
|
+
catch (error) {
|
|
40
|
+
console.warn('Failed to read usage data:', error);
|
|
41
|
+
return createEmptyUsageData();
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
export function writeUsageData(data) {
|
|
45
|
+
try {
|
|
46
|
+
ensureConfigDir();
|
|
47
|
+
data.lastUpdated = Date.now();
|
|
48
|
+
const filePath = getUsageFilePath();
|
|
49
|
+
fs.writeFileSync(filePath, JSON.stringify(data, null, 2), 'utf-8');
|
|
50
|
+
}
|
|
51
|
+
catch (error) {
|
|
52
|
+
console.warn('Failed to write usage data:', error);
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
export function addSession(session) {
|
|
56
|
+
const data = readUsageData();
|
|
57
|
+
// Add session to the beginning (most recent first)
|
|
58
|
+
data.sessions.unshift(session);
|
|
59
|
+
// Keep only last MAX_SESSIONS
|
|
60
|
+
if (data.sessions.length > MAX_SESSIONS) {
|
|
61
|
+
data.sessions = data.sessions.slice(0, MAX_SESSIONS);
|
|
62
|
+
}
|
|
63
|
+
// Update lifetime total
|
|
64
|
+
data.totalLifetime += session.tokens.total;
|
|
65
|
+
// Update daily aggregate
|
|
66
|
+
updateDailyAggregate(data, session);
|
|
67
|
+
writeUsageData(data);
|
|
68
|
+
}
|
|
69
|
+
function updateDailyAggregate(data, session) {
|
|
70
|
+
const dateParts = new Date(session.timestamp).toISOString().split('T');
|
|
71
|
+
const dateStr = dateParts[0] || new Date().toISOString().split('T')[0] || '';
|
|
72
|
+
// Find or create daily aggregate
|
|
73
|
+
let dailyAggregate = data.dailyAggregates.find(agg => agg.date === dateStr);
|
|
74
|
+
if (!dailyAggregate) {
|
|
75
|
+
dailyAggregate = {
|
|
76
|
+
date: dateStr,
|
|
77
|
+
sessions: 0,
|
|
78
|
+
totalTokens: 0,
|
|
79
|
+
providers: {},
|
|
80
|
+
models: {},
|
|
81
|
+
};
|
|
82
|
+
data.dailyAggregates.push(dailyAggregate);
|
|
83
|
+
}
|
|
84
|
+
// Update aggregate
|
|
85
|
+
dailyAggregate.sessions += 1;
|
|
86
|
+
dailyAggregate.totalTokens += session.tokens.total;
|
|
87
|
+
// Update provider stats
|
|
88
|
+
dailyAggregate.providers[session.provider] =
|
|
89
|
+
(dailyAggregate.providers[session.provider] || 0) + session.tokens.total;
|
|
90
|
+
// Update model stats
|
|
91
|
+
dailyAggregate.models[session.model] =
|
|
92
|
+
(dailyAggregate.models[session.model] || 0) + session.tokens.total;
|
|
93
|
+
// Sort by date (newest first) and keep only last MAX_DAILY_AGGREGATES
|
|
94
|
+
data.dailyAggregates.sort((a, b) => b.date.localeCompare(a.date));
|
|
95
|
+
if (data.dailyAggregates.length > MAX_DAILY_AGGREGATES) {
|
|
96
|
+
data.dailyAggregates = data.dailyAggregates.slice(0, MAX_DAILY_AGGREGATES);
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
export function getTodayAggregate() {
|
|
100
|
+
const data = readUsageData();
|
|
101
|
+
const todayParts = new Date().toISOString().split('T');
|
|
102
|
+
const today = todayParts[0] || '';
|
|
103
|
+
return data.dailyAggregates.find(agg => agg.date === today) || null;
|
|
104
|
+
}
|
|
105
|
+
export function getLastNDaysAggregate(days) {
|
|
106
|
+
const data = readUsageData();
|
|
107
|
+
const cutoffDate = new Date();
|
|
108
|
+
cutoffDate.setDate(cutoffDate.getDate() - days);
|
|
109
|
+
const cutoffParts = cutoffDate.toISOString().split('T');
|
|
110
|
+
const cutoffStr = cutoffParts[0] || '';
|
|
111
|
+
const relevantAggregates = data.dailyAggregates.filter(agg => agg.date >= cutoffStr);
|
|
112
|
+
const totalTokens = relevantAggregates.reduce((sum, agg) => sum + agg.totalTokens, 0);
|
|
113
|
+
const totalSessions = relevantAggregates.reduce((sum, agg) => sum + agg.sessions, 0);
|
|
114
|
+
return {
|
|
115
|
+
totalTokens,
|
|
116
|
+
totalSessions,
|
|
117
|
+
avgTokensPerDay: Math.round(totalTokens / (days || 1)),
|
|
118
|
+
};
|
|
119
|
+
}
|
|
120
|
+
/**
|
|
121
|
+
* Clear all usage data
|
|
122
|
+
*/
|
|
123
|
+
export function clearUsageData() {
|
|
124
|
+
try {
|
|
125
|
+
const filePath = getUsageFilePath();
|
|
126
|
+
if (fs.existsSync(filePath)) {
|
|
127
|
+
fs.unlinkSync(filePath);
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
catch (error) {
|
|
131
|
+
console.warn('Failed to clear usage data:', error);
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
//# sourceMappingURL=storage.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage.js","sourceRoot":"","sources":["../../source/usage/storage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAC,cAAc,EAAC,MAAM,gBAAgB,CAAC;AAG9C,MAAM,eAAe,GAAG,YAAY,CAAC;AACrC,MAAM,YAAY,GAAG,GAAG,CAAC;AACzB,MAAM,oBAAoB,GAAG,EAAE,CAAC;AAEhC,SAAS,gBAAgB;IACxB,MAAM,SAAS,GAAG,cAAc,EAAE,CAAC;IACnC,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,eAAe;IACvB,MAAM,SAAS,GAAG,cAAc,EAAE,CAAC;IACnC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAC,SAAS,EAAE,IAAI,EAAC,CAAC,CAAC;IAC5C,CAAC;AACF,CAAC;AAED,SAAS,oBAAoB;IAC5B,OAAO;QACN,QAAQ,EAAE,EAAE;QACZ,eAAe,EAAE,EAAE;QACnB,aAAa,EAAE,CAAC;QAChB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;KACvB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa;IAC5B,IAAI,CAAC;QACJ,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QAEpC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9B,OAAO,oBAAoB,EAAE,CAAC;QAC/B,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAc,CAAC;QAE9C,OAAO,IAAI,CAAC;IACb,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,IAAI,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;QAClD,OAAO,oBAAoB,EAAE,CAAC;IAC/B,CAAC;AACF,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAe;IAC7C,IAAI,CAAC;QACJ,eAAe,EAAE,CAAC;QAElB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE9B,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QACpC,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IACpE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;AACF,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,OAAqB;IAC/C,MAAM,IAAI,GAAG,aAAa,EAAE,CAAC;IAE7B,mDAAmD;IACnD,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAE/B,8BAA8B;IAC9B,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,YAAY,EAAE,CAAC;QACzC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;IACtD,CAAC;IAED,wBAAwB;IACxB,IAAI,CAAC,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;IAE3C,yBAAyB;IACzB,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAEpC,cAAc,CAAC,IAAI,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAe,EAAE,OAAqB;IACnE,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvE,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAE7E,iCAAiC;IACjC,IAAI,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;IAE5E,IAAI,CAAC,cAAc,EAAE,CAAC;QACrB,cAAc,GAAG;YAChB,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,CAAC;YACX,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,EAAE;YACb,MAAM,EAAE,EAAE;SACV,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC3C,CAAC;IAED,mBAAmB;IACnB,cAAc,CAAC,QAAQ,IAAI,CAAC,CAAC;IAC7B,cAAc,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;IAEnD,wBAAwB;IACxB,cAAc,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC;QACzC,CAAC,cAAc,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;IAE1E,qBAAqB;IACrB,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;QACnC,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;IAEpE,sEAAsE;IACtE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAClE,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,oBAAoB,EAAE,CAAC;QACxD,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAC;IAC5E,CAAC;AACF,CAAC;AAED,MAAM,UAAU,iBAAiB;IAChC,MAAM,IAAI,GAAG,aAAa,EAAE,CAAC;IAC7B,MAAM,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvD,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAElC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,IAAI,CAAC;AACrE,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,IAAY;IAKjD,MAAM,IAAI,GAAG,aAAa,EAAE,CAAC;IAC7B,MAAM,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;IAC9B,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IAChD,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACxD,MAAM,SAAS,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAEvC,MAAM,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CACrD,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,IAAI,SAAS,CAC5B,CAAC;IAEF,MAAM,WAAW,GAAG,kBAAkB,CAAC,MAAM,CAC5C,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,WAAW,EACnC,CAAC,CACD,CAAC;IACF,MAAM,aAAa,GAAG,kBAAkB,CAAC,MAAM,CAC9C,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,QAAQ,EAChC,CAAC,CACD,CAAC;IAEF,OAAO;QACN,WAAW;QACX,aAAa;QACb,eAAe,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;KACtD,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc;IAC7B,IAAI,CAAC;QACJ,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QACpC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7B,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACzB,CAAC;IACF,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;AACF,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage.spec.d.ts","sourceRoot":"","sources":["../../source/usage/storage.spec.ts"],"names":[],"mappings":""}
|