diff --git a/swarms/__init__.py b/swarms/__init__.py index c318c2e1..b64b3cd9 100644 --- a/swarms/__init__.py +++ b/swarms/__init__.py @@ -20,3 +20,7 @@ from swarms.workers.worker import Worker #workflows from swarms.structs.workflow import Workflow + +# swarms +from swarms.swarms.orchestrate import Orchestrator +from swarms.swarms.autoscaler import AutoScaler \ No newline at end of file diff --git a/swarms/swarms/orchestrate.py b/swarms/swarms/orchestrate.py index 7eee9a27..6bdd8530 100644 --- a/swarms/swarms/orchestrate.py +++ b/swarms/swarms/orchestrate.py @@ -7,7 +7,7 @@ from typing import Any, Dict, List import chromadb from chromadb.utils import embedding_functions -from swarms.workers.worker import Worker + class Orchestrator: @@ -245,14 +245,3 @@ class Orchestrator: documents=[message], ids=[f"{sender_id}_to_{receiver_id}"] ) - -orchestrate = Orchestrator( - Worker, - Worker, - task_queue=["what is your name"], -) -orchestrate.chat( - sender_id=1, - receiver_id=2, - message="Hello, how are you " -) \ No newline at end of file