Revert "Fix step endpoint to accept openenv-core wrapped action format and add GET /state"

This reverts commit 498deff8a9511218a0e5dcbf4eb73668ffe95da7.

client.py

@@ -90,7 +90,7 @@ class DataCleaningEnvClient:
 
     def state(self) -> DataCleaningState:
         """Return current episode metadata without modifying state."""
-        resp = self._client.get("/state")
+        resp = self._client.post("/state")
         resp.raise_for_status()
         return DataCleaningState(**resp.json())
 

server/app.py

@@ -1,11 +1,11 @@
 """
 FastAPI application exposing the OpenEnv-compatible HTTP API.
 Endpoints: GET /health, GET /metadata, GET /schema,
-           POST /reset, POST /step, GET /state, POST /state, GET /docs
+           POST /reset, POST /step, POST /state, GET /docs
 """
 
-from typing import Any, Dict, Optional
-from fastapi import Body, FastAPI, HTTPException
+from typing import Optional
+from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import uvicorn
 
@@ -123,31 +123,16 @@ def reset(req: ResetRequest = ResetRequest()):
 
 
 @app.post("/step", response_model=StepResponse)
-async def step(body: Dict[str, Any] = Body(...)):
-    """
-    Accept both openenv-core wrapped format:
-        {"action": {"operation": "...", ...}, "timeout_s": 15}
-    and direct format (for backward compat with our own client/inference):
-        {"operation": "...", "column": "...", "params": {...}}
-    """
-    action_data = body.get("action", body)
+def step(action: DataCleaningAction):
     try:
-        action = DataCleaningAction(**action_data)
         obs = env.step(action)
-    except (TypeError, KeyError, Exception) as e:
+    except RuntimeError as e:
         raise HTTPException(status_code=400, detail=str(e))
     return StepResponse(observation=obs, reward=obs.reward, done=obs.done)
 
 
-@app.get("/state", response_model=DataCleaningState)
-def state_get():
-    """GET /state — openenv-core spec."""
-    return env.state()
-
-
 @app.post("/state", response_model=DataCleaningState)
-def state_post():
-    """POST /state — backward compatibility."""
+def state():
     return env.state()
 
 
@@ -160,4 +145,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()