From 33687cc222efbf1213f5cef876b93cc9fa1da16d Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 6 Oct 2023 12:31:06 -0400 Subject: [PATCH] openai api key --- playground/swarms/group_chat.py | 3 +-- swarms/workers/worker.py | 3 ++- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/playground/swarms/group_chat.py b/playground/swarms/group_chat.py index fc44e294..8b01bd34 100644 --- a/playground/swarms/group_chat.py +++ b/playground/swarms/group_chat.py @@ -1,3 +1,2 @@ -from swarms.swarms import GroupChatManager -from swarms.agents.base import AbstractAgent +from swarms.swarms import GroupChat diff --git a/swarms/workers/worker.py b/swarms/workers/worker.py index e01866e1..99b502b1 100644 --- a/swarms/workers/worker.py +++ b/swarms/workers/worker.py @@ -56,17 +56,18 @@ class Worker: """ def __init__( self, - ai_name: str = "Autobot Swarm Worker", ai_role: str = "Worker in a swarm", external_tools = None, human_in_the_loop = False, temperature: float = 0.5, llm = None, + openai_api_key: str = None, ): self.temperature = temperature self.human_in_the_loop = human_in_the_loop self.llm = llm + self.openai_api_key = openai_api_key self.ai_name = ai_name self.ai_role = ai_role self.setup_tools(external_tools)