FIX: Add middleware to correct /reset response schema - add 'info' field, remove reward/done

server/app.py

@@ -5,6 +5,8 @@ Uses OpenEnv's create_fastapi_app() for standard framework compatibility
 """
 import json
 import gradio as gr
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.responses import JSONResponse
 
 from openenv.core.env_server import create_fastapi_app
 from server.models import ConfigDebugAction, ConfigDebugObservation, ConfigDebugState
@@ -27,6 +29,39 @@ for i, route in enumerate(app.router.routes):
         print("[APP_INIT] Removed default /metadata route for override")
         break
 
+# ---- Middleware to fix /reset response schema ----
+# OpenEnv returns {"observation": {...}, "reward": 0.0, "done": false}
+# but validator expects {"observation": {...}, "info": {}}
+class ResetSchemaFixMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request, call_next):
+        response = await call_next(request)
+        
+        # Only fix /reset responses
+        if request.url.path == "/reset" and request.method == "POST":
+            if response.status_code == 200:
+                try:
+                    # Get response body
+                    body = b""
+                    async for chunk in response.body_iterator:
+                        body += chunk
+                    
+                    data = json.loads(body)
+                    
+                    # Fix schema: keep only observation and add info
+                    if isinstance(data, dict) and "observation" in data:
+                        fixed_data = {
+                            "observation": data["observation"],
+                            "info": {}
+                        }
+                        print("[MIDDLEWARE] Fixed /reset response schema")
+                        return JSONResponse(fixed_data, status_code=200)
+                except Exception as e:
+                    print(f"[MIDDLEWARE] Error fixing reset response: {e}")
+        
+        return response
+
+app.add_middleware(ResetSchemaFixMiddleware)
+
 # ---- Startup Diagnostics ----
 print("[APP_INIT] ConfigDebugEnvironment initialization started")
 print(f"[APP_INIT] Loaded {len(TASK_ORDER)} tasks: {TASK_ORDER}")
@@ -153,7 +188,7 @@ def metadata():
     print("[VALIDATOR] GET /metadata called")
     return {
         "name": "ConfigDebugEnvironment",
-        "description": "An RL environment for training AI agents to debug broken configuration files",
+        "description": "An environment for training AI agents to debug broken configuration files",
         "version": "1.0.0",
         "tasks": [
             {