Fix vertex ai function calls

2026-03-02 19:43:24 +05:30 · 2026-03-02 19:43:24 +05:30 · fc41f46f0f
commit fc41f46f0f
parent e40e913622
1 changed files with 2 additions and 2 deletions
--- 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."
                                    },
                                }
-                            ]
+                            }
                        ],
                    },
                ],