From 83d2abbf0a62f8bda0cb4b5ef499eff11dd81a20 Mon Sep 17 00:00:00 2001
From: Greg Slepak <contact@taoeffect.com>
Date: Tue, 12 May 2026 17:04:10 -0700
Subject: [PATCH] fix(agent): clear stale summary token counts

---
 internal/agent/agent.go | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/internal/agent/agent.go b/internal/agent/agent.go
index 80d8ae64a359fec27b0b712fc5ecef5514d579b4..15b0c1af161a7e5631ba00e55e988a19aa28fe0e 100644
--- a/internal/agent/agent.go
+++ b/internal/agent/agent.go
@@ -760,9 +760,7 @@ func (a *sessionAgent) Summarize(ctx context.Context, sessionID string, opts fan
 	// Just in case, get just the last usage info.
 	usage := resp.Response.Usage
 	currentSession.SummaryMessageID = summaryMessage.ID
-	if completionTokens := summaryCompletionTokens(usage, summaryMessage); completionTokens != 0 {
-		currentSession.CompletionTokens = completionTokens
-	}
+	currentSession.CompletionTokens = summaryCompletionTokens(usage, summaryMessage)
 	currentSession.PromptTokens = 0
 	_, err = a.sessions.Save(genCtx, currentSession)
 	if err != nil {