openlit 1.13.0 → 1.14.0

package/dist/instrumentation/gpu/index.d.ts

@@ -0,0 +1,38 @@
+import type { MeterType } from '../../types';
+export type GpuType = 'nvidia' | 'amd';
+export interface GpuStats {
+    index: string;
+    uuid: string;
+    name: string;
+    utilization: number;
+    utilization_enc: number;
+    utilization_dec: number;
+    temperature: number;
+    fan_speed: number;
+    memory_available: number;
+    memory_total: number;
+    memory_used: number;
+    memory_free: number;
+    power_draw: number;
+    power_limit: number;
+}
+export type CommandRunner = (cmd: string, args: string[]) => Promise<string>;
+export declare class GpuMetricsCollector {
+    private runner;
+    private gpuType;
+    constructor(runner?: CommandRunner);
+    detectGpuType(): Promise<GpuType | null>;
+    collectStats(): Promise<GpuStats[]>;
+    private collectNvidiaStats;
+    private collectAmdStats;
+    /**
+     * Detect a supported GPU and register observable gauges on the meter.
+     * Returns the detected GPU type, or null if no supported GPU was found.
+     */
+    setup(options: {
+        meter: MeterType;
+        environment: string;
+        applicationName: string;
+    }): Promise<GpuType | null>;
+}
+export default GpuMetricsCollector;

package/dist/instrumentation/gpu/index.js

