@zgsm/costrict-cli 0.0.1

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

@@ -0,0 +1,171 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useState, useCallback, useRef, useEffect, useMemo } from 'react';
+import { debugLogger, getResponseText } from '@google/gemini-cli-core';
+import { isSlashCommand } from '../utils/commandUtils.js';
+export const PROMPT_COMPLETION_MIN_LENGTH = 5;
+export const PROMPT_COMPLETION_DEBOUNCE_MS = 250;
+export function usePromptCompletion({ buffer, config, enabled, }) {
+    const [ghostText, setGhostText] = useState('');
+    const [isLoadingGhostText, setIsLoadingGhostText] = useState(false);
+    const abortControllerRef = useRef(null);
+    const [justSelectedSuggestion, setJustSelectedSuggestion] = useState(false);
+    const lastSelectedTextRef = useRef('');
+    const lastRequestedTextRef = useRef('');
+    const isPromptCompletionEnabled = enabled && (config?.getEnablePromptCompletion() ?? false);
+    const clearGhostText = useCallback(() => {
+        setGhostText('');
+        setIsLoadingGhostText(false);
+    }, []);
+    const acceptGhostText = useCallback(() => {
+        if (ghostText && ghostText.length > buffer.text.length) {
+            buffer.setText(ghostText);
+            setGhostText('');
+            setJustSelectedSuggestion(true);
+            lastSelectedTextRef.current = ghostText;
+        }
+    }, [ghostText, buffer]);
+    const markSuggestionSelected = useCallback((selectedText) => {
+        setJustSelectedSuggestion(true);
+        lastSelectedTextRef.current = selectedText;
+    }, []);
+    const generatePromptSuggestions = useCallback(async () => {
+        const trimmedText = buffer.text.trim();
+        const geminiClient = config?.getGeminiClient();
+        if (trimmedText === lastRequestedTextRef.current) {
+            return;
+        }
+        if (abortControllerRef.current) {
+            abortControllerRef.current.abort();
+        }
+        if (trimmedText.length < PROMPT_COMPLETION_MIN_LENGTH ||
+            !geminiClient ||
+            isSlashCommand(trimmedText) ||
+            trimmedText.includes('@') ||
+            !isPromptCompletionEnabled) {
+            clearGhostText();
+            lastRequestedTextRef.current = '';
+            return;
+        }
+        lastRequestedTextRef.current = trimmedText;
+        setIsLoadingGhostText(true);
+        abortControllerRef.current = new AbortController();
+        const signal = abortControllerRef.current.signal;
+        try {
+            const contents = [
+                {
+                    role: 'user',
+                    parts: [
+                        {
+                            text: `You are a professional prompt engineering assistant. Complete the user's partial prompt with expert precision and clarity. User's input: "${trimmedText}" Continue this prompt by adding specific, actionable details that align with the user's intent. Focus on: clear, precise language; structured requirements; professional terminology; measurable outcomes. Length Guidelines: Keep suggestions concise (ideally 10-20 characters); prioritize brevity while maintaining clarity; use essential keywords only; avoid redundant phrases. Start your response with the exact user text ("${trimmedText}") followed by your completion. Provide practical, implementation-focused suggestions rather than creative interpretations. Format: Plain text only. Single completion. Match the user's language. Emphasize conciseness over elaboration.`,
+                        },
+                    ],
+                },
+            ];
+            const response = await geminiClient.generateContent({ model: 'prompt-completion' }, contents, signal);
+            if (signal.aborted) {
+                return;
+            }
+            if (response) {
+                const responseText = getResponseText(response);
+                if (responseText) {
+                    const suggestionText = responseText.trim();
+                    if (suggestionText.length > 0 &&
+                        suggestionText.startsWith(trimmedText)) {
+                        setGhostText(suggestionText);
+                    }
+                    else {
+                        clearGhostText();
+                    }
+                }
+            }
+        }
+        catch (error) {
+            if (!(signal.aborted ||
+                (error instanceof Error && error.name === 'AbortError'))) {
+                debugLogger.warn(`[WARN] prompt completion failed: : (${error instanceof Error ? error.message : String(error)})`);
+                // Clear the last requested text to allow retry only on real errors
+                lastRequestedTextRef.current = '';
+            }
+            clearGhostText();
+        }
+        finally {
+            if (!signal.aborted) {
+                setIsLoadingGhostText(false);
+            }
+        }
+    }, [buffer.text, config, clearGhostText, isPromptCompletionEnabled]);
+    const isCursorAtEnd = useCallback(() => {
+        const [cursorRow, cursorCol] = buffer.cursor;
+        const totalLines = buffer.lines.length;
+        if (cursorRow !== totalLines - 1) {
+            return false;
+        }
+        const lastLine = buffer.lines[cursorRow] || '';
+        return cursorCol === lastLine.length;
+    }, [buffer.cursor, buffer.lines]);
+    const handlePromptCompletion = useCallback(() => {
+        if (!isCursorAtEnd()) {
+            clearGhostText();
+            return;
+        }
+        const trimmedText = buffer.text.trim();
+        if (justSelectedSuggestion && trimmedText === lastSelectedTextRef.current) {
+            return;
+        }
+        if (trimmedText !== lastSelectedTextRef.current) {
+            setJustSelectedSuggestion(false);
+            lastSelectedTextRef.current = '';
+        }
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        generatePromptSuggestions();
+    }, [
+        buffer.text,
+        generatePromptSuggestions,
+        justSelectedSuggestion,
+        isCursorAtEnd,
+        clearGhostText,
+    ]);
+    // Debounce prompt completion
+    useEffect(() => {
+        const timeoutId = setTimeout(handlePromptCompletion, PROMPT_COMPLETION_DEBOUNCE_MS);
+        return () => clearTimeout(timeoutId);
+    }, [buffer.text, buffer.cursor, handlePromptCompletion]);
+    // Ghost text validation - clear if it doesn't match current text or cursor not at end
+    useEffect(() => {
+        const currentText = buffer.text.trim();
+        if (ghostText && !isCursorAtEnd()) {
+            clearGhostText();
+            return;
+        }
+        if (ghostText &&
+            currentText.length > 0 &&
+            !ghostText.startsWith(currentText)) {
+            clearGhostText();
+        }
+    }, [buffer.text, buffer.cursor, ghostText, clearGhostText, isCursorAtEnd]);
+    // Cleanup on unmount
+    useEffect(() => () => abortControllerRef.current?.abort(), []);
+    const isActive = useMemo(() => {
+        if (!isPromptCompletionEnabled)
+            return false;
+        if (!isCursorAtEnd())
+            return false;
+        const trimmedText = buffer.text.trim();
+        return (trimmedText.length >= PROMPT_COMPLETION_MIN_LENGTH &&
+            !isSlashCommand(trimmedText) &&
+            !trimmedText.includes('@'));
+    }, [buffer.text, isPromptCompletionEnabled, isCursorAtEnd]);
+    return {
+        text: ghostText,
+        isLoading: isLoadingGhostText,
+        isActive,
+        accept: acceptGhostText,
+        clear: clearGhostText,
+        markSelected: markSuggestionSelected,
+    };
+}
+//# sourceMappingURL=usePromptCompletion.js.map

