From 00805cb2cdd9786451b9a3a6746f11f47535288d Mon Sep 17 00:00:00 2001
From: Jacob Richman <jacob314@gmail.com>
Date: Wed, 28 May 2025 19:46:08 +0000
Subject: Cleanup: Remove low value StreamingContextType interface. (#585)

---
 packages/cli/src/ui/components/messages/ToolMessage.test.tsx | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

(limited to 'packages/cli/src/ui/components/messages/ToolMessage.test.tsx')

diff --git a/packages/cli/src/ui/components/messages/ToolMessage.test.tsx b/packages/cli/src/ui/components/messages/ToolMessage.test.tsx
index a40ca31b..2b96f18a 100644
--- a/packages/cli/src/ui/components/messages/ToolMessage.test.tsx
+++ b/packages/cli/src/ui/components/messages/ToolMessage.test.tsx
@@ -9,10 +9,7 @@ import { render } from 'ink-testing-library';
 import { ToolMessage, ToolMessageProps } from './ToolMessage.js';
 import { StreamingState, ToolCallStatus } from '../../types.js';
 import { Text } from 'ink';
-import {
-  StreamingContext,
-  StreamingContextType,
-} from '../../contexts/StreamingContext.js';
+import { StreamingContext } from '../../contexts/StreamingContext.js';
 
 // Mock child components or utilities if they are complex or have side effects
 vi.mock('../GeminiRespondingSpinner.js', () => ({
@@ -21,7 +18,7 @@ vi.mock('../GeminiRespondingSpinner.js', () => ({
   }: {
     nonRespondingDisplay?: string;
   }) => {
-    const { streamingState } = React.useContext(StreamingContext)!;
+    const streamingState = React.useContext(StreamingContext)!;
     if (streamingState === StreamingState.Responding) {
       return <Text>MockRespondingSpinner</Text>;
     }
@@ -48,7 +45,7 @@ const renderWithContext = (
   ui: React.ReactElement,
   streamingState: StreamingState,
 ) => {
-  const contextValue: StreamingContextType = { streamingState };
+  const contextValue: StreamingState = streamingState;
   return render(
     <StreamingContext.Provider value={contextValue}>
       {ui}
-- 
cgit v1.2.3

