File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -236,7 +236,7 @@ async def test_batched_function_calls(self):
236236 function_call = genai_types .FunctionCall (
237237 name = 'sleep_async' ,
238238 id = 'sleep_async-2' ,
239- args = {'sleep_seconds' : 0.1 },
239+ args = {'sleep_seconds' : 0.2 },
240240 )
241241 ),
242242 role = 'model' ,
@@ -288,15 +288,17 @@ async def test_batched_function_calls(self):
288288 role = 'user' ,
289289 substream_name = function_calling .FUNCTION_CALL_SUBSTREAM_NAME ,
290290 ),
291+ ]
292+ + model_output_1
293+ + [
291294 content_api .ProcessorPart .from_function_response (
292295 name = 'sleep_async' ,
293296 function_call_id = 'sleep_async-2' ,
294- response = 'Slept for 0.1 seconds' ,
297+ response = 'Slept for 0.2 seconds' ,
295298 role = 'user' ,
296299 substream_name = function_calling .FUNCTION_CALL_SUBSTREAM_NAME ,
297300 ),
298301 ]
299- + model_output_1
300302 + model_output_2 ,
301303 )
302304
You can’t perform that action at this time.
0 commit comments