from swarm.repl import run_demo_loop
...
run_demo_loop(agent, stream=True)