From fc41f46f0f03ea023698ad5da45731a9ffc9af26 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Mon, 2 Mar 2026 19:43:24 +0530 Subject: [PATCH] Fix vertex ai function calls --- tests/local_testing/test_amazing_vertex_completion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 8974275ca83..4d3b356bac4 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -2899,7 +2899,7 @@ def test_gemini_function_call_parameter_in_messages(): } with patch( - "litellm.llms.vertex_ai.vertex_llm_base.VertexLLMBase._ensure_access_token", + "litellm.llms.vertex_ai.vertex_llm_base.VertexBase._ensure_access_token", return_value=({"Authorization": "Bearer fake"}, "test-project"), ): with patch.object(client, "post", new=MagicMock()) as mock_client: @@ -2943,7 +2943,7 @@ def test_gemini_function_call_parameter_in_messages(): "content": "The current weather in Boston is 22°F." }, } - ] + } ], }, ],