Connect to your GPU reservation via VM

curl --fail-with-body -X PUT \
  "$ORNN_API_URL/commerce/reservations/$RESERVATION_ID/disk" \
  -H "Authorization: Bearer $ORNN_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"vmDiskId": "77777777-7777-4777-8777-777777777777"}'

response = requests.put(
    f"{API}/commerce/reservations/{reservation_id}/disk",
    headers=HEADERS,
    json={"vmDiskId": "77777777-7777-4777-8777-777777777777"},
    timeout=30,
)
response.raise_for_status()
print(response.json()["vmDiskId"])

curl --fail-with-body -X POST \
  "$ORNN_API_URL/commerce/reservations/$RESERVATION_ID/activate" \
  -H "Authorization: Bearer $ORNN_API_KEY"

response = requests.post(
    f"{API}/commerce/reservations/{reservation_id}/activate",
    headers=HEADERS,
    timeout=60,
)
response.raise_for_status()
print(response.json()["deployedAt"])

ornn launch "$RESERVATION_ID" --mode vm --key "$SSH_KEY_ID" --wait

curl --fail-with-body -X POST \
  "$ORNN_API_URL/v1/orchestrator/nodes/reservations/$RESERVATION_ID/launch" \
  -H "Authorization: Bearer $ORNN_API_KEY" \
  -H "Content-Type: application/json" \
  -d "{\"access_mode\": \"virtual-machine\", \"ssh_key_ids\": [\"$SSH_KEY_ID\"]}"

response = requests.post(
    f"{API}/v1/orchestrator/nodes/reservations/{reservation_id}/launch",
    headers=HEADERS,
    json={"access_mode": "virtual-machine", "ssh_key_ids": [ssh_key_id]},
    timeout=60,
)
response.raise_for_status()
print(response.status_code, [node["node_id"] for node in response.json()["nodes"]])

ornn machines wait <node-id> --timeout 30m

curl --fail-with-body \
  "$ORNN_API_URL/v1/orchestrator/reservations/$RESERVATION_ID/virtual-machines" \
  -H "Authorization: Bearer $ORNN_API_KEY" \
  | jq '.virtual_machines[] | {id, status, ip, port, tenant_username, failure}'

deadline = time.monotonic() + 30 * 60
while True:
    response = requests.get(
        f"{API}/v1/orchestrator/reservations/{reservation_id}/virtual-machines",
        headers=HEADERS,
        timeout=30,
    )
    response.raise_for_status()
    vms = response.json()["virtual_machines"]
    failed = [vm for vm in vms if vm["failure"]]
    if failed:
        raise RuntimeError(failed[0]["failure"]["message"])
    if vms and all(vm["status"] == "running" for vm in vms):
        break
    if time.monotonic() > deadline:
        raise TimeoutError("VMs are still starting; check the reservation in Nodes")
    time.sleep(5)

for vm in vms:
    target = f"{vm['tenant_username']}@{vm['ip']}"
    while True:
        # Existing VMs get keys from a separate push that can land after `running`.
        probe = subprocess.run(
            ["ssh", "-o", "BatchMode=yes", "-o", "ConnectTimeout=10", "-p", str(vm["port"]), target, "true"],
            capture_output=True,
        )
        if probe.returncode == 0:
            break
        if time.monotonic() > deadline:
            raise TimeoutError(f"{target} is running but did not accept your key yet")
        time.sleep(10)
    print(f"ssh -p {vm['port']} {target}")

ornn connect <node-id> --identity-file ~/.ssh/id_ed25519

curl --fail-with-body -s \
  "$ORNN_API_URL/v1/orchestrator/reservations/$RESERVATION_ID/virtual-machines" \
  -H "Authorization: Bearer $ORNN_API_KEY" \
  | jq -r '.virtual_machines[] | select(.status == "running")
      | "ssh -p \(.port) \(.tenant_username)@\(.ip)"'

curl --fail-with-body -X POST \
  "$ORNN_API_URL/v1/orchestrator/nodes/$NODE_ID/virtual-machines/$VM_ID/reboot" \
  -H "Authorization: Bearer $ORNN_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"mode": "reboot"}'

response = requests.post(
    f"{API}/v1/orchestrator/nodes/{node_id}/virtual-machines/{vm_id}/reboot",
    headers=HEADERS,
    json={"mode": "reboot"},
    timeout=60,
)
response.raise_for_status()
print(response.json()["operation"])  # reboot

curl --fail-with-body -X POST \
  "$ORNN_API_URL/v1/orchestrator/nodes/$NODE_ID/virtual-machines/$VM_ID/reimage" \
  -H "Authorization: Bearer $ORNN_API_KEY" \
  -H "Content-Type: application/json" \
  -d "{\"disk_id\": \"$DISK_ID\"}"

response = requests.post(
    f"{API}/v1/orchestrator/nodes/{node_id}/virtual-machines/{vm_id}/reimage",
    headers=HEADERS,
    json={"disk_id": disk_id},
    timeout=60,
)
response.raise_for_status()
print(response.json()["status"])  # provisioning

curl --fail-with-body -X POST \
  "$ORNN_API_URL/commerce/reservations/$RESERVATION_ID/stop" \
  -H "Authorization: Bearer $ORNN_API_KEY"

response = requests.post(
    f"{API}/commerce/reservations/{reservation_id}/stop",
    headers=HEADERS,
    timeout=60,
)
response.raise_for_status()
print(response.json()["status"])