@didim365/agent-cli-core 0.1.0

package/dist/src/code_assist/admin/admin_controls.js

@@ -0,0 +1,88 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { debugLogger } from '../../utils/debugLogger.js';
+import { isDeepStrictEqual } from 'node:util';
+import { FetchAdminControlsResponseSchema, } from '../types.js';
+let pollingInterval;
+let currentSettings;
+export function sanitizeAdminSettings(settings) {
+    const result = FetchAdminControlsResponseSchema.safeParse(settings);
+    if (!result.success) {
+        return {};
+    }
+    return result.data;
+}
+/**
+ * Fetches the admin controls from the server if enabled by experiment flag.
+ * Safely handles polling start/stop based on the flag and server availability.
+ *
+ * @param server The CodeAssistServer instance.
+ * @param cachedSettings The cached settings to use if available.
+ * @param adminControlsEnabled Whether admin controls are enabled.
+ * @param onSettingsChanged Callback to invoke when settings change during polling.
+ * @returns The fetched settings if enabled and successful, otherwise undefined.
+ */
+export async function fetchAdminControls(server, cachedSettings, adminControlsEnabled, onSettingsChanged) {
+    if (!server || !server.projectId || !adminControlsEnabled) {
+        stopAdminControlsPolling();
+        currentSettings = undefined;
+        return {};
+    }
+    // If we already have settings (e.g. from IPC during relaunch), use them
+    // to avoid blocking startup with another fetch. We'll still start polling.
+    if (cachedSettings) {
+        currentSettings = cachedSettings;
+        startAdminControlsPolling(server, server.projectId, onSettingsChanged);
+        return cachedSettings;
+    }
+    try {
+        const rawSettings = await server.fetchAdminControls({
+            project: server.projectId,
+        });
+        const sanitizedSettings = sanitizeAdminSettings(rawSettings);
+        currentSettings = sanitizedSettings;
+        startAdminControlsPolling(server, server.projectId, onSettingsChanged);
+        return sanitizedSettings;
+    }
+    catch (e) {
+        debugLogger.error('Failed to fetch admin controls: ', e);
+        // If initial fetch fails, start polling to retry.
+        currentSettings = {};
+        startAdminControlsPolling(server, server.projectId, onSettingsChanged);
+        return {};
+    }
+}
+/**
+ * Starts polling for admin controls.
+ */
+function startAdminControlsPolling(server, project, onSettingsChanged) {
+    stopAdminControlsPolling();
+    pollingInterval = setInterval(async () => {
+        try {
+            const rawSettings = await server.fetchAdminControls({
+                project,
+            });
+            const newSettings = sanitizeAdminSettings(rawSettings);
+            if (!isDeepStrictEqual(newSettings, currentSettings)) {
+                currentSettings = newSettings;
+                onSettingsChanged(newSettings);
+            }
+        }
+        catch (e) {
+            debugLogger.error('Failed to poll admin controls: ', e);
+        }
+    }, 5 * 60 * 1000); // 5 minutes
+}
+/**
+ * Stops polling for admin controls.
+ */
+export function stopAdminControlsPolling() {
+    if (pollingInterval) {
+        clearInterval(pollingInterval);
+        pollingInterval = undefined;
+    }
+}
+//# sourceMappingURL=admin_controls.js.map

