@google/gemini-cli-core 0.21.0-nightly.20251216.bb0c0d8ee → 0.21.0-nightly.20251218.739c02bd6

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

@@ -0,0 +1,300 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { createConversationOffered, formatProtoJsonDuration, recordConversationOffered, recordToolCallInteractions, } from './telemetry.js';
+import { ActionStatus, ConversationInteractionInteraction, } from './types.js';
+import { FinishReason, GenerateContentResponse, } from '@google/genai';
+import * as codeAssist from './codeAssist.js';
+import { ToolConfirmationOutcome, } from '../tools/tools.js';
+function createMockResponse(candidates = [], ok = true, functionCalls = undefined) {
+    const response = new GenerateContentResponse();
+    response.candidates = candidates;
+    response.sdkHttpResponse = {
+        responseInternal: {
+            ok,
+        },
+        json: async () => ({}),
+    };
+    // If functionCalls is explicitly provided, mock the getter.
+    // Otherwise, let the default behavior (if any) or undefined prevail.
+    // In the real SDK, functionCalls is a getter derived from candidates.
+    // For testing `createConversationOffered` which guards on functionCalls,
+    // we often need to force it to be present.
+    if (functionCalls !== undefined) {
+        Object.defineProperty(response, 'functionCalls', {
+            get: () => functionCalls,
+            configurable: true,
+        });
+    }
+    return response;
+}
+describe('telemetry', () => {
+    describe('createConversationOffered', () => {
+        it('should create a ConversationOffered object with correct values', () => {
+            const response = createMockResponse([
+                {
+                    index: 0,
+                    content: {
+                        role: 'model',
+                        parts: [{ text: 'response with ```code```' }],
+                    },
+                    citationMetadata: {
+                        citations: [
+                            { uri: 'https://example.com', startIndex: 0, endIndex: 10 },
+                        ],
+                    },
+                    finishReason: FinishReason.STOP,
+                },
+            ], true, [{ name: 'someTool', args: {} }]);
+            const traceId = 'test-trace-id';
+            const streamingLatency = { totalLatency: '1s' };
+            const result = createConversationOffered(response, traceId, undefined, streamingLatency);
+            expect(result).toEqual({
+                citationCount: '1',
+                includedCode: true,
+                status: ActionStatus.ACTION_STATUS_NO_ERROR,
+                traceId,
+                streamingLatency,
+                isAgentic: true,
+            });
+        });
+        it('should return undefined if no function calls', () => {
+            const response = createMockResponse([
+                {
+                    index: 0,
+                    content: {
+                        role: 'model',
+                        parts: [{ text: 'response without function calls' }],
+                    },
+                },
+            ], true, []);
+            const result = createConversationOffered(response, 'trace-id', undefined, {});
+            expect(result).toBeUndefined();
+        });
+        it('should set status to CANCELLED if signal is aborted', () => {
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
+            const signal = new AbortController().signal;
+            vi.spyOn(signal, 'aborted', 'get').mockReturnValue(true);
+            const result = createConversationOffered(response, 'trace-id', signal, {});
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_CANCELLED);
+        });
+        it('should set status to ERROR_UNKNOWN if response has error (non-OK SDK response)', () => {
+            const response = createMockResponse([], false, [
+                { name: 'tool', args: {} },
+            ]);
+            const result = createConversationOffered(response, 'trace-id', undefined, {});
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_ERROR_UNKNOWN);
+        });
+        it('should set status to ERROR_UNKNOWN if finishReason is not STOP or MAX_TOKENS', () => {
+            const response = createMockResponse([
+                {
+                    index: 0,
+                    finishReason: FinishReason.SAFETY,
+                },
+            ], true, [{ name: 'tool', args: {} }]);
+            const result = createConversationOffered(response, 'trace-id', undefined, {});
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_ERROR_UNKNOWN);
+        });
+        it('should set status to EMPTY if candidates is empty', () => {
+            // We force functionCalls to be present to bypass the guard,
+            // simulating a state where we want to test the candidates check.
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
+            const result = createConversationOffered(response, 'trace-id', undefined, {});
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_EMPTY);
+        });
+        it('should detect code in response', () => {
+            const response = createMockResponse([
+                {
+                    index: 0,
+                    content: {
+                        parts: [
+                            { text: 'Here is some code:\n```js\nconsole.log("hi")\n```' },
+                        ],
+                    },
+                },
+            ], true, [{ name: 'tool', args: {} }]);
+            const result = createConversationOffered(response, 'id', undefined, {});
+            expect(result?.includedCode).toBe(true);
+        });
+        it('should not detect code if no backticks', () => {
+            const response = createMockResponse([
+                {
+                    index: 0,
+                    content: {
+                        parts: [{ text: 'Here is some text.' }],
+                    },
+                },
+            ], true, [{ name: 'tool', args: {} }]);
+            const result = createConversationOffered(response, 'id', undefined, {});
+            expect(result?.includedCode).toBe(false);
+        });
+    });
+    describe('formatProtoJsonDuration', () => {
+        it('should format milliseconds to seconds string', () => {
+            expect(formatProtoJsonDuration(1500)).toBe('1.5s');
+            expect(formatProtoJsonDuration(100)).toBe('0.1s');
+        });
+    });
+    describe('recordConversationOffered', () => {
+        it('should call server.recordConversationOffered if traceId is present', async () => {
+            const serverMock = {
+                recordConversationOffered: vi.fn(),
+            };
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
+            const streamingLatency = {};
+            await recordConversationOffered(serverMock, 'trace-id', response, streamingLatency, undefined);
+            expect(serverMock.recordConversationOffered).toHaveBeenCalledWith(expect.objectContaining({
+                traceId: 'trace-id',
+            }));
+        });
+        it('should not call server.recordConversationOffered if traceId is undefined', async () => {
+            const serverMock = {
+                recordConversationOffered: vi.fn(),
+            };
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
+            await recordConversationOffered(serverMock, undefined, response, {}, undefined);
+            expect(serverMock.recordConversationOffered).not.toHaveBeenCalled();
+        });
+    });
+    describe('recordToolCallInteractions', () => {
+        let mockServer;
+        beforeEach(() => {
+            mockServer = {
+                recordConversationInteraction: vi.fn(),
+            };
+            vi.spyOn(codeAssist, 'getCodeAssistServer').mockReturnValue(mockServer);
+        });
+        afterEach(() => {
+            vi.restoreAllMocks();
+        });
+        it('should record ACCEPT_FILE interaction for accepted edit tools', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'replace', // in EDIT_TOOL_NAMES
+                        args: {},
+                        callId: 'call-1',
+                        isClientInitiated: false,
+                        prompt_id: 'p1',
+                        traceId: 'trace-1',
+                    },
+                    outcome: ToolConfirmationOutcome.ProceedOnce,
+                    status: 'success',
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).toHaveBeenCalledWith({
+                traceId: 'trace-1',
+                status: ActionStatus.ACTION_STATUS_NO_ERROR,
+                interaction: ConversationInteractionInteraction.ACCEPT_FILE,
+                isAgentic: true,
+            });
+        });
+        it('should record UNKNOWN interaction for other accepted tools', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'read_file', // NOT in EDIT_TOOL_NAMES
+                        args: {},
+                        callId: 'call-2',
+                        isClientInitiated: false,
+                        prompt_id: 'p2',
+                        traceId: 'trace-2',
+                    },
+                    outcome: ToolConfirmationOutcome.ProceedOnce,
+                    status: 'success',
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).toHaveBeenCalledWith({
+                traceId: 'trace-2',
+                status: ActionStatus.ACTION_STATUS_NO_ERROR,
+                interaction: ConversationInteractionInteraction.UNKNOWN,
+                isAgentic: true,
+            });
+        });
+        it('should not record interaction for cancelled status', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'tool',
+                        args: {},
+                        callId: 'call-3',
+                        isClientInitiated: false,
+                        prompt_id: 'p3',
+                        traceId: 'trace-3',
+                    },
+                    status: 'cancelled',
+                    response: {},
+                    tool: {},
+                    invocation: {},
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).not.toHaveBeenCalled();
+        });
+        it('should not record interaction for error status', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'tool',
+                        args: {},
+                        callId: 'call-4',
+                        isClientInitiated: false,
+                        prompt_id: 'p4',
+                        traceId: 'trace-4',
+                    },
+                    status: 'error',
+                    response: {
+                        error: new Error('fail'),
+                    },
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).not.toHaveBeenCalled();
+        });
+        it('should not record interaction if tool calls are mixed or not 100% accepted', async () => {
+            // Logic: traceId && acceptedToolCalls / toolCalls.length >= 1
+            const toolCalls = [
+                {
+                    request: {
+                        name: 't1',
+                        args: {},
+                        callId: 'c1',
+                        isClientInitiated: false,
+                        prompt_id: 'p1',
+                        traceId: 't1',
+                    },
+                    outcome: ToolConfirmationOutcome.ProceedOnce,
+                    status: 'success',
+                },
+                {
+                    request: {
+                        name: 't2',
+                        args: {},
+                        callId: 'c2',
+                        isClientInitiated: false,
+                        prompt_id: 'p1',
+                        traceId: 't1',
+                    },
+                    outcome: ToolConfirmationOutcome.Cancel, // Rejected
+                    status: 'success',
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).not.toHaveBeenCalled();
+        });
+    });
+});
+//# sourceMappingURL=telemetry.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.test.js","sourceRoot":"","sources":["../../../src/code_assist/telemetry.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,EACL,yBAAyB,EACzB,uBAAuB,EACvB,yBAAyB,EACzB,0BAA0B,GAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,YAAY,EACZ,kCAAkC,GAEnC,MAAM,YAAY,CAAC;AACpB,OAAO,EACL,YAAY,EACZ,uBAAuB,GAExB,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,UAAU,MAAM,iBAAiB,CAAC;AAG9C,OAAO,EACL,uBAAuB,GAGxB,MAAM,mBAAmB,CAAC;AAI3B,SAAS,kBAAkB,CACzB,aAAoD,EAAE,EACtD,EAAE,GAAG,IAAI,EACT,gBAA4C,SAAS;IAErD,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;IAC/C,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,QAAQ,CAAC,eAAe,GAAG;QACzB,gBAAgB,EAAE;YAChB,EAAE;SACoB;QACxB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;KACvB,CAAC;IAEF,4DAA4D;IAC5D,qEAAqE;IACrE,sEAAsE;IACtE,yEAAyE;IACzE,2CAA2C;IAC3C,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,eAAe,EAAE;YAC/C,GAAG,EAAE,GAAG,EAAE,CAAC,aAAa;YACxB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC;qBAC9C;oBACD,gBAAgB,EAAE;wBAChB,SAAS,EAAE;4BACT,EAAE,GAAG,EAAE,qBAAqB,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;yBAC5D;qBACF;oBACD,YAAY,EAAE,YAAY,CAAC,IAAI;iBAChC;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CACjC,CAAC;YACF,MAAM,OAAO,GAAG,eAAe,CAAC;YAChC,MAAM,gBAAgB,GAAqB,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;YAElE,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,OAAO,EACP,SAAS,EACT,gBAAgB,CACjB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,aAAa,EAAE,GAAG;gBAClB,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE,YAAY,CAAC,sBAAsB;gBAC3C,OAAO;gBACP,gBAAgB;gBAChB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iCAAiC,EAAE,CAAC;qBACrD;iBACF;aACF,EACD,IAAI,EACJ,EAAE,CACH,CAAC;YACF,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YAC5C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,MAAM,EACN,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,uBAAuB,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,KAAK,EAAE;gBAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,YAAY,EAAE,YAAY,CAAC,MAAM;iBAClC;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAC7B,CAAC;YAEF,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,4DAA4D;YAC5D,iEAAiE;YACjE,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,KAAK,EAAE;4BACL,EAAE,IAAI,EAAE,mDAAmD,EAAE;yBAC9D;qBACF;iBACF;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAC7B,CAAC;YACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;qBACxC;iBACF;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAC7B,CAAC;YACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,UAAU,GAAG;gBACjB,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE;aACJ,CAAC;YAEjC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YACH,MAAM,gBAAgB,GAAG,EAAE,CAAC;YAE5B,MAAM,yBAAyB,CAC7B,UAAU,EACV,UAAU,EACV,QAAQ,EACR,gBAAgB,EAChB,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,oBAAoB,CAC/D,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,UAAU;aACpB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,UAAU,GAAG;gBACjB,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE;aACJ,CAAC;YACjC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,yBAAyB,CAC7B,UAAU,EACV,SAAS,EACT,QAAQ,EACR,EAAE,EACF,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,IAAI,UAAuE,CAAC;QAE5E,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,6BAA6B,EAAE,EAAE,CAAC,EAAE,EAAE;aACvC,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,eAAe,CACzD,UAAyC,CAC1C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,SAAS,EAAE,qBAAqB;wBACtC,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,OAAO,EAAE,uBAAuB,CAAC,WAAW;oBAC5C,MAAM,EAAE,SAAS;iBACc;aAClC,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,oBAAoB,CAAC;gBACpE,OAAO,EAAE,SAAS;gBAClB,MAAM,EAAE,YAAY,CAAC,sBAAsB;gBAC3C,WAAW,EAAE,kCAAkC,CAAC,WAAW;gBAC3D,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW,EAAE,yBAAyB;wBAC5C,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,OAAO,EAAE,uBAAuB,CAAC,WAAW;oBAC5C,MAAM,EAAE,SAAS;iBACc;aAClC,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,oBAAoB,CAAC;gBACpE,OAAO,EAAE,SAAS;gBAClB,MAAM,EAAE,YAAY,CAAC,sBAAsB;gBAC3C,WAAW,EAAE,kCAAkC,CAAC,OAAO;gBACvD,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,MAAM,EAAE,WAAW;oBACnB,QAAQ,EAAE,EAAqC;oBAC/C,IAAI,EAAE,EAAmC;oBACzC,UAAU,EAAE,EAAkC;iBAC1B;aACvB,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE;wBACR,KAAK,EAAE,IAAI,KAAK,CAAC,MAAM,CAAC;qBACU;iBAChB;aACvB,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;YAC1F,8DAA8D;YAC9D,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,IAAI;wBACV,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,IAAI;wBACZ,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI;qBACd;oBACD,OAAO,EAAE,uBAAuB,CAAC,WAAW;oBAC5C,MAAM,EAAE,SAAS;iBAClB;gBACD;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,IAAI;wBACV,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,IAAI;wBACZ,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI;qBACd;oBACD,OAAO,EAAE,uBAAuB,CAAC,MAAM,EAAE,WAAW;oBACpD,MAAM,EAAE,SAAS;iBAClB;aACgC,CAAC;YAEpC,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -140,7 +140,7 @@ export interface HelpLinkUrl {
 }
 export interface SetCodeAssistGlobalUserSettingRequest {
     cloudaicompanionProject?: string;
-    freeTierDataCollectionOptin: boolean;
+    freeTierDataCollectionOptin?: boolean;
 }
 export interface CodeAssistGlobalUserSettingResponse {
     cloudaicompanionProject?: string;
@@ -174,4 +174,55 @@ export interface BucketInfo {
 export interface RetrieveUserQuotaResponse {
     buckets?: BucketInfo[];
 }
+export interface RecordCodeAssistMetricsRequest {
+    project: string;
+    requestId?: string;
+    metadata?: ClientMetadata;
+    metrics?: CodeAssistMetric[];
+}
+export interface CodeAssistMetric {
+    timestamp?: string;
+    metricMetadata?: Map<string, string>;
+    conversationOffered?: ConversationOffered;
+    conversationInteraction?: ConversationInteraction;
+}
+export declare enum ConversationInteractionInteraction {
+    UNKNOWN = 0,
+    THUMBSUP = 1,
+    THUMBSDOWN = 2,
+    COPY = 3,
+    INSERT = 4,
+    ACCEPT_CODE_BLOCK = 5,
+    ACCEPT_ALL = 6,
+    ACCEPT_FILE = 7,
+    DIFF = 8,
+    ACCEPT_RANGE = 9
+}
+export declare enum ActionStatus {
+    ACTION_STATUS_UNSPECIFIED = 0,
+    ACTION_STATUS_NO_ERROR = 1,
+    ACTION_STATUS_ERROR_UNKNOWN = 2,
+    ACTION_STATUS_CANCELLED = 3,
+    ACTION_STATUS_EMPTY = 4
+}
+export interface ConversationOffered {
+    citationCount?: string;
+    includedCode?: boolean;
+    status?: ActionStatus;
+    traceId?: string;
+    streamingLatency?: StreamingLatency;
+    isAgentic?: boolean;
+}
+export interface StreamingLatency {
+    firstMessageLatency?: string;
+    totalLatency?: string;
+}
+export interface ConversationInteraction {
+    traceId: string;
+    status?: ActionStatus;
+    interaction?: ConversationInteractionInteraction;
+    acceptedLines?: string;
+    language?: string;
+    isAgentic?: boolean;
+}
 export {};

package/dist/src/code_assist/types.js

@@ -43,4 +43,25 @@ export var OnboardUserStatusCode;
     OnboardUserStatusCode["Warning"] = "WARNING";
     OnboardUserStatusCode["Error"] = "ERROR";
 })(OnboardUserStatusCode || (OnboardUserStatusCode = {}));
