From 063481faa4b1c86868689580ff0fbd8cb04141e3 Mon Sep 17 00:00:00 2001
From: uttamkanodia14 <codenote@google.com>
Date: Thu, 10 Jul 2025 00:19:30 +0530
Subject: Adding TurnId to Tool call and API responses and error logs. (#3039)

Co-authored-by: Scott Densmore <scottdensmore@mac.com>
---
 packages/cli/src/ui/components/ModelStatsDisplay.test.tsx     | 4 ++++
 packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx | 4 ++++
 packages/cli/src/ui/components/StatsDisplay.test.tsx          | 8 ++++++++
 packages/cli/src/ui/components/ToolStatsDisplay.test.tsx      | 4 ++++
 4 files changed, 20 insertions(+)

(limited to 'packages/cli/src/ui/components')

diff --git a/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx b/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx
index 6c41b775..57382d91 100644
--- a/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx
+++ b/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx
@@ -27,7 +27,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => {
       sessionStartTime: new Date(),
       metrics,
       lastPromptTokenCount: 0,
+      promptCount: 5,
     },
+
+    getPromptCount: () => 5,
+    startNewPrompt: vi.fn(),
   });
 
   return render(<ModelStatsDisplay />);
diff --git a/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx b/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx
index f3c0764e..38400caf 100644
--- a/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx
+++ b/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx
@@ -26,7 +26,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => {
       sessionStartTime: new Date(),
       metrics,
       lastPromptTokenCount: 0,
+      promptCount: 5,
     },
+
+    getPromptCount: () => 5,
+    startNewPrompt: vi.fn(),
   });
 
   return render(<SessionSummaryDisplay duration="1h 23m 45s" />);
diff --git a/packages/cli/src/ui/components/StatsDisplay.test.tsx b/packages/cli/src/ui/components/StatsDisplay.test.tsx
index a62815d9..a0ed3858 100644
--- a/packages/cli/src/ui/components/StatsDisplay.test.tsx
+++ b/packages/cli/src/ui/components/StatsDisplay.test.tsx
@@ -27,7 +27,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => {
       sessionStartTime: new Date(),
       metrics,
       lastPromptTokenCount: 0,
+      promptCount: 5,
     },
+
+    getPromptCount: () => 5,
+    startNewPrompt: vi.fn(),
   });
 
   return render(<StatsDisplay duration="1s" />);
@@ -288,7 +292,11 @@ describe('<StatsDisplay />', () => {
           sessionStartTime: new Date(),
           metrics: zeroMetrics,
           lastPromptTokenCount: 0,
+          promptCount: 5,
         },
+
+        getPromptCount: () => 5,
+        startNewPrompt: vi.fn(),
       });
 
       const { lastFrame } = render(
diff --git a/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx b/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx
index 54902788..e48fcc83 100644
--- a/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx
+++ b/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx
@@ -27,7 +27,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => {
       sessionStartTime: new Date(),
       metrics,
       lastPromptTokenCount: 0,
+      promptCount: 5,
     },
+
+    getPromptCount: () => 5,
+    startNewPrompt: vi.fn(),
   });
 
   return render(<ToolStatsDisplay />);
-- 
cgit v1.2.3