[CLEANUP]

kyegomez · Apr 2, 2024 · d8b42f0 · d8b42f0
1 parent fc587b3
commit d8b42f0
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 77 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "4.7.0"
+version = "4.7.1"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]
@@ -68,7 +68,3 @@ unfixable = []
 line-length = 70
 target-version = ['py38']
 preview = true
-
-
-[tool.poetry.scripts]
-swarms = 'swarms.cli._cli:main'
diff --git a/rearrange_agent_example.py b/rearrange_agent_example.py
@@ -2,17 +2,18 @@
 from collections import defaultdict
 from typing import Callable, Sequence
 from swarms import Agent, Anthropic
-
+from swarms.structs.base_swarm import BaseSwarm
 
 # Assuming the existence of an appropriate Agent class and logger setup
-class AgentRearrange:
+class AgentRearrange(BaseSwarm):
     def __init__(
         self,
         agents: Sequence[Agent] = None,
         verbose: bool = False,
         custom_prompt: str = None,
         callbacks: Sequence[Callable] = None,
     ):
+        super().__init__()
         if not all(isinstance(agent, Agent) for agent in agents):
             raise ValueError(
                 "All elements must be instances of the Agent class."

diff --git a/swarms/cli/__init__.py b/swarms/cli/__init__.py
diff --git a/swarms/cli/_cli.py b/swarms/cli/_cli.py
diff --git a/swarms/cli/run_file.py b/swarms/cli/run_file.py