From ca6d99e1adb3c04ba653bd0e38fa2db76a6a6b8a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 16 Sep 2024 17:59:59 -0700 Subject: [PATCH] fix gemini 1.5 flash supports_response_schema --- litellm/tests/test_amazing_vertex_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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), ], )