Changes for app.py

app.py

@@ -12,10 +12,8 @@ from typing import Dict, Any, Optional
 
 import os
 import uvicorn
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from fastapi import FastAPI, HTTPException, Body
 
-# Import your environments
 from tasks.task_simple import create_env as create_simple
 from tasks.task_multi_service import create_env as create_multi
 from tasks.task_critical import create_env as create_critical
@@ -23,15 +21,8 @@ from tasks.task_critical import create_env as create_critical
 
 app = FastAPI(title="DebugOps AI Environment", version="1.0.0")
 
-# Global env instance
-_env = None
-
-class ResetRequest(BaseModel):
-    task: str = "simple"   # simple | multi_service | critical
-
-
-class StepRequest(BaseModel):
-    action: str
+# Global environment instance
+_env: Optional[Any] = None
 
 def create_env(task: str):
     if task == "simple":
@@ -65,10 +56,19 @@ def health():
 
 
 @app.post("/reset")
-def reset(request: ResetRequest):
+@app.post("/reset/")
+def reset(payload: Optional[Dict[str, Any]] = Body(default={})):
+    """
+    Accepts BOTH:
+      {}                         (validator case)
+      {"task": "critical"}       (manual case)
+    """
     global _env
+
     try:
-        _env = create_env(request.task)
+        task = payload.get("task", "simple") if payload else "simple"
+
+        _env = create_env(task)
         obs = _env.reset()
 
         return {
@@ -81,11 +81,16 @@ def reset(request: ResetRequest):
 
 
 @app.post("/step")
-def step(request: StepRequest):
+@app.post("/step/")
+def step(payload: Dict[str, Any] = Body(...)):
     env = get_env()
 
     try:
-        obs, reward, done, info = env.step(request.action)
+        action = payload.get("action")
+        if not action:
+            raise ValueError("Missing 'action' field")
+
+        obs, reward, done, info = env.step(action)
 
         return {
             "observation": obs,
@@ -99,6 +104,7 @@ def step(request: StepRequest):
 
 
 @app.get("/state")
+@app.get("/state/")
 def state():
     env = get_env()
     try: