@pikku/core 0.12.107 → 0.12.112

package/src/wirings/agent/agent-agui.test.ts

@@ -1,1213 +0,0 @@
-import { describe, it, mock } from 'node:test'
-import assert from 'node:assert/strict'
-import { EventSchemas } from '@ag-ui/core'
-import { wrapChannelWithAGUI } from './agent-agui.js'
-import type { AgentStreamChannel, AgentStreamEvent } from './agent.types.js'
-
-function makeChannel(): { channel: AgentStreamChannel; events: unknown[] } {
-  const events: unknown[] = []
-  const channel: AgentStreamChannel = {
-    channelId: 'test-id',
-    openingData: null,
-    state: 'open',
-    setState: () => {},
-    getState: () => undefined as any,
-    clearState: () => {},
-    sendBinary: () => {},
-    close: () => {},
-    send: (e) => events.push(e),
-  }
-  return { channel, events }
-}
-
-function types(events: unknown[]): string[] {
-  return (events as any[]).map((e) => e.type)
-}
-
-function find<T = any>(
-  events: unknown[],
-  type: string,
-  name?: string
-): T | undefined {
-  return (events as any[]).find(
-    (e) => e.type === type && (name === undefined || e.name === name)
-  ) as T
-}
-
-describe('wrapChannelWithAGUI — text streaming', () => {
-  it('sends TEXT_MESSAGE_START only once for consecutive text-deltas', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'a' } as AgentStreamEvent)
-    wrapped.send({ type: 'text-delta', text: 'b' } as AgentStreamEvent)
-    wrapped.send({ type: 'text-delta', text: 'c' } as AgentStreamEvent)
-
-    assert.equal(
-      events.filter((e: any) => e.type === 'TEXT_MESSAGE_START').length,
-      1
-    )
-    assert.equal(
-      events.filter((e: any) => e.type === 'TEXT_MESSAGE_CONTENT').length,
-      3
-    )
-  })
-
-  it('uses the same messageId for START, all CONTENTs, and END', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'hello' } as AgentStreamEvent)
-    wrapped.send({ type: 'text-delta', text: ' world' } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const start = find(events, 'TEXT_MESSAGE_START')
-    const end = find(events, 'TEXT_MESSAGE_END')
-    const contents = (events as any[]).filter(
-      (e) => e.type === 'TEXT_MESSAGE_CONTENT'
-    )
-
-    assert.ok(start)
-    assert.ok(end)
-    assert.equal(end.messageId, start.messageId)
-    for (const c of contents) {
-      assert.equal(c.messageId, start.messageId)
-    }
-  })
-
-  it('closes open text message when done arrives', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'hi' } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    assert.ok(find(events, 'TEXT_MESSAGE_END'))
-  })
-
-  it('closes open text message when error arrives', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'partial' } as AgentStreamEvent)
-    wrapped.send({ type: 'error', message: 'oops' } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(t.indexOf('TEXT_MESSAGE_END') < t.indexOf('RUN_ERROR'))
-  })
-
-  it('does not emit TEXT_MESSAGE_END when no text was sent', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    assert.equal(
-      events.filter((e: any) => e.type === 'TEXT_MESSAGE_END').length,
-      0
-    )
-  })
-})
-
-describe('wrapChannelWithAGUI — reasoning streaming', () => {
-  it('sends THINKING_TEXT_MESSAGE_START only once for consecutive reasoning-deltas', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'reasoning-delta',
-      text: 'step 1',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'reasoning-delta',
-      text: 'step 2',
-    } as AgentStreamEvent)
-
-    assert.equal(
-      events.filter((e: any) => e.type === 'THINKING_TEXT_MESSAGE_START')
-        .length,
-      1
-    )
-    assert.equal(
-      events.filter((e: any) => e.type === 'THINKING_TEXT_MESSAGE_CONTENT')
-        .length,
-      2
-    )
-  })
-
-  it('closes thinking message and opens text message when switching', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'reasoning-delta',
-      text: 'thinking',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'text-delta', text: 'answer' } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(t.includes('THINKING_TEXT_MESSAGE_END'))
-    assert.ok(
-      t.indexOf('THINKING_TEXT_MESSAGE_END') < t.indexOf('TEXT_MESSAGE_START')
-    )
-  })
-
-  it('closes thinking message before tool-call', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'reasoning-delta',
-      text: 'reasoning',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'fn',
-      args: {},
-    } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(
-      t.indexOf('THINKING_TEXT_MESSAGE_END') < t.indexOf('TOOL_CALL_START')
-    )
-  })
-
-  it('uses consistent messageId across THINKING START, CONTENT, END', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'reasoning-delta',
-      text: 'reason',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const start = find(events, 'THINKING_TEXT_MESSAGE_START')
-    const end = find(events, 'THINKING_TEXT_MESSAGE_END')
-    const content = find(events, 'THINKING_TEXT_MESSAGE_CONTENT')
-
-    assert.ok(start && end && content)
-    assert.equal(content.messageId, start.messageId)
-    assert.equal(end.messageId, start.messageId)
-  })
-})
-
-describe('wrapChannelWithAGUI — tool calls', () => {
-  it('closes text message before TOOL_CALL_START', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'preamble' } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'myTool',
-      args: { x: 1 },
-    } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(t.indexOf('TEXT_MESSAGE_END') < t.indexOf('TOOL_CALL_START'))
-  })
-
-  it('emits TOOL_CALL_START, TOOL_CALL_ARGS, TOOL_CALL_END in sequence', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'search',
-      args: { q: 'hi' },
-    } as AgentStreamEvent)
-
-    const t = types(events)
-    const startIdx = t.indexOf('TOOL_CALL_START')
-    const argsIdx = t.indexOf('TOOL_CALL_ARGS')
-    const endIdx = t.indexOf('TOOL_CALL_END')
-    assert.ok(startIdx !== -1 && argsIdx !== -1 && endIdx !== -1)
-    assert.equal(argsIdx, startIdx + 1)
-    assert.equal(endIdx, argsIdx + 1)
-  })
-
-  it('propagates toolCallId and toolCallName to START, ARGS and END', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'abc',
-      toolName: 'myFunc',
-      args: { n: 42 },
-    } as AgentStreamEvent)
-
-    const start = find(events, 'TOOL_CALL_START')
-    const args = find(events, 'TOOL_CALL_ARGS')
-    const end = find(events, 'TOOL_CALL_END')
-    assert.equal(start.toolCallId, 'abc')
-    assert.equal(start.toolCallName, 'myFunc')
-    assert.equal(args.toolCallId, 'abc')
-    assert.equal(args.delta, '{"n":42}')
-    assert.equal(end.toolCallId, 'abc')
-  })
-
-  it('handles multiple tool calls in sequence', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'fn1',
-      args: {},
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc2',
-      toolName: 'fn2',
-      args: {},
-    } as AgentStreamEvent)
-
-    const starts = (events as any[]).filter((e) => e.type === 'TOOL_CALL_START')
-    const ends = (events as any[]).filter((e) => e.type === 'TOOL_CALL_END')
-    assert.equal(starts.length, 2)
-    assert.equal(ends.length, 2)
-    assert.equal(starts[0].toolCallId, 'tc1')
-    assert.equal(starts[1].toolCallId, 'tc2')
-  })
-
-  it('converts object tool-result to JSON string in content', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'tool-result',
-      toolCallId: 'tc1',
-      toolName: 'fn',
-      result: { ok: true },
-    } as AgentStreamEvent)
-
-    const r = find(events, 'TOOL_CALL_RESULT')
-    assert.equal(r.content, '{"ok":true}')
-    assert.equal(r.role, 'tool')
-  })
-
-  it('passes string tool-result through unchanged', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'tool-result',
-      toolCallId: 'tc1',
-      toolName: 'fn',
-      result: 'plain text',
-    } as AgentStreamEvent)
-
-    const r = find(events, 'TOOL_CALL_RESULT')
-    assert.equal(r.content, 'plain text')
-  })
-})
-
-describe('wrapChannelWithAGUI — run lifecycle', () => {
-  it('emits RUN_FINISHED with full usage fields', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'usage',
-      tokens: { input: 100, output: 50 },
-      model: 'claude-3-5',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const f = find(events, 'RUN_FINISHED')
-    assert.ok(f)
-    assert.equal(f.model, 'claude-3-5')
-    assert.equal(f.usage.promptTokens, 100)
-    assert.equal(f.usage.completionTokens, 50)
-    assert.equal(f.usage.totalTokens, 150)
-  })
-
-  it('emits RUN_FINISHED exactly once when both usage and done arrive', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'usage',
-      tokens: { input: 1, output: 1 },
-      model: 'm',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    assert.equal(events.filter((e: any) => e.type === 'RUN_FINISHED').length, 1)
-  })
-
-  it('emits RUN_FINISHED on done when there was no usage event', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    assert.ok(find(events, 'RUN_FINISHED'))
-  })
-
-  it('usage event closes any open text message before RUN_FINISHED', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'hi' } as AgentStreamEvent)
-    wrapped.send({
-      type: 'usage',
-      tokens: { input: 1, output: 1 },
-      model: 'm',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(t.indexOf('TEXT_MESSAGE_END') < t.indexOf('RUN_FINISHED'))
-  })
-
-  it('emits RUN_ERROR and closes open text on error', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'partial' } as AgentStreamEvent)
-    wrapped.send({
-      type: 'error',
-      message: 'network timeout',
-    } as AgentStreamEvent)
-
-    const err = find(events, 'RUN_ERROR')
-    assert.ok(err)
-    assert.equal(err.message, 'network timeout')
-    const t = types(events)
-    assert.ok(t.indexOf('TEXT_MESSAGE_END') < t.indexOf('RUN_ERROR'))
-  })
-
-  it('emits RUN_ERROR and closes open thinking message on error', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'reasoning-delta',
-      text: 'thinking',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'error', message: 'fail' } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(t.indexOf('THINKING_TEXT_MESSAGE_END') < t.indexOf('RUN_ERROR'))
-  })
-})
-
-describe('wrapChannelWithAGUI — step events', () => {
-  it('emits STEP_STARTED for step-start with the agent in the step name', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'step-start',
-      stepNumber: 1,
-      agent: 'researcher',
-    } as AgentStreamEvent)
-
-    const s = find(events, 'STEP_STARTED')
-    assert.ok(s)
-    assert.ok(
-      typeof s.stepName === 'string' && s.stepName.includes('researcher')
-    )
-  })
-
-  it('emits STEP_STARTED with a defined stepName when agent is absent', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'step-start', stepNumber: 1 } as AgentStreamEvent)
-
-    const s = find(events, 'STEP_STARTED')
-    assert.ok(s)
-    assert.ok(typeof s.stepName === 'string' && s.stepName.length > 0)
-  })
-})
-
-describe('wrapChannelWithAGUI — Pikku CUSTOM events', () => {
-  it('emits CUSTOM pikku:approval-request with all fields', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'approval-request',
-      toolCallId: 'tc1',
-      toolName: 'deleteUser',
-      args: { userId: 99 },
-      reason: 'irreversible action',
-      agent: 'admin-agent',
-      session: 'sess-1',
-    } as AgentStreamEvent)
-
-    const c = find(events, 'CUSTOM', 'pikku:approval-request')
-    assert.ok(c)
-    assert.equal(c.value.toolCallId, 'tc1')
-    assert.equal(c.value.toolName, 'deleteUser')
-    assert.deepEqual(c.value.args, { userId: 99 })
-    assert.equal(c.value.reason, 'irreversible action')
-    assert.equal(c.value.agent, 'admin-agent')
-    assert.equal(c.value.session, 'sess-1')
-  })
-
-  it('closes open text message before emitting approval-request', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'text-delta',
-      text: 'about to ask',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'approval-request',
-      toolCallId: 'tc1',
-      toolName: 'fn',
-      args: {},
-    } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(t.indexOf('TEXT_MESSAGE_END') < t.indexOf('CUSTOM'))
-  })
-
-  it('emits CUSTOM pikku:credential-request with all fields', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'credential-request',
-      toolCallId: 'tc2',
-      toolName: 'githubSearch',
-      args: { repo: 'pikku' },
-      credentialName: 'github-token',
-      credentialType: 'oauth2',
-      connectUrl: 'https://github.com/oauth',
-      runId: 'run-abc',
-      agent: 'dev-agent',
-    } as AgentStreamEvent)
-
-    const c = find(events, 'CUSTOM', 'pikku:credential-request')
-    assert.ok(c)
-    assert.equal(c.value.credentialName, 'github-token')
-    assert.equal(c.value.credentialType, 'oauth2')
-    assert.equal(c.value.connectUrl, 'https://github.com/oauth')
-    assert.equal(c.value.runId, 'run-abc')
-    assert.equal(c.value.toolCallId, 'tc2')
-  })
-
-  it('emits CUSTOM pikku:generative-ui', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'generative-ui',
-      spec: { component: 'Chart', props: { data: [1, 2] } },
-    } as AgentStreamEvent)
-
-    const c = find(events, 'CUSTOM', 'pikku:generative-ui')
-    assert.ok(c)
-    assert.deepEqual(c.value.spec, {
-      component: 'Chart',
-      props: { data: [1, 2] },
-    })
-  })
-
-  it('emits CUSTOM pikku:data', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'data',
-      name: 'search-results',
-      data: [{ title: 'foo' }],
-    } as AgentStreamEvent)
-
-    const c = find(events, 'CUSTOM', 'pikku:data')
-    assert.ok(c)
-    assert.equal(c.value.name, 'search-results')
-    assert.deepEqual(c.value.data, [{ title: 'foo' }])
-  })
-
-  it('emits CUSTOM pikku:agent-call', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'agent-call',
-      agentName: 'sub-agent',
-      session: 'sess-2',
-      input: { query: 'hello' },
-    } as AgentStreamEvent)
-
-    const c = find(events, 'CUSTOM', 'pikku:agent-call')
-    assert.ok(c)
-    assert.equal(c.value.agentName, 'sub-agent')
-    assert.equal(c.value.session, 'sess-2')
-    assert.deepEqual(c.value.input, { query: 'hello' })
-  })
-
-  it('emits CUSTOM pikku:agent-result', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'agent-result',
-      agentName: 'sub-agent',
-      session: 'sess-2',
-      result: 'done',
-    } as AgentStreamEvent)
-
-    const c = find(events, 'CUSTOM', 'pikku:agent-result')
-    assert.ok(c)
-    assert.equal(c.value.agentName, 'sub-agent')
-    assert.equal(c.value.result, 'done')
-  })
-
-  it('emits CUSTOM pikku:suspended', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'suspended',
-      reason: 'rpc-missing',
-      missingRpcs: ['stripe.charge'],
-    } as AgentStreamEvent)
-
-    const c = find(events, 'CUSTOM', 'pikku:suspended')
-    assert.ok(c)
-    assert.equal(c.value.reason, 'rpc-missing')
-    assert.deepEqual(c.value.missingRpcs, ['stripe.charge'])
-  })
-})
-
-describe('wrapChannelWithAGUI — speech', () => {
-  it('forwards audio-delta as a CUSTOM event', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'audio-delta',
-      data: 'base64abc',
-      format: 'mp3',
-    } as AgentStreamEvent)
-
-    // AG-UI has no speech event, but dropping it makes a voice agent reached
-    // over HTTP inaudible: every sentence is synthesized and billed, then
-    // thrown away here.
-    const custom = find(events, 'CUSTOM', 'pikku:audio-delta')
-    assert.ok(custom)
-    assert.deepEqual(custom.value, { data: 'base64abc', format: 'mp3' })
-  })
-
-  it('forwards audio-done as a CUSTOM event', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'audio-done' } as AgentStreamEvent)
-
-    assert.ok(find(events, 'CUSTOM', 'pikku:audio-done'))
-  })
-
-  it('forwards the transcript as a CUSTOM event', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'transcript',
-      text: 'what the user said',
-    } as AgentStreamEvent)
-
-    // The client sent audio, so this is the only way it learns what its own
-    // turn was. Dropped, the user's message renders as an empty bubble.
-    const custom = find(events, 'CUSTOM', 'pikku:transcript')
-    assert.ok(custom)
-    assert.deepEqual(custom.value, { text: 'what the user said' })
-  })
-
-  it('leaves an open text message open, so speech cannot split a reply', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'text-delta', text: 'hello' } as AgentStreamEvent)
-    wrapped.send({
-      type: 'audio-delta',
-      data: 'abc',
-      format: 'mp3',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'text-delta', text: ' there' } as AgentStreamEvent)
-
-    // Audio arrives interleaved with the text it is saying. Ending the message
-    // around it would render one reply as several.
-    const starts = events.filter(
-      (event: any) => event.type === 'TEXT_MESSAGE_START'
-    )
-    assert.equal(starts.length, 1)
-  })
-})
-
-describe('wrapChannelWithAGUI — channel proxy', () => {
-  it('exposes the inner channelId', () => {
-    const { channel } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-    assert.equal(wrapped.channelId, 'test-id')
-  })
-
-  it('delegates setState / getState / clearState to inner', () => {
-    let stored: unknown
-    const { channel } = makeChannel()
-    channel.setState = (v) => {
-      stored = v
-    }
-    channel.getState = () => stored as any
-    channel.clearState = () => {
-      stored = undefined
-    }
-
-    const wrapped = wrapChannelWithAGUI(channel)
-    wrapped.setState('my-state')
-    assert.equal(wrapped.getState(), 'my-state')
-    wrapped.clearState()
-    assert.equal(wrapped.getState(), undefined)
-  })
-
-  it('delegates sendBinary to inner', () => {
-    const binaries: unknown[] = []
-    const { channel } = makeChannel()
-    channel.sendBinary = (d) => binaries.push(d)
-
-    const wrapped = wrapChannelWithAGUI(channel)
-    wrapped.sendBinary(new Uint8Array([1, 2, 3]))
-    assert.equal(binaries.length, 1)
-  })
-
-  it('delegates close to inner', () => {
-    let closed = false
-    const { channel } = makeChannel()
-    channel.close = () => {
-      closed = true
-    }
-
-    const wrapped = wrapChannelWithAGUI(channel)
-    wrapped.close()
-    assert.ok(closed)
-  })
-
-  it('reflects inner state property', () => {
-    const { channel } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-    assert.equal(wrapped.state, 'open')
-    channel.state = 'closed'
-    assert.equal(wrapped.state, 'closed')
-  })
-})
-
-function assertClientAcceptsStream(events: any[]): void {
-  let started = false
-  let finished = false
-  let errored = false
-  const activeTexts = new Set<string>()
-  const activeTools = new Set<string>()
-  const activeSteps = new Set<string>()
-  let thinkingStep = false
-  let thinkingMessage = false
-
-  for (const e of events) {
-    const parsed = EventSchemas.safeParse(e)
-    assert.ok(
-      parsed.success,
-      `event failed @ag-ui/core schema: ${JSON.stringify(e)} — ${JSON.stringify(parsed.success ? [] : parsed.error.issues)}`
-    )
-    assert.ok(!errored, `event '${e.type}' sent after RUN_ERROR`)
-    assert.ok(
-      !finished || e.type === 'RUN_ERROR' || e.type === 'RUN_STARTED',
-      `event '${e.type}' sent after RUN_FINISHED`
-    )
-    if (!started) {
-      started = true
-      assert.ok(
-        e.type === 'RUN_STARTED' || e.type === 'RUN_ERROR',
-        `first event must be RUN_STARTED, got '${e.type}'`
-      )
-    }
-    switch (e.type) {
-      case 'TEXT_MESSAGE_START':
-        assert.ok(!activeTexts.has(e.messageId), 'text already active')
-        activeTexts.add(e.messageId)
-        break
-      case 'TEXT_MESSAGE_CONTENT':
-        assert.ok(activeTexts.has(e.messageId), 'text content without start')
-        break
-      case 'TEXT_MESSAGE_END':
-        assert.ok(activeTexts.delete(e.messageId), 'text end without start')
-        break
-      case 'TOOL_CALL_START':
-        assert.ok(!activeTools.has(e.toolCallId), 'tool already active')
-        activeTools.add(e.toolCallId)
-        break
-      case 'TOOL_CALL_ARGS':
-        assert.ok(activeTools.has(e.toolCallId), 'tool args without start')
-        break
-      case 'TOOL_CALL_END':
-        assert.ok(activeTools.delete(e.toolCallId), 'tool end without start')
-        break
-      case 'STEP_STARTED':
-        assert.ok(
-          !activeSteps.has(e.stepName),
-          `step '${e.stepName}' already active`
-        )
-        activeSteps.add(e.stepName)
-        break
-      case 'STEP_FINISHED':
-        assert.ok(
-          activeSteps.delete(e.stepName),
-          `step '${e.stepName}' not started`
-        )
-        break
-      case 'THINKING_START':
-        assert.ok(!thinkingStep, 'thinking step already active')
-        thinkingStep = true
-        break
-      case 'THINKING_END':
-        assert.ok(thinkingStep, 'thinking end without start')
-        thinkingStep = false
-        break
-      case 'THINKING_TEXT_MESSAGE_START':
-        assert.ok(thinkingStep, 'thinking message outside THINKING_START')
-        assert.ok(!thinkingMessage, 'thinking message already active')
-        thinkingMessage = true
-        break
-      case 'THINKING_TEXT_MESSAGE_CONTENT':
-        assert.ok(thinkingMessage, 'thinking content without start')
-        break
-      case 'THINKING_TEXT_MESSAGE_END':
-        assert.ok(thinkingMessage, 'thinking message end without start')
-        thinkingMessage = false
-        break
-      case 'RUN_FINISHED':
-        assert.equal(activeSteps.size, 0, 'RUN_FINISHED with active steps')
-        assert.equal(activeTexts.size, 0, 'RUN_FINISHED with active text')
-        assert.equal(activeTools.size, 0, 'RUN_FINISHED with active tools')
-        finished = true
-        break
-      case 'RUN_ERROR':
-        errored = true
-        break
-    }
-  }
-}
-
-describe('wrapChannelWithAGUI — AG-UI protocol conformance', () => {
-  it('emits RUN_STARTED with threadId and runId as the first event', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel, {
-      threadId: 'thread-1',
-      runId: 'run-1',
-    })
-
-    wrapped.send({ type: 'text-delta', text: 'hi' } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const first = events[0] as any
-    assert.equal(first.type, 'RUN_STARTED')
-    assert.equal(first.threadId, 'thread-1')
-    assert.equal(first.runId, 'run-1')
-  })
-
-  it('carries the same threadId and runId on RUN_FINISHED', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel, {
-      threadId: 'thread-1',
-      runId: 'run-1',
-    })
-
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const f = find(events, 'RUN_FINISHED')
-    assert.equal(f.threadId, 'thread-1')
-    assert.equal(f.runId, 'run-1')
-  })
-
-  it('does not emit RUN_FINISHED on mid-run usage; accumulates into final RUN_FINISHED', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'step-start', stepNumber: 0 } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'listTodos',
-      args: {},
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-result',
-      toolCallId: 'tc1',
-      toolName: 'listTodos',
-      result: ['a'],
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'usage',
-      tokens: { input: 10, output: 5 },
-      model: 'gpt-4o',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'step-start', stepNumber: 1 } as AgentStreamEvent)
-    wrapped.send({
-      type: 'text-delta',
-      text: 'You have 1 todo',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'usage',
-      tokens: { input: 20, output: 8 },
-      model: 'gpt-4o',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const finishes = (events as any[]).filter((e) => e.type === 'RUN_FINISHED')
-    assert.equal(finishes.length, 1)
-    assert.equal((events as any[]).at(-1).type, 'RUN_FINISHED')
-    assert.equal(finishes[0].usage.promptTokens, 30)
-    assert.equal(finishes[0].usage.completionTokens, 13)
-    assert.equal(finishes[0].usage.totalTokens, 43)
-    const t = types(events)
-    assert.ok(
-      t.lastIndexOf('TEXT_MESSAGE_CONTENT') < t.indexOf('RUN_FINISHED'),
-      'text must arrive before RUN_FINISHED'
-    )
-    assertClientAcceptsStream(events as any[])
-  })
-
-  it('swallows events after done so nothing follows RUN_FINISHED', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-    wrapped.send({
-      type: 'usage',
-      tokens: { input: 1, output: 1 },
-      model: 'm',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'text-delta', text: 'late' } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    assert.equal((events as any[]).at(-1).type, 'RUN_FINISHED')
-    assert.equal(events.filter((e: any) => e.type === 'RUN_FINISHED').length, 1)
-  })
-
-  it('gives TOOL_CALL_RESULT a messageId', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'tool-result',
-      toolCallId: 'tc1',
-      toolName: 'fn',
-      result: 'ok',
-    } as AgentStreamEvent)
-
-    const r = find(events, 'TOOL_CALL_RESULT')
-    assert.ok(r.messageId, 'TOOL_CALL_RESULT requires a messageId')
-  })
-
-  it('streams tool args via TOOL_CALL_ARGS between START and END', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'addTodo',
-      args: { text: 'buy milk' },
-    } as AgentStreamEvent)
-
-    const t = types(events)
-    const argsIdx = t.indexOf('TOOL_CALL_ARGS')
-    assert.ok(argsIdx !== -1, 'TOOL_CALL_ARGS must be emitted')
-    assert.ok(t.indexOf('TOOL_CALL_START') < argsIdx)
-    assert.ok(argsIdx < t.indexOf('TOOL_CALL_END'))
-    const args = find(events, 'TOOL_CALL_ARGS')
-    assert.equal(args.delta, '{"text":"buy milk"}')
-  })
-
-  it('wraps thinking messages in THINKING_START / THINKING_END', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'reasoning-delta', text: 'hmm' } as AgentStreamEvent)
-    wrapped.send({ type: 'text-delta', text: 'answer' } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(
-      t.indexOf('THINKING_START') < t.indexOf('THINKING_TEXT_MESSAGE_START')
-    )
-    assert.ok(
-      t.indexOf('THINKING_TEXT_MESSAGE_END') < t.indexOf('THINKING_END')
-    )
-    assertClientAcceptsStream(events as any[])
-  })
-
-  it('closes the open step before RUN_FINISHED and keeps step names unique', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'step-start', stepNumber: 0 } as AgentStreamEvent)
-    wrapped.send({
-      type: 'step-start',
-      stepNumber: 0,
-      agent: 'todoAgent',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'step-start', stepNumber: 1 } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const startNames = (events as any[])
-      .filter((e) => e.type === 'STEP_STARTED')
-      .map((e) => e.stepName)
-    assert.equal(new Set(startNames).size, startNames.length)
-    assertClientAcceptsStream(events as any[])
-  })
-
-  it('accepts the full multi-agent delegate wire sequence', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel, {
-      threadId: 't1',
-      runId: 'r1',
-    })
-
-    const send = (e: any) => wrapped.send(e as AgentStreamEvent)
-    send({ type: 'step-start', stepNumber: 0 })
-    send({
-      type: 'tool-call',
-      toolCallId: 'p1',
-      toolName: 'todoAgent',
-      args: { message: 'list' },
-    })
-    send({
-      type: 'agent-call',
-      agentName: 'todoAgent',
-      session: 's',
-      input: 'list',
-    })
-    send({ type: 'step-start', stepNumber: 0, agent: 'todoAgent' })
-    send({
-      type: 'tool-call',
-      toolCallId: 'c1',
-      toolName: 'todos__listTodos',
-      args: {},
-    })
-    send({
-      type: 'tool-result',
-      toolCallId: 'c1',
-      toolName: 'todos__listTodos',
-      result: ['x'],
-    })
-    send({
-      type: 'usage',
-      tokens: { input: 5, output: 2 },
-      model: 'gpt-4o-mini',
-    })
-    send({ type: 'step-start', stepNumber: 1, agent: 'todoAgent' })
-    for (const chunk of ['You ', 'have ', 'one ', 'todo']) {
-      send({ type: 'text-delta', text: chunk })
-    }
-    send({
-      type: 'usage',
-      tokens: { input: 7, output: 9 },
-      model: 'gpt-4o-mini',
-    })
-    send({
-      type: 'agent-result',
-      agentName: 'todoAgent',
-      session: 's',
-      result: 'You have one todo',
-    })
-    send({
-      type: 'tool-result',
-      toolCallId: 'p1',
-      toolName: 'todoAgent',
-      result: 'You have one todo',
-    })
-    send({
-      type: 'usage',
-      tokens: { input: 11, output: 3 },
-      model: 'gpt-4o-mini',
-    })
-    send({ type: 'step-start', stepNumber: 1 })
-    send({
-      type: 'usage',
-      tokens: { input: 4, output: 1 },
-      model: 'gpt-4o-mini',
-    })
-    send({ type: 'done' })
-
-    assertClientAcceptsStream(events as any[])
-    assert.equal((events as any[]).at(-1).type, 'RUN_FINISHED')
-    assert.ok(find(events, 'CUSTOM', 'pikku:agent-call'))
-    assert.ok(find(events, 'CUSTOM', 'pikku:agent-result'))
-    assert.equal(
-      (events as any[]).filter((e) => e.type === 'TEXT_MESSAGE_CONTENT').length,
-      4
-    )
-  })
-
-  it('approval flow ends with CUSTOM approval-request then RUN_FINISHED', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({ type: 'step-start', stepNumber: 0 } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'addTodo',
-      args: { text: 'x' },
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'approval-request',
-      toolCallId: 'tc1',
-      toolName: 'addTodo',
-      args: { text: 'x' },
-      reason: 'Add a todo',
-      runId: 'run-9',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    assertClientAcceptsStream(events as any[])
-    const c = find(events, 'CUSTOM', 'pikku:approval-request')
-    assert.equal(c.value.runId, 'run-9')
-    const t = types(events)
-    assert.ok(t.indexOf('CUSTOM') < t.indexOf('RUN_FINISHED'))
-  })
-})
-
-describe('wrapChannelWithAGUI — realistic full turn', () => {
-  it('handles think → text → tool-call → tool-result → usage → done sequence', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel)
-
-    wrapped.send({
-      type: 'reasoning-delta',
-      text: 'let me search',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'text-delta',
-      text: 'I will look that up.',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-call',
-      toolCallId: 'tc1',
-      toolName: 'search',
-      args: { q: 'pikku' },
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'tool-result',
-      toolCallId: 'tc1',
-      toolName: 'search',
-      result: 'found it',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'text-delta',
-      text: 'Here is what I found.',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'usage',
-      tokens: { input: 200, output: 80 },
-      model: 'gpt-4o',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const t = types(events)
-    assert.ok(t.includes('THINKING_TEXT_MESSAGE_START'))
-    assert.ok(t.includes('THINKING_TEXT_MESSAGE_END'))
-    assert.ok(t.includes('TEXT_MESSAGE_START'))
-    assert.ok(t.includes('TEXT_MESSAGE_END'))
-    assert.ok(t.includes('TOOL_CALL_START'))
-    assert.ok(t.includes('TOOL_CALL_END'))
-    assert.ok(t.includes('TOOL_CALL_RESULT'))
-    assert.ok(t.includes('RUN_FINISHED'))
-    assert.equal(t.filter((x) => x === 'RUN_FINISHED').length, 1)
-    assert.equal(t.filter((x) => x === 'TEXT_MESSAGE_START').length, 2)
-  })
-})
-
-describe('wrapChannelWithAGUI — late-bound runId', () => {
-  it('resolves getRunId at the first event, not at wrap time', () => {
-    const { channel, events } = makeChannel()
-    let runId: string | undefined
-    const wrapped = wrapChannelWithAGUI(channel, {
-      threadId: 't1',
-      getRunId: () => runId,
-    })
-
-    runId = 'run-from-state-service'
-    wrapped.send({ type: 'text-delta', text: 'hi' } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    const started = find(events, 'RUN_STARTED')
-    const finished = find(events, 'RUN_FINISHED')
-    assert.equal(started.runId, 'run-from-state-service')
-    assert.equal(finished.runId, 'run-from-state-service')
-  })
-
-  it('prefers an explicit runId over getRunId and falls back to a UUID', () => {
-    const explicit = makeChannel()
-    wrapChannelWithAGUI(explicit.channel, {
-      runId: 'explicit-run',
-      getRunId: () => 'ignored',
-    }).send({ type: 'done' } as AgentStreamEvent)
-    assert.equal(find(explicit.events, 'RUN_STARTED').runId, 'explicit-run')
-
-    const fallback = makeChannel()
-    wrapChannelWithAGUI(fallback.channel, {
-      getRunId: () => undefined,
-    }).send({ type: 'done' } as AgentStreamEvent)
-    const started = find(fallback.events, 'RUN_STARTED')
-    assert.ok(started.runId)
-    assert.equal(started.runId, find(fallback.events, 'RUN_FINISHED').runId)
-  })
-})
-
-describe('wrapChannelWithAGUI — interruption', () => {
-  it('closes the open text message and forwards the fragment as a custom event', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel, { threadId: 't', runId: 'r' })
-
-    wrapped.send({
-      type: 'text-delta',
-      text: 'I will delete the ',
-    } as AgentStreamEvent)
-    wrapped.send({
-      type: 'interrupted',
-      runId: 'r',
-      text: 'I will delete the ',
-      reason: 'speech',
-    } as AgentStreamEvent)
-
-    // Leaving TEXT_MESSAGE_END unsent would strand the assistant bubble in a
-    // streaming state that never resolves.
-    assert.ok(types(events).includes('TEXT_MESSAGE_END'))
-    const custom = find(events, 'CUSTOM', 'pikku:interrupted')
-    assert.deepEqual(custom.value, {
-      runId: 'r',
-      text: 'I will delete the ',
-      reason: 'speech',
-    })
-  })
-
-  it('still ends the run exactly once, so an interrupted run is not terminal on its own', () => {
-    const { channel, events } = makeChannel()
-    const wrapped = wrapChannelWithAGUI(channel, { threadId: 't', runId: 'r' })
-
-    wrapped.send({ type: 'text-delta', text: 'partial' } as AgentStreamEvent)
-    wrapped.send({
-      type: 'interrupted',
-      runId: 'r',
-      text: 'partial',
-      reason: 'speech',
-    } as AgentStreamEvent)
-    wrapped.send({ type: 'done' } as AgentStreamEvent)
-
-    assert.equal(events.filter((e: any) => e.type === 'RUN_FINISHED').length, 1)
-    assert.equal(events.filter((e: any) => e.type === 'RUN_ERROR').length, 0)
-  })
-})