@google/gemini-cli 0.0.1

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

@@ -0,0 +1,267 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach, } from 'vitest';
+import { act, renderHook } from '@testing-library/react';
+import { UserTierId, AuthType, isGenericQuotaExceededError, isProQuotaExceededError, makeFakeConfig, } from '@google/gemini-cli-core';
+import { useQuotaAndFallback } from './useQuotaAndFallback.js';
+import { AuthState, MessageType } from '../types.js';
+// Mock the error checking functions from the core package to control test scenarios
+vi.mock('@google/gemini-cli-core', async (importOriginal) => {
+    const original = await importOriginal();
+    return {
+        ...original,
+        isGenericQuotaExceededError: vi.fn(),
+        isProQuotaExceededError: vi.fn(),
+    };
+});
+describe('useQuotaAndFallback', () => {
+    let mockConfig;
+    let mockHistoryManager;
+    let mockSetAuthState;
+    let mockSetModelSwitchedFromQuotaError;
+    let setFallbackHandlerSpy;
+    const mockedIsGenericQuotaExceededError = isGenericQuotaExceededError;
+    const mockedIsProQuotaExceededError = isProQuotaExceededError;
+    beforeEach(() => {
+        mockConfig = makeFakeConfig();
+        // 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(),
+        };
+        mockSetAuthState = vi.fn();
+        mockSetModelSwitchedFromQuotaError = vi.fn();
+        setFallbackHandlerSpy = vi.spyOn(mockConfig, 'setFallbackModelHandler');
+        vi.spyOn(mockConfig, 'setQuotaErrorOccurred');
+        mockedIsGenericQuotaExceededError.mockReturnValue(false);
+        mockedIsProQuotaExceededError.mockReturnValue(false);
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should register a fallback handler on initialization', () => {
+        renderHook(() => useQuotaAndFallback({
+            config: mockConfig,
+            historyManager: mockHistoryManager,
+            userTier: UserTierId.FREE,
+            setAuthState: mockSetAuthState,
+            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 = (userTier = UserTierId.FREE) => {
+            renderHook((props) => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: props.userTier,
+                setAuthState: mockSetAuthState,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }), { initialProps: { userTier } });
+            return setFallbackHandlerSpy.mock.calls[0][0];
+        };
+        it('should return null and take no action if already in fallback mode', async () => {
+            vi.spyOn(mockConfig, 'isInFallbackMode').mockReturnValue(true);
+            const handler = getRegisteredHandler();
+            const result = await handler('gemini-pro', 'gemini-flash', new Error());
+            expect(result).toBeNull();
+            expect(mockHistoryManager.addItem).not.toHaveBeenCalled();
+        });
+        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('Automatic Fallback Scenarios', () => {
+            const testCases = [
+                {
+                    errorType: 'generic',
+                    tier: UserTierId.FREE,
+                    expectedMessageSnippets: [
+                        'Automatically switching from model-A to model-B',
+                        'upgrade to a Gemini Code Assist Standard or Enterprise plan',
+                    ],
+                },
+                {
+                    errorType: 'generic',
+                    tier: UserTierId.STANDARD, // Paid tier
+                    expectedMessageSnippets: [
+                        'Automatically switching from model-A to model-B',
+                        'switch to using a paid API key from AI Studio',
+                    ],
+                },
+                {
+                    errorType: 'other',
+                    tier: UserTierId.FREE,
+                    expectedMessageSnippets: [
+                        'Automatically switching from model-A to model-B for faster responses',
+                        'upgrade to a Gemini Code Assist Standard or Enterprise plan',
+                    ],
+                },
+                {
+                    errorType: 'other',
+                    tier: UserTierId.LEGACY, // Paid tier
+                    expectedMessageSnippets: [
+                        'Automatically switching from model-A to model-B for faster responses',
+                        'switch to using a paid API key from AI Studio',
+                    ],
+                },
+            ];
+            for (const { errorType, tier, expectedMessageSnippets } of testCases) {
+                it(`should handle ${errorType} error for ${tier} tier correctly`, async () => {
+                    mockedIsGenericQuotaExceededError.mockReturnValue(errorType === 'generic');
+                    const handler = getRegisteredHandler(tier);
+                    const result = await handler('model-A', 'model-B', new Error('quota exceeded'));
+                    // Automatic fallbacks should return 'stop'
+                    expect(result).toBe('stop');
+                    expect(mockHistoryManager.addItem).toHaveBeenCalledWith(expect.objectContaining({ type: MessageType.INFO }), expect.any(Number));
+                    const message = mockHistoryManager.addItem.mock.calls[0][0]
+                        .text;
+                    for (const snippet of expectedMessageSnippets) {
+                        expect(message).toContain(snippet);
+                    }
+                    expect(mockSetModelSwitchedFromQuotaError).toHaveBeenCalledWith(true);
+                    expect(mockConfig.setQuotaErrorOccurred).toHaveBeenCalledWith(true);
+                });
+            }
+        });
+        describe('Interactive Fallback (Pro Quota Error)', () => {
+            beforeEach(() => {
+                mockedIsProQuotaExceededError.mockReturnValue(true);
+            });
+            it('should set an interactive request and wait for user choice', async () => {
+                const { result } = renderHook(() => useQuotaAndFallback({
+                    config: mockConfig,
+                    historyManager: mockHistoryManager,
+                    userTier: UserTierId.FREE,
+                    setAuthState: mockSetAuthState,
+                    setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+                }));
+                const handler = setFallbackHandlerSpy.mock
+                    .calls[0][0];
+                // Call the handler but do not await it, to check the intermediate state
+                const promise = handler('gemini-pro', 'gemini-flash', new Error('pro quota'));
+                await act(async () => { });
+                // The hook should now have a pending request for the UI to handle
+                expect(result.current.proQuotaRequest).not.toBeNull();
+                expect(result.current.proQuotaRequest?.failedModel).toBe('gemini-pro');
+                // Simulate the user choosing to continue with the fallback model
+                act(() => {
+                    result.current.handleProQuotaChoice('continue');
+                });
+                // The original promise from the handler should now resolve
+                const intent = await promise;
+                expect(intent).toBe('retry');
+                // The pending request should be cleared from the state
+                expect(result.current.proQuotaRequest).toBeNull();
+            });
+            it('should handle race conditions by stopping subsequent requests', async () => {
+                const { result } = renderHook(() => useQuotaAndFallback({
+                    config: mockConfig,
+                    historyManager: mockHistoryManager,
+                    userTier: UserTierId.FREE,
+                    setAuthState: mockSetAuthState,
+                    setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+                }));
+                const handler = setFallbackHandlerSpy.mock
+                    .calls[0][0];
+                const promise1 = handler('gemini-pro', 'gemini-flash', new Error('pro quota 1'));
+                await act(async () => { });
+                const firstRequest = result.current.proQuotaRequest;
+                expect(firstRequest).not.toBeNull();
+                const result2 = await handler('gemini-pro', 'gemini-flash', new Error('pro quota 2'));
+                // The lock should have stopped the second request
+                expect(result2).toBe('stop');
+                expect(result.current.proQuotaRequest).toBe(firstRequest);
+                act(() => {
+                    result.current.handleProQuotaChoice('continue');
+                });
+                const intent1 = await promise1;
+                expect(intent1).toBe('retry');
+                expect(result.current.proQuotaRequest).toBeNull();
+            });
+        });
+    });
+    describe('handleProQuotaChoice', () => {
+        beforeEach(() => {
+            mockedIsProQuotaExceededError.mockReturnValue(true);
+        });
+        it('should do nothing if there is no pending pro quota request', () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setAuthState: mockSetAuthState,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            act(() => {
+                result.current.handleProQuotaChoice('auth');
+            });
+            expect(mockSetAuthState).not.toHaveBeenCalled();
+            expect(mockHistoryManager.addItem).not.toHaveBeenCalled();
+        });
+        it('should resolve intent to "auth" and trigger auth state update', async () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setAuthState: mockSetAuthState,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            const handler = setFallbackHandlerSpy.mock
+                .calls[0][0];
+            const promise = handler('gemini-pro', 'gemini-flash', new Error('pro quota'));
+            await act(async () => { }); // Allow state to update
+            act(() => {
+                result.current.handleProQuotaChoice('auth');
+            });
+            const intent = await promise;
+            expect(intent).toBe('auth');
+            expect(mockSetAuthState).toHaveBeenCalledWith(AuthState.Updating);
+            expect(result.current.proQuotaRequest).toBeNull();
+        });
+        it('should resolve intent to "retry" and add info message on continue', async () => {
+            const { result } = renderHook(() => useQuotaAndFallback({
+                config: mockConfig,
+                historyManager: mockHistoryManager,
+                userTier: UserTierId.FREE,
+                setAuthState: mockSetAuthState,
+                setModelSwitchedFromQuotaError: mockSetModelSwitchedFromQuotaError,
+            }));
+            const handler = setFallbackHandlerSpy.mock
+                .calls[0][0];
+            // The first `addItem` call is for the initial quota error message
+            const promise = handler('gemini-pro', 'gemini-flash', new Error('pro quota'));
+            await act(async () => { }); // Allow state to update
+            act(() => {
+                result.current.handleProQuotaChoice('continue');
+            });
+            const intent = await promise;
+            expect(intent).toBe('retry');
+            expect(result.current.proQuotaRequest).toBeNull();
+            // Check for the second "Switched to fallback model" message
+            expect(mockHistoryManager.addItem).toHaveBeenCalledTimes(2);
+            const lastCall = mockHistoryManager.addItem.mock.calls[1][0];
+            expect(lastCall.type).toBe(MessageType.INFO);
+            expect(lastCall.text).toContain('Switched to fallback model.');
+        });
+    });
+});
+//# sourceMappingURL=useQuotaAndFallback.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useQuotaAndFallback.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useQuotaAndFallback.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,EAAE,EACF,QAAQ,EACR,EAAE,EACF,MAAM,EACN,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAGL,UAAU,EACV,QAAQ,EACR,2BAA2B,EAC3B,uBAAuB,EACvB,cAAc,GACf,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE/D,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAErD,oFAAoF;AACpF,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1D,MAAM,QAAQ,GACZ,MAAM,cAAc,EAA4C,CAAC;IACnE,OAAO;QACL,GAAG,QAAQ;QACX,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE;QACpC,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE;KACjC,CAAC;AACJ,CAAC,CAAC,CAAC;AAKH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,UAAkB,CAAC;IACvB,IAAI,kBAA2C,CAAC;IAChD,IAAI,gBAAsB,CAAC;IAC3B,IAAI,kCAAwC,CAAC;IAC7C,IAAI,qBAAkC,CAAC;IAEvC,MAAM,iCAAiC,GAAG,2BAAmC,CAAC;IAC9E,MAAM,6BAA6B,GAAG,uBAA+B,CAAC;IAEtE,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG,cAAc,EAAE,CAAC;QAE9B,yEAAyE;QACzE,6DAA6D;QAC7D,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,2BAA2B,CAAC,CAAC,eAAe,CAAC;YAChE,QAAQ,EAAE,QAAQ,CAAC,iBAAiB;SACrC,CAAC,CAAC;QAEH,kBAAkB,GAAG;YACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,OAAO,EAAE,EAAE;YACX,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;SACrB,CAAC;QACF,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,kCAAkC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAE7C,qBAAqB,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,yBAAyB,CAAC,CAAC;QACxE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,uBAAuB,CAAC,CAAC;QAE9C,iCAAiC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACzD,6BAA6B,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,UAAU,CAAC,GAAG,EAAE,CACd,mBAAmB,CAAC;YAClB,MAAM,EAAE,UAAU;YAClB,cAAc,EAAE,kBAAkB;YAClC,QAAQ,EAAE,UAAU,CAAC,IAAI;YACzB,YAAY,EAAE,gBAAgB;YAC9B,8BAA8B,EAAE,kCAAkC;SACnE,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,wEAAwE;QACxE,MAAM,oBAAoB,GAAG,CAC3B,WAAuB,UAAU,CAAC,IAAI,EAChB,EAAE;YACxB,UAAU,CACR,CAAC,KAAK,EAAE,EAAE,CACR,mBAAmB,CAAC;gBAClB,MAAM,EAAE,UAAU;gBAClB,cAAc,EAAE,kBAAkB;gBAClC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,YAAY,EAAE,gBAAgB;gBAC9B,8BAA8B,EAAE,kCAAkC;aACnE,CAAC,EACJ,EAAE,YAAY,EAAE,EAAE,QAAQ,EAAE,EAAE,CAC/B,CAAC;YACF,OAAO,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAyB,CAAC;QACxE,CAAC,CAAC;QAEF,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/D,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAC;YACvC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,cAAc,EAAE,IAAI,KAAK,EAAE,CAAC,CAAC;YAExE,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC1B,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;YAC1F,mEAAmE;YACnE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,2BAA2B,CAAC,CAAC,eAAe,CAAC;gBAChE,QAAQ,EAAE,QAAQ,CAAC,UAAU;aAC9B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAC;YACvC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,cAAc,EAAE,IAAI,KAAK,EAAE,CAAC,CAAC;YAExE,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC1B,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;YAC5C,MAAM,SAAS,GAAG;gBAChB;oBACE,SAAS,EAAE,SAAS;oBACpB,IAAI,EAAE,UAAU,CAAC,IAAI;oBACrB,uBAAuB,EAAE;wBACvB,iDAAiD;wBACjD,6DAA6D;qBAC9D;iBACF;gBACD;oBACE,SAAS,EAAE,SAAS;oBACpB,IAAI,EAAE,UAAU,CAAC,QAAQ,EAAE,YAAY;oBACvC,uBAAuB,EAAE;wBACvB,iDAAiD;wBACjD,+CAA+C;qBAChD;iBACF;gBACD;oBACE,SAAS,EAAE,OAAO;oBAClB,IAAI,EAAE,UAAU,CAAC,IAAI;oBACrB,uBAAuB,EAAE;wBACvB,sEAAsE;wBACtE,6DAA6D;qBAC9D;iBACF;gBACD;oBACE,SAAS,EAAE,OAAO;oBAClB,IAAI,EAAE,UAAU,CAAC,MAAM,EAAE,YAAY;oBACrC,uBAAuB,EAAE;wBACvB,sEAAsE;wBACtE,+CAA+C;qBAChD;iBACF;aACF,CAAC;YAEF,KAAK,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,uBAAuB,EAAE,IAAI,SAAS,EAAE,CAAC;gBACrE,EAAE,CAAC,iBAAiB,SAAS,cAAc,IAAI,iBAAiB,EAAE,KAAK,IAAI,EAAE;oBAC3E,iCAAiC,CAAC,eAAe,CAC/C,SAAS,KAAK,SAAS,CACxB,CAAC;oBAEF,MAAM,OAAO,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;oBAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAC1B,SAAS,EACT,SAAS,EACT,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAC5B,CAAC;oBAEF,2CAA2C;oBAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAE5B,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACrD,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC,EACnD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;oBAEF,MAAM,OAAO,GAAI,kBAAkB,CAAC,OAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;yBAClE,IAAI,CAAC;oBACR,KAAK,MAAM,OAAO,IAAI,uBAAuB,EAAE,CAAC;wBAC9C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;oBACrC,CAAC;oBAED,MAAM,CAAC,kCAAkC,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;oBACtE,MAAM,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;gBACtE,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;YACtD,UAAU,CAAC,GAAG,EAAE;gBACd,6BAA6B,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;gBAC1E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,mBAAmB,CAAC;oBAClB,MAAM,EAAE,UAAU;oBAClB,cAAc,EAAE,kBAAkB;oBAClC,QAAQ,EAAE,UAAU,CAAC,IAAI;oBACzB,YAAY,EAAE,gBAAgB;oBAC9B,8BAA8B,EAAE,kCAAkC;iBACnE,CAAC,CACH,CAAC;gBAEF,MAAM,OAAO,GAAG,qBAAqB,CAAC,IAAI;qBACvC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAyB,CAAC;gBAEvC,wEAAwE;gBACxE,MAAM,OAAO,GAAG,OAAO,CACrB,YAAY,EACZ,cAAc,EACd,IAAI,KAAK,CAAC,WAAW,CAAC,CACvB,CAAC;gBAEF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAC;gBAE1B,kEAAkE;gBAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAEvE,iEAAiE;gBACjE,GAAG,CAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;gBAClD,CAAC,CAAC,CAAC;gBAEH,2DAA2D;gBAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;gBAC7B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAE7B,uDAAuD;gBACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAAC;YACpD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;gBAC7E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,mBAAmB,CAAC;oBAClB,MAAM,EAAE,UAAU;oBAClB,cAAc,EAAE,kBAAkB;oBAClC,QAAQ,EAAE,UAAU,CAAC,IAAI;oBACzB,YAAY,EAAE,gBAAgB;oBAC9B,8BAA8B,EAAE,kCAAkC;iBACnE,CAAC,CACH,CAAC;gBAEF,MAAM,OAAO,GAAG,qBAAqB,CAAC,IAAI;qBACvC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAyB,CAAC;gBAEvC,MAAM,QAAQ,GAAG,OAAO,CACtB,YAAY,EACZ,cAAc,EACd,IAAI,KAAK,CAAC,aAAa,CAAC,CACzB,CAAC;gBACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAC;gBAE1B,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC;gBACpD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;gBAEpC,MAAM,OAAO,GAAG,MAAM,OAAO,CAC3B,YAAY,EACZ,cAAc,EACd,IAAI,KAAK,CAAC,aAAa,CAAC,CACzB,CAAC;gBAEF,kDAAkD;gBAClD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC7B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAE1D,GAAG,CAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;gBAClD,CAAC,CAAC,CAAC;gBAEH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC;gBAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC9B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,UAAU,CAAC,GAAG,EAAE;YACd,6BAA6B,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,mBAAmB,CAAC;gBAClB,MAAM,EAAE,UAAU;gBAClB,cAAc,EAAE,kBAAkB;gBAClC,QAAQ,EAAE,UAAU,CAAC,IAAI;gBACzB,YAAY,EAAE,gBAAgB;gBAC9B,8BAA8B,EAAE,kCAAkC;aACnE,CAAC,CACH,CAAC;YAEF,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAChD,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,mBAAmB,CAAC;gBAClB,MAAM,EAAE,UAAU;gBAClB,cAAc,EAAE,kBAAkB;gBAClC,QAAQ,EAAE,UAAU,CAAC,IAAI;gBACzB,YAAY,EAAE,gBAAgB;gBAC9B,8BAA8B,EAAE,kCAAkC;aACnE,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,qBAAqB,CAAC,IAAI;iBACvC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAyB,CAAC;YACvC,MAAM,OAAO,GAAG,OAAO,CACrB,YAAY,EACZ,cAAc,EACd,IAAI,KAAK,CAAC,WAAW,CAAC,CACvB,CAAC;YACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;YAEnD,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,mBAAmB,CAAC;gBAClB,MAAM,EAAE,UAAU;gBAClB,cAAc,EAAE,kBAAkB;gBAClC,QAAQ,EAAE,UAAU,CAAC,IAAI;gBACzB,YAAY,EAAE,gBAAgB;gBAC9B,8BAA8B,EAAE,kCAAkC;aACnE,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,qBAAqB,CAAC,IAAI;iBACvC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAyB,CAAC;YACvC,kEAAkE;YAClE,MAAM,OAAO,GAAG,OAAO,CACrB,YAAY,EACZ,cAAc,EACd,IAAI,KAAK,CAAC,WAAW,CAAC,CACvB,CAAC;YACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;YAEnD,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAAC;YAElD,4DAA4D;YAC5D,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAI,kBAAkB,CAAC,OAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config, ToolCallRequestInfo, ExecutingToolCall, ScheduledToolCall, ValidatingToolCall, WaitingToolCall, CompletedToolCall, CancelledToolCall, EditorType } from '@google/gemini-cli-core';
+import type { HistoryItemToolGroup } from '../types.js';
+export type ScheduleFn = (request: ToolCallRequestInfo | ToolCallRequestInfo[], signal: AbortSignal) => void;
+export type MarkToolsAsSubmittedFn = (callIds: string[]) => void;
+export type TrackedScheduledToolCall = ScheduledToolCall & {
+    responseSubmittedToGemini?: boolean;
+};
+export type TrackedValidatingToolCall = ValidatingToolCall & {
+    responseSubmittedToGemini?: boolean;
+};
+export type TrackedWaitingToolCall = WaitingToolCall & {
+    responseSubmittedToGemini?: boolean;
+};
+export type TrackedExecutingToolCall = ExecutingToolCall & {
+    responseSubmittedToGemini?: boolean;
+    pid?: number;
+};
+export type TrackedCompletedToolCall = CompletedToolCall & {
+    responseSubmittedToGemini?: boolean;
+};
+export type TrackedCancelledToolCall = CancelledToolCall & {
+    responseSubmittedToGemini?: boolean;
+};
+export type TrackedToolCall = TrackedScheduledToolCall | TrackedValidatingToolCall | TrackedWaitingToolCall | TrackedExecutingToolCall | TrackedCompletedToolCall | TrackedCancelledToolCall;
+export declare function useReactToolScheduler(onComplete: (tools: CompletedToolCall[]) => Promise<void>, config: Config, getPreferredEditor: () => EditorType | undefined, onEditorClose: () => void): [TrackedToolCall[], ScheduleFn, MarkToolsAsSubmittedFn];
+/**
+ * Transforms `TrackedToolCall` objects into `HistoryItemToolGroup` objects for UI display.
+ */
+export declare function mapToDisplay(toolOrTools: TrackedToolCall[] | TrackedToolCall): HistoryItemToolGroup;

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