package/dist/src/code_assist/admin/admin_controls.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"admin_controls.js","sourceRoot":"","sources":["../../../../src/code_assist/admin/admin_controls.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC9C,OAAO,EAEL,gCAAgC,GACjC,MAAM,aAAa,CAAC;AAErB,IAAI,eAA2C,CAAC;AAChD,IAAI,eAAuD,CAAC;AAE5D,MAAM,UAAU,qBAAqB,CACnC,QAAoC;IAEpC,MAAM,MAAM,GAAG,gCAAgC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACpE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC;AACrB,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,MAAoC,EACpC,cAAsD,EACtD,oBAA6B,EAC7B,iBAAiE;IAEjE,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1D,wBAAwB,EAAE,CAAC;QAC3B,eAAe,GAAG,SAAS,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,wEAAwE;IACxE,2EAA2E;IAC3E,IAAI,cAAc,EAAE,CAAC;QACnB,eAAe,GAAG,cAAc,CAAC;QACjC,yBAAyB,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;QACvE,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAClD,OAAO,EAAE,MAAM,CAAC,SAAS;SAC1B,CAAC,CAAC;QACH,MAAM,iBAAiB,GAAG,qBAAqB,CAAC,WAAW,CAAC,CAAC;QAC7D,eAAe,GAAG,iBAAiB,CAAC;QACpC,yBAAyB,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;QACvE,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,WAAW,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;QACzD,kDAAkD;QAClD,eAAe,GAAG,EAAE,CAAC;QACrB,yBAAyB,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;QACvE,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,yBAAyB,CAChC,MAAwB,EACxB,OAAe,EACf,iBAAiE;IAEjE,wBAAwB,EAAE,CAAC;IAE3B,eAAe,GAAG,WAAW,CAC3B,KAAK,IAAI,EAAE;QACT,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC;gBAClD,OAAO;aACR,CAAC,CAAC;YACH,MAAM,WAAW,GAAG,qBAAqB,CAAC,WAAW,CAAC,CAAC;YAEvD,IAAI,CAAC,iBAAiB,CAAC,WAAW,EAAE,eAAe,CAAC,EAAE,CAAC;gBACrD,eAAe,GAAG,WAAW,CAAC;gBAC9B,iBAAiB,CAAC,WAAW,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,WAAW,CAAC,KAAK,CAAC,iCAAiC,EAAE,CAAC,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,EACD,CAAC,GAAG,EAAE,GAAG,IAAI,CACd,CAAC,CAAC,YAAY;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB;IACtC,IAAI,eAAe,EAAE,CAAC;QACpB,aAAa,CAAC,eAAe,CAAC,CAAC;QAC/B,eAAe,GAAG,SAAS,CAAC;IAC9B,CAAC;AACH,CAAC"}

package/dist/src/code_assist/admin/admin_controls.test.d.ts

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

package/dist/src/code_assist/admin/admin_controls.test.js

