diff --git a/swarms/__init__.py b/swarms/__init__.py index 32d4aad3..2c730324 100644 --- a/swarms/__init__.py +++ b/swarms/__init__.py @@ -1,3 +1,3 @@ # from swarms import Swarms, swarm from swarms.swarms import Swarms, swarm -from swarms.agents.workers.worker_agent import worker_node \ No newline at end of file +from swarms.agents import worker_node \ No newline at end of file diff --git a/swarms/agents/__init__.py b/swarms/agents/__init__.py index 798bf167..9568a2b1 100644 --- a/swarms/agents/__init__.py +++ b/swarms/agents/__init__.py @@ -1 +1,2 @@ -"""Agents, workers and bosses""" \ No newline at end of file +"""Agents, workers and bosses""" +from swarms.agents.workers.worker_agent import worker_node \ No newline at end of file diff --git a/swarms/agents/workers/worker_agent.py b/swarms/agents/workers/worker_agent.py index 5a74ed70..55342e02 100644 --- a/swarms/agents/workers/worker_agent.py +++ b/swarms/agents/workers/worker_agent.py @@ -91,4 +91,6 @@ class WorkerNodeInitializer: def worker_node(openai_api_key): initializer = WorkerNodeInitializer(openai_api_key) worker_node = initializer.create_worker_node() - return worker_node \ No newline at end of file + return worker_node + +