From c944cd607483cd7bfbe60221a4d9cafc2c03e0df Mon Sep 17 00:00:00 2001
From: Murillo Camargo <murillo@murillocamargo.com.br>
Date: Mon, 30 Mar 2026 13:55:51 -0700
Subject: [PATCH] Real time context tracker count

---
 client/src/components/Chat/Input/ContextTracker.tsx | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/client/src/components/Chat/Input/ContextTracker.tsx b/client/src/components/Chat/Input/ContextTracker.tsx
index 7db5767c28..02cbf16c72 100644
--- a/client/src/components/Chat/Input/ContextTracker.tsx
+++ b/client/src/components/Chat/Input/ContextTracker.tsx
@@ -100,10 +100,12 @@ export default function ContextTracker({
   const prevIsSubmitting = useRef(isSubmitting);
   useEffect(() => {
     if (prevIsSubmitting.current && !isSubmitting) {
-      setUsedTokens(getUsedTokens(getMessages()));
+      // Messages from SSE don't include tokenCount (server strips it).
+      // Invalidate to refetch from API which includes tokenCount from DB.
+      queryClient.invalidateQueries({ queryKey: [QueryKeys.messages] });
     }
     prevIsSubmitting.current = isSubmitting;
-  }, [isSubmitting, getMessages]);
+  }, [isSubmitting, queryClient]);
 
   const maxContextTokens =
     typeof conversation?.maxContextTokens === 'number' &&