@google/gemini-cli-core 0.1.0-development.1

package/dist/src/telemetry/metrics.js

@@ -0,0 +1,144 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { metrics, ValueType, } from '@opentelemetry/api';
+import { SERVICE_NAME, METRIC_TOOL_CALL_COUNT, METRIC_TOOL_CALL_LATENCY, METRIC_API_REQUEST_COUNT, METRIC_API_REQUEST_LATENCY, METRIC_TOKEN_USAGE, METRIC_SESSION_COUNT, METRIC_FILE_OPERATION_COUNT, } from './constants.js';
+export var FileOperation;
+(function (FileOperation) {
+    FileOperation["CREATE"] = "create";
+    FileOperation["READ"] = "read";
+    FileOperation["UPDATE"] = "update";
+})(FileOperation || (FileOperation = {}));
+let cliMeter;
+let toolCallCounter;
+let toolCallLatencyHistogram;
+let apiRequestCounter;
+let apiRequestLatencyHistogram;
+let tokenUsageCounter;
+let fileOperationCounter;
+let isMetricsInitialized = false;
+function getCommonAttributes(config) {
+    return {
+        'session.id': config.getSessionId(),
+    };
+}
+export function getMeter() {
+    if (!cliMeter) {
+        cliMeter = metrics.getMeter(SERVICE_NAME);
+    }
+    return cliMeter;
+}
+export function initializeMetrics(config) {
+    if (isMetricsInitialized)
+        return;
+    const meter = getMeter();
+    if (!meter)
+        return;
+    toolCallCounter = meter.createCounter(METRIC_TOOL_CALL_COUNT, {
+        description: 'Counts tool calls, tagged by function name and success.',
+        valueType: ValueType.INT,
+    });
+    toolCallLatencyHistogram = meter.createHistogram(METRIC_TOOL_CALL_LATENCY, {
+        description: 'Latency of tool calls in milliseconds.',
+        unit: 'ms',
+        valueType: ValueType.INT,
+    });
+    apiRequestCounter = meter.createCounter(METRIC_API_REQUEST_COUNT, {
+        description: 'Counts API requests, tagged by model and status.',
+        valueType: ValueType.INT,
+    });
+    apiRequestLatencyHistogram = meter.createHistogram(METRIC_API_REQUEST_LATENCY, {
+        description: 'Latency of API requests in milliseconds.',
+        unit: 'ms',
+        valueType: ValueType.INT,
+    });
+    tokenUsageCounter = meter.createCounter(METRIC_TOKEN_USAGE, {
+        description: 'Counts the total number of tokens used.',
+        valueType: ValueType.INT,
+    });
+    fileOperationCounter = meter.createCounter(METRIC_FILE_OPERATION_COUNT, {
+        description: 'Counts file operations (create, read, update).',
+        valueType: ValueType.INT,
+    });
+    const sessionCounter = meter.createCounter(METRIC_SESSION_COUNT, {
+        description: 'Count of CLI sessions started.',
+        valueType: ValueType.INT,
+    });
+    sessionCounter.add(1, getCommonAttributes(config));
+    isMetricsInitialized = true;
+}
+export function recordToolCallMetrics(config, functionName, durationMs, success, decision) {
+    if (!toolCallCounter || !toolCallLatencyHistogram || !isMetricsInitialized)
+        return;
+    const metricAttributes = {
+        ...getCommonAttributes(config),
+        function_name: functionName,
+        success,
+        decision,
+    };
+    toolCallCounter.add(1, metricAttributes);
+    toolCallLatencyHistogram.record(durationMs, {
+        ...getCommonAttributes(config),
+        function_name: functionName,
+    });
+}
+export function recordTokenUsageMetrics(config, model, tokenCount, type) {
+    if (!tokenUsageCounter || !isMetricsInitialized)
+        return;
+    tokenUsageCounter.add(tokenCount, {
+        ...getCommonAttributes(config),
+        model,
+        type,
+    });
+}
+export function recordApiResponseMetrics(config, model, durationMs, statusCode, error) {
+    if (!apiRequestCounter ||
+        !apiRequestLatencyHistogram ||
+        !isMetricsInitialized)
+        return;
+    const metricAttributes = {
+        ...getCommonAttributes(config),
+        model,
+        status_code: statusCode ?? (error ? 'error' : 'ok'),
+    };
+    apiRequestCounter.add(1, metricAttributes);
+    apiRequestLatencyHistogram.record(durationMs, {
+        ...getCommonAttributes(config),
+        model,
+    });
+}
+export function recordApiErrorMetrics(config, model, durationMs, statusCode, errorType) {
+    if (!apiRequestCounter ||
+        !apiRequestLatencyHistogram ||
+        !isMetricsInitialized)
+        return;
+    const metricAttributes = {
+        ...getCommonAttributes(config),
+        model,
+        status_code: statusCode ?? 'error',
+        error_type: errorType ?? 'unknown',
+    };
+    apiRequestCounter.add(1, metricAttributes);
+    apiRequestLatencyHistogram.record(durationMs, {
+        ...getCommonAttributes(config),
+        model,
+    });
+}
+export function recordFileOperationMetric(config, operation, lines, mimetype, extension) {
+    if (!fileOperationCounter || !isMetricsInitialized)
+        return;
+    const attributes = {
+        ...getCommonAttributes(config),
+        operation,
+    };
+    if (lines !== undefined)
+        attributes.lines = lines;
+    if (mimetype !== undefined)
+        attributes.mimetype = mimetype;
+    if (extension !== undefined)
+        attributes.extension = extension;
+    fileOperationCounter.add(1, attributes);
+}
+//# sourceMappingURL=metrics.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"metrics.js","sourceRoot":"","sources":["../../../src/telemetry/metrics.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,OAAO,EAEP,SAAS,GAIV,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,YAAY,EACZ,sBAAsB,EACtB,wBAAwB,EACxB,wBAAwB,EACxB,0BAA0B,EAC1B,kBAAkB,EAClB,oBAAoB,EACpB,2BAA2B,GAC5B,MAAM,gBAAgB,CAAC;AAGxB,MAAM,CAAN,IAAY,aAIX;AAJD,WAAY,aAAa;IACvB,kCAAiB,CAAA;IACjB,8BAAa,CAAA;IACb,kCAAiB,CAAA;AACnB,CAAC,EAJW,aAAa,KAAb,aAAa,QAIxB;AAED,IAAI,QAA2B,CAAC;AAChC,IAAI,eAAoC,CAAC;AACzC,IAAI,wBAA+C,CAAC;AACpD,IAAI,iBAAsC,CAAC;AAC3C,IAAI,0BAAiD,CAAC;AACtD,IAAI,iBAAsC,CAAC;AAC3C,IAAI,oBAAyC,CAAC;AAC9C,IAAI,oBAAoB,GAAG,KAAK,CAAC;AAEjC,SAAS,mBAAmB,CAAC,MAAc;IACzC,OAAO;QACL,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE;KACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,QAAQ;IACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,MAAc;IAC9C,IAAI,oBAAoB;QAAE,OAAO;IAEjC,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;IACzB,IAAI,CAAC,KAAK;QAAE,OAAO;IAEnB,eAAe,GAAG,KAAK,CAAC,aAAa,CAAC,sBAAsB,EAAE;QAC5D,WAAW,EAAE,yDAAyD;QACtE,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,wBAAwB,GAAG,KAAK,CAAC,eAAe,CAAC,wBAAwB,EAAE;QACzE,WAAW,EAAE,wCAAwC;QACrD,IAAI,EAAE,IAAI;QACV,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,iBAAiB,GAAG,KAAK,CAAC,aAAa,CAAC,wBAAwB,EAAE;QAChE,WAAW,EAAE,kDAAkD;QAC/D,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,0BAA0B,GAAG,KAAK,CAAC,eAAe,CAChD,0BAA0B,EAC1B;QACE,WAAW,EAAE,0CAA0C;QACvD,IAAI,EAAE,IAAI;QACV,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CACF,CAAC;IACF,iBAAiB,GAAG,KAAK,CAAC,aAAa,CAAC,kBAAkB,EAAE;QAC1D,WAAW,EAAE,yCAAyC;QACtD,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,oBAAoB,GAAG,KAAK,CAAC,aAAa,CAAC,2BAA2B,EAAE;QACtE,WAAW,EAAE,gDAAgD;QAC7D,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,MAAM,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC,oBAAoB,EAAE;QAC/D,WAAW,EAAE,gCAAgC;QAC7C,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC;IACnD,oBAAoB,GAAG,IAAI,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,MAAc,EACd,YAAoB,EACpB,UAAkB,EAClB,OAAgB,EAChB,QAAyC;IAEzC,IAAI,CAAC,eAAe,IAAI,CAAC,wBAAwB,IAAI,CAAC,oBAAoB;QACxE,OAAO;IAET,MAAM,gBAAgB,GAAe;QACnC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,aAAa,EAAE,YAAY;QAC3B,OAAO;QACP,QAAQ;KACT,CAAC;IACF,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IACzC,wBAAwB,CAAC,MAAM,CAAC,UAAU,EAAE;QAC1C,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,aAAa,EAAE,YAAY;KAC5B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,MAAc,EACd,KAAa,EACb,UAAkB,EAClB,IAAuD;IAEvD,IAAI,CAAC,iBAAiB,IAAI,CAAC,oBAAoB;QAAE,OAAO;IACxD,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE;QAChC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;QACL,IAAI;KACL,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,MAAc,EACd,KAAa,EACb,UAAkB,EAClB,UAA4B,EAC5B,KAAc;IAEd,IACE,CAAC,iBAAiB;QAClB,CAAC,0BAA0B;QAC3B,CAAC,oBAAoB;QAErB,OAAO;IACT,MAAM,gBAAgB,GAAe;QACnC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;QACL,WAAW,EAAE,UAAU,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;KACpD,CAAC;IACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC3C,0BAA0B,CAAC,MAAM,CAAC,UAAU,EAAE;QAC5C,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;KACN,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,MAAc,EACd,KAAa,EACb,UAAkB,EAClB,UAA4B,EAC5B,SAAkB;IAElB,IACE,CAAC,iBAAiB;QAClB,CAAC,0BAA0B;QAC3B,CAAC,oBAAoB;QAErB,OAAO;IACT,MAAM,gBAAgB,GAAe;QACnC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;QACL,WAAW,EAAE,UAAU,IAAI,OAAO;QAClC,UAAU,EAAE,SAAS,IAAI,SAAS;KACnC,CAAC;IACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC3C,0BAA0B,CAAC,MAAM,CAAC,UAAU,EAAE;QAC5C,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;KACN,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,MAAc,EACd,SAAwB,EACxB,KAAc,EACd,QAAiB,EACjB,SAAkB;IAElB,IAAI,CAAC,oBAAoB,IAAI,CAAC,oBAAoB;QAAE,OAAO;IAC3D,MAAM,UAAU,GAAe;QAC7B,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,SAAS;KACV,CAAC;IACF,IAAI,KAAK,KAAK,SAAS;QAAE,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;IAClD,IAAI,QAAQ,KAAK,SAAS;QAAE,UAAU,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3D,IAAI,SAAS,KAAK,SAAS;QAAE,UAAU,CAAC,SAAS,GAAG,SAAS,CAAC;IAC9D,oBAAoB,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;AAC1C,CAAC"}

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

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

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