@@ -0,0 +1,233 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GpuMetricsCollector = void 0;
+/**
+ * GPU metrics collection for Node.js services, mirroring the Python SDK's
+ * `gpu` instrumentor (sdk/python/src/openlit/instrumentation/gpu/).
+ *
+ * Instead of NVML/AMD-SMI native bindings, this collector shells out to the
+ * vendor CLI tools that ship with the drivers:
+ *   - NVIDIA: `nvidia-smi --query-gpu=... --format=csv,noheader,nounits`
+ *   - AMD:    `rocm-smi --json` with the relevant --show* flags
+ *
+ * Metric names and attributes match the Python SDK exactly so dashboards
+ * stay shared across languages.
+ */
+const child_process_1 = require("child_process");
+const api_1 = require("@opentelemetry/api");
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const constant_1 = require("../../constant");
+const GPU_METRICS = [
+    { semconv: semantic_convention_1.default.GPU_UTILIZATION, key: 'utilization', description: 'GPU Utilization' },
+    { semconv: semantic_convention_1.default.GPU_UTILIZATION_ENC, key: 'utilization_enc', description: 'GPU Utilization Enc' },
+    { semconv: semantic_convention_1.default.GPU_UTILIZATION_DEC, key: 'utilization_dec', description: 'GPU Utilization Dec' },
+    { semconv: semantic_convention_1.default.GPU_TEMPERATURE, key: 'temperature', description: 'GPU Temperature' },
+    { semconv: semantic_convention_1.default.GPU_FAN_SPEED, key: 'fan_speed', description: 'GPU Fan Speed' },
+    { semconv: semantic_convention_1.default.GPU_MEMORY_AVAILABLE, key: 'memory_available', description: 'GPU Memory Available' },
+    { semconv: semantic_convention_1.default.GPU_MEMORY_TOTAL, key: 'memory_total', description: 'GPU Memory Total' },
+    { semconv: semantic_convention_1.default.GPU_MEMORY_USED, key: 'memory_used', description: 'GPU Memory Used' },
+    { semconv: semantic_convention_1.default.GPU_MEMORY_FREE, key: 'memory_free', description: 'GPU Memory Free' },
+    { semconv: semantic_convention_1.default.GPU_POWER_DRAW, key: 'power_draw', description: 'GPU Power Draw' },
+    { semconv: semantic_convention_1.default.GPU_POWER_LIMIT, key: 'power_limit', description: 'GPU Power Limit' },
+];
+/**
+ * Field order must stay aligned with the CSV column indices in collectNvidiaStats.
+ * `enforced.power.limit` matches Python's nvmlDeviceGetEnforcedPowerLimit;
+ * `power.limit` is a fallback when enforced is N/A.
+ */
+const NVIDIA_SMI_FIELDS = [
+    'index',
+    'uuid',
+    'name',
+    'utilization.gpu',
+    'utilization.encoder',
+    'utilization.decoder',
+    'temperature.gpu',
+    'fan.speed',
+    'memory.total',
+    'memory.used',
+    'memory.free',
+    'power.draw',
+    'enforced.power.limit',
+    'power.limit',
+];
+const defaultRunner = (cmd, args) => new Promise((resolve, reject) => {
+    (0, child_process_1.execFile)(cmd, args, { timeout: 10000, windowsHide: true }, (error, stdout) => {
+        if (error)
+            reject(error);
+        else
+            resolve(stdout);
+    });
+});
+function toNumber(value) {
+    const n = parseFloat(String(value));
+    return Number.isFinite(n) ? n : 0;
+}
+class GpuMetricsCollector {
+    constructor(runner = defaultRunner) {
+        this.gpuType = null;
+        this.runner = runner;
+    }
+    async detectGpuType() {
+        try {
+            await this.runner('nvidia-smi', ['-L']);
+            return 'nvidia';
+        }
+        catch {
+            /* not nvidia */
+        }
+        try {
+            await this.runner('rocm-smi', ['--showid', '--json']);
+            return 'amd';
+        }
+        catch {
+            /* not amd */
+        }
+        return null;
+    }
+    async collectStats() {
+        if (this.gpuType === 'nvidia')
+            return this.collectNvidiaStats();
+        if (this.gpuType === 'amd')
+            return this.collectAmdStats();
+        return [];
+    }
+    async collectNvidiaStats() {
+        const stdout = await this.runner('nvidia-smi', [
+            `--query-gpu=${NVIDIA_SMI_FIELDS.join(',')}`,
+            '--format=csv,noheader,nounits',
+        ]);
+        return stdout
+            .split('\n')
+            .map((line) => line.trim())
+            .filter((line) => line.length > 0)
+            .map((line) => {
+            const parts = line.split(',').map((p) => p.trim());
+            const memoryFree = toNumber(parts[10]);
+            // Prefer enforced power limit (Python NVML parity); fall back to power.limit.
+            const powerLimit = toNumber(parts[12]) || toNumber(parts[13]);
+            return {
+                index: parts[0] ?? '0',
+                uuid: parts[1] ?? '',
+                name: parts[2] ?? '',
+                utilization: toNumber(parts[3]),
+                utilization_enc: toNumber(parts[4]),
+                utilization_dec: toNumber(parts[5]),
+                temperature: toNumber(parts[6]),
+                // Python NVML path always reports 0 for NVIDIA fan_speed (many
+                // datacenter cards have no fan sensor). nvidia-smi exposes the value
+                // when present; [N/A] becomes 0 via toNumber, matching Python.
+                fan_speed: toNumber(parts[7]),
+                memory_total: toNumber(parts[8]),
+                memory_used: toNumber(parts[9]),
+                memory_free: memoryFree,
+                // Python NVIDIA: memory_available == memory_free
+                memory_available: memoryFree,
+                power_draw: toNumber(parts[11]),
+                power_limit: powerLimit,
+            };
+        });
+    }
+    async collectAmdStats() {
+        const stdout = await this.runner('rocm-smi', [
+            '--showid',
+            '--showuniqueid',
+            '--showproductname',
+            '--showuse',
+            '--showmeminfo',
+            'vram',
+            '--showtemp',
+            '--showfan',
+            '--showpower',
+            '--json',
+        ]);
+        let parsed;
+        try {
+            parsed = JSON.parse(stdout);
+        }
+        catch (e) {
+            throw new Error(`rocm-smi returned invalid JSON: ${e}`);
+        }
+        const megaBytes = 1024 * 1024;
+        return Object.entries(parsed)
+            .filter(([card]) => card.toLowerCase().startsWith('card'))
+            .map(([card, data], i) => {
+            const memoryTotal = toNumber(data['VRAM Total Memory (B)']) / megaBytes;
+            const memoryUsed = toNumber(data['VRAM Total Used Memory (B)']) / megaBytes;
+            const memoryFree = memoryTotal - memoryUsed;
+            // Prefer ordinal from "card0" / "card1" (Python uses xgmi index 0..N).
+            const cardIndex = card.replace(/^card/i, '');
+            return {
+                index: String(cardIndex !== '' ? cardIndex : i),
+                uuid: String(data['Unique ID'] ?? ''),
+                name: String(data['Card series'] ?? data['Card SKU'] ?? ''),
+                utilization: toNumber(data['GPU use (%)']),
+                // Python AMD stubs encoder/decoder utilization to 0
+                utilization_enc: 0,
+                utilization_dec: 0,
+                temperature: toNumber(data['Temperature (Sensor edge) (C)']),
+                fan_speed: toNumber(data['Fan speed (%)']),
+                memory_total: memoryTotal,
+                memory_used: memoryUsed,
+                memory_free: memoryFree,
+                // Python AMD sets memory_available to total (not free) — keep parity
+                memory_available: memoryTotal,
+                power_draw: toNumber(data['Average Graphics Package Power (W)'] ??
+                    data['Current Socket Graphics Package Power (W)']),
+                power_limit: toNumber(data['Max Graphics Package Power (W)']),
+            };
+        });
+    }
+    /**
+     * Detect a supported GPU and register observable gauges on the meter.
+     * Returns the detected GPU type, or null if no supported GPU was found.
+     */
+    async setup(options) {
+        this.gpuType = await this.detectGpuType();
+        if (!this.gpuType) {
+            // Match Python openlit.init(): skip quietly-ish when no GPU is present.
+            // Use warn because OpenLIT sets DiagLogLevel.WARN (info is filtered).
+            api_1.diag.warn('OpenLIT: No supported GPUs found; skipping GPU metrics collection. ' +
+                'If this is a non-GPU host, set `collectGpuStats: false` to disable GPU stats.');
+            return null;
+        }
+        const gauges = [];
+        const gaugeByKey = new Map();
+        for (const { semconv, key, description } of GPU_METRICS) {
+            const gauge = options.meter.createObservableGauge(semconv, { description });
+            gauges.push(gauge);
+            gaugeByKey.set(gauge, key);
+        }
+        options.meter.addBatchObservableCallback(async (result) => {
+            let stats;
+            try {
+                stats = await this.collectStats();
+            }
+            catch (e) {
+                api_1.diag.error(`Error in GPU metrics collection: ${e}`);
+                return;
+            }
+            for (const gpu of stats) {
+                const attributes = {
+                    'telemetry.sdk.name': constant_1.SDK_NAME,
+                    'service.name': options.applicationName,
+                    [semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]: options.environment,
+                    [semantic_convention_1.default.GPU_INDEX]: gpu.index,
+                    [semantic_convention_1.default.GPU_UUID]: gpu.uuid,
+                    [semantic_convention_1.default.GPU_NAME]: gpu.name,
+                };
+                for (const gauge of gauges) {
+                    const key = gaugeByKey.get(gauge);
+                    result.observe(gauge, gpu[key], attributes);
+                }
+            }
+        }, gauges);
+        return this.gpuType;
+    }
+}
+exports.GpuMetricsCollector = GpuMetricsCollector;
+exports.default = GpuMetricsCollector;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/gpu/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/gpu/index.ts"],"names":[],"mappings":";;;;;;AAAA;;;;;;;;;;;GAWG;AACH,iDAAyC;AACzC,4CAA0C;AAE1C,oFAA2D;AAC3D,6CAA0C;AAwB1C,MAAM,WAAW,GAAuE;IACtF,EAAE,OAAO,EAAE,6BAAkB,CAAC,eAAe,EAAE,GAAG,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE;IACnG,EAAE,OAAO,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,GAAG,EAAE,iBAAiB,EAAE,WAAW,EAAE,qBAAqB,EAAE;IAC/G,EAAE,OAAO,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,GAAG,EAAE,iBAAiB,EAAE,WAAW,EAAE,qBAAqB,EAAE;IAC/G,EAAE,OAAO,EAAE,6BAAkB,CAAC,eAAe,EAAE,GAAG,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE;IACnG,EAAE,OAAO,EAAE,6BAAkB,CAAC,aAAa,EAAE,GAAG,EAAE,WAAW,EAAE,WAAW,EAAE,eAAe,EAAE;IAC7F,EAAE,OAAO,EAAE,6BAAkB,CAAC,oBAAoB,EAAE,GAAG,EAAE,kBAAkB,EAAE,WAAW,EAAE,sBAAsB,EAAE;IAClH,EAAE,OAAO,EAAE,6BAAkB,CAAC,gBAAgB,EAAE,GAAG,EAAE,cAAc,EAAE,WAAW,EAAE,kBAAkB,EAAE;IACtG,EAAE,OAAO,EAAE,6BAAkB,CAAC,eAAe,EAAE,GAAG,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE;IACnG,EAAE,OAAO,EAAE,6BAAkB,CAAC,eAAe,EAAE,GAAG,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE;IACnG,EAAE,OAAO,EAAE,6BAAkB,CAAC,cAAc,EAAE,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,gBAAgB,EAAE;IAChG,EAAE,OAAO,EAAE,6BAAkB,CAAC,eAAe,EAAE,GAAG,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE;CACpG,CAAC;AAEF;;;;GAIG;AACH,MAAM,iBAAiB,GAAG;IACxB,OAAO;IACP,MAAM;IACN,MAAM;IACN,iBAAiB;IACjB,qBAAqB;IACrB,qBAAqB;IACrB,iBAAiB;IACjB,WAAW;IACX,cAAc;IACd,aAAa;IACb,aAAa;IACb,YAAY;IACZ,sBAAsB;IACtB,aAAa;CACL,CAAC;AAIX,MAAM,aAAa,GAAkB,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CACjD,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;IAC9B,IAAA,wBAAQ,EAAC,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;QAC3E,IAAI,KAAK;YAAE,MAAM,CAAC,KAAK,CAAC,CAAC;;YACpB,OAAO,CAAC,MAAM,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEL,SAAS,QAAQ,CAAC,KAAc;IAC9B,MAAM,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACpC,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACpC,CAAC;AAED,MAAa,mBAAmB;IAI9B,YAAY,SAAwB,aAAa;QAFzC,YAAO,GAAmB,IAAI,CAAC;QAGrC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;YACxC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,MAAM,CAAC;YACP,gBAAgB;QAClB,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC;YACtD,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,MAAM,CAAC;YACP,aAAa;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,IAAI,CAAC,OAAO,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAChE,IAAI,IAAI,CAAC,OAAO,KAAK,KAAK;YAAE,OAAO,IAAI,CAAC,eAAe,EAAE,CAAC;QAC1D,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,KAAK,CAAC,kBAAkB;QAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE;YAC7C,eAAe,iBAAiB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;YAC5C,+BAA+B;SAChC,CAAC,CAAC;QAEH,OAAO,MAAM;aACV,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;aAC1B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;aACjC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACZ,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACnD,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YACvC,8EAA8E;YAC9E,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YAC9D,OAAO;gBACL,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG;gBACtB,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;gBACpB,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;gBACpB,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC/B,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACnC,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACnC,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC/B,+DAA+D;gBAC/D,qEAAqE;gBACrE,+DAA+D;gBAC/D,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC7B,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAChC,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC/B,WAAW,EAAE,UAAU;gBACvB,iDAAiD;gBACjD,gBAAgB,EAAE,UAAU;gBAC5B,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC/B,WAAW,EAAE,UAAU;aACxB,CAAC;QACJ,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE;YAC3C,UAAU;YACV,gBAAgB;YAChB,mBAAmB;YACnB,WAAW;YACX,eAAe;YACf,MAAM;YACN,YAAY;YACZ,WAAW;YACX,aAAa;YACb,QAAQ;SACT,CAAC,CAAC;QAEH,IAAI,MAA+C,CAAC;QACpD,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAA4C,CAAC;QACzE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,GAAG,IAAI,CAAC;QAE9B,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;aAC1B,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;aACzD,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;YACvB,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,GAAG,SAAS,CAAC;YACxE,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC,GAAG,SAAS,CAAC;YAC5E,MAAM,UAAU,GAAG,WAAW,GAAG,UAAU,CAAC;YAC5C,uEAAuE;YACvE,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAC7C,OAAO;gBACL,KAAK,EAAE,MAAM,CAAC,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/C,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;gBACrC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gBAC3D,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAC1C,oDAAoD;gBACpD,eAAe,EAAE,CAAC;gBAClB,eAAe,EAAE,CAAC;gBAClB,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;gBAC5D,SAAS,EAAE,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;gBAC1C,YAAY,EAAE,WAAW;gBACzB,WAAW,EAAE,UAAU;gBACvB,WAAW,EAAE,UAAU;gBACvB,qEAAqE;gBACrE,gBAAgB,EAAE,WAAW;gBAC7B,UAAU,EAAE,QAAQ,CAClB,IAAI,CAAC,oCAAoC,CAAC;oBACxC,IAAI,CAAC,2CAA2C,CAAC,CACpD;gBACD,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;aAC9D,CAAC;QACJ,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK,CAAC,OAIX;QACC,IAAI,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC1C,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,wEAAwE;YACxE,sEAAsE;YACtE,UAAI,CAAC,IAAI,CACP,qEAAqE;gBACnE,+EAA+E,CAClF,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAsB,EAAE,CAAC;QACrC,MAAM,UAAU,GAAG,IAAI,GAAG,EAAiC,CAAC;QAC5D,KAAK,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,IAAI,WAAW,EAAE,CAAC;YACxD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnB,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,KAAK,EAAE,MAA6B,EAAE,EAAE;YAC/E,IAAI,KAAiB,CAAC;YACtB,IAAI,CAAC;gBACH,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YACpC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,UAAI,CAAC,KAAK,CAAC,oCAAoC,CAAC,EAAE,CAAC,CAAC;gBACpD,OAAO;YACT,CAAC;YAED,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;gBACxB,MAAM,UAAU,GAAG;oBACjB,oBAAoB,EAAE,mBAAQ;oBAC9B,cAAc,EAAE,OAAO,CAAC,eAAe;oBACvC,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,EAAE,OAAO,CAAC,WAAW;oBACrE,CAAC,6BAAkB,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,KAAK;oBACzC,CAAC,6BAAkB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,IAAI;oBACvC,CAAC,6BAAkB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,IAAI;iBACxC,CAAC;gBACF,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,CAAiB,CAAC;oBAClD,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,UAAU,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;QACH,CAAC,EAAE,MAAM,CAAC,CAAC;QAEX,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;CACF;AAlLD,kDAkLC;AAED,kBAAe,mBAAmB,CAAC"}

