@zgsm/costrict-cli 0.0.1

package/dist/src/ui/hooks/useGeminiStream.test.js

@@ -0,0 +1,1827 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { act } from 'react';
+import { renderHook } from '../../test-utils/render.js';
+import { waitFor } from '../../test-utils/async.js';
+import { useGeminiStream } from './useGeminiStream.js';
+import { useKeypress } from './useKeypress.js';
+import * as atCommandProcessor from './atCommandProcessor.js';
+import { useReactToolScheduler } from './useReactToolScheduler.js';
+import { ApprovalMode, AuthType, GeminiEventType as ServerGeminiEventType, ToolErrorType, ToolConfirmationOutcome, tokenLimit, debugLogger, } from '@google/gemini-cli-core';
+import { MessageType, StreamingState } from '../types.js';
+// --- MOCKS ---
+const mockSendMessageStream = vi
+    .fn()
+    .mockReturnValue((async function* () { })());
+const mockStartChat = vi.fn();
+const MockedGeminiClientClass = vi.hoisted(() => vi.fn().mockImplementation(function (_config) {
+    // _config
+    this.startChat = mockStartChat;
+    this.sendMessageStream = mockSendMessageStream;
+    this.addHistory = vi.fn();
+    this.getChat = vi.fn().mockReturnValue({
+        recordCompletedToolCalls: vi.fn(),
+    });
+    this.getChatRecordingService = vi.fn().mockReturnValue({
+        recordThought: vi.fn(),
+        initialize: vi.fn(),
+        recordMessage: vi.fn(),
+        recordMessageTokens: vi.fn(),
+        recordToolCalls: vi.fn(),
+        getConversationFile: vi.fn(),
+    });
+}));
+const MockedUserPromptEvent = vi.hoisted(() => vi.fn().mockImplementation(() => { }));
+const mockParseAndFormatApiError = vi.hoisted(() => vi.fn());
+vi.mock('@google/gemini-cli-core', async (importOriginal) => {
+    const actualCoreModule = (await importOriginal());
+    return {
+        ...actualCoreModule,
+        GitService: vi.fn(),
+        GeminiClient: MockedGeminiClientClass,
+        UserPromptEvent: MockedUserPromptEvent,
+        parseAndFormatApiError: mockParseAndFormatApiError,
+        tokenLimit: vi.fn().mockReturnValue(100), // Mock tokenLimit
+    };
+});
+const mockUseReactToolScheduler = useReactToolScheduler;
+vi.mock('./useReactToolScheduler.js', async (importOriginal) => {
+    const actualSchedulerModule = (await importOriginal());
+    return {
+        ...(actualSchedulerModule || {}),
+        useReactToolScheduler: vi.fn(),
+    };
+});
+vi.mock('./useKeypress.js', () => ({
+    useKeypress: vi.fn(),
+}));
+vi.mock('./shellCommandProcessor.js', () => ({
+    useShellCommandProcessor: vi.fn().mockReturnValue({
+        handleShellCommand: vi.fn(),
+    }),
+}));
+vi.mock('./atCommandProcessor.js');
+vi.mock('../utils/markdownUtilities.js', () => ({
+    findLastSafeSplitPoint: vi.fn((s) => s.length),
+}));
+vi.mock('./useStateAndRef.js', () => ({
+    useStateAndRef: vi.fn((initial) => {
+        let val = initial;
+        const ref = { current: val };
+        const setVal = vi.fn((updater) => {
+            if (typeof updater === 'function') {
+                val = updater(val);
+            }
+            else {
+                val = updater;
+            }
+            ref.current = val;
+        });
+        return [val, ref, setVal];
+    }),
+}));
+vi.mock('./useLogger.js', () => ({
+    useLogger: vi.fn().mockReturnValue({
+        logMessage: vi.fn().mockResolvedValue(undefined),
+    }),
+}));
+const mockStartNewPrompt = vi.fn();
+const mockAddUsage = vi.fn();
+vi.mock('../contexts/SessionContext.js', () => ({
+    useSessionStats: vi.fn(() => ({
+        startNewPrompt: mockStartNewPrompt,
+        addUsage: mockAddUsage,
+        getPromptCount: vi.fn(() => 5),
+    })),
+}));
+vi.mock('./slashCommandProcessor.js', () => ({
+    handleSlashCommand: vi.fn().mockReturnValue(false),
+}));
+vi.mock('./useAlternateBuffer.js', () => ({
+    useAlternateBuffer: vi.fn(() => false),
+}));
+// --- END MOCKS ---
+// --- Tests for useGeminiStream Hook ---
+describe('useGeminiStream', () => {
+    let mockAddItem;
+    let mockConfig;
+    let mockOnDebugMessage;
+    let mockHandleSlashCommand;
+    let mockScheduleToolCalls;
+    let mockCancelAllToolCalls;
+    let mockMarkToolsAsSubmitted;
+    let handleAtCommandSpy;
+    beforeEach(() => {
+        vi.clearAllMocks(); // Clear mocks before each test
+        mockAddItem = vi.fn();
+        // Define the mock for getGeminiClient
+        const mockGetGeminiClient = vi.fn().mockImplementation(() => {
+            // MockedGeminiClientClass is defined in the module scope by the previous change.
+            // It will use the mockStartChat and mockSendMessageStream that are managed within beforeEach.
+            const clientInstance = new MockedGeminiClientClass(mockConfig);
+            return clientInstance;
+        });
+        const contentGeneratorConfig = {
+            model: 'test-model',
+            apiKey: 'test-key',
+            vertexai: false,
+            authType: AuthType.USE_GEMINI,
+        };
+        mockConfig = {
+            apiKey: 'test-api-key',
+            model: 'gemini-pro',
+            sandbox: false,
+            targetDir: '/test/dir',
+            debugMode: false,
+            question: undefined,
+            coreTools: [],
+            toolDiscoveryCommand: undefined,
+            toolCallCommand: undefined,
+            mcpServerCommand: undefined,
+            mcpServers: undefined,
+            userAgent: 'test-agent',
+            userMemory: '',
+            geminiMdFileCount: 0,
+            alwaysSkipModificationConfirmation: false,
+            vertexai: false,
+            showMemoryUsage: false,
+            contextFileName: undefined,
+            getToolRegistry: vi.fn(() => ({ getToolSchemaList: vi.fn(() => []) })),
+            getProjectRoot: vi.fn(() => '/test/dir'),
+            getCheckpointingEnabled: vi.fn(() => false),
+            getGeminiClient: mockGetGeminiClient,
+            getApprovalMode: () => ApprovalMode.DEFAULT,
+            getUsageStatisticsEnabled: () => true,
+            getDebugMode: () => false,
+            addHistory: vi.fn(),
+            getSessionId() {
+                return 'test-session-id';
+            },
+            setQuotaErrorOccurred: vi.fn(),
+            getQuotaErrorOccurred: vi.fn(() => false),
+            getModel: vi.fn(() => 'gemini-2.5-pro'),
+            getContentGeneratorConfig: vi
+                .fn()
+                .mockReturnValue(contentGeneratorConfig),
+            getUseSmartEdit: () => false,
+            isInteractive: () => false,
+            getExperiments: () => { },
+        };
+        mockOnDebugMessage = vi.fn();
+        mockHandleSlashCommand = vi.fn().mockResolvedValue(false);
+        // Mock return value for useReactToolScheduler
+        mockScheduleToolCalls = vi.fn();
+        mockCancelAllToolCalls = vi.fn();
+        mockMarkToolsAsSubmitted = vi.fn();
+        // Default mock for useReactToolScheduler to prevent toolCalls being undefined initially
+        mockUseReactToolScheduler.mockReturnValue([
+            [], // Default to empty array for toolCalls
+            mockScheduleToolCalls,
+            mockMarkToolsAsSubmitted,
+            vi.fn(), // setToolCallsForDisplay
+            mockCancelAllToolCalls,
+        ]);
+        // Reset mocks for GeminiClient instance methods (startChat and sendMessageStream)
+        // The GeminiClient constructor itself is mocked at the module level.
+        mockStartChat.mockClear().mockResolvedValue({
+            sendMessageStream: mockSendMessageStream,
+        }); // GeminiChat -> any
+        mockSendMessageStream
+            .mockClear()
+            .mockReturnValue((async function* () { })());
+        handleAtCommandSpy = vi.spyOn(atCommandProcessor, 'handleAtCommand');
+    });
+    const mockLoadedSettings = {
+        merged: { preferredEditor: 'vscode' },
+        user: { path: '/user/settings.json', settings: {} },
+        workspace: { path: '/workspace/.gemini/settings.json', settings: {} },
+        errors: [],
+        forScope: vi.fn(),
+        setValue: vi.fn(),
+    };
+    const renderTestHook = (initialToolCalls = [], geminiClient) => {
+        const client = geminiClient || mockConfig.getGeminiClient();
+        const initialProps = {
+            client,
+            history: [],
+            addItem: mockAddItem,
+            config: mockConfig,
+            onDebugMessage: mockOnDebugMessage,
+            handleSlashCommand: mockHandleSlashCommand,
+            shellModeActive: false,
+            loadedSettings: mockLoadedSettings,
+            toolCalls: initialToolCalls,
+        };
+        const { result, rerender } = renderHook((props) => {
+            // This mock needs to be stateful. When setToolCallsForDisplay is called,
+            // it should trigger a rerender with the new state.
+            const mockSetToolCallsForDisplay = vi.fn((updater) => {
+                const newToolCalls = typeof updater === 'function' ? updater(props.toolCalls) : updater;
+                rerender({ ...props, toolCalls: newToolCalls });
+            });
+            // Create a stateful mock for cancellation that updates the toolCalls state.
+            const statefulCancelAllToolCalls = vi.fn((...args) => {
+                // Call the original spy so `toHaveBeenCalled` checks still work.
+                mockCancelAllToolCalls(...args);
+                const newToolCalls = props.toolCalls.map((tc) => {
+                    // Only cancel tools that are in a cancellable state.
+                    if (tc.status === 'awaiting_approval' ||
+                        tc.status === 'executing' ||
+                        tc.status === 'scheduled' ||
+                        tc.status === 'validating') {
+                        // A real cancelled tool call has a response object.
+                        // We need to simulate this to avoid type errors downstream.
+                        return {
+                            ...tc,
+                            status: 'cancelled',
+                            response: {
+                                callId: tc.request.callId,
+                                responseParts: [],
+                                resultDisplay: 'Request cancelled.',
+                            },
+                            responseSubmittedToGemini: true, // Mark as "processed"
+                        };
+                    }
+                    return tc;
+                });
+                rerender({ ...props, toolCalls: newToolCalls });
+            });
+            mockUseReactToolScheduler.mockImplementation(() => [
+                props.toolCalls,
+                mockScheduleToolCalls,
+                mockMarkToolsAsSubmitted,
+                mockSetToolCallsForDisplay,
+                statefulCancelAllToolCalls, // Use the stateful mock
+            ]);
+            return useGeminiStream(props.client, props.history, props.addItem, props.config, props.loadedSettings, props.onDebugMessage, props.handleSlashCommand, props.shellModeActive, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24);
+        }, {
+            initialProps,
+        });
+        return {
+            result,
+            rerender,
+            mockMarkToolsAsSubmitted,
+            mockSendMessageStream,
+            client,
+        };
+    };
+    // Helper to create mock tool calls - reduces boilerplate
+    const createMockToolCall = (toolName, callId, confirmationType, mockOnConfirm, status = 'awaiting_approval') => ({
+        request: {
+            callId,
+            name: toolName,
+            args: {},
+            isClientInitiated: false,
+            prompt_id: 'prompt-id-1',
+        },
+        status: status,
+        responseSubmittedToGemini: false,
+        confirmationDetails: confirmationType === 'edit'
+            ? {
+                type: 'edit',
+                title: 'Confirm Edit',
+                onConfirm: mockOnConfirm,
+                fileName: 'file.txt',
+                filePath: '/test/file.txt',
+                fileDiff: 'fake diff',
+                originalContent: 'old',
+                newContent: 'new',
+            }
+            : {
+                type: 'info',
+                title: `${toolName} confirmation`,
+                onConfirm: mockOnConfirm,
+                prompt: `Execute ${toolName}?`,
+            },
+        tool: {
+            name: toolName,
+            displayName: toolName,
+            description: `${toolName} description`,
+            build: vi.fn(),
+        },
+        invocation: {
+            getDescription: () => 'Mock description',
+        },
+    });
+    // Helper to render hook with default parameters - reduces boilerplate
+    const renderHookWithDefaults = (options = {}) => {
+        const { shellModeActive = false, onCancelSubmit = () => { }, setShellInputFocused = () => { }, performMemoryRefresh = () => Promise.resolve(), onAuthError = () => { }, setModelSwitched = vi.fn(), modelSwitched = false, } = options;
+        return renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, shellModeActive, () => 'vscode', onAuthError, performMemoryRefresh, modelSwitched, setModelSwitched, onCancelSubmit, setShellInputFocused, 80, 24));
+    };
+    it('should not submit tool responses if not all tool calls are completed', () => {
+        const toolCalls = [
+            {
+                request: {
+                    callId: 'call1',
+                    name: 'tool1',
+                    args: {},
+                    isClientInitiated: false,
+                    prompt_id: 'prompt-id-1',
+                },
+                status: 'success',
+                responseSubmittedToGemini: false,
+                response: {
+                    callId: 'call1',
+                    responseParts: [{ text: 'tool 1 response' }],
+                    error: undefined,
+                    errorType: undefined, // FIX: Added missing property
+                    resultDisplay: 'Tool 1 success display',
+                },
+                tool: {
+                    name: 'tool1',
+                    displayName: 'tool1',
+                    description: 'desc1',
+                    build: vi.fn(),
+                },
+                invocation: {
+                    getDescription: () => `Mock description`,
+                },
+                startTime: Date.now(),
+                endTime: Date.now(),
+            },
+            {
+                request: {
+                    callId: 'call2',
+                    name: 'tool2',
+                    args: {},
+                    prompt_id: 'prompt-id-1',
+                },
+                status: 'executing',
+                responseSubmittedToGemini: false,
+                tool: {
+                    name: 'tool2',
+                    displayName: 'tool2',
+                    description: 'desc2',
+                    build: vi.fn(),
+                },
+                invocation: {
+                    getDescription: () => `Mock description`,
+                },
+                startTime: Date.now(),
+                liveOutput: '...',
+            },
+        ];
+        const { mockMarkToolsAsSubmitted, mockSendMessageStream } = renderTestHook(toolCalls);
+        // Effect for submitting tool responses depends on toolCalls and isResponding
+        // isResponding is initially false, so the effect should run.
+        expect(mockMarkToolsAsSubmitted).not.toHaveBeenCalled();
+        expect(mockSendMessageStream).not.toHaveBeenCalled(); // submitQuery uses this
+    });
+    it('should submit tool responses when all tool calls are completed and ready', async () => {
+        const toolCall1ResponseParts = [{ text: 'tool 1 final response' }];
+        const toolCall2ResponseParts = [{ text: 'tool 2 final response' }];
+        const completedToolCalls = [
+            {
+                request: {
+                    callId: 'call1',
+                    name: 'tool1',
+                    args: {},
+                    isClientInitiated: false,
+                    prompt_id: 'prompt-id-2',
+                },
+                status: 'success',
+                responseSubmittedToGemini: false,
+                response: {
+                    callId: 'call1',
+                    responseParts: toolCall1ResponseParts,
+                    errorType: undefined, // FIX: Added missing property
+                },
+                tool: {
+                    displayName: 'MockTool',
+                },
+                invocation: {
+                    getDescription: () => `Mock description`,
+                },
+            },
+            {
+                request: {
+                    callId: 'call2',
+                    name: 'tool2',
+                    args: {},
+                    isClientInitiated: false,
+                    prompt_id: 'prompt-id-2',
+                },
+                status: 'error',
+                responseSubmittedToGemini: false,
+                response: {
+                    callId: 'call2',
+                    responseParts: toolCall2ResponseParts,
+                    errorType: ToolErrorType.UNHANDLED_EXCEPTION, // FIX: Added missing property
+                },
+            },
+        ];
+        // Capture the onComplete callback
+        let capturedOnComplete = null;
+        mockUseReactToolScheduler.mockImplementation((onComplete) => {
+            capturedOnComplete = onComplete;
+            return [[], mockScheduleToolCalls, mockMarkToolsAsSubmitted, vi.fn()];
+        });
+        renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+        // Trigger the onComplete callback with completed tools
+        await act(async () => {
+            if (capturedOnComplete) {
+                await capturedOnComplete(completedToolCalls);
+            }
+        });
+        await waitFor(() => {
+            expect(mockMarkToolsAsSubmitted).toHaveBeenCalledTimes(1);
+            expect(mockSendMessageStream).toHaveBeenCalledTimes(1);
+        });
+        const expectedMergedResponse = [
+            ...toolCall1ResponseParts,
+            ...toolCall2ResponseParts,
+        ];
+        expect(mockSendMessageStream).toHaveBeenCalledWith(expectedMergedResponse, expect.any(AbortSignal), 'prompt-id-2');
+    });
+    it('should handle all tool calls being cancelled', async () => {
+        const cancelledToolCalls = [
+            {
+                request: {
+                    callId: '1',
+                    name: 'testTool',
+                    args: {},
+                    isClientInitiated: false,
+                    prompt_id: 'prompt-id-3',
+                },
+                status: 'cancelled',
+                response: {
+                    callId: '1',
+                    responseParts: [{ text: 'cancelled' }],
+                    errorType: undefined, // FIX: Added missing property
+                },
+                responseSubmittedToGemini: false,
+                tool: {
+                    displayName: 'mock tool',
+                },
+                invocation: {
+                    getDescription: () => `Mock description`,
+                },
+            },
+        ];
+        const client = new MockedGeminiClientClass(mockConfig);
+        // Capture the onComplete callback
+        let capturedOnComplete = null;
+        mockUseReactToolScheduler.mockImplementation((onComplete) => {
+            capturedOnComplete = onComplete;
+            return [[], mockScheduleToolCalls, mockMarkToolsAsSubmitted, vi.fn()];
+        });
+        renderHook(() => useGeminiStream(client, [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+        // Trigger the onComplete callback with cancelled tools
+        await act(async () => {
+            if (capturedOnComplete) {
+                await capturedOnComplete(cancelledToolCalls);
+            }
+        });
+        await waitFor(() => {
+            expect(mockMarkToolsAsSubmitted).toHaveBeenCalledWith(['1']);
+            expect(client.addHistory).toHaveBeenCalledWith({
+                role: 'user',
+                parts: [{ text: 'cancelled' }],
+            });
+            // Ensure we do NOT call back to the API
+            expect(mockSendMessageStream).not.toHaveBeenCalled();
+        });
+    });
+    it('should group multiple cancelled tool call responses into a single history entry', async () => {
+        const cancelledToolCall1 = {
+            request: {
+                callId: 'cancel-1',
+                name: 'toolA',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'prompt-id-7',
+            },
+            tool: {
+                name: 'toolA',
+                displayName: 'toolA',
+                description: 'descA',
+                build: vi.fn(),
+            },
+            invocation: {
+                getDescription: () => `Mock description`,
+            },
+            status: 'cancelled',
+            response: {
+                callId: 'cancel-1',
+                responseParts: [
+                    { functionResponse: { name: 'toolA', id: 'cancel-1' } },
+                ],
+                resultDisplay: undefined,
+                error: undefined,
+                errorType: undefined, // FIX: Added missing property
+            },
+            responseSubmittedToGemini: false,
+        };
+        const cancelledToolCall2 = {
+            request: {
+                callId: 'cancel-2',
+                name: 'toolB',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'prompt-id-8',
+            },
+            tool: {
+                name: 'toolB',
+                displayName: 'toolB',
+                description: 'descB',
+                build: vi.fn(),
+            },
+            invocation: {
+                getDescription: () => `Mock description`,
+            },
+            status: 'cancelled',
+            response: {
+                callId: 'cancel-2',
+                responseParts: [
+                    { functionResponse: { name: 'toolB', id: 'cancel-2' } },
+                ],
+                resultDisplay: undefined,
+                error: undefined,
+                errorType: undefined, // FIX: Added missing property
+            },
+            responseSubmittedToGemini: false,
+        };
+        const allCancelledTools = [cancelledToolCall1, cancelledToolCall2];
+        const client = new MockedGeminiClientClass(mockConfig);
+        let capturedOnComplete = null;
+        mockUseReactToolScheduler.mockImplementation((onComplete) => {
+            capturedOnComplete = onComplete;
+            return [[], mockScheduleToolCalls, mockMarkToolsAsSubmitted, vi.fn()];
+        });
+        renderHook(() => useGeminiStream(client, [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+        // Trigger the onComplete callback with multiple cancelled tools
+        await act(async () => {
+            if (capturedOnComplete) {
+                await capturedOnComplete(allCancelledTools);
+            }
+        });
+        await waitFor(() => {
+            // The tools should be marked as submitted locally
+            expect(mockMarkToolsAsSubmitted).toHaveBeenCalledWith([
+                'cancel-1',
+                'cancel-2',
+            ]);
+            // Crucially, addHistory should be called only ONCE
+            expect(client.addHistory).toHaveBeenCalledTimes(1);
+            // And that single call should contain BOTH function responses
+            expect(client.addHistory).toHaveBeenCalledWith({
+                role: 'user',
+                parts: [
+                    ...cancelledToolCall1.response.responseParts,
+                    ...cancelledToolCall2.response.responseParts,
+                ],
+            });
+            // No message should be sent back to the API for a turn with only cancellations
+            expect(mockSendMessageStream).not.toHaveBeenCalled();
+        });
+    });
+    it('should not flicker streaming state to Idle between tool completion and submission', async () => {
+        const toolCallResponseParts = [
+            { text: 'tool 1 final response' },
+        ];
+        const initialToolCalls = [
+            {
+                request: {
+                    callId: 'call1',
+                    name: 'tool1',
+                    args: {},
+                    isClientInitiated: false,
+                    prompt_id: 'prompt-id-4',
+                },
+                status: 'executing',
+                responseSubmittedToGemini: false,
+                tool: {
+                    name: 'tool1',
+                    displayName: 'tool1',
+                    description: 'desc',
+                    build: vi.fn(),
+                },
+                invocation: {
+                    getDescription: () => `Mock description`,
+                },
+                startTime: Date.now(),
+            },
+        ];
+        const completedToolCalls = [
+            {
+                ...initialToolCalls[0],
+                status: 'success',
+                response: {
+                    callId: 'call1',
+                    responseParts: toolCallResponseParts,
+                    error: undefined,
+                    errorType: undefined, // FIX: Added missing property
+                    resultDisplay: 'Tool 1 success display',
+                },
+                endTime: Date.now(),
+            },
+        ];
+        // Capture the onComplete callback
+        let capturedOnComplete = null;
+        let currentToolCalls = initialToolCalls;
+        mockUseReactToolScheduler.mockImplementation((onComplete) => {
+            capturedOnComplete = onComplete;
+            return [
+                currentToolCalls,
+                mockScheduleToolCalls,
+                mockMarkToolsAsSubmitted,
+                vi.fn(), // setToolCallsForDisplay
+            ];
+        });
+        const { result, rerender } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+        // 1. Initial state should be Responding because a tool is executing.
+        expect(result.current.streamingState).toBe(StreamingState.Responding);
+        // 2. Update the tool calls to completed state and rerender
+        currentToolCalls = completedToolCalls;
+        mockUseReactToolScheduler.mockImplementation((onComplete) => {
+            capturedOnComplete = onComplete;
+            return [
+                completedToolCalls,
+                mockScheduleToolCalls,
+                mockMarkToolsAsSubmitted,
+                vi.fn(), // setToolCallsForDisplay
+            ];
+        });
+        act(() => {
+            rerender();
+        });
+        // 3. The state should *still* be Responding, not Idle.
+        // This is because the completed tool's response has not been submitted yet.
+        expect(result.current.streamingState).toBe(StreamingState.Responding);
+        // 4. Trigger the onComplete callback to simulate tool completion
+        await act(async () => {
+            if (capturedOnComplete) {
+                await capturedOnComplete(completedToolCalls);
+            }
+        });
+        // 5. Wait for submitQuery to be called
+        await waitFor(() => {
+            expect(mockSendMessageStream).toHaveBeenCalledWith(toolCallResponseParts, expect.any(AbortSignal), 'prompt-id-4');
+        });
+        // 6. After submission, the state should remain Responding until the stream completes.
+        expect(result.current.streamingState).toBe(StreamingState.Responding);
+    });
+    describe('User Cancellation', () => {
+        let keypressCallback;
+        const mockUseKeypress = useKeypress;
+        beforeEach(() => {
+            // Capture the callback passed to useKeypress
+            mockUseKeypress.mockImplementation((callback, options) => {
+                if (options.isActive) {
+                    keypressCallback = callback;
+                }
+                else {
+                    keypressCallback = () => { };
+                }
+            });
+        });
+        const simulateEscapeKeyPress = () => {
+            act(() => {
+                keypressCallback({ name: 'escape' });
+            });
+        };
+        it('should cancel an in-progress stream when escape is pressed', async () => {
+            const mockStream = (async function* () {
+                yield { type: 'content', value: 'Part 1' };
+                // Keep the stream open
+                await new Promise(() => { });
+            })();
+            mockSendMessageStream.mockReturnValue(mockStream);
+            const { result } = renderTestHook();
+            // Start a query
+            await act(async () => {
+                // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                result.current.submitQuery('test query');
+            });
+            // Wait for the first part of the response
+            await waitFor(() => {
+                expect(result.current.streamingState).toBe(StreamingState.Responding);
+            });
+            // Simulate escape key press
+            simulateEscapeKeyPress();
+            // Verify cancellation message is added
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith({
+                    type: MessageType.INFO,
+                    text: 'Request cancelled.',
+                }, expect.any(Number));
+            });
+            // Verify state is reset
+            expect(result.current.streamingState).toBe(StreamingState.Idle);
+        });
+        it('should call onCancelSubmit handler when escape is pressed', async () => {
+            const cancelSubmitSpy = vi.fn();
+            const mockStream = (async function* () {
+                yield { type: 'content', value: 'Part 1' };
+                // Keep the stream open
+                await new Promise(() => { });
+            })();
+            mockSendMessageStream.mockReturnValue(mockStream);
+            const { result } = renderHook(() => useGeminiStream(mockConfig.getGeminiClient(), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, cancelSubmitSpy, () => { }, 80, 24));
+            // Start a query
+            await act(async () => {
+                // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                result.current.submitQuery('test query');
+            });
+            simulateEscapeKeyPress();
+            expect(cancelSubmitSpy).toHaveBeenCalledWith(false);
+        });
+        it('should call setShellInputFocused(false) when escape is pressed', async () => {
+            const setShellInputFocusedSpy = vi.fn();
+            const mockStream = (async function* () {
+                yield { type: 'content', value: 'Part 1' };
+                await new Promise(() => { }); // Keep stream open
+            })();
+            mockSendMessageStream.mockReturnValue(mockStream);
+            const { result } = renderHook(() => useGeminiStream(mockConfig.getGeminiClient(), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, vi.fn(), setShellInputFocusedSpy, // Pass the spy here
+            80, 24));
+            // Start a query
+            await act(async () => {
+                // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                result.current.submitQuery('test query');
+            });
+            simulateEscapeKeyPress();
+            expect(setShellInputFocusedSpy).toHaveBeenCalledWith(false);
+        });
+        it('should not do anything if escape is pressed when not responding', () => {
+            const { result } = renderTestHook();
+            expect(result.current.streamingState).toBe(StreamingState.Idle);
+            // Simulate escape key press
+            simulateEscapeKeyPress();
+            // No change should happen, no cancellation message
+            expect(mockAddItem).not.toHaveBeenCalledWith(expect.objectContaining({
+                text: 'Request cancelled.',
+            }), expect.any(Number));
+        });
+        it('should prevent further processing after cancellation', async () => {
+            let continueStream;
+            const streamPromise = new Promise((resolve) => {
+                continueStream = resolve;
+            });
+            const mockStream = (async function* () {
+                yield { type: 'content', value: 'Initial' };
+                await streamPromise; // Wait until we manually continue
+                yield { type: 'content', value: ' Canceled' };
+            })();
+            mockSendMessageStream.mockReturnValue(mockStream);
+            const { result } = renderTestHook();
+            await act(async () => {
+                // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                result.current.submitQuery('long running query');
+            });
+            await waitFor(() => {
+                expect(result.current.streamingState).toBe(StreamingState.Responding);
+            });
+            // Cancel the request
+            simulateEscapeKeyPress();
+            // Allow the stream to continue
+            await act(async () => {
+                continueStream();
+                // Wait a bit to see if the second part is processed
+                await new Promise((resolve) => setTimeout(resolve, 50));
+            });
+            // The text should not have been updated with " Canceled"
+            const lastCall = mockAddItem.mock.calls.find((call) => call[0].type === 'gemini');
+            expect(lastCall?.[0].text).toBe('Initial');
+            // The final state should be idle after cancellation
+            expect(result.current.streamingState).toBe(StreamingState.Idle);
+        });
+        it('should cancel if a tool call is in progress', async () => {
+            const toolCalls = [
+                {
+                    request: { callId: 'call1', name: 'tool1', args: {} },
+                    status: 'executing',
+                    responseSubmittedToGemini: false,
+                    tool: {
+                        name: 'tool1',
+                        description: 'desc1',
+                        build: vi.fn().mockImplementation((_) => ({
+                            getDescription: () => `Mock description`,
+                        })),
+                    },
+                    invocation: {
+                        getDescription: () => `Mock description`,
+                    },
+                    startTime: Date.now(),
+                    liveOutput: '...',
+                },
+            ];
+            const { result } = renderTestHook(toolCalls);
+            // State is `Responding` because a tool is running
+            expect(result.current.streamingState).toBe(StreamingState.Responding);
+            // Try to cancel
+            simulateEscapeKeyPress();
+            // The cancel function should be called
+            expect(mockCancelAllToolCalls).toHaveBeenCalled();
+        });
+        it('should cancel a request when a tool is awaiting confirmation', async () => {
+            const mockOnConfirm = vi.fn().mockResolvedValue(undefined);
+            const toolCalls = [
+                {
+                    request: {
+                        callId: 'confirm-call',
+                        name: 'some_tool',
+                        args: {},
+                        isClientInitiated: false,
+                        prompt_id: 'prompt-id-1',
+                    },
+                    status: 'awaiting_approval',
+                    responseSubmittedToGemini: false,
+                    tool: {
+                        name: 'some_tool',
+                        description: 'a tool',
+                        build: vi.fn().mockImplementation((_) => ({
+                            getDescription: () => `Mock description`,
+                        })),
+                    },
+                    invocation: {
+                        getDescription: () => `Mock description`,
+                    },
+                    confirmationDetails: {
+                        type: 'edit',
+                        title: 'Confirm Edit',
+                        onConfirm: mockOnConfirm,
+                        fileName: 'file.txt',
+                        filePath: '/test/file.txt',
+                        fileDiff: 'fake diff',
+                        originalContent: 'old',
+                        newContent: 'new',
+                    },
+                },
+            ];
+            const { result } = renderTestHook(toolCalls);
+            // State is `WaitingForConfirmation` because a tool is awaiting approval
+            expect(result.current.streamingState).toBe(StreamingState.WaitingForConfirmation);
+            // Try to cancel
+            simulateEscapeKeyPress();
+            // The imperative cancel function should be called on the scheduler
+            expect(mockCancelAllToolCalls).toHaveBeenCalled();
+            // A cancellation message should be added to history
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
+                    text: 'Request cancelled.',
+                }), expect.any(Number));
+            });
+            // The final state should be idle
+            expect(result.current.streamingState).toBe(StreamingState.Idle);
+        });
+    });
+    describe('Slash Command Handling', () => {
+        it('should schedule a tool call when the command processor returns a schedule_tool action', async () => {
+            const clientToolRequest = {
+                type: 'schedule_tool',
+                toolName: 'save_memory',
+                toolArgs: { fact: 'test fact' },
+            };
+            mockHandleSlashCommand.mockResolvedValue(clientToolRequest);
+            const { result } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('/memory add "test fact"');
+            });
+            await waitFor(() => {
+                expect(mockScheduleToolCalls).toHaveBeenCalledWith([
+                    expect.objectContaining({
+                        name: 'save_memory',
+                        args: { fact: 'test fact' },
+                        isClientInitiated: true,
+                    }),
+                ], expect.any(AbortSignal));
+                expect(mockSendMessageStream).not.toHaveBeenCalled();
+            });
+        });
+        it('should stop processing and not call Gemini when a command is handled without a tool call', async () => {
+            const uiOnlyCommandResult = {
+                type: 'handled',
+            };
+            mockHandleSlashCommand.mockResolvedValue(uiOnlyCommandResult);
+            const { result } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('/help');
+            });
+            await waitFor(() => {
+                expect(mockHandleSlashCommand).toHaveBeenCalledWith('/help');
+                expect(mockScheduleToolCalls).not.toHaveBeenCalled();
+                expect(mockSendMessageStream).not.toHaveBeenCalled(); // No LLM call made
+            });
+        });
+        it('should call Gemini with prompt content when slash command returns a `submit_prompt` action', async () => {
+            const customCommandResult = {
+                type: 'submit_prompt',
+                content: 'This is the actual prompt from the command file.',
+            };
+            mockHandleSlashCommand.mockResolvedValue(customCommandResult);
+            const { result, mockSendMessageStream: localMockSendMessageStream } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('/my-custom-command');
+            });
+            await waitFor(() => {
+                expect(mockHandleSlashCommand).toHaveBeenCalledWith('/my-custom-command');
+                expect(localMockSendMessageStream).not.toHaveBeenCalledWith('/my-custom-command', expect.anything(), expect.anything());
+                expect(localMockSendMessageStream).toHaveBeenCalledWith('This is the actual prompt from the command file.', expect.any(AbortSignal), expect.any(String));
+                expect(mockScheduleToolCalls).not.toHaveBeenCalled();
+            });
+        });
+        it('should correctly handle a submit_prompt action with empty content', async () => {
+            const emptyPromptResult = {
+                type: 'submit_prompt',
+                content: '',
+            };
+            mockHandleSlashCommand.mockResolvedValue(emptyPromptResult);
+            const { result, mockSendMessageStream: localMockSendMessageStream } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('/emptycmd');
+            });
+            await waitFor(() => {
+                expect(mockHandleSlashCommand).toHaveBeenCalledWith('/emptycmd');
+                expect(localMockSendMessageStream).toHaveBeenCalledWith('', expect.any(AbortSignal), expect.any(String));
+            });
+        });
+        it('should not call handleSlashCommand for line comments', async () => {
+            const { result, mockSendMessageStream: localMockSendMessageStream } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('// This is a line comment');
+            });
+            await waitFor(() => {
+                expect(mockHandleSlashCommand).not.toHaveBeenCalled();
+                expect(localMockSendMessageStream).toHaveBeenCalledWith('// This is a line comment', expect.any(AbortSignal), expect.any(String));
+            });
+        });
+        it('should not call handleSlashCommand for block comments', async () => {
+            const { result, mockSendMessageStream: localMockSendMessageStream } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('/* This is a block comment */');
+            });
+            await waitFor(() => {
+                expect(mockHandleSlashCommand).not.toHaveBeenCalled();
+                expect(localMockSendMessageStream).toHaveBeenCalledWith('/* This is a block comment */', expect.any(AbortSignal), expect.any(String));
+            });
+        });
+        it('should not call handleSlashCommand is shell mode is active', async () => {
+            const { result } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, () => { }, mockHandleSlashCommand, true, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+            await act(async () => {
+                await result.current.submitQuery('/about');
+            });
+            await waitFor(() => {
+                expect(mockHandleSlashCommand).not.toHaveBeenCalled();
+            });
+        });
+    });
+    describe('Memory Refresh on save_memory', () => {
+        it('should call performMemoryRefresh when a save_memory tool call completes successfully', async () => {
+            const mockPerformMemoryRefresh = vi.fn();
+            const completedToolCall = {
+                request: {
+                    callId: 'save-mem-call-1',
+                    name: 'save_memory',
+                    args: { fact: 'test' },
+                    isClientInitiated: true,
+                    prompt_id: 'prompt-id-6',
+                },
+                status: 'success',
+                responseSubmittedToGemini: false,
+                response: {
+                    callId: 'save-mem-call-1',
+                    responseParts: [{ text: 'Memory saved' }],
+                    resultDisplay: 'Success: Memory saved',
+                    error: undefined,
+                    errorType: undefined, // FIX: Added missing property
+                },
+                tool: {
+                    name: 'save_memory',
+                    displayName: 'save_memory',
+                    description: 'Saves memory',
+                    build: vi.fn(),
+                },
+                invocation: {
+                    getDescription: () => `Mock description`,
+                },
+            };
+            // Capture the onComplete callback
+            let capturedOnComplete = null;
+            mockUseReactToolScheduler.mockImplementation((onComplete) => {
+                capturedOnComplete = onComplete;
+                return [[], mockScheduleToolCalls, mockMarkToolsAsSubmitted, vi.fn()];
+            });
+            renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, mockPerformMemoryRefresh, false, () => { }, () => { }, () => { }, 80, 24));
+            // Trigger the onComplete callback with the completed save_memory tool
+            await act(async () => {
+                if (capturedOnComplete) {
+                    await capturedOnComplete([completedToolCall]);
+                }
+            });
+            await waitFor(() => {
+                expect(mockPerformMemoryRefresh).toHaveBeenCalledTimes(1);
+            });
+        });
+    });
+    describe('Error Handling', () => {
+        it('should call parseAndFormatApiError with the correct authType on stream initialization failure', async () => {
+            // 1. Setup
+            const mockError = new Error('Rate limit exceeded');
+            const mockAuthType = AuthType.LOGIN_WITH_GOOGLE;
+            mockParseAndFormatApiError.mockClear();
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield { type: 'content', value: '' };
+                throw mockError;
+            })());
+            const testConfig = {
+                ...mockConfig,
+                getContentGeneratorConfig: vi.fn(() => ({
+                    authType: mockAuthType,
+                })),
+                getModel: vi.fn(() => 'gemini-2.5-pro'),
+            };
+            const { result } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(testConfig), [], mockAddItem, testConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+            // 2. Action
+            await act(async () => {
+                await result.current.submitQuery('test query');
+            });
+            // 3. Assertion
+            await waitFor(() => {
+                expect(mockParseAndFormatApiError).toHaveBeenCalledWith('Rate limit exceeded', mockAuthType, undefined, 'gemini-2.5-pro', 'gemini-2.5-flash');
+            });
+        });
+    });
+    describe('handleApprovalModeChange', () => {
+        it('should auto-approve all pending tool calls when switching to YOLO mode', async () => {
+            const mockOnConfirm = vi.fn().mockResolvedValue(undefined);
+            const awaitingApprovalToolCalls = [
+                createMockToolCall('replace', 'call1', 'edit', mockOnConfirm),
+                createMockToolCall('read_file', 'call2', 'info', mockOnConfirm),
+            ];
+            const { result } = renderTestHook(awaitingApprovalToolCalls);
+            await act(async () => {
+                await result.current.handleApprovalModeChange(ApprovalMode.YOLO);
+            });
+            // Both tool calls should be auto-approved
+            expect(mockOnConfirm).toHaveBeenCalledTimes(2);
+            expect(mockOnConfirm).toHaveBeenCalledWith(ToolConfirmationOutcome.ProceedOnce);
+        });
+        it('should only auto-approve edit tools when switching to AUTO_EDIT mode', async () => {
+            const mockOnConfirmReplace = vi.fn().mockResolvedValue(undefined);
+            const mockOnConfirmWrite = vi.fn().mockResolvedValue(undefined);
+            const mockOnConfirmRead = vi.fn().mockResolvedValue(undefined);
+            const awaitingApprovalToolCalls = [
+                createMockToolCall('replace', 'call1', 'edit', mockOnConfirmReplace),
+                createMockToolCall('write_file', 'call2', 'edit', mockOnConfirmWrite),
+                createMockToolCall('read_file', 'call3', 'info', mockOnConfirmRead),
+            ];
+            const { result } = renderTestHook(awaitingApprovalToolCalls);
+            await act(async () => {
+                await result.current.handleApprovalModeChange(ApprovalMode.AUTO_EDIT);
+            });
+            // Only replace and write_file should be auto-approved
+            expect(mockOnConfirmReplace).toHaveBeenCalledWith(ToolConfirmationOutcome.ProceedOnce);
+            expect(mockOnConfirmWrite).toHaveBeenCalledWith(ToolConfirmationOutcome.ProceedOnce);
+            // read_file should not be auto-approved
+            expect(mockOnConfirmRead).not.toHaveBeenCalled();
+        });
+        it('should not auto-approve any tools when switching to REQUIRE_CONFIRMATION mode', async () => {
+            const mockOnConfirm = vi.fn().mockResolvedValue(undefined);
+            const awaitingApprovalToolCalls = [
+                createMockToolCall('replace', 'call1', 'edit', mockOnConfirm),
+            ];
+            const { result } = renderTestHook(awaitingApprovalToolCalls);
+            await act(async () => {
+                await result.current.handleApprovalModeChange(ApprovalMode.DEFAULT);
+            });
+            // No tools should be auto-approved
+            expect(mockOnConfirm).not.toHaveBeenCalled();
+        });
+        it('should handle errors gracefully when auto-approving tool calls', async () => {
+            const debuggerSpy = vi
+                .spyOn(debugLogger, 'warn')
+                .mockImplementation(() => { });
+            const mockOnConfirmSuccess = vi.fn().mockResolvedValue(undefined);
+            const mockOnConfirmError = vi
+                .fn()
+                .mockRejectedValue(new Error('Approval failed'));
+            const awaitingApprovalToolCalls = [
+                createMockToolCall('replace', 'call1', 'edit', mockOnConfirmSuccess),
+                createMockToolCall('write_file', 'call2', 'edit', mockOnConfirmError),
+            ];
+            const { result } = renderTestHook(awaitingApprovalToolCalls);
+            await act(async () => {
+                await result.current.handleApprovalModeChange(ApprovalMode.YOLO);
+            });
+            // Both confirmation methods should be called
+            expect(mockOnConfirmSuccess).toHaveBeenCalled();
+            expect(mockOnConfirmError).toHaveBeenCalled();
+            // Error should be logged
+            expect(debuggerSpy).toHaveBeenCalledWith('Failed to auto-approve tool call call2:', expect.any(Error));
+            debuggerSpy.mockRestore();
+        });
+        it('should skip tool calls without confirmationDetails', async () => {
+            const awaitingApprovalToolCalls = [
+                {
+                    request: {
+                        callId: 'call1',
+                        name: 'replace',
+                        args: { old_string: 'old', new_string: 'new' },
+                        isClientInitiated: false,
+                        prompt_id: 'prompt-id-1',
+                    },
+                    status: 'awaiting_approval',
+                    responseSubmittedToGemini: false,
+                    // No confirmationDetails
+                    tool: {
+                        name: 'replace',
+                        displayName: 'replace',
+                        description: 'Replace text',
+                        build: vi.fn(),
+                    },
+                    invocation: {
+                        getDescription: () => 'Mock description',
+                    },
+                },
+            ];
+            const { result } = renderTestHook(awaitingApprovalToolCalls);
+            // Should not throw an error
+            await act(async () => {
+                await result.current.handleApprovalModeChange(ApprovalMode.YOLO);
+            });
+        });
+        it('should skip tool calls without onConfirm method in confirmationDetails', async () => {
+            const awaitingApprovalToolCalls = [
+                {
+                    request: {
+                        callId: 'call1',
+                        name: 'replace',
+                        args: { old_string: 'old', new_string: 'new' },
+                        isClientInitiated: false,
+                        prompt_id: 'prompt-id-1',
+                    },
+                    status: 'awaiting_approval',
+                    responseSubmittedToGemini: false,
+                    confirmationDetails: {
+                        type: 'edit',
+                        title: 'Confirm Edit',
+                        // No onConfirm method
+                        fileName: 'file.txt',
+                        filePath: '/test/file.txt',
+                        fileDiff: 'fake diff',
+                        originalContent: 'old',
+                        newContent: 'new',
+                    },
+                    tool: {
+                        name: 'replace',
+                        displayName: 'replace',
+                        description: 'Replace text',
+                        build: vi.fn(),
+                    },
+                    invocation: {
+                        getDescription: () => 'Mock description',
+                    },
+                },
+            ];
+            const { result } = renderTestHook(awaitingApprovalToolCalls);
+            // Should not throw an error
+            await act(async () => {
+                await result.current.handleApprovalModeChange(ApprovalMode.YOLO);
+            });
+        });
+        it('should only process tool calls with awaiting_approval status', async () => {
+            const mockOnConfirmAwaiting = vi.fn().mockResolvedValue(undefined);
+            const mockOnConfirmExecuting = vi.fn().mockResolvedValue(undefined);
+            const mixedStatusToolCalls = [
+                {
+                    request: {
+                        callId: 'call1',
+                        name: 'replace',
+                        args: { old_string: 'old', new_string: 'new' },
+                        isClientInitiated: false,
+                        prompt_id: 'prompt-id-1',
+                    },
+                    status: 'awaiting_approval',
+                    responseSubmittedToGemini: false,
+                    confirmationDetails: {
+                        type: 'edit',
+                        title: 'Confirm Edit',
+                        onConfirm: mockOnConfirmAwaiting,
+                        fileName: 'file.txt',
+                        filePath: '/test/file.txt',
+                        fileDiff: 'fake diff',
+                        originalContent: 'old',
+                        newContent: 'new',
+                    },
+                    tool: {
+                        name: 'replace',
+                        displayName: 'replace',
+                        description: 'Replace text',
+                        build: vi.fn(),
+                    },
+                    invocation: {
+                        getDescription: () => 'Mock description',
+                    },
+                },
+                {
+                    request: {
+                        callId: 'call2',
+                        name: 'write_file',
+                        args: { path: '/test/file.txt', content: 'content' },
+                        isClientInitiated: false,
+                        prompt_id: 'prompt-id-1',
+                    },
+                    status: 'executing',
+                    responseSubmittedToGemini: false,
+                    tool: {
+                        name: 'write_file',
+                        displayName: 'write_file',
+                        description: 'Write file',
+                        build: vi.fn(),
+                    },
+                    invocation: {
+                        getDescription: () => 'Mock description',
+                    },
+                    startTime: Date.now(),
+                    liveOutput: 'Writing...',
+                },
+            ];
+            const { result } = renderTestHook(mixedStatusToolCalls);
+            await act(async () => {
+                await result.current.handleApprovalModeChange(ApprovalMode.YOLO);
+            });
+            // Only the awaiting_approval tool should be processed
+            expect(mockOnConfirmAwaiting).toHaveBeenCalledTimes(1);
+            expect(mockOnConfirmExecuting).not.toHaveBeenCalled();
+        });
+    });
+    describe('handleFinishedEvent', () => {
+        it('should add info message for MAX_TOKENS finish reason', async () => {
+            // Setup mock to return a stream with MAX_TOKENS finish reason
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: 'This is a truncated response...',
+                };
+                yield {
+                    type: ServerGeminiEventType.Finished,
+                    value: { reason: 'MAX_TOKENS', usageMetadata: undefined },
+                };
+            })());
+            const { result } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+            // Submit a query
+            await act(async () => {
+                await result.current.submitQuery('Generate long text');
+            });
+            // Check that the info message was added
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith({
+                    type: 'info',
+                    text: '⚠️  Response truncated due to token limits.',
+                }, expect.any(Number));
+            });
+        });
+        describe('ContextWindowWillOverflow event', () => {
+            beforeEach(() => {
+                vi.mocked(tokenLimit).mockReturnValue(100);
+            });
+            it.each([
+                {
+                    name: 'without suggestion when remaining tokens are > 75% of limit',
+                    requestTokens: 20,
+                    remainingTokens: 80,
+                    expectedMessage: 'Sending this message (20 tokens) might exceed the remaining context window limit (80 tokens).',
+                },
+                {
+                    name: 'with suggestion when remaining tokens are < 75% of limit',
+                    requestTokens: 30,
+                    remainingTokens: 70,
+                    expectedMessage: 'Sending this message (30 tokens) might exceed the remaining context window limit (70 tokens). Please try reducing the size of your message or use the `/compress` command to compress the chat history.',
+                },
+            ])('should add message $name', async ({ requestTokens, remainingTokens, expectedMessage }) => {
+                mockSendMessageStream.mockReturnValue((async function* () {
+                    yield {
+                        type: ServerGeminiEventType.ContextWindowWillOverflow,
+                        value: {
+                            estimatedRequestTokenCount: requestTokens,
+                            remainingTokenCount: remainingTokens,
+                        },
+                    };
+                })());
+                const { result } = renderHookWithDefaults();
+                await act(async () => {
+                    await result.current.submitQuery('Test overflow');
+                });
+                await waitFor(() => {
+                    expect(mockAddItem).toHaveBeenCalledWith({
+                        type: 'info',
+                        text: expectedMessage,
+                    }, expect.any(Number));
+                });
+            });
+        });
+        it('should call onCancelSubmit when ContextWindowWillOverflow event is received', async () => {
+            const onCancelSubmitSpy = vi.fn();
+            // Setup mock to return a stream with ContextWindowWillOverflow event
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.ContextWindowWillOverflow,
+                    value: {
+                        estimatedRequestTokenCount: 100,
+                        remainingTokenCount: 50,
+                    },
+                };
+            })());
+            const { result } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, onCancelSubmitSpy, () => { }, 80, 24));
+            // Submit a query
+            await act(async () => {
+                await result.current.submitQuery('Test overflow');
+            });
+            // Check that onCancelSubmit was called
+            await waitFor(() => {
+                expect(onCancelSubmitSpy).toHaveBeenCalledWith(true);
+            });
+        });
+        it.each([
+            {
+                reason: 'STOP',
+                shouldAddMessage: false,
+            },
+            {
+                reason: 'FINISH_REASON_UNSPECIFIED',
+                shouldAddMessage: false,
+            },
+            {
+                reason: 'SAFETY',
+                message: '⚠️  Response stopped due to safety reasons.',
+            },
+            {
+                reason: 'RECITATION',
+                message: '⚠️  Response stopped due to recitation policy.',
+            },
+            {
+                reason: 'LANGUAGE',
+                message: '⚠️  Response stopped due to unsupported language.',
+            },
+            {
+                reason: 'BLOCKLIST',
+                message: '⚠️  Response stopped due to forbidden terms.',
+            },
+            {
+                reason: 'PROHIBITED_CONTENT',
+                message: '⚠️  Response stopped due to prohibited content.',
+            },
+            {
+                reason: 'SPII',
+                message: '⚠️  Response stopped due to sensitive personally identifiable information.',
+            },
+            {
+                reason: 'OTHER',
+                message: '⚠️  Response stopped for other reasons.',
+            },
+            {
+                reason: 'MALFORMED_FUNCTION_CALL',
+                message: '⚠️  Response stopped due to malformed function call.',
+            },
+            {
+                reason: 'IMAGE_SAFETY',
+                message: '⚠️  Response stopped due to image safety violations.',
+            },
+            {
+                reason: 'UNEXPECTED_TOOL_CALL',
+                message: '⚠️  Response stopped due to unexpected tool call.',
+            },
+        ])('should handle $reason finish reason correctly', async ({ reason, shouldAddMessage = true, message }) => {
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: `Response for ${reason}`,
+                };
+                yield {
+                    type: ServerGeminiEventType.Finished,
+                    value: { reason, usageMetadata: undefined },
+                };
+            })());
+            const { result } = renderHookWithDefaults();
+            await act(async () => {
+                await result.current.submitQuery(`Test ${reason}`);
+            });
+            if (shouldAddMessage) {
+                await waitFor(() => {
+                    expect(mockAddItem).toHaveBeenCalledWith({
+                        type: 'info',
+                        text: message,
+                    }, expect.any(Number));
+                });
+            }
+            else {
+                // Verify state returns to idle without any info messages
+                await waitFor(() => {
+                    expect(result.current.streamingState).toBe(StreamingState.Idle);
+                });
+                const infoMessages = mockAddItem.mock.calls.filter((call) => call[0].type === 'info');
+                expect(infoMessages).toHaveLength(0);
+            }
+        });
+    });
+    it('should process @include commands, adding user turn after processing to prevent race conditions', async () => {
+        const rawQuery = '@include file.txt Summarize this.';
+        const processedQueryParts = [
+            { text: 'Summarize this with content from @file.txt' },
+            { text: 'File content...' },
+        ];
+        const userMessageTimestamp = Date.now();
+        vi.spyOn(Date, 'now').mockReturnValue(userMessageTimestamp);
+        handleAtCommandSpy.mockResolvedValue({
+            processedQuery: processedQueryParts,
+            shouldProceed: true,
+        });
+        const { result } = renderHook(() => useGeminiStream(mockConfig.getGeminiClient(), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, // shellModeActive
+        vi.fn(), // getPreferredEditor
+        vi.fn(), // onAuthError
+        vi.fn(), // performMemoryRefresh
+        false, // modelSwitched
+        vi.fn(), // setModelSwitched
+        vi.fn(), // onCancelSubmit
+        vi.fn(), // setShellInputFocused
+        80, // terminalWidth
+        24));
+        await act(async () => {
+            await result.current.submitQuery(rawQuery);
+        });
+        expect(handleAtCommandSpy).toHaveBeenCalledWith(expect.objectContaining({
+            query: rawQuery,
+        }));
+        expect(mockAddItem).toHaveBeenCalledWith({
+            type: MessageType.USER,
+            text: rawQuery,
+        }, userMessageTimestamp);
+        // FIX: The expectation now matches the actual call signature.
+        expect(mockSendMessageStream).toHaveBeenCalledWith(processedQueryParts, // Argument 1: The parts array directly
+        expect.any(AbortSignal), // Argument 2: An AbortSignal
+        expect.any(String));
+    });
+    describe('Thought Reset', () => {
+        it('should reset thought to null when starting a new prompt', async () => {
+            // First, simulate a response with a thought
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Thought,
+                    value: {
+                        subject: 'Previous thought',
+                        description: 'Old description',
+                    },
+                };
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: 'Some response content',
+                };
+                yield {
+                    type: ServerGeminiEventType.Finished,
+                    value: { reason: 'STOP', usageMetadata: undefined },
+                };
+            })());
+            const { result } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+            // Submit first query to set a thought
+            await act(async () => {
+                await result.current.submitQuery('First query');
+            });
+            // Wait for the first response to complete
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
+                    type: 'gemini',
+                    text: 'Some response content',
+                }), expect.any(Number));
+            });
+            // Now simulate a new response without a thought
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: 'New response content',
+                };
+                yield {
+                    type: ServerGeminiEventType.Finished,
+                    value: { reason: 'STOP', usageMetadata: undefined },
+                };
+            })());
+            // Submit second query - thought should be reset
+            await act(async () => {
+                await result.current.submitQuery('Second query');
+            });
+            // The thought should be reset to null when starting the new prompt
+            // We can verify this by checking that the LoadingIndicator would not show the previous thought
+            // The actual thought state is internal to the hook, but we can verify the behavior
+            // by ensuring the second response doesn't show the previous thought
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
+                    type: 'gemini',
+                    text: 'New response content',
+                }), expect.any(Number));
+            });
+        });
+        it('should memoize pendingHistoryItems', () => {
+            mockUseReactToolScheduler.mockReturnValue([
+                [],
+                mockScheduleToolCalls,
+                mockCancelAllToolCalls,
+                mockMarkToolsAsSubmitted,
+            ]);
+            const { result, rerender } = renderHook(() => useGeminiStream(mockConfig.getGeminiClient(), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+            const firstResult = result.current.pendingHistoryItems;
+            rerender();
+            const secondResult = result.current.pendingHistoryItems;
+            expect(firstResult).toStrictEqual(secondResult);
+            const newToolCalls = [
+                {
+                    request: { callId: 'call1', name: 'tool1', args: {} },
+                    status: 'executing',
+                    tool: {
+                        name: 'tool1',
+                        displayName: 'tool1',
+                        description: 'desc1',
+                        build: vi.fn(),
+                    },
+                    invocation: {
+                        getDescription: () => 'Mock description',
+                    },
+                },
+            ];
+            mockUseReactToolScheduler.mockReturnValue([
+                newToolCalls,
+                mockScheduleToolCalls,
+                mockCancelAllToolCalls,
+                mockMarkToolsAsSubmitted,
+            ]);
+            rerender();
+            const thirdResult = result.current.pendingHistoryItems;
+            expect(thirdResult).not.toStrictEqual(secondResult);
+        });
+        it('should reset thought to null when user cancels', async () => {
+            // Mock a stream that yields a thought then gets cancelled
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Thought,
+                    value: { subject: 'Some thought', description: 'Description' },
+                };
+                yield { type: ServerGeminiEventType.UserCancelled };
+            })());
+            const { result } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+            // Submit query
+            await act(async () => {
+                await result.current.submitQuery('Test query');
+            });
+            // Verify cancellation message was added
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
+                    type: 'info',
+                    text: 'User cancelled the request.',
+                }), expect.any(Number));
+            });
+            // Verify state is reset to idle
+            expect(result.current.streamingState).toBe(StreamingState.Idle);
+        });
+        it('should reset thought to null when there is an error', async () => {
+            // Mock a stream that yields a thought then encounters an error
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Thought,
+                    value: { subject: 'Some thought', description: 'Description' },
+                };
+                yield {
+                    type: ServerGeminiEventType.Error,
+                    value: { error: { message: 'Test error' } },
+                };
+            })());
+            const { result } = renderHook(() => useGeminiStream(new MockedGeminiClientClass(mockConfig), [], mockAddItem, mockConfig, mockLoadedSettings, mockOnDebugMessage, mockHandleSlashCommand, false, () => 'vscode', () => { }, () => Promise.resolve(), false, () => { }, () => { }, () => { }, 80, 24));
+            // Submit query
+            await act(async () => {
+                await result.current.submitQuery('Test query');
+            });
+            // Verify error message was added
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
+                    type: 'error',
+                }), expect.any(Number));
+            });
+            // Verify parseAndFormatApiError was called
+            expect(mockParseAndFormatApiError).toHaveBeenCalledWith({ message: 'Test error' }, expect.any(String), undefined, 'gemini-2.5-pro', 'gemini-2.5-flash');
+        });
+    });
+    describe('Loop Detection Confirmation', () => {
+        beforeEach(() => {
+            // Add mock for getLoopDetectionService to the config
+            const mockLoopDetectionService = {
+                disableForSession: vi.fn(),
+            };
+            mockConfig.getGeminiClient = vi.fn().mockReturnValue({
+                ...new MockedGeminiClientClass(mockConfig),
+                getLoopDetectionService: () => mockLoopDetectionService,
+            });
+        });
+        it('should set loopDetectionConfirmationRequest when LoopDetected event is received', async () => {
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: 'Some content',
+                };
+                yield {
+                    type: ServerGeminiEventType.LoopDetected,
+                };
+            })());
+            const { result } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('test query');
+            });
+            await waitFor(() => {
+                expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
+                expect(typeof result.current.loopDetectionConfirmationRequest?.onComplete).toBe('function');
+            });
+        });
+        it('should disable loop detection and show message when user selects "disable"', async () => {
+            const mockLoopDetectionService = {
+                disableForSession: vi.fn(),
+            };
+            const mockClient = {
+                ...new MockedGeminiClientClass(mockConfig),
+                getLoopDetectionService: () => mockLoopDetectionService,
+            };
+            mockConfig.getGeminiClient = vi.fn().mockReturnValue(mockClient);
+            // Mock for the initial request
+            mockSendMessageStream.mockReturnValueOnce((async function* () {
+                yield {
+                    type: ServerGeminiEventType.LoopDetected,
+                };
+            })());
+            // Mock for the retry request
+            mockSendMessageStream.mockReturnValueOnce((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: 'Retry successful',
+                };
+                yield {
+                    type: ServerGeminiEventType.Finished,
+                    value: { reason: 'STOP' },
+                };
+            })());
+            const { result } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('test query');
+            });
+            // Wait for confirmation request to be set
+            await waitFor(() => {
+                expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
+            });
+            // Simulate user selecting "disable"
+            await act(async () => {
+                result.current.loopDetectionConfirmationRequest?.onComplete({
+                    userSelection: 'disable',
+                });
+            });
+            // Verify loop detection was disabled
+            expect(mockLoopDetectionService.disableForSession).toHaveBeenCalledTimes(1);
+            // Verify confirmation request was cleared
+            expect(result.current.loopDetectionConfirmationRequest).toBeNull();
+            // Verify appropriate message was added
+            expect(mockAddItem).toHaveBeenCalledWith({
+                type: 'info',
+                text: 'Loop detection has been disabled for this session. Retrying request...',
+            }, expect.any(Number));
+            // Verify that the request was retried
+            await waitFor(() => {
+                expect(mockSendMessageStream).toHaveBeenCalledTimes(2);
+                expect(mockSendMessageStream).toHaveBeenNthCalledWith(2, 'test query', expect.any(AbortSignal), expect.any(String));
+            });
+        });
+        it('should keep loop detection enabled and show message when user selects "keep"', async () => {
+            const mockLoopDetectionService = {
+                disableForSession: vi.fn(),
+            };
+            const mockClient = {
+                ...new MockedGeminiClientClass(mockConfig),
+                getLoopDetectionService: () => mockLoopDetectionService,
+            };
+            mockConfig.getGeminiClient = vi.fn().mockReturnValue(mockClient);
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.LoopDetected,
+                };
+            })());
+            const { result } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('test query');
+            });
+            // Wait for confirmation request to be set
+            await waitFor(() => {
+                expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
+            });
+            // Simulate user selecting "keep"
+            await act(async () => {
+                result.current.loopDetectionConfirmationRequest?.onComplete({
+                    userSelection: 'keep',
+                });
+            });
+            // Verify loop detection was NOT disabled
+            expect(mockLoopDetectionService.disableForSession).not.toHaveBeenCalled();
+            // Verify confirmation request was cleared
+            expect(result.current.loopDetectionConfirmationRequest).toBeNull();
+            // Verify appropriate message was added
+            expect(mockAddItem).toHaveBeenCalledWith({
+                type: 'info',
+                text: 'A potential loop was detected. This can happen due to repetitive tool calls or other model behavior. The request has been halted.',
+            }, expect.any(Number));
+            // Verify that the request was NOT retried
+            expect(mockSendMessageStream).toHaveBeenCalledTimes(1);
+        });
+        it('should handle multiple loop detection events properly', async () => {
+            const { result } = renderTestHook();
+            // First loop detection - set up fresh mock for first call
+            mockSendMessageStream.mockReturnValueOnce((async function* () {
+                yield {
+                    type: ServerGeminiEventType.LoopDetected,
+                };
+            })());
+            // First loop detection
+            await act(async () => {
+                await result.current.submitQuery('first query');
+            });
+            await waitFor(() => {
+                expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
+            });
+            // Simulate user selecting "keep" for first request
+            await act(async () => {
+                result.current.loopDetectionConfirmationRequest?.onComplete({
+                    userSelection: 'keep',
+                });
+            });
+            expect(result.current.loopDetectionConfirmationRequest).toBeNull();
+            // Verify first message was added
+            expect(mockAddItem).toHaveBeenCalledWith({
+                type: 'info',
+                text: 'A potential loop was detected. This can happen due to repetitive tool calls or other model behavior. The request has been halted.',
+            }, expect.any(Number));
+            // Second loop detection - set up fresh mock for second call
+            mockSendMessageStream.mockReturnValueOnce((async function* () {
+                yield {
+                    type: ServerGeminiEventType.LoopDetected,
+                };
+            })());
+            // Mock for the retry request
+            mockSendMessageStream.mockReturnValueOnce((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: 'Retry successful',
+                };
+                yield {
+                    type: ServerGeminiEventType.Finished,
+                    value: { reason: 'STOP' },
+                };
+            })());
+            // Second loop detection
+            await act(async () => {
+                await result.current.submitQuery('second query');
+            });
+            await waitFor(() => {
+                expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
+            });
+            // Simulate user selecting "disable" for second request
+            await act(async () => {
+                result.current.loopDetectionConfirmationRequest?.onComplete({
+                    userSelection: 'disable',
+                });
+            });
+            expect(result.current.loopDetectionConfirmationRequest).toBeNull();
+            // Verify second message was added
+            expect(mockAddItem).toHaveBeenCalledWith({
+                type: 'info',
+                text: 'Loop detection has been disabled for this session. Retrying request...',
+            }, expect.any(Number));
+            // Verify that the request was retried
+            await waitFor(() => {
+                expect(mockSendMessageStream).toHaveBeenCalledTimes(3); // 1st query, 2nd query, retry of 2nd query
+                expect(mockSendMessageStream).toHaveBeenNthCalledWith(3, 'second query', expect.any(AbortSignal), expect.any(String));
+            });
+        });
+        it('should process LoopDetected event after moving pending history to history', async () => {
+            mockSendMessageStream.mockReturnValue((async function* () {
+                yield {
+                    type: ServerGeminiEventType.Content,
+                    value: 'Some response content',
+                };
+                yield {
+                    type: ServerGeminiEventType.LoopDetected,
+                };
+            })());
+            const { result } = renderTestHook();
+            await act(async () => {
+                await result.current.submitQuery('test query');
+            });
+            // Verify that the content was added to history before the loop detection dialog
+            await waitFor(() => {
+                expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
+                    type: 'gemini',
+                    text: 'Some response content',
+                }), expect.any(Number));
+            });
+            // Then verify loop detection confirmation request was set
+            await waitFor(() => {
+                expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
+            });
+        });
+    });
+});
+//# sourceMappingURL=useGeminiStream.test.js.map