File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -157,7 +157,7 @@ async def test_send_message_callsite_config_overrides_non_streaming(
157157 task , _ = events [0 ]
158158 assert task .id == 'task-cfg-ns-1'
159159
160- params = mock_transport .send_message .await_args . args [0 ]
160+ params = mock_transport .send_message .call_args [ 0 ] [0 ]
161161 assert params .configuration .history_length == 2
162162 assert params .configuration .blocking is False
163163 assert params .configuration .accepted_output_modes == ['application/json' ]
@@ -197,7 +197,7 @@ async def create_stream(*args, **kwargs):
197197 task , _ = events [0 ]
198198 assert task .id == 'task-cfg-s-1'
199199
200- params = mock_transport .send_message_streaming .call_args . args [0 ]
200+ params = mock_transport .send_message_streaming .call_args [ 0 ] [0 ]
201201 assert params .configuration .history_length == 0
202202 assert params .configuration .blocking is True
203203 assert params .configuration .accepted_output_modes == ['text/plain' ]
You can’t perform that action at this time.
0 commit comments