Dev Shah commited on
Commit
c3935be
·
1 Parent(s): 4ed3421

fix: make /reset body optional so bare POST (no Content-Type/body) returns JSON not 422 HTML

Browse files
Files changed (1) hide show
  1. server.py +5 -4
server.py CHANGED
@@ -3,7 +3,7 @@ FastAPI server exposing the Email Triage environment via HTTP.
3
  Endpoints mirror the OpenEnv spec.
4
  """
5
 
6
- from fastapi import FastAPI, HTTPException
7
  from fastapi.middleware.cors import CORSMiddleware
8
  from pydantic import BaseModel
9
  from typing import Optional
@@ -45,10 +45,11 @@ def health():
45
 
46
 
47
  @app.post("/reset")
48
- def reset(req: ResetRequest):
49
- env = EmailTriageEnv(task=req.task)
 
50
  obs = env.reset()
51
- _envs[req.task] = env
52
  return {"observation": obs.model_dump(), "state": env.state()}
53
 
54
 
 
3
  Endpoints mirror the OpenEnv spec.
4
  """
5
 
6
+ from fastapi import FastAPI, HTTPException, Body
7
  from fastapi.middleware.cors import CORSMiddleware
8
  from pydantic import BaseModel
9
  from typing import Optional
 
45
 
46
 
47
  @app.post("/reset")
48
+ def reset(req: Optional[ResetRequest] = Body(default=None)):
49
+ task = req.task if req else 1
50
+ env = EmailTriageEnv(task=task)
51
  obs = env.reset()
52
+ _envs[task] = env
53
  return {"observation": obs.model_dump(), "state": env.state()}
54
 
55