@umsai/ums-code 0.0.14-v1 → 0.1.4-v2

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

@@ -0,0 +1,42 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useCallback, useContext, useMemo } from 'react';
+import { Box, Text } from 'ink';
+import { AuthType, ModelSlashCommandEvent, logModelSlashCommand, } from '@umsai/ums-code-core';
+import { useKeypress } from '../hooks/useKeypress.js';
+import { theme } from '../semantic-colors.js';
+import { DescriptiveRadioButtonSelect } from './shared/DescriptiveRadioButtonSelect.js';
+import { ConfigContext } from '../contexts/ConfigContext.js';
+import { getAvailableModelsForAuthType, MAINLINE_CODER, } from '../models/availableModels.js';
+export function ModelDialog({ onClose }) {
+    const config = useContext(ConfigContext);
+    // Get auth type from config, default to QWEN_OAUTH if not available
+    const authType = config?.getAuthType() ?? AuthType.QWEN_OAUTH;
+    // Get available models based on auth type
+    const availableModels = useMemo(() => getAvailableModelsForAuthType(authType), [authType]);
+    const MODEL_OPTIONS = useMemo(() => availableModels.map((model) => ({
+        value: model.id,
+        title: model.label,
+        description: model.description || '',
+        key: model.id,
+    })), [availableModels]);
+    // Determine the Preferred Model (read once when the dialog opens).
+    const preferredModel = config?.getModel() || MAINLINE_CODER;
+    useKeypress((key) => {
+        if (key.name === 'escape') {
+            onClose();
+        }
+    }, { isActive: true });
+    // Calculate the initial index based on the preferred model.
+    const initialIndex = useMemo(() => MODEL_OPTIONS.findIndex((option) => option.value === preferredModel), [MODEL_OPTIONS, preferredModel]);
+    // Handle selection internally (Autonomous Dialog).
+    const handleSelect = useCallback((model) => {
+        if (config) {
+            config.setModel(model);
+            const event = new ModelSlashCommandEvent(model);
+            logModelSlashCommand(config, event);
+        }
+        onClose();
+    }, [config, onClose]);
+    return (_jsxs(Box, { borderStyle: "round", borderColor: theme.border.default, flexDirection: "column", padding: 1, width: "100%", children: [_jsx(Text, { bold: true, children: "Select Model" }), _jsx(Box, { marginTop: 1, children: _jsx(DescriptiveRadioButtonSelect, { items: MODEL_OPTIONS, onSelect: handleSelect, initialIndex: initialIndex, showNumbers: true }) }), _jsx(Box, { marginTop: 1, flexDirection: "column", children: _jsx(Text, { color: theme.text.secondary, children: "(Press Esc to close)" }) })] }));
+}
+//# sourceMappingURL=ModelDialog.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelDialog.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelDialog.tsx"],"names":[],"mappings":";AAOA,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EACL,QAAQ,EACR,sBAAsB,EACtB,oBAAoB,GACrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAC9C,OAAO,EAAE,4BAA4B,EAAE,MAAM,0CAA0C,CAAC;AACxF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EACL,6BAA6B,EAC7B,cAAc,GACf,MAAM,8BAA8B,CAAC;AAMtC,MAAM,UAAU,WAAW,CAAC,EAAE,OAAO,EAAoB;IACvD,MAAM,MAAM,GAAG,UAAU,CAAC,aAAa,CAAC,CAAC;IAEzC,oEAAoE;IACpE,MAAM,QAAQ,GAAG,MAAM,EAAE,WAAW,EAAE,IAAI,QAAQ,CAAC,UAAU,CAAC;IAE9D,0CAA0C;IAC1C,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAG,EAAE,CAAC,6BAA6B,CAAC,QAAQ,CAAC,EAC7C,CAAC,QAAQ,CAAC,CACX,CAAC;IAEF,MAAM,aAAa,GAAG,OAAO,CAC3B,GAAG,EAAE,CACH,eAAe,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9B,KAAK,EAAE,KAAK,CAAC,EAAE;QACf,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,EAAE;QACpC,GAAG,EAAE,KAAK,CAAC,EAAE;KACd,CAAC,CAAC,EACL,CAAC,eAAe,CAAC,CAClB,CAAC;IAEF,mEAAmE;IACnE,MAAM,cAAc,GAAG,MAAM,EAAE,QAAQ,EAAE,IAAI,cAAc,CAAC;IAE5D,WAAW,CACT,CAAC,GAAG,EAAE,EAAE;QACN,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC,EACD,EAAE,QAAQ,EAAE,IAAI,EAAE,CACnB,CAAC;IAEF,4DAA4D;IAC5D,MAAM,YAAY,GAAG,OAAO,CAC1B,GAAG,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,KAAK,cAAc,CAAC,EAC1E,CAAC,aAAa,EAAE,cAAc,CAAC,CAChC,CAAC;IAEF,mDAAmD;IACnD,MAAM,YAAY,GAAG,WAAW,CAC9B,CAAC,KAAa,EAAE,EAAE;QAChB,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACvB,MAAM,KAAK,GAAG,IAAI,sBAAsB,CAAC,KAAK,CAAC,CAAC;YAChD,oBAAoB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,EACD,CAAC,MAAM,EAAE,OAAO,CAAC,CAClB,CAAC;IAEF,OAAO,CACL,MAAC,GAAG,IACF,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,EACjC,aAAa,EAAC,QAAQ,EACtB,OAAO,EAAE,CAAC,EACV,KAAK,EAAC,MAAM,aAEZ,KAAC,IAAI,IAAC,IAAI,mCAAoB,EAC9B,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,YACf,KAAC,4BAA4B,IAC3B,KAAK,EAAE,aAAa,EACpB,QAAQ,EAAE,YAAY,EACtB,YAAY,EAAE,YAAY,EAC1B,WAAW,EAAE,IAAI,GACjB,GACE,EACN,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,EAAE,aAAa,EAAC,QAAQ,YACvC,KAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,qCAA6B,GAC1D,IACF,CACP,CAAC;AACJ,CAAC"}

package/dist/src/ui/components/ModelDialog.test.d.ts

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

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

