From 48315c9621b72c12119a246a691d408190fe585c Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Tue, 11 Feb 2025 10:56:15 +0000 Subject: [PATCH] fix: count read API tokens as completion tokens Co-Authored-By: sha.zhou@jina.ai --- src/tools/read.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tools/read.ts b/src/tools/read.ts index a46530b..77cc449 100644 --- a/src/tools/read.ts +++ b/src/tools/read.ts @@ -74,12 +74,12 @@ export function readUrl(url: string, tracker?: TokenTracker): Promise<{ response if (tracker) { // Track API response tokens - tracker.trackUsage('read_api', apiTokens, TOKEN_CATEGORIES.PROMPT); + tracker.trackUsage('read_api', apiTokens, TOKEN_CATEGORIES.ACCEPTED); // Track content length tokens using the same estimation method if (response.data.content) { const contentTokens = Math.ceil(Buffer.byteLength(response.data.content, 'utf-8') / 4); - tracker.trackUsage('read_content', contentTokens, TOKEN_CATEGORIES.PROMPT); + tracker.trackUsage('read_content', contentTokens); } }