From f08976d9c0fa97b2cedde58fc7832d1142033ed7 Mon Sep 17 00:00:00 2001
From: Yanlong Wang <yanlong.wang@naiver.org>
Date: Wed, 12 Feb 2025 12:58:25 +0800
Subject: [PATCH] jina-ai: fix token charge amount observation

---
 src/utils/token-tracker.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/utils/token-tracker.ts b/src/utils/token-tracker.ts
index bdc29fe..336593d 100644
--- a/src/utils/token-tracker.ts
+++ b/src/utils/token-tracker.ts
@@ -15,7 +15,7 @@ export class TokenTracker extends EventEmitter {
       const asyncLocalContext = process.asyncLocalContext as any;
       this.on('usage', () => {
         if (asyncLocalContext.available()) {
-          asyncLocalContext.ctx.chargeAmount = this.getTotalUsage();
+          asyncLocalContext.ctx.chargeAmount = this.getTotalUsage().totalTokens;
         }
       });