vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-05-01 21:20:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

fix: ensure message_start and message_stop are always paired in SDK stream events

Browse source 
Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>
This commit is contained in:
mingholy.lmh 2026-03-17 23:33:37 +08:00
parent 4c0e4a9611
commit eea92fc8db
2 changed files with 464 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

30
packages/cli/src/nonInteractive/io/StreamJsonOutputAdapter.ts
View file

@ -69,26 +69,44 @@ export class StreamJsonOutputAdapter
}
finalizeAssistantMessage(): CLIAssistantMessage {
const state = this.mainAgentMessageState;
return this.finalizeAssistantMessageInternal(
this.mainAgentMessageState,
null,
);
}
/**
* Overrides base class to emit message_stop event when message is finalized.
* This ensures message_start and message_stop are always paired.
*/
protected override finalizeAssistantMessageInternal(
state: MessageState,
parentToolUseId: string | null,
): CLIAssistantMessage {
if (state.finalized) {
return this.buildMessage(null);
return this.buildMessage(parentToolUseId);
}
state.finalized = true;
this.finalizePendingBlocks(state, null);
this.finalizePendingBlocks(state, parentToolUseId);
const orderedOpenBlocks = Array.from(state.openBlocks).sort(
(a, b) => a - b,
);
for (const index of orderedOpenBlocks) {
this.onBlockClosed(state, index, null);
this.onBlockClosed(state, index, parentToolUseId);
this.closeBlock(state, index);
}
if (state.messageStarted && this.includePartialMessages) {
// Emit message_stop for main agent when message was started and partial messages are enabled
if (
state.messageStarted &&
this.includePartialMessages &&
parentToolUseId === null
) {
this.emitStreamEventIfEnabled({ type: 'message_stop' }, null);
}
const message = this.buildMessage(null);
const message = this.buildMessage(parentToolUseId);
this.updateLastAssistantMessage(message);
this.emitMessageImpl(message);
return message;