fix: include Gemini thinking tokens in output token count for cost calculation

dgageot · krissetto · commit b272de9aff29 · 2026-04-09T11:56:37.000+02:00
Assisted-By: docker-agent
diff --git 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),
 			}

Original file line number	Diff line number	Diff line change
`@@ -171,7 +171,7 @@ func (g *StreamAdapter) Recv() (chat.MessageStreamResponse, error) {`
`171`	`171`	`if res.resp.UsageMetadata != nil && g.trackUsage {`
`172`	`172`	`resp.Usage = &chat.Usage{`
`173`	`173`	`InputTokens: int64(res.resp.UsageMetadata.PromptTokenCount - res.resp.UsageMetadata.CachedContentTokenCount),`
`174`		`- OutputTokens: int64(res.resp.UsageMetadata.CandidatesTokenCount),`
	`174`	`+ OutputTokens: int64(res.resp.UsageMetadata.CandidatesTokenCount + res.resp.UsageMetadata.ThoughtsTokenCount),`
`175`	`175`	`CachedInputTokens: int64(res.resp.UsageMetadata.CachedContentTokenCount),`
`176`	`176`	`ReasoningTokens: int64(res.resp.UsageMetadata.ThoughtsTokenCount),`
`177`	`177`	`}`