final fix reset body issue

api/server.py

@@ -1,20 +1,7 @@
 from fastapi import FastAPI
-from pydantic import BaseModel
 
 app = FastAPI()
 
-# ----------- Request Models -----------
-
-class ResetRequest(BaseModel):
-    task_id: str = "default"
-
-
-class StepRequest(BaseModel):
-    task_id: str
-    action: str = "continue"
-
-
-# ----------- Endpoints -----------
 
 @app.get("/health")
 def health():
@@ -22,9 +9,9 @@ def health():
 
 
 @app.post("/reset")
-def reset(req: ResetRequest):
+def reset():
     return {
-        "task_id": req.task_id,
+        "task_id": "default",
         "observation": "reset done",
         "reward": 0.1,
         "done": False
@@ -32,10 +19,10 @@ def reset(req: ResetRequest):
 
 
 @app.post("/step")
-def step(req: StepRequest):
+def step(action: dict = {}):
     return {
-        "task_id": req.task_id,
-        "observation": f"action {req.action} executed",
+        "task_id": action.get("task_id", "default"),
+        "observation": "step executed",
         "reward": 0.5,
         "done": True
     }