Spaces:
Sleeping
Sleeping
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
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 |
-
|
|
|
|
| 50 |
obs = env.reset()
|
| 51 |
-
_envs[
|
| 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 |
|