feat(telemetry): add controller telemetry tracking and display

- Add telemetry callback to controller execution
- Implement controller telemetry state updates
- Display controller and step token counts in output window
- Refactor telemetry update logic into shared utility

Author: moazbuilds

src/cli/tui/routes/workflow/components/output/output-window.tsx

@@ -425,11 +425,25 @@ export function OutputWindow(props: OutputWindowProps) {
               </Show>
             </box>
             <text fg={themeCtx.theme.textMuted}>
-              context: {(() => {
-                const total = (props.currentAgent?.telemetry.tokensIn ?? 0) + (props.currentAgent?.telemetry.tokensOut ?? 0)
-                if (total >= 1000000) return `${(total / 1000000).toFixed(1)}M`
-                if (total >= 1000) return `${(total / 1000).toFixed(1)}K`
-                return `${total}`
+              {(() => {
+                const formatTokens = (total: number) => {
+                  if (total >= 1000000) return `${(total / 1000000).toFixed(1)}M`
+                  if (total >= 1000) return `${(total / 1000).toFixed(1)}K`
+                  return `${total}`
+                }
+                const controllerTotal = (props.controllerState?.telemetry?.tokensIn ?? 0) + (props.controllerState?.telemetry?.tokensOut ?? 0)
+                const stepTotal = (props.currentAgent?.telemetry.tokensIn ?? 0) + (props.currentAgent?.telemetry.tokensOut ?? 0)
+
+                // Controller view mode - show only controller
+                if (isControllerViewMode()) {
+                  return `controller: ${formatTokens(controllerTotal)}`
+                }
+                // Workflow has controller configured - show both
+                if (props.controllerState != null) {
+                  return `controller: ${formatTokens(controllerTotal)} │ step: ${formatTokens(stepTotal)}`
+                }
+                // No controller - show step only
+                return `context: ${formatTokens(stepTotal)}`
               })()}
             </text>
           </box>

src/cli/tui/routes/workflow/context/ui-state/actions/workflow-actions.ts

@@ -7,6 +7,7 @@
 import type { WorkflowState, WorkflowStatus, WorkflowView, LoopState, ChainedState, InputState, TriggeredAgentState, ControllerState,
   AutonomousMode, AgentTelemetry, AgentStatus } from "../types"
 import { debug } from "../../../../../../../shared/logging/logger.js"
+import { updateControllerTelemetry as updateControllerTelemetryUtil } from "../../../state/utils"
 
 export type WorkflowActionsContext = {
   getState(): WorkflowState
@@ -159,32 +160,9 @@ export function createWorkflowActions(ctx: WorkflowActionsContext) {
     const state = ctx.getState()
     if (!state.controllerState) return
 
-    const current = state.controllerState.telemetry
-    // Context = input (uncached) + cached tokens
-    const currentContext = (telemetry.tokensIn ?? 0) + (telemetry.cached ?? 0)
-    // Replace telemetry values (like step agent) - shows current context, not accumulated
-    const newTokensIn = currentContext > 0 ? currentContext : current.tokensIn
-    const newTokensOut = telemetry.tokensOut ?? current.tokensOut
-    const newCached = telemetry.cached ?? current.cached
-    // Cost accumulates
-    const newCost = (current.cost ?? 0) + (telemetry.cost ?? 0) || undefined
-
-    debug('[TELEMETRY:5-ACTIONS] [CONTROLLER] updateControllerTelemetry')
-    debug('[TELEMETRY:5-ACTIONS] [CONTROLLER]   INPUT: uncached=%d + cached=%d = context=%d, output=%d',
-      telemetry.tokensIn ?? 0, telemetry.cached ?? 0, currentContext, telemetry.tokensOut ?? 0)
-    debug('[TELEMETRY:5-ACTIONS] [CONTROLLER]   RESULT: context=%d, output=%d (REPLACES previous)', newTokensIn, newTokensOut)
-
     ctx.setState({
       ...state,
-      controllerState: {
-        ...state.controllerState,
-        telemetry: {
-          tokensIn: newTokensIn,
-          tokensOut: newTokensOut,
-          cached: newCached,
-          cost: newCost,
-        },
-      },
+      controllerState: updateControllerTelemetryUtil(state.controllerState, telemetry),
     })
     ctx.notify()
   }

src/cli/tui/routes/workflow/state/utils.ts

@@ -1,4 +1,4 @@
-import type { AgentState, AgentStatus, AgentTelemetry } from "./types"
+import type { AgentState, AgentStatus, AgentTelemetry, ControllerState } from "./types"
 import { debug } from "../../../../../shared/logging/logger.js"
 
 export function updateAgentTelemetryInList(
@@ -53,3 +53,35 @@ export function updateAgentStatusInList(
       : agent,
   )
 }
+
+export function updateControllerTelemetry(
+  controller: ControllerState,
+  telemetry: Partial<AgentTelemetry>
+): ControllerState {
+  // Context = input tokens (total)
+  // Note: cached tokens are already INCLUDED in tokensIn, not separate
+  // cached is just metadata showing how many of those tokens were served from cache
+  const currentContext = telemetry.tokensIn ?? 0
+  const newTokensIn = currentContext > 0 ? currentContext : controller.telemetry.tokensIn
+  const newTokensOut = telemetry.tokensOut ?? controller.telemetry.tokensOut
+  const newCached = telemetry.cached ?? controller.telemetry.cached
+  // Cost accumulates
+  const newCost = (controller.telemetry.cost ?? 0) + (telemetry.cost ?? 0) || undefined
+
+  debug('[TELEMETRY:5-UTILS] [CONTROLLER] updateControllerTelemetry')
+  debug('[TELEMETRY:5-UTILS] [CONTROLLER]   INPUT: context=%d (cached=%d), output=%d',
+    telemetry.tokensIn ?? 0, telemetry.cached ?? 0, telemetry.tokensOut ?? 0)
+  debug('[TELEMETRY:5-UTILS] [CONTROLLER]   RESULT: context=%d, output=%d (REPLACES previous)',
+    newTokensIn, newTokensOut)
+
+  return {
+    ...controller,
+    telemetry: {
+      tokensIn: newTokensIn,
+      tokensOut: newTokensOut,
+      cached: newCached,
+      cost: newCost,
+      duration: telemetry.duration ?? controller.telemetry.duration,
+    },
+  }
+}

src/workflows/controller/view.ts

@@ -20,6 +20,7 @@ import { formatUserInput } from '../../shared/formatters/outputMarkers.js';
 import { controllerPrefixUser } from '../../shared/prompts/index.js';
 import { AgentLoggerService, AgentMonitorService } from '../../agents/monitoring/index.js';
 import { executeAgent } from '../../agents/runner/runner.js';
+import { createTelemetryCallback } from '../../agents/execution/telemetry.js';
 
 export interface ControllerViewOptions {
   cwd: string;
@@ -172,6 +173,11 @@ export async function runControllerView(
               resumeMonitoringId: controllerConfig.monitoringId,
               engine: controllerConfig.engine,
               model: controllerConfig.model,
+              onTelemetry: createTelemetryCallback({
+                uniqueAgentId: '',
+                emitter,
+                isController: true,
+              }),
             });
 
             // After response, go back to awaiting input
@@ -368,6 +374,11 @@ export async function runControllerView(
             resumeMonitoringId: controllerConfig.monitoringId,
             engine: controllerConfig.engine,
             model: controllerConfig.model,
+            onTelemetry: createTelemetryCallback({
+              uniqueAgentId: '',
+              emitter,
+              isController: true,
+            }),
           });
 
           // After response, go back to awaiting input