From 301ff85fd57dd018b7446798b924073cbee3eaf1 Mon Sep 17 00:00:00 2001
From: Andres Caicedo <Andrezz1997@gmail.com>
Date: Tue, 4 Apr 2023 00:15:17 +0200
Subject: [PATCH] Update main.py

Introduce check to OPENAI_API_KEY
---
 scripts/main.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/scripts/main.py b/scripts/main.py
index 4e79974369..e334753f37 100644
--- a/scripts/main.py
+++ b/scripts/main.py
@@ -17,6 +17,16 @@ import traceback
 import yaml
 import argparse
 
+def check_openai_api_key():
+    """Check if the OpenAI API key is set in config.py or as an environment variable."""
+    if not cfg.openai_api_key:
+        print(
+            Fore.RED +
+            "Please set your OpenAI API key in config.py or as an environment variable."
+        )
+        print("You can get your key from https://beta.openai.com/account/api-keys")
+        exit(1)
+
 
 def print_to_console(
         title,
@@ -275,7 +285,7 @@ def parse_arguments():
 
 
 # TODO: fill in llm values here
-
+check_openai_api_key()
 cfg = Config()
 parse_arguments()
 ai_name = ""