Preserve reported token usage presence

2026-09-23 16:39:43 +00:00
parent ea94d66fc2
commit d31a563ca1
8 changed files with 166 additions and 16 deletions
--- a/internal/domain/domain.go
+++ b/internal/domain/domain.go
@@ -306,8 +306,18 @@ type StructuredOutputJSONSpec struct {

 // GenerateResponse is the response received from the LLM client.
 type GenerateResponse struct {
-	Content string
-	Usage   TokenUsage
+	Content       string
+	Usage         TokenUsage
+	UsagePresence *TokenUsagePresence
+}
+
+// TokenUsagePresence records which token fields a provider reported.
+type TokenUsagePresence struct {
+	PromptTokens     bool
+	CompletionTokens bool
+	TotalTokens      bool
+	CachedTokens     bool
+	CacheWriteTokens bool
 }

 // TokenUsage tracks token consumption.