@terminai/cli 0.21.0

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

@@ -0,0 +1,860 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useState, useRef, useCallback, useEffect, useMemo } from 'react';
+import { GeminiEventType as ServerGeminiEventType, getErrorMessage, isNodeError, MessageSenderType, logUserPrompt, GitService, UnauthorizedError, UserPromptEvent, DEFAULT_GEMINI_FLASH_MODEL, logConversationFinishedEvent, ConversationFinishedEvent, ApprovalMode, parseAndFormatApiError, ToolConfirmationOutcome, promptIdContext, tokenLimit, debugLogger, runInDevTraceSpan, EDIT_TOOL_NAMES, processRestorableToolCalls, recordToolCallInteractions, SHELL_TOOL_NAME, } from '@terminai/core';
+import { FinishReason } from '@google/genai';
+import { StreamingState, MessageType, ToolCallStatus } from '../types.js';
+import { isAtCommand, isSlashCommand } from '../utils/commandUtils.js';
+import { useShellCommandProcessor } from './shellCommandProcessor.js';
+import { handleAtCommand } from './atCommandProcessor.js';
+import { findLastSafeSplitPoint } from '../utils/markdownUtilities.js';
+import { useStateAndRef } from './useStateAndRef.js';
+import { useLogger } from './useLogger.js';
+import { SHELL_COMMAND_NAME } from '../constants.js';
+import { useReactToolScheduler, mapToDisplay as mapTrackedToolCallsToDisplay, } from './useReactToolScheduler.js';
+import { promises as fs } from 'node:fs';
+import path from 'node:path';
+import { useSessionStats } from '../contexts/SessionContext.js';
+import { useKeypress } from './useKeypress.js';
+var StreamProcessingStatus;
+(function (StreamProcessingStatus) {
+    StreamProcessingStatus[StreamProcessingStatus["Completed"] = 0] = "Completed";
+    StreamProcessingStatus[StreamProcessingStatus["UserCancelled"] = 1] = "UserCancelled";
+    StreamProcessingStatus[StreamProcessingStatus["Error"] = 2] = "Error";
+})(StreamProcessingStatus || (StreamProcessingStatus = {}));
+function showCitations(settings, config) {
+    // Check if settings explicitly disable citations
+    const enabled = settings?.merged?.ui?.showCitations;
+    if (enabled !== undefined && !enabled) {
+        return false;
+    }
+    // Check if provider supports citations
+    if (config) {
+        const capabilities = config.getProviderCapabilities?.();
+        if (capabilities && !capabilities.supportsCitations) {
+            return false;
+        }
+    }
+    return true;
+}
+/**
+ * Manages the Gemini stream, including user input, command processing,
+ * API interaction, and tool call lifecycle.
+ */
+export const useGeminiStream = (geminiClient, history, addItem, config, settings, onDebugMessage, handleSlashCommand, shellModeActive, getPreferredEditor, onAuthError, performMemoryRefresh, modelSwitchedFromQuotaError, setModelSwitchedFromQuotaError, onCancelSubmit, setShellInputFocused, terminalWidth, terminalHeight, isShellFocused) => {
+    const [initError, setInitError] = useState(null);
+    const abortControllerRef = useRef(null);
+    const turnCancelledRef = useRef(false);
+    const activeQueryIdRef = useRef(null);
+    const [isResponding, setIsResponding] = useState(false);
+    const [thought, setThought] = useState(null);
+    const [pendingHistoryItem, pendingHistoryItemRef, setPendingHistoryItem] = useStateAndRef(null);
+    const processedMemoryToolsRef = useRef(new Set());
+    const { startNewPrompt, getPromptCount } = useSessionStats();
+    const storage = config.storage;
+    const logger = useLogger(storage);
+    const gitService = useMemo(() => {
+        if (!config.getProjectRoot()) {
+            return;
+        }
+        return new GitService(config.getProjectRoot(), storage);
+    }, [config, storage]);
+    const [toolCalls, scheduleToolCalls, markToolsAsSubmitted, setToolCallsForDisplay, cancelAllToolCalls, lastToolOutputTime,] = useReactToolScheduler(async (completedToolCallsFromScheduler) => {
+        // This onComplete is called when ALL scheduled tools for a given batch are done.
+        if (completedToolCallsFromScheduler.length > 0) {
+            // Add the final state of these tools to the history for display.
+            addItem(mapTrackedToolCallsToDisplay(completedToolCallsFromScheduler), Date.now());
+            // Clear the live-updating display now that the final state is in history.
+            setToolCallsForDisplay([]);
+            // Record tool calls with full metadata before sending responses.
+            try {
+                const currentModel = config.getGeminiClient().getCurrentSequenceModel() ??
+                    config.getModel();
+                config
+                    .getGeminiClient()
+                    .getChat()
+                    .recordCompletedToolCalls(currentModel, completedToolCallsFromScheduler);
+                await recordToolCallInteractions(config, completedToolCallsFromScheduler);
+            }
+            catch (error) {
+                debugLogger.warn(`Error recording completed tool call information: ${error}`);
+            }
+            // Handle tool response submission immediately when tools complete
+            await handleCompletedTools(completedToolCallsFromScheduler);
+        }
+    }, config, getPreferredEditor, logger || undefined);
+    useEffect(() => {
+        if (logger) {
+            void logger.logEventFull('session_start', {
+                model: config.getModel(),
+                cwd: config.getProjectRoot(),
+            });
+        }
+    }, [logger, config]);
+    const pendingToolCallGroupDisplay = useMemo(() => toolCalls.length ? mapTrackedToolCallsToDisplay(toolCalls) : undefined, [toolCalls]);
+    const activeToolPtyId = useMemo(() => {
+        const executingShellTool = toolCalls?.find((tc) => tc.status === 'executing' &&
+            (tc.request.name === SHELL_TOOL_NAME ||
+                tc.request.name === 'run_shell_command'));
+        if (executingShellTool) {
+            return executingShellTool.pid;
+        }
+        return undefined;
+    }, [toolCalls]);
+    const lastQueryRef = useRef(null);
+    const lastPromptIdRef = useRef(null);
+    const loopDetectedRef = useRef(false);
+    const [loopDetectionConfirmationRequest, setLoopDetectionConfirmationRequest,] = useState(null);
+    const onExec = useCallback(async (done) => {
+        setIsResponding(true);
+        await done;
+        setIsResponding(false);
+    }, []);
+    const { handleShellCommand, activeShellPtyId, lastShellOutputTime, interactivePasswordPrompt, isFullScreen, clearInteractivePasswordPrompt, } = useShellCommandProcessor(addItem, setPendingHistoryItem, onExec, onDebugMessage, config, geminiClient, setShellInputFocused, terminalWidth, terminalHeight);
+    const activePtyId = activeShellPtyId || activeToolPtyId;
+    useEffect(() => {
+        if (!activePtyId) {
+            setShellInputFocused(false);
+        }
+    }, [activePtyId, setShellInputFocused]);
+    const prevActiveShellPtyIdRef = useRef(null);
+    useEffect(() => {
+        if (turnCancelledRef.current &&
+            prevActiveShellPtyIdRef.current !== null &&
+            activeShellPtyId === null) {
+            addItem({ type: MessageType.INFO, text: 'Request cancelled.' }, Date.now());
+            setIsResponding(false);
+        }
+        prevActiveShellPtyIdRef.current = activeShellPtyId;
+    }, [activeShellPtyId, addItem]);
+    const streamingState = useMemo(() => {
+        if (toolCalls.some((tc) => tc.status === 'awaiting_approval')) {
+            return StreamingState.WaitingForConfirmation;
+        }
+        if (isResponding ||
+            toolCalls.some((tc) => tc.status === 'executing' ||
+                tc.status === 'scheduled' ||
+                tc.status === 'validating' ||
+                ((tc.status === 'success' ||
+                    tc.status === 'error' ||
+                    tc.status === 'cancelled') &&
+                    !tc
+                        .responseSubmittedToGemini))) {
+            return StreamingState.Responding;
+        }
+        return StreamingState.Idle;
+    }, [isResponding, toolCalls]);
+    useEffect(() => {
+        if (config.getApprovalMode() === ApprovalMode.YOLO &&
+            streamingState === StreamingState.Idle) {
+            const lastUserMessageIndex = history.findLastIndex((item) => item.type === MessageType.USER);
+            const turnCount = lastUserMessageIndex === -1 ? 0 : history.length - lastUserMessageIndex;
+            if (turnCount > 0) {
+                logConversationFinishedEvent(config, new ConversationFinishedEvent(config.getApprovalMode(), turnCount));
+            }
+        }
+    }, [streamingState, config, history]);
+    const cancelOngoingRequest = useCallback(() => {
+        if (streamingState !== StreamingState.Responding &&
+            streamingState !== StreamingState.WaitingForConfirmation) {
+            return;
+        }
+        if (turnCancelledRef.current) {
+            return;
+        }
+        turnCancelledRef.current = true;
+        // A full cancellation means no tools have produced a final result yet.
+        // This determines if we show a generic "Request cancelled" message.
+        const isFullCancellation = !toolCalls.some((tc) => tc.status === 'success' || tc.status === 'error');
+        // Ensure we have an abort controller, creating one if it doesn't exist.
+        if (!abortControllerRef.current) {
+            abortControllerRef.current = new AbortController();
+        }
+        // The order is important here.
+        // 1. Fire the signal to interrupt any active async operations.
+        abortControllerRef.current.abort();
+        // 2. Call the imperative cancel to clear the queue of pending tools.
+        cancelAllToolCalls(abortControllerRef.current.signal);
+        if (pendingHistoryItemRef.current) {
+            const isShellCommand = pendingHistoryItemRef.current.type === 'tool_group' &&
+                pendingHistoryItemRef.current.tools.some((t) => t.name === SHELL_COMMAND_NAME);
+            // If it is a shell command, we update the status to Canceled and clear the output
+            // to avoid artifacts, then add it to history immediately.
+            if (isShellCommand) {
+                const toolGroup = pendingHistoryItemRef.current;
+                const updatedTools = toolGroup.tools.map((tool) => {
+                    if (tool.name === SHELL_COMMAND_NAME) {
+                        return {
+                            ...tool,
+                            status: ToolCallStatus.Canceled,
+                            resultDisplay: tool.resultDisplay,
+                        };
+                    }
+                    return tool;
+                });
+                addItem({ ...toolGroup, tools: updatedTools }, Date.now());
+            }
+            else {
+                addItem(pendingHistoryItemRef.current, Date.now());
+            }
+        }
+        setPendingHistoryItem(null);
+        // If it was a full cancellation, add the info message now.
+        // Otherwise, we let handleCompletedTools figure out the next step,
+        // which might involve sending partial results back to the model.
+        if (isFullCancellation) {
+            // If shell is active, we delay this message to ensure correct ordering
+            // (Shell item first, then Info message).
+            if (!activeShellPtyId) {
+                addItem({
+                    type: MessageType.INFO,
+                    text: 'Request cancelled.',
+                }, Date.now());
+                setIsResponding(false);
+            }
+        }
+        onCancelSubmit(false);
+        setShellInputFocused(false);
+    }, [
+        streamingState,
+        addItem,
+        setPendingHistoryItem,
+        onCancelSubmit,
+        pendingHistoryItemRef,
+        setShellInputFocused,
+        cancelAllToolCalls,
+        toolCalls,
+        activeShellPtyId,
+    ]);
+    useKeypress((key) => {
+        if (key.name === 'escape' && !isShellFocused) {
+            cancelOngoingRequest();
+        }
+    }, {
+        isActive: streamingState === StreamingState.Responding ||
+            streamingState === StreamingState.WaitingForConfirmation,
+    });
+    const prepareQueryForGemini = useCallback(async (query, userMessageTimestamp, abortSignal, prompt_id) => {
+        if (turnCancelledRef.current) {
+            return { queryToSend: null, shouldProceed: false };
+        }
+        if (typeof query === 'string' && query.trim().length === 0) {
+            return { queryToSend: null, shouldProceed: false };
+        }
+        let localQueryToSendToGemini = null;
+        if (typeof query === 'string') {
+            const trimmedQuery = query.trim();
+            await logger?.logMessage(MessageSenderType.USER, trimmedQuery);
+            await logger?.logEventFull('user_prompt', { prompt: trimmedQuery });
+            if (!shellModeActive) {
+                // Handle UI-only commands first
+                const slashCommandResult = isSlashCommand(trimmedQuery)
+                    ? await handleSlashCommand(trimmedQuery)
+                    : false;
+                if (slashCommandResult) {
+                    switch (slashCommandResult.type) {
+                        case 'schedule_tool': {
+                            const { toolName, toolArgs } = slashCommandResult;
+                            const toolCallRequest = {
+                                callId: `${toolName}-${Date.now()}-${Math.random().toString(16).slice(2)}`,
+                                name: toolName,
+                                args: toolArgs,
+                                isClientInitiated: true,
+                                prompt_id,
+                                provenance: ['local_user', ...config.getSessionProvenance()],
+                            };
+                            scheduleToolCalls([toolCallRequest], abortSignal);
+                            return { queryToSend: null, shouldProceed: false };
+                        }
+                        case 'submit_prompt': {
+                            localQueryToSendToGemini = slashCommandResult.content;
+                            return {
+                                queryToSend: localQueryToSendToGemini,
+                                shouldProceed: true,
+                            };
+                        }
+                        case 'handled': {
+                            return { queryToSend: null, shouldProceed: false };
+                        }
+                        default: {
+                            const unreachable = slashCommandResult;
+                            throw new Error(`Unhandled slash command result type: ${unreachable}`);
+                        }
+                    }
+                }
+            }
+            if (shellModeActive && handleShellCommand(trimmedQuery, abortSignal)) {
+                return { queryToSend: null, shouldProceed: false };
+            }
+            // Handle @-commands (which might involve tool calls)
+            if (isAtCommand(trimmedQuery)) {
+                const atCommandResult = await handleAtCommand({
+                    query: trimmedQuery,
+                    config,
+                    addItem,
+                    onDebugMessage,
+                    messageId: userMessageTimestamp,
+                    signal: abortSignal,
+                });
+                // Add user's turn after @ command processing is done.
+                addItem({ type: MessageType.USER, text: trimmedQuery }, userMessageTimestamp);
+                if (atCommandResult.error) {
+                    onDebugMessage(atCommandResult.error);
+                    return { queryToSend: null, shouldProceed: false };
+                }
+                localQueryToSendToGemini = atCommandResult.processedQuery;
+            }
+            else {
+                // Normal query for Gemini
+                addItem({ type: MessageType.USER, text: trimmedQuery }, userMessageTimestamp);
+                localQueryToSendToGemini = trimmedQuery;
+            }
+        }
+        else {
+            // It's a function response (PartListUnion that isn't a string)
+            localQueryToSendToGemini = query;
+        }
+        if (localQueryToSendToGemini === null) {
+            onDebugMessage('Query processing resulted in null, not sending to Gemini.');
+            return { queryToSend: null, shouldProceed: false };
+        }
+        return { queryToSend: localQueryToSendToGemini, shouldProceed: true };
+    }, [
+        config,
+        addItem,
+        onDebugMessage,
+        handleShellCommand,
+        handleSlashCommand,
+        logger,
+        shellModeActive,
+        scheduleToolCalls,
+    ]);
+    // --- Stream Event Handlers ---
+    const handleContentEvent = useCallback((eventValue, currentGeminiMessageBuffer, userMessageTimestamp) => {
+        if (turnCancelledRef.current) {
+            // Prevents additional output after a user initiated cancel.
+            return '';
+        }
+        let newGeminiMessageBuffer = currentGeminiMessageBuffer + eventValue;
+        if (pendingHistoryItemRef.current?.type !== 'gemini' &&
+            pendingHistoryItemRef.current?.type !== 'gemini_content') {
+            if (pendingHistoryItemRef.current) {
+                addItem(pendingHistoryItemRef.current, userMessageTimestamp);
+            }
+            setPendingHistoryItem({ type: 'gemini', text: '' });
+            newGeminiMessageBuffer = eventValue;
+        }
+        // Split large messages for better rendering performance. Ideally,
+        // we should maximize the amount of output sent to <Static />.
+        const splitPoint = findLastSafeSplitPoint(newGeminiMessageBuffer);
+        if (splitPoint === newGeminiMessageBuffer.length) {
+            // Update the existing message with accumulated content
+            setPendingHistoryItem((item) => ({
+                type: item?.type,
+                text: newGeminiMessageBuffer,
+            }));
+        }
+        else {
+            // This indicates that we need to split up this Gemini Message.
+            // Splitting a message is primarily a performance consideration. There is a
+            // <Static> component at the root of App.tsx which takes care of rendering
+            // content statically or dynamically. Everything but the last message is
+            // treated as static in order to prevent re-rendering an entire message history
+            // multiple times per-second (as streaming occurs). Prior to this change you'd
+            // see heavy flickering of the terminal. This ensures that larger messages get
+            // broken up so that there are more "statically" rendered.
+            const beforeText = newGeminiMessageBuffer.substring(0, splitPoint);
+            const afterText = newGeminiMessageBuffer.substring(splitPoint);
+            addItem({
+                type: pendingHistoryItemRef.current?.type,
+                text: beforeText,
+            }, userMessageTimestamp);
+            setPendingHistoryItem({ type: 'gemini_content', text: afterText });
+            newGeminiMessageBuffer = afterText;
+        }
+        return newGeminiMessageBuffer;
+    }, [addItem, pendingHistoryItemRef, setPendingHistoryItem]);
+    const handleUserCancelledEvent = useCallback((userMessageTimestamp) => {
+        if (turnCancelledRef.current) {
+            return;
+        }
+        if (pendingHistoryItemRef.current) {
+            if (pendingHistoryItemRef.current.type === 'tool_group') {
+                const updatedTools = pendingHistoryItemRef.current.tools.map((tool) => tool.status === ToolCallStatus.Pending ||
+                    tool.status === ToolCallStatus.Confirming ||
+                    tool.status === ToolCallStatus.Executing
+                    ? { ...tool, status: ToolCallStatus.Canceled }
+                    : tool);
+                const pendingItem = {
+                    ...pendingHistoryItemRef.current,
+                    tools: updatedTools,
+                };
+                addItem(pendingItem, userMessageTimestamp);
+            }
+            else {
+                addItem(pendingHistoryItemRef.current, userMessageTimestamp);
+            }
+            setPendingHistoryItem(null);
+        }
+        addItem({ type: MessageType.INFO, text: 'User cancelled the request.' }, userMessageTimestamp);
+        setIsResponding(false);
+        setThought(null); // Reset thought when user cancels
+    }, [addItem, pendingHistoryItemRef, setPendingHistoryItem, setThought]);
+    const handleErrorEvent = useCallback((eventValue, userMessageTimestamp) => {
+        if (pendingHistoryItemRef.current) {
+            addItem(pendingHistoryItemRef.current, userMessageTimestamp);
+            setPendingHistoryItem(null);
+        }
+        addItem({
+            type: MessageType.ERROR,
+            text: parseAndFormatApiError(eventValue.error, config.getContentGeneratorConfig()?.authType, undefined, config.getModel(), DEFAULT_GEMINI_FLASH_MODEL),
+        }, userMessageTimestamp);
+        setThought(null); // Reset thought when there's an error
+    }, [addItem, pendingHistoryItemRef, setPendingHistoryItem, config, setThought]);
+    const handleCitationEvent = useCallback((text, userMessageTimestamp) => {
+        if (!showCitations(settings, config)) {
+            return;
+        }
+        if (pendingHistoryItemRef.current) {
+            addItem(pendingHistoryItemRef.current, userMessageTimestamp);
+            setPendingHistoryItem(null);
+        }
+        addItem({ type: MessageType.INFO, text }, userMessageTimestamp);
+    }, [addItem, pendingHistoryItemRef, setPendingHistoryItem, settings, config]);
+    const handleFinishedEvent = useCallback((event, userMessageTimestamp) => {
+        const finishReason = event.value.reason;
+        if (!finishReason) {
+            return;
+        }
+        const finishReasonMessages = {
+            [FinishReason.FINISH_REASON_UNSPECIFIED]: undefined,
+            [FinishReason.STOP]: undefined,
+            [FinishReason.MAX_TOKENS]: 'Response truncated due to token limits.',
+            [FinishReason.SAFETY]: 'Response stopped due to safety reasons.',
+            [FinishReason.RECITATION]: 'Response stopped due to recitation policy.',
+            [FinishReason.LANGUAGE]: 'Response stopped due to unsupported language.',
+            [FinishReason.BLOCKLIST]: 'Response stopped due to forbidden terms.',
+            [FinishReason.PROHIBITED_CONTENT]: 'Response stopped due to prohibited content.',
+            [FinishReason.SPII]: 'Response stopped due to sensitive personally identifiable information.',
+            [FinishReason.OTHER]: 'Response stopped for other reasons.',
+            [FinishReason.MALFORMED_FUNCTION_CALL]: 'Response stopped due to malformed function call.',
+            [FinishReason.IMAGE_SAFETY]: 'Response stopped due to image safety violations.',
+            [FinishReason.UNEXPECTED_TOOL_CALL]: 'Response stopped due to unexpected tool call.',
+            [FinishReason.IMAGE_PROHIBITED_CONTENT]: 'Response stopped due to prohibited image content.',
+            [FinishReason.NO_IMAGE]: 'Response stopped because no image was generated.',
+        };
+        const message = finishReasonMessages[finishReason];
+        if (message) {
+            addItem({
+                type: 'info',
+                text: `⚠️  ${message}`,
+            }, userMessageTimestamp);
+        }
+    }, [addItem]);
+    const handleChatCompressionEvent = useCallback((eventValue, userMessageTimestamp) => {
+        if (pendingHistoryItemRef.current) {
+            addItem(pendingHistoryItemRef.current, userMessageTimestamp);
+            setPendingHistoryItem(null);
+        }
+        return addItem({
+            type: 'info',
+            text: `IMPORTANT: This conversation exceeded the compress threshold. ` +
+                `A compressed context will be sent for future messages (compressed from: ` +
+                `${eventValue?.originalTokenCount ?? 'unknown'} to ` +
+                `${eventValue?.newTokenCount ?? 'unknown'} tokens).`,
+        }, Date.now());
+    }, [addItem, pendingHistoryItemRef, setPendingHistoryItem]);
+    const handleMaxSessionTurnsEvent = useCallback(() => addItem({
+        type: 'info',
+        text: `The session has reached the maximum number of turns: ${config.getMaxSessionTurns()}. ` +
+            `Please update this limit in your setting.json file.`,
+    }, Date.now()), [addItem, config]);
+    const handleContextWindowWillOverflowEvent = useCallback((estimatedRequestTokenCount, remainingTokenCount) => {
+        onCancelSubmit(true);
+        const limit = tokenLimit(config.getModel());
+        const isLessThan75Percent = limit > 0 && remainingTokenCount < limit * 0.75;
+        let text = `Sending this message (${estimatedRequestTokenCount} tokens) might exceed the remaining context window limit (${remainingTokenCount} tokens).`;
+        if (isLessThan75Percent) {
+            text +=
+                ' Please try reducing the size of your message or use the `/compress` command to compress the chat history.';
+        }
+        addItem({
+            type: 'info',
+            text,
+        }, Date.now());
+    }, [addItem, onCancelSubmit, config]);
+    const handleChatModelEvent = useCallback((eventValue, userMessageTimestamp) => {
+        if (!settings?.merged?.ui?.showModelInfoInChat) {
+            return;
+        }
+        if (pendingHistoryItemRef.current) {
+            addItem(pendingHistoryItemRef.current, userMessageTimestamp);
+            setPendingHistoryItem(null);
+        }
+        addItem({
+            type: 'model',
+            model: eventValue,
+        }, userMessageTimestamp);
+    }, [addItem, pendingHistoryItemRef, setPendingHistoryItem, settings]);
+    const processGeminiStreamEvents = useCallback(async (stream, userMessageTimestamp, signal) => {
+        let geminiMessageBuffer = '';
+        const toolCallRequests = [];
+        for await (const event of stream) {
+            switch (event.type) {
+                case ServerGeminiEventType.Thought:
+                    setThought(event.value);
+                    break;
+                case ServerGeminiEventType.Content:
+                    geminiMessageBuffer = handleContentEvent(event.value, geminiMessageBuffer, userMessageTimestamp);
+                    break;
+                case ServerGeminiEventType.ToolCallRequest:
+                    toolCallRequests.push(event.value);
+                    await logger?.logEventFull('tool_call', {
+                        callId: event.value.callId,
+                        name: event.value.name,
+                        args: event.value.args,
+                    });
+                    break;
+                case ServerGeminiEventType.UserCancelled:
+                    handleUserCancelledEvent(userMessageTimestamp);
+                    break;
+                case ServerGeminiEventType.Error:
+                    handleErrorEvent(event.value, userMessageTimestamp);
+                    break;
+                case ServerGeminiEventType.ChatCompressed:
+                    handleChatCompressionEvent(event.value, userMessageTimestamp);
+                    break;
+                case ServerGeminiEventType.ToolCallConfirmation:
+                case ServerGeminiEventType.ToolCallResponse:
+                    // do nothing
+                    break;
+                case ServerGeminiEventType.MaxSessionTurns:
+                    handleMaxSessionTurnsEvent();
+                    break;
+                case ServerGeminiEventType.ContextWindowWillOverflow:
+                    handleContextWindowWillOverflowEvent(event.value.estimatedRequestTokenCount, event.value.remainingTokenCount);
+                    break;
+                case ServerGeminiEventType.Finished:
+                    handleFinishedEvent(event, userMessageTimestamp);
+                    break;
+                case ServerGeminiEventType.Citation:
+                    handleCitationEvent(event.value, userMessageTimestamp);
+                    break;
+                case ServerGeminiEventType.ModelInfo:
+                    handleChatModelEvent(event.value, userMessageTimestamp);
+                    break;
+                case ServerGeminiEventType.LoopDetected:
+                    // handle later because we want to move pending history to history
+                    // before we add loop detected message to history
+                    loopDetectedRef.current = true;
+                    break;
+                case ServerGeminiEventType.Retry:
+                case ServerGeminiEventType.InvalidStream:
+                    // Will add the missing logic later
+                    break;
+                default: {
+                    // enforces exhaustive switch-case
+                    const unreachable = event;
+                    return unreachable;
+                }
+            }
+        }
+        if (toolCallRequests.length > 0) {
+            setIsResponding(false);
+            scheduleToolCalls(toolCallRequests, signal);
+            return StreamProcessingStatus.Completed;
+        }
+        await logger?.logEventFull('model_response', {
+            text: geminiMessageBuffer,
+        });
+        setIsResponding(false);
+        return StreamProcessingStatus.Completed;
+    }, [
+        handleContentEvent,
+        handleUserCancelledEvent,
+        handleErrorEvent,
+        scheduleToolCalls,
+        handleChatCompressionEvent,
+        handleFinishedEvent,
+        handleMaxSessionTurnsEvent,
+        handleContextWindowWillOverflowEvent,
+        handleCitationEvent,
+        handleChatModelEvent,
+        logger,
+    ]);
+    const submitQuery = useCallback(async (query, options, prompt_id) => runInDevTraceSpan({ name: 'submitQuery' }, async ({ metadata: spanMetadata }) => {
+        spanMetadata.input = query;
+        const queryId = `${Date.now()}-${Math.random()}`;
+        activeQueryIdRef.current = queryId;
+        if ((streamingState === StreamingState.Responding ||
+            streamingState === StreamingState.WaitingForConfirmation) &&
+            !options?.isContinuation)
+            return;
+        const userMessageTimestamp = Date.now();
+        // Reset quota error flag when starting a new query (not a continuation)
+        if (!options?.isContinuation) {
+            setModelSwitchedFromQuotaError(false);
+            config.setQuotaErrorOccurred(false);
+        }
+        abortControllerRef.current = new AbortController();
+        const abortSignal = abortControllerRef.current.signal;
+        turnCancelledRef.current = false;
+        if (!prompt_id) {
+            prompt_id = config.getSessionId() + '########' + getPromptCount();
+        }
+        return promptIdContext.run(prompt_id, async () => {
+            const { queryToSend, shouldProceed } = await prepareQueryForGemini(query, userMessageTimestamp, abortSignal, prompt_id);
+            if (!shouldProceed || queryToSend === null) {
+                return;
+            }
+            if (!options?.isContinuation) {
+                if (typeof queryToSend === 'string') {
+                    // logging the text prompts only for now
+                    const promptText = queryToSend;
+                    logUserPrompt(config, new UserPromptEvent(promptText.length, prompt_id, config.getContentGeneratorConfig()?.authType, promptText));
+                }
+                startNewPrompt();
+                setThought(null); // Reset thought when starting a new prompt
+            }
+            setIsResponding(true);
+            setInitError(null);
+            // Store query and prompt_id for potential retry on loop detection
+            lastQueryRef.current = queryToSend;
+            lastPromptIdRef.current = prompt_id;
+            try {
+                const stream = geminiClient.sendMessageStream(queryToSend, abortSignal, prompt_id);
+                const processingStatus = await processGeminiStreamEvents(stream, userMessageTimestamp, abortSignal);
+                if (processingStatus === StreamProcessingStatus.UserCancelled) {
+                    return;
+                }
+                if (pendingHistoryItemRef.current) {
+                    addItem(pendingHistoryItemRef.current, userMessageTimestamp);
+                    setPendingHistoryItem(null);
+                }
+                if (loopDetectedRef.current) {
+                    loopDetectedRef.current = false;
+                    // Show the confirmation dialog to choose whether to disable loop detection
+                    setLoopDetectionConfirmationRequest({
+                        onComplete: (result) => {
+                            setLoopDetectionConfirmationRequest(null);
+                            if (result.userSelection === 'disable') {
+                                config
+                                    .getGeminiClient()
+                                    .getLoopDetectionService()
+                                    .disableForSession();
+                                addItem({
+                                    type: 'info',
+                                    text: `Loop detection has been disabled for this session. Retrying request...`,
+                                }, Date.now());
+                                if (lastQueryRef.current && lastPromptIdRef.current) {
+                                    // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                                    submitQuery(lastQueryRef.current, { isContinuation: true }, lastPromptIdRef.current);
+                                }
+                            }
+                            else {
+                                addItem({
+                                    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.`,
+                                }, Date.now());
+                            }
+                        },
+                    });
+                }
+            }
+            catch (error) {
+                spanMetadata.error = error;
+                if (error instanceof UnauthorizedError) {
+                    onAuthError('Session expired or is unauthorized.');
+                }
+                else if (!isNodeError(error) || error.name !== 'AbortError') {
+                    addItem({
+                        type: MessageType.ERROR,
+                        text: parseAndFormatApiError(getErrorMessage(error) || 'Unknown error', config.getContentGeneratorConfig()?.authType, undefined, config.getModel(), DEFAULT_GEMINI_FLASH_MODEL),
+                    }, userMessageTimestamp);
+                }
+            }
+            finally {
+                if (activeQueryIdRef.current === queryId) {
+                    setIsResponding(false);
+                }
+            }
+        });
+    }), [
+        streamingState,
+        setModelSwitchedFromQuotaError,
+        prepareQueryForGemini,
+        processGeminiStreamEvents,
+        pendingHistoryItemRef,
+        addItem,
+        setPendingHistoryItem,
+        setInitError,
+        geminiClient,
+        onAuthError,
+        config,
+        startNewPrompt,
+        getPromptCount,
+    ]);
+    const handleApprovalModeChange = useCallback(async (newApprovalMode) => {
+        // Auto-approve pending tool calls when switching to auto-approval modes
+        if (newApprovalMode === ApprovalMode.YOLO ||
+            newApprovalMode === ApprovalMode.AUTO_EDIT) {
+            let awaitingApprovalCalls = toolCalls.filter((call) => call.status === 'awaiting_approval');
+            // For AUTO_EDIT mode, only approve edit tools (replace, write_file)
+            if (newApprovalMode === ApprovalMode.AUTO_EDIT) {
+                awaitingApprovalCalls = awaitingApprovalCalls.filter((call) => EDIT_TOOL_NAMES.has(call.request.name));
+            }
+            // Process pending tool calls sequentially to reduce UI chaos
+            for (const call of awaitingApprovalCalls) {
+                if (call.confirmationDetails?.onConfirm) {
+                    try {
+                        await call.confirmationDetails.onConfirm(ToolConfirmationOutcome.ProceedOnce);
+                    }
+                    catch (error) {
+                        debugLogger.warn(`Failed to auto-approve tool call ${call.request.callId}:`, error);
+                    }
+                }
+            }
+        }
+    }, [toolCalls]);
+    const handleCompletedTools = useCallback(async (completedToolCallsFromScheduler) => {
+        const completedAndReadyToSubmitTools = completedToolCallsFromScheduler.filter((tc) => {
+            const isTerminalState = tc.status === 'success' ||
+                tc.status === 'error' ||
+                tc.status === 'cancelled';
+            if (isTerminalState) {
+                const completedOrCancelledCall = tc;
+                return (completedOrCancelledCall.response?.responseParts !== undefined);
+            }
+            return false;
+        });
+        // Finalize any client-initiated tools as soon as they are done.
+        const clientTools = completedAndReadyToSubmitTools.filter((t) => t.request.isClientInitiated);
+        if (clientTools.length > 0) {
+            markToolsAsSubmitted(clientTools.map((t) => t.request.callId));
+        }
+        // Identify new, successful save_memory calls that we haven't processed yet.
+        const newSuccessfulMemorySaves = completedAndReadyToSubmitTools.filter((t) => t.request.name === 'save_memory' &&
+            t.status === 'success' &&
+            !processedMemoryToolsRef.current.has(t.request.callId));
+        if (newSuccessfulMemorySaves.length > 0) {
+            // Perform the refresh only if there are new ones.
+            void performMemoryRefresh();
+            // Mark them as processed so we don't do this again on the next render.
+            newSuccessfulMemorySaves.forEach((t) => processedMemoryToolsRef.current.add(t.request.callId));
+        }
+        const geminiTools = completedAndReadyToSubmitTools.filter((t) => !t.request.isClientInitiated);
+        if (geminiTools.length === 0) {
+            return;
+        }
+        // If all the tools were cancelled, don't submit a response to Gemini.
+        const allToolsCancelled = geminiTools.every((tc) => tc.status === 'cancelled');
+        if (allToolsCancelled) {
+            // If the turn was cancelled via the imperative escape key flow,
+            // the cancellation message is added there. We check the ref to avoid duplication.
+            if (!turnCancelledRef.current) {
+                addItem({
+                    type: MessageType.INFO,
+                    text: 'Request cancelled.',
+                }, Date.now());
+            }
+            setIsResponding(false);
+            if (geminiClient) {
+                // We need to manually add the function responses to the history
+                // so the model knows the tools were cancelled.
+                const combinedParts = geminiTools.flatMap((toolCall) => toolCall.response.responseParts);
+                // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                geminiClient.addHistory({
+                    role: 'user',
+                    parts: combinedParts,
+                });
+            }
+            const callIdsToMarkAsSubmitted = geminiTools.map((toolCall) => toolCall.request.callId);
+            markToolsAsSubmitted(callIdsToMarkAsSubmitted);
+            return;
+        }
+        const responsesToSend = geminiTools.flatMap((toolCall) => toolCall.response.responseParts);
+        const callIdsToMarkAsSubmitted = geminiTools.map((toolCall) => toolCall.request.callId);
+        const prompt_ids = geminiTools.map((toolCall) => toolCall.request.prompt_id);
+        markToolsAsSubmitted(callIdsToMarkAsSubmitted);
+        // Don't continue if model was switched due to quota error
+        if (modelSwitchedFromQuotaError) {
+            return;
+        }
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        submitQuery(responsesToSend, {
+            isContinuation: true,
+        }, prompt_ids[0]);
+    }, [
+        submitQuery,
+        markToolsAsSubmitted,
+        geminiClient,
+        performMemoryRefresh,
+        modelSwitchedFromQuotaError,
+        addItem,
+    ]);
+    const pendingHistoryItems = useMemo(() => [pendingHistoryItem, pendingToolCallGroupDisplay].filter((i) => i !== undefined && i !== null), [pendingHistoryItem, pendingToolCallGroupDisplay]);
+    useEffect(() => {
+        const saveRestorableToolCalls = async () => {
+            if (!config.getCheckpointingEnabled()) {
+                return;
+            }
+            const restorableToolCalls = toolCalls.filter((toolCall) => EDIT_TOOL_NAMES.has(toolCall.request.name) &&
+                toolCall.status === 'awaiting_approval');
+            if (restorableToolCalls.length > 0) {
+                if (!gitService) {
+                    onDebugMessage('Checkpointing is enabled but Git service is not available. Failed to create snapshot. Ensure Git is installed and working properly.');
+                    return;
+                }
+                const { checkpointsToWrite, errors } = await processRestorableToolCalls(restorableToolCalls.map((call) => call.request), gitService, geminiClient, history);
+                if (errors.length > 0) {
+                    errors.forEach(onDebugMessage);
+                }
+                if (checkpointsToWrite.size > 0) {
+                    const checkpointDir = storage.getProjectTempCheckpointsDir();
+                    try {
+                        await fs.mkdir(checkpointDir, { recursive: true });
+                        for (const [fileName, content] of checkpointsToWrite) {
+                            const filePath = path.join(checkpointDir, fileName);
+                            await fs.writeFile(filePath, content);
+                        }
+                    }
+                    catch (error) {
+                        onDebugMessage(`Failed to write checkpoint file: ${getErrorMessage(error)}`);
+                    }
+                }
+            }
+        };
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        saveRestorableToolCalls();
+    }, [
+        toolCalls,
+        config,
+        onDebugMessage,
+        gitService,
+        history,
+        geminiClient,
+        storage,
+    ]);
+    const lastOutputTime = Math.max(lastToolOutputTime, lastShellOutputTime);
+    return {
+        streamingState,
+        submitQuery,
+        initError,
+        pendingHistoryItems,
+        thought,
+        cancelOngoingRequest,
+        pendingToolCalls: toolCalls,
+        handleApprovalModeChange,
+        activePtyId,
+        loopDetectionConfirmationRequest,
+        lastOutputTime,
+        interactivePasswordPrompt,
+        isFullScreen,
+        clearInteractivePasswordPrompt,
+    };
+};
+//# sourceMappingURL=useGeminiStream.js.map