@@ -93,6 +93,26 @@ def __stream__(self) -> Iterator[_T]:
9393 or sse .event == "content_block_delta"
9494 or sse .event == "content_block_stop"
9595 or sse .event == "message"
96+ or sse .event == "user.message"
97+ or sse .event == "user.interrupt"
98+ or sse .event == "user.tool_confirmation"
99+ or sse .event == "user.custom_tool_result"
100+ or sse .event == "agent.message"
101+ or sse .event == "agent.thinking"
102+ or sse .event == "agent.tool_use"
103+ or sse .event == "agent.tool_result"
104+ or sse .event == "agent.mcp_tool_use"
105+ or sse .event == "agent.mcp_tool_result"
106+ or sse .event == "agent.custom_tool_use"
107+ or sse .event == "agent.thread_context_compacted"
108+ or sse .event == "session.status_running"
109+ or sse .event == "session.status_idle"
110+ or sse .event == "session.status_rescheduled"
111+ or sse .event == "session.status_terminated"
112+ or sse .event == "session.error"
113+ or sse .event == "session.deleted"
114+ or sse .event == "span.model_request_start"
115+ or sse .event == "span.model_request_end"
96116 ):
97117 data = sse .json ()
98118 if is_dict (data ) and "type" not in data :
@@ -214,6 +234,26 @@ async def __stream__(self) -> AsyncIterator[_T]:
214234 or sse .event == "content_block_delta"
215235 or sse .event == "content_block_stop"
216236 or sse .event == "message"
237+ or sse .event == "user.message"
238+ or sse .event == "user.interrupt"
239+ or sse .event == "user.tool_confirmation"
240+ or sse .event == "user.custom_tool_result"
241+ or sse .event == "agent.message"
242+ or sse .event == "agent.thinking"
243+ or sse .event == "agent.tool_use"
244+ or sse .event == "agent.tool_result"
245+ or sse .event == "agent.mcp_tool_use"
246+ or sse .event == "agent.mcp_tool_result"
247+ or sse .event == "agent.custom_tool_use"
248+ or sse .event == "agent.thread_context_compacted"
249+ or sse .event == "session.status_running"
250+ or sse .event == "session.status_idle"
251+ or sse .event == "session.status_rescheduled"
252+ or sse .event == "session.status_terminated"
253+ or sse .event == "session.error"
254+ or sse .event == "session.deleted"
255+ or sse .event == "span.model_request_start"
256+ or sse .event == "span.model_request_end"
217257 ):
218258 data = sse .json ()
219259 if is_dict (data ) and "type" not in data :
0 commit comments