From e37e9d3ad5dc27e7b4f43d2a00a2b08e19073a22 Mon Sep 17 00:00:00 2001 From: DoSun Date: Sat, 9 May 2026 09:49:37 +0800 Subject: [PATCH] feat: enable stream_event forwarding for serve mode streaming QueryEngine previously guarded stream_event yields behind includePartialMessages, which was undefined in serve mode. Remove the guard so stream_event messages always flow to the child process stdout. Add stream_event handling in sessionHandle.routeMessage to emit via EventBus for downstream consumers. --- src/QueryEngine.ts | 14 ++++++-------- src/server/sessionHandle.ts | 5 +++++ 2 files changed, 11 insertions(+), 8 deletions(-) diff --git a/src/QueryEngine.ts b/src/QueryEngine.ts index db5732489..6704dac68 100644 --- a/src/QueryEngine.ts +++ b/src/QueryEngine.ts @@ -837,14 +837,12 @@ export class QueryEngine { ) } - if (includePartialMessages) { - yield { - type: 'stream_event' as const, - event, - session_id: getSessionId(), - parent_tool_use_id: null, - uuid: randomUUID(), - } + yield { + type: 'stream_event' as const, + event, + session_id: getSessionId(), + parent_tool_use_id: null, + uuid: randomUUID(), } break diff --git a/src/server/sessionHandle.ts b/src/server/sessionHandle.ts index 93749dbf0..638115db5 100644 --- a/src/server/sessionHandle.ts +++ b/src/server/sessionHandle.ts @@ -521,6 +521,11 @@ export class SessionHandle { this.pendingQuestions.delete(cancelRequestId) break } + case 'stream_event': { + this.lastActiveAt = Date.now() + this.emitEvent('stream_event', msg) + break + } default: { break }