fix: reset accepts empty body

api/main.py

@@ -7,24 +7,16 @@ import sys
 import os
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
-from typing import Optional, Dict
+from typing import Optional
 import uvicorn
 
 from env.cache import CDNCacheEnv, TASK_CONFIGS
 from env.models import Action, StepResult
 
-app = FastAPI(
-    title="CDN Cache Optimizer - OpenEnv",
-    description=(
-        "RL environment simulating edge CDN cache management. "
-        "Agent decides which files to evict when cache is full. "
-        "Implements full OpenEnv spec."
-    ),
-    version="1.0.0",
-)
+app = FastAPI(title="CDN Cache Optimizer - OpenEnv", version="1.0.0")
 
 app.add_middleware(
     CORSMiddleware,
@@ -33,23 +25,16 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Global env instance (stateful per session)
 _env: Optional[CDNCacheEnv] = None
 
 
-class ResetRequest(BaseModel):
-    task_id: str = "task_easy"
-    seed: int = 42
-
-
-class StepRequest(BaseModel):
-    evict_file_id: Optional[str] = None
-
-
 @app.get("/health")
 def health():
     return {"status": "ok", "env": "cdn-cache-optimizer"}
 
+@app.post("/health")
+def health_post():
+    return {"status": "ok", "env": "cdn-cache-optimizer"}
 
 @app.get("/tasks")
 def list_tasks():
@@ -64,33 +49,40 @@ def list_tasks():
         for task_id, cfg in TASK_CONFIGS.items()
     }
 
-
 @app.post("/reset")
-def reset(req: ResetRequest):
+async def reset(request: Request):
     global _env
-    if req.task_id not in TASK_CONFIGS:
-        raise HTTPException(
-            status_code=400,
-            detail=f"Unknown task_id '{req.task_id}'. Valid: {list(TASK_CONFIGS.keys())}"
-        )
-    _env = CDNCacheEnv(task_id=req.task_id, seed=req.seed)
+    task_id = "task_easy"
+    seed = 42
+    try:
+        body = await request.json()
+        task_id = body.get("task_id", "task_easy")
+        seed = body.get("seed", 42)
+    except Exception:
+        pass
+    if task_id not in TASK_CONFIGS:
+        raise HTTPException(status_code=400, detail=f"Unknown task_id '{task_id}'.")
+    _env = CDNCacheEnv(task_id=task_id, seed=seed)
     obs = _env.reset()
     return {"observation": obs.dict(), "task": _env.config.dict()}
 
-
 @app.post("/step")
-def step(req: StepRequest):
+async def step(request: Request):
     global _env
     if _env is None:
         raise HTTPException(status_code=400, detail="Call /reset first.")
     if _env._done:
         raise HTTPException(status_code=400, detail="Episode done. Call /reset.")
-
-    action = Action(evict_file_id=req.evict_file_id)
+    evict_file_id = None
+    try:
+        body = await request.json()
+        evict_file_id = body.get("evict_file_id", None)
+    except Exception:
+        pass
+    action = Action(evict_file_id=evict_file_id)
     result: StepResult = _env.step(action)
     return result.dict()
 
-
 @app.get("/state")
 def state():
     global _env
@@ -98,7 +90,6 @@ def state():
         raise HTTPException(status_code=400, detail="Call /reset first.")
     return _env.state()
 
-
 @app.get("/")
 def root():
     return {
@@ -108,6 +99,5 @@ def root():
         "tasks": list(TASK_CONFIGS.keys()),
     }
 
-
 if __name__ == "__main__":
-    uvicorn.run("api.main:app", host="0.0.0.0", port=7860, reload=False)
+    uvicorn.run("api.main:app", host="0.0.0.0", port=7860, reload=False)