@@ -760,9 +760,7 @@ func (a *sessionAgent) Summarize(ctx context.Context, sessionID string, opts fan
760760 // Just in case, get just the last usage info.
761761 usage := resp.Response.Usage
762762 currentSession.SummaryMessageID = summaryMessage.ID
763− if completionTokens := summaryCompletionTokens(usage, summaryMessage); completionTokens != 0 {
764− currentSession.CompletionTokens = completionTokens
765− }
763+ currentSession.CompletionTokens = summaryCompletionTokens(usage, summaryMessage)
766764 currentSession.PromptTokens = 0
767765 _, err = a.sessions.Save(genCtx, currentSession)
768766 if err != nil {