test: reduce vertex test to avoid exhausting limits

This commit is contained in:
Krrish Dholakia
2024-06-12 22:32:45 -07:00
parent 4bf33ef267
commit 9c4e6d85ed
@@ -503,7 +503,7 @@ async def test_async_vertexai_streaming_response():
# asyncio.run(test_async_vertexai_streaming_response())
@pytest.mark.parametrize("provider", ["vertex_ai", "vertex_ai_beta"])
@pytest.mark.parametrize("provider", ["vertex_ai"]) # "vertex_ai_beta"
@pytest.mark.parametrize("sync_mode", [True, False])
@pytest.mark.asyncio
async def test_gemini_pro_vision(provider, sync_mode):