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('', () => { ); 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( + , + ); + expect(lastFrame()).not.toContain('Tool Use Tokens'); + }); }); describe('', () => { 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()} /> - + {stats.toolUseTokens > 0 && ( + + )} > 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[` > 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 │ │ ───────────────────────────────────────────── ───────────────────────────────────────────── │