shield360 1.0.0

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

@@ -0,0 +1,345 @@
+"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("../mistral/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('MistralWrapper', () => {
+    let span;
+    beforeEach(() => {
+        span = mockTracer.startSpan('test-span');
+        span.setAttribute = jest.fn();
+        span.addEvent = jest.fn();
+        jest.clearAllMocks();
+    });
+    afterEach(() => {
+        span.end();
+    });
+    describe('_chatCompletion', () => {
+        it('should call recordMetrics after span ends', async () => {
+            const mockArgs = [{ messages: [{ role: 'user', content: 'test message' }] }];
+            const mockResponse = {
+                id: '123',
+                model: 'mistral-small-latest',
+                usage: { prompt_tokens: 10, completion_tokens: 20, total_tokens: 30 },
+                choices: [
+                    {
+                        message: { content: 'response text', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+            };
+            const mockGenAIEndpoint = 'mistral.chat.completions';
+            jest
+                .spyOn(wrapper_1.default, '_chatCompletionCommonSetter')
+                .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: 'mistral-small-latest',
+                    user: 'test-user',
+                    cost: 0.5,
+                    aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_MISTRAL,
+                };
+            });
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: mockGenAIEndpoint,
+                response: mockResponse,
+                span,
+            });
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledWith(span, {
+                genAIEndpoint: mockGenAIEndpoint,
+                model: 'mistral-small-latest',
+                user: 'test-user',
+                cost: 0.5,
+                aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_MISTRAL,
+            });
+        });
+    });
+    describe('_chatCompletionCommonSetter', () => {
+        it('should set span attributes and return metric parameters', async () => {
+            const mockArgs = [
+                {
+                    model: 'mistral-small-latest',
+                    messages: [{ role: 'user', content: 'test message' }],
+                    max_tokens: 100,
+                    temperature: 0.7,
+                    top_p: 1,
+                    user: 'test-user',
+                    presence_penalty: 2,
+                    frequency_penalty: 3,
+                    seed: 3,
+                    stream: false,
+                    stop: ['STOP'],
+                },
+            ];
+            const mockResult = {
+                id: '123',
+                model: 'mistral-small-latest',
+                usage: {
+                    prompt_tokens: 10,
+                    completion_tokens: 20,
+                    total_tokens: 30,
+                },
+                choices: [
+                    {
+                        message: { content: 'response text', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+            };
+            const mockGenAIEndpoint = 'mistral.chat.completions';
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.5);
+            const metricParams = await wrapper_1.default._chatCompletionCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: mockGenAIEndpoint,
+                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_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, 'mistral-small-latest');
+            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_RESPONSE_FINISH_REASON, ['stop']);
+            expect(metricParams).toEqual({
+                genAIEndpoint: mockGenAIEndpoint,
+                model: 'mistral-small-latest',
+                user: 'test-user',
+                cost: 0.5,
+                aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_MISTRAL,
+            });
+        });
+        it('should NOT set sentinel values for optional request params', async () => {
+            const mockArgs = [
+                {
+                    model: 'mistral-small-latest',
+                    messages: [{ role: 'user', content: 'test' }],
+                    stream: false,
+                },
+            ];
+            const mockResult = {
+                id: '456',
+                model: 'mistral-small-latest',
+                usage: { prompt_tokens: 5, completion_tokens: 10, total_tokens: 15 },
+                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._chatCompletionCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'mistral.chat.completions',
+                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 handle tool calls properly', async () => {
+            const mockArgs = [
+                {
+                    messages: [{ role: 'user', content: 'test message' }],
+                    tools: [{ type: 'function', function: { name: 'get_weather' } }],
+                },
+            ];
+            const mockResult = {
+                id: '123',
+                model: 'mistral-small-latest',
+                usage: { prompt_tokens: 10, completion_tokens: 20, total_tokens: 30 },
+                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.5);
+            await wrapper_1.default._chatCompletionCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'mistral.chat.completions',
+                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: 'mistral-small-latest',
+                    messages: [{ role: 'user', content: 'test message' }],
+                    stream: false,
+                },
+            ];
+            const mockResult = {
+                id: '789',
+                model: 'mistral-small-latest',
+                usage: { prompt_tokens: 10, completion_tokens: 20, total_tokens: 30 },
+                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._chatCompletionCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'mistral.chat.completions',
+                result: mockResult,
+                span,
+            });
+            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]: 'mistral-small-latest',
+                [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: 'mistral-small-latest',
+                [semantic_convention_1.default.SERVER_ADDRESS]: 'api.mistral.ai',
+                [semantic_convention_1.default.SERVER_PORT]: 443,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_ID]: '789',
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: 10,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: 20,
+            }));
+        });
+    });
+    describe('Cross-Language Trace Comparison', () => {
+        it('should use mistral_ai as provider name (matching Python SDK)', () => {
+            expect(wrapper_1.default.aiSystem).toBe('mistral_ai');
+            expect(wrapper_1.default.aiSystem).toBe(semantic_convention_1.default.GEN_AI_SYSTEM_MISTRAL);
+        });
+        it('should set same attributes as Python SDK for chat completion', async () => {
+            const mockArgs = [
+                {
+                    messages: [{ role: 'user', content: 'What is Mistral AI?' }],
+                    model: 'mistral-small-latest',
+                    max_tokens: 50,
+                    temperature: 0.7,
+                    stream: false,
+                },
+            ];
+            const mockResponse = {
+                id: 'mistral-test-id',
+                created: Date.now(),
+                model: 'mistral-small-latest',
+                choices: [
+                    {
+                        index: 0,
+                        finish_reason: 'stop',
+                        message: { role: 'assistant', content: 'Mistral AI is...' },
+                    },
+                ],
+                usage: {
+                    prompt_tokens: 8,
+                    completion_tokens: 15,
+                    total_tokens: 23,
+                },
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.captureMessageContent = true;
+            config_1.default.disableEvents = true;
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.001);
+            jest.spyOn(helpers_1.default, 'buildInputMessages').mockReturnValue('[]');
+            jest.spyOn(helpers_1.default, 'buildOutputMessages').mockReturnValue('[]');
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'mistral.chat.completions',
+                response: mockResponse,
+                span,
+            });
+            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, 'mistral-small-latest');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+            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);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE);
+        });
+        it('should set embedding attributes matching Python SDK', async () => {
+            const mockArgs = [
+                {
+                    model: 'mistral-embed',
+                    input: 'Test embedding text',
+                },
+            ];
+            const mockResponse = {
+                model: 'mistral-embed',
+                data: [{ embedding: [0.1, 0.2, 0.3] }],
+                usage: {
+                    prompt_tokens: 3,
+                    total_tokens: 3,
+                },
+            };
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            config_1.default.captureMessageContent = true;
+            jest.spyOn(helpers_1.default, 'getEmbedModelCost').mockReturnValue(0.0001);
+            const mockTracer = {
+                startSpan: jest.fn().mockReturnValue(span),
+            };
+            const patchMethod = wrapper_1.default._patchEmbedding(mockTracer);
+            const wrappedMethod = patchMethod(async () => mockResponse);
+            await wrappedMethod.call({}, ...mockArgs);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 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_ENCODING_FORMATS, ['float']);
+            const setAttrCalls = span.setAttribute.mock.calls;
+            const attrKeys = setAttrCalls.map((c) => c[0]);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_SERVER_TTFT);
+            expect(attrKeys).not.toContain(semantic_convention_1.default.GEN_AI_SERVER_TBT);
+        });
+    });
+});
+//# sourceMappingURL=mistral-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"mistral-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/mistral-trace-comparison.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,iEAAgD;AAChD,0DAA2C;AAC3C,4DAA4C;AAC5C,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,gBAAgB,EAAE,GAAG,EAAE;IAC9B,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;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,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,KAAK;gBACT,KAAK,EAAE,sBAAsB;gBAC7B,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACrE,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,0BAA0B,CAAC;YACrD,IAAI;iBACD,KAAK,CAAC,iBAAc,EAAE,6BAA6B,CAAC;iBACpD,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,sBAAsB;oBAC7B,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,GAAG;oBACT,QAAQ,EAAE,6BAAkB,CAAC,qBAAqB;iBACnD,CAAC;YACJ,CAAC,CAAC,CAAC;YAEL,MAAM,iBAAc,CAAC,eAAe,CAAC;gBACnC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,iBAAiB;gBAChC,QAAQ,EAAE,YAAY;gBACtB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE;gBAC3D,aAAa,EAAE,iBAAiB;gBAChC,KAAK,EAAE,sBAAsB;gBAC7B,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,6BAAkB,CAAC,qBAAqB;aACnD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;QAC3C,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,sBAAsB;oBAC7B,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,IAAI,EAAE,WAAW;oBACjB,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,KAAK;gBACT,KAAK,EAAE,sBAAsB;gBAC7B,KAAK,EAAE;oBACL,aAAa,EAAE,EAAE;oBACjB,iBAAiB,EAAE,EAAE;oBACrB,YAAY,EAAE,EAAE;iBACjB;gBACD,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,0BAA0B,CAAC;YAErD,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YACzC,gBAAuB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAErE,MAAM,YAAY,GAAG,MAAM,iBAAc,CAAC,2BAA2B,CAAC;gBACpE,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,iBAAiB;gBAChC,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,CAC5C,6BAAkB,CAAC,yBAAyB,EAC5C,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,0BAA0B,EAC7C,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,+BAA+B,EAClD,CAAC,CACF,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,gCAAgC,EACnD,CAAC,CACF,CAAC;YACF,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,CAC5C,6BAAkB,CAAC,wBAAwB,EAC3C,KAAK,CACN,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,MAAM,CAAC,CACT,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,qBAAqB,EACxC,sBAAsB,CACvB,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,yBAAyB,EAC5C,EAAE,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,0BAA0B,EAC7C,EAAE,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,MAAM,CAAC,CACT,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBAC3B,aAAa,EAAE,iBAAiB;gBAChC,KAAK,EAAE,sBAAsB;gBAC7B,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,6BAAkB,CAAC,qBAAqB;aACnD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,sBAAsB;oBAC7B,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,KAAK;gBACT,KAAK,EAAE,sBAAsB;gBAC7B,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACpE,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,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YACzC,gBAAuB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEnE,MAAM,iBAAc,CAAC,2BAA2B,CAAC;gBAC/C,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,0BAA0B;gBACzC,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,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,KAAK;gBACT,KAAK,EAAE,sBAAsB;gBAC7B,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACrE,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,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YACzC,gBAAuB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAErE,MAAM,iBAAc,CAAC,2BAA2B,CAAC;gBAC/C,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,0BAA0B;gBACzC,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,gBAAgB,EACnC,aAAa,CACd,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,mBAAmB,EACtC,UAAU,CACX,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,gBAAgB,EACnC,mBAAmB,CACpB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,sBAAsB;oBAC7B,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,KAAK;gBACT,KAAK,EAAE,sBAAsB;gBAC7B,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACrE,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,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YACzC,gBAAuB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACtD,gBAAuB,CAAC,aAAa,GAAG,KAAK,CAAC;YAC/C,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACnE,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,oBAAoB,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAE/E,MAAM,iBAAc,CAAC,2BAA2B,CAAC;gBAC/C,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,0BAA0B;gBACzC,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAe,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAC7D,IAAI,EACJ,MAAM,CAAC,gBAAgB,CAAC;gBACtB,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;gBACpF,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,sBAAsB;gBACjE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,sBAAsB;gBAClE,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,gBAAgB;gBACrD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,GAAG;gBACrC,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,KAAK;gBAC9C,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,EAAE;gBAClD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,EAAE;aACpD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,CAAC,iBAAc,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnD,MAAM,CAAC,iBAAc,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC;QACjF,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,qBAAqB,EAAE,CAAC;oBAC5D,KAAK,EAAE,sBAAsB;oBAC7B,UAAU,EAAE,EAAE;oBACd,WAAW,EAAE,GAAG;oBAChB,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,iBAAiB;gBACrB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;gBACnB,KAAK,EAAE,sBAAsB;gBAC7B,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC;wBACR,aAAa,EAAE,MAAM;wBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,kBAAkB,EAAE;qBAC5D;iBACF;gBACD,KAAK,EAAE;oBACL,aAAa,EAAE,CAAC;oBAChB,iBAAiB,EAAE,EAAE;oBACrB,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YACzC,gBAAuB,CAAC,qBAAqB,GAAG,IAAI,CAAC;YACrD,gBAAuB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACvE,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACxE,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAEzE,MAAM,iBAAc,CAAC,eAAe,CAAC;gBACnC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,0BAA0B;gBACzC,QAAQ,EAAE,YAAY;gBACtB,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,EAAE,CAAC,CAAC;YAClG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,sBAAsB,CAAC,CAAC;YACjH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAE3G,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;YAC7E,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;QAC/E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,eAAe;oBACtB,KAAK,EAAE,qBAAqB;iBAC7B;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE;oBACL,aAAa,EAAE,CAAC;oBAChB,YAAY,EAAE,CAAC;iBAChB;aACF,CAAC;YAEF,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YACzC,gBAAuB,CAAC,qBAAqB,GAAG,IAAI,CAAC;YACtD,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,mBAAmB,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAEzE,MAAM,UAAU,GAAQ;gBACtB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;aAC3C,CAAC;YAEF,MAAM,WAAW,GAAG,iBAAc,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YAC/D,MAAM,aAAa,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE,CAAC,YAAY,CAAC,CAAC;YAE5D,MAAM,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,QAAQ,CAAC,CAAC;YAE1C,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,wBAAwB,EAAE,KAAK,CAAC,CAAC;YACnG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YAE9G,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,kBAAkB,CAAC,CAAC;YACtE,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

