fix: fixed error when no OPENAI_API_KEY set

ErikBjare · Oct 10, 2023 · 9b8f4e1 · 9b8f4e1
1 parent edb1982
commit 9b8f4e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gptme/llm.py b/gptme/llm.py
@@ -27,7 +27,7 @@ def init_llm(llm: str):
     if llm == "openai":
         if "OPENAI_API_KEY" in os.environ:
             api_key = os.environ["OPENAI_API_KEY"]
-        elif api_key := config["env"]["OPENAI_API_KEY"]:
+        elif api_key := config["env"].get("OPENAI_API_KEY", None):
             pass
         else:
             print("Error: OPENAI_API_KEY not set in env or config, see README.")