shank commited on
Commit
f2ee2fc
·
1 Parent(s): e766743

made changes to server.py

Browse files
Files changed (1) hide show
  1. env/server.py +5 -4
env/server.py CHANGED
@@ -28,7 +28,7 @@ env = DebuggerEnvironment()
28
 
29
 
30
  class ResetRequest(BaseModel):
31
- task_id: str
32
 
33
 
34
  @app.get("/health")
@@ -38,10 +38,11 @@ async def health():
38
 
39
 
40
  @app.post("/reset")
41
- async def reset(request: ResetRequest):
42
- """Start a fresh episode. Returns initial Observation."""
43
  try:
44
- observation = env.reset(request.task_id)
 
45
  return JSONResponse(content=observation, status_code=200)
46
  except ValueError as e:
47
  return JSONResponse(
 
28
 
29
 
30
  class ResetRequest(BaseModel):
31
+ task_id: Optional[str] = "easy"
32
 
33
 
34
  @app.get("/health")
 
38
 
39
 
40
  @app.post("/reset")
41
+ async def reset(request: Optional[ResetRequest] = None):
42
+ """Start a fresh episode. Returns initial Observation. Default to 'easy' task if body is missing."""
43
  try:
44
+ task_id = request.task_id if request else "easy"
45
+ observation = env.reset(task_id)
46
  return JSONResponse(content=observation, status_code=200)
47
  except ValueError as e:
48
  return JSONResponse(