CRITICAL FIX: Replace /reset middleware with direct endpoint override - ensure base fields included

server/app.py

@@ -7,6 +7,8 @@ import json
 import gradio as gr
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.responses import JSONResponse
+from starlette.types import ASGIApp, Receive, Scope, Send
+from typing import Callable
 
 from openenv.core.env_server import create_fastapi_app
 from server.models import ConfigDebugAction, ConfigDebugObservation, ConfigDebugState
@@ -23,6 +25,14 @@ app = create_fastapi_app(
 
 # ---- Override OpenEnv's default /metadata route ----
 # Remove the built-in metadata endpoint so we can replace it with task enumeration
+reset_endpoint = None
+for i, route in enumerate(app.router.routes):
+    if hasattr(route, "path") and route.path == "/reset":
+        reset_endpoint = route.endpoint
+        app.router.routes.pop(i)
+        print("[APP_INIT] Removed default /reset route for schema fix")
+        break
+
 for i, route in enumerate(app.router.routes):
     if hasattr(route, "path") and route.path == "/metadata":
         app.router.routes.pop(i)
@@ -204,6 +214,39 @@ def metadata():
     }
 
 
+@app.post("/reset")
+async def reset_env(request):
+    """Override /reset endpoint to return correct OpenEnv contract schema.
+    
+    Must include: observation, done, reward, metadata, info
+    """
+    print("[VALIDATOR] POST /reset called - CUSTOM OVERRIDE")
+    
+    if reset_endpoint:
+        try:
+            # Call the original OpenEnv reset endpoint
+            response = await reset_endpoint(request)
+            
+            # If it's a dict, wrap it correctly
+            if isinstance(response, dict) and "observation" in response:
+                fixed_response = {
+                    "observation": response["observation"],
+                    "done": response.get("done", False),
+                    "reward": response.get("reward"),
+                    "metadata": response.get("metadata", {}),
+                    "info": {}
+                }
+                print("[VALIDATOR] Reset response wrapped with base fields")
+                return fixed_response
+            return response
+        except Exception as e:
+            print(f"[VALIDATOR] Error in custom reset: {e}")
+            raise
+    else:
+        print("[VALIDATOR] ERROR: reset_endpoint not saved")
+        raise RuntimeError("Reset endpoint initialization failed")
+
+
 @app.get("/tasks")
 def tasks():
     return {