File tree Expand file tree Collapse file tree 2 files changed +8
-4
lines changed
Expand file tree Collapse file tree 2 files changed +8
-4
lines changed Original file line number Diff line number Diff 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 (
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments