File size: 882 Bytes
6de1b61 | 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 | from __future__ import annotations
import json
import os
import urllib.request
class CadForgeClient:
def __init__(self, base_url: str | None = None) -> None:
self.base_url = (base_url or os.getenv("OPENENV_BASE_URL", "http://localhost:8000")).rstrip("/")
def reset(self, **payload: object) -> dict:
return self._post("/reset", payload)
def step(self, action: dict) -> dict:
return self._post("/step", {"action": action})
def _post(self, path: str, payload: dict) -> dict:
request = urllib.request.Request(
f"{self.base_url}{path}",
data=json.dumps(payload).encode("utf-8"),
headers={"Content-Type": "application/json"},
method="POST",
)
with urllib.request.urlopen(request, timeout=240) as response:
return json.loads(response.read().decode("utf-8"))
|