@@ -139,7 +139,9 @@ async def test_send_message_uses_callsite_configuration_partial_override_non_str
139139
140140 mock_transport .send_message .assert_called_once ()
141141 assert not mock_transport .send_message_streaming .called
142- assert len (events ) == 1 and events [0 ][0 ].id == 'task-cfg-ns-1'
142+ assert len (events ) == 1
143+ task , _ = events [0 ]
144+ assert task .id == 'task-cfg-ns-1'
143145
144146 params = mock_transport .send_message .await_args .args [0 ]
145147 assert params .configuration .history_length == 2
@@ -170,7 +172,9 @@ async def test_send_message_ignores_none_fields_in_callsite_configuration_non_st
170172 ]
171173
172174 mock_transport .send_message .assert_called_once ()
173- assert len (events ) == 1 and events [0 ][0 ].id == 'task-cfg-ns-2'
175+ assert len (events ) == 1
176+ task , _ = events [0 ]
177+ assert task .id == 'task-cfg-ns-2'
174178
175179 params = mock_transport .send_message .await_args .args [0 ]
176180 assert params .configuration .history_length is None
@@ -208,8 +212,8 @@ async def create_stream(*args, **kwargs):
208212 mock_transport .send_message_streaming .assert_called_once ()
209213 assert not mock_transport .send_message .called
210214 assert len (events ) == 1
211- first = events [ 0 ][ 0 ] if isinstance ( events [ 0 ], tuple ) else events [0 ]
212- assert first .id == 'task-cfg-s-1'
215+ task , _ = events [0 ]
216+ assert task .id == 'task-cfg-s-1'
213217
214218 params = mock_transport .send_message_streaming .call_args .args [0 ]
215219 assert params .configuration .history_length == 0
0 commit comments