From 765c11d5981f75ab047293ddd8dc3fc4da218a8c Mon Sep 17 00:00:00 2001
From: David Gageot <david.gageot@docker.com>
Date: Thu, 9 Apr 2026 10:55:36 +0200
Subject: [PATCH] fix: include Gemini thinking tokens in output token count for
 cost calculation

Assisted-By: docker-agent
---
 pkg/model/provider/gemini/adapter.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/model/provider/gemini/adapter.go b/pkg/model/provider/gemini/adapter.go
index f2cb64f0d..67f88b289 100644
--- a/pkg/model/provider/gemini/adapter.go
+++ b/pkg/model/provider/gemini/adapter.go
@@ -171,7 +171,7 @@ func (g *StreamAdapter) Recv() (chat.MessageStreamResponse, error) {
 		if res.resp.UsageMetadata != nil && g.trackUsage {
 			resp.Usage = &chat.Usage{
 				InputTokens:       int64(res.resp.UsageMetadata.PromptTokenCount - res.resp.UsageMetadata.CachedContentTokenCount),
-				OutputTokens:      int64(res.resp.UsageMetadata.CandidatesTokenCount),
+				OutputTokens:      int64(res.resp.UsageMetadata.CandidatesTokenCount + res.resp.UsageMetadata.ThoughtsTokenCount),
 				CachedInputTokens: int64(res.resp.UsageMetadata.CachedContentTokenCount),
 				ReasoningTokens:   int64(res.resp.UsageMetadata.ThoughtsTokenCount),
 			}