@machina.ai/cell-cli-core 1.11.0-rc1 → 1.13.0-rc2

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

@@ -0,0 +1,387 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { toChatMessage, toInputMessages, toSystemInstruction, toOutputMessages, toFinishReasons, OTelFinishReason, toOutputType, OTelOutputType, } from './semantic.js';
+import { Language, Outcome, FinishReason, } from '@google/genai';
+describe('toChatMessage', () => {
+    it('should correctly handle text parts', () => {
+        const content = {
+            role: 'user',
+            parts: [{ text: 'Hello' }],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'user',
+            parts: [
+                {
+                    type: 'text',
+                    content: 'Hello',
+                },
+            ],
+        });
+    });
+    it('should correctly handle function call parts', () => {
+        const content = {
+            role: 'model',
+            parts: [
+                {
+                    functionCall: {
+                        name: 'test-function',
+                        args: {
+                            arg1: 'test-value',
+                        },
+                        id: '12345',
+                    },
+                    // include field not specified in semantic specification that could be present
+                    thoughtSignature: '1234',
+                },
+            ],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'system',
+            parts: [
+                {
+                    type: 'tool_call',
+                    name: 'test-function',
+                    arguments: '{"arg1":"test-value"}',
+                    id: '12345',
+                },
+            ],
+        });
+    });
+    it('should correctly handle function response parts', () => {
+        const content = {
+            role: 'user',
+            parts: [
+                {
+                    functionResponse: {
+                        name: 'test-function',
+                        response: {
+                            result: 'success',
+                        },
+                        id: '12345',
+                    },
+                    // include field not specified in semantic specification that could be present
+                    fileData: {
+                        displayName: 'greatfile',
+                    },
+                },
+            ],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'user',
+            parts: [
+                {
+                    type: 'tool_call_response',
+                    response: '{"result":"success"}',
+                    id: '12345',
+                },
+            ],
+        });
+    });
+    it('should correctly handle reasoning parts with text', () => {
+        const content = {
+            role: 'system',
+            parts: [{ text: 'Hmm', thought: true }],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'system',
+            parts: [
+                {
+                    type: 'reasoning',
+                    content: 'Hmm',
+                },
+            ],
+        });
+    });
+    it('should correctly handle reasoning parts without text', () => {
+        const content = {
+            role: 'system',
+            parts: [
+                {
+                    thought: true,
+                    // include field not specified in semantic specification that could be present
+                    inlineData: {
+                        displayName: 'wowdata',
+                    },
+                },
+            ],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'system',
+            parts: [
+                {
+                    type: 'reasoning',
+                    content: '',
+                },
+            ],
+        });
+    });
+    it('should correctly handle text parts that are not reasoning parts', () => {
+        const content = {
+            role: 'user',
+            parts: [{ text: 'what a nice day', thought: false }],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'user',
+            parts: [
+                {
+                    type: 'text',
+                    content: 'what a nice day',
+                },
+            ],
+        });
+    });
+    it('should correctly handle "generic" parts', () => {
+        const content = {
+            role: 'model',
+            parts: [
+                {
+                    executableCode: {
+                        code: 'print("foo")',
+                        language: Language.PYTHON,
+                    },
+                },
+                {
+                    codeExecutionResult: {
+                        outcome: Outcome.OUTCOME_OK,
+                        output: 'foo',
+                    },
+                    // include field not specified in semantic specification that could be present
+                    videoMetadata: {
+                        fps: 5,
+                    },
+                },
+            ],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'system',
+            parts: [
+                {
+                    type: 'executableCode',
+                    code: 'print("foo")',
+                    language: 'PYTHON',
+                },
+                {
+                    type: 'codeExecutionResult',
+                    outcome: 'OUTCOME_OK',
+                    output: 'foo',
+                    videoMetadata: {
+                        fps: 5,
+                    },
+                },
+            ],
+        });
+    });
+    it('should correctly handle unknown parts', () => {
+        const content = {
+            role: 'model',
+            parts: [
+                {
+                    fileData: {
+                        displayName: 'superfile',
+                    },
+                },
+            ],
+        };
+        expect(toChatMessage(content)).toEqual({
+            role: 'system',
+            parts: [
+                {
+                    type: 'unknown',
+                    fileData: {
+                        displayName: 'superfile',
+                    },
+                },
+            ],
+        });
+    });
+});
+describe('toSystemInstruction', () => {
+    it('should correctly handle a string', () => {
+        const content = 'Hello';
+        expect(toSystemInstruction(content)).toEqual([
+            {
+                type: 'text',
+                content: 'Hello',
+            },
+        ]);
+    });
+    it('should correctly handle a Content object with a text part', () => {
+        const content = {
+            role: 'user',
+            parts: [{ text: 'Hello' }],
+        };
+        expect(toSystemInstruction(content)).toEqual([
+            {
+                type: 'text',
+                content: 'Hello',
+            },
+        ]);
+    });
+    it('should correctly handle a Content object with multiple parts', () => {
+        const content = {
+            role: 'user',
+            parts: [{ text: 'Hello' }, { text: 'Hmm', thought: true }],
+        };
+        expect(toSystemInstruction(content)).toEqual([
+            {
+                type: 'text',
+                content: 'Hello',
+            },
+            {
+                type: 'reasoning',
+                content: 'Hmm',
+            },
+        ]);
+    });
+});
+describe('toInputMessages', () => {
+    it('should correctly convert an array of Content objects', () => {
+        const contents = [
+            {
+                role: 'user',
+                parts: [{ text: 'Hello' }],
+            },
+            {
+                role: 'model',
+                parts: [{ text: 'Hi there!' }],
+            },
+        ];
+        expect(toInputMessages(contents)).toEqual([
+            {
+                role: 'user',
+                parts: [
+                    {
+                        type: 'text',
+                        content: 'Hello',
+                    },
+                ],
+            },
+            {
+                role: 'system',
+                parts: [
+                    {
+                        type: 'text',
+                        content: 'Hi there!',
+                    },
+                ],
+            },
+        ]);
+    });
+});
+describe('toOutputMessages', () => {
+    it('should correctly convert an array of Candidate objects', () => {
+        const candidates = [
+            {
+                index: 0,
+                finishReason: FinishReason.STOP,
+                content: {
+                    role: 'model',
+                    parts: [{ text: 'This is the first candidate.' }],
+                },
+            },
+            {
+                index: 1,
+                finishReason: FinishReason.MAX_TOKENS,
+                content: {
+                    role: 'model',
+                    parts: [{ text: 'This is the second candidate.' }],
+                },
+            },
+        ];
+        expect(toOutputMessages(candidates)).toEqual([
+            {
+                role: 'system',
+                finish_reason: 'stop',
+                parts: [
+                    {
+                        type: 'text',
+                        content: 'This is the first candidate.',
+                    },
+                ],
+            },
+            {
+                role: 'system',
+                finish_reason: 'length',
+                parts: [
+                    {
+                        type: 'text',
+                        content: 'This is the second candidate.',
+                    },
+                ],
+            },
+        ]);
+    });
+});
+describe('toFinishReasons', () => {
+    it('should return an empty array if candidates is undefined', () => {
+        expect(toFinishReasons(undefined)).toEqual([]);
+    });
+    it('should return an empty array if candidates is an empty array', () => {
+        expect(toFinishReasons([])).toEqual([]);
+    });
+    it('should correctly convert a single candidate', () => {
+        const candidates = [
+            {
+                index: 0,
+                finishReason: FinishReason.STOP,
+                content: {
+                    role: 'model',
+                    parts: [{ text: 'This is the first candidate.' }],
+                },
+            },
+        ];
+        expect(toFinishReasons(candidates)).toEqual([OTelFinishReason.STOP]);
+    });
+    it('should correctly convert multiple candidates', () => {
+        const candidates = [
+            {
+                index: 0,
+                finishReason: FinishReason.STOP,
+                content: {
+                    role: 'model',
+                    parts: [{ text: 'This is the first candidate.' }],
+                },
+            },
+            {
+                index: 1,
+                finishReason: FinishReason.MAX_TOKENS,
+                content: {
+                    role: 'model',
+                    parts: [{ text: 'This is the second candidate.' }],
+                },
+            },
+            {
+                index: 2,
+                finishReason: FinishReason.SAFETY,
+                content: {
+                    role: 'model',
+                    parts: [{ text: 'This is the third candidate.' }],
+                },
+            },
+        ];
+        expect(toFinishReasons(candidates)).toEqual([
+            OTelFinishReason.STOP,
+            OTelFinishReason.LENGTH,
+            OTelFinishReason.CONTENT_FILTER,
+        ]);
+    });
+});
+describe('toOutputType', () => {
+    it('should return TEXT for text/plain', () => {
+        expect(toOutputType('text/plain')).toBe(OTelOutputType.TEXT);
+    });
+    it('should return JSON for application/json', () => {
+        expect(toOutputType('application/json')).toBe(OTelOutputType.JSON);
+    });
+    it('should return the custom mime type for other strings', () => {
+        expect(toOutputType('application/vnd.custom-type')).toBe('application/vnd.custom-type');
+    });
+    it('should return undefined for undefined input', () => {
+        expect(toOutputType(undefined)).toBeUndefined();
+    });
+});
+//# sourceMappingURL=semantic.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"semantic.test.js","sourceRoot":"","sources":["../../../src/telemetry/semantic.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,aAAa,EACb,eAAe,EACf,mBAAmB,EACnB,gBAAgB,EAChB,eAAe,EACf,gBAAgB,EAChB,YAAY,EACZ,cAAc,GACf,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,QAAQ,EAER,OAAO,EAEP,YAAY,GACb,MAAM,eAAe,CAAC;AAEvB,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;SAC3B,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;iBACjB;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,OAAO;YACb,KAAK,EAAE;gBACL;oBACE,YAAY,EAAE;wBACZ,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE;4BACJ,IAAI,EAAE,YAAY;yBACnB;wBACD,EAAE,EAAE,OAAO;qBACZ;oBACD,8EAA8E;oBAC9E,gBAAgB,EAAE,MAAM;iBACzB;aACF;SACF,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,eAAe;oBACrB,SAAS,EAAE,uBAAuB;oBAClC,EAAE,EAAE,OAAO;iBACZ;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL;oBACE,gBAAgB,EAAE;wBAChB,IAAI,EAAE,eAAe;wBACrB,QAAQ,EAAE;4BACR,MAAM,EAAE,SAAS;yBAClB;wBACD,EAAE,EAAE,OAAO;qBACZ;oBACD,8EAA8E;oBAC9E,QAAQ,EAAE;wBACR,WAAW,EAAE,WAAW;qBACzB;iBACF;aACF;SACF,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,oBAAoB;oBAC1B,QAAQ,EAAE,sBAAsB;oBAChC,EAAE,EAAE,OAAO;iBACZ;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SACxC,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,KAAK;iBACf;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE;gBACL;oBACE,OAAO,EAAE,IAAI;oBACb,8EAA8E;oBAC9E,UAAU,EAAE;wBACV,WAAW,EAAE,SAAS;qBACvB;iBACF;aACF;SACF,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,EAAE;iBACZ;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;SACrD,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,iBAAiB;iBAC3B;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,OAAO;YACb,KAAK,EAAE;gBACL;oBACE,cAAc,EAAE;wBACd,IAAI,EAAE,cAAc;wBACpB,QAAQ,EAAE,QAAQ,CAAC,MAAM;qBAC1B;iBACF;gBACD;oBACE,mBAAmB,EAAE;wBACnB,OAAO,EAAE,OAAO,CAAC,UAAU;wBAC3B,MAAM,EAAE,KAAK;qBACd;oBACD,8EAA8E;oBAC9E,aAAa,EAAE;wBACb,GAAG,EAAE,CAAC;qBACP;iBACF;aACF;SACF,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,IAAI,EAAE,cAAc;oBACpB,QAAQ,EAAE,QAAQ;iBACnB;gBACD;oBACE,IAAI,EAAE,qBAAqB;oBAC3B,OAAO,EAAE,YAAY;oBACrB,MAAM,EAAE,KAAK;oBACb,aAAa,EAAE;wBACb,GAAG,EAAE,CAAC;qBACP;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,OAAO;YACb,KAAK,EAAE;gBACL;oBACE,QAAQ,EAAE;wBACR,WAAW,EAAE,WAAW;qBACzB;iBACF;aACF;SACF,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE;wBACR,WAAW,EAAE,WAAW;qBACzB;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,OAAO,GAAG,OAAO,CAAC;QACxB,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3C;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;aACjB;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;SAC3B,CAAC;QACF,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3C;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;aACjB;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SAC3D,CAAC;QACF,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3C;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;aACjB;YACD;gBACE,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,KAAK;aACf;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,QAAQ,GAAc;YAC1B;gBACE,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3B;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;aAC/B;SACF,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC;gBACE,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO;qBACjB;iBACF;aACF;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,WAAW;qBACrB;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,UAAU,GAAgB;YAC9B;gBACE,KAAK,EAAE,CAAC;gBACR,YAAY,EAAE,YAAY,CAAC,IAAI;gBAC/B,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;iBAClD;aACF;YACD;gBACE,KAAK,EAAE,CAAC;gBACR,YAAY,EAAE,YAAY,CAAC,UAAU;gBACrC,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,+BAA+B,EAAE,CAAC;iBACnD;aACF;SACF,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3C;gBACE,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,MAAM;gBACrB,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,8BAA8B;qBACxC;iBACF;aACF;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,QAAQ;gBACvB,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,+BAA+B;qBACzC;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,UAAU,GAAgB;YAC9B;gBACE,KAAK,EAAE,CAAC;gBACR,YAAY,EAAE,YAAY,CAAC,IAAI;gBAC/B,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;iBAClD;aACF;SACF,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,UAAU,GAAgB;YAC9B;gBACE,KAAK,EAAE,CAAC;gBACR,YAAY,EAAE,YAAY,CAAC,IAAI;gBAC/B,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;iBAClD;aACF;YACD;gBACE,KAAK,EAAE,CAAC;gBACR,YAAY,EAAE,YAAY,CAAC,UAAU;gBACrC,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,+BAA+B,EAAE,CAAC;iBACnD;aACF;YACD;gBACE,KAAK,EAAE,CAAC;gBACR,YAAY,EAAE,YAAY,CAAC,MAAM;gBACjC,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;iBAClD;aACF;SACF,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1C,gBAAgB,CAAC,IAAI;YACrB,gBAAgB,CAAC,MAAM;YACvB,gBAAgB,CAAC,cAAc;SAChC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,YAAY,CAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,CACtD,6BAA6B,CAC9B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -6,35 +6,36 @@
 import { describe, it, expect } from 'vitest';
 import { getProgrammingLanguage } from './telemetry-utils.js';
 describe('getProgrammingLanguage', () => {
-    it('should return the programming language when file_path is present', () => {
-        const args = { file_path: 'src/test.ts' };
+    it.each([
+        {
+            name: 'file_path is present',
+            args: { file_path: 'src/test.ts' },
+            expected: 'TypeScript',
+        },
+        {
+            name: 'absolute_path is present',
+            args: { absolute_path: 'src/test.py' },
+            expected: 'Python',
+        },
+        { name: 'path is present', args: { path: 'src/test.go' }, expected: 'Go' },
+        {
+            name: 'no file path is present',
+            args: {},
+            expected: undefined,
+        },
+        {
+            name: 'unknown file extensions',
+            args: { file_path: 'src/test.unknown' },
+            expected: undefined,
+        },
+        {
+            name: 'files with no extension',
+            args: { file_path: 'src/test' },
+            expected: undefined,
+        },
+    ])('should return $expected when $name', ({ args, expected }) => {
         const language = getProgrammingLanguage(args);
-        expect(language).toBe('TypeScript');
-    });
-    it('should return the programming language when absolute_path is present', () => {
-        const args = { absolute_path: 'src/test.py' };
-        const language = getProgrammingLanguage(args);
-        expect(language).toBe('Python');
-    });
-    it('should return the programming language when path is present', () => {
-        const args = { path: 'src/test.go' };
-        const language = getProgrammingLanguage(args);
-        expect(language).toBe('Go');
-    });
-    it('should return undefined when no file path is present', () => {
-        const args = {};
-        const language = getProgrammingLanguage(args);
-        expect(language).toBeUndefined();
-    });
-    it('should handle unknown file extensions gracefully', () => {
-        const args = { file_path: 'src/test.unknown' };
-        const language = getProgrammingLanguage(args);
-        expect(language).toBeUndefined();
-    });
-    it('should handle files with no extension', () => {
-        const args = { file_path: 'src/test' };
-        const language = getProgrammingLanguage(args);
-        expect(language).toBeUndefined();
+        expect(language).toBe(expected);
     });
 });
 //# sourceMappingURL=telemetry-utils.test.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"telemetry-utils.test.js","sourceRoot":"","sources":["../../../src/telemetry/telemetry-utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAE9D,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,IAAI,GAAG,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC9E,MAAM,IAAI,GAAG,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,IAAI,GAAG,EAAE,CAAC;QAChB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,IAAI,GAAG,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;QACvC,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"telemetry-utils.test.js","sourceRoot":"","sources":["../../../src/telemetry/telemetry-utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAE9D,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IAOtC,EAAE,CAAC,IAAI,CAA8B;QACnC;YACE,IAAI,EAAE,sBAAsB;YAC5B,IAAI,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE;YAClC,QAAQ,EAAE,YAAY;SACvB;QACD;YACE,IAAI,EAAE,0BAA0B;YAChC,IAAI,EAAE,EAAE,aAAa,EAAE,aAAa,EAAE;YACtC,QAAQ,EAAE,QAAQ;SACnB;QACD,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;QAC1E;YACE,IAAI,EAAE,yBAAyB;YAC/B,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,SAAS;SACpB;QACD;YACE,IAAI,EAAE,yBAAyB;YAC/B,IAAI,EAAE,EAAE,SAAS,EAAE,kBAAkB,EAAE;YACvC,QAAQ,EAAE,SAAS;SACpB;QACD;YACE,IAAI,EAAE,yBAAyB;YAC/B,IAAI,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE;YAC/B,QAAQ,EAAE,SAAS;SACpB;KACF,CAAC,CAAC,oCAAoC,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC9D,MAAM,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type AttributeValue, type SpanOptions } from '@opentelemetry/api';
+/**
+ * Metadata for a span.
+ */
+export interface SpanMetadata {
+    /** The name of the span. */
+    name: string;
+    /** The input to the span. */
+    input?: unknown;
+    /** The output of the span. */
+    output?: unknown;
+    error?: unknown;
+    /** Additional attributes for the span. */
+    attributes: Record<string, AttributeValue>;
+}
+/**
+ * Runs a function in a new OpenTelemetry span.
+ *
+ * The `meta` object will be automatically used to set the span's status and attributes upon completion.
+ *
+ * @example
+ * ```typescript
+ * runInDevTraceSpan({ name: 'my-operation' }, ({ metadata }) => {
+ *   metadata.input = { foo: 'bar' };
+ *   // ... do work ...
+ *   metadata.output = { result: 'baz' };
+ *   metadata.attributes['my.custom.attribute'] = 'some-value';
+ * });
+ * ```
+ *
+ * @param opts The options for the span.
+ * @param fn The function to run in the span.
+ * @returns The result of the function.
+ */
+export declare function runInDevTraceSpan<R>(opts: SpanOptions & {
+    name: string;
+    noAutoEnd?: boolean;
+}, fn: ({ metadata, }: {
+    metadata: SpanMetadata;
+    endSpan: () => void;
+}) => Promise<R>): Promise<R>;

package/dist/src/telemetry/trace.js

@@ -0,0 +1,121 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { diag, SpanStatusCode, trace, } from '@opentelemetry/api';
+import { safeJsonStringify } from '../utils/safeJsonStringify.js';
+const TRACER_NAME = 'gemini-cli';
+const TRACER_VERSION = 'v1';
+/**
+ * Runs a function in a new OpenTelemetry span.
+ *
+ * The `meta` object will be automatically used to set the span's status and attributes upon completion.
+ *
+ * @example
+ * ```typescript
+ * runInDevTraceSpan({ name: 'my-operation' }, ({ metadata }) => {
+ *   metadata.input = { foo: 'bar' };
+ *   // ... do work ...
+ *   metadata.output = { result: 'baz' };
+ *   metadata.attributes['my.custom.attribute'] = 'some-value';
+ * });
+ * ```
+ *
+ * @param opts The options for the span.
+ * @param fn The function to run in the span.
+ * @returns The result of the function.
+ */
+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({
+            metadata: {
+                name: spanName,
+                attributes: {},
+            },
+            endSpan: () => {
+                // noop
+            },
+        });
+    }
+    const tracer = trace.getTracer(TRACER_NAME, TRACER_VERSION);
+    return await tracer.startActiveSpan(opts.name, restOfSpanOpts, async (span) => {
+        const meta = {
+            name: spanName,
+            attributes: {},
+        };
+        const endSpan = () => {
+            try {
+                if (meta.input !== undefined) {
+                    span.setAttribute('input-json', safeJsonStringify(meta.input));
+                }
+                if (meta.output !== undefined) {
+                    span.setAttribute('output-json', safeJsonStringify(meta.output));
+                }
+                for (const [key, value] of Object.entries(meta.attributes)) {
+                    span.setAttribute(key, value);
+                }
+                if (meta.error) {
+                    span.setStatus({
+                        code: SpanStatusCode.ERROR,
+                        message: getErrorMessage(meta.error),
+                    });
+                    if (meta.error instanceof Error) {
+                        span.recordException(meta.error);
+                    }
+                }
+                else {
+                    span.setStatus({ code: SpanStatusCode.OK });
+                }
+            }
+            catch (e) {
+                // Log the error but don't rethrow, to ensure span.end() is called.
+                diag.error('Error setting span attributes in endSpan', e);
+                span.setStatus({
+                    code: SpanStatusCode.ERROR,
+                    message: `Error in endSpan: ${getErrorMessage(e)}`,
+                });
+            }
+            finally {
+                span.end();
+            }
+        };
+        try {
+            return await fn({ metadata: meta, endSpan });
+        }
+        catch (e) {
+            meta.error = e;
+            if (noAutoEnd) {
+                // For streaming operations, the delegated endSpan call will not be reached
+                // on an exception, so we must end the span here to prevent a leak.
+                endSpan();
+            }
+            throw e;
+        }
+        finally {
+            if (!noAutoEnd) {
+                // For non-streaming operations, this ensures the span is always closed,
+                // and if an error occurred, it will be recorded correctly by endSpan.
+                endSpan();
+            }
+        }
+    });
+}
+/**
+ * Gets the error message from an error object.
+ *
+ * @param e The error object.
+ * @returns The error message.
+ */
+function getErrorMessage(e) {
+    if (e instanceof Error) {
+        return e.message;
+    }
+    if (typeof e === 'string') {
+        return e;
+    }
+    return safeJsonStringify(e);
+}
+//# sourceMappingURL=trace.js.map

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

@@ -0,0 +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"}