openlit 1.13.0 → 1.14.0

package/dist/instrumentation/__tests__/gpu-collector.test.js

@@ -0,0 +1,247 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const gpu_1 = require("../gpu");
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+// Columns match NVIDIA_SMI_FIELDS order:
+// index, uuid, name, util.gpu, util.encoder, util.decoder, temp, fan,
+// mem.total, mem.used, mem.free, power.draw, enforced.power.limit, power.limit
+const NVIDIA_CSV = '0, GPU-11111111-2222-3333-4444-555555555555, NVIDIA A100-SXM4-40GB, 87, 15, 5, 65, 30, 40960, 30000, 10960, 250.50, 400.00, 400.00\n' +
+    '1, GPU-aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee, NVIDIA A100-SXM4-40GB, 12, 0, 0, 40, 25, 40960, 1024, 39936, 60.25, 0, 400.00\n';
+const ROCM_JSON = JSON.stringify({
+    card0: {
+        'GPU ID': '0x740f',
+        'Unique ID': '0x123456789abcdef',
+        'Card series': 'AMD Instinct MI210',
+        'GPU use (%)': '42',
+        'Temperature (Sensor edge) (C)': '55.0',
+        'Fan speed (%)': '35',
+        'VRAM Total Memory (B)': String(64 * 1024 * 1024 * 1024),
+        'VRAM Total Used Memory (B)': String(16 * 1024 * 1024 * 1024),
+        'Average Graphics Package Power (W)': '180.0',
+        'Max Graphics Package Power (W)': '300.0',
+    },
+    system: { 'Driver version': '6.2' },
+});
+function nvidiaRunner() {
+    return async (cmd, args) => {
+        if (cmd !== 'nvidia-smi')
+            throw new Error(`${cmd}: command not found`);
+        if (args[0] === '-L')
+            return 'GPU 0: NVIDIA A100\nGPU 1: NVIDIA A100\n';
+        return NVIDIA_CSV;
+    };
+}
+function amdRunner() {
+    return async (cmd) => {
+        if (cmd !== 'rocm-smi')
+            throw new Error(`${cmd}: command not found`);
+        return ROCM_JSON;
+    };
+}
+const noGpuRunner = async (cmd) => {
+    throw new Error(`${cmd}: command not found`);
+};
+function makeFakeMeter() {
+    const gauges = [];
+    const observations = [];
+    let callback = null;
+    const meter = {
+        createObservableGauge: (name) => {
+            const token = { name };
+            gauges.push({ name, token });
+            return token;
+        },
+        addBatchObservableCallback: (cb) => {
+            callback = cb;
+        },
+    };
+    const runCallback = async () => {
+        if (!callback)
+            throw new Error('no callback registered');
+        await callback({
+            observe: (gauge, value, attributes) => observations.push({ gauge, value, attributes }),
+        });
+    };
+    return { meter: meter, gauges, observations, runCallback };
+}
+describe('GpuMetricsCollector', () => {
+    it('detects nvidia GPUs', async () => {
+        expect(await new gpu_1.GpuMetricsCollector(nvidiaRunner()).detectGpuType()).toBe('nvidia');
+    });
+    it('detects amd GPUs when nvidia-smi is missing', async () => {
+        expect(await new gpu_1.GpuMetricsCollector(amdRunner()).detectGpuType()).toBe('amd');
+    });
+    it('returns null when no supported GPU tooling exists', async () => {
+        expect(await new gpu_1.GpuMetricsCollector(noGpuRunner).detectGpuType()).toBeNull();
+    });
+    it('setup registers all 11 gauges with Python-matching metric names', async () => {
+        const { meter, gauges } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(nvidiaRunner());
+        const gpuType = await collector.setup({
+            meter,
+            environment: 'staging',
+            applicationName: 'my-app',
+        });
+        expect(gpuType).toBe('nvidia');
+        expect(gauges.map((g) => g.name)).toEqual([
+            'gpu.utilization',
+            'gpu.enc.utilization',
+            'gpu.dec.utilization',
+            'gpu.temperature',
+            'gpu.fan_speed',
+            'gpu.memory.available',
+            'gpu.memory.total',
+            'gpu.memory.used',
+            'gpu.memory.free',
+            'gpu.power.draw',
+            'gpu.power.limit',
+        ]);
+    });
+    it('setup returns null and registers nothing without a GPU', async () => {
+        const { meter, gauges } = makeFakeMeter();
+        const gpuType = await new gpu_1.GpuMetricsCollector(noGpuRunner).setup({
+            meter,
+            environment: 'staging',
+            applicationName: 'my-app',
+        });
+        expect(gpuType).toBeNull();
+        expect(gauges).toHaveLength(0);
+    });
+    it('observes nvidia stats per GPU with Python-matching attributes', async () => {
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(nvidiaRunner());
+        await collector.setup({ meter, environment: 'staging', applicationName: 'my-app' });
+        await runCallback();
+        // 2 GPUs x 11 metrics
+        expect(observations).toHaveLength(22);
+        const gpu0 = observations.filter((o) => o.attributes[semantic_convention_1.default.GPU_INDEX] === '0');
+        expect(gpu0).toHaveLength(11);
+        expect(gpu0[0].attributes).toEqual({
+            'telemetry.sdk.name': 'openlit',
+            'service.name': 'my-app',
+            [semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]: 'staging',
+            [semantic_convention_1.default.GPU_INDEX]: '0',
+            [semantic_convention_1.default.GPU_UUID]: 'GPU-11111111-2222-3333-4444-555555555555',
+            [semantic_convention_1.default.GPU_NAME]: 'NVIDIA A100-SXM4-40GB',
+        });
+        const valueOf = (metric, index) => observations.find((o) => o.gauge.name === metric &&
+            o.attributes[semantic_convention_1.default.GPU_INDEX] === index)?.value;
+        expect(valueOf('gpu.utilization', '0')).toBe(87);
+        expect(valueOf('gpu.enc.utilization', '0')).toBe(15);
+        expect(valueOf('gpu.dec.utilization', '0')).toBe(5);
+        expect(valueOf('gpu.temperature', '0')).toBe(65);
+        expect(valueOf('gpu.fan_speed', '0')).toBe(30);
+        expect(valueOf('gpu.memory.total', '0')).toBe(40960);
+        expect(valueOf('gpu.memory.used', '0')).toBe(30000);
+        expect(valueOf('gpu.memory.free', '0')).toBe(10960);
+        expect(valueOf('gpu.memory.available', '0')).toBe(10960);
+        expect(valueOf('gpu.power.draw', '0')).toBe(250.5);
+        expect(valueOf('gpu.power.limit', '0')).toBe(400);
+        expect(valueOf('gpu.utilization', '1')).toBe(12);
+        // GPU 1: enforced.power.limit is 0 / N/A → fall back to power.limit
+        expect(valueOf('gpu.power.limit', '1')).toBe(400);
+        expect(valueOf('gpu.enc.utilization', '1')).toBe(0);
+        expect(valueOf('gpu.dec.utilization', '1')).toBe(0);
+    });
+    it('observes amd stats from rocm-smi JSON', async () => {
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(amdRunner());
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        await runCallback();
+        expect(observations).toHaveLength(11);
+        const attrs = observations[0].attributes;
+        expect(attrs[semantic_convention_1.default.GPU_INDEX]).toBe('0');
+        expect(attrs[semantic_convention_1.default.GPU_NAME]).toBe('AMD Instinct MI210');
+        expect(attrs[semantic_convention_1.default.GPU_UUID]).toBe('0x123456789abcdef');
+        const valueOf = (metric) => observations.find((o) => o.gauge.name === metric)?.value;
+        expect(valueOf('gpu.utilization')).toBe(42);
+        expect(valueOf('gpu.enc.utilization')).toBe(0);
+        expect(valueOf('gpu.dec.utilization')).toBe(0);
+        expect(valueOf('gpu.temperature')).toBe(55);
+        expect(valueOf('gpu.fan_speed')).toBe(35);
+        expect(valueOf('gpu.memory.total')).toBe(64 * 1024);
+        expect(valueOf('gpu.memory.used')).toBe(16 * 1024);
+        expect(valueOf('gpu.memory.free')).toBe(48 * 1024);
+        // Python AMD memory_available == total (not free)
+        expect(valueOf('gpu.memory.available')).toBe(64 * 1024);
+        expect(valueOf('gpu.power.draw')).toBe(180);
+        expect(valueOf('gpu.power.limit')).toBe(300);
+    });
+    it('swallows collection errors without throwing from the callback', async () => {
+        let detect = true;
+        const flakyRunner = async (cmd, args) => {
+            if (cmd !== 'nvidia-smi')
+                throw new Error('not found');
+            if (args[0] === '-L' && detect)
+                return 'GPU 0: NVIDIA A100';
+            throw new Error('nvidia-smi crashed');
+        };
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(flakyRunner);
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        detect = false;
+        await expect(runCallback()).resolves.toBeUndefined();
+        expect(observations).toHaveLength(0);
+    });
+    it('swallows invalid AMD JSON without throwing from the callback', async () => {
+        const badJsonRunner = async (cmd, args) => {
+            if (cmd === 'nvidia-smi')
+                throw new Error('not found');
+            if (cmd !== 'rocm-smi')
+                throw new Error('not found');
+            if (args.includes('--showid') && args.length === 2)
+                return '{}';
+            return 'not-json{';
+        };
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(badJsonRunner);
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        await expect(runCallback()).resolves.toBeUndefined();
+        expect(observations).toHaveLength(0);
+    });
+    it('parses malformed numeric fields as 0', async () => {
+        const badCsvRunner = async (cmd, args) => {
+            if (cmd !== 'nvidia-smi')
+                throw new Error('not found');
+            if (args[0] === '-L')
+                return 'GPU 0: NVIDIA A100';
+            return '0, GPU-uuid, NVIDIA A100, [N/A], [N/A], [N/A], 65, [N/A], 40960, 30000, 10960, [N/A], [N/A], 400.00\n';
+        };
+        const collector = new gpu_1.GpuMetricsCollector(badCsvRunner);
+        const { meter, runCallback, observations } = makeFakeMeter();
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        await runCallback();
+        const valueOf = (metric) => observations.find((o) => o.gauge.name === metric)?.value;
+        expect(valueOf('gpu.utilization')).toBe(0);
+        expect(valueOf('gpu.enc.utilization')).toBe(0);
+        expect(valueOf('gpu.dec.utilization')).toBe(0);
+        expect(valueOf('gpu.power.draw')).toBe(0);
+        // enforced N/A → fall back to power.limit
+        expect(valueOf('gpu.power.limit')).toBe(400);
+    });
+});
+describe('GpuStats typing', () => {
+    it('exposes all metric fields', () => {
+        const stats = {
+            index: '0',
+            uuid: 'u',
+            name: 'n',
+            utilization: 0,
+            utilization_enc: 0,
+            utilization_dec: 0,
+            temperature: 0,
+            fan_speed: 0,
+            memory_available: 0,
+            memory_total: 0,
+            memory_used: 0,
+            memory_free: 0,
+            power_draw: 0,
+            power_limit: 0,
+        };
+        expect(Object.keys(stats)).toHaveLength(14);
+    });
+});
+//# sourceMappingURL=gpu-collector.test.js.map

