@didim365/agent-cli-core 0.1.0

package/dist/src/code_assist/experiments/client_metadata.test.js

@@ -0,0 +1,96 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { ReleaseChannel, getReleaseChannel } from '../../utils/channel.js';
+import { getVersion } from '../../utils/version.js';
+// Mock dependencies before importing the module under test
+vi.mock('../../utils/channel.js', async () => {
+    const actual = await vi.importActual('../../utils/channel.js');
+    return {
+        ...actual,
+        getReleaseChannel: vi.fn(),
+    };
+});
+vi.mock('../../utils/version.js', async () => ({
+    getVersion: vi.fn(),
+}));
+describe('client_metadata', () => {
+    const originalPlatform = process.platform;
+    const originalArch = process.arch;
+    const originalCliVersion = process.env['CLI_VERSION'];
+    const originalNodeVersion = process.version;
+    beforeEach(async () => {
+        // Reset modules to clear the cached `clientMetadataPromise`
+        vi.resetModules();
+        // Re-import the module to get a fresh instance
+        await import('./client_metadata.js');
+        // Provide a default mock implementation for each test
+        vi.mocked(getReleaseChannel).mockResolvedValue(ReleaseChannel.STABLE);
+        vi.mocked(getVersion).mockResolvedValue('0.0.0');
+    });
+    afterEach(() => {
+        // Restore original process properties to avoid side-effects between tests
+        Object.defineProperty(process, 'platform', { value: originalPlatform });
+        Object.defineProperty(process, 'arch', { value: originalArch });
+        process.env['CLI_VERSION'] = originalCliVersion;
+        Object.defineProperty(process, 'version', { value: originalNodeVersion });
+        vi.clearAllMocks();
+    });
+    describe('getPlatform', () => {
+        const testCases = [
+            { platform: 'darwin', arch: 'x64', expected: 'DARWIN_AMD64' },
+            { platform: 'darwin', arch: 'arm64', expected: 'DARWIN_ARM64' },
+            { platform: 'linux', arch: 'x64', expected: 'LINUX_AMD64' },
+            { platform: 'linux', arch: 'arm64', expected: 'LINUX_ARM64' },
+            { platform: 'win32', arch: 'x64', expected: 'WINDOWS_AMD64' },
+            { platform: 'sunos', arch: 'x64', expected: 'PLATFORM_UNSPECIFIED' },
+            { platform: 'win32', arch: 'arm', expected: 'PLATFORM_UNSPECIFIED' },
+        ];
+        for (const { platform, arch, expected } of testCases) {
+            it(`should return ${expected} for platform ${platform} and arch ${arch}`, async () => {
+                Object.defineProperty(process, 'platform', { value: platform });
+                Object.defineProperty(process, 'arch', { value: arch });
+                const { getClientMetadata } = await import('./client_metadata.js');
+                const metadata = await getClientMetadata();
+                expect(metadata.platform).toBe(expected);
+            });
+        }
+    });
+    describe('getClientMetadata', () => {
+        it('should use version from getCliVersion for ideVersion', async () => {
+            vi.mocked(getVersion).mockResolvedValue('1.2.3');
+            const { getClientMetadata } = await import('./client_metadata.js');
+            const metadata = await getClientMetadata();
+            expect(metadata.ideVersion).toBe('1.2.3');
+        });
+        it('should call getReleaseChannel to get the update channel', async () => {
+            vi.mocked(getReleaseChannel).mockResolvedValue(ReleaseChannel.NIGHTLY);
+            const { getClientMetadata } = await import('./client_metadata.js');
+            const metadata = await getClientMetadata();
+            expect(metadata.updateChannel).toBe('nightly');
+            expect(getReleaseChannel).toHaveBeenCalled();
+        });
+        it('should cache the client metadata promise', async () => {
+            const { getClientMetadata } = await import('./client_metadata.js');
+            const firstCall = await getClientMetadata();
+            const secondCall = await getClientMetadata();
+            expect(firstCall).toBe(secondCall);
+            // Ensure the underlying functions are only called once
+            expect(getReleaseChannel).toHaveBeenCalledTimes(1);
+        });
+        it('should always return the IDE name as IDE_UNSPECIFIED', async () => {
+            const { getClientMetadata } = await import('./client_metadata.js');
+            const metadata = await getClientMetadata();
+            expect(metadata.ideName).toBe('IDE_UNSPECIFIED');
+        });
+        it('should always return the pluginType as GEMINI', async () => {
+            const { getClientMetadata } = await import('./client_metadata.js');
+            const metadata = await getClientMetadata();
+            expect(metadata.pluginType).toBe('GEMINI');
+        });
+    });
+});
+//# sourceMappingURL=client_metadata.test.js.map

