From f7914468a9e05dd1ce3a7d8b8bd674b401817969 Mon Sep 17 00:00:00 2001 From: Carlos Alexandro Becker Date: Mon, 26 Jan 2026 12:58:56 -0300 Subject: [PATCH] fix: stats chart don't account for cached tokens Signed-off-by: Carlos Alexandro Becker --- internal/agent/agent.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 }