Skip to content

feat: AgentLite streaming partial content events#66

Open
DorianZheng wants to merge 3 commits into
mainfrom
feat/streaming-partial-content
Open

feat: AgentLite streaming partial content events#66
DorianZheng wants to merge 3 commits into
mainfrom
feat/streaming-partial-content

Conversation

@DorianZheng
Copy link
Copy Markdown
Member

Adds run.partial_content, run.partial_tool_call, run.partial_content.done, and run.partial_content.interrupted events for token-by-token streaming output.

What's in this PR

  • New event interfaces in src/api/events.ts (4 types)
  • Streaming event extraction in src/agent/message-processor.ts
  • Per-jid in-memory ring buffer (last 2 KB) for reconnection replay
  • resumePartialContent() method: replays buffered deltas or emits buffer_evicted
  • Fix: run.partial_content.done only emits when jid is actively streaming
  • 11 test cases in src/streaming-events.test.ts covering all event types, reconnection, and concurrent streams

DorianZheng and others added 3 commits April 24, 2026 13:07
Adds four new AgentLite events that enable the Dune UI to render
agent responses token-by-token as they arrive from the Claude API:

- run.partial_content — text token delta (content_block_delta/text_delta)
- run.partial_tool_call — tool arg JSON delta (input_json_delta)
- run.partial_content.done — stream completed normally (message_stop)
- run.partial_content.interrupted — stream cut short (container crash/exit)

includePartialMessages:true was already set; this extracts the stream_event
SDK messages and surfaces them as typed events. Tracks in-flight jids via
streamingJids set; drains all on container error or stopped state.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
…ent events

Covers all four new event types (run.partial_content, run.partial_tool_call,
run.partial_content.done, run.partial_content.interrupted):

1. text_delta → run.partial_content with delta, contentBlockIndex, agentId, jid, timestamp
2. input_json_delta → run.partial_tool_call with jsonDelta
3. non-delta stream events (message_start) → no partial events emitted
4. message_stop → run.partial_content.done
5. container error during streaming → run.partial_content.interrupted (container_error)
6. container stopped during streaming → run.partial_content.interrupted (container_exit)
7. contentBlockIndex omitted when streamEvent.index is undefined
8. multiple sequential text_deltas → one event per delta + one done event

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
…ts 6-8

- Add per-jid in-memory ring buffer (last 2 KB) in MessageProcessor
- appendToRingBuffer: appends text/tool deltas; marks evicted when >2 KB
- resumePartialContent: replays buffered deltas on reconnect, or emits
  run.partial_content.interrupted (reason: buffer_evicted) when stream
  completed normally or buffer overflowed
- Fix run.partial_content.done guard: only emit if streamingJids.has(chatJid)
- Clear ring buffer on message_stop (stream completed normally)
- Expose resumePartialContent via AgentImpl.resumePartialContent
- Add test 6: ring-buffer replay replays deltas in order on reconnect
- Add test 7: buffer_evicted emitted when stream completed before reconnect
- Add test 8: two concurrent jids each get own events; drain interrupts both

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant