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