diff --git a/tests/test_agent.py b/tests/test_agent.py index e3c3064f..40228285 100644 --- a/tests/test_agent.py +++ b/tests/test_agent.py @@ -817,6 +817,15 @@ def empty(m: list[ModelMessage], _info: AgentInfo) -> ModelResponse: ) +try: + import google.genai # type: ignore # noqa + + skip_gemini_test = False +except ImportError: + skip_gemini_test = True + + +@pytest.mark.skipif(skip_gemini_test, reason='Need google-genai installed.') def test_model_requests_blocked(env: TestEnv): env.set('GEMINI_API_KEY', 'foobar') agent = Agent('google-gla:gemini-1.5-flash', result_type=tuple[str, str], defer_model_check=True)