shield360 1.0.0

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

@@ -0,0 +1,214 @@
+"use strict";
+/**
+ * Cross-Language Trace Comparison Tests for local HuggingFace inference
+ * (Transformers.js) instrumentation.
+ *
+ * Verifies that the TypeScript SDK produces traces consistent with the Python
+ * `transformers` instrumentation and the OTel GenAI semantic conventions:
+ *   - text-generation reports the `chat` operation (Python parity)
+ *   - other local pipelines map to the closest OTel operation
+ *   - token usage, cache tokens, timing, and package version are stamped
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const wrapper_1 = __importDefault(require("../transformers/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('../../config');
+jest.mock('../../helpers');
+jest.mock('../base-wrapper');
+describe('Transformers.js Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    let mockTracer;
+    const makeInstance = (task, model = 'Xenova/distilgpt2') => ({
+        task,
+        model: { config: { _name_or_path: model } },
+    });
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            addEvent: jest.fn(),
+            end: jest.fn(),
+            setStatus: jest.fn(),
+        };
+        mockTracer = {
+            startSpan: jest.fn().mockReturnValue(mockSpan),
+        };
+        config_1.default.environment = 'shield360-testing';
+        config_1.default.applicationName = 'shield360-test';
+        config_1.default.captureMessageContent = true;
+        config_1.default.pricingInfo = {};
+        config_1.default.disableEvents = false;
+        helpers_1.default.getChatModelCost = jest.fn().mockReturnValue(0.0);
+        helpers_1.default.generalTokens = jest.fn().mockReturnValue(7);
+        helpers_1.default.buildInputMessages = jest.fn().mockReturnValue('[{"role":"user"}]');
+        helpers_1.default.buildOutputMessages = jest.fn().mockReturnValue('[{"role":"assistant"}]');
+        helpers_1.default.handleException = jest.fn();
+        helpers_1.default.emitInferenceEvent = jest.fn();
+        helpers_1.default.computeAgentVersionHash = jest.fn().mockReturnValue('');
+        base_wrapper_1.default.recordMetrics = jest.fn();
+        base_wrapper_1.default.setBaseSpanAttributes = jest.fn().mockImplementation((span, attrs) => {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL, attrs.aiSystem);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, attrs.model);
+            if (attrs.cost !== undefined)
+                span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_COST, attrs.cost);
+            // setBaseSpanAttributes stamps Shield360's SDK version; the wrapper must
+            // override it with the transformers package version afterwards.
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, 'shield360-sdk-version');
+        });
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    // ── Span creation ──────────────────────────────────────────────────────────
+    describe('Span creation', () => {
+        it('creates a "chat {model}" span for text-generation (Python parity)', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'TextGenerationPipeline', '3.0.0');
+            const original = jest.fn().mockResolvedValue([{ generated_text: 'hello world' }]);
+            const wrapped = patched(original);
+            await wrapped.call(makeInstance('text-generation'), 'say hi', { temperature: 0.7 });
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('chat Xenova/distilgpt2', expect.objectContaining({
+                attributes: expect.objectContaining({
+                    [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+                    [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_HUGGING_FACE,
+                    [semantic_convention_1.default.SERVER_ADDRESS]: '127.0.0.1',
+                    [semantic_convention_1.default.SERVER_PORT]: 80,
+                }),
+            }), expect.anything());
+        });
+        it('maps summarization to the text_completion operation', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'SummarizationPipeline');
+            const original = jest.fn().mockResolvedValue([{ summary_text: 'short' }]);
+            const wrapped = patched(original);
+            await wrapped.call(makeInstance('summarization'), 'a long text', {});
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('text_completion Xenova/distilgpt2', expect.objectContaining({
+                attributes: expect.objectContaining({
+                    [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION,
+                }),
+            }), expect.anything());
+        });
+        it('maps feature-extraction to the embeddings operation', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'FeatureExtractionPipeline');
+            const original = jest.fn().mockResolvedValue({ data: [0.1, 0.2] });
+            const wrapped = patched(original);
+            await wrapped.call(makeInstance('feature-extraction'), 'embed me', {});
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('embeddings Xenova/distilgpt2', expect.objectContaining({
+                attributes: expect.objectContaining({
+                    [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING,
+                }),
+            }), expect.anything());
+        });
+    });
+    // ── Attribute parity ─────────────────────────────────────────────────────────
+    describe('Span attributes (Python parity)', () => {
+        const callTextGen = async (sdkVersion) => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'TextGenerationPipeline', sdkVersion);
+            const original = jest.fn().mockResolvedValue([{ generated_text: 'the answer is 42' }]);
+            const wrapped = patched(original);
+            await wrapped.call(makeInstance('text-generation'), 'question', {
+                temperature: 0.5,
+                top_k: 40,
+                top_p: 0.9,
+                max_new_tokens: 64,
+            });
+        };
+        it('stamps token usage including client.token.usage and cache tokens (0)', async () => {
+            await callTextGen();
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 7);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 7);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, 14);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, 0);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, 0);
+        });
+        it('always stamps TTFT and TBT', async () => {
+            await callTextGen();
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_SERVER_TTFT, expect.any(Number));
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_SERVER_TBT, 0);
+        });
+        it('stamps gen_ai.system and request params', async () => {
+            await callTextGen();
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, 'huggingface');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.5);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K, 40);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, 0.9);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 64);
+        });
+        it('overrides gen_ai.sdk.version with the transformers package version', async () => {
+            await callTextGen('3.2.1');
+            // The last write to GEN_AI_SDK_VERSION must be the package version, not Shield360's.
+            const versionCalls = mockSpan.setAttribute.mock.calls.filter((c) => c[0] === semantic_convention_1.default.GEN_AI_SDK_VERSION);
+            expect(versionCalls[versionCalls.length - 1][1]).toBe('3.2.1');
+        });
+        it('sets text output type and finish reason for generative tasks', async () => {
+            await callTextGen();
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+        });
+    });
+    describe('Embedding tasks', () => {
+        it('does not set output messages or finish reason and reports 0 output tokens', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'FeatureExtractionPipeline');
+            const original = jest.fn().mockResolvedValue({ data: [0.1, 0.2, 0.3] });
+            const wrapped = patched(original);
+            await wrapped.call(makeInstance('feature-extraction'), 'embed me', {});
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 0);
+            expect(mockSpan.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, expect.anything());
+            expect(mockSpan.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, expect.anything());
+        });
+    });
+    // ── Content capture & events ─────────────────────────────────────────────────
+    describe('Content capture and events', () => {
+        it('captures input/output messages only when enabled', async () => {
+            config_1.default.captureMessageContent = false;
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'TextGenerationPipeline');
+            const original = jest.fn().mockResolvedValue([{ generated_text: 'hi' }]);
+            await patched(original).call(makeInstance('text-generation'), 'in', {});
+            expect(mockSpan.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, expect.anything());
+        });
+        it('emits an inference event when events are enabled', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'TextGenerationPipeline');
+            const original = jest.fn().mockResolvedValue([{ generated_text: 'hi' }]);
+            await patched(original).call(makeInstance('text-generation'), 'in', {});
+            expect(helpers_1.default.emitInferenceEvent).toHaveBeenCalledWith(mockSpan, 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]: 'Xenova/distilgpt2',
+            }));
+        });
+    });
+    // ── Error path ───────────────────────────────────────────────────────────────
+    describe('Error handling', () => {
+        it('records error metrics once, ends the span, and rethrows', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'TextGenerationPipeline');
+            const err = new Error('inference failed');
+            const original = jest.fn().mockRejectedValue(err);
+            const wrapped = patched(original);
+            await expect(wrapped.call(makeInstance('text-generation'), 'in', {})).rejects.toThrow('inference failed');
+            expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, err);
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledTimes(1);
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledWith(mockSpan, expect.objectContaining({ errorType: 'Error' }));
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        });
+    });
+    // ── Response parsing per task ─────────────────────────────────────────────────
+    describe('Response parsing', () => {
+        it('extracts nested chat-message generated_text (text-generation)', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'TextGenerationPipeline');
+            const original = jest.fn().mockResolvedValue([
+                { generated_text: [{ role: 'user', content: 'hi' }, { role: 'assistant', content: 'hello there' }] },
+            ]);
+            await patched(original).call(makeInstance('text-generation'), 'hi', {});
+            expect(helpers_1.default.buildOutputMessages).toHaveBeenCalledWith('hello there', 'stop');
+        });
+        it('extracts translation_text (translation)', async () => {
+            const patched = wrapper_1.default._patchPipelineCall(mockTracer, 'TranslationPipeline');
+            const original = jest.fn().mockResolvedValue([{ translation_text: 'bonjour' }]);
+            await patched(original).call(makeInstance('translation'), 'hello', {});
+            expect(helpers_1.default.buildOutputMessages).toHaveBeenCalledWith('bonjour', 'stop');
+        });
+    });
+});
+//# sourceMappingURL=transformers-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"transformers-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/transformers-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;AAEH,sEAA0D;AAC1D,0DAA2C;AAC3C,4DAA4C;AAC5C,mEAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAC3B,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAE7B,QAAQ,CAAC,iDAAiD,EAAE,GAAG,EAAE;IAC/D,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,MAAM,YAAY,GAAG,CAAC,IAAY,EAAE,KAAK,GAAG,mBAAmB,EAAE,EAAE,CAAC,CAAC;QACnE,IAAI;QACJ,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;KAC5C,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;SACrB,CAAC;QAEF,UAAU,GAAG;YACX,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;SAC/C,CAAC;QAED,gBAAuB,CAAC,WAAW,GAAG,mBAAmB,CAAC;QAC1D,gBAAuB,CAAC,eAAe,GAAG,gBAAgB,CAAC;QAC3D,gBAAuB,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACrD,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;QACzC,gBAAuB,CAAC,aAAa,GAAG,KAAK,CAAC;QAE9C,iBAAuB,CAAC,gBAAgB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAC1E,iBAAuB,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACrE,iBAAuB,CAAC,kBAAkB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAC5F,iBAAuB,CAAC,mBAAmB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,wBAAwB,CAAC,CAAC;QAClG,iBAAuB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACpD,iBAAuB,CAAC,kBAAkB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACvD,iBAAuB,CAAC,uBAAuB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAEhF,sBAAmB,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9C,sBAAmB,CAAC,qBAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAS,EAAE,KAAU,EAAE,EAAE;YAClG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YAChF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACxE,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAClG,yEAAyE;YACzE,gEAAgE;YAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,uBAAuB,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,8EAA8E;IAE9E,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,wBAAwB,EAAE,OAAO,CAAC,CAAC;YACtG,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,cAAc,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;YAClF,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;YAEpF,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAC/C,wBAAwB,EACxB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAClC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;oBACpF,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;oBAC7F,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,WAAW;oBAChD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,EAAE;iBACrC,CAAC;aACH,CAAC,EACF,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,uBAAuB,CAAC,CAAC;YAC5F,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;YAC1E,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAC/C,mCAAmC,EACnC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAClC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,qCAAqC;iBAChG,CAAC;aACH,CAAC,EACF,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,2BAA2B,CAAC,CAAC;YAChG,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAC/C,8BAA8B,EAC9B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAClC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,+BAA+B;iBAC1F,CAAC;aACH,CAAC,EACF,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,gFAAgF;IAEhF,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,MAAM,WAAW,GAAG,KAAK,EAAE,UAAmB,EAAE,EAAE;YAChD,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,wBAAwB,EAAE,UAAU,CAAC,CAAC;YACzG,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YACvF,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAClC,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,EAAE,UAAU,EAAE;gBAC9D,WAAW,EAAE,GAAG;gBAChB,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,GAAG;gBACV,cAAc,EAAE,EAAE;aACnB,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,WAAW,EAAE,CAAC;YAEpB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YACpG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,CAAC,CAAC,CAAC;YACrG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YACrG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;YAC/G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAC;QACrH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,WAAW,EAAE,CAAC;YACpB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YAC9G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;QAC9F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,WAAW,EAAE,CAAC;YACpB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,aAAa,CAAC,CAAC;YAC3G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACvG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;YAChG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,GAAG,CAAC,CAAC;YACjG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;QACvG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,WAAW,CAAC,OAAO,CAAC,CAAC;YAC3B,qFAAqF;YACrF,MAAM,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAC1D,CAAC,CAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,6BAAkB,CAAC,kBAAkB,CAC7D,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,WAAW,EAAE,CAAC;YACpB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,6BAAkB,CAAC,uBAAuB,CAC3C,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QACjH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,2BAA2B,CAAC,CAAC;YAChG,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YACxE,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,CAAC,CAAC,CAAC;YACrG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CACpD,6BAAkB,CAAC,6BAA6B,EAChD,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CACpD,6BAAkB,CAAC,sBAAsB,EACzC,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,gFAAgF;IAEhF,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAC/D,gBAAuB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACvD,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,wBAAwB,CAAC,CAAC;YAC7F,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACzE,MAAM,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAExE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CACpD,6BAAkB,CAAC,qBAAqB,EACxC,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,wBAAwB,CAAC,CAAC;YAC7F,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACzE,MAAM,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAExE,MAAM,CAAC,iBAAe,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAC7D,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC;gBACtB,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;gBACpF,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,mBAAmB;aAC/D,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,gFAAgF;IAEhF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,wBAAwB,CAAC,CAAC;YAC7F,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAClD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YAE1G,MAAM,CAAC,iBAAe,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YAC5E,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACpD,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAChD,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,iFAAiF;IAEjF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,wBAAwB,CAAC,CAAC;YAC7F,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC3C,EAAE,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE;aACrG,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAExE,MAAM,CAAC,iBAAe,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,OAAO,GAAG,iBAAmB,CAAC,kBAAkB,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC;YAC1F,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,gBAAgB,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;YAChF,MAAM,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,iBAAe,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,8 @@
+/**
+ * Cross-Language Trace Comparison Tests for Vercel AI Integration
+ *
+ * Vercel AI is JS-only (no Python equivalent). These tests verify that the
+ * Vercel AI instrumentation follows the same OTel conventions and patterns
+ * as the OpenAI reference wrapper.
+ */
+export {};