Skip to content

Commit 573042c

Browse files
committed
reformat
1 parent 7d240ca commit 573042c

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
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
else None
7474
),
7575
)
76-
params = MessageSendParams(message=request, configuration=config, metadata=request_metadata)
76+
params = MessageSendParams(
77+
message=request, configuration=config, metadata=request_metadata
78+
)
7779

7880
if not self._config.streaming or not self._card.capabilities.streaming:
7981
response = await self._transport.send_message(

tests/client/test_base_client.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,12 +73,14 @@ async def create_stream(*args, **kwargs):
7373

7474
mock_transport.send_message_streaming.return_value = create_stream()
7575

76-
meta = {"test":1}
76+
meta = {'test': 1}
7777
stream = base_client.send_message(sample_message, request_metadata=meta)
7878
events = [event async for event in stream]
7979

8080
mock_transport.send_message_streaming.assert_called_once()
81-
assert mock_transport.send_message_streaming.call_args[0][0].metadata == meta
81+
assert (
82+
mock_transport.send_message_streaming.call_args[0][0].metadata == meta
83+
)
8284
assert not mock_transport.send_message.called
8385
assert len(events) == 1
8486
assert events[0][0].id == 'task-123'
@@ -95,7 +97,7 @@ async def test_send_message_non_streaming(
9597
status=TaskStatus(state=TaskState.completed),
9698
)
9799

98-
meta = {"test":1}
100+
meta = {'test': 1}
99101
stream = base_client.send_message(sample_message, request_metadata=meta)
100102
events = [event async for event in stream]
101103

0 commit comments

Comments
 (0)