|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
""" |
|
|
FastAPI application for the OpenSpiel Environment. |
|
|
|
|
|
This module creates an HTTP server that exposes OpenSpiel games |
|
|
over HTTP endpoints, making them compatible with HTTPEnvClient. |
|
|
|
|
|
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 |
|
|
|
|
|
Environment variables: |
|
|
OPENSPIEL_GAME: Game name to serve (default: "catch") |
|
|
OPENSPIEL_AGENT_PLAYER: Agent player ID (default: 0) |
|
|
OPENSPIEL_OPPONENT_POLICY: Opponent policy (default: "random") |
|
|
""" |
|
|
|
|
|
import os |
|
|
|
|
|
try: |
|
|
from openenv_core.env_server.http_server import create_app |
|
|
except Exception as e: |
|
|
raise ImportError("openenv_core is required for the web interface. Install dependencies with '\n uv sync\n'") from e |
|
|
|
|
|
from .openspiel_environment import OpenSpielEnvironment |
|
|
from models import OpenSpielAction, OpenSpielObservation |
|
|
|
|
|
|
|
|
game_name = os.getenv("OPENSPIEL_GAME", "catch") |
|
|
agent_player = int(os.getenv("OPENSPIEL_AGENT_PLAYER", "0")) |
|
|
opponent_policy = os.getenv("OPENSPIEL_OPPONENT_POLICY", "random") |
|
|
|
|
|
|
|
|
env = OpenSpielEnvironment( |
|
|
game_name=game_name, |
|
|
agent_player=agent_player, |
|
|
opponent_policy=opponent_policy, |
|
|
) |
|
|
|
|
|
|
|
|
app = create_app(env, OpenSpielAction, OpenSpielObservation, env_name="openspiel") |
|
|
|
|
|
def main(host: str = "0.0.0.0", port: int = 8000): |
|
|
""" |
|
|
Entry point for direct execution via uv run or python -m. |
|
|
|
|
|
This function enables running the server without Docker: |
|
|
uv run --project . server |
|
|
uv run --project . server --port 8001 |
|
|
python -m .server.app |
|
|
|
|
|
Args: |
|
|
host: Host address to bind to (default: "0.0.0.0") |
|
|
port: Port number to listen on (default: 8000) |
|
|
|
|
|
For production deployments, consider using uvicorn directly with |
|
|
multiple workers: |
|
|
uvicorn openspiel.server.app:app --workers 4 |
|
|
""" |
|
|
import uvicorn |
|
|
|
|
|
uvicorn.run(app, host=host, port=port) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
import argparse |
|
|
|
|
|
parser = argparse.ArgumentParser() |
|
|
parser.add_argument("--port", type=int, default=8000) |
|
|
args = parser.parse_args() |
|
|
main(port=args.port) |
|
|
|