package/dist/instrumentation/gradient/index.d.ts

@@ -0,0 +1,12 @@
+import { InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+import { InstrumentationConfig } from '@opentelemetry/instrumentation';
+export interface GradientInstrumentationConfig extends InstrumentationConfig {
+}
+export default class OpenlitGradientInstrumentation extends InstrumentationBase {
+    constructor(config?: GradientInstrumentationConfig);
+    private safeWrap;
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    manualPatch(gradient: any): void;
+    protected _patch(moduleExports: any): void;
+    protected _unpatch(moduleExports: any): void;
+}

package/dist/instrumentation/gradient/index.js

@@ -0,0 +1,74 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+class OpenlitGradientInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-gradient`, '1.0.0', config);
+    }
+    safeWrap(target, method, patcher) {
+        if (!target?.[method])
+            return;
+        if ((0, instrumentation_1.isWrapped)(target[method])) {
+            this._unwrap(target, method);
+        }
+        this._wrap(target, method, patcher(this.tracer));
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@digitalocean/gradient', 
+        // @digitalocean/gradient currently ships only as a prerelease (0.1.0-alpha.*).
+        // A plain '>=0.1.0' range does NOT match prereleases under semver, so the hook
+        // would load the module but silently skip patching. Anchor at the prerelease
+        // tag (mirrors azure-ai-inference's '>=1.0.0-beta.1'); this still matches the
+        // eventual stable 0.1.0 and later. See semver.satisfies prerelease semantics.
+        ['>=0.1.0-alpha.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(gradient) {
+        this._patch(gradient);
+    }
+    _patch(moduleExports) {
+        try {
+            const Gradient = moduleExports?.Gradient;
+            if (!Gradient)
+                return;
+            // Chat completions (inference)
+            this.safeWrap(Gradient.Chat?.Completions?.prototype, 'create', wrapper_1.default._patchChatCompletionCreate);
+            // Agent chat completions
+            this.safeWrap(Gradient.Agents?.Chat?.Completions?.prototype, 'create', wrapper_1.default._patchAgentChatCompletionCreate);
+            // Image generation
+            this.safeWrap(Gradient.Images?.prototype, 'generate', wrapper_1.default._patchImageGenerate);
+        }
+        catch (e) {
+            console.error('Error in _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        const Gradient = moduleExports?.Gradient;
+        if (!Gradient)
+            return;
+        const targets = [
+            [Gradient.Chat?.Completions?.prototype, 'create'],
+            [Gradient.Agents?.Chat?.Completions?.prototype, 'create'],
+            [Gradient.Images?.prototype, 'generate'],
+        ];
+        for (const [target, method] of targets) {
+            if (target?.[method]) {
+                this._unwrap(target, method);
+            }
+        }
+    }
+}
+exports.default = OpenlitGradientInstrumentation;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/gradient/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/gradient/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAwC;AAIxC,MAAqB,8BAA+B,SAAQ,qCAAmB;IAC7E,YAAY,SAAwC,EAAE;QACpD,KAAK,CAAC,GAAG,iCAAsB,2BAA2B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAEO,QAAQ,CAAC,MAAW,EAAE,MAAc,EAAE,OAA6B;QACzE,IAAI,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC;YAAE,OAAO;QAC9B,IAAI,IAAA,2BAAS,EAAC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IACnD,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,wBAAwB;QACxB,+EAA+E;QAC/E,+EAA+E;QAC/E,6EAA6E;QAC7E,8EAA8E;QAC9E,8EAA8E;QAC9E,CAAC,iBAAiB,CAAC,EACnB,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,QAAa;QAC9B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,aAAa,EAAE,QAAQ,CAAC;YACzC,IAAI,CAAC,QAAQ;gBAAE,OAAO;YAEtB,+BAA+B;YAC/B,IAAI,CAAC,QAAQ,CACX,QAAQ,CAAC,IAAI,EAAE,WAAW,EAAE,SAAS,EACrC,QAAQ,EACR,iBAAe,CAAC,0BAA0B,CAC3C,CAAC;YAEF,yBAAyB;YACzB,IAAI,CAAC,QAAQ,CACX,QAAQ,CAAC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAC7C,QAAQ,EACR,iBAAe,CAAC,+BAA+B,CAChD,CAAC;YAEF,mBAAmB;YACnB,IAAI,CAAC,QAAQ,CACX,QAAQ,CAAC,MAAM,EAAE,SAAS,EAC1B,UAAU,EACV,iBAAe,CAAC,mBAAmB,CACpC,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,MAAM,QAAQ,GAAG,aAAa,EAAE,QAAQ,CAAC;QACzC,IAAI,CAAC,QAAQ;YAAE,OAAO;QAEtB,MAAM,OAAO,GAAyB;YACpC,CAAC,QAAQ,CAAC,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,CAAC;YACjD,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,CAAC;YACzD,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC;SACzC,CAAC;QAEF,KAAK,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACvC,IAAI,MAAM,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;gBACrB,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;IACH,CAAC;CACF;AArFD,iDAqFC"}

package/dist/instrumentation/gradient/utils.d.ts

@@ -0,0 +1,14 @@
+import { Attributes } from '@opentelemetry/api';
+export type GradientEndpointKind = 'inference' | 'agent';
+/**
+ * Walk the Gradient APIResource `_client` chain and read the per-route endpoint
+ * from the root client (mirrors Python `_resolve_endpoint`).
+ */
+export declare function resolveGradientEndpoint(instance: any, kind: GradientEndpointKind): [string, number];
+/** Extract agent UUID from `{uuid}.agents.do-ai.run` hostnames (Python parity). */
+export declare function agentIdFromHost(host: string): string | undefined;
+export declare function gradientSpanCreationAttrs(operationName: string, requestModel: string, serverAddress: string, serverPort: number): Attributes;
+/** Normalize OpenAI-style stop sequences (Python `_normalize_stop`). */
+export declare function normalizeStopSequences(stop: unknown): string[] | undefined;
+/** Request attributes shared by chat and agent-chat surfaces. */
+export declare function applyGradientChatRequestAttributes(span: any, body: Record<string, any>): void;

package/dist/instrumentation/gradient/utils.js

@@ -0,0 +1,119 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveGradientEndpoint = resolveGradientEndpoint;
+exports.agentIdFromHost = agentIdFromHost;
+exports.gradientSpanCreationAttrs = gradientSpanCreationAttrs;
+exports.normalizeStopSequences = normalizeStopSequences;
+exports.applyGradientChatRequestAttributes = applyGradientChatRequestAttributes;
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const DEFAULT_HOSTS = {
+    inference: ['inference.do-ai.run', 443],
+    agent: ['agents.do-ai.run', 443],
+};
+const ENDPOINT_ATTRS = {
+    inference: 'inferenceEndpoint',
+    agent: 'agentEndpoint',
+};
+/**
+ * Walk the Gradient APIResource `_client` chain and read the per-route endpoint
+ * from the root client (mirrors Python `_resolve_endpoint`).
+ */
+function resolveGradientEndpoint(instance, kind) {
+    const [defaultHost, defaultPort] = DEFAULT_HOSTS[kind];
+    const attr = ENDPOINT_ATTRS[kind];
+    const candidates = [];
+    const seen = new Set();
+    let current = instance;
+    while (current && !seen.has(current)) {
+        seen.add(current);
+        candidates.push(current);
+        current = current._client;
+    }
+    for (const cand of candidates) {
+        const raw = cand?.[attr];
+        if (!raw)
+            continue;
+        try {
+            const url = new URL(String(raw).includes('://') ? String(raw) : `https://${raw}`);
+            const host = url.hostname || defaultHost;
+            const port = url.port
+                ? Number(url.port)
+                : url.protocol === 'http:'
+                    ? 80
+                    : 443;
+            return [host, port];
+        }
+        catch {
+            const host = String(raw).split(':')[0] || defaultHost;
+            return [host, defaultPort];
+        }
+    }
+    return DEFAULT_HOSTS[kind];
+}
+/** Extract agent UUID from `{uuid}.agents.do-ai.run` hostnames (Python parity). */
+function agentIdFromHost(host) {
+    if (!host || !host.includes('agents.do-ai.run'))
+        return undefined;
+    const head = host.split('.agents.do-ai.run')[0];
+    return head || undefined;
+}
+function gradientSpanCreationAttrs(operationName, requestModel, serverAddress, serverPort) {
+    return {
+        [semantic_convention_1.default.GEN_AI_OPERATION]: operationName,
+        [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_DIGITALOCEAN,
+        [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+        [semantic_convention_1.default.SERVER_ADDRESS]: serverAddress,
+        [semantic_convention_1.default.SERVER_PORT]: serverPort,
+    };
+}
+/** Normalize OpenAI-style stop sequences (Python `_normalize_stop`). */
+function normalizeStopSequences(stop) {
+    if (stop == null)
+        return undefined;
+    if (Array.isArray(stop)) {
+        const out = stop.filter((s) => s != null && String(s).length > 0).map(String);
+        return out.length > 0 ? out : undefined;
+    }
+    const s = String(stop);
+    return s ? [s] : undefined;
+}
+/** Request attributes shared by chat and agent-chat surfaces. */
+function applyGradientChatRequestAttributes(span, body) {
+    if (body.temperature != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, body.temperature);
+    }
+    if (body.top_p != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, body.top_p);
+    }
+    const maxTokens = body.max_completion_tokens ?? body.max_tokens;
+    if (maxTokens != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, maxTokens);
+    }
+    if (body.frequency_penalty != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, body.frequency_penalty);
+    }
+    if (body.presence_penalty != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY, body.presence_penalty);
+    }
+    const stopSequences = normalizeStopSequences(body.stop ?? body.stop_sequences);
+    if (stopSequences) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, stopSequences);
+    }
+    if (body.seed != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_SEED, Number(body.seed));
+    }
+    if (body.n != null && body.n !== 1) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_CHOICE_COUNT, body.n);
+    }
+    if (body.user) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_USER, body.user);
+    }
+    if (body.reasoning_effort) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_REASONING_EFFORT, body.reasoning_effort);
+    }
+    span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, Boolean(body.stream));
+}
+//# sourceMappingURL=utils.js.map

