Spaces:
Running on Zero
Running on Zero
File size: 2,359 Bytes
a4fb1d3 | 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 55 56 57 58 59 60 61 62 63 64 65 66 67 | """Live integration test: a local node joins the public HF Space relay and
calls the Space node's capabilities all-to-all over the internet.
Run: python scripts/live_mesh_test.py
"""
from __future__ import annotations
import asyncio
import secrets
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
SPACE = "https://build-small-hackathon-hearthnet.hf.space"
async def main() -> int:
from hearthnet.node import HearthNode
node = HearthNode(f"ed25519:tester-{secrets.token_hex(3)}", "LiveTester", "ed25519:hf-space-community")
await node.start(host="127.0.0.1", port=7099, data_dir=str(Path(__file__).resolve().parent.parent / ".live_test_node"))
print(f"[local] node up: {node.node_id}")
print(f"[mesh] joining live Space relay {SPACE} ...")
result = await node.join_relay(SPACE)
roster = result.get("roster", [])
print(f"[mesh] joined. {len(roster)} other member(s):")
space_node = None
for m in roster:
nid = m.get("node_id", "")
caps = m.get("capabilities", [])
print(f" - {nid} ({m.get('display_name','')}) — {len(caps)} caps")
if nid.startswith("hf-space"):
space_node = nid
if not space_node:
print("[FAIL] Space node not found in roster")
await node.stop()
return 1
# Verify the local bus registry now knows the Space node's capabilities.
remote = sorted({e.descriptor.name for e in node.bus.registry.all_remote()})
print(f"[bus] local registry now has {len(remote)} remote capabilities, e.g. {remote[:6]}")
# All-to-all call: ask the Space node to run llm.chat over the relay.
print("[call] llm.chat -> Space node over the relay ...")
try:
resp = await node.bus.call(
"llm.chat",
(1, 0),
{"input": {"messages": [{"role": "user", "content": "Say hi from the mesh"}]}},
)
msg = resp.get("output", {}).get("message", {}).get("content", resp)
print(f"[call] OK -> {msg!r}")
except Exception as exc:
print(f"[call] remote llm.chat failed (Space backend may gate ZeroGPU): {exc}")
await node.stop()
print("[done] live mesh verified: local node meshed all-to-all with the HF Space.")
return 0
if __name__ == "__main__":
raise SystemExit(asyncio.run(main()))
|