Allow /reset params via body or query; add tests

Make the /reset endpoint more robust by accepting task_id and seed from either a JSON body or query parameters. ResetRequest.task_id is now optional (defaults to BUG_DETECTION). The endpoint resolves final values with precedence: default < body < query, then calls env.reset(final_task_id, final_seed). Added fastapi Body import, an info log on reset, and a test (test_api_reset_robustness) that verifies no-body, empty-body, query-only, and query-overrides-body behavior.

app.py

@@ -6,7 +6,7 @@ from typing import Dict, List, Optional, AsyncGenerator
 from datetime import datetime, timezone
 from contextlib import asynccontextmanager
 
-from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect, Depends, Security, Query, BackgroundTasks, Request
+from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect, Depends, Security, Query, BackgroundTasks, Request, Body
 from fastapi.responses import JSONResponse, FileResponse
 from fastapi.exceptions import RequestValidationError
 from fastapi.security.api_key import APIKeyHeader
@@ -148,7 +148,7 @@ async def cleanup_expired_episodes():
 
 # ── Models ────────────────────────────────────────────────────────────────────
 class ResetRequest(BaseModel):
-    task_id: TaskId
+    task_id: Optional[TaskId] = TaskId.BUG_DETECTION
     seed:    int = 42
 
 class ResetResponse(BaseModel):
@@ -216,12 +216,35 @@ def health_check():
 
 @app.post("/reset", response_model=ResetResponse)
 @limiter.limit(f"{settings.rate_limit_per_minute}/minute")
-def reset_env(request: Request, req: ResetRequest, _: None = Depends(verify_api_key)):
+def reset_env(
+    request: Request,
+    req: Optional[ResetRequest] = Body(None),
+    task_id: Optional[TaskId] = Query(None),
+    seed: Optional[int] = Query(None),
+    _: None = Depends(verify_api_key)
+):
+    # Determine task_id: Body (if present) > Query params > Default
+    final_task_id = TaskId.BUG_DETECTION
+    final_seed = 42
+
+    if req:
+        if req.task_id:
+            final_task_id = req.task_id
+        final_seed = req.seed
+    
+    # Query parameters override body if provided explicitly
+    if task_id:
+        final_task_id = task_id
+    if seed is not None:
+        final_seed = seed
+
     episode_id = str(uuid.uuid4())
     env        = CodeLensEnv()
-    result     = env.reset(req.task_id, req.seed)
+    result     = env.reset(final_task_id, final_seed)
     episodes[episode_id] = env
     episode_timestamps[episode_id] = datetime.now(timezone.utc)
+    
+    logger.info(f"Reset environment: task={final_task_id.value}, seed={final_seed}, id={episode_id}")
     return ResetResponse(episode_id=episode_id, result=result)
 
 @app.post("/step/{episode_id}", response_model=StepResult)

tests/test_api.py

@@ -139,3 +139,29 @@ def test_api_leaderboard_pagination(client):
     
     # Test ordering (best first)
     assert data["entries"][0]["score"] >= data["entries"][1]["score"]
+
+def test_api_reset_robustness(client):
+    # 1. No body at all
+    resp = client.post("/reset")
+    assert resp.status_code == 200
+    assert resp.json()["result"]["task_id"] == "bug_detection"
+
+    # 2. Empty JSON body
+    resp = client.post("/reset", json={})
+    assert resp.status_code == 200
+    assert resp.json()["result"]["task_id"] == "bug_detection"
+
+    # 3. Query params only
+    resp = client.post("/reset?task_id=security_audit&seed=100")
+    assert resp.status_code == 200
+    assert resp.json()["result"]["task_id"] == "security_audit"
+    assert resp.json()["result"]["seed"] == 100
+
+    # 4. Query params overriding body
+    resp = client.post(
+        "/reset?task_id=architectural_review", 
+        json={"task_id": "bug_detection", "seed": 50}
+    )
+    assert resp.status_code == 200
+    assert resp.json()["result"]["task_id"] == "architectural_review"
+    assert resp.json()["result"]["seed"] == 50