diff --git a/packages/core/src/core/loggingContentGenerator.ts b/packages/core/src/core/loggingContentGenerator.ts index 2abe3dce..6d7a29b9 100644 --- a/packages/core/src/core/loggingContentGenerator.ts +++ b/packages/core/src/core/loggingContentGenerator.ts @@ -149,9 +149,12 @@ export class LoggingContentGenerator implements ContentGenerator { userPromptId: string, ): AsyncGenerator { let lastResponse: GenerateContentResponse | undefined; + const responses: GenerateContentResponse[] = []; + let lastUsageMetadata: GenerateContentResponseUsageMetadata | undefined; try { for await (const response of stream) { + responses.push(response); lastResponse = response; if (response.usageMetadata) { lastUsageMetadata = response.usageMetadata; @@ -169,7 +172,7 @@ export class LoggingContentGenerator implements ContentGenerator { durationMs, userPromptId, lastUsageMetadata, - JSON.stringify(lastResponse), + JSON.stringify(responses), ); } }