| """ |
| client.py — OpenEnv-Compliant Client for Autonomy Calibration |
| This client provides a strictly separated interface to the environment, |
| satisfying the hackathon's architectural requirements. |
| """ |
|
|
| from openenv.core.env_client import EnvClient |
| from models import Action, Observation, StepResult |
|
|
| class AutonomyCalibrationClient(EnvClient): |
| """ |
| Client for interacting with the Autonomy Calibration Space. |
| Inheriting from OpenEnv's EnvClient ensures standard communication protocols. |
| """ |
| def __init__(self, base_url: str = "http://localhost:7860"): |
| super().__init__(base_url=base_url) |
|
|
| def reset_env(self, task: str = "email_triage", seed: int = None) -> Observation: |
| """Resets the remote environment and returns a Pydantic Observation.""" |
| response = self._post("/api/reset", json={"task": task, "seed": seed}) |
| return Observation(**response.json()) |
|
|
| def step_env(self, action_type: str) -> StepResult: |
| """Takes a step in the remote environment and returns a Pydantic StepResult.""" |
| action = {"type": action_type, "payload": {}} |
| response = self._post("/api/step", json=action) |
| return StepResult(**response.json()) |
|
|
