From 57b87763c8085d1bbbee2290865e5a98676bb89e Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 8 Jul 2023 16:37:15 -0400 Subject: [PATCH] worker_node init --- swarms/__init__.py | 2 +- swarms/agents/__init__.py | 3 ++- swarms/agents/workers/worker_agent.py | 4 +++- 3 files changed, 6 insertions(+), 3 deletions(-) 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 + +