package/dist/src/ui/hooks/usePromptCompletion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePromptCompletion.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePromptCompletion.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAE1E,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAGvE,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,CAAC;AAC9C,MAAM,CAAC,MAAM,6BAA6B,GAAG,GAAG,CAAC;AAiBjD,MAAM,UAAU,mBAAmB,CAAC,EAClC,MAAM,EACN,MAAM,EACN,OAAO,GACoB;IAC3B,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAS,EAAE,CAAC,CAAC;IACvD,MAAM,CAAC,kBAAkB,EAAE,qBAAqB,CAAC,GAAG,QAAQ,CAAU,KAAK,CAAC,CAAC;IAC7E,MAAM,kBAAkB,GAAG,MAAM,CAAyB,IAAI,CAAC,CAAC;IAChE,MAAM,CAAC,sBAAsB,EAAE,yBAAyB,CAAC,GACvD,QAAQ,CAAU,KAAK,CAAC,CAAC;IAC3B,MAAM,mBAAmB,GAAG,MAAM,CAAS,EAAE,CAAC,CAAC;IAC/C,MAAM,oBAAoB,GAAG,MAAM,CAAS,EAAE,CAAC,CAAC;IAEhD,MAAM,yBAAyB,GAC7B,OAAO,IAAI,CAAC,MAAM,EAAE,yBAAyB,EAAE,IAAI,KAAK,CAAC,CAAC;IAE5D,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;QACtC,YAAY,CAAC,EAAE,CAAC,CAAC;QACjB,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;QACvC,IAAI,SAAS,IAAI,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC1B,YAAY,CAAC,EAAE,CAAC,CAAC;YACjB,yBAAyB,CAAC,IAAI,CAAC,CAAC;YAChC,mBAAmB,CAAC,OAAO,GAAG,SAAS,CAAC;QAC1C,CAAC;IACH,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;IAExB,MAAM,sBAAsB,GAAG,WAAW,CAAC,CAAC,YAAoB,EAAE,EAAE;QAClE,yBAAyB,CAAC,IAAI,CAAC,CAAC;QAChC,mBAAmB,CAAC,OAAO,GAAG,YAAY,CAAC;IAC7C,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,yBAAyB,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;QACvD,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,YAAY,GAAG,MAAM,EAAE,eAAe,EAAE,CAAC;QAE/C,IAAI,WAAW,KAAK,oBAAoB,CAAC,OAAO,EAAE,CAAC;YACjD,OAAO;QACT,CAAC;QAED,IAAI,kBAAkB,CAAC,OAAO,EAAE,CAAC;YAC/B,kBAAkB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrC,CAAC;QAED,IACE,WAAW,CAAC,MAAM,GAAG,4BAA4B;YACjD,CAAC,YAAY;YACb,cAAc,CAAC,WAAW,CAAC;YAC3B,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC;YACzB,CAAC,yBAAyB,EAC1B,CAAC;YACD,cAAc,EAAE,CAAC;YACjB,oBAAoB,CAAC,OAAO,GAAG,EAAE,CAAC;YAClC,OAAO;QACT,CAAC;QAED,oBAAoB,CAAC,OAAO,GAAG,WAAW,CAAC;QAC3C,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAE5B,kBAAkB,CAAC,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE;wBACL;4BACE,IAAI,EAAE,6IAA6I,WAAW,0aAA0a,WAAW,4OAA4O;yBACh0B;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,eAAe,CACjD,EAAE,KAAK,EAAE,mBAAmB,EAAE,EAC9B,QAAQ,EACR,MAAM,CACP,CAAC;YAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;gBAE/C,IAAI,YAAY,EAAE,CAAC;oBACjB,MAAM,cAAc,GAAG,YAAY,CAAC,IAAI,EAAE,CAAC;oBAE3C,IACE,cAAc,CAAC,MAAM,GAAG,CAAC;wBACzB,cAAc,CAAC,UAAU,CAAC,WAAW,CAAC,EACtC,CAAC;wBACD,YAAY,CAAC,cAAc,CAAC,CAAC;oBAC/B,CAAC;yBAAM,CAAC;wBACN,cAAc,EAAE,CAAC;oBACnB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IACE,CAAC,CACC,MAAM,CAAC,OAAO;gBACd,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CACxD,EACD,CAAC;gBACD,WAAW,CAAC,IAAI,CACd,uCAAuC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CACjG,CAAC;gBACF,mEAAmE;gBACnE,oBAAoB,CAAC,OAAO,GAAG,EAAE,CAAC;YACpC,CAAC;YACD,cAAc,EAAE,CAAC;QACnB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;IACH,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,cAAc,EAAE,yBAAyB,CAAC,CAAC,CAAC;IAErE,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE;QACrC,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;QAC7C,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC;QACvC,IAAI,SAAS,KAAK,UAAU,GAAG,CAAC,EAAE,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QAC/C,OAAO,SAAS,KAAK,QAAQ,CAAC,MAAM,CAAC;IACvC,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAElC,MAAM,sBAAsB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC9C,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACrB,cAAc,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAEvC,IAAI,sBAAsB,IAAI,WAAW,KAAK,mBAAmB,CAAC,OAAO,EAAE,CAAC;YAC1E,OAAO;QACT,CAAC;QAED,IAAI,WAAW,KAAK,mBAAmB,CAAC,OAAO,EAAE,CAAC;YAChD,yBAAyB,CAAC,KAAK,CAAC,CAAC;YACjC,mBAAmB,CAAC,OAAO,GAAG,EAAE,CAAC;QACnC,CAAC;QAED,mEAAmE;QACnE,yBAAyB,EAAE,CAAC;IAC9B,CAAC,EAAE;QACD,MAAM,CAAC,IAAI;QACX,yBAAyB;QACzB,sBAAsB;QACtB,aAAa;QACb,cAAc;KACf,CAAC,CAAC;IAEH,6BAA6B;IAC7B,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,SAAS,GAAG,UAAU,CAC1B,sBAAsB,EACtB,6BAA6B,CAC9B,CAAC;QACF,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,sBAAsB,CAAC,CAAC,CAAC;IAEzD,sFAAsF;IACtF,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAEvC,IAAI,SAAS,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YAClC,cAAc,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,IACE,SAAS;YACT,WAAW,CAAC,MAAM,GAAG,CAAC;YACtB,CAAC,SAAS,CAAC,UAAU,CAAC,WAAW,CAAC,EAClC,CAAC;YACD,cAAc,EAAE,CAAC;QACnB,CAAC;IACH,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,aAAa,CAAC,CAAC,CAAC;IAE3E,qBAAqB;IACrB,SAAS,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;IAE/D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE;QAC5B,IAAI,CAAC,yBAAyB;YAAE,OAAO,KAAK,CAAC;QAE7C,IAAI,CAAC,aAAa,EAAE;YAAE,OAAO,KAAK,CAAC;QAEnC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACvC,OAAO,CACL,WAAW,CAAC,MAAM,IAAI,4BAA4B;YAClD,CAAC,cAAc,CAAC,WAAW,CAAC;YAC5B,CAAC,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAC3B,CAAC;IACJ,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,yBAAyB,EAAE,aAAa,CAAC,CAAC,CAAC;IAE5D,OAAO;QACL,IAAI,EAAE,SAAS;QACf,SAAS,EAAE,kBAAkB;QAC7B,QAAQ;QACR,MAAM,EAAE,eAAe;QACvB,KAAK,EAAE,cAAc;QACrB,YAAY,EAAE,sBAAsB;KACrC,CAAC;AACJ,CAAC"}