package/dist/instrumentation/__tests__/gpu-collector.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpu-collector.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/gpu-collector.test.ts"],"names":[],"mappings":";;;;;AAAA,gCAAsE;AACtE,oFAA2D;AAE3D,yCAAyC;AACzC,sEAAsE;AACtE,+EAA+E;AAC/E,MAAM,UAAU,GACd,sIAAsI;IACtI,8HAA8H,CAAC;AAEjI,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;IAC/B,KAAK,EAAE;QACL,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,mBAAmB;QAChC,aAAa,EAAE,oBAAoB;QACnC,aAAa,EAAE,IAAI;QACnB,+BAA+B,EAAE,MAAM;QACvC,eAAe,EAAE,IAAI;QACrB,uBAAuB,EAAE,MAAM,CAAC,EAAE,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;QACxD,4BAA4B,EAAE,MAAM,CAAC,EAAE,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;QAC7D,oCAAoC,EAAE,OAAO;QAC7C,gCAAgC,EAAE,OAAO;KAC1C;IACD,MAAM,EAAE,EAAE,gBAAgB,EAAE,KAAK,EAAE;CACpC,CAAC,CAAC;AAEH,SAAS,YAAY;IACnB,OAAO,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QACzB,IAAI,GAAG,KAAK,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;QACvE,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;YAAE,OAAO,0CAA0C,CAAC;QACxE,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,SAAS;IAChB,OAAO,KAAK,EAAE,GAAG,EAAE,EAAE;QACnB,IAAI,GAAG,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;QACrE,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,WAAW,GAAkB,KAAK,EAAE,GAAG,EAAE,EAAE;IAC/C,MAAM,IAAI,KAAK,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;AAC/C,CAAC,CAAC;AAQF,SAAS,aAAa;IACpB,MAAM,MAAM,GAA2C,EAAE,CAAC;IAC1D,MAAM,YAAY,GAAsB,EAAE,CAAC;IAC3C,IAAI,QAAQ,GAID,IAAI,CAAC;IAEhB,MAAM,KAAK,GAAG;QACZ,qBAAqB,EAAE,CAAC,IAAY,EAAE,EAAE;YACtC,MAAM,KAAK,GAAG,EAAE,IAAI,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAC7B,OAAO,KAAK,CAAC;QACf,CAAC;QACD,0BAA0B,EAAE,CAAC,EAAmB,EAAE,EAAE;YAClD,QAAQ,GAAG,EAAE,CAAC;QAChB,CAAC;KACF,CAAC;IAEF,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;QAC7B,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QACzD,MAAM,QAAQ,CAAC;YACb,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;SACvF,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,EAAE,KAAK,EAAE,KAAY,EAAE,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;AACpE,CAAC;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,MAAM,CAAC,MAAM,IAAI,yBAAmB,CAAC,YAAY,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,CAAC,MAAM,IAAI,yBAAmB,CAAC,SAAS,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,CAAC,MAAM,IAAI,yBAAmB,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;IAChF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,YAAY,EAAE,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC;YACpC,KAAK;YACL,WAAW,EAAE,SAAS;YACtB,eAAe,EAAE,QAAQ;SAC1B,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC,iBAAiB;YACjB,qBAAqB;YACrB,qBAAqB;YACrB,iBAAiB;YACjB,eAAe;YACf,sBAAsB;YACtB,kBAAkB;YAClB,iBAAiB;YACjB,iBAAiB;YACjB,gBAAgB;YAChB,iBAAiB;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QAC1C,MAAM,OAAO,GAAG,MAAM,IAAI,yBAAmB,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC;YAC/D,KAAK;YACL,WAAW,EAAE,SAAS;YACtB,eAAe,EAAE,QAAQ;SAC1B,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,YAAY,EAAE,CAAC,CAAC;QAC1D,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpF,MAAM,WAAW,EAAE,CAAC;QAEpB,sBAAsB;QACtB,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAEtC,MAAM,IAAI,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,6BAAkB,CAAC,SAAS,CAAC,KAAK,GAAG,CAAC,CAAC;QAC5F,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YACjC,oBAAoB,EAAE,SAAS;YAC/B,cAAc,EAAE,QAAQ;YACxB,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,EAAE,SAAS;YAC3D,CAAC,6BAAkB,CAAC,SAAS,CAAC,EAAE,GAAG;YACnC,CAAC,6BAAkB,CAAC,QAAQ,CAAC,EAAE,0CAA0C;YACzE,CAAC,6BAAkB,CAAC,QAAQ,CAAC,EAAE,uBAAuB;SACvD,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,CAAC,MAAc,EAAE,KAAa,EAAE,EAAE,CAChD,YAAY,CAAC,IAAI,CACf,CAAC,CAAC,EAAE,EAAE,CACH,CAAC,CAAC,KAA0B,CAAC,IAAI,KAAK,MAAM;YAC7C,CAAC,CAAC,UAAU,CAAC,6BAAkB,CAAC,SAAS,CAAC,KAAK,KAAK,CACvD,EAAE,KAAK,CAAC;QAEX,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,oEAAoE;QACpE,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,SAAS,EAAE,CAAC,CAAC;QACvD,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,WAAW,EAAE,CAAC;QAEpB,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,6BAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtD,MAAM,CAAC,KAAK,CAAC,6BAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,6BAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAErE,MAAM,OAAO,GAAG,CAAC,MAAc,EAAE,EAAE,CACjC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,KAA0B,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC;QACjF,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACnD,kDAAkD;QAClD,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACxD,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,IAAI,MAAM,GAAG,IAAI,CAAC;QAClB,MAAM,WAAW,GAAkB,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACrD,IAAI,GAAG,KAAK,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACvD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,MAAM;gBAAE,OAAO,oBAAoB,CAAC;YAC5D,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,WAAW,CAAC,CAAC;QACvD,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,GAAG,KAAK,CAAC;QACf,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,aAAa,GAAkB,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACvD,IAAI,GAAG,KAAK,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACvD,IAAI,GAAG,KAAK,UAAU;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACrD,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YAChE,OAAO,WAAW,CAAC;QACrB,CAAC,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,YAAY,GAAkB,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACtD,IAAI,GAAG,KAAK,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACvD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;gBAAE,OAAO,oBAAoB,CAAC;YAClD,OAAO,uGAAuG,CAAC;QACjH,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,WAAW,EAAE,CAAC;QAEpB,MAAM,OAAO,GAAG,CAAC,MAAc,EAAE,EAAE,CACjC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,KAA0B,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC;QAEjF,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1C,0CAA0C;QAC1C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,KAAK,GAAa;YACtB,KAAK,EAAE,GAAG;YACV,IAAI,EAAE,GAAG;YACT,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,CAAC;YACd,eAAe,EAAE,CAAC;YAClB,eAAe,EAAE,CAAC;YAClB,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,CAAC;YACZ,gBAAgB,EAAE,CAAC;YACnB,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;YACd,WAAW,EAAE,CAAC;YACd,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;SACf,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/gradient-trace-comparison.test.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Cross-Language Trace Comparison Tests for the DigitalOcean Gradient Integration
+ *
+ * These verify that the TypeScript Gradient instrumentation emits the same span
+ * attributes / events as the Python SDK reference
+ * (sdk/python/src/openlit/instrumentation/gradient). Gradient is OpenAI-shaped:
+ * responses carry a `model` field, requests support seed / frequency_penalty /
+ * presence_penalty, and streaming usage arrives on the final chunk as `usage`.
+ * The provider name is `digitalocean` and chat is served from inference.do-ai.run.
+ */
+export {};