Shinegupta commited on
Commit
232c3f8
·
1 Parent(s): 3414acf

Allow POST /reset with empty body for validator compatibility

Browse files
Files changed (2) hide show
  1. app.py +6 -5
  2. server/app.py +6 -5
app.py CHANGED
@@ -17,7 +17,7 @@ UI_FILE = Path(__file__).resolve().parent / "ui" / "index.html"
17
 
18
 
19
  class ResetRequest(BaseModel):
20
- task_id: str
21
 
22
 
23
  @app.get("/", response_class=HTMLResponse)
@@ -33,10 +33,11 @@ def health() -> dict:
33
 
34
 
35
  @app.post("/reset")
36
- def reset(req: ResetRequest):
37
- if req.task_id not in TASKS:
38
- raise HTTPException(status_code=400, detail=f"Unknown task_id: {req.task_id}")
39
- return env.reset(task_id=req.task_id).model_dump()
 
40
 
41
 
42
  @app.post("/step")
 
17
 
18
 
19
  class ResetRequest(BaseModel):
20
+ task_id: str | None = None
21
 
22
 
23
  @app.get("/", response_class=HTMLResponse)
 
33
 
34
 
35
  @app.post("/reset")
36
+ def reset(req: ResetRequest | None = None):
37
+ task_id = req.task_id if req and req.task_id else env.state().task_id
38
+ if task_id not in TASKS:
39
+ raise HTTPException(status_code=400, detail=f"Unknown task_id: {task_id}")
40
+ return env.reset(task_id=task_id).model_dump()
41
 
42
 
43
  @app.post("/step")
server/app.py CHANGED
@@ -18,7 +18,7 @@ UI_FILE = Path(__file__).resolve().parents[1] / "ui" / "index.html"
18
 
19
 
20
  class ResetRequest(BaseModel):
21
- task_id: str
22
 
23
 
24
  @app.get("/", response_class=HTMLResponse)
@@ -34,10 +34,11 @@ def health() -> dict:
34
 
35
 
36
  @app.post("/reset")
37
- def reset(req: ResetRequest):
38
- if req.task_id not in TASKS:
39
- raise HTTPException(status_code=400, detail=f"Unknown task_id: {req.task_id}")
40
- return env.reset(task_id=req.task_id).model_dump()
 
41
 
42
 
43
  @app.post("/step")
 
18
 
19
 
20
  class ResetRequest(BaseModel):
21
+ task_id: str | None = None
22
 
23
 
24
  @app.get("/", response_class=HTMLResponse)
 
34
 
35
 
36
  @app.post("/reset")
37
+ def reset(req: ResetRequest | None = None):
38
+ task_id = req.task_id if req and req.task_id else env.state().task_id
39
+ if task_id not in TASKS:
40
+ raise HTTPException(status_code=400, detail=f"Unknown task_id: {task_id}")
41
+ return env.reset(task_id=task_id).model_dump()
42
 
43
 
44
  @app.post("/step")