Real time context tracker count

This commit is contained in:
Murillo Camargo 2026-03-30 13:55:51 -07:00
parent ec7f7908ed
commit c944cd6074

View file

@ -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' &&