Fix OpenEnv validation issues

Dockerfile

@@ -5,4 +5,4 @@ COPY . .
 
 RUN pip install --no-cache-dir numpy fastapi uvicorn
 
-CMD ["uvicorn", "inference:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "server.app:app", "--host", "0.0.0.0", "--port", "7860"]

pyproject.toml

@@ -8,9 +8,13 @@ authors = [
 dependencies = [
     "numpy",
     "fastapi",
-    "uvicorn"
+    "uvicorn",
+    "openenv-core>=0.2.0"
 ]
 
+[project.scripts]
+server = "server.app:app"
+
 [build-system]
 requires = ["setuptools"]
 build-backend = "setuptools.build_meta"

server/app.py

@@ -0,0 +1,33 @@
+from fastapi import FastAPI
+import json
+
+from env.feed_env import FeedRankingEnv
+from agents.random_agent import RandomAgent
+
+app = FastAPI()
+
+with open("data/posts.json", "r") as f:
+    posts = json.load(f)
+
+env = FeedRankingEnv(posts, task="hard")
+agent = RandomAgent()
+
+@app.get("/")
+def root():
+    return {"message": "OpenEnv server running"}
+
+@app.post("/reset")
+def reset():
+    state = env.reset()
+    return {"state": state.__dict__}
+
+@app.post("/step")
+def step():
+    action = agent.act(env.state(), env.posts)
+    state, reward, done, _ = env.step(action)
+
+    return {
+        "state": state.__dict__,
+        "reward": reward,
+        "done": done
+    }