package/dist/src/code_assist/experiments/client_metadata.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client_metadata.test.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/client_metadata.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3E,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAEpD,2DAA2D;AAC3D,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;IAC3C,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,YAAY,CAAC,wBAAwB,CAAC,CAAC;IAC/D,OAAO;QACL,GAAI,MAAiB;QACrB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC3B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;IAC7C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;CACpB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,gBAAgB,GAAG,OAAO,CAAC,QAAQ,CAAC;IAC1C,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAClC,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACtD,MAAM,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC;IAE5C,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,4DAA4D;QAC5D,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,+CAA+C;QAC/C,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QACrC,sDAAsD;QACtD,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACtE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,0EAA0E;QAC1E,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CAAC;QACxE,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAChE,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,GAAG,kBAAkB,CAAC;QAChD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;QAC1E,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,MAAM,SAAS,GAAG;YAChB,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE;YAC7D,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE;YAC/D,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAE;YAC3D,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE;YAC7D,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE;YAC7D,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,sBAAsB,EAAE;YACpE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,sBAAsB,EAAE;SACrE,CAAC;QAEF,KAAK,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,SAAS,EAAE,CAAC;YACrD,EAAE,CAAC,iBAAiB,QAAQ,iBAAiB,QAAQ,aAAa,IAAI,EAAE,EAAE,KAAK,IAAI,EAAE;gBACnF,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;gBAChE,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACxD,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;gBAEnE,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;gBAC3C,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAEnE,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;YAC3C,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACvE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAEnE,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;YAE3C,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/C,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAEnE,MAAM,SAAS,GAAG,MAAM,iBAAiB,EAAE,CAAC;YAC5C,MAAM,UAAU,GAAG,MAAM,iBAAiB,EAAE,CAAC;YAE7C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACnC,uDAAuD;YACvD,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;YAC3C,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;YAC3C,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/experiments/experiments.d.ts

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { CodeAssistServer } from '../server.js';
+import type { Flag } from './types.js';
+export interface Experiments {
+    flags: Record<string, Flag>;
+    experimentIds: number[];
+}
+/**
+ * Gets the experiments from the server.
+ *
+ * The experiments are cached so that they are only fetched once.
+ */
+export declare function getExperiments(server?: CodeAssistServer): Promise<Experiments>;

package/dist/src/code_assist/experiments/experiments.js

@@ -0,0 +1,57 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { getClientMetadata } from './client_metadata.js';
+import * as fs from 'node:fs';
+import { debugLogger } from '../../utils/debugLogger.js';
+let experimentsPromise;
+/**
+ * Gets the experiments from the server.
+ *
+ * The experiments are cached so that they are only fetched once.
+ */
+export async function getExperiments(server) {
+    if (experimentsPromise) {
+        return experimentsPromise;
+    }
+    experimentsPromise = (async () => {
+        if (process.env['GEMINI_EXP']) {
+            try {
+                const expPath = process.env['GEMINI_EXP'];
+                debugLogger.debug('Reading experiments from', expPath);
+                const content = await fs.promises.readFile(expPath, 'utf8');
+                const response = JSON.parse(content);
+                if ((response.flags && !Array.isArray(response.flags)) ||
+                    (response.experimentIds && !Array.isArray(response.experimentIds))) {
+                    throw new Error('Invalid format for experiments file: `flags` and `experimentIds` must be arrays if present.');
+                }
+                return parseExperiments(response);
+            }
+            catch (e) {
+                debugLogger.debug('Failed to read experiments from GEMINI_EXP', e);
+            }
+        }
+        if (!server) {
+            return { flags: {}, experimentIds: [] };
+        }
+        const metadata = await getClientMetadata();
+        const response = await server.listExperiments(metadata);
+        return parseExperiments(response);
+    })();
+    return experimentsPromise;
+}
+function parseExperiments(response) {
+    const flags = {};
+    for (const flag of response.flags ?? []) {
+        if (flag.flagId) {
+            flags[flag.flagId] = flag;
+        }
+    }
+    return {
+        flags,
+        experimentIds: response.experimentIds ?? [],
+    };
+}
+//# sourceMappingURL=experiments.js.map

