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

fix: make reset request body optional

Browse files
Files changed (1) hide show
  1. server/app.py +5 -2
server/app.py CHANGED
@@ -2,6 +2,8 @@ from fastapi import FastAPI, HTTPException
2
  from pydantic import BaseModel
3
  from server.environment import CodeReviewEnv
4
 
 
 
5
  app = FastAPI(title="OpenEnv Code Review Space")
6
 
7
  _env = None
@@ -17,9 +19,10 @@ def ping():
17
  return {"status": "ok", "message": "Hugging Face Space is running"}
18
 
19
  @app.post("/reset")
20
- def reset(req: ResetRequest):
21
  global _env
22
- _env = CodeReviewEnv(difficulty=req.difficulty)
 
23
  obs = _env.reset()
24
  return {"observation": obs, "status": "reset_successful"}
25
 
 
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
 
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"}
28