clean up

Former-commit-id: 99c4f48eb29f24f885fa7360f1a11bca8b27f6b0
2 years ago · 154ad85c6a
parent 0fb3915273
commit 154ad85c6a
1 changed files with 1 additions and 1 deletions
--- a/swarms/agents/workers/WorkerUltraNode.py
+++ b/swarms/agents/workers/WorkerUltraNode.py
@ -30,7 +30,7 @@ except Exception as e:
    
 class WorkerUltraNode:
    def __init__(self, objective: str, openai_api_key: str):
-        self.openai_api_key = openai_api_key
+        self.openai_api_key = openai_api_key or os.getenv("OPENAI_API_KEY")

        if not isinstance(objective, str):
            raise TypeError("Objective must be a string")