chore: fix ut for test_functions_request_euc.py

seanzhougoogle · copybara-github · commit f07abcc09c04 · 2025-06-02T11:15:13.000-07:00
PiperOrigin-RevId: 766268126
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -360,6 +360,9 @@ def test_function_get_auth_response_partial():`
`360`	`360`	`[`
`361`	`361`	`types.Part.from_text(text='response2'),`
`362`	`362`	`],`
	`363`	`+ [`
	`364`	`+ types.Part.from_text(text='final response'),`
	`365`	`+ ],`
`363`	`366`	`]`
`364`	`367`
`365`	`368`	`mock_model = testing_utils.MockModel.create(responses=responses)`