package/dist/instrumentation/gradient/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/instrumentation/gradient/utils.ts"],"names":[],"mappings":";;;;;AAmBA,0DAkCC;AAGD,0CAIC;AAED,8DAaC;AAGD,wDAQC;AAGD,gFAkCC;AA1HD,oFAA2D;AAI3D,MAAM,aAAa,GAAmD;IACpE,SAAS,EAAE,CAAC,qBAAqB,EAAE,GAAG,CAAC;IACvC,KAAK,EAAE,CAAC,kBAAkB,EAAE,GAAG,CAAC;CACjC,CAAC;AAEF,MAAM,cAAc,GAAyC;IAC3D,SAAS,EAAE,mBAAmB;IAC9B,KAAK,EAAE,eAAe;CACvB,CAAC;AAEF;;;GAGG;AACH,SAAgB,uBAAuB,CACrC,QAAa,EACb,IAA0B;IAE1B,MAAM,CAAC,WAAW,EAAE,WAAW,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,UAAU,GAAU,EAAE,CAAC;IAC7B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAO,CAAC;IAC5B,IAAI,OAAO,GAAG,QAAQ,CAAC;IACvB,OAAO,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QACrC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAClB,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzB,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAC5B,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC;QACzB,IAAI,CAAC,GAAG;YAAE,SAAS;QACnB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,GAAG,EAAE,CAAC,CAAC;YAClF,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,IAAI,WAAW,CAAC;YACzC,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI;gBACnB,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;gBAClB,CAAC,CAAC,GAAG,CAAC,QAAQ,KAAK,OAAO;oBACxB,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,GAAG,CAAC;YACV,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC;YACtD,OAAO,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;AAC7B,CAAC;AAED,mFAAmF;AACnF,SAAgB,eAAe,CAAC,IAAY;IAC1C,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QAAE,OAAO,SAAS,CAAC;IAClE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,OAAO,IAAI,IAAI,SAAS,CAAC;AAC3B,CAAC;AAED,SAAgB,yBAAyB,CACvC,aAAqB,EACrB,YAAoB,EACpB,aAAqB,EACrB,UAAkB;IAElB,OAAO;QACL,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;QACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;QAC7F,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;QACvD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,aAAa;QAClD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,UAAU;KAC7C,CAAC;AACJ,CAAC;AAED,wEAAwE;AACxE,SAAgB,sBAAsB,CAAC,IAAa;IAClD,IAAI,IAAI,IAAI,IAAI;QAAE,OAAO,SAAS,CAAC;IACnC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9E,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1C,CAAC;IACD,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IACvB,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7B,CAAC;AAED,iEAAiE;AACjE,SAAgB,kCAAkC,CAAC,IAAS,EAAE,IAAyB;IACrF,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IACrF,CAAC;IACD,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IACzE,CAAC;IACD,MAAM,SAAS,GAAG,IAAI,CAAC,qBAAqB,IAAI,IAAI,CAAC,UAAU,CAAC;IAChE,IAAI,SAAS,IAAI,IAAI,EAAE,CAAC;QACtB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,SAAS,CAAC,CAAC;IAC7E,CAAC;IACD,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACjG,CAAC;IACD,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC/F,CAAC;IACD,MAAM,aAAa,GAAG,sBAAsB,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,CAAC;IAC/E,IAAI,aAAa,EAAE,CAAC;QAClB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,aAAa,CAAC,CAAC;IACrF,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;QACtB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/E,CAAC;IACD,IAAI,IAAI,CAAC,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IAC5E,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IACvE,CAAC;IACD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC/F,CAAC;IACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;AACvF,CAAC"}

