fix context display

2025-06-28 21:01:53 -04:00
parent 196f42cbff
commit 3a9584a419
2 changed files with 12 additions and 10 deletions
--- a/packages/tui/internal/components/status/status.go
+++ b/packages/tui/internal/components/status/status.go
@@ -100,16 +100,18 @@ func (m statusComponent) View() string {
 		contextWindow := m.app.Model.Limit.Context

 		for _, message := range m.app.Messages {
-			if message.Metadata.Assistant.Cost > 0 {
-				cost += message.Metadata.Assistant.Cost
-				usage := message.Metadata.Assistant.Tokens
-				if usage.Output > 0 {
-					tokens = (usage.Input +
-						usage.Cache.Write +
-						usage.Cache.Read +
-						usage.Output +
-						usage.Reasoning)
+			cost += message.Metadata.Assistant.Cost
+			usage := message.Metadata.Assistant.Tokens
+			if usage.Output > 0 {
+				if message.Metadata.Assistant.Summary {
+					tokens = usage.Output
+					continue
 				}
+				tokens = (usage.Input +
+					usage.Cache.Write +
+					usage.Cache.Read +
+					usage.Output +
+					usage.Reasoning)
 			}
 		}