shreyas-garg commited on
Commit
cf77682
·
1 Parent(s): f73e70d

Make reset body optional for OpenEnv validator

Browse files
Files changed (1) hide show
  1. email_env/server/app.py +4 -2
email_env/server/app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from fastapi import FastAPI, HTTPException
2
  from pydantic import BaseModel
3
 
@@ -32,9 +33,10 @@ class GradeRequest(BaseModel):
32
 
33
 
34
  @app.post("/reset", response_model=Observation)
35
- def reset(req: ResetRequest):
 
36
  try:
37
- obs = env.reset(task_id=req.task_id)
38
  except ValueError as e:
39
  raise HTTPException(status_code=400, detail=str(e))
40
  return obs
 
1
+ from typing import Optional
2
  from fastapi import FastAPI, HTTPException
3
  from pydantic import BaseModel
4
 
 
33
 
34
 
35
  @app.post("/reset", response_model=Observation)
36
+ def reset(req: Optional[ResetRequest] = None):
37
+ task_id = req.task_id if req else "task_1"
38
  try:
39
+ obs = env.reset(task_id=task_id)
40
  except ValueError as e:
41
  raise HTTPException(status_code=400, detail=str(e))
42
  return obs