openlit 1.12.0 → 1.14.0

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

@@ -0,0 +1,160 @@
+"use strict";
+/**
+ * Cross-language trace comparison tests for Astra DB integration.
+ *
+ * Verifies TypeScript SDK trace attributes match the Python SDK's Astra
+ * instrumentation behavior for Collection operations.
+ */
+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("../astra/wrapper"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+jest.mock('../../config');
+jest.mock('../../helpers');
+describe('Astra DB 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;
+        helpers_1.default.handleException = jest.fn();
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    async function invokeAsyncWrapped(dbOperation, args, returnValue = {}, instance = { name: 'movies' }) {
+        const patchFn = wrapper_1.default._patchCollectionMethod(mockTracer, dbOperation);
+        const originalMethod = jest.fn().mockResolvedValue(returnValue);
+        const wrapped = patchFn(originalMethod);
+        const response = await wrapped.call(instance, ...args);
+        return { originalMethod, response };
+    }
+    describe('common attributes', () => {
+        it('sets Python parity common database and resource attributes', async () => {
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_INSERT, [{ _id: '1', title: 'Dune' }], { insertedId: '1' });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_SYSTEM_NAME, 'astra');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'INSERT');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_COLLECTION_NAME, 'movies');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith('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(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT, 'openlit-testing');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith('service.name', 'openlit-test');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_SDK_VERSION, expect.any(String));
+        });
+        it('extracts server address and port from a URL endpoint', async () => {
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_SELECT, [{ genre: 'sci-fi' }], { _id: '1' }, { name: 'movies', database: { apiEndpoint: 'https://db.example.apps.astra.datastax.com:8443' } });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'db.example.apps.astra.datastax.com');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 8443);
+        });
+        it('extracts server address and port from a host:port endpoint', async () => {
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_SELECT, [{ genre: 'sci-fi' }], { _id: '1' }, { name: 'movies', client: { api_endpoint: 'astra.local:9090' } });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'astra.local');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 9090);
+        });
+        it('falls back to the Astra default server when no endpoint is present', async () => {
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_SELECT, [{}], null);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'astra.datastax.com');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 443);
+        });
+    });
+    describe('insert operations', () => {
+        it('sets documents count, query text, returned rows, and summary', async () => {
+            const documents = [{ _id: '1' }, { _id: '2' }];
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_INSERT, [documents], { insertedIds: ['1', '2'] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_DOCUMENTS_COUNT, 2);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(documents));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 2);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, 'INSERT movies documents_count=2');
+        });
+    });
+    describe('update operations', () => {
+        it('sets filter, update query text, returned rows, and summary', async () => {
+            const filter = { status: 'draft' };
+            const update = { $set: { status: 'published' } };
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_UPDATE, [filter, update], { update_info: { nModified: 3 } });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, JSON.stringify(filter));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(update));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 3);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, `UPDATE movies filter=${JSON.stringify(filter)} update=${JSON.stringify(update)}`);
+        });
+    });
+    describe('replace operations', () => {
+        it('sets replace filter and summary without upsert', async () => {
+            const filter = { _id: '1' };
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_REPLACE, [filter, { _id: '1', title: 'Dune Messiah' }, { upsert: false }], { modifiedCount: 1 });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'findAndModify');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, JSON.stringify(filter));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(filter));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, `findAndModify movies filter=${JSON.stringify(filter)} upsert=false`);
+        });
+        it('detects replace upsert and sets db.operation.name = UPSERT', async () => {
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_REPLACE, [{ _id: 'missing' }, { _id: 'missing' }, { upsert: true }], { upsertedId: 'missing' });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, semantic_convention_1.default.DB_OPERATION_UPSERT);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, 'UPSERT movies filter={"_id":"missing"} upsert=true');
+        });
+    });
+    describe('select operations', () => {
+        it('sets findOne filter, query text, returned rows, and summary', async () => {
+            const filter = { genre: 'sci-fi' };
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_SELECT, [filter], { _id: '1' });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, JSON.stringify(filter));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(filter));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 1);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, `SELECT movies filter=${JSON.stringify(filter)}`);
+        });
+        it('keeps find() synchronous and returns the original cursor', () => {
+            const cursor = { toArray: jest.fn() };
+            const patchFn = wrapper_1.default._patchSyncFindMethod(mockTracer);
+            const originalMethod = jest.fn().mockReturnValue(cursor);
+            const wrapped = patchFn(originalMethod);
+            const result = wrapped.call({ name: 'movies' }, { genre: 'sci-fi' });
+            expect(result).toBe(cursor);
+            expect(result.then).toBeUndefined();
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, 'SELECT movies filter={"genre":"sci-fi"}');
+            expect(mockSpan.end).toHaveBeenCalled();
+        });
+    });
+    describe('delete operations', () => {
+        it('sets delete filter, query text, returned rows, and summary', async () => {
+            const filter = { expired: true };
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_DELETE, [filter], { deletedCount: 4 });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, JSON.stringify(filter));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(filter));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 4);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_SUMMARY, `DELETE movies filter=${JSON.stringify(filter)}`);
+        });
+        it('sets findOneAndDelete returned rows from the deleted document response', async () => {
+            await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_FIND_AND_DELETE, [{ _id: '1' }], { _id: '1' });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, 1);
+        });
+    });
+    it('does not set db.query.text when captureMessageContent is false', async () => {
+        config_1.default.captureMessageContent = false;
+        await invokeAsyncWrapped(semantic_convention_1.default.DB_OPERATION_SELECT, [{ genre: 'sci-fi' }], { _id: '1' });
+        expect(mockSpan.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.DB_QUERY_TEXT, expect.anything());
+    });
+    it('calls handleException and ends the span on error', async () => {
+        const error = new Error('Astra request failed');
+        const patchFn = wrapper_1.default._patchCollectionMethod(mockTracer, semantic_convention_1.default.DB_OPERATION_SELECT);
+        const originalMethod = jest.fn().mockRejectedValue(error);
+        const wrapped = patchFn(originalMethod);
+        await expect(wrapped.call({ name: 'movies' }, { genre: 'sci-fi' })).rejects.toThrow('Astra request failed');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, error);
+        expect(mockSpan.end).toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=astra-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"astra-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/astra-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;AAEH,+DAA4C;AAC5C,0DAAyC;AACzC,4DAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAE3B,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxD,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,iBAAqB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,KAAK,UAAU,kBAAkB,CAC/B,WAAmB,EACnB,IAAW,EACX,cAAmB,EAAE,EACrB,WAAgB,EAAE,IAAI,EAAE,QAAQ,EAAE;QAElC,MAAM,OAAO,GAAG,iBAAY,CAAC,sBAAsB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAC7E,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,CAAC;QAEvD,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC;IACtC,CAAC;IAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,mBAAmB,EACtC,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,EAC7B,EAAE,UAAU,EAAE,GAAG,EAAE,CACpB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;YAC/F,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;YACnG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;YACpG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE,SAAS,CAAC,CAAC;YACpF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,6BAAkB,CAAC,8BAA8B,CAClD,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,2BAA2B,EAC9C,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;YACnF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5G,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,mBAAmB,EACtC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EACrB,EAAE,GAAG,EAAE,GAAG,EAAE,EACZ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,WAAW,EAAE,iDAAiD,EAAE,EAAE,CACjG,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,cAAc,EACjC,oCAAoC,CACrC,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC3F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,mBAAmB,EACtC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EACrB,EAAE,GAAG,EAAE,GAAG,EAAE,EACZ,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;YACrG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC3F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,kBAAkB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;YAE7E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,oBAAoB,CAAC,CAAC;YAC5G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,SAAS,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;YAC/C,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,mBAAmB,EACtC,CAAC,SAAS,CAAC,EACX,EAAE,WAAW,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAC5B,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC7F,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;YAChH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YACpG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,iCAAiC,CAClC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC;YACjD,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,mBAAmB,EACtC,CAAC,MAAM,EAAE,MAAM,CAAC,EAChB,EAAE,WAAW,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,CAClC,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YACzG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YAC7G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YACpG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,wBAAwB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,WAAW,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAClF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;YAC5B,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,oBAAoB,EACvC,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAChE,EAAE,aAAa,EAAE,CAAC,EAAE,CACrB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;YAC1G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YACzG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YAC7G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,+BAA+B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,CACrE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,oBAAoB,EACvC,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAC1D,EAAE,UAAU,EAAE,SAAS,EAAE,CAC1B,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,6BAAkB,CAAC,mBAAmB,CACvC,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YACnC,MAAM,kBAAkB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;YAEzF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YACzG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YAC7G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YACpG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,wBAAwB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CACjD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,iBAAY,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAC9D,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YACzD,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAE,MAAc,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YAC7C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,yCAAyC,CAC1C,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACjC,MAAM,kBAAkB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAEhG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YACzG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YAC7G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YACpG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,wBAAwB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CACjD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,kBAAkB,CACtB,6BAAkB,CAAC,4BAA4B,EAC/C,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EACd,EAAE,GAAG,EAAE,GAAG,EAAE,CACb,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;QACtG,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC7E,gBAAqB,CAAC,qBAAqB,GAAG,KAAK,CAAC;QAErD,MAAM,kBAAkB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;QAEtG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CACpD,6BAAkB,CAAC,aAAa,EAChC,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAChD,MAAM,OAAO,GAAG,iBAAY,CAAC,sBAAsB,CAAC,UAAU,EAAE,6BAAkB,CAAC,mBAAmB,CAAC,CAAC;QACxG,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QAExC,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC5G,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;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/browser-use-wrapper.test.js

@@ -0,0 +1,284 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const api_1 = require("@opentelemetry/api");
+const core_1 = require("@opentelemetry/core");
+const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const wrapper_1 = __importDefault(require("../browser-use/wrapper"));
+const browser_use_1 = __importDefault(require("../browser-use"));
+jest.mock('../../config');
+jest.mock('../../helpers', () => ({
+    __esModule: true,
+    default: {
+        handleException: jest.fn(),
+        getChatModelCost: jest.fn().mockReturnValue(0.0025),
+    },
+    applyCustomSpanAttributes: jest.fn(),
+}));
+jest.mock('@opentelemetry/core', () => ({
+    isTracingSuppressed: jest.fn().mockReturnValue(false),
+}));
+describe('browser-use wrapper', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            setStatus: jest.fn(),
+            end: jest.fn(),
+        };
+        mockTracer = { startSpan: jest.fn(() => 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.pricingInfo = {
+            chat: { 'gpt-4o-mini': { promptPrice: 0.15, completionPrice: 0.6 } },
+        };
+        core_1.isTracingSuppressed.mockReturnValue(false);
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    function attrs() {
+        return Object.fromEntries(mockSpan.setAttribute.mock.calls);
+    }
+    it('wraps Agent.run into an invoke_agent span with browser-use metadata', async () => {
+        const mainSpan = {
+            setAttribute: jest.fn(),
+            setStatus: jest.fn(),
+            end: jest.fn(),
+        };
+        mockTracer.startSpan = jest.fn((name) => {
+            if (name === 'invoke_agent shopping-bot') {
+                return mainSpan;
+            }
+            return {
+                setAttribute: jest.fn(),
+                setStatus: jest.fn(),
+                end: jest.fn(),
+            };
+        });
+        class StubAgent {
+            constructor() {
+                this.name = 'shopping-bot';
+                this.task = 'Buy the cheapest laptop under $1000';
+                this.id = 'agent-1';
+                this.task_id = 'task-123';
+                this.session_id = 'session-456';
+                this.llm = { model: 'gpt-4o-mini' };
+                this.use_vision = true;
+                this.browser_profile = { headless: true, allowed_domains: ['example.com'] };
+            }
+            async run() {
+                return {
+                    history: [{ result: [{ is_success: true }] }],
+                    usage: { total_input_tokens: 10, total_output_tokens: 5 },
+                    final_result: () => 'done',
+                    total_duration_seconds: () => 1.5,
+                };
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(StubAgent.prototype.run);
+        await wrapped.call(agent, 10);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('invoke_agent shopping-bot', {
+            kind: api_1.SpanKind.CLIENT,
+            attributes: {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT,
+                [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_BROWSER_USE,
+            },
+        });
+        const a = Object.fromEntries(mainSpan.setAttribute.mock.calls);
+        expect(a[semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]).toBe('openlit');
+        expect(a[semantic_conventions_1.ATTR_SERVICE_NAME]).toBe('openlit-test');
+        expect(a[semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]).toBe('openlit-testing');
+        expect(a[semantic_convention_1.default.GEN_AI_PROVIDER_NAME]).toBe('browser_use');
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('invoke_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('invoke_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_TYPE]).toBe('browser');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_ID]).toBe('agent-1');
+        expect(a[semantic_convention_1.default.GEN_AI_BROWSER_AGENT_TASK_ID]).toBe('task-123');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_SESSION_ID]).toBe('session-456');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_USE_VISION]).toBe(true);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_HEADLESS]).toBe(true);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_MAX_STEPS]).toBe(10);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_STEP_COUNT]).toBe(1);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_FINAL_RESULT]).toBe('done');
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]).toBe(10);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]).toBe(5);
+        expect(typeof a[semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION]).toBe('number');
+        expect(mainSpan.setStatus).toHaveBeenCalledWith({ code: 1 });
+        expect(mainSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('records Agent.run errors and still ends the span', async () => {
+        class StubAgent {
+            constructor() {
+                this.task = 'fail task';
+            }
+            async run() {
+                throw new Error('agent run failed');
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(StubAgent.prototype.run);
+        await expect(wrapped.call(agent)).rejects.toThrow('agent run failed');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, expect.objectContaining({ message: 'agent run failed' }));
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('bypasses instrumentation when tracing is suppressed', async () => {
+        core_1.isTracingSuppressed.mockReturnValue(true);
+        const original = jest.fn().mockResolvedValue({ ok: true });
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(original);
+        await wrapped.call({ task: 'x' });
+        expect(original).toHaveBeenCalled();
+        expect(mockTracer.startSpan).not.toHaveBeenCalled();
+    });
+    it('wraps Controller.act into invoke_agent action spans (Python parity)', async () => {
+        class StubController {
+            constructor() {
+                this.registry = {
+                    get_action: () => ({ description: 'Click element by index.' }),
+                };
+            }
+            async act(action) {
+                return { is_success: true, extracted_content: `done:${Object.keys(action)[0]}` };
+            }
+        }
+        const controller = new StubController();
+        const wrapped = wrapper_1.default._patchControllerAct(mockTracer, '0.7.3')(StubController.prototype.act);
+        await wrapped.call(controller, { go_to_url: { url: 'https://example.com/page' } }, {});
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('invoke_agent go_to_url', {
+            kind: api_1.SpanKind.INTERNAL,
+            attributes: {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT,
+                [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_BROWSER_USE,
+            },
+        });
+        const a = attrs();
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('invoke_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('execute_task');
+        expect(a[semantic_convention_1.default.GEN_AI_ACTION_TYPE]).toBe('go_to_url');
+        expect(a[semantic_convention_1.default.GEN_AI_ACTION_INDEX]).toBe(1);
+        expect(a['gen_ai.action.url']).toBe('https://example.com/page');
+        expect(a[semantic_convention_1.default.GEN_AI_ACTION_SUCCESS]).toBe(true);
+        expect(a[semantic_convention_1.default.GEN_AI_TOOL_CALL_ARGUMENTS]).toBe('{"url":"https://example.com/page"}');
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('records controller errors and still ends the action span', async () => {
+        class StubController {
+            constructor() {
+                this.registry = {
+                    get_action: () => ({ description: 'Navigate to URL.' }),
+                };
+            }
+            async act() {
+                throw new Error('navigation failed');
+            }
+        }
+        const controller = new StubController();
+        const wrapped = wrapper_1.default._patchControllerAct(mockTracer, '0.7.3')(StubController.prototype.act);
+        await expect(wrapped.call(controller, { go_to_url: { url: 'https://example.com' } }, {})).rejects.toThrow('navigation failed');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, expect.objectContaining({ message: 'navigation failed' }));
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('wraps Agent.step into execute_task spans', async () => {
+        class StubAgent {
+            constructor() {
+                this.name = 'research-bot';
+                this.step_count = 2;
+                this.llm = { model: 'gpt-4o' };
+            }
+            async step() {
+                return { is_success: true };
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentStep(mockTracer, '0.7.3')(StubAgent.prototype.step);
+        await wrapped.call(agent);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('execute_task step 2', { kind: api_1.SpanKind.CLIENT });
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('step');
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('execute_task');
+    });
+    it('wraps Agent.pause into browser lifecycle spans', async () => {
+        class StubAgent {
+            constructor() {
+                this.name = 'research-bot';
+            }
+            async pause() {
+                return undefined;
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentLifecycle(mockTracer, 'pause', '0.7.3')(StubAgent.prototype.pause);
+        await wrapped.call(agent);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('browser pause', { kind: api_1.SpanKind.CLIENT });
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('pause');
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('invoke_workflow');
+    });
+    it('creates step and action spans from run history when step hooks do not fire', async () => {
+        class StubAgent {
+            constructor() {
+                this.name = 'history-bot';
+                this.task = 'complete checkout';
+            }
+            async run(_options) {
+                return {
+                    history: [
+                        {
+                            model_output: { action: [{ click: { index: 1 } }] },
+                            result: [{ is_success: true }],
+                            state: { url: 'https://shop.example/cart', title: 'Cart' },
+                        },
+                    ],
+                    usage: { total_input_tokens: 3, total_output_tokens: 2 },
+                    final_result: () => 'checked out',
+                };
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(StubAgent.prototype.run);
+        await wrapped.call(agent);
+        const spanNames = mockTracer.startSpan.mock.calls.map((call) => call[0]);
+        expect(spanNames).toContain('invoke_agent history-bot');
+        expect(spanNames).toContain('execute_task step 1');
+        expect(spanNames).toContain('invoke_agent click');
+    });
+});
+describe('BrowserUseInstrumentation patch targets', () => {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const { isWrapped } = require('@opentelemetry/instrumentation');
+    it('wraps Agent lifecycle methods and Controller.act', () => {
+        class Agent {
+            async run() {
+                return null;
+            }
+            async step() {
+                return null;
+            }
+            async pause() { }
+            async resume() { }
+            async stop() { }
+        }
+        class Controller {
+            async act() {
+                return null;
+            }
+        }
+        const fakeModule = { Agent, Controller };
+        const instrumentation = new browser_use_1.default();
+        instrumentation.manualPatch(fakeModule, '0.7.3');
+        expect(isWrapped(fakeModule.Agent.prototype.run)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.step)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.pause)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.resume)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.stop)).toBe(true);
+        expect(isWrapped(fakeModule.Controller.prototype.act)).toBe(true);
+    });
+});
+//# sourceMappingURL=browser-use-wrapper.test.js.map

package/dist/instrumentation/__tests__/browser-use-wrapper.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"browser-use-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/browser-use-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAA8C;AAC9C,8CAA0D;AAC1D,8EAAiG;AACjG,0DAAyC;AACzC,4DAA0C;AAC1C,oFAA2D;AAC3D,qEAAuD;AACvD,iEAAuD;AAEvD,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;QACP,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE;QAC1B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC;KACpD;IACD,yBAAyB,EAAE,IAAI,CAAC,EAAE,EAAE;CACrC,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,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;SACf,CAAC;QACF,UAAU,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAEnD,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,WAAW,GAAG;YACnC,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,EAAE,EAAE;SACrE,CAAC;QACD,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,SAAS,KAAK;QACZ,OAAO,MAAM,CAAC,WAAW,CAAE,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7E,CAAC;IAED,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,QAAQ,GAAG;YACf,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;SACf,CAAC;QACF,UAAU,CAAC,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,IAAY,EAAE,EAAE;YAC9C,IAAI,IAAI,KAAK,2BAA2B,EAAE,CAAC;gBACzC,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,OAAO;gBACL,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;aACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,cAAc,CAAC;gBACtB,SAAI,GAAG,qCAAqC,CAAC;gBAC7C,OAAE,GAAG,SAAS,CAAC;gBACf,YAAO,GAAG,UAAU,CAAC;gBACrB,eAAU,GAAG,aAAa,CAAC;gBAC3B,QAAG,GAAG,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;gBAC/B,eAAU,GAAG,IAAI,CAAC;gBAClB,oBAAe,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC;YAUzE,CAAC;YARC,KAAK,CAAC,GAAG;gBACP,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;oBAC7C,KAAK,EAAE,EAAE,kBAAkB,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC,EAAE;oBACzD,YAAY,EAAE,GAAG,EAAE,CAAC,MAAM;oBAC1B,sBAAsB,EAAE,GAAG,EAAE,CAAC,GAAG;iBAClC,CAAC;YACJ,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC/F,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAE9B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,EAAE;YAC7E,IAAI,EAAE,cAAQ,CAAC,MAAM;YACrB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,2BAA2B;gBACrF,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAC5C,6BAAkB,CAAC,yBAAyB;aAC/C;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,MAAM,CAAC,WAAW,CAAE,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9E,MAAM,CAAC,CAAC,CAAC,8CAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,CAAC,CAAC,CAAC,wCAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAClF,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACvE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACzE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5E,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1E,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,OAAO,CAAC,CAAC,6BAAkB,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrF,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,WAAW,CAAC;YAKrB,CAAC;YAHC,KAAK,CAAC,GAAG;gBACP,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtC,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAE/F,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACtE,MAAM,CAAC,iBAAa,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACxD,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CACzD,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QAClE,0BAAiC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEhF,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;QAClC,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,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,cAAc;YAApB;gBACE,aAAQ,GAAG;oBACT,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,yBAAyB,EAAE,CAAC;iBAC/D,CAAC;YAKJ,CAAC;YAHC,KAAK,CAAC,GAAG,CAAC,MAA+B;gBACvC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;YACnF,CAAC;SACF;QAED,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,CACxE,cAAc,CAAC,SAAS,CAAC,GAAG,CAC7B,CAAC;QACF,MAAM,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,0BAA0B,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAEvF,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,EAAE;YAC1E,IAAI,EAAE,cAAQ,CAAC,QAAQ;YACvB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,2BAA2B;gBACrF,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAC5C,6BAAkB,CAAC,yBAAyB;aAC/C;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;QAClB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACzE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAC3D,oCAAoC,CACrC,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,cAAc;YAApB;gBACE,aAAQ,GAAG;oBACT,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC;iBACxD,CAAC;YAKJ,CAAC;YAHC,KAAK,CAAC,GAAG;gBACP,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC;SACF;QAED,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,CACxE,cAAc,CAAC,SAAS,CAAC,GAAG,CAC7B,CAAC;QAEF,MAAM,MAAM,CACV,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,qBAAqB,EAAE,EAAE,EAAE,EAAE,CAAC,CAC5E,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAEvC,MAAM,CAAC,iBAAa,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACxD,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAC1D,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,cAAc,CAAC;gBACtB,eAAU,GAAG,CAAC,CAAC;gBACf,QAAG,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YAK5B,CAAC;YAHC,KAAK,CAAC,IAAI;gBACR,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;YAC9B,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACjG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE1B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACpG,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,cAAc,CAAC;YAKxB,CAAC;YAHC,KAAK,CAAC,KAAK;gBACT,OAAO,SAAS,CAAC;YACnB,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,oBAAoB,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAClF,SAAS,CAAC,SAAS,CAAC,KAAK,CAC1B,CAAC;QACF,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE1B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9F,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,aAAa,CAAC;gBACrB,SAAI,GAAG,mBAAmB,CAAC;YAe7B,CAAC;YAbC,KAAK,CAAC,GAAG,CAAC,QAAkC;gBAC1C,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,YAAY,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE;4BACnD,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;4BAC9B,KAAK,EAAE,EAAE,GAAG,EAAE,2BAA2B,EAAE,KAAK,EAAE,MAAM,EAAE;yBAC3D;qBACF;oBACD,KAAK,EAAE,EAAE,kBAAkB,EAAE,CAAC,EAAE,mBAAmB,EAAE,CAAC,EAAE;oBACxD,YAAY,EAAE,GAAG,EAAE,CAAC,aAAa;iBAClC,CAAC;YACJ,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC/F,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE1B,MAAM,SAAS,GAAI,UAAU,CAAC,SAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAW,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/F,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;QACxD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QACnD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;IACvD,8DAA8D;IAC9D,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC;IAEhE,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,KAAK;YACT,KAAK,CAAC,GAAG;gBACP,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,IAAI;gBACR,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,KAAK,KAAI,CAAC;YAChB,KAAK,CAAC,MAAM,KAAI,CAAC;YACjB,KAAK,CAAC,IAAI,KAAI,CAAC;SAChB;QAED,MAAM,UAAU;YACd,KAAK,CAAC,GAAG;gBACP,OAAO,IAAI,CAAC;YACd,CAAC;SACF;QAED,MAAM,UAAU,GAAG,EAAE,KAAK,EAAE,UAAU,EAAS,CAAC;QAChD,MAAM,eAAe,GAAG,IAAI,qBAAyB,EAAE,CAAC;QACxD,eAAe,CAAC,WAAW,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAEjD,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/cache-pricing.test.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Unit tests for cache-aware cost calculation in `OpenLitHelper.getChatModelCost`.
+ *
+ * Mirrors the Python `test_cache_pricing.py` coverage and validates the two
+ * provider token-accounting conventions:
+ *   - Anthropic native API -> promptTokens is exclusive of cache tokens.
+ *   - OpenAI / LangChain -> promptTokens is inclusive of cache tokens.
+ *
+ * The same cached tokens must never be billed twice, and models without cache
+ * pricing must produce exactly the legacy cost regardless of convention.
+ */
+export {};

package/dist/instrumentation/__tests__/cache-pricing.test.js

@@ -0,0 +1,95 @@
+"use strict";
+/**
+ * Unit tests for cache-aware cost calculation in `OpenLitHelper.getChatModelCost`.
+ *
+ * Mirrors the Python `test_cache_pricing.py` coverage and validates the two
+ * provider token-accounting conventions:
+ *   - Anthropic native API -> promptTokens is exclusive of cache tokens.
+ *   - OpenAI / LangChain -> promptTokens is inclusive of cache tokens.
+ *
+ * The same cached tokens must never be billed twice, and models without cache
+ * pricing must produce exactly the legacy cost regardless of convention.
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const helpers_1 = __importDefault(require("../../helpers"));
+const PRICING = {
+    chat: {
+        'claude-3-5-sonnet-20241022': {
+            promptPrice: 0.003,
+            completionPrice: 0.015,
+            cacheReadPrice: 0.0003,
+            cacheCreationPrice: 0.00375,
+        },
+        'model-no-cache-price': {
+            promptPrice: 0.003,
+            completionPrice: 0.015,
+        },
+    },
+};
+const MODEL = 'claude-3-5-sonnet-20241022';
+describe('OpenLitHelper.getChatModelCost — cache pricing', () => {
+    describe('no cache tokens', () => {
+        it('matches the legacy formula', () => {
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 1000, 500);
+            expect(cost).toBeCloseTo((1000 / 1000) * 0.003 + (500 / 1000) * 0.015, 10);
+        });
+        it('returns 0 for an unknown model', () => {
+            expect(helpers_1.default.getChatModelCost('nope', PRICING, 10, 10)).toBe(0);
+        });
+        it('returns 0 when pricing info is missing', () => {
+            expect(helpers_1.default.getChatModelCost(MODEL, {}, 10, 10)).toBe(0);
+        });
+    });
+    describe('Anthropic convention (promptTokens exclusive of cache)', () => {
+        it('adds cache cost on top of the prompt base', () => {
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 800, 500, 100, 100);
+            const expected = (800 / 1000) * 0.003 +
+                (500 / 1000) * 0.015 +
+                (100 / 1000) * 0.0003 +
+                (100 / 1000) * 0.00375;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+        it('leaves the prompt base untouched when no cache price exists', () => {
+            const cost = helpers_1.default.getChatModelCost('model-no-cache-price', PRICING, 800, 500, 100, 100);
+            expect(cost).toBeCloseTo((800 / 1000) * 0.003 + (500 / 1000) * 0.015, 10);
+        });
+    });
+    describe('OpenAI / LangChain convention (promptTokens inclusive of cache)', () => {
+        it('does not bill cache tokens twice', () => {
+            // 1000 total input = 800 uncached + 100 cache read + 100 cache creation
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 1000, 500, 100, 100, true);
+            const expected = (800 / 1000) * 0.003 +
+                (500 / 1000) * 0.015 +
+                (100 / 1000) * 0.0003 +
+                (100 / 1000) * 0.00375;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+        it('matches the exclusive call describing the same request', () => {
+            const inclusive = helpers_1.default.getChatModelCost(MODEL, PRICING, 1000, 500, 100, 100, true);
+            const exclusive = helpers_1.default.getChatModelCost(MODEL, PRICING, 800, 500, 100, 100);
+            expect(inclusive).toBeCloseTo(exclusive, 10);
+        });
+        it('matches the legacy total when no cache price exists (no regression)', () => {
+            const cost = helpers_1.default.getChatModelCost('model-no-cache-price', PRICING, 1000, 500, 100, 100, true);
+            expect(cost).toBeCloseTo((1000 / 1000) * 0.003 + (500 / 1000) * 0.015, 10);
+        });
+        it('handles OpenAI-style cache read only (no cache creation)', () => {
+            // OpenAI: prompt_tokens includes cached_tokens; no cache-creation charge.
+            const pricing = {
+                chat: { 'gpt-cache': { promptPrice: 0.0025, completionPrice: 0.01, cacheReadPrice: 0.00125 } },
+            };
+            const cost = helpers_1.default.getChatModelCost('gpt-cache', pricing, 1000, 200, 400, 0, true);
+            const expected = (600 / 1000) * 0.0025 + (200 / 1000) * 0.01 + (400 / 1000) * 0.00125;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+        it('clamps the prompt base at 0 for malformed token data', () => {
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 100, 0, 200, 200, true);
+            const expected = (200 / 1000) * 0.0003 + (200 / 1000) * 0.00375;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+    });
+});
+//# sourceMappingURL=cache-pricing.test.js.map

package/dist/instrumentation/__tests__/cache-pricing.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache-pricing.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/cache-pricing.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;GAUG;;;;;AAEH,4DAA0C;AAE1C,MAAM,OAAO,GAAG;IACd,IAAI,EAAE;QACJ,4BAA4B,EAAE;YAC5B,WAAW,EAAE,KAAK;YAClB,eAAe,EAAE,KAAK;YACtB,cAAc,EAAE,MAAM;YACtB,kBAAkB,EAAE,OAAO;SAC5B;QACD,sBAAsB,EAAE;YACtB,WAAW,EAAE,KAAK;YAClB,eAAe,EAAE,KAAK;SACvB;KACF;CACF,CAAC;AAEF,MAAM,KAAK,GAAG,4BAA4B,CAAC;AAE3C,QAAQ,CAAC,gDAAgD,EAAE,GAAG,EAAE;IAC9D,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACvE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,iBAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,CAAC,iBAAa,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACtE,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAChF,MAAM,QAAQ,GACZ,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM;gBACrB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CACzC,sBAAsB,EACtB,OAAO,EACP,GAAG,EACH,GAAG,EACH,GAAG,EACH,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iEAAiE,EAAE,GAAG,EAAE;QAC/E,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,wEAAwE;YACxE,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YACvF,MAAM,QAAQ,GACZ,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM;gBACrB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,SAAS,GAAG,iBAAa,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YAC5F,MAAM,SAAS,GAAG,iBAAa,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrF,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CACzC,sBAAsB,EACtB,OAAO,EACP,IAAI,EACJ,GAAG,EACH,GAAG,EACH,GAAG,EACH,IAAI,CACL,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,0EAA0E;YAC1E,MAAM,OAAO,GAAG;gBACd,IAAI,EAAE,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,eAAe,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,EAAE;aAC/F,CAAC;YACF,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;YAC3F,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YACtF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YACpF,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YAChE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/elevenlabs-wrapper.test.d.ts

@@ -0,0 +1 @@
+export {};