From ea016cb0ca89fc6c3c7aff6f62d55e89db97018f Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Sat, 29 Jul 2023 12:37:56 -0400
Subject: [PATCH] args in create worker node
Former-commit-id: 7d2594e03f1855b772353bc2515e819c3a3beffb
---
swarms/workers/worker_node.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py
index a1a7a7dc..1b52f4ea 100644
--- a/swarms/workers/worker_node.py
+++ b/swarms/workers/worker_node.py
@@ -160,7 +160,7 @@ class WorkerNode:
except Exception as e:
logging.error(f"Failed to initialize tools: {e}")
- def create_worker_node(self, llm_class=ChatOpenAI, worker_name, worker_role, human_in_the_loop, search_kwargs={}, verbose):
+ def create_worker_node(self, worker_name, worker_role, human_in_the_loop, llm_class=ChatOpenAI, search_kwargs={}, **kwargs):
if not llm_class:
logging.error("llm_class cannot be None.")
raise ValueError("llm_class cannot be None.")
@@ -175,14 +175,14 @@ class WorkerNode:
ai_name=worker_name,
ai_role=worker_role,
human_in_the_loop=human_in_the_loop,
- search_kwargs=search_kwargs,
- verbose=verbose
+ search_kwargs=search_kwargs,
)
return worker_node
except Exception as e:
logging.error(f"Failed to create worker node: {e}")
raise
+
def worker_node(openai_api_key):
if not openai_api_key:
logging.error("OpenAI API key is not provided")