Upload 2 files

server/app.py

@@ -0,0 +1,79 @@
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
+
+from fastapi import FastAPI, HTTPException
+from typing import Optional
+from environment import CodeReviewEnv
+from models import Action, Observation, State
+
+app = FastAPI(title="Code Review Environment", version="1.0.0")
+env = CodeReviewEnv()  # default task "easy" is set in __init__
+
+# ------------------------- OpenEnv required endpoints -------------------------
+
+@app.get("/health")
+def health():
+    return {"status": "healthy"}
+
+@app.get("/metadata")
+def metadata():
+    return {
+        "name": "Code Review Environment",
+        "description": "Simulate reviewing a PR and adding helpful comments. Three tasks with increasing difficulty."
+    }
+
+@app.get("/schema")
+def schema():
+    return {
+        "action": Action.model_json_schema(),
+        "observation": Observation.model_json_schema(),
+        "state": State.model_json_schema()
+    }
+
+@app.post("/mcp")
+def mcp():
+    # Minimal JSON-RPC 2.0 response
+    return {"jsonrpc": "2.0", "result": None}
+
+# ------------------------- Your environment endpoints -------------------------
+
+@app.post("/reset")
+def reset(task: Optional[str] = None):
+    if task:
+        try:
+            env.set_task(task)
+        except ValueError as e:
+            raise HTTPException(status_code=400, detail=str(e))
+    obs = env.reset()
+    return obs.dict()
+
+@app.post("/set_task")
+def set_task(task: str):
+    try:
+        env.set_task(task)
+        return {"status": "ok", "task": task}
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+@app.post("/step")
+def step(action: dict):
+    action_obj = Action(**action)
+    obs, reward, done, info = env.step(action_obj)
+    return {
+        "observation": obs.dict(),
+        "reward": reward.value,
+        "done": done,
+        "info": info
+    }
+
+@app.get("/state")
+def state():
+    return env.state().dict()
+
+def main():
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+
+if __name__ == "__main__":
+    main()