krishnachoudhary-hclguvi commited on
fix: make reset request body optional
Browse files- 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 |
-
|
|
|
|
| 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 |
|