Dar3devil commited on
Commit
1f76703
·
verified ·
1 Parent(s): ae5e31e

Allow empty POST body on reset

Browse files
Files changed (1) hide show
  1. server/app.py +5 -4
server/app.py CHANGED
@@ -4,7 +4,7 @@ from threading import Lock
4
  from uuid import uuid4
5
  from typing import Any
6
 
7
- from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
8
  from fastapi.responses import HTMLResponse, RedirectResponse
9
  from pydantic import BaseModel, ConfigDict
10
 
@@ -75,9 +75,10 @@ def root() -> RedirectResponse:
75
 
76
 
77
  @app.post("/reset")
78
- def reset(request: ResetRequest) -> dict[str, Any]:
79
- session_id, env = manager.create_or_reuse(request.session_id, request.task_id)
80
- result = env.reset(request.task_id)
 
81
  return _step_payload(result, session_id)
82
 
83
 
 
4
  from uuid import uuid4
5
  from typing import Any
6
 
7
+ from fastapi import Body, FastAPI, HTTPException, WebSocket, WebSocketDisconnect
8
  from fastapi.responses import HTMLResponse, RedirectResponse
9
  from pydantic import BaseModel, ConfigDict
10
 
 
75
 
76
 
77
  @app.post("/reset")
78
+ def reset(request: ResetRequest | None = Body(default=None)) -> dict[str, Any]:
79
+ request = request or ResetRequest()
80
+ session_id, env = manager.create_or_reuse(request.session_id, request.task_id)
81
+ result = env.reset(request.task_id)
82
  return _step_payload(result, session_id)
83
 
84