package/dist/src/ui/hooks/useQuotaAndFallback.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type Config, type FallbackIntent, type UserTierId } from '@google/gemini-cli-core';
+import { type UseHistoryManagerReturn } from './useHistoryManager.js';
+import { type ProQuotaDialogRequest } from '../contexts/UIStateContext.js';
+interface UseQuotaAndFallbackArgs {
+    config: Config;
+    historyManager: UseHistoryManagerReturn;
+    userTier: UserTierId | undefined;
+    setModelSwitchedFromQuotaError: (value: boolean) => void;
+}
+export declare function useQuotaAndFallback({ config, historyManager, userTier, setModelSwitchedFromQuotaError, }: UseQuotaAndFallbackArgs): {
+    proQuotaRequest: ProQuotaDialogRequest | null;
+    handleProQuotaChoice: (choice: FallbackIntent) => void;
+};
+export {};

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

@@ -0,0 +1,112 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType, TerminalQuotaError, ModelNotFoundError, PREVIEW_GEMINI_MODEL, DEFAULT_GEMINI_MODEL, } from '@google/gemini-cli-core';
+import { useCallback, useEffect, useRef, useState } from 'react';
+import {} from './useHistoryManager.js';
+import { MessageType } from '../types.js';
+import {} from '../contexts/UIStateContext.js';
+export function useQuotaAndFallback({ config, historyManager, userTier, setModelSwitchedFromQuotaError, }) {
+    const [proQuotaRequest, setProQuotaRequest] = useState(null);
+    const isDialogPending = useRef(false);
+    // Set up Flash fallback handler
+    useEffect(() => {
+        const fallbackHandler = async (failedModel, fallbackModel, error) => {
+            // Fallbacks are currently only handled for OAuth users.
+            const contentGeneratorConfig = config.getContentGeneratorConfig();
+            if (!contentGeneratorConfig ||
+                contentGeneratorConfig.authType !== AuthType.LOGIN_WITH_GOOGLE) {
+                return null;
+            }
+            let message;
+            let isTerminalQuotaError = false;
+            let isModelNotFoundError = false;
+            const usageLimitReachedModel = failedModel === DEFAULT_GEMINI_MODEL ||
+                failedModel === PREVIEW_GEMINI_MODEL
+                ? 'all Pro models'
+                : failedModel;
+            if (error instanceof TerminalQuotaError) {
+                isTerminalQuotaError = true;
+                // Common part of the message for both tiers
+                const messageLines = [
+                    `Usage limit reached for ${usageLimitReachedModel}.`,
+                    error.retryDelayMs ? getResetTimeMessage(error.retryDelayMs) : null,
+                    `/stats for usage details`,
+                    `/auth to switch to API key.`,
+                ].filter(Boolean);
+                message = messageLines.join('\n');
+            }
+            else if (error instanceof ModelNotFoundError) {
+                isModelNotFoundError = true;
+                const messageLines = [
+                    `It seems like you don't have access to Gemini 3.`,
+                    `Learn more at https://goo.gle/enable-preview-features`,
+                    `To disable Gemini 3, disable "Preview features" in /settings.`,
+                ];
+                message = messageLines.join('\n');
+            }
+            else {
+                message = `${failedModel} is currently experiencing high demand. We apologize and appreciate your patience.`;
+            }
+            setModelSwitchedFromQuotaError(true);
+            config.setQuotaErrorOccurred(true);
+            if (isDialogPending.current) {
+                return 'stop'; // A dialog is already active, so just stop this request.
+            }
+            isDialogPending.current = true;
+            const intent = await new Promise((resolve) => {
+                setProQuotaRequest({
+                    failedModel,
+                    fallbackModel,
+                    resolve,
+                    message,
+                    isTerminalQuotaError,
+                    isModelNotFoundError,
+                });
+            });
+            return intent;
+        };
+        config.setFallbackModelHandler(fallbackHandler);
+    }, [config, historyManager, userTier, setModelSwitchedFromQuotaError]);
+    const handleProQuotaChoice = useCallback((choice) => {
+        if (!proQuotaRequest)
+            return;
+        const intent = choice;
+        proQuotaRequest.resolve(intent);
+        setProQuotaRequest(null);
+        isDialogPending.current = false; // Reset the flag here
+        if (choice === 'retry_always') {
+            // If we were recovering from a Preview Model failure, show a specific message.
+            if (proQuotaRequest.failedModel === PREVIEW_GEMINI_MODEL) {
+                const showPeriodicalCheckMessage = !proQuotaRequest.isModelNotFoundError &&
+                    proQuotaRequest.fallbackModel === DEFAULT_GEMINI_MODEL;
+                historyManager.addItem({
+                    type: MessageType.INFO,
+                    text: `Switched to fallback model ${proQuotaRequest.fallbackModel}. ${showPeriodicalCheckMessage ? `We will periodically check if ${PREVIEW_GEMINI_MODEL} is available again.` : ''}`,
+                }, Date.now());
+            }
+            else {
+                historyManager.addItem({
+                    type: MessageType.INFO,
+                    text: 'Switched to fallback model.',
+                }, Date.now());
+            }
+        }
+    }, [proQuotaRequest, historyManager]);
+    return {
+        proQuotaRequest,
+        handleProQuotaChoice,
+    };
+}
+function getResetTimeMessage(delayMs) {
+    const resetDate = new Date(Date.now() + delayMs);
+    const timeFormatter = new Intl.DateTimeFormat('en-US', {
+        hour: 'numeric',
+        minute: '2-digit',
+        timeZoneName: 'short',
+    });
+    return `Access resets at ${timeFormatter.format(resetDate)}.`;
+}
+//# sourceMappingURL=useQuotaAndFallback.js.map

