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.
54 lines
1.4 KiB
54 lines
1.4 KiB
#!/usr/bin/env python3
|
|
"""
|
|
Basic Hierarchical Swarm Streaming Demo
|
|
|
|
Minimal example showing the core streaming callback functionality.
|
|
"""
|
|
|
|
from swarms.structs.hiearchical_swarm import HierarchicalSwarm
|
|
from swarms.agents import Agent
|
|
|
|
|
|
def simple_callback(agent_name: str, chunk: str, is_final: bool):
|
|
"""Simple callback that shows agent progress."""
|
|
if chunk.strip():
|
|
if is_final:
|
|
print(f"✅ {agent_name} finished")
|
|
else:
|
|
print(f"🔄 {agent_name}: {chunk}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("🎯 BASIC HIERARCHICAL SWARM STREAMING")
|
|
|
|
# Create a simple agent
|
|
agent = Agent(
|
|
agent_name="Simple_Agent",
|
|
agent_description="A simple agent for demonstration",
|
|
system_prompt="You are a helpful assistant.",
|
|
model_name="gpt-4o-mini",
|
|
max_loops=1,
|
|
)
|
|
|
|
# Create swarm
|
|
swarm = HierarchicalSwarm(
|
|
name="Basic_Swarm",
|
|
description="Basic streaming demo",
|
|
agents=[agent],
|
|
max_loops=1,
|
|
director_model_name="gpt-4o-mini",
|
|
)
|
|
|
|
# Simple task
|
|
task = "Explain what artificial intelligence is in simple terms."
|
|
|
|
print(f"Task: {task}")
|
|
print("\nExecuting with streaming callback:\n")
|
|
|
|
# Run with streaming
|
|
result = swarm.run(task=task, streaming_callback=simple_callback)
|
|
|
|
print("\n" + "=" * 30)
|
|
print("Final result:")
|
|
print(result)
|