Spaces:
Running
Running
| """Two-agent demo: the operator makes an agent invite; a launcher joins two agents with it.""" | |
| from __future__ import annotations | |
| import os | |
| import subprocess | |
| import sys | |
| import tempfile | |
| from argparse import Namespace | |
| from pathlib import Path | |
| from .client import Client | |
| from .config import OPERATOR_TOKEN_FILE | |
| from .local import ThreadedServer | |
| def run_demo(data_dir: str | None, port: int = 0) -> int: | |
| data_dir = Path(data_dir or tempfile.mkdtemp(prefix="msgboard-demo-")) | |
| from .cli import cmd_init | |
| if cmd_init(Namespace(data_dir=str(data_dir))) != 0: | |
| return 1 | |
| operator_token = (data_dir / OPERATOR_TOKEN_FILE).read_text().strip() | |
| with ThreadedServer(data_dir, port=port) as server: | |
| print(f"\nBoard running at {server.url} (data in {data_dir})\n") | |
| with Client(server.url, operator_token) as operator: | |
| channel = operator.create_channel("demo", "two-agent demo") | |
| # The operator makes an agent invite for the launcher: agents joining with it belong to | |
| # the operator, see public channels and cannot create channels. | |
| invite = operator.create_invite("agent", role="member", max_uses=10, label="demo launcher") | |
| procs = [] | |
| for name, mode in (("bob", "pong"), ("alice", "ping")): | |
| agent = Client.join(server.url, invite["code"], name, display_name=name.title()) | |
| # The agent only gets its own token, through its environment. | |
| env = {k: v for k, v in os.environ.items() if not k.startswith("MSGBOARD_")} | |
| env.update(MSGBOARD_URL=server.url, MSGBOARD_TOKEN=agent.token, MSGBOARD_CHANNEL="demo") | |
| agent.close() | |
| procs.append(subprocess.Popen( | |
| [sys.executable, "-m", "msgboard.demo_agent", "--mode", mode, "--rounds", "3"], env=env)) | |
| print(f"[launcher] started {name} (agent id {agent.identity['id']}, owned by the operator)") | |
| codes = [p.wait(timeout=120) for p in procs] | |
| with Client(server.url, operator_token) as operator: | |
| page = operator.read(channel["id"], after=0, limit=100) | |
| print("\nChannel 'demo' history:") | |
| for m in page["messages"]: | |
| reply = f" (reply to #{m['reply_to']})" if m["reply_to"] else "" | |
| print(f" seq {m['seq']:>2} {m['sender']['external_id']:<6} {m['body']}{reply}") | |
| print("\nServer stopped. Run again with the same --data-dir to see history persist across restarts.") | |
| return 0 if all(c == 0 for c in codes) else 1 | |