diff --git a/src/main/native-chat/agent-session-wire/structured-agent-session-host.ts b/src/main/native-chat/agent-session-wire/structured-agent-session-host.ts index d8daaa65953..e2f59123e86 100644 --- a/src/main/native-chat/agent-session-wire/structured-agent-session-host.ts +++ b/src/main/native-chat/agent-session-wire/structured-agent-session-host.ts @@ -260,8 +260,7 @@ export class StructuredAgentSessionHost { } /** Test barrier: every write has landed by its call's return, so no production path needs it. */ - flushStreamedEvents = (sessionId: string): Promise => - this.runtimeState.flushEventSink(sessionId) + flushStreamedEvents = (id: string): Promise => this.runtimeState.flushEventSink(id) // Trigger inlined rather than imported: `AgentSessionResumeTrigger` in shared is the canonical // type, and this file has no line budget left for the import. diff --git a/src/shared/structured-agent-session-live-turn.test.ts b/src/shared/structured-agent-session-live-turn.test.ts index 5a4d1f7d43c..389e6b2aff3 100644 --- a/src/shared/structured-agent-session-live-turn.test.ts +++ b/src/shared/structured-agent-session-live-turn.test.ts @@ -39,9 +39,13 @@ describe('isStructuredAgentSessionThinking', () => { state, ...(state === 'completed' ? { completedAt: 3 } : {}) }) - expect(isStructuredAgentSessionThinking([turnStart, withState('running')])).toBe(true) + expect(isStructuredAgentSessionThinking({ items: [turnStart, withState('running')] })).toBe( + true + ) // Ended reasoning stays the newest row while Claude streams a tool's input after it. - expect(isStructuredAgentSessionThinking([turnStart, withState('completed')])).toBe(false) + expect(isStructuredAgentSessionThinking({ items: [turnStart, withState('completed')] })).toBe( + false + ) }) it('is false once a tool call, a message or a diff lands after the reasoning', () => {