feat(tui): show router inference metrics beside router lines

2026-06-03 15:41:51 +04:00
parent c2fb17f6ee
commit 80a72d370e
6 changed files with 45 additions and 7 deletions
@@ -98,7 +98,11 @@ func (m *Model) applyServer(msg protocol.ServerMessage) {
 		if msg.Role == "USER" {
 			role = "user"
 		}
-		m.appendRouter(msg.SessionID, RouterEntry{role, msg.Content})
+		entry := RouterEntry{Role: role, Content: msg.Content}
+		if msg.LatencyMs != nil && msg.TotalTokens != nil {
+			entry.Metrics = &TurnMetrics{LatencyMs: *msg.LatencyMs, TotalTokens: *msg.TotalTokens}
+		}
+		m.appendRouter(msg.SessionID, entry)
 		if s := m.session(msg.SessionID); s != nil {
 			s.LastEventAt = nowMillis()
 		}
@@ -182,7 +186,7 @@ func (m *Model) applyServer(msg protocol.ServerMessage) {
 			s.addEvent(msg.OccurredAt, "ToolCompleted", msg.ToolName)
 		}
 		if msg.Diff != nil && *msg.Diff != "" {
-			m.appendRouter(msg.SessionID, RouterEntry{"tool", *msg.Diff})
+			m.appendRouter(msg.SessionID, RouterEntry{Role: "tool", Content: *msg.Diff})
 		}
 	case protocol.TypeToolAssessed:
 		if s := m.session(msg.SessionID); s != nil {