Spaces:
Sleeping
Sleeping
File size: 2,380 Bytes
f4e02da da00adf f4e02da da00adf f4e02da da00adf f4e02da d8abf58 f4e02da 232c3f8 f4e02da da00adf f4e02da 232c3f8 f4e02da d8abf58 f4e02da 397179a d8abf58 e1c9a45 f4e02da 397179a d8abf58 397179a e1c9a45 397179a f4e02da | 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 | from __future__ import annotations
from pathlib import Path
from fastapi import FastAPI, HTTPException
from fastapi.responses import HTMLResponse
from pydantic import BaseModel
import uvicorn
from openenv_support_triage.environment import SupportTriageEnv
from openenv_support_triage.models import ActionModel
from openenv_support_triage.tasks import TASKS
app = FastAPI(title="OpenEnv Support Triage", version="0.1.0")
env = SupportTriageEnv()
UI_FILE = Path(__file__).resolve().parents[1] / "ui" / "index.html"
def _one_decimal_score(value: float) -> float:
return min(0.9, max(0.1, round(float(value), 1)))
class ResetRequest(BaseModel):
task_id: str | None = None
@app.get("/", response_class=HTMLResponse)
def home() -> str:
if UI_FILE.exists():
return UI_FILE.read_text(encoding="utf-8")
return "<h1>OpenEnv Support Triage</h1><p>UI file missing.</p>"
@app.get("/health")
def health() -> dict:
return {"status": "ok", "tasks": sorted(list(TASKS.keys()))}
@app.post("/reset")
def reset(req: ResetRequest | None = None):
task_id = req.task_id if req and req.task_id else env.state().task_id
if task_id not in TASKS:
raise HTTPException(status_code=400, detail=f"Unknown task_id: {task_id}")
return env.reset(task_id=task_id).model_dump()
@app.post("/step")
def step(action: ActionModel):
observation, reward, done, info = env.step(action)
score = _one_decimal_score(info.get("score", info.get("task_score", info.get("running_score", 0.1))))
return {
"observation": observation.model_dump(),
"reward": reward.model_dump(),
"done": done,
"info": info,
"score": score,
"task_score": _one_decimal_score(info.get("task_score", score)),
"final_score": _one_decimal_score(info.get("final_score", score)),
"normalized_score": _one_decimal_score(info.get("normalized_score", score)),
}
@app.get("/state")
def state():
state = env.state().model_dump()
grade = state.get("metadata", {}).get("grade", {})
score = _one_decimal_score(grade.get("overall", 0.1))
state["score"] = score
state["task_score"] = score
state["final_score"] = score
state["normalized_score"] = score
return state
def main() -> None:
uvicorn.run("server.app:app", host="0.0.0.0", port=7860)
if __name__ == "__main__":
main()
|