| 84 | return SSEDecoder().iter_bytes(response.iter_bytes()) |
| 85 | |
| 86 | def __stream__(self) -> Iterator[_T]: |
| 87 | cast_to = cast(Any, self._cast_to) |
| 88 | response = self.response |
| 89 | process_data = self._client._process_response_data |
| 90 | iterator = self._iter_events() |
| 91 | |
| 92 | try: |
| 93 | for sse in iterator: |
| 94 | if sse.event == "completion": |
| 95 | yield process_data(data=sse.json(), cast_to=cast_to, response=response) |
| 96 | |
| 97 | if ( |
| 98 | sse.event == "message_start" |
| 99 | or sse.event == "message_delta" |
| 100 | or sse.event == "message_stop" |
| 101 | or sse.event == "content_block_start" |
| 102 | or sse.event == "content_block_delta" |
| 103 | or sse.event == "content_block_stop" |
| 104 | or sse.event == "message" |
| 105 | or sse.event == "user.message" |
| 106 | or sse.event == "user.interrupt" |
| 107 | or sse.event == "user.tool_confirmation" |
| 108 | or sse.event == "user.custom_tool_result" |
| 109 | or sse.event == "user.tool_result" |
| 110 | or sse.event == "agent.message" |
| 111 | or sse.event == "agent.thinking" |
| 112 | or sse.event == "agent.tool_use" |
| 113 | or sse.event == "agent.tool_result" |
| 114 | or sse.event == "agent.mcp_tool_use" |
| 115 | or sse.event == "agent.mcp_tool_result" |
| 116 | or sse.event == "agent.custom_tool_use" |
| 117 | or sse.event == "agent.thread_context_compacted" |
| 118 | or sse.event == "session.status_running" |
| 119 | or sse.event == "session.status_idle" |
| 120 | or sse.event == "session.status_rescheduled" |
| 121 | or sse.event == "session.status_terminated" |
| 122 | or sse.event == "session.error" |
| 123 | or sse.event == "session.deleted" |
| 124 | or sse.event == "session.updated" |
| 125 | or sse.event == "span.model_request_start" |
| 126 | or sse.event == "span.model_request_end" |
| 127 | or sse.event == "span.outcome_evaluation_start" |
| 128 | or sse.event == "span.outcome_evaluation_ongoing" |
| 129 | or sse.event == "span.outcome_evaluation_end" |
| 130 | or sse.event == "user.define_outcome" |
| 131 | or sse.event == "agent.thread_message_received" |
| 132 | or sse.event == "agent.thread_message_sent" |
| 133 | or sse.event == "agent.session_thread_message_received" |
| 134 | or sse.event == "agent.session_thread_message_sent" |
| 135 | or sse.event == "session.thread_created" |
| 136 | or sse.event == "session.thread_status_created" |
| 137 | or sse.event == "session.thread_status_running" |
| 138 | or sse.event == "session.thread_status_idle" |
| 139 | or sse.event == "session.thread_status_rescheduled" |
| 140 | or sse.event == "session.thread_status_terminated" |
| 141 | or sse.event == "event_start" |
| 142 | or sse.event == "event_delta" |
| 143 | or sse.event == "system.message" |