diff options
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 │ │ ───────────────────────────────────────────── ───────────────────────────────────────────── │ |