pull/435/head
Kye 10 months ago
parent 2b13e86cf8
commit 4e1100e15a

@ -996,8 +996,9 @@ 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
@ -1005,8 +1006,8 @@ class MySwarm(BaseSwarm):
# 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,

@ -3,8 +3,9 @@ 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
@ -15,12 +16,14 @@ class MySwarm(BaseSwarm):
# 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,

@ -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,

Loading…
Cancel
Save