@machina.ai/cell-cli-core 1.19.4-rc3 → 1.22.5-rc1

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

@@ -0,0 +1,289 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { StartupProfiler } from './startupProfiler.js';
+import { debugLogger } from '../utils/debugLogger.js';
+// Mock the metrics module
+vi.mock('./metrics.js', () => ({
+    recordStartupPerformance: vi.fn(),
+}));
+// Mock loggers module
+vi.mock('./loggers.js', () => ({
+    logStartupStats: vi.fn(),
+}));
+// Mock os module
+vi.mock('node:os', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        platform: vi.fn(() => 'darwin'),
+        arch: vi.fn(() => 'x64'),
+        release: vi.fn(() => '22.6.0'),
+    };
+});
+// Mock fs module
+vi.mock('node:fs', () => ({
+    existsSync: vi.fn(() => false),
+    createWriteStream: vi.fn(() => ({
+        write: vi.fn(),
+        on: vi.fn(),
+    })),
+}));
+describe('StartupProfiler', () => {
+    let profiler;
+    let mockConfig;
+    let recordStartupPerformance;
+    let logStartupStats;
+    beforeEach(async () => {
+        vi.resetAllMocks();
+        // Get the mocked function
+        const metricsModule = await import('./metrics.js');
+        recordStartupPerformance =
+            metricsModule.recordStartupPerformance;
+        const loggersModule = await import('./loggers.js');
+        logStartupStats = loggersModule.logStartupStats;
+        // Create a fresh profiler instance
+        profiler = StartupProfiler.getInstance();
+        // Clear any existing phases and performance entries
+        profiler['phases'].clear();
+        performance.clearMarks();
+        performance.clearMeasures();
+        mockConfig = {
+            getSessionId: () => 'test-session-id',
+            getTelemetryEnabled: () => true,
+        };
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('getInstance', () => {
+        it('should return a singleton instance', () => {
+            const instance1 = StartupProfiler.getInstance();
+            const instance2 = StartupProfiler.getInstance();
+            expect(instance1).toBe(instance2);
+        });
+    });
+    describe('start', () => {
+        it('should create a performance mark for a phase', () => {
+            profiler.start('test_phase');
+            const phase = profiler['phases'].get('test_phase');
+            expect(phase).toBeDefined();
+            expect(phase?.name).toBe('test_phase');
+            // Verify performance mark was created
+            const marks = performance.getEntriesByType('mark');
+            const startMark = marks.find((m) => m.name === 'startup:test_phase:start');
+            expect(startMark).toBeDefined();
+        });
+        it('should record start time with details', () => {
+            const details = { key: 'value', count: 42 };
+            profiler.start('test_phase', details);
+            const phase = profiler['phases'].get('test_phase');
+            expect(phase?.details).toEqual(details);
+        });
+        it('should return undefined when starting a phase that is already active', () => {
+            profiler.start('test_phase');
+            const handle = profiler.start('test_phase');
+            expect(handle).toBeUndefined();
+        });
+    });
+    describe('end', () => {
+        it('should create a performance measure for a started phase', () => {
+            const handle = profiler.start('test_phase');
+            handle?.end();
+            // Verify performance measure was created
+            const measures = performance.getEntriesByType('measure');
+            const measure = measures.find((m) => m.name === 'test_phase');
+            expect(measure).toBeDefined();
+            expect(measure?.duration).toBeGreaterThan(0);
+        });
+        it('should merge details when ending a phase', () => {
+            const handle = profiler.start('test_phase', { initial: 'value' });
+            handle?.end({ additional: 'data' });
+            const phase = profiler['phases'].get('test_phase');
+            expect(phase?.details).toEqual({
+                initial: 'value',
+                additional: 'data',
+            });
+        });
+        it('should overwrite details with same key', () => {
+            const handle = profiler.start('test_phase', { key: 'original' });
+            handle?.end({ key: 'updated' });
+            const phase = profiler['phases'].get('test_phase');
+            expect(phase?.details).toEqual({ key: 'updated' });
+        });
+    });
+    describe('flush', () => {
+        it('should call recordStartupPerformance for each completed phase', () => {
+            const handle1 = profiler.start('phase1');
+            handle1?.end();
+            const handle2 = profiler.start('phase2');
+            handle2?.end();
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).toHaveBeenCalledTimes(2);
+        });
+        it('should not record phases without duration', () => {
+            profiler.start('incomplete_phase');
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).not.toHaveBeenCalled();
+        });
+        it('should include common details in all metrics', () => {
+            const handle = profiler.start('test_phase');
+            handle?.end();
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).toHaveBeenCalledWith(mockConfig, expect.any(Number), expect.objectContaining({
+                phase: 'test_phase',
+                details: expect.objectContaining({
+                    os_platform: 'darwin',
+                    os_arch: 'x64',
+                    os_release: '22.6.0',
+                    is_docker: false,
+                    cpu_usage_user: expect.any(Number),
+                    cpu_usage_system: expect.any(Number),
+                }),
+            }));
+        });
+        it('should merge phase-specific details with common details', () => {
+            const handle = profiler.start('test_phase', { custom: 'value' });
+            handle?.end();
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).toHaveBeenCalledWith(mockConfig, expect.any(Number), expect.objectContaining({
+                phase: 'test_phase',
+                details: expect.objectContaining({
+                    custom: 'value',
+                    os_platform: 'darwin',
+                }),
+            }));
+        });
+        it('should clear phases after flushing', () => {
+            const handle = profiler.start('test_phase');
+            handle?.end();
+            profiler.flush(mockConfig);
+            expect(profiler['phases'].size).toBe(0);
+        });
+        it('should detect Docker environment', async () => {
+            const fs = await import('node:fs');
+            fs.existsSync.mockReturnValue(true);
+            const handle = profiler.start('test_phase');
+            handle?.end();
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).toHaveBeenCalledWith(mockConfig, expect.any(Number), expect.objectContaining({
+                details: expect.objectContaining({
+                    is_docker: true,
+                }),
+            }));
+        });
+        it('should calculate CPU usage correctly', () => {
+            const cpuUsageSpy = vi.spyOn(process, 'cpuUsage');
+            // Mock start usage
+            cpuUsageSpy.mockReturnValueOnce({ user: 1000, system: 500 });
+            // Mock diff usage (this is what process.cpuUsage(startUsage) returns)
+            cpuUsageSpy.mockReturnValueOnce({ user: 100, system: 50 });
+            const handle = profiler.start('cpu_test_phase');
+            handle?.end();
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).toHaveBeenCalledWith(mockConfig, expect.any(Number), expect.objectContaining({
+                phase: 'cpu_test_phase',
+                details: expect.objectContaining({
+                    cpu_usage_user: 100,
+                    cpu_usage_system: 50,
+                }),
+            }));
+        });
+        it('should use debug logging instead of standard logging', () => {
+            const logSpy = vi.spyOn(debugLogger, 'log');
+            const debugSpy = vi.spyOn(debugLogger, 'debug');
+            const handle = profiler.start('test_phase');
+            handle?.end();
+            profiler.flush(mockConfig);
+            expect(logSpy).not.toHaveBeenCalled();
+            expect(debugSpy).toHaveBeenCalled();
+        });
+    });
+    describe('integration scenarios', () => {
+        it('should handle a complete startup profiling workflow', () => {
+            // Simulate startup sequence
+            const totalHandle = profiler.start('total_startup');
+            const settingsHandle = profiler.start('load_settings');
+            settingsHandle?.end();
+            const argsHandle = profiler.start('parse_arguments');
+            argsHandle?.end();
+            const appHandle = profiler.start('initialize_app');
+            appHandle?.end();
+            totalHandle?.end();
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).toHaveBeenCalledTimes(4);
+            expect(recordStartupPerformance).toHaveBeenCalledWith(mockConfig, expect.any(Number), expect.objectContaining({ phase: 'total_startup' }));
+        });
+        it('should handle nested timing correctly', () => {
+            const outerHandle = profiler.start('outer');
+            const innerHandle = profiler.start('inner');
+            innerHandle?.end();
+            outerHandle?.end();
+            profiler.flush(mockConfig);
+            const calls = recordStartupPerformance.mock.calls;
+            const outerCall = calls.find((call) => call[2].phase === 'outer');
+            const innerCall = calls.find((call) => call[2].phase === 'inner');
+            expect(outerCall).toBeDefined();
+            expect(innerCall).toBeDefined();
+            // Outer duration should be >= inner duration
+            expect(outerCall[1]).toBeGreaterThanOrEqual(innerCall[1]);
+        });
+    });
+    describe('sanity checking', () => {
+        it('should return undefined when starting a phase that is already active', () => {
+            profiler.start('test_phase');
+            const handle = profiler.start('test_phase');
+            expect(handle).toBeUndefined();
+        });
+        it('should allow restarting a phase after it has ended', () => {
+            const handle1 = profiler.start('test_phase');
+            handle1?.end();
+            // Should not throw
+            expect(() => profiler.start('test_phase')).not.toThrow();
+        });
+        it('should not throw error when ending a phase that is already ended', () => {
+            const handle = profiler.start('test_phase');
+            handle?.end();
+            // Calling end() again on the same handle should not throw
+            expect(() => handle?.end()).not.toThrow();
+        });
+        it('should not record metrics for incomplete phases', () => {
+            profiler.start('incomplete_phase');
+            // Never call end()
+            profiler.flush(mockConfig);
+            expect(recordStartupPerformance).not.toHaveBeenCalled();
+        });
+        it('should handle mix of complete and incomplete phases', () => {
+            const completeHandle = profiler.start('complete_phase');
+            completeHandle?.end();
+            profiler.start('incomplete_phase');
+            // Never call end()
+            profiler.flush(mockConfig);
+            // Should only record the complete phase
+            expect(recordStartupPerformance).toHaveBeenCalledTimes(1);
+            expect(recordStartupPerformance).toHaveBeenCalledWith(mockConfig, expect.any(Number), expect.objectContaining({ phase: 'complete_phase' }));
+        });
+        it('should log startup stats event', () => {
+            const handle = profiler.start('test_phase');
+            handle?.end();
+            profiler.flush(mockConfig);
+            expect(logStartupStats).toHaveBeenCalledWith(mockConfig, expect.objectContaining({
+                phases: expect.arrayContaining([
+                    expect.objectContaining({
+                        name: 'test_phase',
+                        duration_ms: expect.any(Number),
+                        start_time_usec: expect.any(Number),
+                        end_time_usec: expect.any(Number),
+                    }),
+                ]),
+                os_platform: 'darwin',
+                os_release: '22.6.0',
+                is_docker: false,
+            }));
+        });
+    });
+});
+//# sourceMappingURL=startupProfiler.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"startupProfiler.test.js","sourceRoot":"","sources":["../../../src/telemetry/startupProfiler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD,0BAA0B;AAC1B,EAAE,CAAC,IAAI,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7B,wBAAwB,EAAE,EAAE,CAAC,EAAE,EAAE;CAClC,CAAC,CAAC,CAAC;AAEJ,sBAAsB;AACtB,EAAE,CAAC,IAAI,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;CACzB,CAAC,CAAC,CAAC;AAEJ,iBAAiB;AACjB,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1C,MAAM,MAAM,GAAG,MAAM,cAAc,EAA4B,CAAC;IAChE,OAAO;QACL,GAAG,MAAM;QACT,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC;QAC/B,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;QACxB,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC;KAC/B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,iBAAiB;AACjB,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;IACxB,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;IAC9B,iBAAiB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;QAC9B,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE;KACZ,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,QAAyB,CAAC;IAC9B,IAAI,UAAkB,CAAC;IACvB,IAAI,wBAAkD,CAAC;IACvD,IAAI,eAAyC,CAAC;IAE9C,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,0BAA0B;QAC1B,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,wBAAwB;YACtB,aAAa,CAAC,wBAAoD,CAAC;QAErE,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,eAAe,GAAG,aAAa,CAAC,eAA2C,CAAC;QAE5E,mCAAmC;QACnC,QAAQ,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;QAEzC,oDAAoD;QACpD,QAAQ,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,CAAC;QAC3B,WAAW,CAAC,UAAU,EAAE,CAAC;QACzB,WAAW,CAAC,aAAa,EAAE,CAAC;QAE5B,UAAU,GAAG;YACX,YAAY,EAAE,GAAG,EAAE,CAAC,iBAAiB;YACrC,mBAAmB,EAAE,GAAG,EAAE,CAAC,IAAI;SACX,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,SAAS,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;YAChD,MAAM,SAAS,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;YAChD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAE7B,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YACnD,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5B,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAEvC,sCAAsC;YACtC,MAAM,KAAK,GAAG,WAAW,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAC1B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,0BAA0B,CAC7C,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAC5C,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAEtC,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YACnD,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,yCAAyC;YACzC,MAAM,QAAQ,GAAG,WAAW,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;YACzD,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;YAC9D,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAClE,MAAM,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;YAEpC,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YACnD,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC;gBAC7B,OAAO,EAAE,OAAO;gBAChB,UAAU,EAAE,MAAM;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,CAAC;YACjE,MAAM,EAAE,GAAG,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC;YAEhC,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YACnD,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACzC,OAAO,EAAE,GAAG,EAAE,CAAC;YAEf,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACzC,OAAO,EAAE,GAAG,EAAE,CAAC;YAEf,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,QAAQ,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACnC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,QAAQ;oBACrB,OAAO,EAAE,KAAK;oBACd,UAAU,EAAE,QAAQ;oBACpB,SAAS,EAAE,KAAK;oBAChB,cAAc,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAClC,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;iBACrC,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YACjE,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,MAAM,EAAE,OAAO;oBACf,WAAW,EAAE,QAAQ;iBACtB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YAClC,EAAE,CAAC,UAAuC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAElE,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,SAAS,EAAE,IAAI;iBAChB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,WAAW,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YAClD,mBAAmB;YACnB,WAAW,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;YAC7D,sEAAsE;YACtE,WAAW,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAChD,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,KAAK,EAAE,gBAAgB;gBACvB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,cAAc,EAAE,GAAG;oBACnB,gBAAgB,EAAE,EAAE;iBACrB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,MAAM,GAAG,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAC5C,MAAM,QAAQ,GAAG,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAEhD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACtC,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,4BAA4B;YAC5B,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;YAEpD,MAAM,cAAc,GAAG,QAAQ,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;YACvD,cAAc,EAAE,GAAG,EAAE,CAAC;YAEtB,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YACrD,UAAU,EAAE,GAAG,EAAE,CAAC;YAElB,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YACnD,SAAS,EAAE,GAAG,EAAE,CAAC;YAEjB,WAAW,EAAE,GAAG,EAAE,CAAC;YAEnB,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5C,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5C,WAAW,EAAE,GAAG,EAAE,CAAC;YACnB,WAAW,EAAE,GAAG,EAAE,CAAC;YAEnB,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,CAAC;YAClD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,CAAC;YAClE,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,CAAC;YAElE,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;YAChC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;YAChC,6CAA6C;YAC7C,MAAM,CAAC,SAAU,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,SAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC7C,OAAO,EAAE,GAAG,EAAE,CAAC;YAEf,mBAAmB;YACnB,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,0DAA0D;YAC1D,MAAM,CAAC,GAAG,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,QAAQ,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACnC,mBAAmB;YAEnB,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,wBAAwB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,cAAc,GAAG,QAAQ,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YACxD,cAAc,EAAE,GAAG,EAAE,CAAC;YAEtB,QAAQ,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACnC,mBAAmB;YAEnB,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,wCAAwC;YACxC,MAAM,CAAC,wBAAwB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC;YAEd,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAE3B,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,UAAU,EACV,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,eAAe,CAAC;oBAC7B,MAAM,CAAC,gBAAgB,CAAC;wBACtB,IAAI,EAAE,YAAY;wBAClB,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;wBAC/B,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;wBACnC,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;qBAClC,CAAC;iBACH,CAAC;gBACF,WAAW,EAAE,QAAQ;gBACrB,UAAU,EAAE,QAAQ;gBACpB,SAAS,EAAE,KAAK;aACjB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -7,13 +7,20 @@ 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';
+import { GoogleAuth } from 'google-auth-library';
 vi.mock('@opentelemetry/sdk-node');
 vi.mock('../config/config.js');
