openlit 1.12.0 → 1.14.0

package/dist/instrumentation/__tests__/pg-trace-comparison.test.js

@@ -0,0 +1,455 @@
+"use strict";
+/**
+ * Cross-Language Trace Comparison Tests for PostgreSQL (`pg`) Integration
+ *
+ * Verifies that the TypeScript SDK generates traces consistent with the Python SDK
+ * psycopg instrumentation for PostgreSQL queries.
+ *
+ * Python SDK reference: sdk/python/src/openlit/instrumentation/psycopg/utils.py
+ *
+ * Key alignment:
+ *   - db.system.name = 'postgresql'
+ *   - db.operation.name parsed from the SQL verb (SELECT/INSERT/UPDATE/DELETE/...)
+ *   - db.collection.name = target table
+ *   - db.namespace = database name
+ *   - server.address, server.port (default localhost:5432)
+ *   - db.query.text (when captureMessageContent is enabled)
+ *   - db.query.parameter.<index> (only when captureDbParameters is enabled)
+ *   - db.search.similarity_metric for pgvector operators (<=>, <->, <#>)
+ *   - db.query.summary
+ *   - db.response.returned_rows from result.rowCount
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const wrapper_1 = __importDefault(require("../pg/wrapper"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importStar(require("../../helpers"));
+const metrics_1 = __importDefault(require("../../otel/metrics"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const api_1 = require("@opentelemetry/api");
+const core_1 = require("@opentelemetry/core");
+jest.mock('../../config');
+jest.mock('../../helpers', () => ({
+    __esModule: true,
+    default: { handleException: jest.fn() },
+    applyCustomSpanAttributes: jest.fn(),
+}));
+jest.mock('../../otel/metrics', () => ({
+    __esModule: true,
+    default: {
+        dbRequests: { add: jest.fn() },
+        dbClientOperationDuration: { record: jest.fn() },
+    },
+}));
+jest.mock('@opentelemetry/core', () => ({
+    isTracingSuppressed: jest.fn().mockReturnValue(false),
+}));
+describe('PostgreSQL (pg) Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            addEvent: jest.fn(),
+            end: jest.fn(),
+            setStatus: jest.fn(),
+        };
+        mockTracer = {
+            startSpan: jest.fn().mockReturnValue(mockSpan),
+        };
+        config_1.default.environment = 'openlit-testing';
+        config_1.default.applicationName = 'openlit-test';
+        config_1.default.captureMessageContent = true;
+        config_1.default.maxContentLength = null;
+        config_1.default.disableMetrics = false;
+        helpers_1.default.handleException = jest.fn();
+        core_1.isTracingSuppressed.mockReturnValue(false);
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    // Fake `pg` Client instance carrying connection parameters.
+    const fakeClient = () => ({
+        connectionParameters: { host: 'db.example.com', port: 5433, database: 'app_db' },
+    });
+    // Invoke the patched query() in promise form and await completion.
+    async function invokePromise(args, result = { rowCount: 0 }, config = {}) {
+        const patchFn = wrapper_1.default._patchQuery(mockTracer, config);
+        const original = jest.fn().mockResolvedValue(result);
+        const wrapped = patchFn(original);
+        await wrapped.apply(fakeClient(), args);
+        return original;
+    }
+    // ── Pure-function parity checks (mirror Python utils) ───────────────────────
+    describe('SQL operation parsing (parseSqlOperation)', () => {
+        it('detects SELECT', () => {
+            expect(wrapper_1.default.parseSqlOperation('SELECT * FROM users')).toBe(semantic_convention_1.default.DB_OPERATION_SELECT);
+        });
+        it('detects INSERT', () => {
+            expect(wrapper_1.default.parseSqlOperation('INSERT INTO users (id) VALUES (1)')).toBe(semantic_convention_1.default.DB_OPERATION_INSERT);
+        });
+        it('falls back to QUERY for unknown statements', () => {
+            expect(wrapper_1.default.parseSqlOperation('EXPLAIN ANALYZE foo')).toBe(semantic_convention_1.default.DB_OPERATION_QUERY);
+        });
+        it('resolves CTE (WITH ... SELECT) to the inner operation', () => {
+            expect(wrapper_1.default.parseSqlOperation('WITH t AS (...) SELECT * FROM t')).toBe(semantic_convention_1.default.DB_OPERATION_SELECT);
+        });
+    });
+    describe('table extraction (extractTableName)', () => {
+        it('extracts table from SELECT ... FROM', () => {
+            expect(wrapper_1.default.extractTableName('SELECT * FROM items WHERE id = $1')).toBe('items');
+        });
+        it('extracts table from INSERT INTO', () => {
+            expect(wrapper_1.default.extractTableName('INSERT INTO documents (x) VALUES ($1)')).toBe('documents');
+        });
+        it('returns unknown when no table matches', () => {
+            expect(wrapper_1.default.extractTableName('COMMIT')).toBe('unknown');
+        });
+    });
+    describe('pgvector similarity detection (detectSimilarityMetric)', () => {
+        it('maps <=> to cosine', () => {
+            expect(wrapper_1.default.detectSimilarityMetric('... ORDER BY embedding <=> $1')).toBe('cosine');
+        });
+        it('maps <-> to l2', () => {
+            expect(wrapper_1.default.detectSimilarityMetric('... ORDER BY embedding <-> $1')).toBe('l2');
+        });
+        it('maps <#> to inner_product', () => {
+            expect(wrapper_1.default.detectSimilarityMetric('... ORDER BY embedding <#> $1')).toBe('inner_product');
+        });
+        it('returns undefined when no operator present', () => {
+            expect(wrapper_1.default.detectSimilarityMetric('SELECT 1')).toBeUndefined();
+        });
+    });
+    // ── Common DB span attributes ───────────────────────────────────────────────
+    it('sets db.system.name = "postgresql"', async () => {
+        await invokePromise(['SELECT * FROM users']);
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_SYSTEM_NAME, 'postgresql');
+    });
+    it('sets server.address and server.port from the client connection params', async () => {
+        await invokePromise(['SELECT 1']);
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'db.example.com');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 5433);
+    });
+    it('defaults to localhost:5432 when connection params are missing (Python defaults)', async () => {
+        const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+        const original = jest.fn().mockResolvedValue({ rowCount: 0 });
+        const wrapped = patchFn(original);
+        await wrapped.apply({}, ['SELECT 1']);
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'localhost');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 5432);
+    });
+    it('sets db.namespace from the database name', async () => {
+        await invokePromise(['SELECT 1']);
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_NAMESPACE_POSTGRESQL, 'app_db');
+    });
+    it('sets deployment.environment and service.name (not gen_ai.* env attrs)', async () => {
+        await invokePromise(['SELECT 1']);
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT, 'openlit-testing');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_conventions_1.ATTR_SERVICE_NAME, 'openlit-test');
+        expect(mockSpan.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_ENVIRONMENT, expect.anything());
+        expect(mockSpan.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_APPLICATION_NAME, expect.anything());
+    });
+    it('sets telemetry.sdk.name, gen_ai.operation.name=vectordb, and custom span attrs', async () => {
+        await invokePromise(['SELECT 1']);
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME, 'openlit');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_VECTORDB);
+        expect(helpers_1.applyCustomSpanAttributes).toHaveBeenCalledWith(mockSpan);
+    });
+    it('records db metrics when disableMetrics is false', async () => {
+        await invokePromise(['SELECT * FROM users'], { rowCount: 1 });
+        expect(metrics_1.default.dbRequests.add).toHaveBeenCalledWith(1, expect.objectContaining({
+            [semantic_convention_1.default.DB_SYSTEM_NAME]: 'postgresql',
+            [semantic_convention_1.default.DB_OPERATION_NAME]: 'SELECT',
+            [semantic_convention_1.default.SERVER_ADDRESS]: 'db.example.com',
+            [semantic_convention_1.default.SERVER_PORT]: 5433,
+            [semantic_conventions_1.ATTR_SERVICE_NAME]: 'openlit-test',
+            [semantic_conventions_1.SEMRESATTRS_DEPLOYMENT_ENVIRONMENT]: 'openlit-testing',
+        }));
+        expect(metrics_1.default.dbClientOperationDuration.record).toHaveBeenCalledWith(expect.any(Number), expect.objectContaining({
+            [semantic_convention_1.default.DB_OPERATION_NAME]: 'SELECT',
+        }));
+    });
+    it('skips db metrics when disableMetrics is true', async () => {
+        config_1.default.disableMetrics = true;
+        await invokePromise(['SELECT 1']);
+        expect(metrics_1.default.dbRequests.add).not.toHaveBeenCalled();
+        expect(metrics_1.default.dbClientOperationDuration.record).not.toHaveBeenCalled();
+    });
+    it('bypasses instrumentation when tracing is suppressed', async () => {
+        core_1.isTracingSuppressed.mockReturnValue(true);
+        const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+        const original = jest.fn().mockResolvedValue({ rowCount: 0 });
+        const wrapped = patchFn(original);
+        await wrapped.apply(fakeClient(), ['SELECT 1']);
+        expect(original).toHaveBeenCalled();
+        expect(mockTracer.startSpan).not.toHaveBeenCalled();
+    });
+    // ── Basic query span ─────────────────────────────────────────────────────────
+    describe('basic query span', () => {
+        it('sets db.operation.name and db.collection.name', async () => {
+            await invokePromise(['SELECT id, name FROM users WHERE id = $1', [42]]);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'SELECT');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_COLLECTION_NAME, 'users');
+        });
+        it('names the span "<operation> <table>"', async () => {
+            await invokePromise(['SELECT * FROM users']);
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('SELECT users', expect.anything());
+        });
+        it('captures query text when captureMessageContent is enabled', async () => {
+            await invokePromise(['SELECT * FROM users']);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_TEXT, 'SELECT * FROM users');
+        });
+        it('sets db.response.returned_rows from result.rowCount', async () => {
+            await invokePromise(['SELECT * FROM users'], { rowCount: 7 });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 7);
+        });
+        it('sets db.query.summary', async () => {
+            await invokePromise(['SELECT * FROM users']);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, 'SELECT users');
+        });
+        it('ends the span and marks status OK', async () => {
+            await invokePromise(['SELECT 1']);
+            expect(mockSpan.setStatus).toHaveBeenCalledWith({ code: 1 });
+            expect(mockSpan.end).toHaveBeenCalled();
+        });
+    });
+    // ── pgvector query ───────────────────────────────────────────────────────────
+    describe('pgvector similarity query', () => {
+        it('sets db.search.similarity_metric = cosine for the <=> operator', async () => {
+            await invokePromise([
+                'SELECT id FROM embeddings ORDER BY vec <=> $1 LIMIT 5',
+                ['[0.1,0.2]'],
+            ]);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_SEARCH_SIMILARITY_METRIC, 'cosine');
+        });
+        it('includes the vector metric in the query summary', async () => {
+            await invokePromise(['SELECT id FROM embeddings ORDER BY vec <-> $1']);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, 'SELECT embeddings (vector l2)');
+        });
+    });
+    // ── Full-text search ─────────────────────────────────────────────────────────
+    describe('full-text search query', () => {
+        it('appends (full-text) to the query summary for TSVECTOR queries', async () => {
+            await invokePromise(['SELECT id FROM docs WHERE body @@ to_tsquery($1)', ['search']]);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, 'SELECT docs (full-text)');
+        });
+        it('detects full-text search via detectFullTextSearch()', () => {
+            expect(wrapper_1.default.detectFullTextSearch('SELECT * FROM docs WHERE col @@ TSQUERY')).toBe(true);
+            expect(wrapper_1.default.detectFullTextSearch('SELECT rank FROM ts_rank(col, q)')).toBe(true);
+            expect(wrapper_1.default.detectFullTextSearch('SELECT * FROM users')).toBe(false);
+        });
+    });
+    // ── Batch queries ────────────────────────────────────────────────────────────
+    describe('batch parameter arrays', () => {
+        it('names the span with (batch) and sets db.batch.size', async () => {
+            await invokePromise(['INSERT INTO users (name) VALUES ($1)', [['alice'], ['bob']]], { rowCount: 2 }, { captureDbParameters: true });
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('INSERT users (batch)', { kind: api_1.SpanKind.CLIENT });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_BATCH_SIZE, 2);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.0`, 'alice');
+        });
+    });
+    // ── captureDbParameters toggle ────────────────────────────────────────────────
+    describe('captureDbParameters toggle', () => {
+        it('does NOT capture query parameters when disabled (default)', async () => {
+            await invokePromise(['SELECT * FROM users WHERE id = $1', [42]], { rowCount: 1 }, {
+                captureDbParameters: false,
+            });
+            const paramCalls = mockSpan.setAttribute.mock.calls.filter((c) => String(c[0]).startsWith(semantic_convention_1.default.DB_QUERY_PARAMETER));
+            expect(paramCalls).toHaveLength(0);
+        });
+        it('captures query parameters per-index when enabled', async () => {
+            await invokePromise(['SELECT * FROM users WHERE id = $1 AND status = $2', [42, 'active']], { rowCount: 1 }, { captureDbParameters: true });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.0`, '42');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.1`, 'active');
+        });
+        it('sanitizes Buffer parameters to <bytes:N> when enabled', async () => {
+            await invokePromise(['INSERT INTO files (data) VALUES ($1)', [Buffer.from('abc')]], { rowCount: 1 }, { captureDbParameters: true });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.0`, '<bytes:3>');
+        });
+    });
+    // ── Query-config object form ──────────────────────────────────────────────────
+    it('supports the { text, values } query-config object form', async () => {
+        await invokePromise([{ text: 'UPDATE accounts SET balance = $1 WHERE id = $2', values: [100, 5] }], { rowCount: 1 }, { captureDbParameters: true });
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'UPDATE');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_COLLECTION_NAME, 'accounts');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.0`, '100');
+    });
+    // ── Callback form ─────────────────────────────────────────────────────────────
+    describe('callback form', () => {
+        it('finalizes the span and invokes the user callback on success', (done) => {
+            const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+            const original = jest.fn((_text, _vals, cb) => cb(null, { rowCount: 3 }));
+            const wrapped = patchFn(original);
+            wrapped.apply(fakeClient(), [
+                'SELECT * FROM logs',
+                [],
+                (err, res) => {
+                    expect(err).toBeNull();
+                    expect(res.rowCount).toBe(3);
+                    expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 3);
+                    expect(mockSpan.end).toHaveBeenCalled();
+                    done();
+                },
+            ]);
+        });
+        it('records an exception and still invokes the callback on error', (done) => {
+            const error = new Error('connection refused');
+            const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+            const original = jest.fn((_text, cb) => cb(error, undefined));
+            const wrapped = patchFn(original);
+            wrapped.apply(fakeClient(), [
+                'SELECT 1',
+                (err) => {
+                    expect(err).toBe(error);
+                    expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, error);
+                    expect(mockSpan.end).toHaveBeenCalled();
+                    done();
+                },
+            ]);
+        });
+    });
+    // ── Error handling (promise form) ───────────────────────────────────────────────
+    it('records an exception when the promise rejects', async () => {
+        const error = new Error('syntax error');
+        const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+        const original = jest.fn().mockRejectedValue(error);
+        const wrapped = patchFn(original);
+        await expect(wrapped.apply(fakeClient(), ['SELET 1'])).rejects.toThrow('syntax error');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, error);
+        expect(mockSpan.end).toHaveBeenCalled();
+    });
+    // ── Submittable / Cursor form (result.on('end' | 'error')) ──────────────────────
+    describe('submittable / cursor form', () => {
+        // Minimal EventEmitter-like fake submittable (neither a promise nor a callback).
+        const makeSubmittable = (extra = {}) => {
+            const listeners = {};
+            return {
+                ...extra,
+                on(event, cb) {
+                    (listeners[event] || (listeners[event] = [])).push(cb);
+                    return this;
+                },
+                emit(event, ...args) {
+                    (listeners[event] || []).forEach((cb) => cb(...args));
+                },
+            };
+        };
+        it('finalizes the span when the submittable emits "end"', () => {
+            const submittable = makeSubmittable({ rowCount: 4 });
+            const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+            const original = jest.fn().mockReturnValue(submittable);
+            const wrapped = patchFn(original);
+            const returned = wrapped.apply(fakeClient(), ['SELECT * FROM events']);
+            expect(returned).toBe(submittable); // submittable passed through unchanged
+            expect(mockSpan.end).not.toHaveBeenCalled(); // not finalized until completion
+            returned.emit('end');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_SYSTEM_NAME, 'postgresql');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 4);
+            expect(mockSpan.setStatus).toHaveBeenCalledWith({ code: 1 }); // SpanStatusCode.OK
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        });
+        it('records an exception when the submittable emits "error"', () => {
+            const submittable = makeSubmittable();
+            const error = new Error('cursor failed');
+            const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+            const original = jest.fn().mockReturnValue(submittable);
+            const wrapped = patchFn(original);
+            const returned = wrapped.apply(fakeClient(), ['SELECT * FROM events']);
+            returned.emit('error', error);
+            expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, error);
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        });
+        it('finalizes the span only once even if "end" fires after "error"', () => {
+            const submittable = makeSubmittable();
+            const error = new Error('boom');
+            const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+            const original = jest.fn().mockReturnValue(submittable);
+            const wrapped = patchFn(original);
+            const returned = wrapped.apply(fakeClient(), ['SELECT 1']);
+            returned.emit('error', error);
+            returned.emit('end');
+            expect(helpers_1.default.handleException).toHaveBeenCalledTimes(1);
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        });
+    });
+    // ── Single-span guarantee ──────────────────────────────────────────────────────
+    describe('exactly one span per logical query', () => {
+        it('creates exactly one span for a promise-form query', async () => {
+            await invokePromise(['SELECT * FROM users'], { rowCount: 1 });
+            expect(mockTracer.startSpan).toHaveBeenCalledTimes(1);
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        });
+        it('creates exactly one span for a pool.query() that delegates to client.query()', async () => {
+            // Model `pg`: Pool.query() checks out a Client and calls client.query().
+            // Only Client.query is patched, so a pool query must yield ONE span.
+            const patchFn = wrapper_1.default._patchQuery(mockTracer, {});
+            const realClientQuery = jest.fn().mockResolvedValue({ rowCount: 2 });
+            const patchedClientQuery = patchFn(realClientQuery);
+            const client = { ...fakeClient(), query: patchedClientQuery };
+            // Unpatched Pool.query delegating to the patched client query.
+            const poolQuery = (...args) => client.query(...args);
+            await poolQuery('SELECT * FROM users WHERE id = $1', [1]);
+            expect(mockTracer.startSpan).toHaveBeenCalledTimes(1);
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        });
+    });
+});
+// ── Instrumentation patching: only Client.query is wrapped, never Pool.query ──────
+describe('OpenlitPgInstrumentation patch targets', () => {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const PgInstrumentation = require('../pg').default;
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const { isWrapped } = require('@opentelemetry/instrumentation');
+    it('wraps Client.prototype.query but leaves Pool.prototype.query untouched', () => {
+        const clientQuery = function () { };
+        const poolQuery = function () { };
+        const fakePgModule = {
+            Client: function () { },
+            Pool: function () { },
+        };
+        fakePgModule.Client.prototype.query = clientQuery;
+        fakePgModule.Pool.prototype.query = poolQuery;
+        const instr = new PgInstrumentation();
+        instr.manualPatch(fakePgModule);
+        expect(isWrapped(fakePgModule.Client.prototype.query)).toBe(true);
+        expect(isWrapped(fakePgModule.Pool.prototype.query)).toBe(false);
+        expect(fakePgModule.Pool.prototype.query).toBe(poolQuery); // unchanged
+    });
+});
+//# sourceMappingURL=pg-trace-comparison.test.js.map

package/dist/instrumentation/__tests__/pg-trace-comparison.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/pg-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;GAmBG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,4DAAsC;AACtC,0DAAyC;AACzC,yDAAyE;AACzE,iEAAyC;AACzC,oFAA2D;AAC3D,8EAAqI;AACrI,4CAA8C;AAC9C,8CAA0D;AAE1D,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,UAAU,EAAE,IAAI;IAChB,OAAO,EAAE,EAAE,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE;IACvC,yBAAyB,EAAE,IAAI,CAAC,EAAE,EAAE;CACrC,CAAC,CAAC,CAAC;AACJ,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,UAAU,EAAE,IAAI;IAChB,OAAO,EAAE;QACP,UAAU,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE;QAC9B,yBAAyB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE;KACjD;CACF,CAAC,CAAC,CAAC;AACJ,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;CACtD,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iDAAiD,EAAE,GAAG,EAAE;IAC/D,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;SACrB,CAAC;QAEF,UAAU,GAAG;YACX,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;SAC/C,CAAC;QAED,gBAAqB,CAAC,WAAW,GAAG,iBAAiB,CAAC;QACtD,gBAAqB,CAAC,eAAe,GAAG,cAAc,CAAC;QACvD,gBAAqB,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACnD,gBAAqB,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC9C,gBAAqB,CAAC,cAAc,GAAG,KAAK,CAAC;QAC7C,iBAAqB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAClD,0BAAiC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,4DAA4D;IAC5D,MAAM,UAAU,GAAG,GAAG,EAAE,CAAC,CAAC;QACxB,oBAAoB,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE;KACjF,CAAC,CAAC;IAEH,mEAAmE;IACnE,KAAK,UAAU,aAAa,CAC1B,IAAW,EACX,SAAc,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC7B,SAA4C,EAAE;QAE9C,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACrD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,IAAI,CAAC,CAAC;QACxC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,+EAA+E;IAE/E,QAAQ,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;YACxB,MAAM,CAAC,iBAAS,CAAC,iBAAiB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAC7D,6BAAkB,CAAC,mBAAmB,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;YACxB,MAAM,CAAC,iBAAS,CAAC,iBAAiB,CAAC,mCAAmC,CAAC,CAAC,CAAC,IAAI,CAC3E,6BAAkB,CAAC,mBAAmB,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,CAAC,iBAAS,CAAC,iBAAiB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAC7D,6BAAkB,CAAC,kBAAkB,CACtC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CAAC,iBAAS,CAAC,iBAAiB,CAAC,iCAAiC,CAAC,CAAC,CAAC,IAAI,CACzE,6BAAkB,CAAC,mBAAmB,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qCAAqC,EAAE,GAAG,EAAE;QACnD,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,iBAAS,CAAC,gBAAgB,CAAC,mCAAmC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,iBAAS,CAAC,gBAAgB,CAAC,uCAAuC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAChG,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,iBAAS,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACtE,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;YAC5B,MAAM,CAAC,iBAAS,CAAC,sBAAsB,CAAC,+BAA+B,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3F,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;YACxB,MAAM,CAAC,iBAAS,CAAC,sBAAsB,CAAC,+BAA+B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,CAAC,iBAAS,CAAC,sBAAsB,CAAC,+BAA+B,CAAC,CAAC,CAAC,IAAI,CAC5E,eAAe,CAChB,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,CAAC,iBAAS,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,+EAA+E;IAE/E,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,cAAc,EACjC,YAAY,CACb,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,aAAa,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,cAAc,EACjC,gBAAgB,CACjB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QACnG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,aAAa,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,uBAAuB,EAC1C,QAAQ,CACT,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,aAAa,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,2BAA2B,EAC9C,iBAAiB,CAClB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,wCAAiB,EAAE,cAAc,CAAC,CAAC;QACtF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CACpD,6BAAkB,CAAC,kBAAkB,EACrC,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CACpD,6BAAkB,CAAC,uBAAuB,EAC1C,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,aAAa,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,8CAAuB,EAAE,SAAS,CAAC,CAAC;QACvF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,6BAAkB,CAAC,8BAA8B,CAClD,CAAC;QACF,MAAM,CAAC,mCAAyB,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,aAAa,CAAC,CAAC,qBAAqB,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,iBAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACjD,CAAC,EACD,MAAM,CAAC,gBAAgB,CAAC;YACtB,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,YAAY;YACjD,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;YAChD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,gBAAgB;YACrD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,IAAI;YACtC,CAAC,wCAAiB,CAAC,EAAE,cAAc;YACnC,CAAC,yDAAkC,CAAC,EAAE,iBAAiB;SACxD,CAAC,CACH,CAAC;QACF,MAAM,CAAC,iBAAO,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACnE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC;YACtB,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;SACjD,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC3D,gBAAqB,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7C,MAAM,aAAa,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,iBAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,CAAC,iBAAO,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QAClE,0BAAiC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,gFAAgF;IAEhF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,CAAC,CAAC,0CAA0C,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACxE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,OAAO,CACR,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,cAAc,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,aAAa,EAChC,qBAAqB,CACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,aAAa,CAAC,CAAC,qBAAqB,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,yBAAyB,EAC5C,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,aAAa,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,gFAAgF;IAEhF,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,aAAa,CAAC;gBAClB,uDAAuD;gBACvD,CAAC,WAAW,CAAC;aACd,CAAC,CAAC;YACH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,2BAA2B,EAC9C,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,aAAa,CAAC,CAAC,+CAA+C,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,+BAA+B,CAChC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,gFAAgF;IAEhF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,aAAa,CAAC,CAAC,kDAAkD,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YACtF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,yBAAyB,CAC1B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,iBAAS,CAAC,oBAAoB,CAAC,yCAAyC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7F,MAAM,CAAC,iBAAS,CAAC,oBAAoB,CAAC,kCAAkC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtF,MAAM,CAAC,iBAAS,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,gFAAgF;IAEhF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,aAAa,CACjB,CAAC,sCAAsC,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAC9D,EAAE,QAAQ,EAAE,CAAC,EAAE,EACf,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAC9B,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,sBAAsB,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YACrG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;YACxF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,EAC5C,OAAO,CACR,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,iFAAiF;IAEjF,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,aAAa,CAAC,CAAC,mCAAmC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE;gBAChF,mBAAmB,EAAE,KAAK;aAC3B,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAQ,EAAE,EAAE,CACtE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,CAC/D,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,aAAa,CACjB,CAAC,mDAAmD,EAAE,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC,EACrE,EAAE,QAAQ,EAAE,CAAC,EAAE,EACf,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAC9B,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,EAC5C,IAAI,CACL,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,EAC5C,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,aAAa,CACjB,CAAC,sCAAsC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAC9D,EAAE,QAAQ,EAAE,CAAC,EAAE,EACf,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAC9B,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,EAC5C,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,iFAAiF;IAEjF,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,aAAa,CACjB,CAAC,EAAE,IAAI,EAAE,gDAAgD,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAC9E,EAAE,QAAQ,EAAE,CAAC,EAAE,EACf,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAC9B,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,UAAU,CACX,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,EAC5C,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,iFAAiF;IAEjF,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,6DAA6D,EAAE,CAAC,IAAI,EAAE,EAAE;YACzE,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,KAAa,EAAE,KAAU,EAAE,EAAO,EAAE,EAAE,CAC9D,EAAE,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAC1B,CAAC;YACF,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAClC,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE;gBAC1B,oBAAoB;gBACpB,EAAE;gBACF,CAAC,GAAQ,EAAE,GAAQ,EAAE,EAAE;oBACrB,MAAM,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;oBACvB,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBAC7B,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,yBAAyB,EAC5C,CAAC,CACF,CAAC;oBACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;oBACxC,IAAI,EAAE,CAAC;gBACT,CAAC;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,CAAC,IAAI,EAAE,EAAE;YAC1E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;YAC9C,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,KAAa,EAAE,EAAO,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;YAC3E,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAClC,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE;gBAC1B,UAAU;gBACV,CAAC,GAAQ,EAAE,EAAE;oBACX,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACxB,MAAM,CAAC,iBAAa,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;oBAC5E,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;oBACxC,IAAI,EAAE,CAAC;gBACT,CAAC;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,mFAAmF;IAEnF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACvF,MAAM,CAAC,iBAAa,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC5E,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,mFAAmF;IAEnF,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,iFAAiF;QACjF,MAAM,eAAe,GAAG,CAAC,QAAiC,EAAE,EAAE,EAAE;YAC9D,MAAM,SAAS,GAAiD,EAAE,CAAC;YACnE,OAAO;gBACL,GAAG,KAAK;gBACR,EAAE,CAAC,KAAa,EAAE,EAAyB;oBACzC,CAAC,SAAS,CAAC,KAAK,MAAf,SAAS,CAAC,KAAK,IAAM,EAAE,EAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACnC,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,IAAI,CAAC,KAAa,EAAE,GAAG,IAAW;oBAChC,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACxD,CAAC;aACF,CAAC;QACJ,CAAC,CAAC;QAEF,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,WAAW,GAAG,eAAe,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YACrD,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YACxD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,QAAQ,GAAQ,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,CAAC;YAC5E,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,uCAAuC;YAC3E,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,iCAAiC;YAE9E,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAErB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,cAAc,EACjC,YAAY,CACb,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,yBAAyB,EAC5C,CAAC,CACF,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB;YAClF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,WAAW,GAAG,eAAe,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACzC,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YACxD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,QAAQ,GAAQ,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,CAAC;YAC5E,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAE9B,MAAM,CAAC,iBAAa,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC5E,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,WAAW,GAAG,eAAe,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;YAChC,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YACxD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,QAAQ,GAAQ,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YAChE,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAC9B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAErB,MAAM,CAAC,iBAAa,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,kFAAkF;IAElF,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,aAAa,CAAC,CAAC,qBAAqB,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACtD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,yEAAyE;YACzE,qEAAqE;YACrE,MAAM,OAAO,GAAG,iBAAS,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtD,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YACrE,MAAM,kBAAkB,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;YAEpD,MAAM,MAAM,GAAG,EAAE,GAAG,UAAU,EAAE,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;YAC9D,+DAA+D;YAC/D,MAAM,SAAS,GAAG,CAAC,GAAG,IAAW,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAE5D,MAAM,SAAS,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACtD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,qFAAqF;AAErF,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;IACtD,8DAA8D;IAC9D,MAAM,iBAAiB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;IACnD,8DAA8D;IAC9D,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC;IAEhE,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;QAChF,MAAM,WAAW,GAAG,cAAa,CAAC,CAAC;QACnC,MAAM,SAAS,GAAG,cAAa,CAAC,CAAC;QACjC,MAAM,YAAY,GAAG;YACnB,MAAM,EAAE,cAAa,CAAC;YACtB,IAAI,EAAE,cAAa,CAAC;SACd,CAAC;QACT,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,GAAG,WAAW,CAAC;QAClD,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,GAAG,SAAS,CAAC;QAE9C,MAAM,KAAK,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACtC,KAAK,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QAEhC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjE,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY;IACzE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,11 @@
+/**
+ * Cross-Language Trace Comparison Tests for local HuggingFace inference
+ * (Transformers.js) instrumentation.
+ *
+ * Verifies that the TypeScript SDK produces traces consistent with the Python
+ * `transformers` instrumentation and the OTel GenAI semantic conventions:
+ *   - text-generation reports the `chat` operation (Python parity)
+ *   - other local pipelines map to the closest OTel operation
+ *   - token usage, cache tokens, timing, and package version are stamped
+ */
+export {};