openlit 1.13.0 → 1.14.0

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

@@ -0,0 +1,519 @@
+"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 wrapper_1 = __importDefault(require("../vllm/wrapper"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+jest.mock('../../../src/config');
+jest.mock('../../../src/helpers');
+jest.mock('../../../src/instrumentation/base-wrapper');
+const mockTracer = api_1.trace.getTracer('test-tracer');
+describe('VllmWrapper', () => {
+    let span;
+    beforeEach(() => {
+        span = mockTracer.startSpan('test-span');
+        span.setAttribute = jest.fn();
+        span.addEvent = jest.fn();
+        jest.clearAllMocks();
+        jest.restoreAllMocks();
+        wrapper_1.default.baseUrlPrefixes = [...wrapper_1.default.defaultBaseUrlPrefixes];
+    });
+    afterEach(() => {
+        span.end();
+    });
+    describe('isVllmClient / extractServerInfo', () => {
+        it('returns false for default OpenAI client (api.openai.com)', () => {
+            expect(wrapper_1.default.isVllmClient({ baseURL: 'https://api.openai.com/v1' })).toBe(false);
+        });
+        it('returns true for localhost:8000 vLLM endpoint', () => {
+            expect(wrapper_1.default.isVllmClient({ baseURL: 'http://127.0.0.1:8000/v1' })).toBe(true);
+        });
+        it('returns true for custom baseUrlPrefixes', () => {
+            wrapper_1.default.baseUrlPrefixes.push('http://gpu-cluster:8080/v1');
+            expect(wrapper_1.default.isVllmClient({ baseURL: 'http://gpu-cluster:8080/v1' })).toBe(true);
+        });
+        it('extracts server address and port from client baseURL', () => {
+            const info = wrapper_1.default.extractServerInfo({ baseURL: 'http://my-vllm:9000/v1' });
+            expect(info).toEqual({ address: 'my-vllm', port: 9000 });
+        });
+    });
+    describe('_patchChat routing', () => {
+        it('delegates non-vLLM clients to the OpenAI handler', async () => {
+            const openaiHandler = jest.fn().mockResolvedValue({ id: 'openai-resp' });
+            const rawCreate = jest.fn();
+            const patchFn = wrapper_1.default._patchChat(mockTracer, openaiHandler, rawCreate);
+            const wrapped = patchFn(jest.fn());
+            const client = { baseURL: 'https://api.openai.com/v1' };
+            const result = await wrapped.call(client, { model: 'gpt-4o', messages: [] });
+            expect(openaiHandler).toHaveBeenCalled();
+            expect(rawCreate).not.toHaveBeenCalled();
+            expect(result).toEqual({ id: 'openai-resp' });
+        });
+        it('creates a vLLM span for vLLM clients', async () => {
+            const openaiHandler = jest.fn();
+            const rawCreate = jest.fn().mockResolvedValue({
+                id: 'vllm-resp',
+                model: 'facebook/opt-125m',
+                choices: [{ message: { content: 'hi', role: 'assistant' }, finish_reason: 'stop' }],
+                usage: { prompt_tokens: 1, completion_tokens: 1 },
+            });
+            const patchFn = wrapper_1.default._patchChat(mockTracer, openaiHandler, rawCreate);
+            const wrapped = patchFn(jest.fn());
+            jest.spyOn(mockTracer, 'startSpan').mockReturnValue(span);
+            jest.spyOn(wrapper_1.default, '_chat').mockResolvedValue({ id: 'vllm-resp' });
+            const client = { baseURL: 'http://127.0.0.1:8000/v1' };
+            await wrapped.call(client, { model: 'facebook/opt-125m', messages: [{ role: 'user', content: 'hi' }] });
+            expect(openaiHandler).not.toHaveBeenCalled();
+            expect(rawCreate).toHaveBeenCalled();
+            expect(mockTracer.startSpan).toHaveBeenCalled();
+            expect(wrapper_1.default._chat).toHaveBeenCalled();
+        });
+    });
+    describe('_chat', () => {
+        it('should call recordMetrics after span ends', async () => {
+            const mockArgs = [{ messages: [{ role: 'user', content: 'test message' }] }];
+            const mockResponse = {
+                id: 'chatcmpl-123',
+                model: 'facebook/opt-125m',
+                usage: { prompt_tokens: 10, completion_tokens: 20 },
+                choices: [
+                    {
+                        message: { content: 'response text', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+            };
+            const mockGenAIEndpoint = 'vllm.chat';
+            jest
+                .spyOn(wrapper_1.default, '_chatCommonSetter')
+                .mockImplementationOnce(async ({ genAIEndpoint, span }) => {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, 1);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
+                return {
+                    genAIEndpoint,
+                    model: 'facebook/opt-125m',
+                    cost: 0.0,
+                    aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_VLLM,
+                    serverAddress: '127.0.0.1',
+                    serverPort: 8000,
+                };
+            });
+            await wrapper_1.default._chat({
+                args: mockArgs,
+                genAIEndpoint: mockGenAIEndpoint,
+                response: mockResponse,
+                span,
+                serverAddress: '127.0.0.1',
+                serverPort: 8000,
+            });
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledWith(span, {
+                genAIEndpoint: mockGenAIEndpoint,
+                model: 'facebook/opt-125m',
+                cost: 0.0,
+                aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_VLLM,
+                serverAddress: '127.0.0.1',
+                serverPort: 8000,
+            });
+        });
+    });
+    describe('_chatGenerator streaming', () => {
+        async function* mockStream() {
+            yield {
+                id: 'chatcmpl-stream',
+                model: 'facebook/opt-125m',
+                choices: [{
+                        delta: { role: 'assistant', content: 'Hello' },
+                        finish_reason: null,
+                    }],
+            };
+            yield {
+                id: 'chatcmpl-stream',
+                model: 'facebook/opt-125m',
+                choices: [{
+                        delta: {
+                            tool_calls: [{ index: 0, id: 'call_1', function: { name: 'get_weather', arguments: '{"loc' } }],
+                        },
+                        finish_reason: null,
+                    }],
+            };
+            yield {
+                id: 'chatcmpl-stream',
+                model: 'facebook/opt-125m',
+                choices: [{
+                        delta: {
+                            tool_calls: [{ index: 0, function: { arguments: '":"SF"}' } }],
+                        },
+                        finish_reason: 'tool_calls',
+                    }],
+                usage: { prompt_tokens: 5, completion_tokens: 10 },
+            };
+        }
+        it('aggregates streaming tool_calls across chunks', async () => {
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            const generator = wrapper_1.default._chatGenerator({
+                args: [{ model: 'facebook/opt-125m', messages: [{ role: 'user', content: 'weather?' }], stream: true }],
+                genAIEndpoint: 'vllm.chat',
+                response: mockStream(),
+                span,
+                serverAddress: '127.0.0.1',
+                serverPort: 8000,
+            });
+            // drain generator
+            // eslint-disable-next-line @typescript-eslint/no-unused-vars
+            for await (const _chunk of generator) { /* consume */ }
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_NAME, 'get_weather');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_CALL_ID, 'call_1');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_ARGS, '{"loc":"SF"}');
+        });
+    });
+    describe('_chatCommonSetter', () => {
+        it('should set span attributes and return metric parameters', async () => {
+            const mockArgs = [
+                {
+                    model: 'facebook/opt-125m',
+                    messages: [{ role: 'user', content: 'test message' }],
+                    max_tokens: 100,
+                    temperature: 0.7,
+                    top_p: 1,
+                    top_k: 40,
+                    presence_penalty: 2,
+                    frequency_penalty: 3,
+                    seed: 3,
+                    stream: false,
+                    stop: ['STOP'],
+                },
+            ];
+            const mockResult = {
+                id: 'chatcmpl-123',
+                model: 'facebook/opt-125m',
+                usage: { prompt_tokens: 10, completion_tokens: 20 },
+                choices: [
+                    {
+                        message: { content: 'response text', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.0);
+            const metricParams = await wrapper_1.default._chatCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'vllm.chat',
+                result: mockResult,
+                span,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, 1);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K, 40);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY, 2);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, 3);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_SEED, 3);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, ['STOP']);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, 'facebook/opt-125m');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 10);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 20);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, 30);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+            expect(metricParams).toEqual({
+                genAIEndpoint: 'vllm.chat',
+                model: 'facebook/opt-125m',
+                cost: 0.0,
+                aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_VLLM,
+                serverAddress: '127.0.0.1',
+                serverPort: 8000,
+            });
+        });
+        it('should record zero penalty values', async () => {
+            const mockArgs = [
+                {
+                    model: 'facebook/opt-125m',
+                    messages: [{ role: 'user', content: 'test' }],
+                    presence_penalty: 0,
+                    frequency_penalty: 0,
+                    stream: false,
+                },
+            ];
+            const mockResult = {
+                id: 'chatcmpl-zero',
+                model: 'facebook/opt-125m',
+                usage: { prompt_tokens: 1, completion_tokens: 1 },
+                choices: [{ message: { content: 'ok', role: 'assistant' }, finish_reason: 'stop' }],
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            await wrapper_1.default._chatCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'vllm.chat',
+                result: mockResult,
+                span,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY, 0);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, 0);
+        });
+        it('should NOT set sentinel values for optional request params', async () => {
+            const mockArgs = [
+                {
+                    model: 'facebook/opt-125m',
+                    messages: [{ role: 'user', content: 'test' }],
+                    stream: false,
+                },
+            ];
+            const mockResult = {
+                id: 'chatcmpl-456',
+                model: 'facebook/opt-125m',
+                usage: { prompt_tokens: 5, completion_tokens: 10 },
+                choices: [
+                    {
+                        message: { content: 'test response', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            await wrapper_1.default._chatCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'vllm.chat',
+                result: mockResult,
+                span,
+            });
+            const setAttrCalls = span.setAttribute.mock.calls;
+            const attrKeys = setAttrCalls.map((c) => c[0]);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_SEED);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_CHOICE_COUNT);
+        });
+        it('should set gen_ai.input.messages on span when capture enabled', async () => {
+            const mockArgs = [
+                {
+                    model: 'facebook/opt-125m',
+                    messages: [{ role: 'user', content: 'hello' }],
+                    stream: false,
+                },
+            ];
+            const mockResult = {
+                id: 'chatcmpl-input',
+                model: 'facebook/opt-125m',
+                usage: { prompt_tokens: 2, completion_tokens: 3 },
+                choices: [{ message: { content: 'hi', role: 'assistant' }, finish_reason: 'stop' }],
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            config_1.default.captureMessageContent = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            jest.spyOn(helpers_1.default, 'buildInputMessages').mockReturnValue('[{"role":"user","content":"hello"}]');
+            jest.spyOn(helpers_1.default, 'buildOutputMessages').mockReturnValue('[]');
+            await wrapper_1.default._chatCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'vllm.chat',
+                result: mockResult,
+                span,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, '[{"role":"user","content":"hello"}]');
+        });
+        it('should handle tool calls properly', async () => {
+            const mockArgs = [
+                {
+                    messages: [{ role: 'user', content: 'test message' }],
+                    tools: [{ type: 'function', function: { name: 'get_weather' } }],
+                },
+            ];
+            const mockResult = {
+                id: 'chatcmpl-123',
+                model: 'facebook/opt-125m',
+                usage: { prompt_tokens: 10, completion_tokens: 20 },
+                choices: [
+                    {
+                        message: {
+                            content: null,
+                            role: 'assistant',
+                            tool_calls: [
+                                {
+                                    id: 'call_123',
+                                    type: 'function',
+                                    function: { name: 'get_weather', arguments: '{"location":"SF"}' },
+                                },
+                            ],
+                        },
+                        finish_reason: 'tool_calls',
+                    },
+                ],
+            };
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.0);
+            await wrapper_1.default._chatCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'vllm.chat',
+                result: mockResult,
+                span,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_NAME, 'get_weather');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_CALL_ID, 'call_123');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_ARGS, '{"location":"SF"}');
+        });
+        it('should emit inference event when events not disabled', async () => {
+            const mockArgs = [
+                {
+                    model: 'facebook/opt-125m',
+                    messages: [{ role: 'user', content: 'test message' }],
+                    stream: false,
+                },
+            ];
+            const mockResult = {
+                id: 'chatcmpl-789',
+                model: 'facebook/opt-125m',
+                usage: { prompt_tokens: 10, completion_tokens: 20 },
+                choices: [
+                    {
+                        message: { content: 'response text', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.captureMessageContent = false;
+            config_1.default.disableEvents = false;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            jest.spyOn(helpers_1.default, 'emitInferenceEvent').mockImplementation(() => { });
+            await wrapper_1.default._chatCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'vllm.chat',
+                result: mockResult,
+                span,
+                serverAddress: 'my-vllm',
+                serverPort: 9000,
+            });
+            expect(helpers_1.default.emitInferenceEvent).toHaveBeenCalledWith(span, expect.objectContaining({
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+                [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: 'facebook/opt-125m',
+                [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: 'facebook/opt-125m',
+                [semantic_convention_1.default.SERVER_ADDRESS]: 'my-vllm',
+                [semantic_convention_1.default.SERVER_PORT]: 9000,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_ID]: 'chatcmpl-789',
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: 10,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: 20,
+            }));
+        });
+        it('falls back to token estimation when usage is missing', async () => {
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            jest.spyOn(helpers_1.default, 'openaiTokens').mockReturnValue(7);
+            await wrapper_1.default._chatCommonSetter({
+                args: [{ model: 'facebook/opt-125m', messages: [{ role: 'user', content: 'hi' }] }],
+                genAIEndpoint: 'vllm.chat',
+                result: {
+                    id: 'no-usage',
+                    model: 'facebook/opt-125m',
+                    choices: [{ message: { content: 'hello', role: 'assistant' }, finish_reason: 'stop' }],
+                },
+                span,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 7);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 7);
+        });
+    });
+    describe('Cross-Language Trace Comparison', () => {
+        it('should use vllm as provider name (matching Python SDK)', () => {
+            expect(wrapper_1.default.aiSystem).toBe('vllm');
+            expect(wrapper_1.default.aiSystem).toBe(semantic_convention_1.default.GEN_AI_SYSTEM_VLLM);
+        });
+        it('should use port 8000 as default (matching vLLM default server port)', () => {
+            expect(wrapper_1.default.serverPort).toBe(8000);
+        });
+        it('should set same attributes as Python SDK for chat completion', async () => {
+            const mockArgs = [
+                {
+                    messages: [{ role: 'user', content: 'What is vLLM?' }],
+                    model: 'facebook/opt-125m',
+                    max_tokens: 50,
+                    temperature: 0.7,
+                    stream: false,
+                },
+            ];
+            const mockResponse = {
+                id: 'chatcmpl-test-id',
+                model: 'facebook/opt-125m',
+                choices: [
+                    {
+                        index: 0,
+                        finish_reason: 'stop',
+                        message: { role: 'assistant', content: 'vLLM is...' },
+                    },
+                ],
+                usage: {
+                    prompt_tokens: 8,
+                    completion_tokens: 15,
+                },
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.captureMessageContent = true;
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.0);
+            jest.spyOn(helpers_1.default, 'buildInputMessages').mockReturnValue('[]');
+            jest.spyOn(helpers_1.default, 'buildOutputMessages').mockReturnValue('[]');
+            await wrapper_1.default._chat({
+                args: mockArgs,
+                genAIEndpoint: 'vllm.chat',
+                response: mockResponse,
+                span,
+                serverAddress: '127.0.0.1',
+                serverPort: 8000,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 8);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 15);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, 'facebook/opt-125m');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 50);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, '[]');
+            const setAttrCalls = span.setAttribute.mock.calls;
+            const attrKeys = setAttrCalls.map((c) => c[0]);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS);
+        });
+    });
+});
+describe('OpenlitVllmInstrumentation patch targets', () => {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const VllmInstrumentation = require('../vllm').default;
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const { isWrapped } = require('@opentelemetry/instrumentation');
+    it('wraps OpenAI.Chat.Completions.prototype.create once with vLLM routing', () => {
+        const rawCreate = jest.fn();
+        const fakeOpenAiModule = {
+            OpenAI: function () { },
+        };
+        fakeOpenAiModule.OpenAI.Chat = { Completions: function () { } };
+        fakeOpenAiModule.OpenAI.Chat.Completions.prototype.create = rawCreate;
+        const instr = new VllmInstrumentation();
+        instr.manualPatch(fakeOpenAiModule);
+        expect(isWrapped(fakeOpenAiModule.OpenAI.Chat.Completions.prototype.create)).toBe(true);
+        expect(fakeOpenAiModule.OpenAI.Chat.Completions.prototype.create).not.toBe(rawCreate);
+    });
+});
+//# sourceMappingURL=vllm-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"vllm-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/vllm-trace-comparison.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,8DAA0C;AAC1C,0DAAyC;AACzC,4DAA0C;AAC1C,mEAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AACjC,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAClC,IAAI,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;AAEvD,MAAM,UAAU,GAAG,WAAK,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;AAElD,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,IAAU,CAAC;IAEf,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC1B,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,iBAAW,CAAC,eAAe,GAAG,CAAC,GAAG,iBAAW,CAAC,sBAAsB,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,CAAC,iBAAW,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,2BAA2B,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,iBAAW,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,0BAA0B,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,iBAAW,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;YAC/D,MAAM,CAAC,iBAAW,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,4BAA4B,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,IAAI,GAAG,iBAAW,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,wBAAwB,EAAE,CAAC,CAAC;YAClF,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC;YACzE,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,iBAAW,CAAC,UAAU,CAAC,UAAU,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;YAC7E,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAEnC,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,2BAA2B,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAE7E,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACzC,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC5C,EAAE,EAAE,WAAW;gBACf,KAAK,EAAE,mBAAmB;gBAC1B,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;gBACnF,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aAClD,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,iBAAW,CAAC,UAAU,CAAC,UAAU,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;YAC7E,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAEnC,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,KAAK,CAAC,iBAAW,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;YAExE,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,0BAA0B,EAAE,CAAC;YACvD,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,mBAAmB,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAExG,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC7C,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACrC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAChD,MAAM,CAAC,iBAAW,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7E,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,cAAc;gBAClB,KAAK,EAAE,mBAAmB;gBAC1B,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE;gBACnD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,WAAW,EAAE;wBACxD,aAAa,EAAE,MAAM;qBACtB;iBACF;aACF,CAAC;YACF,MAAM,iBAAiB,GAAG,WAAW,CAAC;YAEtC,IAAI;iBACD,KAAK,CAAC,iBAAW,EAAE,mBAAmB,CAAC;iBACvC,sBAAsB,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,EAAE;gBACxD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,CAAC,CAAC,CAAC;gBAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;gBAEtE,OAAO;oBACL,aAAa;oBACb,KAAK,EAAE,mBAAmB;oBAC1B,IAAI,EAAE,GAAG;oBACT,QAAQ,EAAE,6BAAkB,CAAC,kBAAkB;oBAC/C,aAAa,EAAE,WAAW;oBAC1B,UAAU,EAAE,IAAI;iBACjB,CAAC;YACJ,CAAC,CAAC,CAAC;YAEL,MAAM,iBAAW,CAAC,KAAK,CAAC;gBACtB,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,iBAAiB;gBAChC,QAAQ,EAAE,YAAY;gBACtB,IAAI;gBACJ,aAAa,EAAE,WAAW;gBAC1B,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE;gBAC3D,aAAa,EAAE,iBAAiB;gBAChC,KAAK,EAAE,mBAAmB;gBAC1B,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,6BAAkB,CAAC,kBAAkB;gBAC/C,aAAa,EAAE,WAAW;gBAC1B,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,KAAK,SAAS,CAAC,CAAC,UAAU;YACxB,MAAM;gBACJ,EAAE,EAAE,iBAAiB;gBACrB,KAAK,EAAE,mBAAmB;gBAC1B,OAAO,EAAE,CAAC;wBACR,KAAK,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;wBAC9C,aAAa,EAAE,IAAI;qBACpB,CAAC;aACH,CAAC;YACF,MAAM;gBACJ,EAAE,EAAE,iBAAiB;gBACrB,KAAK,EAAE,mBAAmB;gBAC1B,OAAO,EAAE,CAAC;wBACR,KAAK,EAAE;4BACL,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,CAAC;yBAChG;wBACD,aAAa,EAAE,IAAI;qBACpB,CAAC;aACH,CAAC;YACF,MAAM;gBACJ,EAAE,EAAE,iBAAiB;gBACrB,KAAK,EAAE,mBAAmB;gBAC1B,OAAO,EAAE,CAAC;wBACR,KAAK,EAAE;4BACL,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;yBAC/D;wBACD,aAAa,EAAE,YAAY;qBAC5B,CAAC;gBACF,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE;aACnD,CAAC;QACJ,CAAC;QAED,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,IAAI,CAAC,eAAe,EAAE,CAAC;YACtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEjE,MAAM,SAAS,GAAG,iBAAW,CAAC,cAAc,CAAC;gBAC3C,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;gBACvG,aAAa,EAAE,WAAW;gBAC1B,QAAQ,EAAE,UAAU,EAAE;gBACtB,IAAI;gBACJ,aAAa,EAAE,WAAW;gBAC1B,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,kBAAkB;YAClB,6DAA6D;YAC7D,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,SAAS,EAAE,CAAC,CAAC,aAAa,CAAC,CAAC;YAEvD,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;YACnG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,QAAQ,CAAC,CAAC;YACjG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,gBAAgB,EACnC,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,mBAAmB;oBAC1B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;oBACrD,UAAU,EAAE,GAAG;oBACf,WAAW,EAAE,GAAG;oBAChB,KAAK,EAAE,CAAC;oBACR,KAAK,EAAE,EAAE;oBACT,gBAAgB,EAAE,CAAC;oBACnB,iBAAiB,EAAE,CAAC;oBACpB,IAAI,EAAE,CAAC;oBACP,MAAM,EAAE,KAAK;oBACb,IAAI,EAAE,CAAC,MAAM,CAAC;iBACf;aACF,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,cAAc;gBAClB,KAAK,EAAE,mBAAmB;gBAC1B,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE;gBACnD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,WAAW,EAAE;wBACxD,aAAa,EAAE,MAAM;qBACtB;iBACF;aACF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAEnE,MAAM,YAAY,GAAG,MAAM,iBAAW,CAAC,iBAAiB,CAAC;gBACvD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,CAAC,CAAC,CAAC;YAC3F,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;YAC5F,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YAClG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACnG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;YACtG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;YACvG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;YAC1F,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YACnG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3G,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,mBAAmB,CAAC,CAAC;YAC9G,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YACjG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;YAClG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YACjG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAE3G,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBAC3B,aAAa,EAAE,WAAW;gBAC1B,KAAK,EAAE,mBAAmB;gBAC1B,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,6BAAkB,CAAC,kBAAkB;gBAC/C,aAAa,EAAE,WAAW;gBAC1B,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,mBAAmB;oBAC1B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;oBAC7C,gBAAgB,EAAE,CAAC;oBACnB,iBAAiB,EAAE,CAAC;oBACpB,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,eAAe;gBACnB,KAAK,EAAE,mBAAmB;gBAC1B,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;gBACjD,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;aACpF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YACtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEjE,MAAM,iBAAW,CAAC,iBAAiB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;YACtG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACzG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,mBAAmB;oBAC1B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;oBAC7C,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,cAAc;gBAClB,KAAK,EAAE,mBAAmB;gBAC1B,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE;gBAClD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,WAAW,EAAE;wBACxD,aAAa,EAAE,MAAM;qBACtB;iBACF;aACF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEjE,MAAM,iBAAW,CAAC,iBAAiB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,YAAY,GAAI,IAAI,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC;YACjE,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAEtD,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;YAC7E,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC;YACvE,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,+BAA+B,CAAC,CAAC;YACnF,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,gCAAgC,CAAC,CAAC;YACpF,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,CAAC;YACjF,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,CAAC;QACjF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,mBAAmB;oBAC1B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;oBAC9C,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,gBAAgB;gBACpB,KAAK,EAAE,mBAAmB;gBAC1B,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;gBACjD,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;aACpF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YACtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3C,gBAAqB,CAAC,qBAAqB,GAAG,IAAI,CAAC;YACpD,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACjE,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,qCAAqC,CAAC,CAAC;YACvG,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAEvE,MAAM,iBAAW,CAAC,iBAAiB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,qBAAqB,EACxC,qCAAqC,CACtC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,GAAG;gBACf;oBACE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;oBACrD,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,CAAC;iBACjE;aACF,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,cAAc;gBAClB,KAAK,EAAE,mBAAmB;gBAC1B,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE;gBACnD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE;4BACP,OAAO,EAAE,IAAI;4BACb,IAAI,EAAE,WAAW;4BACjB,UAAU,EAAE;gCACV;oCACE,EAAE,EAAE,UAAU;oCACd,IAAI,EAAE,UAAU;oCAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,mBAAmB,EAAE;iCAClE;6BACF;yBACF;wBACD,aAAa,EAAE,YAAY;qBAC5B;iBACF;aACF,CAAC;YAED,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAEnE,MAAM,iBAAW,CAAC,iBAAiB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;YACnG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;YACnG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;QAC3G,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,mBAAmB;oBAC1B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;oBACrD,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,cAAc;gBAClB,KAAK,EAAE,mBAAmB;gBAC1B,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE;gBACnD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,WAAW,EAAE;wBACxD,aAAa,EAAE,MAAM;qBACtB;iBACF;aACF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACpD,gBAAqB,CAAC,aAAa,GAAG,KAAK,CAAC;YAC7C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACjE,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,oBAAoB,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAE7E,MAAM,iBAAW,CAAC,iBAAiB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE,UAAU;gBAClB,IAAI;gBACJ,aAAa,EAAE,SAAS;gBACxB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAa,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAC3D,IAAI,EACJ,MAAM,CAAC,gBAAgB,CAAC;gBACtB,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;gBACpF,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,mBAAmB;gBAC9D,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,mBAAmB;gBAC/D,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,SAAS;gBAC9C,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,IAAI;gBACtC,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,cAAc;gBACvD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,EAAE;gBAClD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,EAAE;aACpD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,IAAI,CAAC,eAAe,EAAE,CAAC;YACtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACjE,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE7D,MAAM,iBAAW,CAAC,iBAAiB,CAAC;gBAClC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;gBACnF,aAAa,EAAE,WAAW;gBAC1B,MAAM,EAAE;oBACN,EAAE,EAAE,UAAU;oBACd,KAAK,EAAE,mBAAmB;oBAC1B,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;iBACvF;gBACD,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YAChG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,CAAC,CAAC,CAAC;QACnG,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,CAAC,iBAAW,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,iBAAW,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,CAAC,iBAAW,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG;gBACf;oBACE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;oBACtD,KAAK,EAAE,mBAAmB;oBAC1B,UAAU,EAAE,EAAE;oBACd,WAAW,EAAE,GAAG;oBAChB,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,kBAAkB;gBACtB,KAAK,EAAE,mBAAmB;gBAC1B,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC;wBACR,aAAa,EAAE,MAAM;wBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,EAAE;qBACtD;iBACF;gBACD,KAAK,EAAE;oBACL,aAAa,EAAE,CAAC;oBAChB,iBAAiB,EAAE,EAAE;iBACtB;aACF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,qBAAqB,GAAG,IAAI,CAAC;YACnD,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YACnE,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACtE,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAEvE,MAAM,iBAAW,CAAC,KAAK,CAAC;gBACtB,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,WAAW;gBAC1B,QAAQ,EAAE,YAAY;gBACtB,IAAI;gBACJ,aAAa,EAAE,WAAW;gBAC1B,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YAChG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;YAClG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,mBAAmB,CAAC,CAAC;YAC9G,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3G,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACnG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YACjG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;YAE/F,MAAM,YAAY,GAAI,IAAI,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC;YACjE,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;QAC/E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxD,8DAA8D;IAC9D,MAAM,mBAAmB,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC;IACvD,8DAA8D;IAC9D,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC;IAEhE,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC5B,MAAM,gBAAgB,GAAG;YACvB,MAAM,EAAE,cAAa,CAAC;SAChB,CAAC;QACT,gBAAgB,CAAC,MAAM,CAAC,IAAI,GAAG,EAAE,WAAW,EAAE,cAAa,CAAC,EAAE,CAAC;QAC/D,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,GAAG,SAAS,CAAC;QAEtE,MAAM,KAAK,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACxC,KAAK,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC;QAEpC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxF,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACxF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