package/dist/instrumentation/__tests__/openai-wrapper.test.js

@@ -0,0 +1,179 @@
+"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("../openai/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('OpenAIWrapper', () => {
+    let span;
+    beforeEach(() => {
+        span = mockTracer.startSpan('test-span');
+        span.setAttribute = jest.fn();
+        span.addEvent = jest.fn();
+        jest.clearAllMocks();
+    });
+    afterEach(() => {
+        span.end();
+    });
+    describe('_chatCompletion', () => {
+        it('should call recordMetrics after span ends', async () => {
+            const mockArgs = [{ messages: [{ role: 'user', content: 'test message' }] }];
+            const mockResponse = {
+                id: '123',
+                model: 'gpt-3.5-turbo',
+                usage: { prompt_tokens: 10, completion_tokens: 20, total_tokens: 30 },
+                choices: [
+                    {
+                        message: { content: 'response text', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+            };
+            const mockGenAIEndpoint = 'openai.resources.chat.completions';
+            jest
+                .spyOn(wrapper_1.default, '_chatCompletionCommonSetter')
+                .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: 'gpt-3.5-turbo',
+                    user: 'test-user',
+                    cost: 0.5,
+                    aiSystem: 'openai',
+                };
+            });
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: mockGenAIEndpoint,
+                response: mockResponse,
+                span,
+            });
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledWith(span, {
+                genAIEndpoint: mockGenAIEndpoint,
+                model: 'gpt-3.5-turbo',
+                user: 'test-user',
+                cost: 0.5,
+                aiSystem: 'openai',
+            });
+        });
+    });
+    describe('_chatCompletionCommonSetter', () => {
+        it('should set span attributes and return metric parameters', async () => {
+            const mockArgs = [
+                {
+                    model: 'gpt-3.5-turbo',
+                    messages: [{ role: 'user', content: 'test message' }],
+                    max_tokens: 100,
+                    temperature: 0.7,
+                    top_p: 1,
+                    user: 'test-user',
+                    presence_penalty: 2,
+                    frequency_penalty: 3,
+                    seed: 3,
+                    stream: false,
+                    stop: ['STOP'],
+                },
+            ];
+            const mockResult = {
+                id: '123',
+                model: 'gpt-3.5-turbo',
+                usage: {
+                    prompt_tokens: 10,
+                    completion_tokens: 20,
+                    total_tokens: 30,
+                    completion_tokens_details: { reasoning_tokens: 5 },
+                },
+                choices: [
+                    {
+                        message: { content: 'response text', role: 'assistant' },
+                        finish_reason: 'stop',
+                    },
+                ],
+                system_fingerprint: 'fp_test',
+                service_tier: 'default',
+            };
+            const mockGenAIEndpoint = 'openai.resources.chat.completions';
+            jest.restoreAllMocks();
+            config_1.default.pricingInfo = {};
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.5);
+            const metricParams = await wrapper_1.default._chatCompletionCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: mockGenAIEndpoint,
+                result: mockResult,
+                span,
+            });
+            // Basic request parameters
+            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_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);
+            // New attributes
+            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, 'gpt-3.5-turbo');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.OPENAI_RESPONSE_SYSTEM_FINGERPRINT, 'fp_test');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.OPENAI_RESPONSE_SERVICE_TIER, 'default');
+            expect(metricParams).toEqual({
+                genAIEndpoint: mockGenAIEndpoint,
+                model: 'gpt-3.5-turbo',
+                user: 'test-user',
+                cost: 0.5,
+                aiSystem: 'openai',
+            });
+        });
+        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: '123',
+                model: 'gpt-3.5-turbo',
+                usage: { prompt_tokens: 10, completion_tokens: 20, total_tokens: 30 },
+                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 = {};
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.5);
+            await wrapper_1.default._chatCompletionCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'openai.resources.chat.completions',
+                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"}');
+        });
+    });
+});
+//# sourceMappingURL=openai-wrapper.test.js.map

