From 32da693b911d0022d2b0f0d6a918e5830f425d54 Mon Sep 17 00:00:00 2001
From: Abhi <43648792+abhipatel12@users.noreply.github.com>
Date: Thu, 12 Jun 2025 15:31:17 -0400
Subject: [PATCH] bug: only show tool tokens if > 0 (#985)

---
 packages/cli/src/ui/components/Stats.test.tsx         | 11 +++++++++++
 packages/cli/src/ui/components/Stats.tsx              | 10 ++++++----
 .../__snapshots__/SessionSummaryDisplay.test.tsx.snap |  1 -
 .../__snapshots__/StatsDisplay.test.tsx.snap          |  1 -
 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  │
 │  ─────────────────────────────────────────────    ─────────────────────────────────────────────  │