Add on_round_update to communicate game state to agent

Author: john-b-yang

codeclash/agents/abstract.py

@@ -21,7 +21,6 @@ def __init__(
         self.name = f"{template_vars['game_id']}_{config['name']}"
         self.environment = environment
         self.template_vars = template_vars
-        self.round = 1  # TODO: This is disconnected from game.round right now
 
     def commit(self):
         """Commit changes to the agent's codebase."""
@@ -32,7 +31,10 @@ def commit(self):
         ]:
             assert_zero_exit_code(self.environment.execute(cmd))
         print(f"Committed changes for {self.name} for round {self.round}/{rounds}")
-        self.round += 1  # TODO: This is disconnected from game.round right now
+
+    def on_round_update(self, new_round: int):
+        """Update the agent's round to match the game round."""
+        self.round = new_round
 
     def push(self):
         """Push codebase to a branch on the game's remote repository."""

codeclash/agents/minisweagent.py

@@ -93,7 +93,7 @@ def run(self):
             save_traj(
                 self.agent,  # type: ignore
                 DIR_LOGS
-                / f"{self.template_vars['game_id']}/{player_id}_r{self.template_vars['round']}.traj.json",
+                / f"{self.template_vars['game_id']}/{player_id}_r{self.round}.traj.json",
                 exit_status=exit_status,
                 result=result,
             )

codeclash/games/abstract.py

@@ -93,6 +93,10 @@ def get_environment(self, branch_name: str | None = None) -> DockerEnvironment:
     def _pre_round_setup(self, agents: list[Player]):
         """Copy agent codebases into game's container and make round log file"""
         self.round += 1
+        # Notify agents of round update
+        for agent in agents:
+            if hasattr(agent, "on_round_update"):
+                agent.on_round_update(self.round)
         print(f"▶️ Running {self.name} round {self.round}...")
 
         # Copy agent codebases into game's container