fix: token usage reset handling in UI

src/features/app/hooks/useAppServerEvents.ts

@@ -77,7 +77,7 @@ type AppServerEventHandlers = {
   onThreadTokenUsageUpdated?: (
     workspaceId: string,
     threadId: string,
-    tokenUsage: Record<string, unknown>,
+    tokenUsage: Record<string, unknown> | null,
   ) => void;
   onAccountRateLimitsUpdated?: (
     workspaceId: string,
@@ -299,9 +299,9 @@ export function useAppServerEvents(handlers: AppServerEventHandlers) {
       if (method === "thread/tokenUsage/updated") {
         const threadId = String(params.threadId ?? params.thread_id ?? "");
         const tokenUsage =
-          (params.tokenUsage as Record<string, unknown> | undefined) ??
-          (params.token_usage as Record<string, unknown> | undefined);
-        if (threadId && tokenUsage) {
+          (params.tokenUsage as Record<string, unknown> | null | undefined) ??
+          (params.token_usage as Record<string, unknown> | null | undefined);
+        if (threadId && tokenUsage !== undefined) {
           handlers.onThreadTokenUsageUpdated?.(workspace_id, threadId, tokenUsage);
         }
         return;

src/features/threads/hooks/useThreadTurnEvents.ts

@@ -171,7 +171,11 @@ export function useThreadTurnEvents({
   );
 
   const onThreadTokenUsageUpdated = useCallback(
-    (workspaceId: string, threadId: string, tokenUsage: Record<string, unknown>) => {
+    (
+      workspaceId: string,
+      threadId: string,
+      tokenUsage: Record<string, unknown> | null,
+    ) => {
       dispatch({ type: "ensureThread", workspaceId, threadId });
       dispatch({
         type: "setThreadTokenUsage",

src/features/threads/utils/threadNormalize.ts

@@ -73,9 +73,12 @@ export function extractReviewThreadId(response: unknown): string | null {
   return threadId || null;
 }
 
-export function normalizeTokenUsage(raw: Record<string, unknown>): ThreadTokenUsage {
-  const total = (raw.total as Record<string, unknown>) ?? {};
-  const last = (raw.last as Record<string, unknown>) ?? {};
+export function normalizeTokenUsage(
+  raw: Record<string, unknown> | null | undefined,
+): ThreadTokenUsage {
+  const source = raw ?? {};
+  const total = (source.total as Record<string, unknown>) ?? {};
+  const last = (source.last as Record<string, unknown>) ?? {};
   return {
     total: {
       totalTokens: asNumber(total.totalTokens ?? total.total_tokens),
@@ -98,7 +101,7 @@ export function normalizeTokenUsage(raw: Record<string, unknown>): ThreadTokenUs
       ),
     },
     modelContextWindow: (() => {
-      const value = raw.modelContextWindow ?? raw.model_context_window;
+      const value = source.modelContextWindow ?? source.model_context_window;
       if (typeof value === "number") {
         return value;
       }