package/dist/src/code_assist/experiments/experiments.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"experiments.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/experiments.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzD,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAOzD,IAAI,kBAAoD,CAAC;AAEzD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAAyB;IAEzB,IAAI,kBAAkB,EAAE,CAAC;QACvB,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,kBAAkB,GAAG,CAAC,KAAK,IAAI,EAAE;QAC/B,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC1C,WAAW,CAAC,KAAK,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC;gBACvD,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACrC,IACE,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;oBAClD,CAAC,QAAQ,CAAC,aAAa,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,EAClE,CAAC;oBACD,MAAM,IAAI,KAAK,CACb,6FAA6F,CAC9F,CAAC;gBACJ,CAAC;gBACD,OAAO,gBAAgB,CAAC,QAAmC,CAAC,CAAC;YAC/D,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,WAAW,CAAC,KAAK,CAAC,4CAA4C,EAAE,CAAC,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC;QAC1C,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC3C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QACxD,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC,CAAC,EAAE,CAAC;IACL,OAAO,kBAAkB,CAAC;AAC5B,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAiC;IACzD,MAAM,KAAK,GAAyB,EAAE,CAAC;IACvC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;QACxC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC;IACD,OAAO;QACL,KAAK;QACL,aAAa,EAAE,QAAQ,CAAC,aAAa,IAAI,EAAE;KAC5C,CAAC;AACJ,CAAC"}

package/dist/src/code_assist/experiments/experiments.test.d.ts

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

package/dist/src/code_assist/experiments/experiments.test.js

@@ -0,0 +1,93 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { getClientMetadata } from './client_metadata.js';
+// Mock dependencies before importing the module under test
+vi.mock('../server.js');
+vi.mock('./client_metadata.js');
+describe('experiments', () => {
+    let mockServer;
+    beforeEach(() => {
+        // Reset modules to clear the cached `experimentsPromise`
+        vi.resetModules();
+        delete process.env['GEMINI_EXP'];
+        // Mock the dependencies that `getExperiments` relies on
+        vi.mocked(getClientMetadata).mockResolvedValue({
+            ideName: 'GEMINI_CLI',
+            ideVersion: '1.0.0',
+            platform: 'LINUX_AMD64',
+            updateChannel: 'stable',
+        });
+        // Create a mock instance of the server for each test
+        mockServer = {
+            listExperiments: vi.fn(),
+        };
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should fetch and parse experiments from the server', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {
+            flags: [
+                { flagId: 234, boolValue: true },
+                { flagId: 345, stringValue: 'value' },
+            ],
+            experimentIds: [123, 456],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const experiments = await getExperiments(mockServer);
+        // Verify that the dependencies were called
+        expect(getClientMetadata).toHaveBeenCalled();
+        expect(mockServer.listExperiments).toHaveBeenCalledWith(await getClientMetadata());
+        // Verify that the response was parsed correctly
+        expect(experiments.flags[234]).toEqual({
+            flagId: 234,
+            boolValue: true,
+        });
+        expect(experiments.flags[345]).toEqual({
+            flagId: 345,
+            stringValue: 'value',
+        });
+        expect(experiments.experimentIds).toEqual([123, 456]);
+    });
+    it('should handle an empty or partial response from the server', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {}; // No flags or experimentIds
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const experiments = await getExperiments(mockServer);
+        expect(experiments.flags).toEqual({});
+        expect(experiments.experimentIds).toEqual([]);
+    });
+    it('should ignore flags that are missing a name', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {
+            flags: [
+                { boolValue: true }, // No name
+                { flagId: 256, stringValue: 'value' },
+            ],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const experiments = await getExperiments(mockServer);
+        expect(Object.keys(experiments.flags)).toHaveLength(1);
+        expect(experiments.flags[256]).toBeDefined();
+        expect(experiments.flags['undefined']).toBeUndefined();
+    });
+    it('should cache the experiments promise to avoid multiple fetches', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {
+            experimentIds: [1, 2, 3],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const firstCall = await getExperiments(mockServer);
+        const secondCall = await getExperiments(mockServer);
+        expect(firstCall).toBe(secondCall); // Should be the exact same promise object
+        // Verify the underlying functions were only called once
+        expect(getClientMetadata).toHaveBeenCalledTimes(1);
+        expect(mockServer.listExperiments).toHaveBeenCalledTimes(1);
+    });
+});
+//# sourceMappingURL=experiments.test.js.map

