88from a2a .types import (
99 AgentCapabilities ,
1010 AgentCard ,
11- MessageSendConfiguration ,
1211 Message ,
12+ MessageSendConfiguration ,
1313 Part ,
1414 Role ,
1515 Task ,
@@ -130,7 +130,12 @@ async def test_send_message_uses_callsite_configuration_partial_override_non_str
130130 )
131131
132132 cfg = MessageSendConfiguration (history_length = 2 )
133- events = [ev async for ev in base_client .send_message (sample_message , configuration = cfg )]
133+ events = [
134+ ev
135+ async for ev in base_client .send_message (
136+ sample_message , configuration = cfg
137+ )
138+ ]
134139
135140 mock_transport .send_message .assert_called_once ()
136141 assert not mock_transport .send_message_streaming .called
@@ -139,7 +144,10 @@ async def test_send_message_uses_callsite_configuration_partial_override_non_str
139144 params = mock_transport .send_message .await_args .args [0 ]
140145 assert params .configuration .history_length == 2
141146 assert params .configuration .blocking == (not base_client ._config .polling )
142- assert params .configuration .accepted_output_modes == base_client ._config .accepted_output_modes
147+ assert (
148+ params .configuration .accepted_output_modes
149+ == base_client ._config .accepted_output_modes
150+ )
143151
144152
145153@pytest .mark .asyncio
@@ -154,15 +162,23 @@ async def test_send_message_ignores_none_fields_in_callsite_configuration_non_st
154162 )
155163
156164 cfg = MessageSendConfiguration (history_length = None , blocking = None )
157- events = [ev async for ev in base_client .send_message (sample_message , configuration = cfg )]
165+ events = [
166+ ev
167+ async for ev in base_client .send_message (
168+ sample_message , configuration = cfg
169+ )
170+ ]
158171
159172 mock_transport .send_message .assert_called_once ()
160173 assert len (events ) == 1 and events [0 ][0 ].id == 'task-cfg-ns-2'
161174
162175 params = mock_transport .send_message .await_args .args [0 ]
163176 assert params .configuration .history_length is None
164177 assert params .configuration .blocking == (not base_client ._config .polling )
165- assert params .configuration .accepted_output_modes == base_client ._config .accepted_output_modes
178+ assert (
179+ params .configuration .accepted_output_modes
180+ == base_client ._config .accepted_output_modes
181+ )
166182
167183
168184@pytest .mark .asyncio
@@ -182,7 +198,12 @@ async def create_stream(*args, **kwargs):
182198 mock_transport .send_message_streaming .return_value = create_stream ()
183199
184200 cfg = MessageSendConfiguration (history_length = 0 )
185- events = [ev async for ev in base_client .send_message (sample_message , configuration = cfg )]
201+ events = [
202+ ev
203+ async for ev in base_client .send_message (
204+ sample_message , configuration = cfg
205+ )
206+ ]
186207
187208 mock_transport .send_message_streaming .assert_called_once ()
188209 assert not mock_transport .send_message .called
@@ -193,5 +214,7 @@ async def create_stream(*args, **kwargs):
193214 params = mock_transport .send_message_streaming .call_args .args [0 ]
194215 assert params .configuration .history_length == 0
195216 assert params .configuration .blocking == (not base_client ._config .polling )
196- assert params .configuration .accepted_output_modes == base_client ._config .accepted_output_modes
197-
217+ assert (
218+ params .configuration .accepted_output_modes
219+ == base_client ._config .accepted_output_modes
220+ )
0 commit comments