fix step format, add tasks endpoint back

inference.py

@@ -216,9 +216,7 @@ async def run_task(task_name: str) -> None:
 
                 # Submit action to environment
                 try:
-                    step_resp = await http.post(
-                        "/step", json={"corrected_query": action_str}
-                    )
+                    step_resp = await http.post("/step", json={"action":{"corrected_query": action_str}})
                     step_resp.raise_for_status()
                     result = step_resp.json()
                 except Exception as exc:

sql_env/server.py

@@ -105,4 +105,17 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()
+from fastapi import Request
+
+@app.get("/tasks")
+async def list_tasks():
+    """Return graded tasks in openenv validator format."""
+    from sql_env.grader import grade
+    return {
+        "tasks": [
+            {"id": "easy", "difficulty": "easy", "description": "Fix a single syntax error.", "steps": 5, "ideal_action": "correct_sql", "has_grader": True},
+            {"id": "medium", "difficulty": "medium", "description": "Fix multiple errors.", "steps": 5, "ideal_action": "correct_sql", "has_grader": True},
+            {"id": "hard", "difficulty": "hard", "description": "Fix complex multi-join queries.", "steps": 4, "ideal_action": "correct_sql", "has_grader": True},
+        ]
+    }