fix gemini 1.5 flash supports_response_schema

This commit is contained in:
Ishaan Jaff 2024-09-16 17:59:59 -07:00
parent ce7a937a53
commit ca6d99e1ad

View file

@ -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), ("vertex_ai_beta/gemini-1.5-pro-001", "us-central1", True),
("gemini/gemini-1.5-pro", None, 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), ("vertex_ai/claude-3-5-sonnet@20240620", "us-east5", False),
], ],
) )