RealBhupesh commited on
Commit ·
60e6d0d
1
Parent(s): f9dd82f
Fix /reset to accept empty POST body
Browse filesMake the reset request body optional so OpenEnv automation that posts without a JSON payload no longer fails with 422 'Field required'. Keep existing behavior unchanged when a body is provided.
Made-with: Cursor
app.py
CHANGED
|
@@ -3,7 +3,7 @@ from __future__ import annotations
|
|
| 3 |
from pathlib import Path
|
| 4 |
from typing import Any
|
| 5 |
|
| 6 |
-
from fastapi import FastAPI, HTTPException
|
| 7 |
from fastapi.responses import PlainTextResponse
|
| 8 |
from pydantic import BaseModel, Field
|
| 9 |
|
|
@@ -67,7 +67,8 @@ def tasks() -> list[dict[str, Any]]:
|
|
| 67 |
|
| 68 |
|
| 69 |
@app.post("/reset")
|
| 70 |
-
def reset(request: ResetRequest) -> dict[str, Any]:
|
|
|
|
| 71 |
env = EmailTriageEnv(task_id=request.task_id, seed=request.seed)
|
| 72 |
_sessions[request.session_id] = env
|
| 73 |
observation = env.reset(scenario_id=request.scenario_id)
|
|
|
|
| 3 |
from pathlib import Path
|
| 4 |
from typing import Any
|
| 5 |
|
| 6 |
+
from fastapi import Body, FastAPI, HTTPException
|
| 7 |
from fastapi.responses import PlainTextResponse
|
| 8 |
from pydantic import BaseModel, Field
|
| 9 |
|
|
|
|
| 67 |
|
| 68 |
|
| 69 |
@app.post("/reset")
|
| 70 |
+
def reset(request: ResetRequest | None = Body(default=None)) -> dict[str, Any]:
|
| 71 |
+
request = request or ResetRequest()
|
| 72 |
env = EmailTriageEnv(task_id=request.task_id, seed=request.seed)
|
| 73 |
_sessions[request.session_id] = env
|
| 74 |
observation = env.reset(scenario_id=request.scenario_id)
|