Merge pull request #858 from pkvach/handle-missing-openai-key

Handle OPENAI_KEY not set error in github_action_runner.py
Codium-ai · Apr 12, 2024 · 654f88b · 654f88b
2 parents af8ca7d + 4c83bf6
commit 654f88b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pr_agent/servers/github_action_runner.py b/pr_agent/servers/github_action_runner.py
@@ -46,15 +46,16 @@ async def run_action():
     if not GITHUB_EVENT_PATH:
         print("GITHUB_EVENT_PATH not set")
         return
-    if not OPENAI_KEY:
-        print("OPENAI_KEY not set")
-        return
     if not GITHUB_TOKEN:
         print("GITHUB_TOKEN not set")
         return
 
     # Set the environment variables in the settings
-    get_settings().set("OPENAI.KEY", OPENAI_KEY)
+    if OPENAI_KEY:
+        get_settings().set("OPENAI.KEY", OPENAI_KEY)
+    else:
+        # Might not be set if the user is using models not from OpenAI
+        print("OPENAI_KEY not set")
     if OPENAI_ORG:
         get_settings().set("OPENAI.ORG", OPENAI_ORG)
     get_settings().set("GITHUB.USER_TOKEN", GITHUB_TOKEN)