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,
): AsyncGenerator<GenerateContentResponse> {
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),
);
}
}