Skip to content

Commit cab68ed

Browse files
committed
style(client): apply linter formatting to BaseClient and related tests
1 parent 41f4456 commit cab68ed

File tree

2 files changed

+34
-9
lines changed

2 files changed

+34
-9
lines changed

src/a2a/client/base_client.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,9 @@ async def send_message(
7373
),
7474
)
7575
if configuration is not None:
76-
overrides = configuration.model_dump(exclude_unset=True, exclude_none=True)
76+
overrides = configuration.model_dump(
77+
exclude_unset=True, exclude_none=True
78+
)
7779
config = base_config.model_copy(update=overrides)
7880
else:
7981
config = base_config

tests/client/test_base_client.py

Lines changed: 31 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
from 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

Comments
 (0)