package/dist/src/ui/hooks/useQuotaAndFallback.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useQuotaAndFallback.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useQuotaAndFallback.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EAIR,kBAAkB,EAClB,kBAAkB,EAElB,oBAAoB,EACpB,oBAAoB,GACrB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjE,OAAO,EAAgC,MAAM,wBAAwB,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAA8B,MAAM,+BAA+B,CAAC;AAS3E,MAAM,UAAU,mBAAmB,CAAC,EAClC,MAAM,EACN,cAAc,EACd,QAAQ,EACR,8BAA8B,GACN;IACxB,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GACzC,QAAQ,CAA+B,IAAI,CAAC,CAAC;IAC/C,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAEtC,gCAAgC;IAChC,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,eAAe,GAAyB,KAAK,EACjD,WAAW,EACX,aAAa,EACb,KAAK,EAC2B,EAAE;YAClC,wDAAwD;YACxD,MAAM,sBAAsB,GAAG,MAAM,CAAC,yBAAyB,EAAE,CAAC;YAClE,IACE,CAAC,sBAAsB;gBACvB,sBAAsB,CAAC,QAAQ,KAAK,QAAQ,CAAC,iBAAiB,EAC9D,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YAED,IAAI,OAAe,CAAC;YACpB,IAAI,oBAAoB,GAAG,KAAK,CAAC;YACjC,IAAI,oBAAoB,GAAG,KAAK,CAAC;YACjC,MAAM,sBAAsB,GAC1B,WAAW,KAAK,oBAAoB;gBACpC,WAAW,KAAK,oBAAoB;gBAClC,CAAC,CAAC,gBAAgB;gBAClB,CAAC,CAAC,WAAW,CAAC;YAClB,IAAI,KAAK,YAAY,kBAAkB,EAAE,CAAC;gBACxC,oBAAoB,GAAG,IAAI,CAAC;gBAC5B,4CAA4C;gBAC5C,MAAM,YAAY,GAAG;oBACnB,2BAA2B,sBAAsB,GAAG;oBACpD,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,mBAAmB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI;oBACnE,0BAA0B;oBAC1B,6BAA6B;iBAC9B,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpC,CAAC;iBAAM,IAAI,KAAK,YAAY,kBAAkB,EAAE,CAAC;gBAC/C,oBAAoB,GAAG,IAAI,CAAC;gBAC5B,MAAM,YAAY,GAAG;oBACnB,kDAAkD;oBAClD,uDAAuD;oBACvD,+DAA+D;iBAChE,CAAC;gBACF,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,GAAG,WAAW,oFAAoF,CAAC;YAC/G,CAAC;YAED,8BAA8B,CAAC,IAAI,CAAC,CAAC;YACrC,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;YAEnC,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;gBAC5B,OAAO,MAAM,CAAC,CAAC,yDAAyD;YAC1E,CAAC;YACD,eAAe,CAAC,OAAO,GAAG,IAAI,CAAC;YAE/B,MAAM,MAAM,GAAmB,MAAM,IAAI,OAAO,CAC9C,CAAC,OAAO,EAAE,EAAE;gBACV,kBAAkB,CAAC;oBACjB,WAAW;oBACX,aAAa;oBACb,OAAO;oBACP,OAAO;oBACP,oBAAoB;oBACpB,oBAAoB;iBACrB,CAAC,CAAC;YACL,CAAC,CACF,CAAC;YAEF,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QAEF,MAAM,CAAC,uBAAuB,CAAC,eAAe,CAAC,CAAC;IAClD,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,8BAA8B,CAAC,CAAC,CAAC;IAEvE,MAAM,oBAAoB,GAAG,WAAW,CACtC,CAAC,MAAsB,EAAE,EAAE;QACzB,IAAI,CAAC,eAAe;YAAE,OAAO;QAE7B,MAAM,MAAM,GAAmB,MAAM,CAAC;QACtC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACzB,eAAe,CAAC,OAAO,GAAG,KAAK,CAAC,CAAC,sBAAsB;QAEvD,IAAI,MAAM,KAAK,cAAc,EAAE,CAAC;YAC9B,+EAA+E;YAC/E,IAAI,eAAe,CAAC,WAAW,KAAK,oBAAoB,EAAE,CAAC;gBACzD,MAAM,0BAA0B,GAC9B,CAAC,eAAe,CAAC,oBAAoB;oBACrC,eAAe,CAAC,aAAa,KAAK,oBAAoB,CAAC;gBACzD,cAAc,CAAC,OAAO,CACpB;oBACE,IAAI,EAAE,WAAW,CAAC,IAAI;oBACtB,IAAI,EAAE,8BAA8B,eAAe,CAAC,aAAa,KAAK,0BAA0B,CAAC,CAAC,CAAC,iCAAiC,oBAAoB,sBAAsB,CAAC,CAAC,CAAC,EAAE,EAAE;iBACtL,EACD,IAAI,CAAC,GAAG,EAAE,CACX,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,cAAc,CAAC,OAAO,CACpB;oBACE,IAAI,EAAE,WAAW,CAAC,IAAI;oBACtB,IAAI,EAAE,6BAA6B;iBACpC,EACD,IAAI,CAAC,GAAG,EAAE,CACX,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC,EACD,CAAC,eAAe,EAAE,cAAc,CAAC,CAClC,CAAC;IAEF,OAAO;QACL,eAAe;QACf,oBAAoB;KACrB,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAe;IAC1C,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,CAAC;IAEjD,MAAM,aAAa,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE;QACrD,IAAI,EAAE,SAAS;QACf,MAAM,EAAE,SAAS;QACjB,YAAY,EAAE,OAAO;KACtB,CAAC,CAAC;IAEH,OAAO,oBAAoB,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC;AAChE,CAAC"}

