diff options
| author | Abhi <[email protected]> | 2025-06-12 15:31:17 -0400 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-06-12 19:31:17 +0000 |
| commit | 32da693b911d0022d2b0f0d6a918e5830f425d54 (patch) | |
| tree | 4341c1e3fa712903a6924735e468a7cc92ed5f4c /packages/cli/src | |
| parent | 6bb3f27f6cddfeeb4295d12830aa3c03214e6b13 (diff) | |
bug: only show tool tokens if > 0 (#985)
Diffstat (limited to 'packages/cli/src')
4 files changed, 17 insertions, 6 deletions
diff --git a/packages/cli/src/ui/components/Stats.test.tsx b/packages/cli/src/ui/components/Stats.test.tsx index 1436d485..27c7d64e 100644 --- a/packages/cli/src/ui/components/Stats.test.tsx +++ b/packages/cli/src/ui/components/Stats.test.tsx @@ -66,6 +66,17 @@ describe('<StatsColumn />', () => { ); expect(lastFrame()).toMatchSnapshot(); }); + + it('hides the tool use row when there are no tool use tokens', () => { + const statsWithNoToolUse: FormattedStats = { + ...mockStats, + toolUseTokens: 0, + }; + const { lastFrame } = render( + <StatsColumn title="Test Stats" stats={statsWithNoToolUse} />, + ); + expect(lastFrame()).not.toContain('Tool Use Tokens'); + }); }); describe('<DurationColumn />', () => { diff --git a/packages/cli/src/ui/components/Stats.tsx b/packages/cli/src/ui/components/Stats.tsx index 92fadd11..64e55d8b 100644 --- a/packages/cli/src/ui/components/Stats.tsx +++ b/packages/cli/src/ui/components/Stats.tsx @@ -66,10 +66,12 @@ export const StatsColumn: React.FC<{ label="Output Tokens" value={stats.outputTokens.toLocaleString()} /> - <StatRow - label="Tool Use Tokens" - value={stats.toolUseTokens.toLocaleString()} - /> + {stats.toolUseTokens > 0 && ( + <StatRow + label="Tool Use Tokens" + value={stats.toolUseTokens.toLocaleString()} + /> + )} <StatRow label="Thoughts Tokens" value={stats.thoughtsTokens.toLocaleString()} diff --git a/packages/cli/src/ui/components/__snapshots__/SessionSummaryDisplay.test.tsx.snap b/packages/cli/src/ui/components/__snapshots__/SessionSummaryDisplay.test.tsx.snap index 74b067b7..e8871d88 100644 --- a/packages/cli/src/ui/components/__snapshots__/SessionSummaryDisplay.test.tsx.snap +++ b/packages/cli/src/ui/components/__snapshots__/SessionSummaryDisplay.test.tsx.snap @@ -32,7 +32,6 @@ exports[`<SessionSummaryDisplay /> > renders zero state correctly 1`] = ` │ │ │ Input Tokens 0 │ │ Output Tokens 0 │ -│ Tool Use Tokens 0 │ │ Thoughts Tokens 0 │ │ Cached Tokens 0 │ │ ────────────────────────── │ diff --git a/packages/cli/src/ui/components/__snapshots__/StatsDisplay.test.tsx.snap b/packages/cli/src/ui/components/__snapshots__/StatsDisplay.test.tsx.snap index f8fa3d4f..8dc73d93 100644 --- a/packages/cli/src/ui/components/__snapshots__/StatsDisplay.test.tsx.snap +++ b/packages/cli/src/ui/components/__snapshots__/StatsDisplay.test.tsx.snap @@ -30,7 +30,6 @@ exports[`<StatsDisplay /> > renders zero state correctly 1`] = ` │ │ │ Input Tokens 0 Input Tokens 0 │ │ Output Tokens 0 Output Tokens 0 │ -│ Tool Use Tokens 0 Tool Use Tokens 0 │ │ Thoughts Tokens 0 Thoughts Tokens 0 │ │ Cached Tokens 0 Cached Tokens 0 │ │ ───────────────────────────────────────────── ───────────────────────────────────────────── │ |