diff --git a/overcooked_simulator/game_server.py b/overcooked_simulator/game_server.py index c5bb5d2577dd0dde98a76f54474893fd0a838b56..d54341a53d2a8100a0a2b9aa12c4bf3879e27f8b 100644 --- a/overcooked_simulator/game_server.py +++ b/overcooked_simulator/game_server.py @@ -675,6 +675,12 @@ class CreateEnvironmentConfig(BaseModel): seed: int +class ManageEnv(BaseModel): + manager_id: str + env_id: str + reason: str + + class AdditionalPlayer(BaseModel): manager_id: str env_id: str @@ -697,8 +703,10 @@ async def additional_player(creation: AdditionalPlayer) -> dict[str, PlayerInfo] @app.post("/manage/stop_env/") -async def stop_env(manager_id: str, env_id: str, reason: str) -> str: - accept = environment_handler.stop_env(manager_id, env_id, reason) +async def stop_env(manage_env: ManageEnv) -> str: + accept = environment_handler.stop_env( + manage_env.manager_id, manage_env.env_id, manage_env.reason + ) if accept: raise HTTPException( status_code=403 if accept == 1 else 409, diff --git a/overcooked_simulator/gui_2d_vis/overcooked_gui.py b/overcooked_simulator/gui_2d_vis/overcooked_gui.py index a4d28c5cb9fe468683db502766b2f2f0c37a7371..8808d1dbb0be8194625b0f1872298fc0b3a5a8de 100644 --- a/overcooked_simulator/gui_2d_vis/overcooked_gui.py +++ b/overcooked_simulator/gui_2d_vis/overcooked_gui.py @@ -860,7 +860,7 @@ class PyGameGUI: def finished_button_press(self): requests.post( - f"{self.request_url}/manage/stop_env", + f"{self.request_url}/manage/stop_env/", json={ "manager_id": self.manager_id, "env_id": self.current_env_id,