@terminai/core 0.21.0

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

@@ -0,0 +1,172 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * 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;;;;;GAKG;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/telemetry.d.ts

@@ -0,0 +1,15 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type GenerateContentResponse } from '@google/genai';
+import { type ConversationOffered, type StreamingLatency } from './types.js';
+import type { CompletedToolCall } from '../core/coreToolScheduler.js';
+import type { Config } from '../config/config.js';
+import type { CodeAssistServer } from './server.js';
+export declare function recordConversationOffered(server: CodeAssistServer, traceId: string | undefined, response: GenerateContentResponse, streamingLatency: StreamingLatency, abortSignal: AbortSignal | undefined): Promise<void>;
+export declare function recordToolCallInteractions(config: Config, toolCalls: CompletedToolCall[]): Promise<void>;
+export declare function createConversationOffered(response: GenerateContentResponse, traceId: string, signal: AbortSignal | undefined, streamingLatency: StreamingLatency): ConversationOffered | undefined;
+export declare function formatProtoJsonDuration(milliseconds: number): string;

package/dist/src/code_assist/telemetry.js

@@ -0,0 +1,157 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FinishReason } from '@google/genai';
+import { getCitations } from '../utils/generateContentResponseUtilities.js';
+import { ActionStatus, ConversationInteractionInteraction, } from './types.js';
+import { debugLogger } from '../utils/debugLogger.js';
+import { getCodeAssistServer } from './codeAssist.js';
+import { EDIT_TOOL_NAMES } from '../tools/tool-names.js';
+import { getErrorMessage } from '../utils/errors.js';
+import { ToolConfirmationOutcome } from '../tools/tools.js';
+export async function recordConversationOffered(server, traceId, response, streamingLatency, abortSignal) {
+    try {
+        if (traceId) {
+            const offered = createConversationOffered(response, traceId, abortSignal, streamingLatency);
+            if (offered) {
+                await server.recordConversationOffered(offered);
+            }
+        }
+    }
+    catch (error) {
+        debugLogger.warn(`Error recording tool call interactions: ${getErrorMessage(error)}`);
+    }
+}
+export async function recordToolCallInteractions(config, toolCalls) {
+    // Only send interaction events for responses that contain function calls.
+    if (toolCalls.length === 0) {
+        return;
+    }
+    try {
+        const server = getCodeAssistServer(config);
+        if (!server) {
+            return;
+        }
+        const interaction = summarizeToolCalls(toolCalls);
+        if (interaction) {
+            await server.recordConversationInteraction(interaction);
+        }
+    }
+    catch (error) {
+        debugLogger.warn(`Error recording tool call interactions: ${getErrorMessage(error)}`);
+    }
+}
+export function createConversationOffered(response, traceId, signal, streamingLatency) {
+    // Only send conversation offered events for responses that contain function
+    // calls. Non-function call events don't represent user actionable
+    // 'suggestions'.
+    if ((response.functionCalls?.length || 0) === 0) {
+        return;
+    }
+    const actionStatus = getStatusFromResponse(response, signal);
+    return {
+        citationCount: String(getCitations(response).length),
+        includedCode: includesCode(response),
+        status: actionStatus,
+        traceId,
+        streamingLatency,
+        isAgentic: true,
+    };
+}
+function summarizeToolCalls(toolCalls) {
+    let acceptedToolCalls = 0;
+    let actionStatus = undefined;
+    let traceId = undefined;
+    // Treat file edits as ACCEPT_FILE and everything else as unknown.
+    let isEdit = false;
+    // Iterate the tool calls and summarize them into a single conversation
+    // interaction so that the ConversationOffered and ConversationInteraction
+    // events are 1:1 in telemetry.
+    for (const toolCall of toolCalls) {
+        traceId ||= toolCall.request.traceId;
+        // If any tool call is canceled, we treat the entire interaction as canceled.
+        if (toolCall.status === 'cancelled') {
+            actionStatus = ActionStatus.ACTION_STATUS_CANCELLED;
+            break;
+        }
+        // If any tool call encounters an error, we treat the entire interaction as
+        // having errored.
+        if (toolCall.status === 'error') {
+            actionStatus = ActionStatus.ACTION_STATUS_ERROR_UNKNOWN;
+            break;
+        }
+        // Record if the tool call was accepted.
+        if (toolCall.outcome !== ToolConfirmationOutcome.Cancel) {
+            acceptedToolCalls++;
+            // Edits are ACCEPT_FILE, everything else is UNKNOWN.
+            if (EDIT_TOOL_NAMES.has(toolCall.request.name)) {
+                isEdit ||= true;
+            }
+        }
+    }
+    // Only file interaction telemetry if 100% of the tool calls were accepted.
+    return traceId && acceptedToolCalls / toolCalls.length >= 1
+        ? createConversationInteraction(traceId, actionStatus || ActionStatus.ACTION_STATUS_NO_ERROR, isEdit
+            ? ConversationInteractionInteraction.ACCEPT_FILE
+            : ConversationInteractionInteraction.UNKNOWN)
+        : undefined;
+}
+function createConversationInteraction(traceId, status, interaction) {
+    return {
+        traceId,
+        status,
+        interaction,
+        isAgentic: true,
+    };
+}
+function includesCode(resp) {
+    if (!resp.candidates) {
+        return false;
+    }
+    for (const candidate of resp.candidates) {
+        if (!candidate.content || !candidate.content.parts) {
+            continue;
+        }
+        for (const part of candidate.content.parts) {
+            if ('text' in part && part?.text?.includes('```')) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+function getStatusFromResponse(response, signal) {
+    if (signal?.aborted) {
+        return ActionStatus.ACTION_STATUS_CANCELLED;
+    }
+    if (hasError(response)) {
+        return ActionStatus.ACTION_STATUS_ERROR_UNKNOWN;
+    }
+    if ((response.candidates?.length ?? 0) <= 0) {
+        return ActionStatus.ACTION_STATUS_EMPTY;
+    }
+    return ActionStatus.ACTION_STATUS_NO_ERROR;
+}
+export function formatProtoJsonDuration(milliseconds) {
+    return `${milliseconds / 1000}s`;
+}
+function hasError(response) {
+    // Non-OK SDK results should be considered an error.
+    if (response.sdkHttpResponse &&
+        !response.sdkHttpResponse?.responseInternal?.ok) {
+        return true;
+    }
+    for (const candidate of response.candidates || []) {
+        // Treat sanitization, SPII, recitation, and forbidden terms as an error.
+        if (candidate.finishReason &&
+            candidate.finishReason !== FinishReason.STOP &&
+            candidate.finishReason !== FinishReason.MAX_TOKENS) {
+            return true;
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=telemetry.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../../src/code_assist/telemetry.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAgC,MAAM,eAAe,CAAC;AAC3E,OAAO,EAAE,YAAY,EAAE,MAAM,8CAA8C,CAAC;AAC5E,OAAO,EACL,YAAY,EACZ,kCAAkC,GAInC,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAE5D,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,MAAwB,EACxB,OAA2B,EAC3B,QAAiC,EACjC,gBAAkC,EAClC,WAAoC;IAEpC,IAAI,CAAC;QACH,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,OAAO,GAAG,yBAAyB,CACvC,QAAQ,EACR,OAAO,EACP,WAAW,EACX,gBAAgB,CACjB,CAAC;YACF,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,WAAW,CAAC,IAAI,CACd,2CAA2C,eAAe,CAAC,KAAK,CAAC,EAAE,CACpE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAAc,EACd,SAA8B;IAE9B,0EAA0E;IAC1E,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,MAAM,WAAW,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAClD,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,MAAM,CAAC,6BAA6B,CAAC,WAAW,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,WAAW,CAAC,IAAI,CACd,2CAA2C,eAAe,CAAC,KAAK,CAAC,EAAE,CACpE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,QAAiC,EACjC,OAAe,EACf,MAA+B,EAC/B,gBAAkC;IAElC,4EAA4E;IAC5E,kEAAkE;IAClE,iBAAiB;IACjB,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,MAAM,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;QAChD,OAAO;IACT,CAAC;IAED,MAAM,YAAY,GAAG,qBAAqB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAE7D,OAAO;QACL,aAAa,EAAE,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;QACpD,YAAY,EAAE,YAAY,CAAC,QAAQ,CAAC;QACpC,MAAM,EAAE,YAAY;QACpB,OAAO;QACP,gBAAgB;QAChB,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,SAA8B;IAE9B,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,IAAI,YAAY,GAAG,SAAS,CAAC;IAC7B,IAAI,OAAO,GAAG,SAAS,CAAC;IAExB,kEAAkE;IAClE,IAAI,MAAM,GAAG,KAAK,CAAC;IAEnB,uEAAuE;IACvE,0EAA0E;IAC1E,+BAA+B;IAC/B,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,OAAO,KAAK,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC;QAErC,6EAA6E;QAC7E,IAAI,QAAQ,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACpC,YAAY,GAAG,YAAY,CAAC,uBAAuB,CAAC;YACpD,MAAM;QACR,CAAC;QAED,2EAA2E;QAC3E,kBAAkB;QAClB,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;YAChC,YAAY,GAAG,YAAY,CAAC,2BAA2B,CAAC;YACxD,MAAM;QACR,CAAC;QAED,wCAAwC;QACxC,IAAI,QAAQ,CAAC,OAAO,KAAK,uBAAuB,CAAC,MAAM,EAAE,CAAC;YACxD,iBAAiB,EAAE,CAAC;YAEpB,qDAAqD;YACrD,IAAI,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/C,MAAM,KAAK,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,OAAO,OAAO,IAAI,iBAAiB,GAAG,SAAS,CAAC,MAAM,IAAI,CAAC;QACzD,CAAC,CAAC,6BAA6B,CAC3B,OAAO,EACP,YAAY,IAAI,YAAY,CAAC,sBAAsB,EACnD,MAAM;YACJ,CAAC,CAAC,kCAAkC,CAAC,WAAW;YAChD,CAAC,CAAC,kCAAkC,CAAC,OAAO,CAC/C;QACH,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,SAAS,6BAA6B,CACpC,OAAe,EACf,MAAoB,EACpB,WAA+C;IAE/C,OAAO;QACL,OAAO;QACP,MAAM;QACN,WAAW;QACX,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,IAA6B;IACjD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACxC,IAAI,CAAC,SAAS,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnD,SAAS;QACX,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YAC3C,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAClD,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,qBAAqB,CAC5B,QAAiC,EACjC,MAA+B;IAE/B,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;QACpB,OAAO,YAAY,CAAC,uBAAuB,CAAC;IAC9C,CAAC;IAED,IAAI,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvB,OAAO,YAAY,CAAC,2BAA2B,CAAC;IAClD,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,OAAO,YAAY,CAAC,mBAAmB,CAAC;IAC1C,CAAC;IAED,OAAO,YAAY,CAAC,sBAAsB,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,YAAoB;IAC1D,OAAO,GAAG,YAAY,GAAG,IAAI,GAAG,CAAC;AACnC,CAAC;AAED,SAAS,QAAQ,CAAC,QAAiC;IACjD,oDAAoD;IACpD,IACE,QAAQ,CAAC,eAAe;QACxB,CAAC,QAAQ,CAAC,eAAe,EAAE,gBAAgB,EAAE,EAAE,EAC/C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;QAClD,yEAAyE;QACzE,IACE,SAAS,CAAC,YAAY;YACtB,SAAS,CAAC,YAAY,KAAK,YAAY,CAAC,IAAI;YAC5C,SAAS,CAAC,YAAY,KAAK,YAAY,CAAC,UAAU,EAClD,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

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

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

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

@@ -0,0 +1,301 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * 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: 'edit_file', // 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