@@ -0,0 +1,158 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { render, cleanup } from '@testing-library/react';
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { ModelDialog } from './ModelDialog.js';
+import { useKeypress } from '../hooks/useKeypress.js';
+import { DescriptiveRadioButtonSelect } from './shared/DescriptiveRadioButtonSelect.js';
+import { ConfigContext } from '../contexts/ConfigContext.js';
+import { AVAILABLE_MODELS_QWEN, MAINLINE_CODER, MAINLINE_VLM, } from '../models/availableModels.js';
+vi.mock('../hooks/useKeypress.js', () => ({
+    useKeypress: vi.fn(),
+}));
+const mockedUseKeypress = vi.mocked(useKeypress);
+vi.mock('./shared/DescriptiveRadioButtonSelect.js', () => ({
+    DescriptiveRadioButtonSelect: vi.fn(() => null),
+}));
+const mockedSelect = vi.mocked(DescriptiveRadioButtonSelect);
+const renderComponent = (props = {}, contextValue = undefined) => {
+    const defaultProps = {
+        onClose: vi.fn(),
+    };
+    const combinedProps = { ...defaultProps, ...props };
+    const mockConfig = contextValue
+        ? {
+            // --- Functions used by ModelDialog ---
+            getModel: vi.fn(() => MAINLINE_CODER),
+            setModel: vi.fn(),
+            getAuthType: vi.fn(() => 'qwen-oauth'),
+            // --- Functions used by ClearcutLogger ---
+            getUsageStatisticsEnabled: vi.fn(() => true),
+            getSessionId: vi.fn(() => 'mock-session-id'),
+            getDebugMode: vi.fn(() => false),
+            getContentGeneratorConfig: vi.fn(() => ({ authType: 'mock' })),
+            getUseSmartEdit: vi.fn(() => false),
+            getUseModelRouter: vi.fn(() => false),
+            getProxy: vi.fn(() => undefined),
+            // --- Spread test-specific overrides ---
+            ...contextValue,
+        }
+        : undefined;
+    const renderResult = render(_jsx(ConfigContext.Provider, { value: mockConfig, children: _jsx(ModelDialog, { ...combinedProps }) }));
+    return {
+        ...renderResult,
+        props: combinedProps,
+        mockConfig,
+    };
+};
+describe('<ModelDialog />', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    afterEach(() => {
+        cleanup();
+    });
+    it('renders the title and help text', () => {
+        const { getByText } = renderComponent();
+        expect(getByText('Select Model')).toBeDefined();
+        expect(getByText('(Press Esc to close)')).toBeDefined();
+    });
+    it('passes all model options to DescriptiveRadioButtonSelect', () => {
+        renderComponent();
+        expect(mockedSelect).toHaveBeenCalledTimes(1);
+        const props = mockedSelect.mock.calls[0][0];
+        expect(props.items).toHaveLength(AVAILABLE_MODELS_QWEN.length);
+        expect(props.items[0].value).toBe(MAINLINE_CODER);
+        expect(props.items[1].value).toBe(MAINLINE_VLM);
+        expect(props.showNumbers).toBe(true);
+    });
+    it('initializes with the model from ConfigContext', () => {
+        const mockGetModel = vi.fn(() => MAINLINE_VLM);
+        renderComponent({}, { getModel: mockGetModel });
+        expect(mockGetModel).toHaveBeenCalled();
+        expect(mockedSelect).toHaveBeenCalledWith(expect.objectContaining({
+            initialIndex: 1,
+        }), undefined);
+    });
+    it('initializes with default coder model if context is not provided', () => {
+        renderComponent({}, undefined);
+        expect(mockedSelect).toHaveBeenCalledWith(expect.objectContaining({
+            initialIndex: 0,
+        }), undefined);
+    });
+    it('initializes with default coder model if getModel returns undefined', () => {
+        const mockGetModel = vi.fn(() => undefined);
+        // @ts-expect-error This test validates component robustness when getModel
+        // returns an unexpected undefined value.
+        renderComponent({}, { getModel: mockGetModel });
+        expect(mockGetModel).toHaveBeenCalled();
+        // When getModel returns undefined, preferredModel falls back to MAINLINE_CODER
+        // which has index 0, so initialIndex should be 0
+        expect(mockedSelect).toHaveBeenCalledWith(expect.objectContaining({
+            initialIndex: 0,
+        }), undefined);
+        expect(mockedSelect).toHaveBeenCalledTimes(1);
+    });
+    it('calls config.setModel and onClose when DescriptiveRadioButtonSelect.onSelect is triggered', () => {
+        const { props, mockConfig } = renderComponent({}, {}); // Pass empty object for contextValue
+        const childOnSelect = mockedSelect.mock.calls[0][0].onSelect;
+        expect(childOnSelect).toBeDefined();
+        childOnSelect(MAINLINE_CODER);
+        // Assert against the default mock provided by renderComponent
+        expect(mockConfig?.setModel).toHaveBeenCalledWith(MAINLINE_CODER);
+        expect(props.onClose).toHaveBeenCalledTimes(1);
+    });
+    it('does not pass onHighlight to DescriptiveRadioButtonSelect', () => {
+        renderComponent();
+        const childOnHighlight = mockedSelect.mock.calls[0][0].onHighlight;
+        expect(childOnHighlight).toBeUndefined();
+    });
+    it('calls onClose prop when "escape" key is pressed', () => {
+        const { props } = renderComponent();
+        expect(mockedUseKeypress).toHaveBeenCalled();
+        const keyPressHandler = mockedUseKeypress.mock.calls[0][0];
+        const options = mockedUseKeypress.mock.calls[0][1];
+        expect(options).toEqual({ isActive: true });
+        keyPressHandler({
+            name: 'escape',
+            ctrl: false,
+            meta: false,
+            shift: false,
+            paste: false,
+            sequence: '',
+        });
+        expect(props.onClose).toHaveBeenCalledTimes(1);
+        keyPressHandler({
+            name: 'a',
+            ctrl: false,
+            meta: false,
+            shift: false,
+            paste: false,
+            sequence: '',
+        });
+        expect(props.onClose).toHaveBeenCalledTimes(1);
+    });
+    it('updates initialIndex when config context changes', () => {
+        const mockGetModel = vi.fn(() => MAINLINE_CODER);
+        const mockGetAuthType = vi.fn(() => 'qwen-oauth');
+        const { rerender } = render(_jsx(ConfigContext.Provider, { value: {
+                getModel: mockGetModel,
+                getAuthType: mockGetAuthType,
+            }, children: _jsx(ModelDialog, { onClose: vi.fn() }) }));
+        expect(mockedSelect.mock.calls[0][0].initialIndex).toBe(0);
+        mockGetModel.mockReturnValue(MAINLINE_VLM);
+        const newMockConfig = {
+            getModel: mockGetModel,
+            getAuthType: mockGetAuthType,
+        };
+        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);
+        expect(mockedSelect.mock.calls[1][0].initialIndex).toBe(1);
+    });
+});
+//# sourceMappingURL=ModelDialog.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelDialog.test.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelDialog.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,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;AAE7D,OAAO,EACL,qBAAqB,EACrB,cAAc,EACd,YAAY,GACb,MAAM,8BAA8B,CAAC;AAEtC,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,cAAc,CAAC;YACrC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,WAAW,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,YAAY,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;YAEhC,yCAAyC;YACzC,GAAG,YAAY;SACM;QACzB,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,GAAG,eAAe,EAAE,CAAC;QACxC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAChD,MAAM,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,eAAe,EAAE,CAAC;QAClB,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,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;QAC/C,eAAe,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEhD,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;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,eAAe,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAE/B,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,YAAY,EAAE,CAAC;SAChB,CAAC,EACF,SAAS,CACV,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC5E,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5C,0EAA0E;QAC1E,yCAAyC;QACzC,eAAe,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEhD,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAExC,+EAA+E;QAC/E,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;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,GAAG,EAAE;QACnG,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,qCAAqC;QAE5F,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,cAAc,CAAC,CAAC;QAE9B,8DAA8D;QAC9D,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAClE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,eAAe,EAAE,CAAC;QAElB,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;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,EAAE,KAAK,EAAE,GAAG,eAAe,EAAE,CAAC;QAEpC,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,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,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,CAAC;QACjD,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,KAAC,aAAa,CAAC,QAAQ,IACrB,KAAK,EACH;gBACE,QAAQ,EAAE,YAAY;gBACtB,WAAW,EAAE,eAAe;aACR,YAGxB,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,YAAY,CAAC,CAAC;QAC3C,MAAM,aAAa,GAAG;YACpB,QAAQ,EAAE,YAAY;YACtB,WAAW,EAAE,eAAe;SACR,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;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -3,5 +3,5 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import React from 'react';
+import type React from 'react';
 export declare const ModelStatsDisplay: React.FC;

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

