From e418fb44a8681d891b1aa69edcd2a57acdf7ad74 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 10 Oct 2023 17:03:18 -0400
Subject: [PATCH] clean up

---
 swarms/workers/worker.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/swarms/workers/worker.py b/swarms/workers/worker.py
index 996b694e..524ecd5d 100644
--- a/swarms/workers/worker.py
+++ b/swarms/workers/worker.py
@@ -22,7 +22,7 @@ from swarms.utils.decorators import error_decorator, log_decorator, timing_decor
 ROOT_DIR = "./data/"
 
 # main
-
+openai_api_key = os.environ["OPENAI_API_KEY"]
 
 class Worker:
     """
@@ -71,7 +71,7 @@ class Worker:
         self.temperature = temperature
         self.human_in_the_loop = human_in_the_loop
         self.llm = llm
-        self.openai_api_key = openai_api_key
+        self.openai_api_key = openai_api_key or os.environ["OPENAI_API_KEY"]
         self.ai_name = ai_name
         self.ai_role = ai_role
         self.setup_tools(external_tools)