@@ -0,0 +1,162 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { FileOperation } from './metrics.js';
+const mockCounterAddFn = vi.fn();
+const mockHistogramRecordFn = vi.fn();
+const mockCreateCounterFn = vi.fn();
+const mockCreateHistogramFn = vi.fn();
+const mockCounterInstance = {
+    add: mockCounterAddFn,
+};
+const mockHistogramInstance = {
+    record: mockHistogramRecordFn,
+};
+const mockMeterInstance = {
+    createCounter: mockCreateCounterFn.mockReturnValue(mockCounterInstance),
+    createHistogram: mockCreateHistogramFn.mockReturnValue(mockHistogramInstance),
+};
+function originalOtelMockFactory() {
+    return {
+        metrics: {
+            getMeter: vi.fn(),
+        },
+        ValueType: {
+            INT: 1,
+        },
+    };
+}
+vi.mock('@opentelemetry/api', originalOtelMockFactory);
+describe('Telemetry Metrics', () => {
+    let initializeMetricsModule;
+    let recordTokenUsageMetricsModule;
+    let recordFileOperationMetricModule;
+    beforeEach(async () => {
+        vi.resetModules();
+        vi.doMock('@opentelemetry/api', () => {
+            const actualApi = originalOtelMockFactory();
+            actualApi.metrics.getMeter.mockReturnValue(mockMeterInstance);
+            return actualApi;
+        });
+        const metricsJsModule = await import('./metrics.js');
+        initializeMetricsModule = metricsJsModule.initializeMetrics;
+        recordTokenUsageMetricsModule = metricsJsModule.recordTokenUsageMetrics;
+        recordFileOperationMetricModule = metricsJsModule.recordFileOperationMetric;
+        const otelApiModule = await import('@opentelemetry/api');
+        mockCounterAddFn.mockClear();
+        mockCreateCounterFn.mockClear();
+        mockCreateHistogramFn.mockClear();
+        mockHistogramRecordFn.mockClear();
+        otelApiModule.metrics.getMeter.mockClear();
+        otelApiModule.metrics.getMeter.mockReturnValue(mockMeterInstance);
+        mockCreateCounterFn.mockReturnValue(mockCounterInstance);
+        mockCreateHistogramFn.mockReturnValue(mockHistogramInstance);
+    });
+    describe('recordTokenUsageMetrics', () => {
+        const mockConfig = {
+            getSessionId: () => 'test-session-id',
+        };
+        it('should not record metrics if not initialized', () => {
+            recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 100, 'input');
+            expect(mockCounterAddFn).not.toHaveBeenCalled();
+        });
+        it('should record token usage with the correct attributes', () => {
+            initializeMetricsModule(mockConfig);
+            recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 100, 'input');
+            expect(mockCounterAddFn).toHaveBeenCalledTimes(2);
+            expect(mockCounterAddFn).toHaveBeenNthCalledWith(1, 1, {
+                'session.id': 'test-session-id',
+            });
+            expect(mockCounterAddFn).toHaveBeenNthCalledWith(2, 100, {
+                'session.id': 'test-session-id',
+                model: 'gemini-pro',
+                type: 'input',
+            });
+        });
+        it('should record token usage for different types', () => {
+            initializeMetricsModule(mockConfig);
+            mockCounterAddFn.mockClear();
+            recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 50, 'output');
+            expect(mockCounterAddFn).toHaveBeenCalledWith(50, {
+                'session.id': 'test-session-id',
+                model: 'gemini-pro',
+                type: 'output',
+            });
+            recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 25, 'thought');
+            expect(mockCounterAddFn).toHaveBeenCalledWith(25, {
+                'session.id': 'test-session-id',
+                model: 'gemini-pro',
+                type: 'thought',
+            });
+            recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 75, 'cache');
+            expect(mockCounterAddFn).toHaveBeenCalledWith(75, {
+                'session.id': 'test-session-id',
+                model: 'gemini-pro',
+                type: 'cache',
+            });
+            recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 125, 'tool');
+            expect(mockCounterAddFn).toHaveBeenCalledWith(125, {
+                'session.id': 'test-session-id',
+                model: 'gemini-pro',
+                type: 'tool',
+            });
+        });
+        it('should handle different models', () => {
+            initializeMetricsModule(mockConfig);
+            mockCounterAddFn.mockClear();
+            recordTokenUsageMetricsModule(mockConfig, 'gemini-ultra', 200, 'input');
+            expect(mockCounterAddFn).toHaveBeenCalledWith(200, {
+                'session.id': 'test-session-id',
+                model: 'gemini-ultra',
+                type: 'input',
+            });
+        });
+    });
+    describe('recordFileOperationMetric', () => {
+        const mockConfig = {
+            getSessionId: () => 'test-session-id',
+        };
+        it('should not record metrics if not initialized', () => {
+            recordFileOperationMetricModule(mockConfig, FileOperation.CREATE, 10, 'text/plain', 'txt');
+            expect(mockCounterAddFn).not.toHaveBeenCalled();
+        });
+        it('should record file creation with all attributes', () => {
+            initializeMetricsModule(mockConfig);
+            recordFileOperationMetricModule(mockConfig, FileOperation.CREATE, 10, 'text/plain', 'txt');
+            expect(mockCounterAddFn).toHaveBeenCalledTimes(2);
+            expect(mockCounterAddFn).toHaveBeenNthCalledWith(1, 1, {
+                'session.id': 'test-session-id',
+            });
+            expect(mockCounterAddFn).toHaveBeenNthCalledWith(2, 1, {
+                'session.id': 'test-session-id',
+                operation: FileOperation.CREATE,
+                lines: 10,
+                mimetype: 'text/plain',
+                extension: 'txt',
+            });
+        });
+        it('should record file read with minimal attributes', () => {
+            initializeMetricsModule(mockConfig);
+            mockCounterAddFn.mockClear();
+            recordFileOperationMetricModule(mockConfig, FileOperation.READ);
+            expect(mockCounterAddFn).toHaveBeenCalledWith(1, {
+                'session.id': 'test-session-id',
+                operation: FileOperation.READ,
+            });
+        });
+        it('should record file update with some attributes', () => {
+            initializeMetricsModule(mockConfig);
+            mockCounterAddFn.mockClear();
+            recordFileOperationMetricModule(mockConfig, FileOperation.UPDATE, undefined, 'application/javascript');
+            expect(mockCounterAddFn).toHaveBeenCalledWith(1, {
+                'session.id': 'test-session-id',
+                operation: FileOperation.UPDATE,
+                mimetype: 'application/javascript',
+            });
+        });
+    });
+});
+//# sourceMappingURL=metrics.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"metrics.test.js","sourceRoot":"","sources":["../../../src/telemetry/metrics.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAa,MAAM,QAAQ,CAAC;AASzE,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,MAAM,gBAAgB,GAElB,EAAE,CAAC,EAAE,EAAE,CAAC;AACZ,MAAM,qBAAqB,GAEvB,EAAE,CAAC,EAAE,EAAE,CAAC;AAEZ,MAAM,mBAAmB,GACvB,EAAE,CAAC,EAAE,EAAE,CAAC;AACV,MAAM,qBAAqB,GAEvB,EAAE,CAAC,EAAE,EAAE,CAAC;AAEZ,MAAM,mBAAmB,GAAG;IAC1B,GAAG,EAAE,gBAAgB;CACA,CAAC;AAExB,MAAM,qBAAqB,GAAG;IAC5B,MAAM,EAAE,qBAAqB;CACN,CAAC;AAE1B,MAAM,iBAAiB,GAAG;IACxB,aAAa,EAAE,mBAAmB,CAAC,eAAe,CAAC,mBAAmB,CAAC;IACvE,eAAe,EAAE,qBAAqB,CAAC,eAAe,CAAC,qBAAqB,CAAC;CAC1D,CAAC;AAEtB,SAAS,uBAAuB;IAC9B,OAAO;QACL,OAAO,EAAE;YACP,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB;QACD,SAAS,EAAE;YACT,GAAG,EAAE,CAAC;SACP;KACF,CAAC;AACJ,CAAC;AAED,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,CAAC;AAEvD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,uBAAwE,CAAC;IAC7E,IAAI,6BAAoF,CAAC;IACzF,IAAI,+BAAwF,CAAC;IAE7F,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,EAAE,CAAC,MAAM,CAAC,oBAAoB,EAAE,GAAG,EAAE;YACnC,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;YAC3C,SAAS,CAAC,OAAO,CAAC,QAAiB,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;YACxE,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;QACrD,uBAAuB,GAAG,eAAe,CAAC,iBAAiB,CAAC;QAC5D,6BAA6B,GAAG,eAAe,CAAC,uBAAuB,CAAC;QACxE,+BAA+B,GAAG,eAAe,CAAC,yBAAyB,CAAC;QAE5E,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAEzD,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC7B,mBAAmB,CAAC,SAAS,EAAE,CAAC;QAChC,qBAAqB,CAAC,SAAS,EAAE,CAAC;QAClC,qBAAqB,CAAC,SAAS,EAAE,CAAC;QACjC,aAAa,CAAC,OAAO,CAAC,QAAiB,CAAC,SAAS,EAAE,CAAC;QAEpD,aAAa,CAAC,OAAO,CAAC,QAAiB,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;QAC5E,mBAAmB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACzD,qBAAqB,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,MAAM,UAAU,GAAG;YACjB,YAAY,EAAE,GAAG,EAAE,CAAC,iBAAiB;SACjB,CAAC;QAEvB,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE;gBACrD,YAAY,EAAE,iBAAiB;aAChC,CAAC,CAAC;YACH,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,GAAG,EAAE;gBACvD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,OAAO;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC;YACtE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE;gBAChD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;YACvE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE;gBAChD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YAEH,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;YACrE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE;gBAChD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,OAAO;aACd,CAAC,CAAC;YAEH,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;YACrE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE;gBACjD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,MAAM;aACb,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,6BAA6B,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YACxE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE;gBACjD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,cAAc;gBACrB,IAAI,EAAE,OAAO;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,MAAM,UAAU,GAAG;YACjB,YAAY,EAAE,GAAG,EAAE,CAAC,iBAAiB;SACjB,CAAC;QAEvB,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,+BAA+B,CAC7B,UAAU,EACV,aAAa,CAAC,MAAM,EACpB,EAAE,EACF,YAAY,EACZ,KAAK,CACN,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,+BAA+B,CAC7B,UAAU,EACV,aAAa,CAAC,MAAM,EACpB,EAAE,EACF,YAAY,EACZ,KAAK,CACN,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE;gBACrD,YAAY,EAAE,iBAAiB;aAChC,CAAC,CAAC;YACH,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE;gBACrD,YAAY,EAAE,iBAAiB;gBAC/B,SAAS,EAAE,aAAa,CAAC,MAAM;gBAC/B,KAAK,EAAE,EAAE;gBACT,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,+BAA+B,CAAC,UAAU,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;YAChE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE;gBAC/C,YAAY,EAAE,iBAAiB;gBAC/B,SAAS,EAAE,aAAa,CAAC,IAAI;aAC9B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,+BAA+B,CAC7B,UAAU,EACV,aAAa,CAAC,MAAM,EACpB,SAAS,EACT,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE;gBAC/C,YAAY,EAAE,iBAAiB;gBAC/B,SAAS,EAAE,aAAa,CAAC,MAAM;gBAC/B,QAAQ,EAAE,wBAAwB;aACnC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/telemetry/sdk.d.ts