+export var ConversationInteractionInteraction;
+(function (ConversationInteractionInteraction) {
+    ConversationInteractionInteraction[ConversationInteractionInteraction["UNKNOWN"] = 0] = "UNKNOWN";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["THUMBSUP"] = 1] = "THUMBSUP";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["THUMBSDOWN"] = 2] = "THUMBSDOWN";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["COPY"] = 3] = "COPY";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["INSERT"] = 4] = "INSERT";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["ACCEPT_CODE_BLOCK"] = 5] = "ACCEPT_CODE_BLOCK";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["ACCEPT_ALL"] = 6] = "ACCEPT_ALL";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["ACCEPT_FILE"] = 7] = "ACCEPT_FILE";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["DIFF"] = 8] = "DIFF";
+    ConversationInteractionInteraction[ConversationInteractionInteraction["ACCEPT_RANGE"] = 9] = "ACCEPT_RANGE";
+})(ConversationInteractionInteraction || (ConversationInteractionInteraction = {}));
+export var ActionStatus;
+(function (ActionStatus) {
+    ActionStatus[ActionStatus["ACTION_STATUS_UNSPECIFIED"] = 0] = "ACTION_STATUS_UNSPECIFIED";
+    ActionStatus[ActionStatus["ACTION_STATUS_NO_ERROR"] = 1] = "ACTION_STATUS_NO_ERROR";
+    ActionStatus[ActionStatus["ACTION_STATUS_ERROR_UNKNOWN"] = 2] = "ACTION_STATUS_ERROR_UNKNOWN";
+    ActionStatus[ActionStatus["ACTION_STATUS_CANCELLED"] = 3] = "ACTION_STATUS_CANCELLED";
+    ActionStatus[ActionStatus["ACTION_STATUS_EMPTY"] = 4] = "ACTION_STATUS_EMPTY";
+})(ActionStatus || (ActionStatus = {}));
 //# sourceMappingURL=types.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/code_assist/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgFH;;;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"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/code_assist/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgFH;;;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;AA2ED,MAAM,CAAN,IAAY,kCAWX;AAXD,WAAY,kCAAkC;IAC5C,iGAAW,CAAA;IACX,mGAAY,CAAA;IACZ,uGAAc,CAAA;IACd,2FAAQ,CAAA;IACR,+FAAU,CAAA;IACV,qHAAqB,CAAA;IACrB,uGAAc,CAAA;IACd,yGAAe,CAAA;IACf,2FAAQ,CAAA;IACR,2GAAgB,CAAA;AAClB,CAAC,EAXW,kCAAkC,KAAlC,kCAAkC,QAW7C;AAED,MAAM,CAAN,IAAY,YAMX;AAND,WAAY,YAAY;IACtB,yFAA6B,CAAA;IAC7B,mFAA0B,CAAA;IAC1B,6FAA+B,CAAA;IAC/B,qFAA2B,CAAA;IAC3B,6EAAuB,CAAA;AACzB,CAAC,EANW,YAAY,KAAZ,YAAY,QAMvB"}