package/dist/src/ui/hooks/useQuotaAndFallback.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

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

@@ -0,0 +1,332 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach, } from 'vitest';
+import { act } from 'react';
+import { renderHook } from '../../test-utils/render.js';
+import { UserTierId, AuthType, TerminalQuotaError, makeFakeConfig, RetryableQuotaError, PREVIEW_GEMINI_MODEL, ModelNotFoundError, DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, } from '@google/gemini-cli-core';
+import { useQuotaAndFallback } from './useQuotaAndFallback.js';
+import { MessageType } from '../types.js';
+describe('useQuotaAndFallback', () => {
+    let mockConfig;
+    let mockHistoryManager;
+    let mockSetModelSwitchedFromQuotaError;
+    let setFallbackHandlerSpy;
+    let mockGoogleApiError;
+    beforeEach(() => {
+        mockConfig = makeFakeConfig();
+        mockGoogleApiError = {
+            code: 429,
+            message: 'mock error',
+            details: [],
+        };
+        // Spy on the method that requires the private field and mock its return.
+        // This is cleaner than modifying the config class for tests.
+        vi.spyOn(mockConfig, 'getContentGeneratorConfig').mockReturnValue({
+            authType: AuthType.LOGIN_WITH_GOOGLE,
+        });
+        mockHistoryManager = {
+            addItem: vi.fn(),
+            history: [],
+            updateItem: vi.fn(),
+            clearItems: vi.fn(),
+            loadHistory: vi.fn(),
+        };
+        mockSetModelSwitchedFromQuotaError = vi.fn();
+        setFallbackHandlerSpy = vi.spyOn(mockConfig, 'setFallbackModelHandler');
+        vi.spyOn(mockConfig, 'setQuotaErrorOccurred');
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should register a fallback handler on initialization', () => {
+        renderHook(() => useQuotaAndFallback({
+            config: mockConfig,
+            historyManager: mockHistoryManager,
+            userTier: UserTierId.FREE,
+            setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+        }));
+        expect(setFallbackHandlerSpy).toHaveBeenCalledTimes(1);
+        expect(setFallbackHandlerSpy.mock.calls[0][0]).toBeInstanceOf(Function);
+    });
+    describe('Fallback Handler Logic', () => {
+        // Helper function to render the hook and extract the registered handler
+        const getRegisteredHandler = () => {
+            renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            return setFallbackHandlerSpy.mock.calls[0][0];
+        };
+        it('should return null and take no action if authType is not LOGIN_WITH_GOOGLE', async () => {
+            // Override the default mock from beforeEach for this specific test
+            vi.spyOn(mockConfig, 'getContentGeneratorConfig').mockReturnValue({
+                authType: AuthType.USE_GEMINI,
+            });
+            const handler = getRegisteredHandler();
+            const result = await handler('gemini-pro', 'gemini-flash', new Error());
+            expect(result).toBeNull();
+            expect(mockHistoryManager.addItem).not.toHaveBeenCalled();
+        });
+        describe('Interactive Fallback', () => {
+            it('should set an interactive request for a terminal quota error', async () => {
+                const { result } = renderHook(() => useQuotaAndFallback({
+                    config: mockConfig,
+                    historyManager: mockHistoryManager,
+                    userTier: UserTierId.FREE,
+                    setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+                }));
+                const handler = setFallbackHandlerSpy.mock
+                    .calls[0][0];
+                let promise;
+                const error = new TerminalQuotaError('pro quota', mockGoogleApiError, 1000 * 60 * 5); // 5 minutes
+                await act(() => {
+                    promise = handler('gemini-pro', 'gemini-flash', error);
+                });
+                // The hook should now have a pending request for the UI to handle
+                const request = result.current.proQuotaRequest;
+                expect(request).not.toBeNull();
+                expect(request?.failedModel).toBe('gemini-pro');
+                expect(request?.isTerminalQuotaError).toBe(true);
+                const message = request.message;
+                expect(message).toContain('Usage limit reached for gemini-pro.');
+                expect(message).toContain('Access resets at'); // From getResetTimeMessage
+                expect(message).toContain('/stats for usage details');
+                expect(message).toContain('/auth to switch to API key.');
+                expect(mockHistoryManager.addItem).not.toHaveBeenCalled();
+                // Simulate the user choosing to continue with the fallback model
+                await act(() => {
+                    result.current.handleProQuotaChoice('retry_always');
+                });
+                // The original promise from the handler should now resolve
+                const intent = await promise;
+                expect(intent).toBe('retry_always');
+                // The pending request should be cleared from the state
+                expect(result.current.proQuotaRequest).toBeNull();
+                expect(mockHistoryManager.addItem).toHaveBeenCalledTimes(1);
+            });
+            it('should handle race conditions by stopping subsequent requests', async () => {
+                const { result } = renderHook(() => useQuotaAndFallback({
+                    config: mockConfig,
+                    historyManager: mockHistoryManager,
+                    userTier: UserTierId.FREE,
+                    setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+                }));
+                const handler = setFallbackHandlerSpy.mock
+                    .calls[0][0];
+                let promise1;
+                await act(() => {
+                    promise1 = handler('gemini-pro', 'gemini-flash', new TerminalQuotaError('pro quota 1', mockGoogleApiError));
+                });
+                const firstRequest = result.current.proQuotaRequest;
+                expect(firstRequest).not.toBeNull();
+                let result2;
+                await act(async () => {
+                    result2 = await handler('gemini-pro', 'gemini-flash', new TerminalQuotaError('pro quota 2', mockGoogleApiError));
+                });
+                // The lock should have stopped the second request
+                expect(result2).toBe('stop');
+                expect(result.current.proQuotaRequest).toBe(firstRequest);
+                await act(() => {
+                    result.current.handleProQuotaChoice('retry_always');
+                });
+                const intent1 = await promise1;
+                expect(intent1).toBe('retry_always');
+                expect(result.current.proQuotaRequest).toBeNull();
+            });
+            // Non-TerminalQuotaError test cases
+            const testCases = [
+                {
+                    description: 'generic error',
+                    error: new Error('some error'),
+                },
+                {
+                    description: 'retryable quota error',
+                    error: new RetryableQuotaError('retryable quota', mockGoogleApiError, 5),
+                },
+            ];
+            for (const { description, error } of testCases) {
+                it(`should handle ${description} correctly`, async () => {
+                    const { result } = renderHook(() => useQuotaAndFallback({
+                        config: mockConfig,
+                        historyManager: mockHistoryManager,
+                        userTier: UserTierId.FREE,
+                        setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+                    }));
+                    const handler = setFallbackHandlerSpy.mock
+                        .calls[0][0];
+                    let promise;
+                    await act(() => {
+                        promise = handler('model-A', 'model-B', error);
+                    });
+                    // The hook should now have a pending request for the UI to handle
+                    const request = result.current.proQuotaRequest;
+                    expect(request).not.toBeNull();
+                    expect(request?.failedModel).toBe('model-A');
+                    expect(request?.isTerminalQuotaError).toBe(false);
+                    // Check that the correct initial message was generated
+                    expect(mockHistoryManager.addItem).not.toHaveBeenCalled();
+                    const message = request.message;
+                    expect(message).toContain('model-A is currently experiencing high demand. We apologize and appreciate your patience.');
+                    // Simulate the user choosing to continue with the fallback model
+                    await act(() => {
+                        result.current.handleProQuotaChoice('retry_always');
+                    });
+                    expect(mockSetModelSwitchedFromQuotaError).toHaveBeenCalledWith(true);
+                    // The original promise from the handler should now resolve
+                    const intent = await promise;
+                    expect(intent).toBe('retry_always');
+                    // The pending request should be cleared from the state
+                    expect(result.current.proQuotaRequest).toBeNull();
+                    expect(mockConfig.setQuotaErrorOccurred).toHaveBeenCalledWith(true);
+                    // Check for the "Switched to fallback model" message
+                    expect(mockHistoryManager.addItem).toHaveBeenCalledTimes(1);
+                    const lastCall = mockHistoryManager.addItem.mock
+                        .calls[0][0];
+                    expect(lastCall.type).toBe(MessageType.INFO);
+                    expect(lastCall.text).toContain('Switched to fallback model.');
+                });
+            }
+            it('should handle ModelNotFoundError correctly', async () => {
+                const { result } = renderHook(() => useQuotaAndFallback({
+                    config: mockConfig,
+                    historyManager: mockHistoryManager,
+                    userTier: UserTierId.FREE,
+                    setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+                }));
+                const handler = setFallbackHandlerSpy.mock
+                    .calls[0][0];
+                let promise;
+                const error = new ModelNotFoundError('model not found', 404);
+                await act(() => {
+                    promise = handler('gemini-3-pro-preview', 'gemini-2.5-pro', error);
+                });
+                // The hook should now have a pending request for the UI to handle
+                const request = result.current.proQuotaRequest;
+                expect(request).not.toBeNull();
+                expect(request?.failedModel).toBe('gemini-3-pro-preview');
+                expect(request?.isTerminalQuotaError).toBe(false);
+                expect(request?.isModelNotFoundError).toBe(true);
+                const message = request.message;
+                expect(message).toBe(`It seems like you don't have access to Gemini 3.
+Learn more at https://goo.gle/enable-preview-features
+To disable Gemini 3, disable "Preview features" in /settings.`);
+                // Simulate the user choosing to switch
+                await act(() => {
+                    result.current.handleProQuotaChoice('retry_always');
+                });
+                const intent = await promise;
+                expect(intent).toBe('retry_always');
+                expect(result.current.proQuotaRequest).toBeNull();
+            });
+        });
+    });
+    describe('handleProQuotaChoice', () => {
+        it('should do nothing if there is no pending pro quota request', () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            act(() => {
+                result.current.handleProQuotaChoice('retry_later');
+            });
+            expect(mockHistoryManager.addItem).not.toHaveBeenCalled();
+        });
+        it('should resolve intent to "retry_later"', async () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            const handler = setFallbackHandlerSpy.mock
+                .calls[0][0];
+            let promise;
+            await act(() => {
+                promise = handler('gemini-pro', 'gemini-flash', new TerminalQuotaError('pro quota', mockGoogleApiError));
+            });
+            await act(() => {
+                result.current.handleProQuotaChoice('retry_later');
+            });
+            const intent = await promise;
+            expect(intent).toBe('retry_later');
+            expect(result.current.proQuotaRequest).toBeNull();
+        });
+        it('should resolve intent to "retry_always" and add info message on continue', async () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            const handler = setFallbackHandlerSpy.mock
+                .calls[0][0];
+            let promise;
+            await act(() => {
+                promise = handler('gemini-pro', 'gemini-flash', new TerminalQuotaError('pro quota', mockGoogleApiError));
+            });
+            await act(() => {
+                result.current.handleProQuotaChoice('retry_always');
+            });
+            const intent = await promise;
+            expect(intent).toBe('retry_always');
+            expect(result.current.proQuotaRequest).toBeNull();
+            // Check for the "Switched to fallback model" message
+            expect(mockHistoryManager.addItem).toHaveBeenCalledTimes(1);
+            const lastCall = mockHistoryManager.addItem.mock.calls[0][0];
+            expect(lastCall.type).toBe(MessageType.INFO);
+            expect(lastCall.text).toContain('Switched to fallback model.');
+        });
+        it('should show a special message when falling back from the preview model', async () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            const handler = setFallbackHandlerSpy.mock
+                .calls[0][0];
+            let promise;
+            await act(() => {
+                promise = handler(PREVIEW_GEMINI_MODEL, DEFAULT_GEMINI_MODEL, new Error('preview model failed'));
+            });
+            await act(() => {
+                result.current.handleProQuotaChoice('retry_always');
+            });
+            await promise;
+            expect(mockHistoryManager.addItem).toHaveBeenCalledTimes(1);
+            const lastCall = mockHistoryManager.addItem.mock.calls[0][0];
+            expect(lastCall.type).toBe(MessageType.INFO);
+            expect(lastCall.text).toContain(`Switched to fallback model gemini-2.5-pro. We will periodically check if ${PREVIEW_GEMINI_MODEL} is available again.`);
+        });
+        it('should show a special message when falling back from the preview model, but do not show periodical check message for flash model fallback', async () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            const handler = setFallbackHandlerSpy.mock
+                .calls[0][0];
+            let promise;
+            await act(() => {
+                promise = handler(PREVIEW_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, new Error('preview model failed'));
+            });
+            await act(() => {
+                result.current.handleProQuotaChoice('retry_always');
+            });
+            await promise;
+            expect(mockHistoryManager.addItem).toHaveBeenCalledTimes(1);
+            const lastCall = mockHistoryManager.addItem.mock.calls[0][0];
+            expect(lastCall.type).toBe(MessageType.INFO);
+            expect(lastCall.text).toContain(`Switched to fallback model gemini-2.5-flash.`);
+        });
+    });
+});
+//# sourceMappingURL=useQuotaAndFallback.test.js.map