diff --git a/internal/agent/agent.go b/internal/agent/agent.go index 15aa1c0a840e8aafd2feced6bd4bb580e17be5f8..d5b0ed5f1cf4333922ef65c6e0a0fafcf92710a2 100644 --- a/internal/agent/agent.go +++ b/internal/agent/agent.go @@ -858,7 +858,7 @@ func (a *sessionAgent) generateTitle(ctx context.Context, sessionID string, user } promptTokens := resp.TotalUsage.InputTokens + resp.TotalUsage.CacheCreationTokens - completionTokens := resp.TotalUsage.OutputTokens + resp.TotalUsage.CacheReadTokens + completionTokens := resp.TotalUsage.OutputTokens // Atomically update only title and usage fields to avoid overriding other // concurrent session updates. @@ -897,7 +897,7 @@ func (a *sessionAgent) updateSessionUsage(model Model, session *session.Session, session.Cost += cost } - session.CompletionTokens = usage.OutputTokens + usage.CacheReadTokens + session.CompletionTokens = usage.OutputTokens session.PromptTokens = usage.InputTokens + usage.CacheCreationTokens }