Fix: support optional task_id/task field and action_type

app.py

@@ -22,84 +22,26 @@ _http_env: Optional[MLOpsEnvironment] = None
 
 
 class ResetRequest(BaseModel):
-    task_id: str = "easy"
+    task_id: Optional[str] = "easy"
     seed: Optional[int] = None
-
-
-class StepResponse(BaseModel):
-    observation: MLOpsObservation
-    reward: float
-    done: bool
-    info: Dict[str, Any]
-
-
-@app.get("/")
-async def root():
-    return {
-        "message": "MLOps Pipeline Debugger API",
-        "version": "1.0.0",
-        "docs": "This is an OpenEnv-compatible RL environment",
-        "endpoints": {
-            "GET /": "This message",
-            "GET /health": "Health check",
-            "GET /tasks": "List available tasks",
-            "GET /openenv/state": "OpenEnv state",
-            "POST /reset": "Start a new episode",
-            "POST /step": "Take an action",
-            "GET /state": "Get current state",
-        },
-        "space_url": "https://angelgupta-mlops-openenv.hf.space",
-    }
-
-
-@app.get("/health")
-async def health():
-    return {"status": "ok", "environment": "mlops_debug_env", "version": "1.0.0"}
-
-
-@app.get("/openenv/state", response_model=OpenEnvState)
-def openenv_state():
-    # Expose the current OpenEnv-like state persisted in memory/state.json
-    return OPENENV_STATE
-
-
-@app.get("/tasks")
-async def list_tasks():
-    return {
-        "tasks": [
-            {
-                "task_id": "easy",
-                "name": "Config Error Diagnosis",
-                "difficulty": "easy",
-                "max_steps": 20,
-            },
-            {
-                "task_id": "medium",
-                "name": "Data Leakage Detection",
-                "difficulty": "medium",
-                "max_steps": 30,
-            },
-            {
-                "task_id": "hard",
-                "name": "Silent Evaluation Bug",
-                "difficulty": "hard",
-                "max_steps": 40,
-            },
-        ]
-    }
+    task: Optional[str] = None  # Support both task_id and task
 
 
 @app.post("/reset", response_model=MLOpsObservation)
 async def reset(req: ResetRequest):
     global _http_env
-    _http_env = MLOpsEnvironment(task_id=req.task_id)
+    # Support both field names for compatibility
+    task_id = req.task_id or req.task or "easy"
+    _http_env = MLOpsEnvironment(task_id=task_id)
     return _http_env.reset(seed=req.seed)
 
 
 @app.post("/step", response_model=StepResponse)
-async def step(action: MLOpsAction):
+async def step(action: Optional[MLOpsAction] = None):
     if _http_env is None:
         raise HTTPException(400, "Call /reset first.")
+    if action is None or action.action_type is None:
+        raise HTTPException(422, "Field required: action_type")
     obs, reward, done, info = _http_env.step(action)
     return StepResponse(observation=obs, reward=reward, done=done, info=info)