package/dist/instrumentation/__tests__/openai-wrapper.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/openai-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,gEAA8C;AAC9C,0DAA2C;AAC3C,4DAA4C;AAC5C,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,eAAe,EAAE,GAAG,EAAE;IAC7B,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;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,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,KAAK;gBACT,KAAK,EAAE,eAAe;gBACtB,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACrE,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,mCAAmC,CAAC;YAC9D,IAAI;iBACD,KAAK,CAAC,iBAAa,EAAE,6BAA6B,CAAC;iBACnD,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,eAAe;oBACtB,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,GAAG;oBACT,QAAQ,EAAE,QAAQ;iBACnB,CAAC;YACJ,CAAC,CAAC,CAAC;YAEL,MAAM,iBAAa,CAAC,eAAe,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,iBAAiB;gBAChC,QAAQ,EAAE,YAAY;gBACtB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE;gBAC3D,aAAa,EAAE,iBAAiB;gBAChC,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,QAAQ;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;QAC3C,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,eAAe;oBACtB,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,IAAI,EAAE,WAAW;oBACjB,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,KAAK;gBACT,KAAK,EAAE,eAAe;gBACtB,KAAK,EAAE;oBACL,aAAa,EAAE,EAAE;oBACjB,iBAAiB,EAAE,EAAE;oBACrB,YAAY,EAAE,EAAE;oBAChB,yBAAyB,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE;iBACnD;gBACD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,WAAW,EAAE;wBACxD,aAAa,EAAE,MAAM;qBACtB;iBACF;gBACD,kBAAkB,EAAE,SAAS;gBAC7B,YAAY,EAAE,SAAS;aACxB,CAAC;YACF,MAAM,iBAAiB,GAAG,mCAAmC,CAAC;YAE9D,IAAI,CAAC,eAAe,EAAE,CAAC;YAEtB,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YAC1C,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAErE,MAAM,YAAY,GAAG,MAAM,iBAAa,CAAC,2BAA2B,CAAC;gBACnE,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,iBAAiB;gBAChC,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,2BAA2B;YAC3B,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,CAC5C,6BAAkB,CAAC,yBAAyB,EAC5C,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,0BAA0B,EAC7C,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,+BAA+B,EAClD,CAAC,CACF,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,gCAAgC,EACnD,CAAC,CACF,CAAC;YACF,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,CAC5C,6BAAkB,CAAC,wBAAwB,EAC3C,KAAK,CACN,CAAC;YAEF,iBAAiB;YACjB,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,MAAM,CAAC,CACT,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,qBAAqB,EACxC,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,kCAAkC,EACrD,SAAS,CACV,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,4BAA4B,EAC/C,SAAS,CACV,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBAC3B,aAAa,EAAE,iBAAiB;gBAChC,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,QAAQ;aACnB,CAAC,CAAC;QACL,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,KAAK;gBACT,KAAK,EAAE,eAAe;gBACtB,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACrE,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,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;YAC1C,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAErE,MAAM,iBAAa,CAAC,2BAA2B,CAAC;gBAC9C,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,mCAAmC;gBAClD,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,gBAAgB,EACnC,aAAa,CACd,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,mBAAmB,EACtC,UAAU,CACX,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,gBAAgB,EACnC,mBAAmB,CACpB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,21 @@
+/**
+ * Cross-Language Trace Comparison Tests for PostgreSQL (`pg`) Integration
+ *
+ * Verifies that the TypeScript SDK generates traces consistent with the Python SDK
+ * psycopg instrumentation for PostgreSQL queries.
+ *
+ * Python SDK reference: sdk/python/src/shield360/instrumentation/psycopg/utils.py
+ *
+ * Key alignment:
+ *   - db.system.name = 'postgresql'
+ *   - db.operation.name parsed from the SQL verb (SELECT/INSERT/UPDATE/DELETE/...)
+ *   - db.collection.name = target table
+ *   - db.namespace = database name
+ *   - server.address, server.port (default localhost:5432)
+ *   - db.query.text (when captureMessageContent is enabled)
+ *   - db.query.parameter.<index> (only when captureDbParameters is enabled)
+ *   - db.search.similarity_metric for pgvector operators (<=>, <->, <#>)
+ *   - db.query.summary
+ *   - db.response.returned_rows from result.rowCount
+ */
+export {};