Upload 3 files

Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.11-slim
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . /code/
+
+ENV PYTHONPATH=/code
+
+# Create server/ folder and copy server_app.py into it as app.py
+RUN mkdir -p /code/server && cp /code/server_app.py /code/server/app.py
+
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+  CMD python -c "from meta import DataCenterEnv; e=DataCenterEnv(); e.reset(); e.step(0); print('ok')"
+
+CMD ["sh", "-c", "python inference.py && uvicorn server.app:app --host 0.0.0.0 --port 7860"]

app.py

@@ -0,0 +1,52 @@
+"""
+app.py — OpenEnv-compatible FastAPI server for DataCenterEnv.
+Exposes /reset, /step, /state endpoints as required by the OpenEnv spec.
+"""
+
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from typing import Any, Dict
+import uvicorn
+
+from meta import DataCenterEnv, EnvState
+
+app = FastAPI(title="DataCenter OpenEnv", version="1.0.0")
+
+# Global environment instance
+env = DataCenterEnv(seed=42)
+
+class StepRequest(BaseModel):
+    action: int
+
+@app.post("/reset")
+def reset():
+    state = env.reset()
+    return {"state": state.model_dump()}
+
+@app.post("/step")
+def step(req: StepRequest):
+    if req.action not in (0, 1, 2, 3):
+        raise HTTPException(status_code=400, detail=f"Invalid action {req.action}. Must be 0-3.")
+    state, reward, done, info = env.step(req.action)
+    return {"state": state.model_dump(), "reward": reward, "done": done, "info": info}
+
+@app.get("/state")
+def state():
+    return {"state": env.state().model_dump()}
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}
+
+@app.get("/")
+def root():
+    return {
+        "name": "DataCenter OpenEnv",
+        "version": "1.0.0",
+        "endpoints": ["/reset", "/step", "/state", "/health"],
+        "action_space": {"type": "Discrete", "n": 4},
+        "observation_space": {"type": "Box", "shape": [5]},
+    }
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)

server_app.py

@@ -0,0 +1,52 @@
+"""
+app.py — OpenEnv-compatible FastAPI server for DataCenterEnv.
+Exposes /reset, /step, /state endpoints as required by the OpenEnv spec.
+"""
+
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from typing import Any, Dict
+import uvicorn
+
+from meta import DataCenterEnv, EnvState
+
+app = FastAPI(title="DataCenter OpenEnv", version="1.0.0")
+
+# Global environment instance
+env = DataCenterEnv(seed=42)
+
+class StepRequest(BaseModel):
+    action: int
+
+@app.post("/reset")
+def reset():
+    state = env.reset()
+    return {"state": state.model_dump()}
+
+@app.post("/step")
+def step(req: StepRequest):
+    if req.action not in (0, 1, 2, 3):
+        raise HTTPException(status_code=400, detail=f"Invalid action {req.action}. Must be 0-3.")
+    state, reward, done, info = env.step(req.action)
+    return {"state": state.model_dump(), "reward": reward, "done": done, "info": info}
+
+@app.get("/state")
+def state():
+    return {"state": env.state().model_dump()}
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}
+
+@app.get("/")
+def root():
+    return {
+        "name": "DataCenter OpenEnv",
+        "version": "1.0.0",
+        "endpoints": ["/reset", "/step", "/state", "/health"],
+        "action_space": {"type": "Discrete", "n": 4},
+        "observation_space": {"type": "Box", "shape": [5]},
+    }
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)