Spaces:
Running
Running
ajaxwin commited on
Commit ·
a503619
1
Parent(s): cfd3cfa
fix: Handle optional request body in reset endpoint and set default task ID
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 167 |
_sessions[session_id] = env
|
| 168 |
-
|
|
|
|
| 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 |
|