+vi.mock('google-auth-library');
 describe('telemetry', () => {
     let mockConfig;
     let mockNodeSdk;
+    const mockGetApplicationDefault = vi.fn();
     beforeEach(() => {
         vi.resetAllMocks();
+        vi.mocked(GoogleAuth).mockImplementation(() => ({
+            getApplicationDefault: mockGetApplicationDefault,
+        }));
+        mockGetApplicationDefault.mockResolvedValue(undefined); // Simulate ADC available
         mockConfig = new Config({
             sessionId: 'test-session-id',
             model: 'test-model',
@@ -36,13 +43,13 @@ describe('telemetry', () => {
             await shutdownTelemetry(mockConfig);
         }
     });
-    it('should initialize the telemetry service', () => {
-        initializeTelemetry(mockConfig);
+    it('should initialize the telemetry service', async () => {
+        await initializeTelemetry(mockConfig);
         expect(NodeSDK).toHaveBeenCalled();
         expect(mockNodeSdk.start).toHaveBeenCalled();
     });
     it('should shutdown the telemetry service', async () => {
-        initializeTelemetry(mockConfig);
+        await initializeTelemetry(mockConfig);
         await shutdownTelemetry(mockConfig);
         expect(mockNodeSdk.shutdown).toHaveBeenCalled();
     });

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

@@ -1 +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,CAAC,UAAU,CAAC,CAAC;QACtC,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,CAAC,UAAU,CAAC,CAAC;QAEpC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"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;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAC/B,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,UAAkB,CAAC;IACvB,IAAI,WAAoB,CAAC;IACzB,MAAM,yBAAyB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAE1C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,kBAAkB,CACtC,GAAG,EAAE,CACH,CAAC;YACC,qBAAqB,EAAE,yBAAyB;SACjD,CAA0B,CAC9B,CAAC;QACF,yBAAyB,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC,CAAC,yBAAyB;QAEjF,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,CAAC,UAAU,CAAC,CAAC;QACtC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,mBAAmB,CAAC,UAAU,CAAC,CAAC;QACtC,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,MAAM,mBAAmB,CAAC,UAAU,CAAC,CAAC;QACtC,MAAM,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAEpC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/telemetry/trace.js

@@ -30,7 +30,7 @@ export async function runInDevTraceSpan(opts, fn) {
     const { name: spanName, noAutoEnd, ...restOfSpanOpts } = opts;
     if (process.env['GEMINI_DEV_TRACING'] !== 'true') {
         // If GEMINI_DEV_TRACING env var not set, we do not trace.
-        return await fn({
+        return fn({
             metadata: {
                 name: spanName,
                 attributes: {},
@@ -41,7 +41,7 @@ export async function runInDevTraceSpan(opts, fn) {
         });
     }
     const tracer = trace.getTracer(TRACER_NAME, TRACER_VERSION);
-    return await tracer.startActiveSpan(opts.name, restOfSpanOpts, async (span) => {
+    return tracer.startActiveSpan(opts.name, restOfSpanOpts, async (span) => {
         const meta = {
             name: spanName,
             attributes: {},

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

@@ -1 +1 @@
-{"version":3,"file":"trace.js","sourceRoot":"","sources":["../../../src/telemetry/trace.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,IAAI,EACJ,cAAc,EACd,KAAK,GAGN,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAElE,MAAM,WAAW,GAAG,YAAY,CAAC;AACjC,MAAM,cAAc,GAAG,IAAI,CAAC;AAiB5B;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAyD,EACzD,EAKgB;IAEhB,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,cAAc,EAAE,GAAG,IAAI,CAAC;IAC9D,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,KAAK,MAAM,EAAE,CAAC;QACjD,0DAA0D;QAC1D,OAAO,MAAM,EAAE,CAAC;YACd,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACf;YACD,OAAO,EAAE,GAAG,EAAE;gBACZ,OAAO;YACT,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IAC5D,OAAO,MAAM,MAAM,CAAC,eAAe,CACjC,IAAI,CAAC,IAAI,EACT,cAAc,EACd,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,MAAM,IAAI,GAAiB;YACzB,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf,CAAC;QACF,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,IAAI,CAAC;gBACH,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;oBAC7B,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;gBACjE,CAAC;gBACD,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC9B,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;gBACnE,CAAC;gBACD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;oBAC3D,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAuB,CAAC,CAAC;gBAClD,CAAC;gBACD,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,SAAS,CAAC;wBACb,IAAI,EAAE,cAAc,CAAC,KAAK;wBAC1B,OAAO,EAAE,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC;qBACrC,CAAC,CAAC;oBACH,IAAI,IAAI,CAAC,KAAK,YAAY,KAAK,EAAE,CAAC;wBAChC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACnC,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,mEAAmE;gBACnE,IAAI,CAAC,KAAK,CAAC,0CAA0C,EAAE,CAAC,CAAC,CAAC;gBAC1D,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAE,cAAc,CAAC,KAAK;oBAC1B,OAAO,EAAE,qBAAqB,eAAe,CAAC,CAAC,CAAC,EAAE;iBACnD,CAAC,CAAC;YACL,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC,CAAC;QACF,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;YACf,IAAI,SAAS,EAAE,CAAC;gBACd,2EAA2E;gBAC3E,mEAAmE;gBACnE,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,wEAAwE;gBACxE,sEAAsE;gBACtE,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAS,eAAe,CAAC,CAAU;IACjC,IAAI,CAAC,YAAY,KAAK,EAAE,CAAC;QACvB,OAAO,CAAC,CAAC,OAAO,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,iBAAiB,CAAC,CAAC,CAAC,CAAC;AAC9B,CAAC"}
+{"version":3,"file":"trace.js","sourceRoot":"","sources":["../../../src/telemetry/trace.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,IAAI,EACJ,cAAc,EACd,KAAK,GAGN,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAElE,MAAM,WAAW,GAAG,YAAY,CAAC;AACjC,MAAM,cAAc,GAAG,IAAI,CAAC;AAiB5B;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAyD,EACzD,EAKgB;IAEhB,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,cAAc,EAAE,GAAG,IAAI,CAAC;IAC9D,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,KAAK,MAAM,EAAE,CAAC;QACjD,0DAA0D;QAC1D,OAAO,EAAE,CAAC;YACR,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACf;YACD,OAAO,EAAE,GAAG,EAAE;gBACZ,OAAO;YACT,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IAC5D,OAAO,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QACtE,MAAM,IAAI,GAAiB;YACzB,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf,CAAC;QACF,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,IAAI,CAAC;gBACH,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;oBAC7B,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;gBACjE,CAAC;gBACD,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC9B,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;gBACnE,CAAC;gBACD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;oBAC3D,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAChC,CAAC;gBACD,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,SAAS,CAAC;wBACb,IAAI,EAAE,cAAc,CAAC,KAAK;wBAC1B,OAAO,EAAE,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC;qBACrC,CAAC,CAAC;oBACH,IAAI,IAAI,CAAC,KAAK,YAAY,KAAK,EAAE,CAAC;wBAChC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACnC,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,mEAAmE;gBACnE,IAAI,CAAC,KAAK,CAAC,0CAA0C,EAAE,CAAC,CAAC,CAAC;gBAC1D,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAE,cAAc,CAAC,KAAK;oBAC1B,OAAO,EAAE,qBAAqB,eAAe,CAAC,CAAC,CAAC,EAAE;iBACnD,CAAC,CAAC;YACL,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC,CAAC;QACF,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;YACf,IAAI,SAAS,EAAE,CAAC;gBACd,2EAA2E;gBAC3E,mEAAmE;gBACnE,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,wEAAwE;gBACxE,sEAAsE;gBACtE,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,SAAS,eAAe,CAAC,CAAU;IACjC,IAAI,CAAC,YAAY,KAAK,EAAE,CAAC;QACvB,OAAO,CAAC,CAAC,OAAO,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,iBAAiB,CAAC,CAAC,CAAC,CAAC;AAC9B,CAAC"}

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

@@ -14,6 +14,7 @@ export { ToolCallDecision };
 import type { ToolRegistry } from '../tools/tool-registry.js';
 import type { OutputFormat } from '../output/types.js';
 import type { AgentTerminateMode } from '../agents/types.js';
+import type { OTelFinishReason } from './semantic.js';
 export interface BaseTelemetryEvent {
     'event.name': string;
     /** Current timestamp in ISO 8601 format */
@@ -98,11 +99,11 @@ export declare class ApiRequestEvent implements BaseTelemetryEvent {
     'event.name': 'api_request';
     'event.timestamp': string;
     model: string;
-    prompt_id: string;
+    prompt: GenAIPromptDetails;
     request_text?: string;
-    constructor(model: string, prompt_id: string, request_text?: string);
-    toOpenTelemetryAttributes(config: Config): LogAttributes;
-    toLogBody(): string;
+    constructor(model: string, prompt_details: GenAIPromptDetails, request_text?: string);
+    toLogRecord(config: Config): LogRecord;
+    toSemanticLogRecord(config: Config): LogRecord;
 }
 export declare const EVENT_API_ERROR = "gemini_cli.api_error";
 export declare class ApiErrorEvent implements BaseTelemetryEvent {
@@ -154,6 +155,7 @@ export declare class ApiResponseEvent implements BaseTelemetryEvent {
     prompt: GenAIPromptDetails;
     response: GenAIResponseDetails;
     usage: GenAIUsageDetails;
+    finish_reasons: OTelFinishReason[];
     constructor(model: string, duration_ms: number, prompt_details: GenAIPromptDetails, response_details: GenAIResponseDetails, auth_type?: string, usage_data?: GenerateContentResponseUsageMetadata, response_text?: string);
     toLogRecord(config: Config): LogRecord;
     toSemanticLogRecord(config: Config): LogRecord;
@@ -334,11 +336,12 @@ export declare class ExtensionInstallEvent implements BaseTelemetryEvent {
     'event.name': 'extension_install';
     'event.timestamp': string;
     extension_name: string;
+    hashed_extension_name: string;
     extension_id: string;
     extension_version: string;
     extension_source: string;
     status: 'success' | 'error';
-    constructor(extension_name: string, extension_id: string, extension_version: string, extension_source: string, status: 'success' | 'error');
+    constructor(extension_name: string, hashed_extension_name: string, extension_id: string, extension_version: string, extension_source: string, status: 'success' | 'error');
     toOpenTelemetryAttributes(config: Config): LogAttributes;
     toLogBody(): string;
 }
@@ -368,9 +371,10 @@ export declare class ExtensionUninstallEvent implements BaseTelemetryEvent {
     'event.name': 'extension_uninstall';
     'event.timestamp': string;
     extension_name: string;
+    hashed_extension_name: string;
     extension_id: string;
     status: 'success' | 'error';
-    constructor(extension_name: string, extension_id: string, status: 'success' | 'error');
+    constructor(extension_name: string, hashed_extension_name: string, extension_id: string, status: 'success' | 'error');
     toOpenTelemetryAttributes(config: Config): LogAttributes;
     toLogBody(): string;
 }
@@ -379,12 +383,13 @@ export declare class ExtensionUpdateEvent implements BaseTelemetryEvent {
     'event.name': 'extension_update';
     'event.timestamp': string;
     extension_name: string;
+    hashed_extension_name: string;
     extension_id: string;
     extension_previous_version: string;
     extension_version: string;
     extension_source: string;
     status: 'success' | 'error';
-    constructor(extension_name: string, extension_id: string, extension_version: string, extension_previous_version: string, extension_source: string, status: 'success' | 'error');
+    constructor(extension_name: string, hashed_extension_name: string, extension_id: string, extension_version: string, extension_previous_version: string, extension_source: string, status: 'success' | 'error');
     toOpenTelemetryAttributes(config: Config): LogAttributes;
     toLogBody(): string;
 }
@@ -393,9 +398,10 @@ export declare class ExtensionEnableEvent implements BaseTelemetryEvent {
     'event.name': 'extension_enable';
     'event.timestamp': string;
     extension_name: string;
+    hashed_extension_name: string;
     extension_id: string;
     setting_scope: string;
-    constructor(extension_name: string, extension_id: string, settingScope: string);
+    constructor(extension_name: string, hashed_extension_name: string, extension_id: string, settingScope: string);
     toOpenTelemetryAttributes(config: Config): LogAttributes;
     toLogBody(): string;
 }
@@ -420,15 +426,16 @@ export declare class LlmLoopCheckEvent implements BaseTelemetryEvent {
     toOpenTelemetryAttributes(config: Config): LogAttributes;
     toLogBody(): string;
 }
-export type TelemetryEvent = StartSessionEvent | EndSessionEvent | UserPromptEvent | ToolCallEvent | ApiRequestEvent | ApiErrorEvent | ApiResponseEvent | FlashFallbackEvent | LoopDetectedEvent | LoopDetectionDisabledEvent | NextSpeakerCheckEvent | MalformedJsonResponseEvent | IdeConnectionEvent | ConversationFinishedEvent | SlashCommandEvent | FileOperationEvent | InvalidChunkEvent | ContentRetryEvent | ContentRetryFailureEvent | ExtensionEnableEvent | ExtensionInstallEvent | ExtensionUninstallEvent | ModelRoutingEvent | ToolOutputTruncatedEvent | ModelSlashCommandEvent | AgentStartEvent | AgentFinishEvent | RecoveryAttemptEvent | LlmLoopCheckEvent | WebFetchFallbackAttemptEvent;
+export type TelemetryEvent = StartSessionEvent | EndSessionEvent | UserPromptEvent | ToolCallEvent | ApiRequestEvent | ApiErrorEvent | ApiResponseEvent | FlashFallbackEvent | LoopDetectedEvent | LoopDetectionDisabledEvent | NextSpeakerCheckEvent | MalformedJsonResponseEvent | IdeConnectionEvent | ConversationFinishedEvent | SlashCommandEvent | FileOperationEvent | InvalidChunkEvent | ContentRetryEvent | ContentRetryFailureEvent | ExtensionEnableEvent | ExtensionInstallEvent | ExtensionUninstallEvent | ModelRoutingEvent | ToolOutputTruncatedEvent | ModelSlashCommandEvent | AgentStartEvent | AgentFinishEvent | RecoveryAttemptEvent | LlmLoopCheckEvent | StartupStatsEvent | WebFetchFallbackAttemptEvent;
 export declare const EVENT_EXTENSION_DISABLE = "gemini_cli.extension_disable";
 export declare class ExtensionDisableEvent implements BaseTelemetryEvent {
     'event.name': 'extension_disable';
     'event.timestamp': string;
     extension_name: string;
+    hashed_extension_name: string;
     extension_id: string;
     setting_scope: string;
-    constructor(extension_name: string, extension_id: string, settingScope: string);
+    constructor(extension_name: string, hashed_extension_name: string, extension_id: string, settingScope: string);
     toOpenTelemetryAttributes(config: Config): LogAttributes;
     toLogBody(): string;
 }
@@ -450,6 +457,26 @@ export declare class SmartEditCorrectionEvent implements BaseTelemetryEvent {
     toOpenTelemetryAttributes(config: Config): LogAttributes;
     toLogBody(): string;
 }
+export interface StartupPhaseStats {
+    name: string;
+    duration_ms: number;
+    cpu_usage_user_usec: number;
+    cpu_usage_system_usec: number;
+    start_time_usec: number;
+    end_time_usec: number;
+}
+export declare const EVENT_STARTUP_STATS = "gemini_cli.startup_stats";
+export declare class StartupStatsEvent implements BaseTelemetryEvent {
+    'event.name': 'startup_stats';
+    'event.timestamp': string;
+    phases: StartupPhaseStats[];
+    os_platform: string;
+    os_release: string;
+    is_docker: boolean;
+    constructor(phases: StartupPhaseStats[], os_platform: string, os_release: string, is_docker: boolean);
+    toOpenTelemetryAttributes(config: Config): LogAttributes;
+    toLogBody(): string;
+}
 declare abstract class BaseAgentEvent implements BaseTelemetryEvent {
     abstract 'event.name': 'agent_start' | 'agent_finish' | 'agent_recovery_attempt';
     'event.timestamp': string;