diff --git a/tests/unit/test_setup.py b/tests/unit/test_setup.py index 67923dbd7..1e718bf95 100644 --- a/tests/unit/test_setup.py +++ b/tests/unit/test_setup.py @@ -8,6 +8,7 @@ from autogpt.setup import ( generate_aiconfig_manual, prompt_user, ) +from tests.utils import requires_api_key class TestAutoGPT(unittest.TestCase): @@ -22,6 +23,7 @@ class TestAutoGPT(unittest.TestCase): self.assertGreaterEqual(len(ai_config.ai_goals), 1) self.assertLessEqual(len(ai_config.ai_goals), 5) + @requires_api_key("OPENAI_API_KEY") def test_generate_aiconfig_automatic_typical(self): user_prompt = "Help me create a rock opera about cybernetic giraffes" ai_config = generate_aiconfig_automatic(user_prompt)