Spaces:
Running
Running
File size: 2,512 Bytes
742e389 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | """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
|