fix pytest env error

UmstadAI · Jan 29, 2024 · 8e2d184 · 8e2d184
1 parent 6992462
commit 8e2d184
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cli/zkappumstad/tools/prd_tool.py b/cli/zkappumstad/tools/prd_tool.py
@@ -5,11 +5,14 @@
 from openai import OpenAI
 from openai.types.chat import ChatCompletion
 from dotenv import load_dotenv, find_dotenv
+import os
 
 load_dotenv(find_dotenv(".env.local"))
 
-
-client = OpenAI()
+if os.getenv("OPENAI_API_KEY"):
+    client = OpenAI()
+else:
+    client = None
 
 function_description = {
     "name": "write_prd",