Stream token usage on every agent message (#5342)

2025-10-29 23:23:25 -04:00
parent c875f13757
commit 37e1bb1d37
18 changed files with 214 additions and 63 deletions
@@ -132,6 +132,7 @@ function BaseChatContent({
    sessionOutputTokens,
    localInputTokens,
    localOutputTokens,
+    tokenState,
    commandHistory,
    toolCallNotifications,
    sessionMetadata,
@@ -442,9 +443,13 @@ function BaseChatContent({
            commandHistory={commandHistory}
            initialValue={input || ''}
            setView={setView}
-            numTokens={sessionTokenCount}
-            inputTokens={sessionInputTokens || localInputTokens}
-            outputTokens={sessionOutputTokens || localOutputTokens}
+            totalTokens={tokenState?.totalTokens ?? sessionTokenCount}
+            accumulatedInputTokens={
+              tokenState?.accumulatedInputTokens ?? sessionInputTokens ?? localInputTokens
+            }
+            accumulatedOutputTokens={
+              tokenState?.accumulatedOutputTokens ?? sessionOutputTokens ?? localOutputTokens
+            }
            droppedFiles={droppedFiles}
            onFilesProcessed={() => setDroppedFiles([])} // Clear dropped files after processing
            messages={messages}