Spaces:
Sleeping
Sleeping
File size: 2,281 Bytes
c14504c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | """
FastAPI server exposing the OpenEnv-compatible HTTP API.
Port: 7860 (Hugging Face Spaces default)
"""
from fastapi import FastAPI, HTTPException
from fastapi.responses import JSONResponse
from ..models import (
DebugState,
ResetRequest,
ResetResponse,
StepRequest,
StepResponse,
)
from .environment import CodeDebugEnvironment
app = FastAPI(
title="Code Debug OpenEnv",
description="An OpenEnv environment where an AI agent debugs broken Python code.",
version="1.0.0",
)
env = CodeDebugEnvironment()
# ------------------------------------------------------------------
# Health & metadata
# ------------------------------------------------------------------
@app.get("/health")
async def health():
return {"status": "healthy", "tasks_loaded": len(env.tasks)}
@app.get("/tasks")
async def list_tasks():
return env.list_tasks()
# ------------------------------------------------------------------
# OpenEnv core endpoints
# ------------------------------------------------------------------
@app.post("/reset", response_model=ResetResponse)
async def reset(req: ResetRequest = None):
if req is None:
req = ResetRequest()
try:
result = env.reset(task_id=req.task_id, seed=req.seed)
return result
except KeyError as e:
raise HTTPException(status_code=404, detail=str(e))
@app.post("/step/{episode_id}", response_model=StepResponse)
async def step(episode_id: str, req: StepRequest):
try:
result = env.step(episode_id, req.action.model_dump())
return result
except KeyError as e:
raise HTTPException(status_code=404, detail=str(e))
except ValueError as e:
raise HTTPException(status_code=400, detail=str(e))
@app.get("/state/{episode_id}", response_model=DebugState)
async def state(episode_id: str):
try:
return env.state(episode_id)
except KeyError as e:
raise HTTPException(status_code=404, detail=str(e))
# ------------------------------------------------------------------
# Entry point for local dev
# ------------------------------------------------------------------
if __name__ == "__main__":
import uvicorn
uvicorn.run("code_debug_env.server.app:app", host="0.0.0.0", port=7860, reload=True)
|