(self)
| 253 | return SSEDecoder().aiter_bytes(response.aiter_bytes()) |
| 254 | |
| 255 | async def __stream__(self) -> AsyncIterator[_T]: |
| 256 | cast_to = cast(Any, self._cast_to) |
| 257 | response = self.response |
| 258 | process_data = self._client._process_response_data |
| 259 | iterator = self._iter_events() |
| 260 | |
| 261 | try: |
| 262 | async for sse in iterator: |
| 263 | if sse.event == "completion": |
| 264 | yield process_data(data=sse.json(), cast_to=cast_to, response=response) |
| 265 | |
| 266 | if ( |
| 267 | sse.event == "message_start" |
| 268 | or sse.event == "message_delta" |
| 269 | or sse.event == "message_stop" |
| 270 | or sse.event == "content_block_start" |
| 271 | or sse.event == "content_block_delta" |
| 272 | or sse.event == "content_block_stop" |
| 273 | or sse.event == "message" |
| 274 | or sse.event == "user.message" |
| 275 | or sse.event == "user.interrupt" |
| 276 | or sse.event == "user.tool_confirmation" |
| 277 | or sse.event == "user.custom_tool_result" |
| 278 | or sse.event == "user.tool_result" |
| 279 | or sse.event == "agent.message" |
| 280 | or sse.event == "agent.thinking" |
| 281 | or sse.event == "agent.tool_use" |
| 282 | or sse.event == "agent.tool_result" |
| 283 | or sse.event == "agent.mcp_tool_use" |
| 284 | or sse.event == "agent.mcp_tool_result" |
| 285 | or sse.event == "agent.custom_tool_use" |
| 286 | or sse.event == "agent.thread_context_compacted" |
| 287 | or sse.event == "session.status_running" |
| 288 | or sse.event == "session.status_idle" |
| 289 | or sse.event == "session.status_rescheduled" |
| 290 | or sse.event == "session.status_terminated" |
| 291 | or sse.event == "session.error" |
| 292 | or sse.event == "session.deleted" |
| 293 | or sse.event == "session.updated" |
| 294 | or sse.event == "span.model_request_start" |
| 295 | or sse.event == "span.model_request_end" |
| 296 | or sse.event == "span.outcome_evaluation_start" |
| 297 | or sse.event == "span.outcome_evaluation_ongoing" |
| 298 | or sse.event == "span.outcome_evaluation_end" |
| 299 | or sse.event == "user.define_outcome" |
| 300 | or sse.event == "agent.thread_message_received" |
| 301 | or sse.event == "agent.thread_message_sent" |
| 302 | or sse.event == "agent.session_thread_message_received" |
| 303 | or sse.event == "agent.session_thread_message_sent" |
| 304 | or sse.event == "session.thread_created" |
| 305 | or sse.event == "session.thread_status_created" |
| 306 | or sse.event == "session.thread_status_running" |
| 307 | or sse.event == "session.thread_status_idle" |
| 308 | or sse.event == "session.thread_status_rescheduled" |
| 309 | or sse.event == "session.thread_status_terminated" |
| 310 | or sse.event == "event_start" |
| 311 | or sse.event == "event_delta" |
| 312 | or sse.event == "system.message" |
no test coverage detected