@google/gemini-cli 0.16.0-preview.4 → 0.16.0

package/dist/src/ui/components/ModelDialog.test.js

@@ -7,7 +7,7 @@ import { jsx as _jsx } from "react/jsx-runtime";
 import { render } from '../../test-utils/render.js';
 import { cleanup } from 'ink-testing-library';
 import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { DEFAULT_GEMINI_FLASH_LITE_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_MODEL_AUTO, } from '@google/gemini-cli-core';
+import { GEMINI_MODEL_ALIAS_FLASH_LITE, GEMINI_MODEL_ALIAS_FLASH, GEMINI_MODEL_ALIAS_PRO, DEFAULT_GEMINI_MODEL_AUTO, } from '@google/gemini-cli-core';
 import { ModelDialog } from './ModelDialog.js';
 import { useKeypress } from '../hooks/useKeypress.js';
 import { DescriptiveRadioButtonSelect } from './shared/DescriptiveRadioButtonSelect.js';
@@ -30,6 +30,7 @@ const renderComponent = (props = {}, contextValue = undefined) => {
             // --- Functions used by ModelDialog ---
             getModel: vi.fn(() => DEFAULT_GEMINI_MODEL_AUTO),
             setModel: vi.fn(),
+            getPreviewFeatures: vi.fn(() => false),
             // --- Functions used by ClearcutLogger ---
             getUsageStatisticsEnabled: vi.fn(() => true),
             getSessionId: vi.fn(() => 'mock-session-id'),
@@ -61,7 +62,7 @@ describe('<ModelDialog />', () => {
         const { lastFrame, unmount } = renderComponent();
         expect(lastFrame()).toContain('Select Model');
         expect(lastFrame()).toContain('(Press Esc to close)');
-        expect(lastFrame()).toContain('> To use a specific Gemini model on startup, use the --model flag.');
+        expect(lastFrame()).toContain('To use a specific Gemini model on startup, use the --model flag.');
         unmount();
     });
     it('passes all model options to DescriptiveRadioButtonSelect', () => {
@@ -70,14 +71,14 @@ describe('<ModelDialog />', () => {
         const props = mockedSelect.mock.calls[0][0];
         expect(props.items).toHaveLength(4);
         expect(props.items[0].value).toBe(DEFAULT_GEMINI_MODEL_AUTO);
-        expect(props.items[1].value).toBe(DEFAULT_GEMINI_MODEL);
-        expect(props.items[2].value).toBe(DEFAULT_GEMINI_FLASH_MODEL);
-        expect(props.items[3].value).toBe(DEFAULT_GEMINI_FLASH_LITE_MODEL);
+        expect(props.items[1].value).toBe(GEMINI_MODEL_ALIAS_PRO);
+        expect(props.items[2].value).toBe(GEMINI_MODEL_ALIAS_FLASH);
+        expect(props.items[3].value).toBe(GEMINI_MODEL_ALIAS_FLASH_LITE);
         expect(props.showNumbers).toBe(true);
         unmount();
     });
     it('initializes with the model from ConfigContext', () => {
-        const mockGetModel = vi.fn(() => DEFAULT_GEMINI_FLASH_MODEL);
+        const mockGetModel = vi.fn(() => GEMINI_MODEL_ALIAS_FLASH);
         const { unmount } = renderComponent({}, { getModel: mockGetModel });
         expect(mockGetModel).toHaveBeenCalled();
         expect(mockedSelect).toHaveBeenCalledWith(expect.objectContaining({
@@ -110,9 +111,9 @@ describe('<ModelDialog />', () => {
         const { props, mockConfig, unmount } = renderComponent({}, {}); // Pass empty object for contextValue
         const childOnSelect = mockedSelect.mock.calls[0][0].onSelect;
         expect(childOnSelect).toBeDefined();
-        childOnSelect(DEFAULT_GEMINI_MODEL);
+        childOnSelect(GEMINI_MODEL_ALIAS_PRO);
         // Assert against the default mock provided by renderComponent
-        expect(mockConfig?.setModel).toHaveBeenCalledWith(DEFAULT_GEMINI_MODEL);
+        expect(mockConfig?.setModel).toHaveBeenCalledWith(GEMINI_MODEL_ALIAS_PRO);
         expect(props.onClose).toHaveBeenCalledTimes(1);
         unmount();
     });
@@ -152,10 +153,17 @@ describe('<ModelDialog />', () => {
     });
     it('updates initialIndex when config context changes', () => {
         const mockGetModel = vi.fn(() => DEFAULT_GEMINI_MODEL_AUTO);
-        const { rerender, unmount } = render(_jsx(ConfigContext.Provider, { value: { getModel: mockGetModel }, children: _jsx(ModelDialog, { onClose: vi.fn() }) }));
+        const oldMockConfig = {
+            getModel: mockGetModel,
+            getPreviewFeatures: vi.fn(() => false),
+        };
+        const { rerender, unmount } = render(_jsx(ConfigContext.Provider, { value: oldMockConfig, children: _jsx(ModelDialog, { onClose: vi.fn() }) }));
         expect(mockedSelect.mock.calls[0][0].initialIndex).toBe(0);
-        mockGetModel.mockReturnValue(DEFAULT_GEMINI_FLASH_LITE_MODEL);
-        const newMockConfig = { getModel: mockGetModel };
+        mockGetModel.mockReturnValue(GEMINI_MODEL_ALIAS_FLASH_LITE);
+        const newMockConfig = {
+            getModel: mockGetModel,
+            getPreviewFeatures: vi.fn(() => false),
+        };
         rerender(_jsx(ConfigContext.Provider, { value: newMockConfig, children: _jsx(ModelDialog, { onClose: vi.fn() }) }));
         // Should be called at least twice: initial render + re-render after context change
         expect(mockedSelect).toHaveBeenCalledTimes(2);

package/dist/src/ui/components/ModelDialog.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"ModelDialog.test.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelDialog.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,+BAA+B,EAC/B,0BAA0B,EAC1B,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,4BAA4B,EAAE,MAAM,0CAA0C,CAAC;AACxF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAG7D,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AACJ,MAAM,iBAAiB,GAAG,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;AAEjD,EAAE,CAAC,IAAI,CAAC,0CAA0C,EAAE,GAAG,EAAE,CAAC,CAAC;IACzD,4BAA4B,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;CAChD,CAAC,CAAC,CAAC;AACJ,MAAM,YAAY,GAAG,EAAE,CAAC,MAAM,CAAC,4BAA4B,CAAC,CAAC;AAE7D,MAAM,eAAe,GAAG,CACtB,QAA2D,EAAE,EAC7D,eAA4C,SAAS,EACrD,EAAE;IACF,MAAM,YAAY,GAAG;QACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;KACjB,CAAC;IACF,MAAM,aAAa,GAAG,EAAE,GAAG,YAAY,EAAE,GAAG,KAAK,EAAE,CAAC;IAEpD,MAAM,UAAU,GAAG,YAAY;QAC7B,CAAC,CAAE;YACC,wCAAwC;YACxC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC;YAChD,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YAEjB,2CAA2C;YAC3C,yBAAyB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YAC5C,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC;YAC5C,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YAChC,yBAAyB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YAC9D,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YACnC,iBAAiB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YACrC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;YAChC,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YAEjC,yCAAyC;YACzC,GAAG,YAAY;SACL;QACd,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,YAAY,GAAG,MAAM,CACzB,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,UAAU,YACvC,KAAC,WAAW,OAAK,aAAa,GAAI,GACX,CAC1B,CAAC;IAEF,OAAO;QACL,GAAG,YAAY;QACf,KAAK,EAAE,aAAa;QACpB,UAAU;KACX,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QACjD,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACtD,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAC3B,oEAAoE,CACrE,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QACtC,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE9C,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAC9D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QACnE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrC,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,0BAA0B,CAAC,CAAC;QAC7D,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,YAAY,EAAE,CAAC;SAChB,CAAC,EACF,SAAS,CACV,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAEnD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,YAAY,EAAE,CAAC;SAChB,CAAC,EACF,SAAS,CACV,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5C,0EAA0E;QAC1E,yCAAyC;QACzC,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAExC,0FAA0F;QAC1F,iDAAiD;QACjD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,YAAY,EAAE,CAAC;SAChB,CAAC,EACF,SAAS,CACV,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,GAAG,EAAE;QACnG,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,qCAAqC;QAErG,MAAM,aAAa,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC7D,MAAM,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;QAEpC,aAAa,CAAC,oBAAoB,CAAC,CAAC;QAEpC,8DAA8D;QAC9D,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,CAAC;QACxE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QAEtC,MAAM,gBAAgB,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;QACnE,MAAM,CAAC,gBAAgB,CAAC,CAAC,aAAa,EAAE,CAAC;QACzC,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QAE7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAE7C,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5C,eAAe,CAAC;YACd,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,KAAK;YACjB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE/C,eAAe,CAAC;YACd,IAAI,EAAE,GAAG;YACT,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,IAAI;YAChB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,CAAC;QAC5D,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAClC,KAAC,aAAa,CAAC,QAAQ,IACrB,KAAK,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAuB,YAEtD,KAAC,WAAW,IAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,GAAI,GACV,CAC1B,CAAC;QAEF,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE3D,YAAY,CAAC,eAAe,CAAC,+BAA+B,CAAC,CAAC;QAC9D,MAAM,aAAa,GAAG,EAAE,QAAQ,EAAE,YAAY,EAAuB,CAAC;QAEtE,QAAQ,CACN,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,aAAa,YAC1C,KAAC,WAAW,IAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,GAAI,GACV,CAC1B,CAAC;QAEF,mFAAmF;QACnF,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3D,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"ModelDialog.test.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelDialog.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,6BAA6B,EAC7B,wBAAwB,EACxB,sBAAsB,EACtB,yBAAyB,GAC1B,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,4BAA4B,EAAE,MAAM,0CAA0C,CAAC;AACxF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAG7D,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AACJ,MAAM,iBAAiB,GAAG,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;AAEjD,EAAE,CAAC,IAAI,CAAC,0CAA0C,EAAE,GAAG,EAAE,CAAC,CAAC;IACzD,4BAA4B,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;CAChD,CAAC,CAAC,CAAC;AACJ,MAAM,YAAY,GAAG,EAAE,CAAC,MAAM,CAAC,4BAA4B,CAAC,CAAC;AAE7D,MAAM,eAAe,GAAG,CACtB,QAA2D,EAAE,EAC7D,eAA4C,SAAS,EACrD,EAAE;IACF,MAAM,YAAY,GAAG;QACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;KACjB,CAAC;IACF,MAAM,aAAa,GAAG,EAAE,GAAG,YAAY,EAAE,GAAG,KAAK,EAAE,CAAC;IAEpD,MAAM,UAAU,GAAG,YAAY;QAC7B,CAAC,CAAE;YACC,wCAAwC;YACxC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC;YAChD,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YAEtC,2CAA2C;YAC3C,yBAAyB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YAC5C,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC;YAC5C,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YAChC,yBAAyB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YAC9D,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YACnC,iBAAiB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YACrC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;YAChC,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;YAEjC,yCAAyC;YACzC,GAAG,YAAY;SACL;QACd,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,YAAY,GAAG,MAAM,CACzB,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,UAAU,YACvC,KAAC,WAAW,OAAK,aAAa,GAAI,GACX,CAC1B,CAAC;IAEF,OAAO;QACL,GAAG,YAAY;QACf,KAAK,EAAE,aAAa;QACpB,UAAU;KACX,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QACjD,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACtD,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAC3B,kEAAkE,CACnE,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QACtC,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE9C,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC5D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QACjE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrC,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,CAAC;QAC3D,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,YAAY,EAAE,CAAC;SAChB,CAAC,EACF,SAAS,CACV,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAEnD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,YAAY,EAAE,CAAC;SAChB,CAAC,EACF,SAAS,CACV,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5C,0EAA0E;QAC1E,yCAAyC;QACzC,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAExC,0FAA0F;QAC1F,iDAAiD;QACjD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,YAAY,EAAE,CAAC;SAChB,CAAC,EACF,SAAS,CACV,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,GAAG,EAAE;QACnG,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,qCAAqC;QAErG,MAAM,aAAa,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC7D,MAAM,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;QAEpC,aAAa,CAAC,sBAAsB,CAAC,CAAC;QAEtC,8DAA8D;QAC9D,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,oBAAoB,CAAC,sBAAsB,CAAC,CAAC;QAC1E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QAEtC,MAAM,gBAAgB,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;QACnE,MAAM,CAAC,gBAAgB,CAAC,CAAC,aAAa,EAAE,CAAC;QACzC,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC;QAE7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAE7C,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5C,eAAe,CAAC;YACd,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,KAAK;YACjB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE/C,eAAe,CAAC;YACd,IAAI,EAAE,GAAG;YACT,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,IAAI;YAChB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,CAAC;QAC5D,MAAM,aAAa,GAAG;YACpB,QAAQ,EAAE,YAAY;YACtB,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;SAClB,CAAC;QACvB,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAClC,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,aAAa,YAC1C,KAAC,WAAW,IAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,GAAI,GACV,CAC1B,CAAC;QAEF,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE3D,YAAY,CAAC,eAAe,CAAC,6BAA6B,CAAC,CAAC;QAC5D,MAAM,aAAa,GAAG;YACpB,QAAQ,EAAE,YAAY;YACtB,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;SAClB,CAAC;QAEvB,QAAQ,CACN,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,aAAa,YAC1C,KAAC,WAAW,IAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,GAAI,GACV,CAC1B,CAAC;QAEF,mFAAmF;QACnF,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3D,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/components/ProQuotaDialog.d.ts

@@ -4,9 +4,15 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import type React from 'react';
+import { UserTierId } from '@google/gemini-cli-core';
 interface ProQuotaDialogProps {
+    failedModel: string;
     fallbackModel: string;
-    onChoice: (choice: 'retry_later' | 'retry') => void;
+    message: string;
+    isTerminalQuotaError: boolean;
+    isModelNotFoundError?: boolean;
+    onChoice: (choice: 'retry_later' | 'retry_once' | 'retry_always' | 'upgrade') => void;
+    userTier: UserTierId | undefined;
 }
-export declare function ProQuotaDialog({ fallbackModel, onChoice, }: ProQuotaDialogProps): React.JSX.Element;
+export declare function ProQuotaDialog({ failedModel, fallbackModel, message, isTerminalQuotaError, isModelNotFoundError, onChoice, userTier, }: ProQuotaDialogProps): React.JSX.Element;
 export {};

package/dist/src/ui/components/ProQuotaDialog.js

@@ -2,22 +2,87 @@ import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import { Box, Text } from 'ink';
 import { RadioButtonSelect } from './shared/RadioButtonSelect.js';
 import { theme } from '../semantic-colors.js';
-export function ProQuotaDialog({ fallbackModel, onChoice, }) {
-    const items = [
-        {
-            label: 'Try again later',
-            value: 'retry_later',
-            key: 'retry_later',
-        },
-        {
-            label: `Switch to ${fallbackModel} for the rest of this session`,
-            value: 'retry',
-            key: 'retry',
-        },
-    ];
+import { DEFAULT_GEMINI_FLASH_LITE_MODEL, DEFAULT_GEMINI_FLASH_MODEL, PREVIEW_GEMINI_MODEL, UserTierId, } from '@google/gemini-cli-core';
+export function ProQuotaDialog({ failedModel, fallbackModel, message, isTerminalQuotaError, isModelNotFoundError, onChoice, userTier, }) {
+    // Use actual user tier if available; otherwise, default to FREE tier behavior (safe default)
+    const isPaidTier = userTier === UserTierId.LEGACY || userTier === UserTierId.STANDARD;
+    let items;
+    // flash and flash lite don't have options to switch or upgrade.
+    if (failedModel === DEFAULT_GEMINI_FLASH_MODEL ||
+        failedModel === DEFAULT_GEMINI_FLASH_LITE_MODEL) {
+        items = [
+            {
+                label: 'Keep trying',
+                value: 'retry_once',
+                key: 'retry_once',
+            },
+            {
+                label: 'Stop',
+                value: 'retry_later',
+                key: 'retry_later',
+            },
+        ];
+    }
+    else if (isModelNotFoundError || (isTerminalQuotaError && isPaidTier)) {
+        // out of quota
+        items = [
+            {
+                label: `Switch to ${fallbackModel}`,
+                value: 'retry_always',
+                key: 'retry_always',
+            },
+            {
+                label: `Stop`,
+                value: 'retry_later',
+                key: 'retry_later',
+            },
+        ];
+    }
+    else if (isTerminalQuotaError && !isPaidTier) {
+        // free user gets an option to upgrade
+        items = [
+            {
+                label: `Switch to ${fallbackModel}`,
+                value: 'retry_always',
+                key: 'retry_always',
+            },
+            {
+                label: 'Upgrade for higher limits',
+                value: 'upgrade',
+                key: 'upgrade',
+            },
+            {
+                label: `Stop`,
+                value: 'retry_later',
+                key: 'retry_later',
+            },
+        ];
+    }
+    else {
+        // capacity error
+        items = [
+            {
+                label: 'Keep trying',
+                value: 'retry_once',
+                key: 'retry_once',
+            },
+            {
+                label: `Switch to ${fallbackModel}`,
+                value: 'retry_always',
+                key: 'retry_always',
+            },
+            {
+                label: 'Stop',
+                value: 'retry_later',
+                key: 'retry_later',
+            },
+        ];
+    }
     const handleSelect = (choice) => {
         onChoice(choice);
     };
-    return (_jsxs(Box, { borderStyle: "round", flexDirection: "column", paddingX: 1, children: [_jsx(Box, { marginTop: 1, marginBottom: 1, children: _jsx(RadioButtonSelect, { items: items, initialIndex: 1, onSelect: handleSelect }) }), _jsx(Text, { color: theme.text.primary, children: "Note: You can always use /model to select a different option." })] }));
+    return (_jsxs(Box, { borderStyle: "round", flexDirection: "column", padding: 1, children: [_jsx(Box, { marginBottom: 1, children: _jsx(Text, { children: message }) }), _jsx(Box, { marginTop: 1, marginBottom: 1, children: _jsx(RadioButtonSelect, { items: items, onSelect: handleSelect }) }), _jsx(Text, { color: theme.text.primary, children: failedModel === PREVIEW_GEMINI_MODEL && !isModelNotFoundError
+                    ? 'Note: We will periodically retry Preview Model to see if congestion has cleared.'
+                    : 'Note: You can always use /model to select a different option.' })] }));
 }
 //# sourceMappingURL=ProQuotaDialog.js.map

package/dist/src/ui/components/ProQuotaDialog.js.map

@@ -1 +1 @@
-{"version":3,"file":"ProQuotaDialog.js","sourceRoot":"","sources":["../../../../src/ui/components/ProQuotaDialog.tsx"],"names":[],"mappings":";AAOA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAO9C,MAAM,UAAU,cAAc,CAAC,EAC7B,aAAa,EACb,QAAQ,GACY;IACpB,MAAM,KAAK,GAAG;QACZ;YACE,KAAK,EAAE,iBAAiB;YACxB,KAAK,EAAE,aAAsB;YAC7B,GAAG,EAAE,aAAa;SACnB;QACD;YACE,KAAK,EAAE,aAAa,aAAa,+BAA+B;YAChE,KAAK,EAAE,OAAgB;YACvB,GAAG,EAAE,OAAO;SACb;KACF,CAAC;IAEF,MAAM,YAAY,GAAG,CAAC,MAA+B,EAAE,EAAE;QACvD,QAAQ,CAAC,MAAM,CAAC,CAAC;IACnB,CAAC,CAAC;IAEF,OAAO,CACL,MAAC,GAAG,IAAC,WAAW,EAAC,OAAO,EAAC,aAAa,EAAC,QAAQ,EAAC,QAAQ,EAAE,CAAC,aACzD,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,YAChC,KAAC,iBAAiB,IAChB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,CAAC,EACf,QAAQ,EAAE,YAAY,GACtB,GACE,EACN,KAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,8EAExB,IACH,CACP,CAAC;AACJ,CAAC"}
+{"version":3,"file":"ProQuotaDialog.js","sourceRoot":"","sources":["../../../../src/ui/components/ProQuotaDialog.tsx"],"names":[],"mappings":";AAOA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAE9C,OAAO,EACL,+BAA+B,EAC/B,0BAA0B,EAC1B,oBAAoB,EACpB,UAAU,GACX,MAAM,yBAAyB,CAAC;AAcjC,MAAM,UAAU,cAAc,CAAC,EAC7B,WAAW,EACX,aAAa,EACb,OAAO,EACP,oBAAoB,EACpB,oBAAoB,EACpB,QAAQ,EACR,QAAQ,GACY;IACpB,6FAA6F;IAC7F,MAAM,UAAU,GACd,QAAQ,KAAK,UAAU,CAAC,MAAM,IAAI,QAAQ,KAAK,UAAU,CAAC,QAAQ,CAAC;IACrE,IAAI,KAAK,CAAC;IACV,gEAAgE;IAChE,IACE,WAAW,KAAK,0BAA0B;QAC1C,WAAW,KAAK,+BAA+B,EAC/C,CAAC;QACD,KAAK,GAAG;YACN;gBACE,KAAK,EAAE,aAAa;gBACpB,KAAK,EAAE,YAAqB;gBAC5B,GAAG,EAAE,YAAY;aAClB;YACD;gBACE,KAAK,EAAE,MAAM;gBACb,KAAK,EAAE,aAAsB;gBAC7B,GAAG,EAAE,aAAa;aACnB;SACF,CAAC;IACJ,CAAC;SAAM,IAAI,oBAAoB,IAAI,CAAC,oBAAoB,IAAI,UAAU,CAAC,EAAE,CAAC;QACxE,eAAe;QACf,KAAK,GAAG;YACN;gBACE,KAAK,EAAE,aAAa,aAAa,EAAE;gBACnC,KAAK,EAAE,cAAuB;gBAC9B,GAAG,EAAE,cAAc;aACpB;YACD;gBACE,KAAK,EAAE,MAAM;gBACb,KAAK,EAAE,aAAsB;gBAC7B,GAAG,EAAE,aAAa;aACnB;SACF,CAAC;IACJ,CAAC;SAAM,IAAI,oBAAoB,IAAI,CAAC,UAAU,EAAE,CAAC;QAC/C,sCAAsC;QACtC,KAAK,GAAG;YACN;gBACE,KAAK,EAAE,aAAa,aAAa,EAAE;gBACnC,KAAK,EAAE,cAAuB;gBAC9B,GAAG,EAAE,cAAc;aACpB;YACD;gBACE,KAAK,EAAE,2BAA2B;gBAClC,KAAK,EAAE,SAAkB;gBACzB,GAAG,EAAE,SAAS;aACf;YACD;gBACE,KAAK,EAAE,MAAM;gBACb,KAAK,EAAE,aAAsB;gBAC7B,GAAG,EAAE,aAAa;aACnB;SACF,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,iBAAiB;QACjB,KAAK,GAAG;YACN;gBACE,KAAK,EAAE,aAAa;gBACpB,KAAK,EAAE,YAAqB;gBAC5B,GAAG,EAAE,YAAY;aAClB;YACD;gBACE,KAAK,EAAE,aAAa,aAAa,EAAE;gBACnC,KAAK,EAAE,cAAuB;gBAC9B,GAAG,EAAE,cAAc;aACpB;YACD;gBACE,KAAK,EAAE,MAAM;gBACb,KAAK,EAAE,aAAsB;gBAC7B,GAAG,EAAE,aAAa;aACnB;SACF,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,CACnB,MAAiE,EACjE,EAAE;QACF,QAAQ,CAAC,MAAM,CAAC,CAAC;IACnB,CAAC,CAAC;IAEF,OAAO,CACL,MAAC,GAAG,IAAC,WAAW,EAAC,OAAO,EAAC,aAAa,EAAC,QAAQ,EAAC,OAAO,EAAE,CAAC,aACxD,KAAC,GAAG,IAAC,YAAY,EAAE,CAAC,YAClB,KAAC,IAAI,cAAE,OAAO,GAAQ,GAClB,EACN,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,YAChC,KAAC,iBAAiB,IAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,YAAY,GAAI,GACvD,EACN,KAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,YAC5B,WAAW,KAAK,oBAAoB,IAAI,CAAC,oBAAoB;oBAC5D,CAAC,CAAC,kFAAkF;oBACpF,CAAC,CAAC,+DAA+D,GAC9D,IACH,CACP,CAAC;AACJ,CAAC"}

package/dist/src/ui/components/ProQuotaDialog.test.js

@@ -9,58 +9,164 @@ import { act } from 'react';
 import { describe, it, expect, vi, beforeEach } from 'vitest';
 import { ProQuotaDialog } from './ProQuotaDialog.js';
 import { RadioButtonSelect } from './shared/RadioButtonSelect.js';
+import { PREVIEW_GEMINI_MODEL, UserTierId, DEFAULT_GEMINI_FLASH_MODEL, } from '@google/gemini-cli-core';
 // Mock the child component to make it easier to test the parent
 vi.mock('./shared/RadioButtonSelect.js', () => ({
     RadioButtonSelect: vi.fn(),
 }));
 describe('ProQuotaDialog', () => {
+    const mockOnChoice = vi.fn();
     beforeEach(() => {
         vi.clearAllMocks();
     });
-    it('should render with correct title and options', () => {
-        const { lastFrame, unmount } = render(_jsx(ProQuotaDialog, { fallbackModel: "gemini-2.5-flash", onChoice: () => { } }));
-        const output = lastFrame();
-        expect(output).toContain('Note: You can always use /model to select a different option.');
-        // Check that RadioButtonSelect was called with the correct items
-        expect(RadioButtonSelect).toHaveBeenCalledWith(expect.objectContaining({
-            items: [
-                {
-                    label: 'Try again later',
-                    value: 'retry_later',
-                    key: 'retry_later',
-                },
-                {
-                    label: `Switch to gemini-2.5-flash for the rest of this session`,
-                    value: 'retry',
-                    key: 'retry',
-                },
-            ],
-        }), undefined);
-        unmount();
+    describe('for flash model failures', () => {
+        it('should render "Keep trying" and "Stop" options', () => {
+            const { unmount } = render(_jsx(ProQuotaDialog, { failedModel: DEFAULT_GEMINI_FLASH_MODEL, fallbackModel: "gemini-2.5-pro", message: "flash error", isTerminalQuotaError: true, onChoice: mockOnChoice, userTier: UserTierId.FREE }));
+            expect(RadioButtonSelect).toHaveBeenCalledWith(expect.objectContaining({
+                items: [
+                    {
+                        label: 'Keep trying',
+                        value: 'retry_once',
+                        key: 'retry_once',
+                    },
+                    {
+                        label: 'Stop',
+                        value: 'retry_later',
+                        key: 'retry_later',
+                    },
+                ],
+            }), undefined);
+            unmount();
+        });
+    });
+    describe('for non-flash model failures', () => {
+        describe('when it is a terminal quota error', () => {
+            it('should render switch and stop options for paid tiers', () => {
+                const { unmount } = render(_jsx(ProQuotaDialog, { failedModel: "gemini-2.5-pro", fallbackModel: "gemini-2.5-flash", message: "paid tier quota error", isTerminalQuotaError: true, isModelNotFoundError: false, onChoice: mockOnChoice, userTier: UserTierId.LEGACY }));
+                expect(RadioButtonSelect).toHaveBeenCalledWith(expect.objectContaining({
+                    items: [
+                        {
+                            label: 'Switch to gemini-2.5-flash',
+                            value: 'retry_always',
+                            key: 'retry_always',
+                        },
+                        {
+                            label: 'Stop',
+                            value: 'retry_later',
+                            key: 'retry_later',
+                        },
+                    ],
+                }), undefined);
+                unmount();
+            });
+            it('should render switch, upgrade, and stop options for free tier', () => {
+                const { unmount } = render(_jsx(ProQuotaDialog, { failedModel: "gemini-2.5-pro", fallbackModel: "gemini-2.5-flash", message: "free tier quota error", isTerminalQuotaError: true, isModelNotFoundError: false, onChoice: mockOnChoice, userTier: UserTierId.FREE }));
+                expect(RadioButtonSelect).toHaveBeenCalledWith(expect.objectContaining({
+                    items: [
+                        {
+                            label: 'Switch to gemini-2.5-flash',
+                            value: 'retry_always',
+                            key: 'retry_always',
+                        },
+                        {
+                            label: 'Upgrade for higher limits',
+                            value: 'upgrade',
+                            key: 'upgrade',
+                        },
+                        {
+                            label: 'Stop',
+                            value: 'retry_later',
+                            key: 'retry_later',
+                        },
+                    ],
+                }), undefined);
+                unmount();
+            });
+        });
+        describe('when it is a capacity error', () => {
+            it('should render keep trying, switch, and stop options', () => {
+                const { unmount } = render(_jsx(ProQuotaDialog, { failedModel: "gemini-2.5-pro", fallbackModel: "gemini-2.5-flash", message: "capacity error", isTerminalQuotaError: false, isModelNotFoundError: false, onChoice: mockOnChoice, userTier: UserTierId.FREE }));
+                expect(RadioButtonSelect).toHaveBeenCalledWith(expect.objectContaining({
+                    items: [
+                        {
+                            label: 'Keep trying',
+                            value: 'retry_once',
+                            key: 'retry_once',
+                        },
+                        {
+                            label: 'Switch to gemini-2.5-flash',
+                            value: 'retry_always',
+                            key: 'retry_always',
+                        },
+                        { label: 'Stop', value: 'retry_later', key: 'retry_later' },
+                    ],
+                }), undefined);
+                unmount();
+            });
+        });
+        describe('when it is a model not found error', () => {
+            it('should render switch and stop options regardless of tier', () => {
+                const { unmount } = render(_jsx(ProQuotaDialog, { failedModel: "gemini-3-pro-preview", fallbackModel: "gemini-2.5-pro", message: "You don't have access to gemini-3-pro-preview yet.", isTerminalQuotaError: false, isModelNotFoundError: true, onChoice: mockOnChoice, userTier: UserTierId.FREE }));
+                expect(RadioButtonSelect).toHaveBeenCalledWith(expect.objectContaining({
+                    items: [
+                        {
+                            label: 'Switch to gemini-2.5-pro',
+                            value: 'retry_always',
+                            key: 'retry_always',
+                        },
+                        {
+                            label: 'Stop',
+                            value: 'retry_later',
+                            key: 'retry_later',
+                        },
+                    ],
+                }), undefined);
+                unmount();
+            });
+            it('should render switch and stop options for paid tier as well', () => {
+                const { unmount } = render(_jsx(ProQuotaDialog, { failedModel: "gemini-3-pro-preview", fallbackModel: "gemini-2.5-pro", message: "You don't have access to gemini-3-pro-preview yet.", isTerminalQuotaError: false, isModelNotFoundError: true, onChoice: mockOnChoice, userTier: UserTierId.LEGACY }));
+                expect(RadioButtonSelect).toHaveBeenCalledWith(expect.objectContaining({
+                    items: [
+                        {
+                            label: 'Switch to gemini-2.5-pro',
+                            value: 'retry_always',
+                            key: 'retry_always',
+                        },
+                        {
+                            label: 'Stop',
+                            value: 'retry_later',
+                            key: 'retry_later',
+                        },
+                    ],
+                }), undefined);
+                unmount();
+            });
+        });
     });
-    it('should call onChoice with "auth" when "Change auth" is selected', () => {
-        const mockOnChoice = vi.fn();
-        const { unmount } = render(_jsx(ProQuotaDialog, { fallbackModel: "gemini-2.5-flash", onChoice: mockOnChoice }));
-        // Get the onSelect function passed to RadioButtonSelect
-        const onSelect = RadioButtonSelect.mock.calls[0][0].onSelect;
-        // Simulate the selection
-        act(() => {
-            onSelect('auth');
+    describe('onChoice handling', () => {
+        it('should call onChoice with the selected value', () => {
+            const { unmount } = render(_jsx(ProQuotaDialog, { failedModel: "gemini-2.5-pro", fallbackModel: "gemini-2.5-flash", message: "", isTerminalQuotaError: false, onChoice: mockOnChoice, userTier: UserTierId.FREE }));
+            const onSelect = RadioButtonSelect.mock.calls[0][0].onSelect;
+            act(() => {
+                onSelect('retry_always');
+            });
+            expect(mockOnChoice).toHaveBeenCalledWith('retry_always');
+            unmount();
         });
-        expect(mockOnChoice).toHaveBeenCalledWith('auth');
-        unmount();
     });
-    it('should call onChoice with "continue" when "Continue with flash" is selected', () => {
-        const mockOnChoice = vi.fn();
-        const { unmount } = render(_jsx(ProQuotaDialog, { fallbackModel: "gemini-2.5-flash", onChoice: mockOnChoice }));
-        // Get the onSelect function passed to RadioButtonSelect
-        const onSelect = RadioButtonSelect.mock.calls[0][0].onSelect;
-        // Simulate the selection
-        act(() => {
-            onSelect('retry');
+    describe('footer note', () => {
+        it('should show a special note for PREVIEW_GEMINI_MODEL', () => {
+            const { lastFrame, unmount } = render(_jsx(ProQuotaDialog, { failedModel: PREVIEW_GEMINI_MODEL, fallbackModel: "gemini-2.5-pro", message: "", isTerminalQuotaError: false, onChoice: mockOnChoice, userTier: UserTierId.FREE }));
+            const output = lastFrame();
+            expect(output).toContain('Note: We will periodically retry Preview Model to see if congestion has cleared.');
+            unmount();
+        });
+        it('should show the default note for other models', () => {
+            const { lastFrame, unmount } = render(_jsx(ProQuotaDialog, { failedModel: "gemini-2.5-pro", fallbackModel: "gemini-2.5-flash", message: "", isTerminalQuotaError: false, onChoice: mockOnChoice, userTier: UserTierId.FREE }));
+            const output = lastFrame();
+            expect(output).toContain('Note: You can always use /model to select a different option.');
+            unmount();
         });
-        expect(mockOnChoice).toHaveBeenCalledWith('retry');
-        unmount();
     });
 });
 //# sourceMappingURL=ProQuotaDialog.test.js.map

package/dist/src/ui/components/ProQuotaDialog.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"ProQuotaDialog.test.js","sourceRoot":"","sources":["../../../../src/ui/components/ProQuotaDialog.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAa,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAElE,gEAAgE;AAChE,EAAE,CAAC,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC3B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,MAAM,CACnC,KAAC,cAAc,IAAC,aAAa,EAAC,kBAAkB,EAAC,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,GAAI,CACxE,CAAC;QAEF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,+DAA+D,CAChE,CAAC;QAEF,iEAAiE;QACjE,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC;YACtB,KAAK,EAAE;gBACL;oBACE,KAAK,EAAE,iBAAiB;oBACxB,KAAK,EAAE,aAAsB;oBAC7B,GAAG,EAAE,aAAa;iBACnB;gBACD;oBACE,KAAK,EAAE,yDAAyD;oBAChE,KAAK,EAAE,OAAgB;oBACvB,GAAG,EAAE,OAAO;iBACb;aACF;SACF,CAAC,EACF,SAAS,CACV,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,aAAa,EAAC,kBAAkB,EAChC,QAAQ,EAAE,YAAY,GACtB,CACH,CAAC;QAEF,wDAAwD;QACxD,MAAM,QAAQ,GAAI,iBAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAEvE,yBAAyB;QACzB,GAAG,CAAC,GAAG,EAAE;YACP,QAAQ,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAClD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;QACrF,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,aAAa,EAAC,kBAAkB,EAChC,QAAQ,EAAE,YAAY,GACtB,CACH,CAAC;QAEF,wDAAwD;QACxD,MAAM,QAAQ,GAAI,iBAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAEvE,yBAAyB;QACzB,GAAG,CAAC,GAAG,EAAE;YACP,QAAQ,CAAC,OAAO,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACnD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"ProQuotaDialog.test.js","sourceRoot":"","sources":["../../../../src/ui/components/ProQuotaDialog.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAa,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAElE,OAAO,EACL,oBAAoB,EACpB,UAAU,EACV,0BAA0B,GAC3B,MAAM,yBAAyB,CAAC;AAEjC,gEAAgE;AAChE,EAAE,CAAC,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC3B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,WAAW,EAAE,0BAA0B,EACvC,aAAa,EAAC,gBAAgB,EAC9B,OAAO,EAAC,aAAa,EACrB,oBAAoB,EAAE,IAAI,EAC1B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,IAAI,GACzB,CACH,CAAC;YAEF,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,KAAK,EAAE;oBACL;wBACE,KAAK,EAAE,aAAa;wBACpB,KAAK,EAAE,YAAY;wBACnB,GAAG,EAAE,YAAY;qBAClB;oBACD;wBACE,KAAK,EAAE,MAAM;wBACb,KAAK,EAAE,aAAa;wBACpB,GAAG,EAAE,aAAa;qBACnB;iBACF;aACF,CAAC,EACF,SAAS,CACV,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;YACjD,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;gBAC9D,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,WAAW,EAAC,gBAAgB,EAC5B,aAAa,EAAC,kBAAkB,EAChC,OAAO,EAAC,uBAAuB,EAC/B,oBAAoB,EAAE,IAAI,EAC1B,oBAAoB,EAAE,KAAK,EAC3B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,MAAM,GAC3B,CACH,CAAC;gBAEF,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE;wBACL;4BACE,KAAK,EAAE,4BAA4B;4BACnC,KAAK,EAAE,cAAc;4BACrB,GAAG,EAAE,cAAc;yBACpB;wBACD;4BACE,KAAK,EAAE,MAAM;4BACb,KAAK,EAAE,aAAa;4BACpB,GAAG,EAAE,aAAa;yBACnB;qBACF;iBACF,CAAC,EACF,SAAS,CACV,CAAC;gBACF,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;gBACvE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,WAAW,EAAC,gBAAgB,EAC5B,aAAa,EAAC,kBAAkB,EAChC,OAAO,EAAC,uBAAuB,EAC/B,oBAAoB,EAAE,IAAI,EAC1B,oBAAoB,EAAE,KAAK,EAC3B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,IAAI,GACzB,CACH,CAAC;gBAEF,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE;wBACL;4BACE,KAAK,EAAE,4BAA4B;4BACnC,KAAK,EAAE,cAAc;4BACrB,GAAG,EAAE,cAAc;yBACpB;wBACD;4BACE,KAAK,EAAE,2BAA2B;4BAClC,KAAK,EAAE,SAAS;4BAChB,GAAG,EAAE,SAAS;yBACf;wBACD;4BACE,KAAK,EAAE,MAAM;4BACb,KAAK,EAAE,aAAa;4BACpB,GAAG,EAAE,aAAa;yBACnB;qBACF;iBACF,CAAC,EACF,SAAS,CACV,CAAC;gBACF,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;YAC3C,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;gBAC7D,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,WAAW,EAAC,gBAAgB,EAC5B,aAAa,EAAC,kBAAkB,EAChC,OAAO,EAAC,gBAAgB,EACxB,oBAAoB,EAAE,KAAK,EAC3B,oBAAoB,EAAE,KAAK,EAC3B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,IAAI,GACzB,CACH,CAAC;gBAEF,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE;wBACL;4BACE,KAAK,EAAE,aAAa;4BACpB,KAAK,EAAE,YAAY;4BACnB,GAAG,EAAE,YAAY;yBAClB;wBACD;4BACE,KAAK,EAAE,4BAA4B;4BACnC,KAAK,EAAE,cAAc;4BACrB,GAAG,EAAE,cAAc;yBACpB;wBACD,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,EAAE,aAAa,EAAE;qBAC5D;iBACF,CAAC,EACF,SAAS,CACV,CAAC;gBACF,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAClD,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;gBAClE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,WAAW,EAAC,sBAAsB,EAClC,aAAa,EAAC,gBAAgB,EAC9B,OAAO,EAAC,oDAAoD,EAC5D,oBAAoB,EAAE,KAAK,EAC3B,oBAAoB,EAAE,IAAI,EAC1B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,IAAI,GACzB,CACH,CAAC;gBAEF,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE;wBACL;4BACE,KAAK,EAAE,0BAA0B;4BACjC,KAAK,EAAE,cAAc;4BACrB,GAAG,EAAE,cAAc;yBACpB;wBACD;4BACE,KAAK,EAAE,MAAM;4BACb,KAAK,EAAE,aAAa;4BACpB,GAAG,EAAE,aAAa;yBACnB;qBACF;iBACF,CAAC,EACF,SAAS,CACV,CAAC;gBACF,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;gBACrE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,WAAW,EAAC,sBAAsB,EAClC,aAAa,EAAC,gBAAgB,EAC9B,OAAO,EAAC,oDAAoD,EAC5D,oBAAoB,EAAE,KAAK,EAC3B,oBAAoB,EAAE,IAAI,EAC1B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,MAAM,GAC3B,CACH,CAAC;gBAEF,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE;wBACL;4BACE,KAAK,EAAE,0BAA0B;4BACjC,KAAK,EAAE,cAAc;4BACrB,GAAG,EAAE,cAAc;yBACpB;wBACD;4BACE,KAAK,EAAE,MAAM;4BACb,KAAK,EAAE,aAAa;4BACpB,GAAG,EAAE,aAAa;yBACnB;qBACF;iBACF,CAAC,EACF,SAAS,CACV,CAAC;gBACF,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,cAAc,IACb,WAAW,EAAC,gBAAgB,EAC5B,aAAa,EAAC,kBAAkB,EAChC,OAAO,EAAC,EAAE,EACV,oBAAoB,EAAE,KAAK,EAC3B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,IAAI,GACzB,CACH,CAAC;YAEF,MAAM,QAAQ,GAAI,iBAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;YACvE,GAAG,CAAC,GAAG,EAAE;gBACP,QAAQ,CAAC,cAAc,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAC1D,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,MAAM,CACnC,KAAC,cAAc,IACb,WAAW,EAAE,oBAAoB,EACjC,aAAa,EAAC,gBAAgB,EAC9B,OAAO,EAAC,EAAE,EACV,oBAAoB,EAAE,KAAK,EAC3B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,IAAI,GACzB,CACH,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,kFAAkF,CACnF,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,MAAM,CACnC,KAAC,cAAc,IACb,WAAW,EAAC,gBAAgB,EAC5B,aAAa,EAAC,kBAAkB,EAChC,OAAO,EAAC,EAAE,EACV,oBAAoB,EAAE,KAAK,EAC3B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,UAAU,CAAC,IAAI,GACzB,CACH,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,+DAA+D,CAChE,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/components/SettingsDialog.test.js

@@ -295,16 +295,16 @@ describe('SettingsDialog', () => {
             const settings = createMockSettings();
             const onSelect = vi.fn();
             const { stdin, unmount, lastFrame } = renderDialog(settings, onSelect);
-            // Wait for initial render and verify we're on Vim Mode (first setting)
+            // Wait for initial render and verify we're on Preview Features (first setting)
             await waitFor(() => {
-                expect(lastFrame()).toContain('Vim Mode');
+                expect(lastFrame()).toContain('Preview Features (e.g., models)');
             });
-            // Navigate to Disable Auto Update setting and verify we're there
+            // Navigate to Vim Mode setting and verify we're there
             act(() => {
                 stdin.write(TerminalKeys.DOWN_ARROW);
             });
             await waitFor(() => {
-                expect(lastFrame()).toContain('Disable Auto Update');
+                expect(lastFrame()).toContain('Vim Mode');
             });
             // Toggle the setting
             act(() => {
@@ -318,9 +318,9 @@ describe('SettingsDialog', () => {
             await waitFor(() => {
                 expect(vi.mocked(saveModifiedSettings)).toHaveBeenCalled();
             });
-            expect(vi.mocked(saveModifiedSettings)).toHaveBeenCalledWith(new Set(['general.disableAutoUpdate']), expect.objectContaining({
+            expect(vi.mocked(saveModifiedSettings)).toHaveBeenCalledWith(new Set(['general.vimMode']), expect.objectContaining({
                 general: expect.objectContaining({
-                    disableAutoUpdate: true,
+                    vimMode: true,
                 }),
             }), expect.any(LoadedSettings), SettingScope.User);
             unmount();
@@ -445,7 +445,7 @@ describe('SettingsDialog', () => {
             const { lastFrame, unmount } = renderDialog(settings, onSelect);
             // Wait for initial render
             await waitFor(() => {
-                expect(lastFrame()).toContain('Hide Window Title');
+                expect(lastFrame()).toContain('Vim Mode');
             });
             // Verify the dialog is rendered properly
             expect(lastFrame()).toContain('Settings');