parent
41c5295239
commit
cd58ffadaa
@ -0,0 +1,33 @@
|
|||||||
|
from swarms import AutoSwarm, AutoSwarmRouter, BaseSwarm
|
||||||
|
|
||||||
|
|
||||||
|
# Build your own Swarm
|
||||||
|
class MySwarm(BaseSwarm):
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
super().__init__(*args, **kwargs)
|
||||||
|
|
||||||
|
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]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# Create an AutoSwarm instance
|
||||||
|
autoswarm = AutoSwarm(
|
||||||
|
name = "AutoSwarm, an API for all swarms",
|
||||||
|
description="A simple API to build and run swarms",
|
||||||
|
verbose=True,
|
||||||
|
router=router,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# Run the AutoSwarm
|
||||||
|
autoswarm.run("Analyze these financial data and give me a summary")
|
Loading…
Reference in new issue