You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
from swarms import HierarchicalSwarm
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
swarm = HierarchicalSwarm(
|
|
|
|
|
openai_api_key="key",
|
|
|
|
|
model_type="openai",
|
|
|
|
|
model_id="gpt-4",
|
|
|
|
|
use_vectorstore=False,
|
|
|
|
|
use_async=False,
|
|
|
|
|
human_in_the_loop=False,
|
|
|
|
|
logging_enabled=False,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# run the swarm with an objective
|
|
|
|
|
result = swarm.run("Design a new car")
|
|
|
|
|
|
|
|
|
|
# or huggingface
|
|
|
|
|
swarm = HierarchicalSwarm(
|
|
|
|
|
model_type="huggingface",
|
|
|
|
|
model_id="tiaueu/falcon",
|
|
|
|
|
use_vectorstore=True,
|
|
|
|
|
embedding_size=768,
|
|
|
|
|
use_async=False,
|
|
|
|
|
human_in_the_loop=True,
|
|
|
|
|
logging_enabled=False,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# Run the swarm with a particular objective
|
|
|
|
|
result = swarm.run("Write a sci-fi short story")
|