@@ -0,0 +1,192 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CoreToolScheduler } from '@google/gemini-cli-core';
+import { useCallback, useState, useMemo } from 'react';
+import { ToolCallStatus } from '../types.js';
+export function useReactToolScheduler(onComplete, config, getPreferredEditor, onEditorClose) {
+    const [toolCallsForDisplay, setToolCallsForDisplay] = useState([]);
+    const outputUpdateHandler = useCallback((toolCallId, outputChunk) => {
+        setToolCallsForDisplay((prevCalls) => prevCalls.map((tc) => {
+            if (tc.request.callId === toolCallId && tc.status === 'executing') {
+                const executingTc = tc;
+                return { ...executingTc, liveOutput: outputChunk };
+            }
+            return tc;
+        }));
+    }, []);
+    const allToolCallsCompleteHandler = useCallback(async (completedToolCalls) => {
+        await onComplete(completedToolCalls);
+    }, [onComplete]);
+    const toolCallsUpdateHandler = useCallback((updatedCoreToolCalls) => {
+        setToolCallsForDisplay((prevTrackedCalls) => updatedCoreToolCalls.map((coreTc) => {
+            const existingTrackedCall = prevTrackedCalls.find((ptc) => ptc.request.callId === coreTc.request.callId);
+            // Start with the new core state, then layer on the existing UI state
+            // to ensure UI-only properties like pid are preserved.
+            const responseSubmittedToGemini = existingTrackedCall?.responseSubmittedToGemini ?? false;
+            if (coreTc.status === 'executing') {
+                return {
+                    ...coreTc,
+                    responseSubmittedToGemini,
+                    liveOutput: existingTrackedCall
+                        ?.liveOutput,
+                    pid: coreTc.pid,
+                };
+            }
+            // For other statuses, explicitly set liveOutput and pid to undefined
+            // to ensure they are not carried over from a previous executing state.
+            return {
+                ...coreTc,
+                responseSubmittedToGemini,
+                liveOutput: undefined,
+                pid: undefined,
+            };
+        }));
+    }, [setToolCallsForDisplay]);
+    const scheduler = useMemo(() => new CoreToolScheduler({
+        outputUpdateHandler,
+        onAllToolCallsComplete: allToolCallsCompleteHandler,
+        onToolCallsUpdate: toolCallsUpdateHandler,
+        getPreferredEditor,
+        config,
+        onEditorClose,
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    }), [
+        config,
+        outputUpdateHandler,
+        allToolCallsCompleteHandler,
+        toolCallsUpdateHandler,
+        getPreferredEditor,
+        onEditorClose,
+    ]);
+    const schedule = useCallback((request, signal) => {
+        void scheduler.schedule(request, signal);
+    }, [scheduler]);
+    const markToolsAsSubmitted = useCallback((callIdsToMark) => {
+        setToolCallsForDisplay((prevCalls) => prevCalls.map((tc) => callIdsToMark.includes(tc.request.callId)
+            ? { ...tc, responseSubmittedToGemini: true }
+            : tc));
+    }, []);
+    return [toolCallsForDisplay, schedule, markToolsAsSubmitted];
+}
+/**
+ * Maps a CoreToolScheduler status to the UI's ToolCallStatus enum.
+ */
+function mapCoreStatusToDisplayStatus(coreStatus) {
+    switch (coreStatus) {
+        case 'validating':
+            return ToolCallStatus.Executing;
+        case 'awaiting_approval':
+            return ToolCallStatus.Confirming;
+        case 'executing':
+            return ToolCallStatus.Executing;
+        case 'success':
+            return ToolCallStatus.Success;
+        case 'cancelled':
+            return ToolCallStatus.Canceled;
+        case 'error':
+            return ToolCallStatus.Error;
+        case 'scheduled':
+            return ToolCallStatus.Pending;
+        default: {
+            const exhaustiveCheck = coreStatus;
+            console.warn(`Unknown core status encountered: ${exhaustiveCheck}`);
+            return ToolCallStatus.Error;
+        }
+    }
+}
+/**
+ * Transforms `TrackedToolCall` objects into `HistoryItemToolGroup` objects for UI display.
+ */
+export function mapToDisplay(toolOrTools) {
+    const toolCalls = Array.isArray(toolOrTools) ? toolOrTools : [toolOrTools];
+    const toolDisplays = toolCalls.map((trackedCall) => {
+        let displayName;
+        let description;
+        let renderOutputAsMarkdown = false;
+        if (trackedCall.status === 'error') {
+            displayName =
+                trackedCall.tool === undefined
+                    ? trackedCall.request.name
+                    : trackedCall.tool.displayName;
+            description = JSON.stringify(trackedCall.request.args);
+        }
+        else {
+            displayName = trackedCall.tool.displayName;
+            description = trackedCall.invocation.getDescription();
+            renderOutputAsMarkdown = trackedCall.tool.isOutputMarkdown;
+        }
+        const baseDisplayProperties = {
+            callId: trackedCall.request.callId,
+            name: displayName,
+            description,
+            renderOutputAsMarkdown,
+        };
+        switch (trackedCall.status) {
+            case 'success':
+                return {
+                    ...baseDisplayProperties,
+                    status: mapCoreStatusToDisplayStatus(trackedCall.status),
+                    resultDisplay: trackedCall.response.resultDisplay,
+                    confirmationDetails: undefined,
+                    outputFile: trackedCall.response.outputFile,
+                };
+            case 'error':
+                return {
+                    ...baseDisplayProperties,
+                    status: mapCoreStatusToDisplayStatus(trackedCall.status),
+                    resultDisplay: trackedCall.response.resultDisplay,
+                    confirmationDetails: undefined,
+                };
+            case 'cancelled':
+                return {
+                    ...baseDisplayProperties,
+                    status: mapCoreStatusToDisplayStatus(trackedCall.status),
+                    resultDisplay: trackedCall.response.resultDisplay,
+                    confirmationDetails: undefined,
+                };
+            case 'awaiting_approval':
+                return {
+                    ...baseDisplayProperties,
+                    status: mapCoreStatusToDisplayStatus(trackedCall.status),
+                    resultDisplay: undefined,
+                    confirmationDetails: trackedCall.confirmationDetails,
+                };
+            case 'executing':
+                return {
+                    ...baseDisplayProperties,
+                    status: mapCoreStatusToDisplayStatus(trackedCall.status),
+                    resultDisplay: trackedCall.liveOutput ?? undefined,
+                    confirmationDetails: undefined,
+                    ptyId: trackedCall.pid,
+                };
+            case 'validating': // Fallthrough
+            case 'scheduled':
+                return {
+                    ...baseDisplayProperties,
+                    status: mapCoreStatusToDisplayStatus(trackedCall.status),
+                    resultDisplay: undefined,
+                    confirmationDetails: undefined,
+                };
+            default: {
+                const exhaustiveCheck = trackedCall;
+                return {
+                    callId: exhaustiveCheck.request.callId,
+                    name: 'Unknown Tool',
+                    description: 'Encountered an unknown tool call state.',
+                    status: ToolCallStatus.Error,
+                    resultDisplay: 'Unknown tool call state',
+                    confirmationDetails: undefined,
+                    renderOutputAsMarkdown: false,
+                };
+            }
+        }
+    });
+    return {
+        type: 'tool_group',
+        tools: toolDisplays,
+    };
+}
+//# sourceMappingURL=useReactToolScheduler.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useReactToolScheduler.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useReactToolScheduler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAkBH,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAKvD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAoC7C,MAAM,UAAU,qBAAqB,CACnC,UAAyD,EACzD,MAAc,EACd,kBAAgD,EAChD,aAAyB;IAEzB,MAAM,CAAC,mBAAmB,EAAE,sBAAsB,CAAC,GAAG,QAAQ,CAE5D,EAAE,CAAC,CAAC;IAEN,MAAM,mBAAmB,GAAwB,WAAW,CAC1D,CAAC,UAAU,EAAE,WAAW,EAAE,EAAE;QAC1B,sBAAsB,CAAC,CAAC,SAAS,EAAE,EAAE,CACnC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;YACnB,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,KAAK,UAAU,IAAI,EAAE,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAClE,MAAM,WAAW,GAAG,EAA8B,CAAC;gBACnD,OAAO,EAAE,GAAG,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC;YACrD,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,2BAA2B,GAAgC,WAAW,CAC1E,KAAK,EAAE,kBAAkB,EAAE,EAAE;QAC3B,MAAM,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACvC,CAAC,EACD,CAAC,UAAU,CAAC,CACb,CAAC;IAEF,MAAM,sBAAsB,GAA2B,WAAW,CAChE,CAAC,oBAAgC,EAAE,EAAE;QACnC,sBAAsB,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAC1C,oBAAoB,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAClC,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,IAAI,CAC/C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,OAAO,CAAC,MAAM,CACtD,CAAC;YACF,qEAAqE;YACrE,uDAAuD;YACvD,MAAM,yBAAyB,GAC7B,mBAAmB,EAAE,yBAAyB,IAAI,KAAK,CAAC;YAE1D,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAClC,OAAO;oBACL,GAAG,MAAM;oBACT,yBAAyB;oBACzB,UAAU,EAAG,mBAAgD;wBAC3D,EAAE,UAAU;oBACd,GAAG,EAAG,MAA4B,CAAC,GAAG;iBACvC,CAAC;YACJ,CAAC;YAED,qEAAqE;YACrE,uEAAuE;YACvE,OAAO;gBACL,GAAG,MAAM;gBACT,yBAAyB;gBACzB,UAAU,EAAE,SAAS;gBACrB,GAAG,EAAE,SAAS;aACf,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC,EACD,CAAC,sBAAsB,CAAC,CACzB,CAAC;IAEF,MAAM,SAAS,GAAG,OAAO,CACvB,GAAG,EAAE,CACH,IAAI,iBAAiB,CAAC;QACpB,mBAAmB;QACnB,sBAAsB,EAAE,2BAA2B;QACnD,iBAAiB,EAAE,sBAAsB;QACzC,kBAAkB;QAClB,MAAM;QACN,aAAa;QACb,8DAA8D;KACxD,CAAC,EACX;QACE,MAAM;QACN,mBAAmB;QACnB,2BAA2B;QAC3B,sBAAsB;QACtB,kBAAkB;QAClB,aAAa;KACd,CACF,CAAC;IAEF,MAAM,QAAQ,GAAe,WAAW,CACtC,CACE,OAAoD,EACpD,MAAmB,EACnB,EAAE;QACF,KAAK,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC,EACD,CAAC,SAAS,CAAC,CACZ,CAAC;IAEF,MAAM,oBAAoB,GAA2B,WAAW,CAC9D,CAAC,aAAuB,EAAE,EAAE;QAC1B,sBAAsB,CAAC,CAAC,SAAS,EAAE,EAAE,CACnC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACnB,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC;YACvC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,yBAAyB,EAAE,IAAI,EAAE;YAC5C,CAAC,CAAC,EAAE,CACP,CACF,CAAC;IACJ,CAAC,EACD,EAAE,CACH,CAAC;IAEF,OAAO,CAAC,mBAAmB,EAAE,QAAQ,EAAE,oBAAoB,CAAC,CAAC;AAC/D,CAAC;AAED;;GAEG;AACH,SAAS,4BAA4B,CAAC,UAAsB;IAC1D,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,YAAY;YACf,OAAO,cAAc,CAAC,SAAS,CAAC;QAClC,KAAK,mBAAmB;YACtB,OAAO,cAAc,CAAC,UAAU,CAAC;QACnC,KAAK,WAAW;YACd,OAAO,cAAc,CAAC,SAAS,CAAC;QAClC,KAAK,SAAS;YACZ,OAAO,cAAc,CAAC,OAAO,CAAC;QAChC,KAAK,WAAW;YACd,OAAO,cAAc,CAAC,QAAQ,CAAC;QACjC,KAAK,OAAO;YACV,OAAO,cAAc,CAAC,KAAK,CAAC;QAC9B,KAAK,WAAW;YACd,OAAO,cAAc,CAAC,OAAO,CAAC;QAChC,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,eAAe,GAAU,UAAU,CAAC;YAC1C,OAAO,CAAC,IAAI,CAAC,oCAAoC,eAAe,EAAE,CAAC,CAAC;YACpE,OAAO,cAAc,CAAC,KAAK,CAAC;QAC9B,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,WAAgD;IAEhD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IAE3E,MAAM,YAAY,GAAG,SAAS,CAAC,GAAG,CAChC,CAAC,WAAW,EAA6B,EAAE;QACzC,IAAI,WAAmB,CAAC;QACxB,IAAI,WAAmB,CAAC;QACxB,IAAI,sBAAsB,GAAG,KAAK,CAAC;QAEnC,IAAI,WAAW,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;YACnC,WAAW;gBACT,WAAW,CAAC,IAAI,KAAK,SAAS;oBAC5B,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI;oBAC1B,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC;YACnC,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzD,CAAC;aAAM,CAAC;YACN,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC;YAC3C,WAAW,GAAG,WAAW,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC;YACtD,sBAAsB,GAAG,WAAW,CAAC,IAAI,CAAC,gBAAgB,CAAC;QAC7D,CAAC;QAED,MAAM,qBAAqB,GAGvB;YACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM;YAClC,IAAI,EAAE,WAAW;YACjB,WAAW;YACX,sBAAsB;SACvB,CAAC;QAEF,QAAQ,WAAW,CAAC,MAAM,EAAE,CAAC;YAC3B,KAAK,SAAS;gBACZ,OAAO;oBACL,GAAG,qBAAqB;oBACxB,MAAM,EAAE,4BAA4B,CAAC,WAAW,CAAC,MAAM,CAAC;oBACxD,aAAa,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa;oBACjD,mBAAmB,EAAE,SAAS;oBAC9B,UAAU,EAAE,WAAW,CAAC,QAAQ,CAAC,UAAU;iBAC5C,CAAC;YACJ,KAAK,OAAO;gBACV,OAAO;oBACL,GAAG,qBAAqB;oBACxB,MAAM,EAAE,4BAA4B,CAAC,WAAW,CAAC,MAAM,CAAC;oBACxD,aAAa,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa;oBACjD,mBAAmB,EAAE,SAAS;iBAC/B,CAAC;YACJ,KAAK,WAAW;gBACd,OAAO;oBACL,GAAG,qBAAqB;oBACxB,MAAM,EAAE,4BAA4B,CAAC,WAAW,CAAC,MAAM,CAAC;oBACxD,aAAa,EAAE,WAAW,CAAC,QAAQ,CAAC,aAAa;oBACjD,mBAAmB,EAAE,SAAS;iBAC/B,CAAC;YACJ,KAAK,mBAAmB;gBACtB,OAAO;oBACL,GAAG,qBAAqB;oBACxB,MAAM,EAAE,4BAA4B,CAAC,WAAW,CAAC,MAAM,CAAC;oBACxD,aAAa,EAAE,SAAS;oBACxB,mBAAmB,EAAE,WAAW,CAAC,mBAAmB;iBACrD,CAAC;YACJ,KAAK,WAAW;gBACd,OAAO;oBACL,GAAG,qBAAqB;oBACxB,MAAM,EAAE,4BAA4B,CAAC,WAAW,CAAC,MAAM,CAAC;oBACxD,aAAa,EACV,WAAwC,CAAC,UAAU,IAAI,SAAS;oBACnE,mBAAmB,EAAE,SAAS;oBAC9B,KAAK,EAAG,WAAwC,CAAC,GAAG;iBACrD,CAAC;YACJ,KAAK,YAAY,CAAC,CAAC,cAAc;YACjC,KAAK,WAAW;gBACd,OAAO;oBACL,GAAG,qBAAqB;oBACxB,MAAM,EAAE,4BAA4B,CAAC,WAAW,CAAC,MAAM,CAAC;oBACxD,aAAa,EAAE,SAAS;oBACxB,mBAAmB,EAAE,SAAS;iBAC/B,CAAC;YACJ,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,eAAe,GAAU,WAAW,CAAC;gBAC3C,OAAO;oBACL,MAAM,EAAG,eAAmC,CAAC,OAAO,CAAC,MAAM;oBAC3D,IAAI,EAAE,cAAc;oBACpB,WAAW,EAAE,yCAAyC;oBACtD,MAAM,EAAE,cAAc,CAAC,KAAK;oBAC5B,aAAa,EAAE,yBAAyB;oBACxC,mBAAmB,EAAE,SAAS;oBAC9B,sBAAsB,EAAE,KAAK;iBAC9B,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC,CACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE,YAAY;KACpB,CAAC;AACJ,CAAC"}

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

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const REFRESH_MEMORY_COMMAND_NAME = "/refreshmemory";

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

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export const REFRESH_MEMORY_COMMAND_NAME = '/refreshmemory';
+//# sourceMappingURL=useRefreshMemoryCommand.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useRefreshMemoryCommand.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useRefreshMemoryCommand.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,2BAA2B,GAAG,gBAAgB,CAAC"}

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

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { TextBuffer } from '../components/shared/text-buffer.js';
+import type { Suggestion } from '../components/SuggestionsDisplay.js';
+export interface UseReverseSearchCompletionReturn {
+    suggestions: Suggestion[];
+    activeSuggestionIndex: number;
+    visibleStartIndex: number;
+    showSuggestions: boolean;
+    isLoadingSuggestions: boolean;
+    navigateUp: () => void;
+    navigateDown: () => void;
+    handleAutocomplete: (i: number) => void;
+    resetCompletionState: () => void;
+}
+export declare function useReverseSearchCompletion(buffer: TextBuffer, history: readonly string[], reverseSearchActive: boolean): UseReverseSearchCompletionReturn;