package/dist/src/code_assist/experiments/experiments.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"experiments.test.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/experiments.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAGzD,2DAA2D;AAC3D,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AACxB,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAEhC,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,UAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,yDAAyD;QACzD,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAEjC,wDAAwD;QACxD,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,iBAAiB,CAAC;YAC7C,OAAO,EAAE,YAAY;YACrB,UAAU,EAAE,OAAO;YACnB,QAAQ,EAAE,aAAa;YACvB,aAAa,EAAE,QAAQ;SACxB,CAAC,CAAC;QAEH,qDAAqD;QACrD,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACM,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B;YAC/C,KAAK,EAAE;gBACL,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE;gBAChC,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,OAAO,EAAE;aACtC;YACD,aAAa,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;SAC1B,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,2CAA2C;QAC3C,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC7C,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACrD,MAAM,iBAAiB,EAAE,CAC1B,CAAC;QAEF,gDAAgD;QAChD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,MAAM,EAAE,GAAG;YACX,WAAW,EAAE,OAAO;SACrB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B,EAAE,CAAC,CAAC,4BAA4B;QACjF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B;YAC/C,KAAK,EAAE;gBACL,EAAE,SAAS,EAAE,IAAI,EAAU,EAAE,UAAU;gBACvC,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,OAAO,EAAE;aACtC;SACF,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B;YAC/C,aAAa,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;SACzB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAEpD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,0CAA0C;QAC9E,wDAAwD;QACxD,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/experiments/experiments_local.test.d.ts

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

package/dist/src/code_assist/experiments/experiments_local.test.js