@@ -1,33 +1,33 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import { Box, Text } from 'ink';
-import { Colors } from '../colors.js';
+import { theme } from '../semantic-colors.js';
 import { formatDuration } from '../utils/formatters.js';
 import { calculateAverageLatency, calculateCacheHitRate, calculateErrorRate, } from '../utils/computeStats.js';
 import { useSessionStats } from '../contexts/SessionContext.js';
 const METRIC_COL_WIDTH = 28;
 const MODEL_COL_WIDTH = 22;
-const StatRow = ({ title, values, isSubtle = false, isSection = false, }) => (_jsxs(Box, { children: [_jsx(Box, { width: METRIC_COL_WIDTH, children: _jsx(Text, { bold: isSection, color: isSection ? undefined : Colors.LightBlue, children: isSubtle ? `  ↳ ${title}` : title }) }), values.map((value, index) => (_jsx(Box, { width: MODEL_COL_WIDTH, children: _jsx(Text, { children: value }) }, index)))] }));
+const StatRow = ({ title, values, isSubtle = false, isSection = false, }) => (_jsxs(Box, { children: [_jsx(Box, { width: METRIC_COL_WIDTH, children: _jsx(Text, { bold: isSection, color: isSection ? theme.text.primary : theme.text.link, children: isSubtle ? `  ↳ ${title}` : title }) }), values.map((value, index) => (_jsx(Box, { width: MODEL_COL_WIDTH, children: _jsx(Text, { color: theme.text.primary, children: value }) }, index)))] }));
 export const ModelStatsDisplay = () => {
     const { stats } = useSessionStats();
     const { models } = stats.metrics;
     const activeModels = Object.entries(models).filter(([, metrics]) => metrics.api.totalRequests > 0);
     if (activeModels.length === 0) {
-        return (_jsx(Box, { borderStyle: "round", borderColor: Colors.Gray, paddingY: 1, paddingX: 2, children: _jsx(Text, { children: "No API calls have been made in this session." }) }));
+        return (_jsx(Box, { borderStyle: "round", borderColor: theme.border.default, paddingY: 1, paddingX: 2, children: _jsx(Text, { color: theme.text.primary, children: "No API calls have been made in this session." }) }));
     }
     const modelNames = activeModels.map(([name]) => name);
     const getModelValues = (getter) => activeModels.map(([, metrics]) => getter(metrics));
     const hasThoughts = activeModels.some(([, metrics]) => metrics.tokens.thoughts > 0);
     const hasTool = activeModels.some(([, metrics]) => metrics.tokens.tool > 0);
     const hasCached = activeModels.some(([, metrics]) => metrics.tokens.cached > 0);
-    return (_jsxs(Box, { borderStyle: "round", borderColor: Colors.Gray, flexDirection: "column", paddingY: 1, paddingX: 2, children: [_jsx(Text, { bold: true, color: Colors.AccentPurple, children: "Model Stats For Nerds" }), _jsx(Box, { height: 1 }), _jsxs(Box, { children: [_jsx(Box, { width: METRIC_COL_WIDTH, children: _jsx(Text, { bold: true, children: "Metric" }) }), modelNames.map((name) => (_jsx(Box, { width: MODEL_COL_WIDTH, children: _jsx(Text, { bold: true, children: name }) }, name)))] }), _jsx(Box, { borderStyle: "single", borderBottom: true, borderTop: false, borderLeft: false, borderRight: false }), _jsx(StatRow, { title: "API", values: [], isSection: true }), _jsx(StatRow, { title: "Requests", values: getModelValues((m) => m.api.totalRequests.toLocaleString()) }), _jsx(StatRow, { title: "Errors", values: getModelValues((m) => {
+    return (_jsxs(Box, { borderStyle: "round", borderColor: theme.border.default, flexDirection: "column", paddingY: 1, paddingX: 2, children: [_jsx(Text, { bold: true, color: theme.text.accent, children: "Model Stats For Nerds" }), _jsx(Box, { height: 1 }), _jsxs(Box, { children: [_jsx(Box, { width: METRIC_COL_WIDTH, children: _jsx(Text, { bold: true, color: theme.text.primary, children: "Metric" }) }), modelNames.map((name) => (_jsx(Box, { width: MODEL_COL_WIDTH, children: _jsx(Text, { bold: true, color: theme.text.primary, children: name }) }, name)))] }), _jsx(Box, { borderStyle: "single", borderBottom: true, borderTop: false, borderLeft: false, borderRight: false, borderColor: theme.border.default }), _jsx(StatRow, { title: "API", values: [], isSection: true }), _jsx(StatRow, { title: "Requests", values: getModelValues((m) => m.api.totalRequests.toLocaleString()) }), _jsx(StatRow, { title: "Errors", values: getModelValues((m) => {
                     const errorRate = calculateErrorRate(m);
-                    return (_jsxs(Text, { color: m.api.totalErrors > 0 ? Colors.AccentRed : Colors.Foreground, children: [m.api.totalErrors.toLocaleString(), " (", errorRate.toFixed(1), "%)"] }));
+                    return (_jsxs(Text, { color: m.api.totalErrors > 0 ? theme.status.error : theme.text.primary, children: [m.api.totalErrors.toLocaleString(), " (", errorRate.toFixed(1), "%)"] }));
                 }) }), _jsx(StatRow, { title: "Avg Latency", values: getModelValues((m) => {
                     const avgLatency = calculateAverageLatency(m);
                     return formatDuration(avgLatency);
-                }) }), _jsx(Box, { height: 1 }), _jsx(StatRow, { title: "Tokens", values: [], isSection: true }), _jsx(StatRow, { title: "Total", values: getModelValues((m) => (_jsx(Text, { color: Colors.AccentYellow, children: m.tokens.total.toLocaleString() }))) }), _jsx(StatRow, { title: "Prompt", isSubtle: true, values: getModelValues((m) => m.tokens.prompt.toLocaleString()) }), hasCached && (_jsx(StatRow, { title: "Cached", isSubtle: true, values: getModelValues((m) => {
+                }) }), _jsx(Box, { height: 1 }), _jsx(StatRow, { title: "Tokens", values: [], isSection: true }), _jsx(StatRow, { title: "Total", values: getModelValues((m) => (_jsx(Text, { color: theme.status.warning, children: m.tokens.total.toLocaleString() }))) }), _jsx(StatRow, { title: "Prompt", isSubtle: true, values: getModelValues((m) => m.tokens.prompt.toLocaleString()) }), hasCached && (_jsx(StatRow, { title: "Cached", isSubtle: true, values: getModelValues((m) => {
                     const cacheHitRate = calculateCacheHitRate(m);
-                    return (_jsxs(Text, { color: Colors.AccentGreen, children: [m.tokens.cached.toLocaleString(), " (", cacheHitRate.toFixed(1), "%)"] }));
+                    return (_jsxs(Text, { color: theme.status.success, children: [m.tokens.cached.toLocaleString(), " (", cacheHitRate.toFixed(1), "%)"] }));
                 }) })), hasThoughts && (_jsx(StatRow, { title: "Thoughts", isSubtle: true, values: getModelValues((m) => m.tokens.thoughts.toLocaleString()) })), hasTool && (_jsx(StatRow, { title: "Tool", isSubtle: true, values: getModelValues((m) => m.tokens.tool.toLocaleString()) })), _jsx(StatRow, { title: "Output", isSubtle: true, values: getModelValues((m) => m.tokens.candidates.toLocaleString()) })] }));
 };
 //# sourceMappingURL=ModelStatsDisplay.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"ModelStatsDisplay.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelStatsDisplay.tsx"],"names":[],"mappings":";AAOA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EACL,uBAAuB,EACvB,qBAAqB,EACrB,kBAAkB,GACnB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,eAAe,EAAgB,MAAM,+BAA+B,CAAC;AAE9E,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAC5B,MAAM,eAAe,GAAG,EAAE,CAAC;AAS3B,MAAM,OAAO,GAA2B,CAAC,EACvC,KAAK,EACL,MAAM,EACN,QAAQ,GAAG,KAAK,EAChB,SAAS,GAAG,KAAK,GAClB,EAAE,EAAE,CAAC,CACJ,MAAC,GAAG,eACF,KAAC,GAAG,IAAC,KAAK,EAAE,gBAAgB,YAC1B,KAAC,IAAI,IAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,YACnE,QAAQ,CAAC,CAAC,CAAC,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,GAC7B,GACH,EACL,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAC5B,KAAC,GAAG,IAAC,KAAK,EAAE,eAAe,YACzB,KAAC,IAAI,cAAE,KAAK,GAAQ,IADY,KAAK,CAEjC,CACP,CAAC,IACE,CACP,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAa,GAAG,EAAE;IAC9C,MAAM,EAAE,KAAK,EAAE,GAAG,eAAe,EAAE,CAAC;IACpC,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC;IACjC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,CAChD,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,CAAC,CAC/C,CAAC;IAEF,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,CACL,KAAC,GAAG,IACF,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,MAAM,CAAC,IAAI,EACxB,QAAQ,EAAE,CAAC,EACX,QAAQ,EAAE,CAAC,YAEX,KAAC,IAAI,+DAAoD,GACrD,CACP,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;IAEtD,MAAM,cAAc,GAAG,CACrB,MAA8D,EAC9D,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAExD,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CACnC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,GAAG,CAAC,CAC7C,CAAC;IACF,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;IAC5E,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CACjC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAC3C,CAAC;IAEF,OAAO,CACL,MAAC,GAAG,IACF,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,MAAM,CAAC,IAAI,EACxB,aAAa,EAAC,QAAQ,EACtB,QAAQ,EAAE,CAAC,EACX,QAAQ,EAAE,CAAC,aAEX,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAE,MAAM,CAAC,YAAY,sCAE9B,EACP,KAAC,GAAG,IAAC,MAAM,EAAE,CAAC,GAAI,EAGlB,MAAC,GAAG,eACF,KAAC,GAAG,IAAC,KAAK,EAAE,gBAAgB,YAC1B,KAAC,IAAI,IAAC,IAAI,6BAAc,GACpB,EACL,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CACxB,KAAC,GAAG,IAAC,KAAK,EAAE,eAAe,YACzB,KAAC,IAAI,IAAC,IAAI,kBAAE,IAAI,GAAQ,IADQ,IAAI,CAEhC,CACP,CAAC,IACE,EAGN,KAAC,GAAG,IACF,WAAW,EAAC,QAAQ,EACpB,YAAY,EAAE,IAAI,EAClB,SAAS,EAAE,KAAK,EAChB,UAAU,EAAE,KAAK,EACjB,WAAW,EAAE,KAAK,GAClB,EAGF,KAAC,OAAO,IAAC,KAAK,EAAC,KAAK,EAAC,MAAM,EAAE,EAAE,EAAE,SAAS,SAAG,EAC7C,KAAC,OAAO,IACN,KAAK,EAAC,UAAU,EAChB,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,cAAc,EAAE,CAAC,GACnE,EACF,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC3B,MAAM,SAAS,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC;oBACxC,OAAO,CACL,MAAC,IAAI,IACH,KAAK,EACH,CAAC,CAAC,GAAG,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,aAG7D,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,QAAI,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,UACtD,CACR,CAAC;gBACJ,CAAC,CAAC,GACF,EACF,KAAC,OAAO,IACN,KAAK,EAAC,aAAa,EACnB,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC3B,MAAM,UAAU,GAAG,uBAAuB,CAAC,CAAC,CAAC,CAAC;oBAC9C,OAAO,cAAc,CAAC,UAAU,CAAC,CAAC;gBACpC,CAAC,CAAC,GACF,EAEF,KAAC,GAAG,IAAC,MAAM,EAAE,CAAC,GAAI,EAGlB,KAAC,OAAO,IAAC,KAAK,EAAC,QAAQ,EAAC,MAAM,EAAE,EAAE,EAAE,SAAS,SAAG,EAChD,KAAC,OAAO,IACN,KAAK,EAAC,OAAO,EACb,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAC5B,KAAC,IAAI,IAAC,KAAK,EAAE,MAAM,CAAC,YAAY,YAC7B,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE,GAC3B,CACR,CAAC,GACF,EACF,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,GAC/D,EACD,SAAS,IAAI,CACZ,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC3B,MAAM,YAAY,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;oBAC9C,OAAO,CACL,MAAC,IAAI,IAAC,KAAK,EAAE,MAAM,CAAC,WAAW,aAC5B,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,QAAI,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,UACvD,CACR,CAAC;gBACJ,CAAC,CAAC,GACF,CACH,EACA,WAAW,IAAI,CACd,KAAC,OAAO,IACN,KAAK,EAAC,UAAU,EAChB,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC,GACjE,CACH,EACA,OAAO,IAAI,CACV,KAAC,OAAO,IACN,KAAK,EAAC,MAAM,EACZ,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,GAC7D,CACH,EACD,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,GACnE,IACE,CACP,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"ModelStatsDisplay.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelStatsDisplay.tsx"],"names":[],"mappings":";AAOA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EACL,uBAAuB,EACvB,qBAAqB,EACrB,kBAAkB,GACnB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAEhE,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAC5B,MAAM,eAAe,GAAG,EAAE,CAAC;AAS3B,MAAM,OAAO,GAA2B,CAAC,EACvC,KAAK,EACL,MAAM,EACN,QAAQ,GAAG,KAAK,EAChB,SAAS,GAAG,KAAK,GAClB,EAAE,EAAE,CAAC,CACJ,MAAC,GAAG,eACF,KAAC,GAAG,IAAC,KAAK,EAAE,gBAAgB,YAC1B,KAAC,IAAI,IACH,IAAI,EAAE,SAAS,EACf,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,YAEtD,QAAQ,CAAC,CAAC,CAAC,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,GAC7B,GACH,EACL,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAC5B,KAAC,GAAG,IAAC,KAAK,EAAE,eAAe,YACzB,KAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,YAAG,KAAK,GAAQ,IADf,KAAK,CAEjC,CACP,CAAC,IACE,CACP,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAa,GAAG,EAAE;IAC9C,MAAM,EAAE,KAAK,EAAE,GAAG,eAAe,EAAE,CAAC;IACpC,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC;IACjC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,CAChD,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,CAAC,CAC/C,CAAC;IAEF,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,CACL,KAAC,GAAG,IACF,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,EACjC,QAAQ,EAAE,CAAC,EACX,QAAQ,EAAE,CAAC,YAEX,KAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,6DAExB,GACH,CACP,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;IAEtD,MAAM,cAAc,GAAG,CACrB,MAA8D,EAC9D,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAExD,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CACnC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,GAAG,CAAC,CAC7C,CAAC;IACF,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;IAC5E,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CACjC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAC3C,CAAC;IAEF,OAAO,CACL,MAAC,GAAG,IACF,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,EACjC,aAAa,EAAC,QAAQ,EACtB,QAAQ,EAAE,CAAC,EACX,QAAQ,EAAE,CAAC,aAEX,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,sCAE5B,EACP,KAAC,GAAG,IAAC,MAAM,EAAE,CAAC,GAAI,EAGlB,MAAC,GAAG,eACF,KAAC,GAAG,IAAC,KAAK,EAAE,gBAAgB,YAC1B,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,uBAE7B,GACH,EACL,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CACxB,KAAC,GAAG,IAAC,KAAK,EAAE,eAAe,YACzB,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,YACjC,IAAI,GACA,IAHyB,IAAI,CAIhC,CACP,CAAC,IACE,EAGN,KAAC,GAAG,IACF,WAAW,EAAC,QAAQ,EACpB,YAAY,EAAE,IAAI,EAClB,SAAS,EAAE,KAAK,EAChB,UAAU,EAAE,KAAK,EACjB,WAAW,EAAE,KAAK,EAClB,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,GACjC,EAGF,KAAC,OAAO,IAAC,KAAK,EAAC,KAAK,EAAC,MAAM,EAAE,EAAE,EAAE,SAAS,SAAG,EAC7C,KAAC,OAAO,IACN,KAAK,EAAC,UAAU,EAChB,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,cAAc,EAAE,CAAC,GACnE,EACF,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC3B,MAAM,SAAS,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC;oBACxC,OAAO,CACL,MAAC,IAAI,IACH,KAAK,EACH,CAAC,CAAC,GAAG,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,aAGhE,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,QAAI,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,UACtD,CACR,CAAC;gBACJ,CAAC,CAAC,GACF,EACF,KAAC,OAAO,IACN,KAAK,EAAC,aAAa,EACnB,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC3B,MAAM,UAAU,GAAG,uBAAuB,CAAC,CAAC,CAAC,CAAC;oBAC9C,OAAO,cAAc,CAAC,UAAU,CAAC,CAAC;gBACpC,CAAC,CAAC,GACF,EAEF,KAAC,GAAG,IAAC,MAAM,EAAE,CAAC,GAAI,EAGlB,KAAC,OAAO,IAAC,KAAK,EAAC,QAAQ,EAAC,MAAM,EAAE,EAAE,EAAE,SAAS,SAAG,EAChD,KAAC,OAAO,IACN,KAAK,EAAC,OAAO,EACb,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAC5B,KAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,YAC9B,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE,GAC3B,CACR,CAAC,GACF,EACF,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,GAC/D,EACD,SAAS,IAAI,CACZ,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC3B,MAAM,YAAY,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;oBAC9C,OAAO,CACL,MAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,aAC9B,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,QAAI,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,UACvD,CACR,CAAC;gBACJ,CAAC,CAAC,GACF,CACH,EACA,WAAW,IAAI,CACd,KAAC,OAAO,IACN,KAAK,EAAC,UAAU,EAChB,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC,GACjE,CACH,EACA,OAAO,IAAI,CACV,KAAC,OAAO,IACN,KAAK,EAAC,MAAM,EACZ,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,GAC7D,CACH,EACD,KAAC,OAAO,IACN,KAAK,EAAC,QAAQ,EACd,QAAQ,QACR,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,GACnE,IACE,CACP,CAAC;AACJ,CAAC,CAAC"}

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

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type React from 'react';
+import { type AuthType } from '@umsai/ums-code-core';
+export declare enum VisionSwitchOutcome {
+    SwitchOnce = "once",
+    SwitchSessionToVL = "session",
+    ContinueWithCurrentModel = "persist"
+}
+export interface ModelSwitchDialogProps {
+    onSelect: (outcome: VisionSwitchOutcome, authType: AuthType) => void;
+    authType: AuthType;
+}
+export declare const ModelSwitchDialog: React.FC<ModelSwitchDialogProps>;

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

@@ -0,0 +1,41 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { Box, Text } from 'ink';
+import { Colors } from '../colors.js';
+import { RadioButtonSelect, } from './shared/RadioButtonSelect.js';
+import { useKeypress } from '../hooks/useKeypress.js';
+import {} from '@umsai/ums-code-core';
+export var VisionSwitchOutcome;
+(function (VisionSwitchOutcome) {
+    VisionSwitchOutcome["SwitchOnce"] = "once";
+    VisionSwitchOutcome["SwitchSessionToVL"] = "session";
+    VisionSwitchOutcome["ContinueWithCurrentModel"] = "persist";
+})(VisionSwitchOutcome || (VisionSwitchOutcome = {}));
+export const ModelSwitchDialog = ({ onSelect, authType, }) => {
+    useKeypress((key) => {
+        if (key.name === 'escape') {
+            onSelect(VisionSwitchOutcome.ContinueWithCurrentModel, authType);
+        }
+    }, { isActive: true });
+    const options = [
+        {
+            key: 'switch-once',
+            label: '只切换当前这一次请求(临时)',
+            value: VisionSwitchOutcome.SwitchOnce,
+        },
+        {
+            key: 'switch-session',
+            label: '切换当前会话默认模型为视觉模型(会话)',
+            value: VisionSwitchOutcome.SwitchSessionToVL,
+        },
+        {
+            key: 'continue',
+            label: '使用当前模型继续(不切换)',
+            value: VisionSwitchOutcome.ContinueWithCurrentModel,
+        },
+    ];
+    const handleSelect = (outcome) => {
+        onSelect(outcome, authType);
+    };
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: Colors.AccentYellow, padding: 1, width: "100%", marginLeft: 1, children: [_jsxs(Box, { flexDirection: "column", marginBottom: 1, children: [_jsx(Text, { bold: true, children: "\u5207\u6362\u4F7F\u7528\u89C6\u89C9\u6A21\u578B" }), _jsx(Text, { children: "\u4E0A\u4E0B\u6587\u4E2D\u5305\u542B\u56FE\u7247\uFF0C\u4F46\u662F\u76EE\u524D\u4F7F\u7528\u7684\u6A21\u578B\u4E0D\u652F\u6301\u56FE\u7247\u3002" }), _jsx(Text, { children: "\u8BF7\u786E\u8BA4\u662F\u5426\u7EE7\u7EED?" })] }), _jsx(Box, { marginBottom: 1, children: _jsx(RadioButtonSelect, { items: options, initialIndex: 0, onSelect: handleSelect, isFocused: true }) }), _jsx(Box, { children: _jsx(Text, { color: Colors.Gray, children: "\u6309 Enter \u952E\u786E\u8BA4, Esc \u952E\u9000\u51FA" }) })] }));
+};
+//# sourceMappingURL=ModelSwitchDialog.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelSwitchDialog.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelSwitchDialog.tsx"],"names":[],"mappings":";AAOA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EACL,iBAAiB,GAElB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAiB,MAAM,sBAAsB,CAAC;AAErD,MAAM,CAAN,IAAY,mBAIX;AAJD,WAAY,mBAAmB;IAC7B,0CAAmB,CAAA;IACnB,oDAA6B,CAAA;IAC7B,2DAAoC,CAAA;AACtC,CAAC,EAJW,mBAAmB,KAAnB,mBAAmB,QAI9B;AAOD,MAAM,CAAC,MAAM,iBAAiB,GAAqC,CAAC,EAClE,QAAQ,EACR,QAAQ,GACT,EAAE,EAAE;IACH,WAAW,CACT,CAAC,GAAG,EAAE,EAAE;QACN,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,QAAQ,CAAC,mBAAmB,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC;QACnE,CAAC;IACH,CAAC,EACD,EAAE,QAAQ,EAAE,IAAI,EAAE,CACnB,CAAC;IAEF,MAAM,OAAO,GAAgD;QAC3D;YACE,GAAG,EAAE,aAAa;YAClB,KAAK,EAAE,gBAAgB;YACvB,KAAK,EAAE,mBAAmB,CAAC,UAAU;SACtC;QACD;YACE,GAAG,EAAE,gBAAgB;YACrB,KAAK,EAAE,qBAAqB;YAC5B,KAAK,EAAE,mBAAmB,CAAC,iBAAiB;SAC7C;QACD;YACE,GAAG,EAAE,UAAU;YACf,KAAK,EAAE,eAAe;YACtB,KAAK,EAAE,mBAAmB,CAAC,wBAAwB;SACpD;KACF,CAAC;IAEF,MAAM,YAAY,GAAG,CAAC,OAA4B,EAAE,EAAE;QACpD,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC9B,CAAC,CAAC;IAEF,OAAO,CACL,MAAC,GAAG,IACF,aAAa,EAAC,QAAQ,EACtB,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,MAAM,CAAC,YAAY,EAChC,OAAO,EAAE,CAAC,EACV,KAAK,EAAC,MAAM,EACZ,UAAU,EAAE,CAAC,aAEb,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,EAAC,YAAY,EAAE,CAAC,aACzC,KAAC,IAAI,IAAC,IAAI,uEAAgB,EAC1B,KAAC,IAAI,mKAEE,EACP,KAAC,IAAI,8DAAgB,IACjB,EAEN,KAAC,GAAG,IAAC,YAAY,EAAE,CAAC,YAClB,KAAC,iBAAiB,IAChB,KAAK,EAAE,OAAO,EACd,YAAY,EAAE,CAAC,EACf,QAAQ,EAAE,YAAY,EACtB,SAAS,SACT,GACE,EAEN,KAAC,GAAG,cACF,KAAC,IAAI,IAAC,KAAK,EAAE,MAAM,CAAC,IAAI,wEAA6B,GACjD,IACF,CACP,CAAC;AACJ,CAAC,CAAC"}

package/dist/src/ui/components/ModelSwitchDialog.test.d.ts

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

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

@@ -0,0 +1,129 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import React from 'react';
+import { render } from 'ink-testing-library';
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { ModelSwitchDialog, VisionSwitchOutcome } from './ModelSwitchDialog.js';
+import { AuthType } from '@umsai/ums-code-core';
+// Mock the useKeypress hook
+const mockUseKeypress = vi.hoisted(() => vi.fn());
+vi.mock('../hooks/useKeypress.js', () => ({
+    useKeypress: mockUseKeypress,
+}));
+// Mock the RadioButtonSelect component
+const mockRadioButtonSelect = vi.hoisted(() => vi.fn());
+vi.mock('./shared/RadioButtonSelect.js', () => ({
+    RadioButtonSelect: mockRadioButtonSelect,
+}));
+describe('ModelSwitchDialog', () => {
+    const mockOnSelect = vi.fn();
+    beforeEach(() => {
+        vi.clearAllMocks();
+        // Mock RadioButtonSelect to return a simple div
+        mockRadioButtonSelect.mockReturnValue(React.createElement('div', { 'data-testid': 'radio-select' }));
+    });
+    it('should setup RadioButtonSelect with correct options', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const expectedItems = [
+            {
+                key: 'switch-once',
+                label: 'Switch for this request only',
+                value: VisionSwitchOutcome.SwitchOnce,
+            },
+            {
+                key: 'switch-session',
+                label: 'Switch session to vision model',
+                value: VisionSwitchOutcome.SwitchSessionToVL,
+            },
+            {
+                key: 'continue',
+                label: 'Continue with current model',
+                value: VisionSwitchOutcome.ContinueWithCurrentModel,
+            },
+        ];
+        const callArgs = mockRadioButtonSelect.mock.calls[0][0];
+        expect(callArgs.items).toEqual(expectedItems);
+        expect(callArgs.initialIndex).toBe(0);
+        expect(callArgs.isFocused).toBe(true);
+    });
+    it('should call onSelect when an option is selected', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const callArgs = mockRadioButtonSelect.mock.calls[0][0];
+        expect(typeof callArgs.onSelect).toBe('function');
+        // Simulate selection of "Switch for this request only"
+        const onSelectCallback = mockRadioButtonSelect.mock.calls[0][0].onSelect;
+        onSelectCallback(VisionSwitchOutcome.SwitchOnce);
+        expect(mockOnSelect).toHaveBeenCalledWith(VisionSwitchOutcome.SwitchOnce);
+    });
+    it('should call onSelect with SwitchSessionToVL when second option is selected', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const onSelectCallback = mockRadioButtonSelect.mock.calls[0][0].onSelect;
+        onSelectCallback(VisionSwitchOutcome.SwitchSessionToVL);
+        expect(mockOnSelect).toHaveBeenCalledWith(VisionSwitchOutcome.SwitchSessionToVL);
+    });
+    it('should call onSelect with ContinueWithCurrentModel when third option is selected', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const onSelectCallback = mockRadioButtonSelect.mock.calls[0][0].onSelect;
+        onSelectCallback(VisionSwitchOutcome.ContinueWithCurrentModel);
+        expect(mockOnSelect).toHaveBeenCalledWith(VisionSwitchOutcome.ContinueWithCurrentModel);
+    });
+    it('should setup escape key handler to call onSelect with ContinueWithCurrentModel', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        expect(mockUseKeypress).toHaveBeenCalledWith(expect.any(Function), {
+            isActive: true,
+        });
+        // Simulate escape key press
+        const keypressHandler = mockUseKeypress.mock.calls[0][0];
+        keypressHandler({ name: 'escape' });
+        expect(mockOnSelect).toHaveBeenCalledWith(VisionSwitchOutcome.ContinueWithCurrentModel);
+    });
+    it('should not call onSelect for non-escape keys', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const keypressHandler = mockUseKeypress.mock.calls[0][0];
+        keypressHandler({ name: 'enter' });
+        expect(mockOnSelect).not.toHaveBeenCalled();
+    });
+    it('should set initial index to 0 (first option)', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const callArgs = mockRadioButtonSelect.mock.calls[0][0];
+        expect(callArgs.initialIndex).toBe(0);
+    });
+    describe('VisionSwitchOutcome enum', () => {
+        it('should have correct enum values', () => {
+            expect(VisionSwitchOutcome.SwitchOnce).toBe('once');
+            expect(VisionSwitchOutcome.SwitchSessionToVL).toBe('session');
+            expect(VisionSwitchOutcome.ContinueWithCurrentModel).toBe('persist');
+        });
+    });
+    it('should handle multiple onSelect calls correctly', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const onSelectCallback = mockRadioButtonSelect.mock.calls[0][0].onSelect;
+        // Call multiple times
+        onSelectCallback(VisionSwitchOutcome.SwitchOnce);
+        onSelectCallback(VisionSwitchOutcome.SwitchSessionToVL);
+        onSelectCallback(VisionSwitchOutcome.ContinueWithCurrentModel);
+        expect(mockOnSelect).toHaveBeenCalledTimes(3);
+        expect(mockOnSelect).toHaveBeenNthCalledWith(1, VisionSwitchOutcome.SwitchOnce);
+        expect(mockOnSelect).toHaveBeenNthCalledWith(2, VisionSwitchOutcome.SwitchSessionToVL);
+        expect(mockOnSelect).toHaveBeenNthCalledWith(3, VisionSwitchOutcome.ContinueWithCurrentModel);
+    });
+    it('should pass isFocused prop to RadioButtonSelect', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const callArgs = mockRadioButtonSelect.mock.calls[0][0];
+        expect(callArgs.isFocused).toBe(true);
+    });
+    it('should handle escape key multiple times', () => {
+        render(_jsx(ModelSwitchDialog, { onSelect: mockOnSelect, authType: AuthType.UMS_OAUTH }));
+        const keypressHandler = mockUseKeypress.mock.calls[0][0];
+        // Call escape multiple times
+        keypressHandler({ name: 'escape' });
+        keypressHandler({ name: 'escape' });
+        expect(mockOnSelect).toHaveBeenCalledTimes(2);
+        expect(mockOnSelect).toHaveBeenCalledWith(VisionSwitchOutcome.ContinueWithCurrentModel);
+    });
+});
+//# sourceMappingURL=ModelSwitchDialog.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelSwitchDialog.test.js","sourceRoot":"","sources":["../../../../src/ui/components/ModelSwitchDialog.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAChF,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEhD,4BAA4B;AAC5B,MAAM,eAAe,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAClD,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE,eAAe;CAC7B,CAAC,CAAC,CAAC;AAEJ,uCAAuC;AACvC,MAAM,qBAAqB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACxD,EAAE,CAAC,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,iBAAiB,EAAE,qBAAqB;CACzC,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,gDAAgD;QAChD,qBAAqB,CAAC,eAAe,CACnC,KAAK,CAAC,aAAa,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC,CAC9D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,aAAa,GAAG;YACpB;gBACE,GAAG,EAAE,aAAa;gBAClB,KAAK,EAAE,8BAA8B;gBACrC,KAAK,EAAE,mBAAmB,CAAC,UAAU;aACtC;YACD;gBACE,GAAG,EAAE,gBAAgB;gBACrB,KAAK,EAAE,gCAAgC;gBACvC,KAAK,EAAE,mBAAmB,CAAC,iBAAiB;aAC7C;YACD;gBACE,GAAG,EAAE,UAAU;gBACf,KAAK,EAAE,6BAA6B;gBACpC,KAAK,EAAE,mBAAmB,CAAC,wBAAwB;aACpD;SACF,CAAC;QAEF,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,OAAO,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAElD,uDAAuD;QACvD,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QACzE,gBAAgB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAEjD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;QACpF,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QACzE,gBAAgB,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;QAExD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,mBAAmB,CAAC,iBAAiB,CACtC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;QAC1F,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QACzE,gBAAgB,CAAC,mBAAmB,CAAC,wBAAwB,CAAC,CAAC;QAE/D,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,mBAAmB,CAAC,wBAAwB,CAC7C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE;YACjE,QAAQ,EAAE,IAAI;SACf,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,eAAe,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACzD,eAAe,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAEpC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,mBAAmB,CAAC,wBAAwB,CAC7C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,eAAe,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACzD,eAAe,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnC,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpD,MAAM,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9D,MAAM,CAAC,mBAAmB,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAEzE,sBAAsB;QACtB,gBAAgB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;QACjD,gBAAgB,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;QACxD,gBAAgB,CAAC,mBAAmB,CAAC,wBAAwB,CAAC,CAAC;QAE/D,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,YAAY,CAAC,CAAC,uBAAuB,CAC1C,CAAC,EACD,mBAAmB,CAAC,UAAU,CAC/B,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,uBAAuB,CAC1C,CAAC,EACD,mBAAmB,CAAC,iBAAiB,CACtC,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,uBAAuB,CAC1C,CAAC,EACD,mBAAmB,CAAC,wBAAwB,CAC7C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,KAAC,iBAAiB,IAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,GAAI,CAAC,CAAC;QAEpF,MAAM,eAAe,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAEzD,6BAA6B;QAC7B,eAAe,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpC,eAAe,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAEpC,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,mBAAmB,CAAC,wBAAwB,CAC7C,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const Notifications: () => import("react/jsx-runtime").JSX.Element | null;

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

