github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2025-12-25 22:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: unify getTotalUsage behavior

Browse Source
This commit is contained in:
Yanlong Wang 2025-09-15 17:31:44 +08:00
parent 58af7eb3a8
commit d6959a6001
No known key found for this signature in database
GPG Key ID: C0A623C0BADF9F37
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/utils/token-tracker.ts
View File

@ -42,9 +42,11 @@ export class TokenTracker extends EventEmitter {
getTotalUsageSnakeCase(): { prompt_tokens: number, completion_tokens: number, total_tokens: number } {
return this.usages.reduce((acc, { usage }) => {
acc.prompt_tokens += usage.promptTokens;
acc.completion_tokens += usage.completionTokens;
acc.total_tokens += usage.totalTokens;
// CompletionTokens > 0 means LLM usage, apply 3x multiplier
const scaler = usage.completionTokens > 0 ? 3 : 1;
acc.prompt_tokens += usage.promptTokens * scaler;
acc.completion_tokens += usage.completionTokens * scaler;
acc.total_tokens += usage.totalTokens * scaler;
return acc;
}, { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 });
}