@@ -0,0 +1,200 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { fetchAdminControls, sanitizeAdminSettings, stopAdminControlsPolling, } from './admin_controls.js';
+describe('Admin Controls', () => {
+    let mockServer;
+    let mockOnSettingsChanged;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        vi.useFakeTimers();
+        mockServer = {
+            projectId: 'test-project',
+            fetchAdminControls: vi.fn(),
+        };
+        mockOnSettingsChanged = vi.fn();
+    });
+    afterEach(() => {
+        stopAdminControlsPolling();
+        vi.useRealTimers();
+    });
+    describe('sanitizeAdminSettings', () => {
+        it('should strip unknown fields', () => {
+            const input = {
+                secureModeEnabled: true,
+                extraField: 'should be removed',
+                mcpSetting: {
+                    mcpEnabled: false,
+                    unknownMcpField: 'remove me',
+                },
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result).toEqual({
+                secureModeEnabled: true,
+                mcpSetting: {
+                    mcpEnabled: false,
+                },
+            });
+            // Explicitly check that unknown fields are gone
+            expect(result['extraField']).toBeUndefined();
+        });
+        it('should preserve valid nested fields', () => {
+            const input = {
+                cliFeatureSetting: {
+                    extensionsSetting: {
+                        extensionsEnabled: true,
+                    },
+                },
+            };
+            expect(sanitizeAdminSettings(input)).toEqual(input);
+        });
+    });
+    describe('fetchAdminControls', () => {
+        it('should return empty object and not poll if server is missing', async () => {
+            const result = await fetchAdminControls(undefined, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should return empty object if project ID is missing', async () => {
+            mockServer = {
+                fetchAdminControls: vi.fn(),
+            };
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should use cachedSettings and start polling if provided', async () => {
+            const cachedSettings = { secureModeEnabled: true };
+            const result = await fetchAdminControls(mockServer, cachedSettings, true, mockOnSettingsChanged);
+            expect(result).toEqual(cachedSettings);
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+            // Should still start polling
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: false,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+        it('should return empty object if admin controls are disabled', async () => {
+            const result = await fetchAdminControls(mockServer, undefined, false, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should fetch from server if no cachedSettings provided', async () => {
+            const serverResponse = { secureModeEnabled: true };
+            mockServer.fetchAdminControls.mockResolvedValue(serverResponse);
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual(serverResponse);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+        it('should return empty object on fetch error and still start polling', async () => {
+            mockServer.fetchAdminControls.mockRejectedValue(new Error('Network error'));
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            // Polling should have been started and should retry
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: true,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2); // Initial + poll
+        });
+        it('should sanitize server response', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: true,
+                unknownField: 'bad',
+            });
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({ secureModeEnabled: true });
+            expect(result['unknownField']).toBeUndefined();
+        });
+        it('should reset polling interval if called again', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({});
+            // First call
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            // Advance time, but not enough to trigger the poll
+            await vi.advanceTimersByTimeAsync(2 * 60 * 1000);
+            // Second call, should reset the timer
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+            // Advance time by 3 mins. If timer wasn't reset, it would have fired (2+3=5)
+            await vi.advanceTimersByTimeAsync(3 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2); // No new poll
+            // Advance time by another 2 mins. Now it should fire.
+            await vi.advanceTimersByTimeAsync(2 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(3); // Poll fires
+        });
+    });
+    describe('polling', () => {
+        it('should poll and emit changes', async () => {
+            // Initial fetch
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: false,
+            });
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            // Update for next poll
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: true,
+            });
+            // Fast forward
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockOnSettingsChanged).toHaveBeenCalledWith({
+                secureModeEnabled: true,
+            });
+        });
+        it('should NOT emit if settings are deeply equal but not the same instance', async () => {
+            const settings = { secureModeEnabled: true };
+            mockServer.fetchAdminControls.mockResolvedValue(settings);
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            mockOnSettingsChanged.mockClear();
+            // Next poll returns a different object with the same values
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: true,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockOnSettingsChanged).not.toHaveBeenCalled();
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+        });
+        it('should continue polling after a fetch error', async () => {
+            // Initial fetch is successful
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: false,
+            });
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            // Next poll fails
+            mockServer.fetchAdminControls.mockRejectedValue(new Error('Poll failed'));
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+            expect(mockOnSettingsChanged).not.toHaveBeenCalled(); // No changes on error
+            // Subsequent poll succeeds with new data
+            mockServer.fetchAdminControls.mockResolvedValue({
+                secureModeEnabled: true,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(3);
+            expect(mockOnSettingsChanged).toHaveBeenCalledWith({
+                secureModeEnabled: true,
+            });
+        });
+    });
+    describe('stopAdminControlsPolling', () => {
+        it('should stop polling after it has started', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({});
+            // Start polling
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            // Stop polling
+            stopAdminControlsPolling();
+            // Advance timer well beyond the polling interval
+            await vi.advanceTimersByTimeAsync(10 * 60 * 1000);
+            // The poll should not have fired again
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+    });
+});
+//# sourceMappingURL=admin_controls.test.js.map

