@google/gemini-cli-core 0.0.3-preview.4

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

@@ -0,0 +1,134 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { beforeEach, describe, it, expect, vi } from 'vitest';
+import { CodeAssistServer } from './server.js';
+import { OAuth2Client } from 'google-auth-library';
+import { UserTierId } from './types.js';
+vi.mock('google-auth-library');
+describe('CodeAssistServer', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+    });
+    it('should be able to be constructed', () => {
+        const auth = new OAuth2Client();
+        const server = new CodeAssistServer(auth, 'test-project', {}, 'test-session', UserTierId.FREE);
+        expect(server).toBeInstanceOf(CodeAssistServer);
+    });
+    it('should call the generateContent endpoint', async () => {
+        const client = new OAuth2Client();
+        const server = new CodeAssistServer(client, 'test-project', {}, 'test-session', UserTierId.FREE);
+        const mockResponse = {
+            response: {
+                candidates: [
+                    {
+                        index: 0,
+                        content: {
+                            role: 'model',
+                            parts: [{ text: 'response' }],
+                        },
+                        finishReason: 'STOP',
+                        safetyRatings: [],
+                    },
+                ],
+            },
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const response = await server.generateContent({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        }, 'user-prompt-id');
+        expect(server.requestPost).toHaveBeenCalledWith('generateContent', expect.any(Object), undefined);
+        expect(response.candidates?.[0]?.content?.parts?.[0]?.text).toBe('response');
+    });
+    it('should call the generateContentStream endpoint', async () => {
+        const client = new OAuth2Client();
+        const server = new CodeAssistServer(client, 'test-project', {}, 'test-session', UserTierId.FREE);
+        const mockResponse = (async function* () {
+            yield {
+                response: {
+                    candidates: [
+                        {
+                            index: 0,
+                            content: {
+                                role: 'model',
+                                parts: [{ text: 'response' }],
+                            },
+                            finishReason: 'STOP',
+                            safetyRatings: [],
+                        },
+                    ],
+                },
+            };
+        })();
+        vi.spyOn(server, 'requestStreamingPost').mockResolvedValue(mockResponse);
+        const stream = await server.generateContentStream({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        }, 'user-prompt-id');
+        for await (const res of stream) {
+            expect(server.requestStreamingPost).toHaveBeenCalledWith('streamGenerateContent', expect.any(Object), undefined);
+            expect(res.candidates?.[0]?.content?.parts?.[0]?.text).toBe('response');
+        }
+    });
+    it('should call the onboardUser endpoint', async () => {
+        const client = new OAuth2Client();
+        const server = new CodeAssistServer(client, 'test-project', {}, 'test-session', UserTierId.FREE);
+        const mockResponse = {
+            name: 'operations/123',
+            done: true,
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const response = await server.onboardUser({
+            tierId: 'test-tier',
+            cloudaicompanionProject: 'test-project',
+            metadata: {},
+        });
+        expect(server.requestPost).toHaveBeenCalledWith('onboardUser', expect.any(Object));
+        expect(response.name).toBe('operations/123');
+    });
+    it('should call the loadCodeAssist endpoint', async () => {
+        const client = new OAuth2Client();
+        const server = new CodeAssistServer(client, 'test-project', {}, 'test-session', UserTierId.FREE);
+        const mockResponse = {
+            currentTier: {
+                id: UserTierId.FREE,
+                name: 'Free',
+                description: 'free tier',
+            },
+            allowedTiers: [],
+            ineligibleTiers: [],
+            cloudaicompanionProject: 'projects/test',
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const response = await server.loadCodeAssist({
+            metadata: {},
+        });
+        expect(server.requestPost).toHaveBeenCalledWith('loadCodeAssist', expect.any(Object));
+        expect(response).toEqual(mockResponse);
+    });
+    it('should return 0 for countTokens', async () => {
+        const client = new OAuth2Client();
+        const server = new CodeAssistServer(client, 'test-project', {}, 'test-session', UserTierId.FREE);
+        const mockResponse = {
+            totalTokens: 100,
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const response = await server.countTokens({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        });
+        expect(response.totalTokens).toBe(100);
+    });
+    it('should throw an error for embedContent', async () => {
+        const client = new OAuth2Client();
+        const server = new CodeAssistServer(client, 'test-project', {}, 'test-session', UserTierId.FREE);
+        await expect(server.embedContent({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        })).rejects.toThrow();
+    });
+});
+//# sourceMappingURL=server.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"server.test.js","sourceRoot":"","sources":["../../../src/code_assist/server.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAExC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,IAAI,GAAG,IAAI,YAAY,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,IAAI,EACJ,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,MAAM,EACN,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,YAAY,GAAG;YACnB,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;yBAC9B;wBACD,YAAY,EAAE,MAAM;wBACpB,aAAa,EAAE,EAAE;qBAClB;iBACF;aACF;SACF,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,eAAe,CAC3C;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,EACD,gBAAgB,CACjB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAC7C,iBAAiB,EACjB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,SAAS,CACV,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAC9D,UAAU,CACX,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,MAAM,EACN,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,YAAY,GAAG,CAAC,KAAK,SAAS,CAAC;YACnC,MAAM;gBACJ,QAAQ,EAAE;oBACR,UAAU,EAAE;wBACV;4BACE,KAAK,EAAE,CAAC;4BACR,OAAO,EAAE;gCACP,IAAI,EAAE,OAAO;gCACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;6BAC9B;4BACD,YAAY,EAAE,MAAM;4BACpB,aAAa,EAAE,EAAE;yBAClB;qBACF;iBACF;aACF,CAAC;QACJ,CAAC,CAAC,EAAE,CAAC;QACL,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,sBAAsB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEzE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAC/C;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,EACD,gBAAgB,CACjB,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CACtD,uBAAuB,EACvB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,SAAS,CACV,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,MAAM,EACN,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,YAAY,GAAG;YACnB,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE,IAAI;SACX,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC;YACxC,MAAM,EAAE,WAAW;YACnB,uBAAuB,EAAE,cAAc;YACvC,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAC7C,aAAa,EACb,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,MAAM,EACN,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,YAAY,GAAG;YACnB,WAAW,EAAE;gBACX,EAAE,EAAE,UAAU,CAAC,IAAI;gBACnB,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,WAAW;aACzB;YACD,YAAY,EAAE,EAAE;YAChB,eAAe,EAAE,EAAE;YACnB,uBAAuB,EAAE,eAAe;SACzC,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC;YAC3C,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAC7C,gBAAgB,EAChB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,MAAM,EACN,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,YAAY,GAAG;YACnB,WAAW,EAAE,GAAG;SACjB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC;YACxC,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,MAAM,EACN,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,MAAM,CACV,MAAM,CAAC,YAAY,CAAC;YAClB,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,20 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { UserTierId } from './types.js';
+import type { OAuth2Client } from 'google-auth-library';
+export declare class ProjectIdRequiredError extends Error {
+    constructor();
+}
+export interface UserData {
+    projectId: string;
+    userTier: UserTierId;
+}
+/**
+ *
+ * @param projectId the user's project id, if any
+ * @returns the user's actual project id
+ */
+export declare function setupUser(client: OAuth2Client): Promise<UserData>;

package/dist/src/code_assist/setup.js

@@ -0,0 +1,101 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { UserTierId } from './types.js';
+import { CodeAssistServer } from './server.js';
+export class ProjectIdRequiredError extends Error {
+    constructor() {
+        super('This account requires setting the GOOGLE_CLOUD_PROJECT env var. See https://goo.gle/gemini-cli-auth-docs#workspace-gca');
+    }
+}
+/**
+ *
+ * @param projectId the user's project id, if any
+ * @returns the user's actual project id
+ */
+export async function setupUser(client) {
+    const projectId = process.env['GOOGLE_CLOUD_PROJECT'] || undefined;
+    const caServer = new CodeAssistServer(client, projectId, {}, '', undefined);
+    const coreClientMetadata = {
+        ideType: 'IDE_UNSPECIFIED',
+        platform: 'PLATFORM_UNSPECIFIED',
+        pluginType: 'GEMINI',
+    };
+    const loadRes = await caServer.loadCodeAssist({
+        cloudaicompanionProject: projectId,
+        metadata: {
+            ...coreClientMetadata,
+            duetProject: projectId,
+        },
+    });
+    if (loadRes.currentTier) {
+        if (!loadRes.cloudaicompanionProject) {
+            if (projectId) {
+                return {
+                    projectId,
+                    userTier: loadRes.currentTier.id,
+                };
+            }
+            throw new ProjectIdRequiredError();
+        }
+        return {
+            projectId: loadRes.cloudaicompanionProject,
+            userTier: loadRes.currentTier.id,
+        };
+    }
+    const tier = getOnboardTier(loadRes);
+    let onboardReq;
+    if (tier.id === UserTierId.FREE) {
+        // The free tier uses a managed google cloud project. Setting a project in the `onboardUser` request causes a `Precondition Failed` error.
+        onboardReq = {
+            tierId: tier.id,
+            cloudaicompanionProject: undefined,
+            metadata: coreClientMetadata,
+        };
+    }
+    else {
+        onboardReq = {
+            tierId: tier.id,
+            cloudaicompanionProject: projectId,
+            metadata: {
+                ...coreClientMetadata,
+                duetProject: projectId,
+            },
+        };
+    }
+    // Poll onboardUser until long running operation is complete.
+    let lroRes = await caServer.onboardUser(onboardReq);
+    while (!lroRes.done) {
+        await new Promise((f) => setTimeout(f, 5000));
+        lroRes = await caServer.onboardUser(onboardReq);
+    }
+    if (!lroRes.response?.cloudaicompanionProject?.id) {
+        if (projectId) {
+            return {
+                projectId,
+                userTier: tier.id,
+            };
+        }
+        throw new ProjectIdRequiredError();
+    }
+    return {
+        projectId: lroRes.response.cloudaicompanionProject.id,
+        userTier: tier.id,
+    };
+}
+function getOnboardTier(res) {
+    for (const tier of res.allowedTiers || []) {
+        if (tier.isDefault) {
+            return tier;
+        }
+    }
+    return {
+        name: '',
+        description: '',
+        id: UserTierId.LEGACY,
+        userDefinedCloudaicompanionProject: true,
+    };
+}
+//# sourceMappingURL=setup.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../../../src/code_assist/setup.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/C,MAAM,OAAO,sBAAuB,SAAQ,KAAK;IAC/C;QACE,KAAK,CACH,wHAAwH,CACzH,CAAC;IACJ,CAAC;CACF;AAOD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,MAAoB;IAClD,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,SAAS,CAAC;IACnE,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;IAC5E,MAAM,kBAAkB,GAAmB;QACzC,OAAO,EAAE,iBAAiB;QAC1B,QAAQ,EAAE,sBAAsB;QAChC,UAAU,EAAE,QAAQ;KACrB,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,cAAc,CAAC;QAC5C,uBAAuB,EAAE,SAAS;QAClC,QAAQ,EAAE;YACR,GAAG,kBAAkB;YACrB,WAAW,EAAE,SAAS;SACvB;KACF,CAAC,CAAC;IAEH,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;YACrC,IAAI,SAAS,EAAE,CAAC;gBACd,OAAO;oBACL,SAAS;oBACT,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,EAAE;iBACjC,CAAC;YACJ,CAAC;YACD,MAAM,IAAI,sBAAsB,EAAE,CAAC;QACrC,CAAC;QACD,OAAO;YACL,SAAS,EAAE,OAAO,CAAC,uBAAuB;YAC1C,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,EAAE;SACjC,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAErC,IAAI,UAA8B,CAAC;IACnC,IAAI,IAAI,CAAC,EAAE,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;QAChC,0IAA0I;QAC1I,UAAU,GAAG;YACX,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,uBAAuB,EAAE,SAAS;YAClC,QAAQ,EAAE,kBAAkB;SAC7B,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,UAAU,GAAG;YACX,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,uBAAuB,EAAE,SAAS;YAClC,QAAQ,EAAE;gBACR,GAAG,kBAAkB;gBACrB,WAAW,EAAE,SAAS;aACvB;SACF,CAAC;IACJ,CAAC;IAED,6DAA6D;IAC7D,IAAI,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IACpD,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;QAC9C,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAClD,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,uBAAuB,EAAE,EAAE,EAAE,CAAC;QAClD,IAAI,SAAS,EAAE,CAAC;YACd,OAAO;gBACL,SAAS;gBACT,QAAQ,EAAE,IAAI,CAAC,EAAE;aAClB,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,sBAAsB,EAAE,CAAC;IACrC,CAAC;IAED,OAAO;QACL,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAAC,EAAE;QACrD,QAAQ,EAAE,IAAI,CAAC,EAAE;KAClB,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,GAA2B;IACjD,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,YAAY,IAAI,EAAE,EAAE,CAAC;QAC1C,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO;QACL,IAAI,EAAE,EAAE;QACR,WAAW,EAAE,EAAE;QACf,EAAE,EAAE,UAAU,CAAC,MAAM;QACrB,kCAAkC,EAAE,IAAI;KACzC,CAAC;AACJ,CAAC"}

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

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

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

@@ -0,0 +1,171 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { setupUser, ProjectIdRequiredError } from './setup.js';
+import { CodeAssistServer } from '../code_assist/server.js';
+import { UserTierId } from './types.js';
+vi.mock('../code_assist/server.js');
+const mockPaidTier = {
+    id: UserTierId.STANDARD,
+    name: 'paid',
+    description: 'Paid tier',
+    isDefault: true,
+};
+const mockFreeTier = {
+    id: UserTierId.FREE,
+    name: 'free',
+    description: 'Free tier',
+    isDefault: true,
+};
+describe('setupUser for existing user', () => {
+    let mockLoad;
+    let mockOnboardUser;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        mockLoad = vi.fn();
+        mockOnboardUser = vi.fn().mockResolvedValue({
+            done: true,
+            response: {
+                cloudaicompanionProject: {
+                    id: 'server-project',
+                },
+            },
+        });
+        vi.mocked(CodeAssistServer).mockImplementation(() => ({
+            loadCodeAssist: mockLoad,
+            onboardUser: mockOnboardUser,
+        }));
+    });
+    afterEach(() => {
+        vi.unstubAllEnvs();
+    });
+    it('should use GOOGLE_CLOUD_PROJECT when set and project from server is undefined', async () => {
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', 'test-project');
+        mockLoad.mockResolvedValue({
+            currentTier: mockPaidTier,
+        });
+        await setupUser({});
+        expect(CodeAssistServer).toHaveBeenCalledWith({}, 'test-project', {}, '', undefined);
+    });
+    it('should ignore GOOGLE_CLOUD_PROJECT when project from server is set', async () => {
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', 'test-project');
+        mockLoad.mockResolvedValue({
+            cloudaicompanionProject: 'server-project',
+            currentTier: mockPaidTier,
+        });
+        const projectId = await setupUser({});
+        expect(CodeAssistServer).toHaveBeenCalledWith({}, 'test-project', {}, '', undefined);
+        expect(projectId).toEqual({
+            projectId: 'server-project',
+            userTier: 'standard-tier',
+        });
+    });
+    it('should throw ProjectIdRequiredError when no project ID is available', async () => {
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', '');
+        // And the server itself requires a project ID internally
+        vi.mocked(CodeAssistServer).mockImplementation(() => {
+            throw new ProjectIdRequiredError();
+        });
+        await expect(setupUser({})).rejects.toThrow(ProjectIdRequiredError);
+    });
+});
+describe('setupUser for new user', () => {
+    let mockLoad;
+    let mockOnboardUser;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        mockLoad = vi.fn();
+        mockOnboardUser = vi.fn().mockResolvedValue({
+            done: true,
+            response: {
+                cloudaicompanionProject: {
+                    id: 'server-project',
+                },
+            },
+        });
+        vi.mocked(CodeAssistServer).mockImplementation(() => ({
+            loadCodeAssist: mockLoad,
+            onboardUser: mockOnboardUser,
+        }));
+    });
+    afterEach(() => {
+        vi.unstubAllEnvs();
+    });
+    it('should use GOOGLE_CLOUD_PROJECT when set and onboard a new paid user', async () => {
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', 'test-project');
+        mockLoad.mockResolvedValue({
+            allowedTiers: [mockPaidTier],
+        });
+        const userData = await setupUser({});
+        expect(CodeAssistServer).toHaveBeenCalledWith({}, 'test-project', {}, '', undefined);
+        expect(mockLoad).toHaveBeenCalled();
+        expect(mockOnboardUser).toHaveBeenCalledWith({
+            tierId: 'standard-tier',
+            cloudaicompanionProject: 'test-project',
+            metadata: {
+                ideType: 'IDE_UNSPECIFIED',
+                platform: 'PLATFORM_UNSPECIFIED',
+                pluginType: 'GEMINI',
+                duetProject: 'test-project',
+            },
+        });
+        expect(userData).toEqual({
+            projectId: 'server-project',
+            userTier: 'standard-tier',
+        });
+    });
+    it('should onboard a new free user when GOOGLE_CLOUD_PROJECT is not set', async () => {
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', '');
+        mockLoad.mockResolvedValue({
+            allowedTiers: [mockFreeTier],
+        });
+        const userData = await setupUser({});
+        expect(CodeAssistServer).toHaveBeenCalledWith({}, undefined, {}, '', undefined);
+        expect(mockLoad).toHaveBeenCalled();
+        expect(mockOnboardUser).toHaveBeenCalledWith({
+            tierId: 'free-tier',
+            cloudaicompanionProject: undefined,
+            metadata: {
+                ideType: 'IDE_UNSPECIFIED',
+                platform: 'PLATFORM_UNSPECIFIED',
+                pluginType: 'GEMINI',
+            },
+        });
+        expect(userData).toEqual({
+            projectId: 'server-project',
+            userTier: 'free-tier',
+        });
+    });
+    it('should use GOOGLE_CLOUD_PROJECT when onboard response has no project ID', async () => {
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', 'test-project');
+        mockLoad.mockResolvedValue({
+            allowedTiers: [mockPaidTier],
+        });
+        mockOnboardUser.mockResolvedValue({
+            done: true,
+            response: {
+                cloudaicompanionProject: undefined,
+            },
+        });
+        const userData = await setupUser({});
+        expect(userData).toEqual({
+            projectId: 'test-project',
+            userTier: 'standard-tier',
+        });
+    });
+    it('should throw ProjectIdRequiredError when no project ID is available', async () => {
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', '');
+        mockLoad.mockResolvedValue({
+            allowedTiers: [mockPaidTier],
+        });
+        mockOnboardUser.mockResolvedValue({
+            done: true,
+            response: {},
+        });
+        await expect(setupUser({})).rejects.toThrow(ProjectIdRequiredError);
+    });
+});
+//# sourceMappingURL=setup.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"setup.test.js","sourceRoot":"","sources":["../../../src/code_assist/setup.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,SAAS,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAG5D,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAExC,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AAEpC,MAAM,YAAY,GAAmB;IACnC,EAAE,EAAE,UAAU,CAAC,QAAQ;IACvB,IAAI,EAAE,MAAM;IACZ,WAAW,EAAE,WAAW;IACxB,SAAS,EAAE,IAAI;CAChB,CAAC;AAEF,MAAM,YAAY,GAAmB;IACnC,EAAE,EAAE,UAAU,CAAC,IAAI;IACnB,IAAI,EAAE,MAAM;IACZ,WAAW,EAAE,WAAW;IACxB,SAAS,EAAE,IAAI;CAChB,CAAC;AAEF,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,QAAkC,CAAC;IACvC,IAAI,eAAyC,CAAC;IAE9C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACnB,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAC1C,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE;gBACR,uBAAuB,EAAE;oBACvB,EAAE,EAAE,gBAAgB;iBACrB;aACF;SACF,CAAC,CAAC;QACH,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,kBAAkB,CAC5C,GAAG,EAAE,CACH,CAAC;YACC,cAAc,EAAE,QAAQ;YACxB,WAAW,EAAE,eAAe;SAC7B,CAAgC,CACpC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAC7F,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,cAAc,CAAC,CAAC;QACnD,QAAQ,CAAC,iBAAiB,CAAC;YACzB,WAAW,EAAE,YAAY;SAC1B,CAAC,CAAC;QACH,MAAM,SAAS,CAAC,EAAkB,CAAC,CAAC;QACpC,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,EAAE,EACF,cAAc,EACd,EAAE,EACF,EAAE,EACF,SAAS,CACV,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,cAAc,CAAC,CAAC;QACnD,QAAQ,CAAC,iBAAiB,CAAC;YACzB,uBAAuB,EAAE,gBAAgB;YACzC,WAAW,EAAE,YAAY;SAC1B,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,EAAkB,CAAC,CAAC;QACtD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,EAAE,EACF,cAAc,EACd,EAAE,EACF,EAAE,EACF,SAAS,CACV,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC;YACxB,SAAS,EAAE,gBAAgB;YAC3B,QAAQ,EAAE,eAAe;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC;QACvC,yDAAyD;QACzD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;YAClD,MAAM,IAAI,sBAAsB,EAAE,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,SAAS,CAAC,EAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,sBAAsB,CACvB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,IAAI,QAAkC,CAAC;IACvC,IAAI,eAAyC,CAAC;IAE9C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACnB,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAC1C,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE;gBACR,uBAAuB,EAAE;oBACvB,EAAE,EAAE,gBAAgB;iBACrB;aACF;SACF,CAAC,CAAC;QACH,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,kBAAkB,CAC5C,GAAG,EAAE,CACH,CAAC;YACC,cAAc,EAAE,QAAQ;YACxB,WAAW,EAAE,eAAe;SAC7B,CAAgC,CACpC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,cAAc,CAAC,CAAC;QACnD,QAAQ,CAAC,iBAAiB,CAAC;YACzB,YAAY,EAAE,CAAC,YAAY,CAAC;SAC7B,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,EAAkB,CAAC,CAAC;QACrD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,EAAE,EACF,cAAc,EACd,EAAE,EACF,EAAE,EACF,SAAS,CACV,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpC,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAC3C,MAAM,EAAE,eAAe;YACvB,uBAAuB,EAAE,cAAc;YACvC,QAAQ,EAAE;gBACR,OAAO,EAAE,iBAAiB;gBAC1B,QAAQ,EAAE,sBAAsB;gBAChC,UAAU,EAAE,QAAQ;gBACpB,WAAW,EAAE,cAAc;aAC5B;SACF,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,SAAS,EAAE,gBAAgB;YAC3B,QAAQ,EAAE,eAAe;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC;QACvC,QAAQ,CAAC,iBAAiB,CAAC;YACzB,YAAY,EAAE,CAAC,YAAY,CAAC;SAC7B,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,EAAkB,CAAC,CAAC;QACrD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,EAAE,EACF,SAAS,EACT,EAAE,EACF,EAAE,EACF,SAAS,CACV,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpC,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC;YAC3C,MAAM,EAAE,WAAW;YACnB,uBAAuB,EAAE,SAAS;YAClC,QAAQ,EAAE;gBACR,OAAO,EAAE,iBAAiB;gBAC1B,QAAQ,EAAE,sBAAsB;gBAChC,UAAU,EAAE,QAAQ;aACrB;SACF,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,SAAS,EAAE,gBAAgB;YAC3B,QAAQ,EAAE,WAAW;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,cAAc,CAAC,CAAC;QACnD,QAAQ,CAAC,iBAAiB,CAAC;YACzB,YAAY,EAAE,CAAC,YAAY,CAAC;SAC7B,CAAC,CAAC;QACH,eAAe,CAAC,iBAAiB,CAAC;YAChC,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE;gBACR,uBAAuB,EAAE,SAAS;aACnC;SACF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,EAAkB,CAAC,CAAC;QACrD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,SAAS,EAAE,cAAc;YACzB,QAAQ,EAAE,eAAe;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC;QACvC,QAAQ,CAAC,iBAAiB,CAAC;YACzB,YAAY,EAAE,CAAC,YAAY,CAAC;SAC7B,CAAC,CAAC;QACH,eAAe,CAAC,iBAAiB,CAAC;YAChC,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,SAAS,CAAC,EAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,sBAAsB,CACvB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,148 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export interface ClientMetadata {
+    ideType?: ClientMetadataIdeType;
+    ideVersion?: string;
+    pluginVersion?: string;
+    platform?: ClientMetadataPlatform;
+    updateChannel?: string;
+    duetProject?: string;
+    pluginType?: ClientMetadataPluginType;
+    ideName?: string;
+}
+export type ClientMetadataIdeType = 'IDE_UNSPECIFIED' | 'VSCODE' | 'INTELLIJ' | 'VSCODE_CLOUD_WORKSTATION' | 'INTELLIJ_CLOUD_WORKSTATION' | 'CLOUD_SHELL';
+export type ClientMetadataPlatform = 'PLATFORM_UNSPECIFIED' | 'DARWIN_AMD64' | 'DARWIN_ARM64' | 'LINUX_AMD64' | 'LINUX_ARM64' | 'WINDOWS_AMD64';
+export type ClientMetadataPluginType = 'PLUGIN_UNSPECIFIED' | 'CLOUD_CODE' | 'GEMINI' | 'AIPLUGIN_INTELLIJ' | 'AIPLUGIN_STUDIO';
+export interface LoadCodeAssistRequest {
+    cloudaicompanionProject?: string;
+    metadata: ClientMetadata;
+}
+/**
+ * Represents LoadCodeAssistResponse proto json field
+ * http://google3/google/internal/cloud/code/v1internal/cloudcode.proto;l=224
+ */
+export interface LoadCodeAssistResponse {
+    currentTier?: GeminiUserTier | null;
+    allowedTiers?: GeminiUserTier[] | null;
+    ineligibleTiers?: IneligibleTier[] | null;
+    cloudaicompanionProject?: string | null;
+}
+/**
+ * GeminiUserTier reflects the structure received from the CodeAssist when calling LoadCodeAssist.
+ */
+export interface GeminiUserTier {
+    id: UserTierId;
+    name: string;
+    description: string;
+    userDefinedCloudaicompanionProject?: boolean | null;
+    isDefault?: boolean;
+    privacyNotice?: PrivacyNotice;
+    hasAcceptedTos?: boolean;
+    hasOnboardedPreviously?: boolean;
+}
+/**
+ * Includes information specifying the reasons for a user's ineligibility for a specific tier.
+ * @param reasonCode mnemonic code representing the reason for in-eligibility.
+ * @param reasonMessage message to display to the user.
+ * @param tierId id of the tier.
+ * @param tierName name of the tier.
+ */
+export interface IneligibleTier {
+    reasonCode: IneligibleTierReasonCode;
+    reasonMessage: string;
+    tierId: UserTierId;
+    tierName: string;
+}
+/**
+ * List of predefined reason codes when a tier is blocked from a specific tier.
+ * https://source.corp.google.com/piper///depot/google3/google/internal/cloud/code/v1internal/cloudcode.proto;l=378
+ */
+export declare enum IneligibleTierReasonCode {
+    DASHER_USER = "DASHER_USER",
+    INELIGIBLE_ACCOUNT = "INELIGIBLE_ACCOUNT",
+    NON_USER_ACCOUNT = "NON_USER_ACCOUNT",
+    RESTRICTED_AGE = "RESTRICTED_AGE",
+    RESTRICTED_NETWORK = "RESTRICTED_NETWORK",
+    UNKNOWN = "UNKNOWN",
+    UNKNOWN_LOCATION = "UNKNOWN_LOCATION",
+    UNSUPPORTED_LOCATION = "UNSUPPORTED_LOCATION"
+}
+/**
+ * UserTierId represents IDs returned from the Cloud Code Private API representing a user's tier
+ *
+ * //depot/google3/cloud/developer_experience/cloudcode/pa/service/usertier.go;l=16
+ */
+export declare enum UserTierId {
+    FREE = "free-tier",
+    LEGACY = "legacy-tier",
+    STANDARD = "standard-tier"
+}
+/**
+ * PrivacyNotice reflects the structure received from the CodeAssist in regards to a tier
+ * privacy notice.
+ */
+export interface PrivacyNotice {
+    showNotice: boolean;
+    noticeText?: string;
+}
+/**
+ * Proto signature of OnboardUserRequest as payload to OnboardUser call
+ */
+export interface OnboardUserRequest {
+    tierId: string | undefined;
+    cloudaicompanionProject: string | undefined;
+    metadata: ClientMetadata | undefined;
+}
+/**
+ * Represents LongRunningOperation proto
+ * http://google3/google/longrunning/operations.proto;rcl=698857719;l=107
+ */
+export interface LongRunningOperationResponse {
+    name: string;
+    done?: boolean;
+    response?: OnboardUserResponse;
+}
+/**
+ * Represents OnboardUserResponse proto
+ * http://google3/google/internal/cloud/code/v1internal/cloudcode.proto;l=215
+ */
+export interface OnboardUserResponse {
+    cloudaicompanionProject?: {
+        id: string;
+        name: string;
+    };
+}
+/**
+ * Status code of user license status
+ * it does not strictly correspond to the proto
+ * Error value is an additional value assigned to error responses from OnboardUser
+ */
+export declare enum OnboardUserStatusCode {
+    Default = "DEFAULT",
+    Notice = "NOTICE",
+    Warning = "WARNING",
+    Error = "ERROR"
+}
+/**
+ * Status of user onboarded to gemini
+ */
+export interface OnboardUserStatus {
+    statusCode: OnboardUserStatusCode;
+    displayMessage: string;
+    helpLink: HelpLinkUrl | undefined;
+}
+export interface HelpLinkUrl {
+    description: string;
+    url: string;
+}
+export interface SetCodeAssistGlobalUserSettingRequest {
+    cloudaicompanionProject?: string;
+    freeTierDataCollectionOptin: boolean;
+}
+export interface CodeAssistGlobalUserSettingResponse {
+    cloudaicompanionProject?: string;
+    freeTierDataCollectionOptin: boolean;
+}

