bug: only show tool tokens if > 0 (#985)

1 files changed, 11 insertions, 0 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 />', () => {