ajaxwin commited on
Commit
a503619
·
1 Parent(s): cfd3cfa

fix: Handle optional request body in reset endpoint and set default task ID

Browse files
Files changed (1) hide show
  1. server/app.py +13 -4
server/app.py CHANGED
@@ -156,16 +156,25 @@ def list_tasks():
156
  ]
157
  return {"tasks": [t.model_dump() for t in tasks]}
158
 
159
-
160
  @app.post("/reset")
161
  def reset(
162
- body: ResetRequest,
163
  session_id: str = Query(default=DEFAULT_SESSION),
164
  ):
165
  """Reset the environment and start a new episode."""
166
- env = _create_env(body.task_id)
 
 
 
 
 
 
 
 
 
167
  _sessions[session_id] = env
168
- result = env.reset(seed=body.seed)
 
169
  return JSONResponse(content=result.model_dump(), status_code=200)
170
 
171
 
 
156
  ]
157
  return {"tasks": [t.model_dump() for t in tasks]}
158
 
 
159
  @app.post("/reset")
160
  def reset(
161
+ body: Optional[ResetRequest] = None,
162
  session_id: str = Query(default=DEFAULT_SESSION),
163
  ):
164
  """Reset the environment and start a new episode."""
165
+
166
+ # Handle missing body (OpenEnv validator case)
167
+ if body is None:
168
+ task_id = "task1_vuln_detection"
169
+ seed = None
170
+ else:
171
+ task_id = body.task_id
172
+ seed = body.seed
173
+
174
+ env = _create_env(task_id)
175
  _sessions[session_id] = env
176
+
177
+ result = env.reset(seed=seed)
178
  return JSONResponse(content=result.model_dump(), status_code=200)
179
 
180