From 01ca9b1655a751f9130d3ec725c61d3d88fb5616 Mon Sep 17 00:00:00 2001
From: Marco Beretta <81851188+berry-13@users.noreply.github.com>
Date: Mon, 15 Dec 2025 23:55:43 +0100
Subject: [PATCH] feat: enhance token formatting and reset logic for new
 conversations

---
 .../src/components/Chat/Input/TokenUsageIndicator.tsx |  4 ++--
 client/src/hooks/useTokenUsage.ts                     | 11 +++++++++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/client/src/components/Chat/Input/TokenUsageIndicator.tsx b/client/src/components/Chat/Input/TokenUsageIndicator.tsx
index 72562044d9..f4a847ef06 100644
--- a/client/src/components/Chat/Input/TokenUsageIndicator.tsx
+++ b/client/src/components/Chat/Input/TokenUsageIndicator.tsx
@@ -5,10 +5,10 @@ import { cn } from '~/utils';
 
 function formatTokens(n: number): string {
   if (n >= 1000000) {
-    return `${(n / 1000000).toFixed(1)}M`;
+    return `${(n / 1000000).toFixed(1).replace(/\.0$/, '')}M`;
   }
   if (n >= 1000) {
-    return `${(n / 1000).toFixed(1)}K`;
+    return `${(n / 1000).toFixed(1).replace(/\.0$/, '')}K`;
   }
   return n.toString();
 }
diff --git a/client/src/hooks/useTokenUsage.ts b/client/src/hooks/useTokenUsage.ts
index 2dce4f87b2..3141c9c454 100644
--- a/client/src/hooks/useTokenUsage.ts
+++ b/client/src/hooks/useTokenUsage.ts
@@ -84,6 +84,17 @@ export function useTokenUsageComputation() {
   useEffect(() => {
     setTokenUsage(tokenData);
   }, [tokenData, setTokenUsage]);
+
+  // Reset token usage when starting a new conversation
+  useEffect(() => {
+    if (paramId === 'new' && effectiveMessages.length === 0) {
+      setTokenUsage({
+        inputTokens: 0,
+        outputTokens: 0,
+        maxContext: null,
+      });
+    }
+  }, [paramId, effectiveMessages.length, setTokenUsage]);
 }
 
 /**