| """FastAPI server for ForgeEnv (OpenEnv-compliant). |
| |
| Exposes /reset, /step, /state HTTP endpoints via OpenEnv's `create_app`. |
| HF Spaces sets PORT=7860 automatically. |
| """ |
| from __future__ import annotations |
|
|
| import os |
|
|
| from fastapi.responses import HTMLResponse |
| from openenv.core import create_app |
|
|
| from forgeenv.env.actions import ForgeAction |
| from forgeenv.env.forge_environment import ForgeEnvironment |
| from forgeenv.env.observations import ForgeObservation |
|
|
| app = create_app( |
| env=ForgeEnvironment, |
| action_cls=ForgeAction, |
| observation_cls=ForgeObservation, |
| env_name="forgeenv", |
| ) |
|
|
|
|
| _LANDING_HTML = """<!doctype html> |
| <html lang="en"> |
| <head> |
| <meta charset="utf-8"> |
| <title>ForgeEnv — OpenEnv server</title> |
| <meta name="viewport" content="width=device-width,initial-scale=1"> |
| <style> |
| :root { color-scheme: light dark; } |
| body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif; |
| max-width: 760px; margin: 2.5rem auto; padding: 0 1.25rem; |
| line-height: 1.55; color: #1f2937; background: #fafafa; } |
| @media (prefers-color-scheme: dark) { body { color: #e5e7eb; background: #0f172a; } } |
| h1 { font-size: 1.65rem; margin-bottom: 0.25rem; } |
| .sub { color: #6b7280; margin-top: 0; } |
| code, pre { font-family: ui-monospace, "SF Mono", Menlo, monospace; } |
| pre { background: rgba(127,127,127,0.12); padding: 0.9rem; border-radius: 8px; |
| overflow-x: auto; } |
| table { border-collapse: collapse; width: 100%; margin: 0.75rem 0 1.25rem; } |
| td, th { text-align: left; padding: 0.5rem 0.75rem; |
| border-bottom: 1px solid rgba(127,127,127,0.25); } |
| th { font-weight: 600; } |
| a { color: #2563eb; text-decoration: none; } a:hover { text-decoration: underline; } |
| .ok { color: #16a34a; font-weight: 600; } |
| .muted { color: #6b7280; font-size: 0.9rem; } |
| .pill { display: inline-block; padding: 0.1rem 0.5rem; border-radius: 999px; |
| background: rgba(34,197,94,0.15); color: #16a34a; font-size: 0.85rem; } |
| </style> |
| </head> |
| <body> |
| <h1>ForgeEnv 🔧 <span class="pill">running</span></h1> |
| <p class="sub">OpenEnv-compliant RL environment for HuggingFace |
| ecosystem repair under library version drift.</p> |
| |
| <p>This URL serves the environment over HTTP. It is not a UI — it's the |
| runtime that <strong>training notebooks connect to</strong>. Open one of |
| the endpoints below, or use the demo Space to try the trained Repair |
| Agent in a browser.</p> |
| |
| <h2>Endpoints</h2> |
| <table> |
| <tr><th>Method</th><th>Path</th><th>Purpose</th></tr> |
| <tr><td>GET </td><td><a href="/health">/health</a></td><td>Health probe</td></tr> |
| <tr><td>POST</td><td><code>/reset</code></td><td>Sample task, return drift-gen observation</td></tr> |
| <tr><td>POST</td><td><code>/step</code></td><td>Apply <code>ForgeAction</code> (breakage or repair)</td></tr> |
| <tr><td>GET </td><td><a href="/state">/state</a></td><td>Current internal state</td></tr> |
| <tr><td>GET </td><td><a href="/metadata">/metadata</a></td><td>Env name + version + schema URLs</td></tr> |
| <tr><td>GET </td><td><a href="/schema">/schema</a></td><td>Action / observation JSON schemas</td></tr> |
| <tr><td>GET </td><td><a href="/docs">/docs</a></td><td>Interactive Swagger UI</td></tr> |
| </table> |
| |
| <h2>Quick start (Python)</h2> |
| <pre><code>import asyncio |
| from openenv.core import GenericEnvClient |
| |
| async def go(): |
| client = GenericEnvClient(base_url="https://akhiilll-forgeenv.hf.space") |
| obs = await client.reset() |
| print(obs.observation["current_phase"], obs.observation["task_id"]) |
| |
| asyncio.run(go())</code></pre> |
| |
| <h2>Project links</h2> |
| <ul> |
| <li>Space card & README: |
| <a href="https://huggingface.co/spaces/akhiilll/forgeenv" target="_blank" rel="noopener noreferrer">huggingface.co/spaces/akhiilll/forgeenv</a></li> |
| <li>Gradio demo: |
| <a href="https://huggingface.co/spaces/akhiilll/forgeenv-demo" target="_blank" rel="noopener noreferrer">huggingface.co/spaces/akhiilll/forgeenv-demo</a></li> |
| <li>Trained model (LoRA) <span class="muted">— published after the Colab training run finishes</span>: |
| <a href="https://huggingface.co/akhiilll/forgeenv-repair-agent" target="_blank" rel="noopener noreferrer">huggingface.co/akhiilll/forgeenv-repair-agent</a></li> |
| </ul> |
| <p class="muted">Tip: if links don't open from inside the embedded Space frame, |
| right-click and choose <em>Open in new tab</em>, or open this URL directly |
| at <a href="https://akhiilll-forgeenv.hf.space/" target="_blank" rel="noopener noreferrer">akhiilll-forgeenv.hf.space</a>.</p> |
| </body> |
| </html>""" |
|
|
|
|
| def _attach_supplementary_routes(_app) -> None: |
| """Add /health and a friendly GET / landing page if not present.""" |
| existing = { |
| getattr(r, "path", None) for r in getattr(_app, "routes", []) |
| } |
|
|
| if "/health" not in existing: |
| @_app.get("/health") |
| def _health() -> dict: |
| return {"status": "ok", "env": "forgeenv"} |
|
|
| if "/" not in existing: |
| @_app.get("/", response_class=HTMLResponse, include_in_schema=False) |
| def _root() -> str: |
| return _LANDING_HTML |
|
|
|
|
| _attach_supplementary_routes(app) |
|
|
|
|
| if __name__ == "__main__": |
| import uvicorn |
|
|
| port = int(os.environ.get("PORT", "7860")) |
| uvicorn.run(app, host="0.0.0.0", port=port) |
|
|
