refactor: Centralize tool scheduling logic and simplify React hook (#670)

1 files changed, 3 insertions, 3 deletions

diff --git a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
index 6959d9a7..44013059 100644
--- a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
+++ b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
@@ -9,11 +9,11 @@ import { mergePartListUnions } from './useGeminiStream.js';
 import { Part, PartListUnion } from '@google/genai';
 
 // Mock useToolScheduler
-vi.mock('./useToolScheduler', async () => {
-  const actual = await vi.importActual('./useToolScheduler');
+vi.mock('./useReactToolScheduler', async () => {
+  const actual = await vi.importActual('./useReactToolScheduler');
   return {
     ...actual, // We need mapToDisplay from actual
-    useToolScheduler: vi.fn(),
+    useReactToolScheduler: vi.fn(),
   };
 });