clean up

multi_agent_debate.py

@@ -2,7 +2,7 @@
 from swarms.workers.worker import Worker
 from langchain.llms import OpenAIChat
 
-llm = OpenAIChat(model_name="gpt-4", openai_api_key="api-key", temperature=0.5)
+llm = OpenAIChat()
 
 worker1 = Worker(
     llm=llm,

swarms/workers/worker.py

@@ -22,9 +22,6 @@
 # cache
 ROOT_DIR = "./data/"
 
-# main
-OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
-
 
 class Worker:
     """
@@ -73,7 +70,7 @@ def __init__(
         self.temperature = temperature
         self.human_in_the_loop = human_in_the_loop
         self.llm = llm
-        self.openai_api_key = openai_api_key or OPENAI_API_KEY
+        self.openai_api_key = openai_api_key
         self.ai_name = ai_name
         self.ai_role = ai_role
         self.setup_tools(external_tools)
@@ -145,9 +142,8 @@ def setup_memory(self):
         """
         Set up memory for the worker.
         """
-        openai_api_key = OPENAI_API_KEY
         try:
-            embeddings_model = OpenAIEmbeddings(openai_api_key=openai_api_key)
+            embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
             embedding_size = 1536
             index = faiss.IndexFlatL2(embedding_size)
 
@@ -283,4 +279,4 @@ def _message_to_dict(message: Union[Dict, str]):
         if isinstance(message, str):
             return {"content": message}
         else:
-            return message
+            return message