package/dist/src/code_assist/admin/admin_controls.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"admin_controls.test.js","sourceRoot":"","sources":["../../../../src/code_assist/admin/admin_controls.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EACL,kBAAkB,EAClB,qBAAqB,EACrB,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAG7B,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,UAA4B,CAAC;IACjC,IAAI,qBAA2B,CAAC;IAEhC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,UAAU,GAAG;YACX,SAAS,EAAE,cAAc;YACzB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;SACG,CAAC;QAEjC,qBAAqB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,wBAAwB,EAAE,CAAC;QAC3B,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,KAAK,GAAG;gBACZ,iBAAiB,EAAE,IAAI;gBACvB,UAAU,EAAE,mBAAmB;gBAC/B,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;oBACjB,eAAe,EAAE,WAAW;iBAC7B;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,iBAAiB,EAAE,IAAI;gBACvB,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;iBAClB;aACF,CAAC,CAAC;YACH,gDAAgD;YAChD,MAAM,CAAE,MAAkC,CAAC,YAAY,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,KAAK,GAAG;gBACZ,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE;wBACjB,iBAAiB,EAAE,IAAI;qBACxB;iBACF;aACF,CAAC;YACF,MAAM,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,SAAS,EACT,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,UAAU,GAAG;gBACX,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;aACG,CAAC;YAEjC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,cAAc,GAAG,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;YACnD,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,cAAc,EACd,IAAI,EACJ,qBAAqB,CACtB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACvC,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAE7D,6BAA6B;YAC5B,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,KAAK;aACzB,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,KAAK,EACL,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,cAAc,GAAG,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;YAClD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACvC,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YAChF,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CACvD,IAAI,KAAK,CAAC,eAAe,CAAC,CAC3B,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAE3B,oDAAoD;YACnD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,IAAI;aACxB,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC9C,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,IAAI;gBACvB,YAAY,EAAE,KAAK;aACpB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YACpD,MAAM,CACH,MAAkC,CAAC,cAAc,CAAC,CACpD,CAAC,aAAa,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC5D,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE9D,aAAa;YACb,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,mDAAmD;YACnD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,sCAAsC;YACtC,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,6EAA6E;YAC7E,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc;YAE9E,sDAAsD;YACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;QAC/E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,gBAAgB;YACf,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,KAAK;aACzB,CAAC,CAAC;YACH,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YAEF,uBAAuB;YACtB,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,IAAI;aACxB,CAAC,CAAC;YAEH,eAAe;YACf,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC;gBACjD,iBAAiB,EAAE,IAAI;aACxB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,QAAQ,GAAG,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;YAC5C,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEpE,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,qBAAqB,CAAC,SAAS,EAAE,CAAC;YAElC,4DAA4D;YAC3D,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,IAAI;aACxB,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,MAAM,CAAC,qBAAqB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACrD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,8BAA8B;YAC7B,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,KAAK;aACzB,CAAC,CAAC;YACH,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,kBAAkB;YACjB,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CACvD,IAAI,KAAK,CAAC,aAAa,CAAC,CACzB,CAAC;YACF,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,sBAAsB;YAE5E,yCAAyC;YACxC,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,iBAAiB,EAAE,IAAI;aACxB,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC;gBACjD,iBAAiB,EAAE,IAAI;aACxB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACvD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE9D,gBAAgB;YAChB,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,eAAe;YACf,wBAAwB,EAAE,CAAC;YAE3B,iDAAiD;YACjD,MAAM,EAAE,CAAC,wBAAwB,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAElD,uCAAuC;YACvC,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,12 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ContentGenerator } from '../core/contentGenerator.js';
+import { AuthType } from '../core/contentGenerator.js';
+import type { HttpOptions } from './server.js';
+import { CodeAssistServer } from './server.js';
+import type { Config } from '../config/config.js';
+export declare function createCodeAssistContentGenerator(httpOptions: HttpOptions, authType: AuthType, config: Config, sessionId?: string): Promise<ContentGenerator>;
+export declare function getCodeAssistServer(config: Config): CodeAssistServer | undefined;

package/dist/src/code_assist/codeAssist.js

@@ -0,0 +1,31 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType } from '../core/contentGenerator.js';
+import { getOauthClient } from './oauth2.js';
+import { setupUser } from './setup.js';
+import { CodeAssistServer } from './server.js';
+import { LoggingContentGenerator } from '../core/loggingContentGenerator.js';
+export async function createCodeAssistContentGenerator(httpOptions, authType, config, sessionId) {
+    if (authType === AuthType.LOGIN_WITH_GOOGLE ||
+        authType === AuthType.COMPUTE_ADC) {
+        const authClient = await getOauthClient(authType, config);
+        const userData = await setupUser(authClient, config.getValidationHandler());
+        return new CodeAssistServer(authClient, userData.projectId, httpOptions, sessionId, userData.userTier, userData.userTierName);
+    }
+    throw new Error(`Unsupported authType: ${authType}`);
+}
+export function getCodeAssistServer(config) {
+    let server = config.getContentGenerator();
+    // Unwrap LoggingContentGenerator if present
+    if (server instanceof LoggingContentGenerator) {
+        server = server.getWrapped();
+    }
+    if (!(server instanceof CodeAssistServer)) {
+        return undefined;
+    }
+    return server;
+}
+//# sourceMappingURL=codeAssist.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"codeAssist.js","sourceRoot":"","sources":["../../../src/code_assist/codeAssist.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE/C,OAAO,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAE7E,MAAM,CAAC,KAAK,UAAU,gCAAgC,CACpD,WAAwB,EACxB,QAAkB,EAClB,MAAc,EACd,SAAkB;IAElB,IACE,QAAQ,KAAK,QAAQ,CAAC,iBAAiB;QACvC,QAAQ,KAAK,QAAQ,CAAC,WAAW,EACjC,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,UAAU,EAAE,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC;QAC5E,OAAO,IAAI,gBAAgB,CACzB,UAAU,EACV,QAAQ,CAAC,SAAS,EAClB,WAAW,EACX,SAAS,EACT,QAAQ,CAAC,QAAQ,EACjB,QAAQ,CAAC,YAAY,CACtB,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,MAAc;IAEd,IAAI,MAAM,GAAG,MAAM,CAAC,mBAAmB,EAAE,CAAC;IAE1C,4CAA4C;IAC5C,IAAI,MAAM,YAAY,uBAAuB,EAAE,CAAC;QAC9C,MAAM,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;IAC/B,CAAC;IAED,IAAI,CAAC,CAAC,MAAM,YAAY,gBAAgB,CAAC,EAAE,CAAC;QAC1C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

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

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

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