package/dist/instrumentation/gradient/wrapper.d.ts

@@ -0,0 +1,76 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+import { GradientEndpointKind } from './utils';
+type ChatPatchOptions = {
+    operationName: string;
+    endpointKind: GradientEndpointKind;
+    genAIEndpoint: string;
+    apiType: string;
+    isAgent?: boolean;
+};
+declare class GradientWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static _patchChatCompletionCreate(tracer: Tracer): any;
+    static _patchAgentChatCompletionCreate(tracer: Tracer): any;
+    static _buildChatPatch(tracer: Tracer, options: ChatPatchOptions): any;
+    static _patchImageGenerate(tracer: Tracer): any;
+    static _chatCompletion({ args, genAIEndpoint, response, span, serverAddress, serverPort, operationName, apiType, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+        serverAddress: string;
+        serverPort: number;
+        operationName: string;
+        apiType: string;
+    }): Promise<any>;
+    static _chatCompletionGenerator({ args, genAIEndpoint, response, span, serverAddress, serverPort, operationName, apiType, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+        serverAddress: string;
+        serverPort: number;
+        operationName: string;
+        apiType: string;
+    }): AsyncGenerator<unknown, any, unknown>;
+    static _imageGenerateCommonSetter({ args, genAIEndpoint, response, span, serverAddress, serverPort, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+        serverAddress: string;
+        serverPort: number;
+    }): {
+        genAIEndpoint: string;
+        model: any;
+        user: any;
+        cost: number;
+        aiSystem: string;
+        serverAddress: string;
+        serverPort: number;
+    };
+    static _chatCompletionCommonSetter({ args, genAIEndpoint, result, span, ttft, tbt, serverAddress, serverPort, operationName, apiType, reasoningText, }: {
+        args: any[];
+        genAIEndpoint: string;
+        result: any;
+        span: Span;
+        ttft?: number;
+        tbt?: number;
+        serverAddress: string;
+        serverPort: number;
+        operationName?: string;
+        apiType?: string;
+        reasoningText?: string;
+    }): Promise<{
+        genAIEndpoint: string;
+        model: any;
+        user: any;
+        cost: number;
+        aiSystem: string;
+        serverAddress: string;
+        serverPort: number;
+    }>;
+    static _buildOutputMessages(text: string, finishReason: string, toolCalls?: any[], reasoning?: string): string;
+}
+export default GradientWrapper;