sdk-dev
d6c2b701b6
[Do] Merge queue: CI checks green, merge queue ready
sop-checklist / all-items-acked SOP checklist acknowledged
Test / test (3.12) (pull_request) Successful in 1m30s
Test / test (3.11) (pull_request) Successful in 1m45s
Test / test (3.13) (pull_request) Successful in 1m33s
The old path `sdk/python/` no longer exists in the standalone molecule-sdk-python repo. Update the local-dev import path to add the repo root to sys.path, which correctly resolves molecule_agent when running the script from a repo checkout. Also update the Usage section to use the correct relative path (`examples/remote-agent/run.py` instead of `python3 run.py`) and update the "What it doesn't do" docstring to reflect that Phase 30.8b (inbound A2A) is now shipped. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
105 lines
4.1 KiB
Python
105 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
||
"""Minimal remote-agent demo — Phase 30.1–30.5 end-to-end from outside the
|
||
platform's Docker network.
|
||
|
||
What this does:
|
||
1. Registers the workspace with the platform (mints + saves a bearer token).
|
||
2. Pulls the merged decrypted secrets via the token-gated 30.2 endpoint.
|
||
3. Runs a heartbeat + state-poll loop; exits cleanly when the platform
|
||
reports the workspace paused or deleted.
|
||
|
||
What it doesn't do (see run_agent_loop for the full version):
|
||
- Host an inbound A2A server. Use run_agent_loop(handler) instead of
|
||
run_heartbeat_loop() to also receive inbound A2A messages and reply
|
||
via the platform's /notify (canvas user) and /a2a (peer agent) endpoints.
|
||
The reply transport is auto-selected based on message source.
|
||
|
||
Usage:
|
||
# One-time setup on the platform side:
|
||
# 1) Create the workspace row (any template is fine — external runtime
|
||
# is cleanest if you don't want Docker to try to start a container):
|
||
curl -s -X POST http://localhost:8080/workspaces \\
|
||
-H 'Content-Type: application/json' \\
|
||
-d '{"name":"remote-demo","tier":2,"runtime":"external"}'
|
||
# 2) Grab the returned workspace id.
|
||
# 3) Optional — seed a secret:
|
||
curl -s -X POST http://localhost:8080/workspaces/<id>/secrets \\
|
||
-H 'Content-Type: application/json' \\
|
||
-d '{"key":"REMOTE_DEMO_KEY","value":"hello-from-remote"}'
|
||
|
||
# Now run this script from any machine that can reach the platform:
|
||
WORKSPACE_ID=<id> PLATFORM_URL=http://localhost:8080 python3 examples/remote-agent/run.py
|
||
|
||
Environment variables:
|
||
WORKSPACE_ID (required)
|
||
PLATFORM_URL (required)
|
||
AGENT_NAME (optional; default derived from workspace id)
|
||
MAX_ITERATIONS (optional; caps loop length for demos)
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
import os
|
||
import sys
|
||
|
||
# Local-dev import path — when installed via pip the molecule_agent package
|
||
# resolves normally; when running from the repo checkout we add the repo root
|
||
# to sys.path so you can run `python3 examples/remote-agent/run.py` without
|
||
# a pip install.
|
||
_here = os.path.dirname(os.path.abspath(__file__))
|
||
_repo_root = os.path.normpath(os.path.join(_here, "..", ".."))
|
||
_molecule_agent = os.path.join(_repo_root, "molecule_agent")
|
||
if os.path.isdir(_molecule_agent) and _repo_root not in sys.path:
|
||
sys.path.insert(0, _repo_root)
|
||
|
||
from molecule_agent import RemoteAgentClient # noqa: E402
|
||
|
||
|
||
def main() -> int:
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format="%(asctime)s [%(levelname)s] %(message)s",
|
||
)
|
||
log = logging.getLogger("remote-agent-demo")
|
||
|
||
workspace_id = os.environ.get("WORKSPACE_ID", "").strip()
|
||
platform_url = os.environ.get("PLATFORM_URL", "").strip()
|
||
if not workspace_id or not platform_url:
|
||
log.error("set WORKSPACE_ID and PLATFORM_URL and re-run")
|
||
return 2
|
||
|
||
agent_name = os.environ.get("AGENT_NAME", f"remote-demo-{workspace_id[:8]}")
|
||
max_iter_env = os.environ.get("MAX_ITERATIONS", "").strip()
|
||
max_iter = int(max_iter_env) if max_iter_env else None
|
||
|
||
client = RemoteAgentClient(
|
||
workspace_id=workspace_id,
|
||
platform_url=platform_url,
|
||
agent_card={"name": agent_name, "skills": []},
|
||
# Shorter intervals for demo visibility; production would leave defaults.
|
||
heartbeat_interval=5.0,
|
||
)
|
||
|
||
log.info("phase 1 — registering workspace %s with %s", workspace_id, platform_url)
|
||
client.register()
|
||
|
||
log.info("phase 2 — pulling secrets via 30.2 token-gated endpoint")
|
||
try:
|
||
secrets = client.pull_secrets()
|
||
except Exception as exc:
|
||
log.error("pull_secrets failed: %s", exc)
|
||
return 1
|
||
log.info("received %d secret(s): keys=%s", len(secrets), sorted(secrets.keys()))
|
||
|
||
log.info("phase 3 — heartbeat + state-poll loop (will exit on pause/delete)")
|
||
terminal = client.run_heartbeat_loop(
|
||
max_iterations=max_iter,
|
||
task_supplier=lambda: {"current_task": "remote-agent demo idle", "active_tasks": 0},
|
||
)
|
||
log.info("loop exited: terminal_status=%s", terminal)
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|