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
@@ -367,6 +367,7 @@ export type MessageContent = (TextContent & {

 export type MessageEvent = {
    message: Message;
+    token_state: TokenState;
    type: 'Message';
 } | {
    error: string;
@@ -789,6 +790,15 @@ export type ThinkingContent = {
    thinking: string;
 };

+export type TokenState = {
+    accumulatedInputTokens: number;
+    accumulatedOutputTokens: number;
+    accumulatedTotalTokens: number;
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+};
+
 export type Tool = {
    annotations?: ToolAnnotations | {
        [key: string]: unknown;