Add check on openai key

uber · Jul 5, 2023 · c4b552f · c4b552f
1 parent d90cdc9
commit c4b552f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/experimental/rule_inference/local.py b/experimental/rule_inference/local.py
@@ -1,6 +1,7 @@
 import logging
 import multiprocessing
 import os
+import sys
 import time
 
 import attr
@@ -11,7 +12,9 @@
 from utils.pretty_toml import PrettyTOML
 
 from experimental.rule_inference.piranha_agent import (
-    PiranhaAgent, run_piranha_with_timeout)
+    PiranhaAgent,
+    run_piranha_with_timeout,
+)
 from experimental.rule_inference.rule_application import CodebaseRefactorer
 from experimental.rule_inference.utils.rule_utils import RawRuleGraph
 
@@ -165,4 +168,8 @@ def test_rule(data):
 
 if __name__ == "__main__":
     openai.api_key = os.getenv("OPENAI_API_KEY")
+    if not openai.api_key:
+        sys.exit(
+            "Please set the OPENAI_API_KEY environment variable to your OpenAI API key."
+        )
     app.run(debug=True)