From 20ce7ccf263fc9c359eedf335355aa18a44bef52 Mon Sep 17 00:00:00 2001
From: Zicheng Zhang <38455236+victkk@users.noreply.github.com>
Date: Tue, 25 Feb 2025 08:24:53 +0800
Subject: [PATCH] fix: count reasoning content into token estimation (#2276)

---
 src/renderer/src/services/TokenService.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/services/TokenService.ts b/src/renderer/src/services/TokenService.ts
index c895792e6e..12b8b0821d 100644
--- a/src/renderer/src/services/TokenService.ts
+++ b/src/renderer/src/services/TokenService.ts
@@ -64,7 +64,8 @@ export async function estimateMessageUsage(message: Message): Promise<Completion
     }
   }
 
-  const tokens = estimateTextTokens(message.content)
+  const combinedContent = [message.content, message.reasoning_content].filter((s) => s !== undefined).join(' ')
+  const tokens = estimateTextTokens(combinedContent)
 
   return {
     prompt_tokens: tokens,