package/dist/src/config/config.d.ts

@@ -36,6 +36,7 @@ import { PolicyEngine } from '../policy/policy-engine.js';
 import type { PolicyEngineConfig } from '../policy/types.js';
 import { HookSystem } from '../hooks/index.js';
 import type { UserTierId } from '../code_assist/types.js';
+import type { RetrieveUserQuotaResponse } from '../code_assist/types.js';
 import type { Experiments } from '../code_assist/experiments/experiments.js';
 import { AgentRegistry } from '../agents/registry.js';
 import { ApprovalMode } from '../policy/types.js';
@@ -229,7 +230,6 @@ export interface ConfigParameters {
     });
     previewFeatures?: boolean;
     enableAgents?: boolean;
-    enableModelAvailabilityService?: boolean;
     experimentalJitContext?: boolean;
 }
 export declare class Config {
@@ -279,10 +279,10 @@ export declare class Config {
     private readonly bugCommand;
     private model;
     private previewFeatures;
+    private hasAccessToPreviewModel;
     private readonly noBrowser;
     private readonly folderTrust;
     private ideMode;
-    private inFallbackMode;
     private _activeModel;
     private readonly maxSessionTurns;
     private readonly listSessions;
@@ -336,9 +336,6 @@ export declare class Config {
     private experiments;
     private experimentsPromise;
     private hookSystem?;
-    private previewModelFallbackMode;
-    private previewModelBypassMode;
-    private readonly enableModelAvailabilityService;
     private readonly enableAgents;
     private readonly experimentalJitContext;
     private contextManager?;
@@ -365,15 +362,9 @@ export declare class Config {
     setModel(newModel: string): void;
     getActiveModel(): string;
     setActiveModel(model: string): void;
-    resetTurn(): void;
-    isInFallbackMode(): boolean;
-    setFallbackMode(active: boolean): void;
     setFallbackModelHandler(handler: FallbackModelHandler): void;
     getFallbackModelHandler(): FallbackModelHandler | undefined;
-    isPreviewModelFallbackMode(): boolean;
-    setPreviewModelFallbackMode(active: boolean): void;
-    isPreviewModelBypassMode(): boolean;
-    setPreviewModelBypassMode(active: boolean): void;
+    resetTurn(): void;
     getMaxSessionTurns(): number;
     setQuotaErrorOccurred(value: boolean): void;
     getQuotaErrorOccurred(): boolean;
@@ -391,6 +382,9 @@ export declare class Config {
     getQuestion(): string | undefined;
     getPreviewFeatures(): boolean | undefined;
     setPreviewFeatures(previewFeatures: boolean): void;
+    getHasAccessToPreviewModel(): boolean;
+    setHasAccessToPreviewModel(hasAccess: boolean): void;
+    refreshUserQuota(): Promise<RetrieveUserQuotaResponse | undefined>;
     getCoreTools(): string[] | undefined;
     getAllowedTools(): string[] | undefined;
     /**
@@ -472,7 +466,6 @@ export declare class Config {
     getExtensionLoader(): ExtensionLoader;
     getEnabledExtensions(): string[];
     getEnableExtensionReloading(): boolean;
-    isModelAvailabilityServiceEnabled(): boolean;
     isAgentsEnabled(): boolean;
     getNoBrowser(): boolean;
     isBrowserLaunchSuppressed(): boolean;