krishnachoudhary-hclguvi commited on
Commit
08b81aa
·
unverified ·
1 Parent(s): 24dd7bb

fix(server): make /reset endpoint ignore body parsing errors fully

Browse files
Files changed (1) hide show
  1. server/app.py +8 -8
server/app.py CHANGED
@@ -1,9 +1,7 @@
1
- from fastapi import FastAPI, HTTPException
2
  from pydantic import BaseModel
3
  from server.environment import CodeReviewEnv
4
 
5
- from typing import Optional
6
-
7
  app = FastAPI(title="OpenEnv Code Review Space")
8
 
9
  _env = None
@@ -11,17 +9,19 @@ _env = None
11
  class ActionRequest(BaseModel):
12
  action: str
13
 
14
- class ResetRequest(BaseModel):
15
- difficulty: str = "medium"
16
-
17
  @app.get("/")
18
  def ping():
19
  return {"status": "ok", "message": "Hugging Face Space is running"}
20
 
21
  @app.post("/reset")
22
- def reset(req: Optional[ResetRequest] = None):
23
  global _env
24
- difficulty = req.difficulty if req else "medium"
 
 
 
 
 
25
  _env = CodeReviewEnv(difficulty=difficulty)
26
  obs = _env.reset()
27
  return {"observation": obs, "status": "reset_successful"}
 
1
+ from fastapi import FastAPI, HTTPException, Request
2
  from pydantic import BaseModel
3
  from server.environment import CodeReviewEnv
4
 
 
 
5
  app = FastAPI(title="OpenEnv Code Review Space")
6
 
7
  _env = None
 
9
  class ActionRequest(BaseModel):
10
  action: str
11
 
 
 
 
12
  @app.get("/")
13
  def ping():
14
  return {"status": "ok", "message": "Hugging Face Space is running"}
15
 
16
  @app.post("/reset")
17
+ async def reset(request: Request):
18
  global _env
19
+ try:
20
+ body = await request.json()
21
+ except Exception:
22
+ body = {}
23
+
24
+ difficulty = (body or {}).get("difficulty", "medium")
25
  _env = CodeReviewEnv(difficulty=difficulty)
26
  obs = _env.reset()
27
  return {"observation": obs, "status": "reset_successful"}