Fix: support multiple action formats

app.py

@@ -37,11 +37,23 @@ async def reset(req: ResetRequest):
 
 
 @app.post("/step", response_model=StepResponse)
-async def step(action: Optional[MLOpsAction] = None):
+async def step(action: Optional[MLOpsAction] = None, data: Optional[dict] = None):
     if _http_env is None:
         raise HTTPException(400, "Call /reset first.")
+
+    # Handle various input formats
+    if action is None and data is not None:
+        # Try different formats
+        if "action_type" in data:
+            action = MLOpsAction(**data)
+        elif "action" in data:
+            action = MLOpsAction(**data["action"])
+        elif "message" in data:
+            action = MLOpsAction(action_type=data["message"])
+
     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)