diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 1a2dab3f6..30bceea7b 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -1398,7 +1398,7 @@ def vertex_httpx_mock_post_invalid_schema_response_anthropic(*args, **kwargs): [ ("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), ], )