Merge pull request #82 from filip-michalsky/chemik-bit-patch-1

Update test_salesgpt.py
filip-michalsky · Dec 22, 2023 · d1ab9e9 · d1ab9e9
2 parents 7cccbb4 + bf392c4
commit d1ab9e9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_salesgpt.py b/tests/test_salesgpt.py
@@ -12,7 +12,10 @@
 load_dotenv(dotenv_path)
 
 # Now, try to load the API key
-openai.api_key = os.getenv('OPENAI_API_KEY')
+api_key = os.getenv('OPENAI_API_KEY')
+if not api_key:
+    raise Exception("OPENAI_API_KEY not found")
+#openai.api_key = os.getenv('OPENAI_API_KEY')
 
 class TestSalesGPT:
     def test_valid_inference_no_tools(self, load_env):