1 parent a845d3f commit cf567dfCopy full SHA for cf567df
1 file changed
tests/test_orchestrator.py
@@ -658,11 +658,11 @@ def test_completed_codex_input_overrun_is_recorded_and_kept(self) -> None:
658
)
659
self.assertTrue(agent_step.data["usage_budget_exceeded"])
660
self.assertEqual(agent_step.data["role_usage"]["input_tokens"], 120)
661
- execution = json.loads(
662
- next((directory / "agent-executions").glob("*.json")).read_text(
663
- encoding="utf-8"
664
- )
665
+ executions = [
+ json.loads(path.read_text(encoding="utf-8"))
+ for path in (directory / "agent-executions").glob("*.json")
+ ]
+ execution = next(row for row in executions if row["role"] == "primary")
666
self.assertEqual(execution["usage"]["cached_input_tokens"], 90)
667
668
self.assertEqual(len(primary.calls), 1)
0 commit comments