From 53963f264dce700fd58c6ad38a3841334daa35c0 Mon Sep 17 00:00:00 2001
From: Yanlong Wang <yanlong.wang@naiver.org>
Date: Tue, 16 Sep 2025 13:56:26 +0800
Subject: [PATCH] saas: suspend llm token multiplier

---
 src/utils/token-tracker.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/utils/token-tracker.ts b/src/utils/token-tracker.ts
index 76d642d..522a777 100644
--- a/src/utils/token-tracker.ts
+++ b/src/utils/token-tracker.ts
@@ -32,7 +32,8 @@ export class TokenTracker extends EventEmitter {
   getTotalUsage(): LanguageModelUsage {
     return this.usages.reduce((acc, { usage }) => {
       // CompletionTokens > 0 means LLM usage, apply 3x multiplier
-      const scaler = usage.completionTokens > 0 ? 3 : 1;
+      // const scaler = usage.completionTokens > 0 ? 3 : 1;
+      const scaler = 1;
       acc.promptTokens += usage.promptTokens * scaler;
       acc.completionTokens += usage.completionTokens * scaler;
       acc.totalTokens += usage.totalTokens * scaler;
@@ -43,7 +44,8 @@ export class TokenTracker extends EventEmitter {
   getTotalUsageSnakeCase(): { prompt_tokens: number, completion_tokens: number, total_tokens: number } {
     return this.usages.reduce((acc, { usage }) => {
       // CompletionTokens > 0 means LLM usage, apply 3x multiplier
-      const scaler = usage.completionTokens > 0 ? 3 : 1;
+      // const scaler = usage.completionTokens > 0 ? 3 
+      const scaler = 1;
       acc.prompt_tokens += usage.promptTokens * scaler;
       acc.completion_tokens += usage.completionTokens * scaler;
       acc.total_tokens += usage.totalTokens * scaler;