@@ -0,0 +1,9 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Config } from '../config/config.js';
+export declare function isTelemetrySdkInitialized(): boolean;
+export declare function initializeTelemetry(config: Config): void;
+export declare function shutdownTelemetry(): Promise<void>;

package/dist/src/telemetry/sdk.js

@@ -0,0 +1,116 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { DiagConsoleLogger, DiagLogLevel, diag } from '@opentelemetry/api';
+import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-grpc';
+import { OTLPLogExporter } from '@opentelemetry/exporter-logs-otlp-grpc';
+import { OTLPMetricExporter } from '@opentelemetry/exporter-metrics-otlp-grpc';
+import { CompressionAlgorithm } from '@opentelemetry/otlp-exporter-base';
+import { NodeSDK } from '@opentelemetry/sdk-node';
+import { SemanticResourceAttributes } from '@opentelemetry/semantic-conventions';
+import { Resource } from '@opentelemetry/resources';
+import { BatchSpanProcessor, ConsoleSpanExporter, } from '@opentelemetry/sdk-trace-node';
+import { BatchLogRecordProcessor, ConsoleLogRecordExporter, } from '@opentelemetry/sdk-logs';
+import { ConsoleMetricExporter, PeriodicExportingMetricReader, } from '@opentelemetry/sdk-metrics';
+import { HttpInstrumentation } from '@opentelemetry/instrumentation-http';
+import { SERVICE_NAME } from './constants.js';
+import { initializeMetrics } from './metrics.js';
+import { ClearcutLogger } from './clearcut-logger/clearcut-logger.js';
+// For troubleshooting, set the log level to DiagLogLevel.DEBUG
+diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.INFO);
+let sdk;
+let telemetryInitialized = false;
+export function isTelemetrySdkInitialized() {
+    return telemetryInitialized;
+}
+function parseGrpcEndpoint(otlpEndpointSetting) {
+    if (!otlpEndpointSetting) {
+        return undefined;
+    }
+    // Trim leading/trailing quotes that might come from env variables
+    const trimmedEndpoint = otlpEndpointSetting.replace(/^["']|["']$/g, '');
+    try {
+        const url = new URL(trimmedEndpoint);
+        // OTLP gRPC exporters expect an endpoint in the format scheme://host:port
+        // The `origin` property provides this, stripping any path, query, or hash.
+        return url.origin;
+    }
+    catch (error) {
+        diag.error('Invalid OTLP endpoint URL provided:', trimmedEndpoint, error);
+        return undefined;
+    }
+}
+export function initializeTelemetry(config) {
+    if (telemetryInitialized || !config.getTelemetryEnabled()) {
+        return;
+    }
+    const resource = new Resource({
+        [SemanticResourceAttributes.SERVICE_NAME]: SERVICE_NAME,
+        [SemanticResourceAttributes.SERVICE_VERSION]: process.version,
+        'session.id': config.getSessionId(),
+    });
+    const otlpEndpoint = config.getTelemetryOtlpEndpoint();
+    const grpcParsedEndpoint = parseGrpcEndpoint(otlpEndpoint);
+    const useOtlp = !!grpcParsedEndpoint;
+    const spanExporter = useOtlp
+        ? new OTLPTraceExporter({
+            url: grpcParsedEndpoint,
+            compression: CompressionAlgorithm.GZIP,
+        })
+        : new ConsoleSpanExporter();
+    const logExporter = useOtlp
+        ? new OTLPLogExporter({
+            url: grpcParsedEndpoint,
+            compression: CompressionAlgorithm.GZIP,
+        })
+        : new ConsoleLogRecordExporter();
+    const metricReader = useOtlp
+        ? new PeriodicExportingMetricReader({
+            exporter: new OTLPMetricExporter({
+                url: grpcParsedEndpoint,
+                compression: CompressionAlgorithm.GZIP,
+            }),
+            exportIntervalMillis: 10000,
+        })
+        : new PeriodicExportingMetricReader({
+            exporter: new ConsoleMetricExporter(),
+            exportIntervalMillis: 10000,
+        });
+    sdk = new NodeSDK({
+        resource,
+        spanProcessors: [new BatchSpanProcessor(spanExporter)],
+        logRecordProcessor: new BatchLogRecordProcessor(logExporter),
+        metricReader,
+        instrumentations: [new HttpInstrumentation()],
+    });
+    try {
+        sdk.start();
+        console.log('OpenTelemetry SDK started successfully.');
+        telemetryInitialized = true;
+        initializeMetrics(config);
+    }
+    catch (error) {
+        console.error('Error starting OpenTelemetry SDK:', error);
+    }
+    process.on('SIGTERM', shutdownTelemetry);
+    process.on('SIGINT', shutdownTelemetry);
+}
+export async function shutdownTelemetry() {
+    if (!telemetryInitialized || !sdk) {
+        return;
+    }
+    try {
+        ClearcutLogger.getInstance()?.shutdown();
+        await sdk.shutdown();
+        console.log('OpenTelemetry SDK shut down successfully.');
+    }
+    catch (error) {
+        console.error('Error shutting down SDK:', error);
+    }
+    finally {
+        telemetryInitialized = false;
+    }
+}
+//# sourceMappingURL=sdk.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"sdk.js","sourceRoot":"","sources":["../../../src/telemetry/sdk.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,eAAe,EAAE,MAAM,wCAAwC,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2CAA2C,CAAC;AAC/E,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,0BAA0B,EAAE,MAAM,qCAAqC,CAAC;AACjF,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EACL,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,uBAAuB,EACvB,wBAAwB,GACzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,qBAAqB,EACrB,6BAA6B,GAC9B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE1E,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AAEtE,+DAA+D;AAC/D,IAAI,CAAC,SAAS,CAAC,IAAI,iBAAiB,EAAE,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;AAE3D,IAAI,GAAwB,CAAC;AAC7B,IAAI,oBAAoB,GAAG,KAAK,CAAC;AAEjC,MAAM,UAAU,yBAAyB;IACvC,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,SAAS,iBAAiB,CACxB,mBAAuC;IAEvC,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,kEAAkE;IAClE,MAAM,eAAe,GAAG,mBAAmB,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IAExE,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,eAAe,CAAC,CAAC;QACrC,0EAA0E;QAC1E,2EAA2E;QAC3E,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,KAAK,CAAC,qCAAqC,EAAE,eAAe,EAAE,KAAK,CAAC,CAAC;QAC1E,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAc;IAChD,IAAI,oBAAoB,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,EAAE,CAAC;QAC1D,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;QAC5B,CAAC,0BAA0B,CAAC,YAAY,CAAC,EAAE,YAAY;QACvD,CAAC,0BAA0B,CAAC,eAAe,CAAC,EAAE,OAAO,CAAC,OAAO;QAC7D,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE;KACpC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,MAAM,CAAC,wBAAwB,EAAE,CAAC;IACvD,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC;IAC3D,MAAM,OAAO,GAAG,CAAC,CAAC,kBAAkB,CAAC;IAErC,MAAM,YAAY,GAAG,OAAO;QAC1B,CAAC,CAAC,IAAI,iBAAiB,CAAC;YACpB,GAAG,EAAE,kBAAkB;YACvB,WAAW,EAAE,oBAAoB,CAAC,IAAI;SACvC,CAAC;QACJ,CAAC,CAAC,IAAI,mBAAmB,EAAE,CAAC;IAC9B,MAAM,WAAW,GAAG,OAAO;QACzB,CAAC,CAAC,IAAI,eAAe,CAAC;YAClB,GAAG,EAAE,kBAAkB;YACvB,WAAW,EAAE,oBAAoB,CAAC,IAAI;SACvC,CAAC;QACJ,CAAC,CAAC,IAAI,wBAAwB,EAAE,CAAC;IACnC,MAAM,YAAY,GAAG,OAAO;QAC1B,CAAC,CAAC,IAAI,6BAA6B,CAAC;YAChC,QAAQ,EAAE,IAAI,kBAAkB,CAAC;gBAC/B,GAAG,EAAE,kBAAkB;gBACvB,WAAW,EAAE,oBAAoB,CAAC,IAAI;aACvC,CAAC;YACF,oBAAoB,EAAE,KAAK;SAC5B,CAAC;QACJ,CAAC,CAAC,IAAI,6BAA6B,CAAC;YAChC,QAAQ,EAAE,IAAI,qBAAqB,EAAE;YACrC,oBAAoB,EAAE,KAAK;SAC5B,CAAC,CAAC;IAEP,GAAG,GAAG,IAAI,OAAO,CAAC;QAChB,QAAQ;QACR,cAAc,EAAE,CAAC,IAAI,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACtD,kBAAkB,EAAE,IAAI,uBAAuB,CAAC,WAAW,CAAC;QAC5D,YAAY;QACZ,gBAAgB,EAAE,CAAC,IAAI,mBAAmB,EAAE,CAAC;KAC9C,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,GAAG,CAAC,KAAK,EAAE,CAAC;QACZ,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;QACvD,oBAAoB,GAAG,IAAI,CAAC;QAC5B,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;IAC5D,CAAC;IAED,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;IACzC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB;IACrC,IAAI,CAAC,oBAAoB,IAAI,CAAC,GAAG,EAAE,CAAC;QAClC,OAAO;IACT,CAAC;IACD,IAAI,CAAC;QACH,cAAc,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,CAAC;QACzC,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;QACrB,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IAC3D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC;YAAS,CAAC;QACT,oBAAoB,GAAG,KAAK,CAAC;IAC/B,CAAC;AACH,CAAC"}

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

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

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

@@ -0,0 +1,50 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { initializeTelemetry, shutdownTelemetry, isTelemetrySdkInitialized, } from './sdk.js';
+import { Config } from '../config/config.js';
+import { NodeSDK } from '@opentelemetry/sdk-node';
+vi.mock('@opentelemetry/sdk-node');
+vi.mock('../config/config.js');
+describe('telemetry', () => {
+    let mockConfig;
+    let mockNodeSdk;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        mockConfig = new Config({
+            sessionId: 'test-session-id',
+            model: 'test-model',
+            targetDir: '/test/dir',
+            debugMode: false,
+            cwd: '/test/dir',
+        });
+        vi.spyOn(mockConfig, 'getTelemetryEnabled').mockReturnValue(true);
+        vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('http://localhost:4317');
+        vi.spyOn(mockConfig, 'getSessionId').mockReturnValue('test-session-id');
+        mockNodeSdk = {
+            start: vi.fn(),
+            shutdown: vi.fn().mockResolvedValue(undefined),
+        };
+        vi.mocked(NodeSDK).mockImplementation(() => mockNodeSdk);
+    });
+    afterEach(async () => {
+        // Ensure we shut down telemetry even if a test fails.
+        if (isTelemetrySdkInitialized()) {
+            await shutdownTelemetry();
+        }
+    });
+    it('should initialize the telemetry service', () => {
+        initializeTelemetry(mockConfig);
+        expect(NodeSDK).toHaveBeenCalled();
+        expect(mockNodeSdk.start).toHaveBeenCalled();
+    });
+    it('should shutdown the telemetry service', async () => {
+        initializeTelemetry(mockConfig);
+        await shutdownTelemetry();
+        expect(mockNodeSdk.shutdown).toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=telemetry.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.test.js","sourceRoot":"","sources":["../../../src/telemetry/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,mBAAmB,EACnB,iBAAiB,EACjB,yBAAyB,GAC1B,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAElD,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,UAAkB,CAAC;IACvB,IAAI,WAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,UAAU,GAAG,IAAI,MAAM,CAAC;YACtB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,WAAW;YACtB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,WAAW;SACjB,CAAC,CAAC;QACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAClE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAC9D,uBAAuB,CACxB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;QACxE,WAAW,GAAG;YACZ,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;SACzB,CAAC;QACxB,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,sDAAsD;QACtD,IAAI,yBAAyB,EAAE,EAAE,CAAC;YAChC,MAAM,iBAAiB,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,iBAAiB,EAAE,CAAC;QAE1B,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,91 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { GenerateContentResponseUsageMetadata } from '@google/genai';
+import { Config } from '../config/config.js';
+import { CompletedToolCall } from '../core/coreToolScheduler.js';
+import { ToolConfirmationOutcome } from '../tools/tools.js';
+export declare enum ToolCallDecision {
+    ACCEPT = "accept",
+    REJECT = "reject",
+    MODIFY = "modify"
+}
+export declare function getDecisionFromOutcome(outcome: ToolConfirmationOutcome): ToolCallDecision;
+export declare class StartSessionEvent {
+    'event.name': 'cli_config';
+    'event.timestamp': string;
+    model: string;
+    embedding_model: string;
+    sandbox_enabled: boolean;
+    core_tools_enabled: string;
+    approval_mode: string;
+    api_key_enabled: boolean;
+    vertex_ai_enabled: boolean;
+    debug_enabled: boolean;
+    mcp_servers: string;
+    telemetry_enabled: boolean;
+    telemetry_log_user_prompts_enabled: boolean;
+    file_filtering_respect_git_ignore: boolean;
+    constructor(config: Config);
+}
+export declare class EndSessionEvent {
+    'event.name': 'end_session';
+    'event.timestamp': string;
+    session_id?: string;
+    constructor(config?: Config);
+}
+export declare class UserPromptEvent {
+    'event.name': 'user_prompt';
+    'event.timestamp': string;
+    prompt_length: number;
+    prompt?: string;
+    constructor(prompt_length: number, prompt?: string);
+}
+export declare class ToolCallEvent {
+    'event.name': 'tool_call';
+    'event.timestamp': string;
+    function_name: string;
+    function_args: Record<string, unknown>;
+    duration_ms: number;
+    success: boolean;
+    decision?: ToolCallDecision;
+    error?: string;
+    error_type?: string;
+    constructor(call: CompletedToolCall);
+}
+export declare class ApiRequestEvent {
+    'event.name': 'api_request';
+    'event.timestamp': string;
+    model: string;
+    request_text?: string;
+    constructor(model: string, request_text?: string);
+}
+export declare class ApiErrorEvent {
+    'event.name': 'api_error';
+    'event.timestamp': string;
+    model: string;
+    error: string;
+    error_type?: string;
+    status_code?: number | string;
+    duration_ms: number;
+    constructor(model: string, error: string, duration_ms: number, error_type?: string, status_code?: number | string);
+}
+export declare class ApiResponseEvent {
+    'event.name': 'api_response';
+    'event.timestamp': string;
+    model: string;
+    status_code?: number | string;
+    duration_ms: number;
+    error?: string;
+    input_token_count: number;
+    output_token_count: number;
+    cached_content_token_count: number;
+    thoughts_token_count: number;
+    tool_token_count: number;
+    total_token_count: number;
+    response_text?: string;
+    constructor(model: string, duration_ms: number, usage_data?: GenerateContentResponseUsageMetadata, response_text?: string, error?: string);
+}
+export type TelemetryEvent = StartSessionEvent | EndSessionEvent | UserPromptEvent | ToolCallEvent | ApiRequestEvent | ApiErrorEvent | ApiResponseEvent;