From d7ef20031b2fe5e08724c156449a711a06c8b02f Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 11 Aug 2023 22:56:36 -0400 Subject: [PATCH] unindent worker swarm --- swarms/workers/worker_node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py index 74adb798..114d1745 100644 --- a/swarms/workers/worker_node.py +++ b/swarms/workers/worker_node.py @@ -109,7 +109,7 @@ class WorkerNodeInitializer: raise e class WorkerNode: - def __init__(self, + def __init__(self, openai_api_key: str, temperature: Optional[int] = None, llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None, @@ -237,7 +237,7 @@ class WorkerNode: try: worker_tools = self.initialize_tools(llm_class) vectorstore = self.initialize_vectorstore() - worker_node = UnifiedWorkerNode( + worker_node = WorkerNode( openai_api_key=self.openai_api_key, llm=self.initialize_llm(llm_class, self.temperature), tools=worker_tools,