Enh: Save exit status/cost/api calls in metadata

klieret · klieret · commit 261483de3cbf · 2025-10-10T14:19:02.000-04:00
diff --git a/codeclash/agents/minisweagent.py b/codeclash/agents/minisweagent.py
@@ -83,3 +83,8 @@ def run(self):
                 traj_path,
                 self.game_context.log_env / "edits" / traj_path.name,
             )
+            self._metadata["agent_stats"][self.game_context.round] = {
+                "exit_status": exit_status,
+                "cost": self.agent.model.cost,
+                "api_calls": self.agent.model.n_calls,
+            }
diff --git a/codeclash/agents/player.py b/codeclash/agents/player.py
@@ -44,6 +44,7 @@ def __init__(
             "initial_commit_hash": self._get_commit_hash(),
             "branch_name": self._branch_name,
             "round_tags": {},  # mapping round -> tag
+            "agent_stats": {},  # mapping round -> agent stats
         }
 
         if self.push:

Original file line number	Diff line number	Diff line change
`@@ -83,3 +83,8 @@ def run(self):`
`83`	`83`	`traj_path,`
`84`	`84`	`self.game_context.log_env / "edits" / traj_path.name,`
`85`	`85`	`)`
	`86`	`+ self._metadata["agent_stats"][self.game_context.round] = {`
	`87`	`+ "exit_status": exit_status,`
	`88`	`+ "cost": self.agent.model.cost,`
	`89`	`+ "api_calls": self.agent.model.n_calls,`
	`90`	`+ }`
Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,7 @@ def __init__(`
`44`	`44`	`"initial_commit_hash": self._get_commit_hash(),`
`45`	`45`	`"branch_name": self._branch_name,`
`46`	`46`	`"round_tags": {}, # mapping round -> tag`
	`47`	`+ "agent_stats": {}, # mapping round -> agent stats`
`47`	`48`	`}`
`48`	`49`
`49`	`50`	`if self.push:`