fix gemini 1.5 flash test

This commit is contained in:
Ishaan Jaff 2024-09-16 19:37:41 -07:00
parent 8762b64b1d
commit 7e9dbcd1a3

View file

@ -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),
],
)