fix gemini 1.5 flash supports_response_schema

2024-09-16 17:59:59 -07:00 · 2024-09-16 17:59:59 -07:00 · ca6d99e1ad
commit ca6d99e1ad
parent ce7a937a53
1 changed files with 1 additions and 1 deletions
--- 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),
    ],
 )