diff --git a/swarms/__init__.py b/swarms/__init__.py index 2c730324..5b5e4280 100644 --- a/swarms/__init__.py +++ b/swarms/__init__.py @@ -1,3 +1,4 @@ # from swarms import Swarms, swarm from swarms.swarms import Swarms, swarm -from swarms.agents import worker_node \ No newline at end of file +from swarms.agents import worker_node +from swarms.agents.workers.worker_ultranode import UltraNode \ No newline at end of file diff --git a/swarms/agents/workers/__init__.py b/swarms/agents/workers/__init__.py index 91ea82ab..eab37564 100644 --- a/swarms/agents/workers/__init__.py +++ b/swarms/agents/workers/__init__.py @@ -1 +1,3 @@ from .worker_agent import worker_node + +from .worker_ultranode import UltraNode diff --git a/swarms/agents/workers/worker_ultranode.py b/swarms/agents/workers/worker_ultranode.py index 39fe1864..08230671 100644 --- a/swarms/agents/workers/worker_ultranode.py +++ b/swarms/agents/workers/worker_ultranode.py @@ -68,7 +68,7 @@ class UltraNode: # The prompt is not needed here either return self.execute_task() -from worker_node import UltraNode +# from worker_node import UltraNode -node = UltraNode('objective') -result = node.execute() +# node = UltraNode('objective') +# result = node.execute()