package/dist/instrumentation/ai21/index.js

@@ -0,0 +1,67 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+class OpenlitAI21Instrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-ai21`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('ai21', ['>=1.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(ai21) {
+        this._patch(ai21);
+    }
+    _patch(moduleExports) {
+        try {
+            // AI21 exposes chat completions as `Completions.prototype.create` (a flat
+            // top-level export, unlike groq-sdk's nested `Groq.Chat.Completions`).
+            // Note: ai21's CJS bundle does not re-export its classes, so under CommonJS
+            // `moduleExports` is empty and the guard below makes this a safe no-op. The
+            // patch takes effect when the SDK is loaded as ESM (OTel's import hook) or
+            // when the module is supplied via `manualPatch`.
+            if (!moduleExports?.Completions?.prototype?.create) {
+                return;
+            }
+            if ((0, instrumentation_1.isWrapped)(moduleExports.Completions.prototype.create)) {
+                this._unwrap(moduleExports.Completions.prototype, 'create');
+            }
+            this._wrap(moduleExports.Completions.prototype, 'create', wrapper_1.default._patchChatCompletionCreate(this.tracer));
+            // Conversational RAG is exported as `ConversationalRag` from ai21 and maps
+            // to `ConversationalRag.prototype.create` (mirrors the Python SDK's
+            // StudioConversationalRag.create). Guarded the same way as Completions so
+            // it is a safe no-op when the SDK is loaded as CommonJS.
+            if (moduleExports?.ConversationalRag?.prototype?.create) {
+                if ((0, instrumentation_1.isWrapped)(moduleExports.ConversationalRag.prototype.create)) {
+                    this._unwrap(moduleExports.ConversationalRag.prototype, 'create');
+                }
+                this._wrap(moduleExports.ConversationalRag.prototype, 'create', wrapper_1.default._patchConversationalRagCreate(this.tracer));
+            }
+        }
+        catch (e) {
+            console.error('Error in _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        if (moduleExports?.Completions?.prototype?.create) {
+            this._unwrap(moduleExports.Completions.prototype, 'create');
+        }
+        if (moduleExports?.ConversationalRag?.prototype?.create) {
+            this._unwrap(moduleExports.ConversationalRag.prototype, 'create');
+        }
+    }
+}
+exports.default = OpenlitAI21Instrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/ai21/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAoC;AAIpC,MAAqB,0BAA2B,SAAQ,qCAAmB;IACzE,YAAY,SAAoC,EAAE;QAChD,KAAK,CAAC,GAAG,iCAAsB,uBAAuB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,MAAM,EACN,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,IAAS;QAC1B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,0EAA0E;YAC1E,uEAAuE;YACvE,4EAA4E;YAC5E,4EAA4E;YAC5E,2EAA2E;YAC3E,iDAAiD;YACjD,IAAI,CAAC,aAAa,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;gBACnD,OAAO;YACT,CAAC;YAED,IAAI,IAAA,2BAAS,EAAC,aAAa,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1D,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAC9D,CAAC;YAED,IAAI,CAAC,KAAK,CACR,aAAa,CAAC,WAAW,CAAC,SAAS,EACnC,QAAQ,EACR,iBAAW,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CACpD,CAAC;YAEF,2EAA2E;YAC3E,oEAAoE;YACpE,0EAA0E;YAC1E,yDAAyD;YACzD,IAAI,aAAa,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;gBACxD,IAAI,IAAA,2BAAS,EAAC,aAAa,CAAC,iBAAiB,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;oBAChE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,iBAAiB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACpE,CAAC;gBAED,IAAI,CAAC,KAAK,CACR,aAAa,CAAC,iBAAiB,CAAC,SAAS,EACzC,QAAQ,EACR,iBAAW,CAAC,6BAA6B,CAAC,IAAI,CAAC,MAAM,CAAC,CACvD,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,aAAa,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;YAClD,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,aAAa,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;YACxD,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,iBAAiB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;CACF;AA5ED,6CA4EC"}