clean up

Former-commit-id: 1a96d7ca12
2 years ago · a454cdfdff
parent fc1a004b1b
commit a454cdfdff
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")