Skip to content

Commit 9d1c597

Browse files
kiberguscopybara-github
authored andcommitted
Fix test_batched_function_calls test.
PiperOrigin-RevId: 881513994
1 parent 06399af commit 9d1c597

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

genai_processors/tests/function_calling_test.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)