Skip to content

Commit d6959a6

Browse files
committed
fix: unify getTotalUsage behavior
1 parent 58af7eb commit d6959a6

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/utils/token-tracker.ts

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,9 +42,11 @@ export class TokenTracker extends EventEmitter {
4242

4343
getTotalUsageSnakeCase(): { prompt_tokens: number, completion_tokens: number, total_tokens: number } {
4444
return this.usages.reduce((acc, { usage }) => {
45-
acc.prompt_tokens += usage.promptTokens;
46-
acc.completion_tokens += usage.completionTokens;
47-
acc.total_tokens += usage.totalTokens;
45+
// CompletionTokens > 0 means LLM usage, apply 3x multiplier
46+
const scaler = usage.completionTokens > 0 ? 3 : 1;
47+
acc.prompt_tokens += usage.promptTokens * scaler;
48+
acc.completion_tokens += usage.completionTokens * scaler;
49+
acc.total_tokens += usage.totalTokens * scaler;
4850
return acc;
4951
}, { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 });
5052
}

0 commit comments

Comments
 (0)