diff --git a/src/modules/aix/server/dispatch/chatGenerate/parsers/gemini.parser.ts b/src/modules/aix/server/dispatch/chatGenerate/parsers/gemini.parser.ts index 4d8e0d47c..769b2d25b 100644 --- a/src/modules/aix/server/dispatch/chatGenerate/parsers/gemini.parser.ts +++ b/src/modules/aix/server/dispatch/chatGenerate/parsers/gemini.parser.ts @@ -217,6 +217,9 @@ export function createGeminiGenerateContentResponseParser(requestedModelName: st TIn: generationChunk.usageMetadata.promptTokenCount, TOut: generationChunk.usageMetadata.candidatesTokenCount, }; + if (generationChunk.usageMetadata.thoughtsTokenCount) + metricsUpdate.TOutR = generationChunk.usageMetadata.thoughtsTokenCount; + if (isStreaming && timeToFirstEvent !== undefined) metricsUpdate.dtStart = timeToFirstEvent;