server/app.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

"""
FastAPI application for the Coding Environment.

This module creates an HTTP server that exposes the PythonCodeActEnv
over HTTP and WebSocket endpoints, compatible with EnvClient.

Usage:
    # Development (with auto-reload):
    uvicorn envs.coding_env.server.app:app --reload --host 0.0.0.0 --port 8000

    # Production:
    uvicorn envs.coding_env.server.app:app --host 0.0.0.0 --port 8000 --workers 4

    # Or run directly:
    python -m envs.coding_env.server.app
"""

from fastapi import Query

try:
    from coding_env.models import CodeAction, CodeObservation
    from coding_env.server.python_codeact_env import PythonCodeActEnv
    from coding_env.server.task_bank import get_episode_score, list_tasks
except ImportError:
    from ..models import CodeAction, CodeObservation
    from .python_codeact_env import PythonCodeActEnv
    from .task_bank import get_episode_score, list_tasks
from openenv.core.env_server import create_app

# Create the app with web interface and README integration
# Pass the class (factory) instead of an instance for WebSocket session support
app = create_app(PythonCodeActEnv, CodeAction, CodeObservation, env_name="coding_env")


@app.get("/tasks", tags=["Environment Info"])
def tasks():
    """Return available benchmark tasks and their difficulty."""
    return {"tasks": list_tasks()}


@app.get("/grader", tags=["Environment Info"])
def grader(
    task_id: str = Query(..., description="Task identifier"),
    episode_id: str = Query(..., description="Episode identifier"),
):
    """Return normalized score in [0.0, 1.0] for task/episode."""
    score = get_episode_score(task_id, episode_id)
    return {"task_id": task_id, "episode_id": episode_id, "score": float(score)}


if __name__ == "__main__":
    import uvicorn

    uvicorn.run(app, host="0.0.0.0", port=8000)


def main():
    """Main entry point for running the server."""
    import uvicorn

    uvicorn.run(app, host="0.0.0.0", port=8000)


if __name__ == "__main__":
    main()