package/dist/src/code_assist/types.js

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * List of predefined reason codes when a tier is blocked from a specific tier.
+ * https://source.corp.google.com/piper///depot/google3/google/internal/cloud/code/v1internal/cloudcode.proto;l=378
+ */
+export var IneligibleTierReasonCode;
+(function (IneligibleTierReasonCode) {
+    // go/keep-sorted start
+    IneligibleTierReasonCode["DASHER_USER"] = "DASHER_USER";
+    IneligibleTierReasonCode["INELIGIBLE_ACCOUNT"] = "INELIGIBLE_ACCOUNT";
+    IneligibleTierReasonCode["NON_USER_ACCOUNT"] = "NON_USER_ACCOUNT";
+    IneligibleTierReasonCode["RESTRICTED_AGE"] = "RESTRICTED_AGE";
+    IneligibleTierReasonCode["RESTRICTED_NETWORK"] = "RESTRICTED_NETWORK";
+    IneligibleTierReasonCode["UNKNOWN"] = "UNKNOWN";
+    IneligibleTierReasonCode["UNKNOWN_LOCATION"] = "UNKNOWN_LOCATION";
+    IneligibleTierReasonCode["UNSUPPORTED_LOCATION"] = "UNSUPPORTED_LOCATION";
+    // go/keep-sorted end
+})(IneligibleTierReasonCode || (IneligibleTierReasonCode = {}));
+/**
+ * UserTierId represents IDs returned from the Cloud Code Private API representing a user's tier
+ *
+ * //depot/google3/cloud/developer_experience/cloudcode/pa/service/usertier.go;l=16
+ */
+export var UserTierId;
+(function (UserTierId) {
+    UserTierId["FREE"] = "free-tier";
+    UserTierId["LEGACY"] = "legacy-tier";
+    UserTierId["STANDARD"] = "standard-tier";
+})(UserTierId || (UserTierId = {}));
+/**
+ * Status code of user license status
+ * it does not strictly correspond to the proto
+ * Error value is an additional value assigned to error responses from OnboardUser
+ */
+export var OnboardUserStatusCode;
+(function (OnboardUserStatusCode) {
+    OnboardUserStatusCode["Default"] = "DEFAULT";
+    OnboardUserStatusCode["Notice"] = "NOTICE";
+    OnboardUserStatusCode["Warning"] = "WARNING";
+    OnboardUserStatusCode["Error"] = "ERROR";
+})(OnboardUserStatusCode || (OnboardUserStatusCode = {}));
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/code_assist/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AA+EH;;;GAGG;AACH,MAAM,CAAN,IAAY,wBAWX;AAXD,WAAY,wBAAwB;IAClC,uBAAuB;IACvB,uDAA2B,CAAA;IAC3B,qEAAyC,CAAA;IACzC,iEAAqC,CAAA;IACrC,6DAAiC,CAAA;IACjC,qEAAyC,CAAA;IACzC,+CAAmB,CAAA;IACnB,iEAAqC,CAAA;IACrC,yEAA6C,CAAA;IAC7C,qBAAqB;AACvB,CAAC,EAXW,wBAAwB,KAAxB,wBAAwB,QAWnC;AACD;;;;GAIG;AACH,MAAM,CAAN,IAAY,UAIX;AAJD,WAAY,UAAU;IACpB,gCAAkB,CAAA;IAClB,oCAAsB,CAAA;IACtB,wCAA0B,CAAA;AAC5B,CAAC,EAJW,UAAU,KAAV,UAAU,QAIrB;AA0CD;;;;GAIG;AACH,MAAM,CAAN,IAAY,qBAKX;AALD,WAAY,qBAAqB;IAC/B,4CAAmB,CAAA;IACnB,0CAAiB,CAAA;IACjB,4CAAmB,CAAA;IACnB,wCAAe,CAAA;AACjB,CAAC,EALW,qBAAqB,KAArB,qBAAqB,QAKhC"}