@@ -0,0 +1,115 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+// Mock dependencies
+vi.mock('node:fs', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        promises: {
+            ...actual.promises,
+            readFile: vi.fn(),
+        },
+        readFileSync: vi.fn(),
+    };
+});
+vi.mock('node:os');
+vi.mock('../server.js');
+vi.mock('./client_metadata.js', () => ({
+    getClientMetadata: vi.fn(),
+}));
+describe('experiments with GEMINI_EXP', () => {
+    let mockServer;
+    beforeEach(() => {
+        vi.resetModules();
+        vi.clearAllMocks();
+        process.env['GEMINI_EXP'] = ''; // Clear env var
+        // Default mocks
+        vi.mocked(os.homedir).mockReturnValue('/home/user');
+        mockServer = {
+            listExperiments: vi.fn(),
+        };
+    });
+    afterEach(() => {
+        delete process.env['GEMINI_EXP'];
+    });
+    it('should read experiments from local file if GEMINI_EXP is set', async () => {
+        process.env['GEMINI_EXP'] = '/tmp/experiments.json';
+        const mockFileContent = JSON.stringify({
+            flags: [{ flagId: 111, boolValue: true }],
+            experimentIds: [999],
+        });
+        vi.mocked(fs.promises.readFile).mockResolvedValue(mockFileContent);
+        const { getExperiments } = await import('./experiments.js');
+        const experiments = await getExperiments(mockServer);
+        expect(fs.promises.readFile).toHaveBeenCalledWith('/tmp/experiments.json', 'utf8');
+        expect(experiments.flags[111]).toEqual({
+            flagId: 111,
+            boolValue: true,
+        });
+        expect(experiments.experimentIds).toEqual([999]);
+        expect(mockServer.listExperiments).not.toHaveBeenCalled();
+    });
+    it('should fall back to server if reading file fails', async () => {
+        process.env['GEMINI_EXP'] = '/tmp/missing.json';
+        vi.mocked(fs.promises.readFile).mockRejectedValue(new Error('File not found'));
+        // Mock server response
+        const mockApiResponse = {
+            flags: [{ flagId: 222, boolValue: true }],
+            experimentIds: [111],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const { getClientMetadata } = await import('./client_metadata.js');
+        vi.mocked(getClientMetadata).mockResolvedValue({});
+        const { getExperiments } = await import('./experiments.js');
+        const experiments = await getExperiments(mockServer);
+        expect(experiments.flags[222]).toBeDefined();
+        expect(mockServer.listExperiments).toHaveBeenCalled();
+    });
+    it('should work without server if file read succeeds', async () => {
+        process.env['GEMINI_EXP'] = '/tmp/experiments.json';
+        const mockFileContent = JSON.stringify({
+            flags: [{ flagId: 333, boolValue: true }],
+            experimentIds: [999],
+        });
+        vi.mocked(fs.promises.readFile).mockResolvedValue(mockFileContent);
+        const { getExperiments } = await import('./experiments.js');
+        const experiments = await getExperiments(undefined);
+        expect(experiments.flags[333]).toEqual({
+            flagId: 333,
+            boolValue: true,
+        });
+    });
+    it('should return empty if no server and no GEMINI_EXP', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const experiments = await getExperiments(undefined);
+        expect(experiments.flags).toEqual({});
+        expect(experiments.experimentIds).toEqual([]);
+    });
+    it('should fallback to server if file has invalid structure', async () => {
+        process.env['GEMINI_EXP'] = '/tmp/invalid.json';
+        const mockFileContent = JSON.stringify({
+            flags: 'invalid-flags-type', // Should be array
+            experimentIds: 123, // Should be array
+        });
+        vi.mocked(fs.promises.readFile).mockResolvedValue(mockFileContent);
+        // Mock server response
+        const mockApiResponse = {
+            flags: [{ flagId: 444, boolValue: true }],
+            experimentIds: [555],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const { getClientMetadata } = await import('./client_metadata.js');
+        vi.mocked(getClientMetadata).mockResolvedValue({});
+        const { getExperiments } = await import('./experiments.js');
+        const experiments = await getExperiments(mockServer);
+        expect(experiments.flags[444]).toBeDefined();
+        expect(mockServer.listExperiments).toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=experiments_local.test.js.map

package/dist/src/code_assist/experiments/experiments_local.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"experiments_local.test.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/experiments_local.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAI9B,oBAAoB;AACpB,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1C,MAAM,MAAM,GAAG,MAAM,cAAc,EAA4B,CAAC;IAChE,OAAO;QACL,GAAG,MAAM;QACT,QAAQ,EAAE;YACR,GAAG,MAAM,CAAC,QAAQ;YAClB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB;QACD,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;KACtB,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACnB,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AACxB,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC3B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,UAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC,CAAC,gBAAgB;QAEhD,gBAAgB;QAChB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;QACpD,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACM,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,uBAAuB,CAAC;QACpD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC;YACrC,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;YACzC,aAAa,EAAE,CAAC,GAAG,CAAC;SACrB,CAAC,CAAC;QACH,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEnE,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAC/C,uBAAuB,EACvB,MAAM,CACP,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,mBAAmB,CAAC;QAChD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAC/C,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAC5B,CAAC;QAEF,uBAAuB;QACvB,MAAM,eAAe,GAAG;YACtB,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;YACzC,aAAa,EAAE,CAAC,GAAG,CAAC;SACrB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CACrD,eAA0C,CAC3C,CAAC;QACF,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QACnE,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,iBAAiB,CAC5C,EAA+B,CAChC,CAAC;QAEF,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,uBAAuB,CAAC;QACpD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC;YACrC,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;YACzC,aAAa,EAAE,CAAC,GAAG,CAAC;SACrB,CAAC,CAAC;QACH,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEnE,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,CAAC;QAEpD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,mBAAmB,CAAC;QAChD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC;YACrC,KAAK,EAAE,oBAAoB,EAAE,kBAAkB;YAC/C,aAAa,EAAE,GAAG,EAAE,kBAAkB;SACvC,CAAC,CAAC;QACH,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEnE,uBAAuB;QACvB,MAAM,eAAe,GAAG;YACtB,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;YACzC,aAAa,EAAE,CAAC,GAAG,CAAC;SACrB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CACrD,eAA0C,CAC3C,CAAC;QACF,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QACnE,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,iBAAiB,CAC5C,EAA+B,CAChC,CAAC;QAEF,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/experiments/flagNames.d.ts

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const ExperimentFlags: {
+    readonly CONTEXT_COMPRESSION_THRESHOLD: 45740197;
+    readonly USER_CACHING: 45740198;
+    readonly BANNER_TEXT_NO_CAPACITY_ISSUES: 45740199;
+    readonly BANNER_TEXT_CAPACITY_ISSUES: 45740200;
+    readonly ENABLE_PREVIEW: 45740196;
+    readonly ENABLE_NUMERICAL_ROUTING: 45750526;
+    readonly CLASSIFIER_THRESHOLD: 45750527;
+    readonly ENABLE_ADMIN_CONTROLS: 45752213;
+};
+export type ExperimentFlagName = (typeof ExperimentFlags)[keyof typeof ExperimentFlags];

package/dist/src/code_assist/experiments/flagNames.js

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export const ExperimentFlags = {
+    CONTEXT_COMPRESSION_THRESHOLD: 45740197,
+    USER_CACHING: 45740198,
+    BANNER_TEXT_NO_CAPACITY_ISSUES: 45740199,
+    BANNER_TEXT_CAPACITY_ISSUES: 45740200,
+    ENABLE_PREVIEW: 45740196,
+    ENABLE_NUMERICAL_ROUTING: 45750526,
+    CLASSIFIER_THRESHOLD: 45750527,
+    ENABLE_ADMIN_CONTROLS: 45752213,
+};
+//# sourceMappingURL=flagNames.js.map

package/dist/src/code_assist/experiments/flagNames.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"flagNames.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/flagNames.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,6BAA6B,EAAE,QAAQ;IACvC,YAAY,EAAE,QAAQ;IACtB,8BAA8B,EAAE,QAAQ;IACxC,2BAA2B,EAAE,QAAQ;IACrC,cAAc,EAAE,QAAQ;IACxB,wBAAwB,EAAE,QAAQ;IAClC,oBAAoB,EAAE,QAAQ;IAC9B,qBAAqB,EAAE,QAAQ;CACvB,CAAC"}

package/dist/src/code_assist/experiments/types.d.ts

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ClientMetadata } from '../types.js';
+export interface ListExperimentsRequest {
+    project: string;
+    metadata?: ClientMetadata;
+}
+export interface ListExperimentsResponse {
+    experimentIds?: number[];
+    flags?: Flag[];
+    filteredFlags?: FilteredFlag[];
+    debugString?: string;
+}
+export interface Flag {
+    flagId?: number;
+    boolValue?: boolean;
+    floatValue?: number;
+    intValue?: string;
+    stringValue?: string;
+    int32ListValue?: Int32List;
+    stringListValue?: StringList;
+}
+export interface Int32List {
+    values?: number[];
+}
+export interface StringList {
+    values?: string[];
+}
+export interface FilteredFlag {
+    name?: string;
+    reason?: string;
+}

package/dist/src/code_assist/experiments/types.js

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/src/code_assist/experiments/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}

package/dist/src/code_assist/oauth-credential-storage.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type Credentials } from 'google-auth-library';
+export declare class OAuthCredentialStorage {
+    private static storage;
+    /**
+     * Load cached OAuth credentials
+     */
+    static loadCredentials(): Promise<Credentials | null>;
+    /**
+     * Save OAuth credentials
+     */
+    static saveCredentials(credentials: Credentials): Promise<void>;
+    /**
+     * Clear cached OAuth credentials
+     */
+    static clearCredentials(): Promise<void>;
+    /**
+     * Migrate credentials from old file-based storage to keychain
+     */
+    private static migrateFromFileStorage;
+}

package/dist/src/code_assist/oauth-credential-storage.js

@@ -0,0 +1,109 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {} from 'google-auth-library';
+import { HybridTokenStorage } from '../mcp/token-storage/hybrid-token-storage.js';
+import { OAUTH_FILE } from '../config/storage.js';
+import * as path from 'node:path';
+import { promises as fs } from 'node:fs';
+import { GEMINI_DIR, homedir } from '../utils/paths.js';
+import { coreEvents } from '../utils/events.js';
+const KEYCHAIN_SERVICE_NAME = 'gemini-cli-oauth';
+const MAIN_ACCOUNT_KEY = 'main-account';
+export class OAuthCredentialStorage {
+    static storage = new HybridTokenStorage(KEYCHAIN_SERVICE_NAME);
+    /**
+     * Load cached OAuth credentials
+     */
+    static async loadCredentials() {
+        try {
+            const credentials = await this.storage.getCredentials(MAIN_ACCOUNT_KEY);
+            if (credentials?.token) {
+                const { accessToken, refreshToken, expiresAt, tokenType, scope } = credentials.token;
+                // Convert from OAuthCredentials format to Google Credentials format
+                const googleCreds = {
+                    access_token: accessToken,
+                    refresh_token: refreshToken || undefined,
+                    token_type: tokenType || undefined,
+                    scope: scope || undefined,
+                };
+                if (expiresAt) {
+                    googleCreds.expiry_date = expiresAt;
+                }
+                return googleCreds;
+            }
+            // Fallback: Try to migrate from old file-based storage
+            return await this.migrateFromFileStorage();
+        }
+        catch (error) {
+            coreEvents.emitFeedback('error', 'Failed to load OAuth credentials', error);
+            throw new Error('Failed to load OAuth credentials', { cause: error });
+        }
+    }
+    /**
+     * Save OAuth credentials
+     */
+    static async saveCredentials(credentials) {
+        if (!credentials.access_token) {
+            throw new Error('Attempted to save credentials without an access token.');
+        }
+        // Convert Google Credentials to OAuthCredentials format
+        const mcpCredentials = {
+            serverName: MAIN_ACCOUNT_KEY,
+            token: {
+                accessToken: credentials.access_token,
+                refreshToken: credentials.refresh_token || undefined,
+                tokenType: credentials.token_type || 'Bearer',
+                scope: credentials.scope || undefined,
+                expiresAt: credentials.expiry_date || undefined,
+            },
+            updatedAt: Date.now(),
+        };
+        await this.storage.setCredentials(mcpCredentials);
+    }
+    /**
+     * Clear cached OAuth credentials
+     */
+    static async clearCredentials() {
+        try {
+            await this.storage.deleteCredentials(MAIN_ACCOUNT_KEY);
+            // Also try to remove the old file if it exists
+            const oldFilePath = path.join(homedir(), GEMINI_DIR, OAUTH_FILE);
+            await fs.rm(oldFilePath, { force: true }).catch(() => { });
+        }
+        catch (error) {
+            coreEvents.emitFeedback('error', 'Failed to clear OAuth credentials', error);
+            throw new Error('Failed to clear OAuth credentials', { cause: error });
+        }
+    }
+    /**
+     * Migrate credentials from old file-based storage to keychain
+     */
+    static async migrateFromFileStorage() {
+        const oldFilePath = path.join(homedir(), GEMINI_DIR, OAUTH_FILE);
+        let credsJson;
+        try {
+            credsJson = await fs.readFile(oldFilePath, 'utf-8');
+        }
+        catch (error) {
+            if (typeof error === 'object' &&
+                error !== null &&
+                'code' in error &&
+                error.code === 'ENOENT') {
+                // File doesn't exist, so no migration.
+                return null;
+            }
+            // Other read errors should propagate.
+            throw error;
+        }
+        const credentials = JSON.parse(credsJson);
+        // Save to new storage
+        await this.saveCredentials(credentials);
+        // Remove old file after successful migration
+        await fs.rm(oldFilePath, { force: true }).catch(() => { });
+        return credentials;
+    }
+}
+//# sourceMappingURL=oauth-credential-storage.js.map