From dcc7b7c882cb1e47dcc5028723a996d473767af3 Mon Sep 17 00:00:00 2001 From: Matheus Date: Wed, 4 Mar 2026 18:59:44 -0300 Subject: [PATCH] fix(ai-gemini): parse usageMetadata in image adapter response The Gemini image adapter was always returning `usage: undefined` even though the API response includes `usageMetadata` with token counts. Map `usageMetadata` fields to the `ImageGenerationResult.usage` object, consistent with how the OpenAI image adapter already handles usage. Closes #330 Co-Authored-By: Claude Opus 4.6 --- packages/typescript/ai-gemini/src/adapters/image.ts | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/packages/typescript/ai-gemini/src/adapters/image.ts b/packages/typescript/ai-gemini/src/adapters/image.ts index 2ccf47b5..fb4f1cdb 100644 --- a/packages/typescript/ai-gemini/src/adapters/image.ts +++ b/packages/typescript/ai-gemini/src/adapters/image.ts @@ -168,7 +168,13 @@ export class GeminiImageAdapter< id: generateId(this.name), model, images, - usage: undefined, + usage: response.usageMetadata + ? { + inputTokens: response.usageMetadata.promptTokenCount, + outputTokens: response.usageMetadata.candidatesTokenCount, + totalTokens: response.usageMetadata.totalTokenCount, + } + : undefined, } }