Skip to content

Commit bb78646

Browse files
authored
Mark test test_generate_aiconfig_automatic_typical as @requires_api_… (Significant-Gravitas#2981)
* Mark test test_generate_aiconfig_automatic_typical as @requires_api_key("OPENAI_API_KEY") * missing import
1 parent bc354a3 commit bb78646

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/test_setup.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
generate_aiconfig_manual,
99
prompt_user,
1010
)
11+
from tests.utils import requires_api_key
1112

1213

1314
class TestAutoGPT(unittest.TestCase):
@@ -22,6 +23,7 @@ def test_generate_aiconfig_automatic_default(self):
2223
self.assertGreaterEqual(len(ai_config.ai_goals), 1)
2324
self.assertLessEqual(len(ai_config.ai_goals), 5)
2425

26+
@requires_api_key("OPENAI_API_KEY")
2527
def test_generate_aiconfig_automatic_typical(self):
2628
user_prompt = "Help me create a rock opera about cybernetic giraffes"
2729
ai_config = generate_aiconfig_automatic(user_prompt)

0 commit comments

Comments
 (0)