@@ -0,0 +1,102 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { AuthType } from '../core/contentGenerator.js';
+import { getOauthClient } from './oauth2.js';
+import { setupUser } from './setup.js';
+import { CodeAssistServer } from './server.js';
+import { createCodeAssistContentGenerator, getCodeAssistServer, } from './codeAssist.js';
+import { LoggingContentGenerator } from '../core/loggingContentGenerator.js';
+import { UserTierId } from './types.js';
+// Mock dependencies
+vi.mock('./oauth2.js');
+vi.mock('./setup.js');
+vi.mock('./server.js');
+vi.mock('../core/loggingContentGenerator.js');
+const mockedGetOauthClient = vi.mocked(getOauthClient);
+const mockedSetupUser = vi.mocked(setupUser);
+const MockedCodeAssistServer = vi.mocked(CodeAssistServer);
+const MockedLoggingContentGenerator = vi.mocked(LoggingContentGenerator);
+describe('codeAssist', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+    });
+    describe('createCodeAssistContentGenerator', () => {
+        const httpOptions = {};
+        const mockValidationHandler = vi.fn();
+        const mockConfig = {
+            getValidationHandler: () => mockValidationHandler,
+        };
+        const mockAuthClient = { a: 'client' };
+        const mockUserData = {
+            projectId: 'test-project',
+            userTier: UserTierId.FREE,
+        };
+        it('should create a server for LOGIN_WITH_GOOGLE', async () => {
+            mockedGetOauthClient.mockResolvedValue(mockAuthClient);
+            mockedSetupUser.mockResolvedValue(mockUserData);
+            const generator = await createCodeAssistContentGenerator(httpOptions, AuthType.LOGIN_WITH_GOOGLE, mockConfig, 'session-123');
+            expect(getOauthClient).toHaveBeenCalledWith(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+            expect(setupUser).toHaveBeenCalledWith(mockAuthClient, mockValidationHandler);
+            expect(MockedCodeAssistServer).toHaveBeenCalledWith(mockAuthClient, 'test-project', httpOptions, 'session-123', 'free-tier', undefined);
+            expect(generator).toBeInstanceOf(MockedCodeAssistServer);
+        });
+        it('should create a server for COMPUTE_ADC', async () => {
+            mockedGetOauthClient.mockResolvedValue(mockAuthClient);
+            mockedSetupUser.mockResolvedValue(mockUserData);
+            const generator = await createCodeAssistContentGenerator(httpOptions, AuthType.COMPUTE_ADC, mockConfig);
+            expect(getOauthClient).toHaveBeenCalledWith(AuthType.COMPUTE_ADC, mockConfig);
+            expect(setupUser).toHaveBeenCalledWith(mockAuthClient, mockValidationHandler);
+            expect(MockedCodeAssistServer).toHaveBeenCalledWith(mockAuthClient, 'test-project', httpOptions, undefined, // No session ID
+            'free-tier', undefined);
+            expect(generator).toBeInstanceOf(MockedCodeAssistServer);
+        });
+        it('should throw an error for unsupported auth types', async () => {
+            await expect(createCodeAssistContentGenerator(httpOptions, 'api-key', // Use literal string to avoid enum resolution issues
+            mockConfig)).rejects.toThrow('Unsupported authType: api-key');
+        });
+    });
+    describe('getCodeAssistServer', () => {
+        it('should return the server if it is a CodeAssistServer', () => {
+            const mockServer = new MockedCodeAssistServer({}, '', {});
+            const mockConfig = {
+                getContentGenerator: () => mockServer,
+            };
+            const server = getCodeAssistServer(mockConfig);
+            expect(server).toBe(mockServer);
+        });
+        it('should unwrap and return the server if it is wrapped in a LoggingContentGenerator', () => {
+            const mockServer = new MockedCodeAssistServer({}, '', {});
+            const mockLogger = new MockedLoggingContentGenerator({}, {});
+            vi.spyOn(mockLogger, 'getWrapped').mockReturnValue(mockServer);
+            const mockConfig = {
+                getContentGenerator: () => mockLogger,
+            };
+            const server = getCodeAssistServer(mockConfig);
+            expect(server).toBe(mockServer);
+            expect(mockLogger.getWrapped).toHaveBeenCalled();
+        });
+        it('should return undefined if the content generator is not a CodeAssistServer', () => {
+            const mockGenerator = { a: 'generator' }; // Not a CodeAssistServer
+            const mockConfig = {
+                getContentGenerator: () => mockGenerator,
+            };
+            const server = getCodeAssistServer(mockConfig);
+            expect(server).toBeUndefined();
+        });
+        it('should return undefined if the wrapped generator is not a CodeAssistServer', () => {
+            const mockGenerator = { a: 'generator' }; // Not a CodeAssistServer
+            const mockLogger = new MockedLoggingContentGenerator({}, {});
+            vi.spyOn(mockLogger, 'getWrapped').mockReturnValue(mockGenerator);
+            const mockConfig = {
+                getContentGenerator: () => mockLogger,
+            };
+            const server = getCodeAssistServer(mockConfig);
+            expect(server).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=codeAssist.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"codeAssist.test.js","sourceRoot":"","sources":["../../../src/code_assist/codeAssist.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EACL,gCAAgC,EAChC,mBAAmB,GACpB,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAExC,oBAAoB;AACpB,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AACvB,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AACvB,EAAE,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;AAE9C,MAAM,oBAAoB,GAAG,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;AACvD,MAAM,eAAe,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AAC7C,MAAM,sBAAsB,GAAG,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;AAC3D,MAAM,6BAA6B,GAAG,EAAE,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;AAEzE,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,MAAM,WAAW,GAAG,EAAE,CAAC;QACvB,MAAM,qBAAqB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,UAAU,GAAG;YACjB,oBAAoB,EAAE,GAAG,EAAE,CAAC,qBAAqB;SAC7B,CAAC;QACvB,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC;QACvC,MAAM,YAAY,GAAG;YACnB,SAAS,EAAE,cAAc;YACzB,QAAQ,EAAE,UAAU,CAAC,IAAI;SAC1B,CAAC;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,oBAAoB,CAAC,iBAAiB,CAAC,cAAuB,CAAC,CAAC;YAChE,eAAe,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAEhD,MAAM,SAAS,GAAG,MAAM,gCAAgC,CACtD,WAAW,EACX,QAAQ,CAAC,iBAAiB,EAC1B,UAAU,EACV,aAAa,CACd,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,QAAQ,CAAC,iBAAiB,EAC1B,UAAU,CACX,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,cAAc,EACd,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CACjD,cAAc,EACd,cAAc,EACd,WAAW,EACX,aAAa,EACb,WAAW,EACX,SAAS,CACV,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,oBAAoB,CAAC,iBAAiB,CAAC,cAAuB,CAAC,CAAC;YAChE,eAAe,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAEhD,MAAM,SAAS,GAAG,MAAM,gCAAgC,CACtD,WAAW,EACX,QAAQ,CAAC,WAAW,EACpB,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,QAAQ,CAAC,WAAW,EACpB,UAAU,CACX,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,cAAc,EACd,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CACjD,cAAc,EACd,cAAc,EACd,WAAW,EACX,SAAS,EAAE,gBAAgB;YAC3B,WAAW,EACX,SAAS,CACV,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,CACV,gCAAgC,CAC9B,WAAW,EACX,SAAqB,EAAE,qDAAqD;YAC5E,UAAU,CACX,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,UAAU,GAAG,IAAI,sBAAsB,CAAC,EAAW,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YACnE,MAAM,UAAU,GAAG;gBACjB,mBAAmB,EAAE,GAAG,EAAE,CAAC,UAAU;aACjB,CAAC;YAEvB,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;YAC3F,MAAM,UAAU,GAAG,IAAI,sBAAsB,CAAC,EAAW,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YACnE,MAAM,UAAU,GAAG,IAAI,6BAA6B,CAClD,EAAW,EACX,EAAW,CACZ,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YAE/D,MAAM,UAAU,GAAG;gBACjB,mBAAmB,EAAE,GAAG,EAAE,CAAC,UAAU;aACjB,CAAC;YAEvB,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,yBAAyB;YACnE,MAAM,UAAU,GAAG;gBACjB,mBAAmB,EAAE,GAAG,EAAE,CAAC,aAAa;aACpB,CAAC;YAEvB,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,yBAAyB;YACnE,MAAM,UAAU,GAAG,IAAI,6BAA6B,CAClD,EAAW,EACX,EAAW,CACZ,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,eAAe,CAChD,aAAsB,CACvB,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,mBAAmB,EAAE,GAAG,EAAE,CAAC,UAAU;aACjB,CAAC;YAEvB,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,75 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Content, ContentListUnion, GenerateContentParameters, CountTokensParameters, CountTokensResponse, GenerationConfigRoutingConfig, MediaResolution, Candidate, ModelSelectionConfig, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, Part, SafetySetting, PartUnion, SpeechConfigUnion, ThinkingConfig, ToolListUnion, ToolConfig } from '@google/genai';
+import { GenerateContentResponse } from '@google/genai';
+export interface CAGenerateContentRequest {
+    model: string;
+    project?: string;
+    user_prompt_id?: string;
+    request: VertexGenerateContentRequest;
+}
+interface VertexGenerateContentRequest {
+    contents: Content[];
+    systemInstruction?: Content;
+    cachedContent?: string;
+    tools?: ToolListUnion;
+    toolConfig?: ToolConfig;
+    labels?: Record<string, string>;
+    safetySettings?: SafetySetting[];
+    generationConfig?: VertexGenerationConfig;
+    session_id?: string;
+}
+interface VertexGenerationConfig {
+    temperature?: number;
+    topP?: number;
+    topK?: number;
+    candidateCount?: number;
+    maxOutputTokens?: number;
+    stopSequences?: string[];
+    responseLogprobs?: boolean;
+    logprobs?: number;
+    presencePenalty?: number;
+    frequencyPenalty?: number;
+    seed?: number;
+    responseMimeType?: string;
+    responseJsonSchema?: unknown;
+    responseSchema?: unknown;
+    routingConfig?: GenerationConfigRoutingConfig;
+    modelSelectionConfig?: ModelSelectionConfig;
+    responseModalities?: string[];
+    mediaResolution?: MediaResolution;
+    speechConfig?: SpeechConfigUnion;
+    audioTimestamp?: boolean;
+    thinkingConfig?: ThinkingConfig;
+}
+export interface CaGenerateContentResponse {
+    response: VertexGenerateContentResponse;
+    traceId?: string;
+}
+interface VertexGenerateContentResponse {
+    candidates: Candidate[];
+    automaticFunctionCallingHistory?: Content[];
+    promptFeedback?: GenerateContentResponsePromptFeedback;
+    usageMetadata?: GenerateContentResponseUsageMetadata;
+    modelVersion?: string;
+}
+export interface CaCountTokenRequest {
+    request: VertexCountTokenRequest;
+}
+interface VertexCountTokenRequest {
+    model: string;
+    contents: Content[];
+}
+export interface CaCountTokenResponse {
+    totalTokens: number;
+}
+export declare function toCountTokenRequest(req: CountTokensParameters): CaCountTokenRequest;
+export declare function fromCountTokenResponse(res: CaCountTokenResponse): CountTokensResponse;
+export declare function toGenerateContentRequest(req: GenerateContentParameters, userPromptId: string, project?: string, sessionId?: string): CAGenerateContentRequest;
+export declare function fromGenerateContentResponse(res: CaGenerateContentResponse): GenerateContentResponse;
+export declare function toContents(contents: ContentListUnion): Content[];
+export declare function toParts(parts: PartUnion[]): Part[];
+export {};