abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-19 09:33:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Log all parts of a streaming response (#6855)

Browse Source
This commit is contained in:
Silvio Junior
2025-08-22 16:01:45 -04:00
committed by GitHub
parent 5030ced9e1
commit 4c1c6d2b0d
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
packages/core/src/core/loggingContentGenerator.ts
View File

@@ -149,9 +149,12 @@ export class LoggingContentGenerator implements ContentGenerator {
userPromptId: string, userPromptId: string,
): AsyncGenerator<GenerateContentResponse> { ): AsyncGenerator<GenerateContentResponse> {
let lastResponse: GenerateContentResponse | undefined; let lastResponse: GenerateContentResponse | undefined;
const responses: GenerateContentResponse[] = [];
let lastUsageMetadata: GenerateContentResponseUsageMetadata | undefined; let lastUsageMetadata: GenerateContentResponseUsageMetadata | undefined;
try { try {
for await (const response of stream) { for await (const response of stream) {
responses.push(response);
lastResponse = response; lastResponse = response;
if (response.usageMetadata) { if (response.usageMetadata) {
lastUsageMetadata = response.usageMetadata; lastUsageMetadata = response.usageMetadata;
@@ -169,7 +172,7 @@ export class LoggingContentGenerator implements ContentGenerator {
durationMs, durationMs,
userPromptId, userPromptId,
lastUsageMetadata, lastUsageMetadata,
JSON.stringify(lastResponse), JSON.stringify(responses),
); );
} }
} }