diff --git a/tests/test_litellm/test_main.py b/tests/test_litellm/test_main.py index 40a3692ac6..f538bc4e2f 100644 --- a/tests/test_litellm/test_main.py +++ b/tests/test_litellm/test_main.py @@ -213,6 +213,8 @@ async def test_url_with_format_param(model, sync_mode, monkeypatch): } ], } + if model.startswith("gemini/"): + args["api_key"] = "test-api-key" with patch.object(client, "post", new=MagicMock()) as mock_client: try: if sync_mode: