Update client.py (#15)

- Update client.py (778057bd00d4be4fda2af0c510396c86781a84c8)


Co-authored-by: tarun saroya <Tarun-sar0ya@users.noreply.huggingface.co>

client.py

@@ -0,0 +1,35 @@
+"""Client for the API Triage Agent environment.
+
+Implements the EnvClient interface for connecting to the
+API Triage environment server over HTTP.
+"""
+
+try:
+    from openenv.core.env_client import EnvClient
+
+    class APITriageClient(EnvClient):
+        """Client for the API Triage Agent OpenEnv environment."""
+        pass
+
+except ImportError:
+    # Fallback if openenv-core not installed locally
+    class APITriageClient:  # type: ignore
+        """Stub client for local development without openenv-core."""
+
+        def __init__(self, base_url: str = "http://localhost:7860"):
+            self.base_url = base_url
+
+        def reset(self):
+            import requests
+            return requests.post(f"{self.base_url}/reset").json()
+
+        def step(self, action: str):
+            import requests
+            return requests.post(
+                f"{self.base_url}/step",
+                json={"action": action}
+            ).json()
+
+        def state(self):
+            import requests
+            return requests.get(f"{self.base_url}/state").json()