From c2aec08a1bc7f81187ec567bc2e2b95210d96959 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 30 Mar 2026 17:15:19 -0700 Subject: [PATCH] test: fix test --- tests/local_testing/test_amazing_vertex_completion.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 34d20832c2..d7356183b9 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -3564,8 +3564,9 @@ def test_gemini_tool_calling_working_demo(): }, } ], + "vertex_location": "global", } - response = completion(model="vertex_ai/gemini-2.0-flash", **args) + response = completion(model="vertex_ai/gemini-3-flash-preview", **args) print(response) @@ -3989,7 +3990,7 @@ def test_vertex_ai_gemini_audio_ogg(): @pytest.mark.asyncio async def test_vertex_ai_deepseek(): """Test that deepseek models use the correct v1 API endpoint instead of v1beta1.""" - # load_vertex_ai_credentials() + load_vertex_ai_credentials() litellm._turn_on_debug() from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler @@ -4065,7 +4066,7 @@ def test_gemini_google_maps_tool_simple(): """ Test googleMaps tool with just enableWidget parameter. """ - # load_vertex_ai_credentials() + load_vertex_ai_credentials() litellm._turn_on_debug() tools = [{"googleMaps": {"enableWidget": True}}]