@@ -0,0 +1,23 @@
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Box, Text } from 'ink';
+import { useAppContext } from '../contexts/AppContext.js';
+import { useUIState } from '../contexts/UIStateContext.js';
+import { theme } from '../semantic-colors.js';
+import { StreamingState } from '../types.js';
+import { UpdateNotification } from './UpdateNotification.js';
+export const Notifications = () => {
+    const { startupWarnings } = useAppContext();
+    const { initError, streamingState, updateInfo } = useUIState();
+    const showStartupWarnings = startupWarnings.length > 0;
+    const showInitError = initError && streamingState !== StreamingState.Responding;
+    if (!showStartupWarnings && !showInitError && !updateInfo) {
+        return null;
+    }
+    return (_jsxs(_Fragment, { children: [updateInfo && _jsx(UpdateNotification, { message: updateInfo.message }), showStartupWarnings && (_jsx(Box, { borderStyle: "round", borderColor: theme.status.warning, paddingX: 1, marginY: 1, flexDirection: "column", children: startupWarnings.map((warning, index) => (_jsx(Text, { color: theme.status.warning, children: warning }, index))) })), showInitError && (_jsxs(Box, { borderStyle: "round", borderColor: theme.status.error, paddingX: 1, marginBottom: 1, children: [_jsxs(Text, { color: theme.status.error, children: ["Initialization Error: ", initError] }), _jsxs(Text, { color: theme.status.error, children: [' ', "Please check API key and configuration."] })] }))] }));
+};
+//# sourceMappingURL=Notifications.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"Notifications.js","sourceRoot":"","sources":["../../../../src/ui/components/Notifications.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,+BAA+B,CAAC;AAC3D,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,MAAM,CAAC,MAAM,aAAa,GAAG,GAAG,EAAE;IAChC,MAAM,EAAE,eAAe,EAAE,GAAG,aAAa,EAAE,CAAC;IAC5C,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,UAAU,EAAE,GAAG,UAAU,EAAE,CAAC;IAE/D,MAAM,mBAAmB,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;IACvD,MAAM,aAAa,GACjB,SAAS,IAAI,cAAc,KAAK,cAAc,CAAC,UAAU,CAAC;IAE5D,IAAI,CAAC,mBAAmB,IAAI,CAAC,aAAa,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CACL,8BACG,UAAU,IAAI,KAAC,kBAAkB,IAAC,OAAO,EAAE,UAAU,CAAC,OAAO,GAAI,EACjE,mBAAmB,IAAI,CACtB,KAAC,GAAG,IACF,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,EACjC,QAAQ,EAAE,CAAC,EACX,OAAO,EAAE,CAAC,EACV,aAAa,EAAC,QAAQ,YAErB,eAAe,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,CACvC,KAAC,IAAI,IAAa,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,YAC1C,OAAO,IADC,KAAK,CAET,CACR,CAAC,GACE,CACP,EACA,aAAa,IAAI,CAChB,MAAC,GAAG,IACF,WAAW,EAAC,OAAO,EACnB,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,KAAK,EAC/B,QAAQ,EAAE,CAAC,EACX,YAAY,EAAE,CAAC,aAEf,MAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,KAAK,uCACN,SAAS,IAC3B,EACP,MAAC,IAAI,IAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,KAAK,aAC5B,GAAG,+CAEC,IACH,CACP,IACA,CACJ,CAAC;AACJ,CAAC,CAAC"}

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

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import React from 'react';
+import type React from 'react';
 interface OpenAIKeyPromptProps {
     onSubmit: (apiKey: string, baseUrl: string, model: string) => void;
     onCancel: () => void;