From f07abcc09c0445a4646b8ef97469e7195e0ff352 Mon Sep 17 00:00:00 2001 From: "Xiang (Sean) Zhou" Date: Mon, 2 Jun 2025 11:14:36 -0700 Subject: [PATCH] chore: fix ut for test_functions_request_euc.py PiperOrigin-RevId: 766268126 --- tests/unittests/flows/llm_flows/test_functions_request_euc.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/unittests/flows/llm_flows/test_functions_request_euc.py b/tests/unittests/flows/llm_flows/test_functions_request_euc.py index 6f8f112..afb3b73 100644 --- a/tests/unittests/flows/llm_flows/test_functions_request_euc.py +++ b/tests/unittests/flows/llm_flows/test_functions_request_euc.py @@ -360,6 +360,9 @@ def test_function_get_auth_response_partial(): [ types.Part.from_text(text='response2'), ], + [ + types.Part.from_text(text='final response'), + ], ] mock_model = testing_utils.MockModel.create(responses=responses)