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 {