[CLEANUP]

kyegomez · Mar 30, 2024 · 4e1100e · 4e1100e
1 parent 2b13e86
commit 4e1100e
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -996,17 +996,18 @@ from swarms import AutoSwarm, AutoSwarmRouter, BaseSwarm
 
 # Build your own Swarm
 class MySwarm(BaseSwarm):
-    def __init__(self, *args, **kwargs):
+    def __init__(self, name="kyegomez/myswarm", *args, **kwargs):
         super().__init__(*args, **kwargs)
+        self.name = name
 
     def run(self, task: str, *args, **kwargs):
         # Add your multi-agent logic here
         # agent 1
         # agent 2
         # agent 3
         return "output of the swarm"
-    
-    
+
+
 # Add your custom swarm to the AutoSwarmRouter
 router = AutoSwarmRouter(
     swarms=[MySwarm]
@@ -1015,7 +1016,7 @@ router = AutoSwarmRouter(
 
 # Create an AutoSwarm instance
 autoswarm = AutoSwarm(
-    name = "AutoSwarm, an API for all swarms",
+    name="kyegomez/myswarm",
     description="A simple API to build and run swarms",
     verbose=True,
     router=router,

diff --git a/full_stack_agent.py → playground/agents/full_stack_agent.py b/full_stack_agent.py → playground/agents/full_stack_agent.py
diff --git a/tool_agent_with_llm.py → playground/agents/tool_agent_with_llm.py b/tool_agent_with_llm.py → playground/agents/tool_agent_with_llm.py
diff --git a/playground/structs/build_your_own_swarm.py b/playground/structs/build_your_own_swarm.py
@@ -3,8 +3,9 @@
 
 # Build your own Swarm
 class MySwarm(BaseSwarm):
-    def __init__(self, *args, **kwargs):
+    def __init__(self, name="kyegomez/myswarm", *args, **kwargs):
         super().__init__(*args, **kwargs)
+        self.name = name
 
     def run(self, task: str, *args, **kwargs):
         # Add your multi-agent logic here
@@ -15,12 +16,14 @@ def run(self, task: str, *args, **kwargs):
 
 
 # Add your custom swarm to the AutoSwarmRouter
-router = AutoSwarmRouter(swarms=[MySwarm])
+router = AutoSwarmRouter(
+    swarms=[MySwarm]
+)
 
 
 # Create an AutoSwarm instance
 autoswarm = AutoSwarm(
-    name="AutoSwarm, an API for all swarms",
+    name="kyegomez/myswarm",
     description="A simple API to build and run swarms",
     verbose=True,
     router=router,

diff --git a/swarm_network_example.py → playground/structs/swarm_network_example.py b/swarm_network_example.py → playground/structs/swarm_network_example.py
diff --git a/swarms/structs/base_swarm.py b/swarms/structs/base_swarm.py
@@ -68,6 +68,7 @@ class BaseSwarm(ABC):
 
     def __init__(
         self,
+        # name: str = "Swarm",
         agents: List[Agent] = None,
         models: List[Any] = None,
         max_loops: int = 200,