diff --git a/src/copilot_usage/report.py b/src/copilot_usage/report.py index 29c2eef..a38fbe0 100644 --- a/src/copilot_usage/report.py +++ b/src/copilot_usage/report.py @@ -156,7 +156,7 @@ def render_live_sessions(sessions: list[SessionSummary]) -> None: model = s.model or "—" running = _format_session_running_time(s) messages = str(s.user_messages) - tokens = f"{_estimated_output_tokens(s):,}" + tokens = format_tokens(_estimated_output_tokens(s)) cwd = s.cwd or "—" table.add_row( diff --git a/tests/copilot_usage/test_report.py b/tests/copilot_usage/test_report.py index 956a9c8..6615e18 100644 --- a/tests/copilot_usage/test_report.py +++ b/tests/copilot_usage/test_report.py @@ -190,7 +190,15 @@ def test_active_session_shows_output_tokens(self) -> None: output_tokens=15000, start_time=now - timedelta(minutes=5) ) output = _capture_output([session]) - assert "15,000" in output + assert "15.0K" in output + + def test_active_session_shows_output_tokens_m_suffix(self) -> None: + now = datetime.now(tz=UTC) + session = _make_session( + output_tokens=1_500_000, start_time=now - timedelta(minutes=5) + ) + output = _capture_output([session]) + assert "1.5M" in output def test_active_session_shows_cwd(self) -> None: now = datetime.now(tz=UTC)