janakb commited on
Commit
efeaffa
·
1 Parent(s): 2573603
Files changed (1) hide show
  1. server.py +5 -2
server.py CHANGED
@@ -41,6 +41,8 @@ class ResetRequest(BaseModel):
41
  task_id: str = "task_1_easy_bug_hunt"
42
  session_id: str | None = None
43
 
 
 
44
 
45
  class ResetResponse(BaseModel):
46
  session_id: str
@@ -86,9 +88,10 @@ def list_tasks():
86
  })
87
  return {"tasks": tasks}
88
 
89
-
90
  @app.post("/reset", response_model=ResetResponse)
91
- def reset(req: ResetRequest):
 
 
92
  if req.task_id not in TASK_IDS:
93
  raise HTTPException(400, f"Unknown task_id {req.task_id!r}. Choose from {TASK_IDS}")
94
  session_id = req.session_id or str(uuid.uuid4())
 
41
  task_id: str = "task_1_easy_bug_hunt"
42
  session_id: str | None = None
43
 
44
+ model_config = {"extra": "allow"}
45
+
46
 
47
  class ResetResponse(BaseModel):
48
  session_id: str
 
88
  })
89
  return {"tasks": tasks}
90
 
 
91
  @app.post("/reset", response_model=ResetResponse)
92
+ def reset(req: ResetRequest | None = None):
93
+ if req is None:
94
+ req = ResetRequest()
95
  if req.task_id not in TASK_IDS:
96
  raise HTTPException(400, f"Unknown task_id {req.task_id!r}. Choose from {TASK_IDS}")
97
  session_id = req.session_id or str(uuid.uuid4())