| """FastAPI app for the SUPERBEAM environment. | |
| uvicorn server.app:app --host 0.0.0.0 --port 8000 (from the env directory) | |
| """ | |
| import os | |
| from openenv.core.env_server import create_app | |
| try: | |
| from ..models import SuperBeamAction, SuperBeamObservation, SuperBeamState | |
| from .superbeam_environment import SuperBeamEnvironment | |
| except ImportError as e: | |
| if "relative import" not in str(e) and "no known parent package" not in str(e): | |
| raise | |
| from models import SuperBeamAction, SuperBeamObservation, SuperBeamState | |
| from server.superbeam_environment import SuperBeamEnvironment | |
| app = create_app( | |
| SuperBeamEnvironment, | |
| SuperBeamAction, | |
| SuperBeamObservation, | |
| env_name="superbeam_env", | |
| state_cls=SuperBeamState, | |
| max_concurrent_envs=int(os.getenv("MAX_CONCURRENT_ENVS", "16")), | |
| ) | |
| def main(): | |
| import uvicorn | |
| uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", "8000"))) | |
| if __name__ == "__main__": | |
| main() | |