diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 30bceea7b..d65954127 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -1506,7 +1506,7 @@ async def test_gemini_pro_json_schema_args_sent_httpx( [ ("vertex_ai_beta/gemini-1.5-pro-001", "us-central1", True), ("gemini/gemini-1.5-pro", None, True), - ("vertex_ai_beta/gemini-1.5-flash", "us-central1", False), + ("vertex_ai_beta/gemini-1.5-flash", "us-central1", True), ("vertex_ai/claude-3-5-sonnet@20240620", "us-east5", False), ], )