From bb786461c76022873eb770802e0bb5dd914cbba0 Mon Sep 17 00:00:00 2001 From: Richard Beales Date: Sun, 23 Apr 2023 07:35:17 +0100 Subject: [PATCH] =?UTF-8?q?Mark=20test=20test=5Fgenerate=5Faiconfig=5Fauto?= =?UTF-8?q?matic=5Ftypical=20=20as=20@requires=5Fapi=5F=E2=80=A6=20(#2981)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Mark test test_generate_aiconfig_automatic_typical as @requires_api_key("OPENAI_API_KEY") * missing import --- tests/unit/test_setup.py | 2 ++ 1 file changed, 2 insertions(+) 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)