File size: 1,905 Bytes
cc6e5ac | 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 | """
FastAPI application for the TICE environment.
This module creates an HTTP server that exposes `TICEEnvironment` over HTTP and WebSocket
endpoints, compatible with OpenEnv `EnvClient`.
Endpoints:
- POST /reset: Reset the environment
- POST /step: Execute an action
- GET /state: Get current environment state
- GET /schema: Get action/observation schemas
- WS /ws: WebSocket endpoint for persistent sessions
Usage:
# Development (with auto-reload):
uvicorn server.app:app --reload --host 0.0.0.0 --port 8000
# Production:
uvicorn server.app:app --host 0.0.0.0 --port 8000 --workers 4
# Or run directly:
python -m server.app
"""
try:
from openenv.core.env_server.http_server import create_app
except Exception as e: # pragma: no cover
raise ImportError(
"openenv is required for the web interface. Install dependencies with '\n uv sync\n'"
) from e
try:
from ..models import TICEAction, TICEObservation
from .tice_environment import TICEEnvironment
except (ImportError, ModuleNotFoundError):
from models import TICEAction, TICEObservation
from server.tice_environment import TICEEnvironment
app = create_app(
TICEEnvironment,
TICEAction,
TICEObservation,
env_name="tice",
max_concurrent_envs=4,
)
def main(host: str = "0.0.0.0", port: int = 8000) -> None:
"""
Entry point for direct execution via `python -m server.app`.
For production deployments, prefer calling uvicorn directly with multiple workers:
uvicorn server.app:app --workers 4
"""
import argparse
import uvicorn
parser = argparse.ArgumentParser()
parser.add_argument("--host", type=str, default=host)
parser.add_argument("--port", type=int, default=port)
args = parser.parse_args()
uvicorn.run(app, host=args.host, port=args.port)
if __name__ == "__main__":
main()
|