Agent in Sandbox (Basic)

AGENT_SCRIPT = textwrap.dedent("""\
    import json
    import subprocess

    TASK_FILE = "/tmp/task.json"
    RESULT_FILE = "/tmp/result.json"

    with open(TASK_FILE) as f:
        task = json.load(f)

    results = []
    for i, step in enumerate(task["steps"], 1):
        proc = subprocess.run(
            step, shell=True, capture_output=True, text=True, timeout=30
        )
        results.append({
            "step": i,
            "command": step,
            "stdout": proc.stdout.strip(),
            "exit_code": proc.returncode,
        })

    output = {"task_name": task["name"], "status": "completed", "step_results": results}
    with open(RESULT_FILE, "w") as f:
        json.dump(output, f, indent=2)
""")

TASK_PAYLOAD = {
    "name": "system-info-gathering",
    "description": "Collect basic system information inside the sandbox",
    "steps": [
        "uname -a",
        "python3 --version",
        "whoami",
        "ls /tmp",
    ],
}

sbx = Sandbox.create(template="python", timeout=300)
try:
    # Upload the agent script
    sbx.files.write("/tmp/agent.py", AGENT_SCRIPT)

    # Upload the task definition
    task_json = json.dumps(TASK_PAYLOAD, indent=2)
    sbx.files.write("/tmp/task.json", task_json)

    # Execute the agent inside the isolated sandbox
    result = sbx.commands.run("python3 /tmp/agent.py", timeout=60)
    print(result.stdout)

    # Read structured output back to the host process
    result_content = sbx.files.read("/tmp/result.json")
    results = json.loads(result_content)
    for step in results["step_results"]:
        print(f"Step {step['step']}: {step['command']} -> exit {step['exit_code']}")
finally:
    sbx.kill()

--- Creating Sandbox ---
Sandbox created: sbx-abc123

--- Uploading Agent Script ---
  Wrote /tmp/agent.py

--- Uploading Task ---
  Task: system-info-gathering
  Steps: 4

--- Running Agent ---
  Agent stdout:
  Agent received task: system-info-gathering
    Description: Collect basic system information inside the sandbox
    Running step 1: uname -a
    Running step 2: python3 --version
    Running step 3: whoami
    Running step 4: ls /tmp
  Agent finished. Results written to /tmp/result.json
  Exit code: 0

--- Reading Agent Results ---
  Task: system-info-gathering
  Status: completed

  Step 1: uname -a
    stdout: Linux ... x86_64 GNU/Linux
    exit_code: 0

  Step 2: python3 --version
    stdout: Python 3.x.x
    exit_code: 0