openlit 1.13.0 → 1.14.0

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

@@ -0,0 +1,203 @@
+"use strict";
+/**
+ * Cross-Language Trace Comparison Tests for the AI21 Integration
+ *
+ * These verify that the TypeScript AI21 instrumentation emits the same span
+ * attributes / events as the Python SDK reference
+ * (sdk/python/src/openlit/instrumentation/ai21). AI21's request surface has no
+ * seed / frequency_penalty / presence_penalty, and its responses carry no
+ * `model` field, so the response model falls back to the request model.
+ */
+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("../ai21/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('AI21 Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            addEvent: jest.fn(),
+            end: jest.fn(),
+            setStatus: jest.fn(),
+        };
+        config_1.default.environment = 'openlit-testing';
+        config_1.default.applicationName = 'openlit-test';
+        config_1.default.captureMessageContent = true;
+        config_1.default.pricingInfo = {};
+        config_1.default.disableEvents = false;
+        helpers_1.default.getChatModelCost = jest.fn().mockReturnValue(0.001);
+        helpers_1.default.openaiTokens = jest.fn().mockReturnValue(5);
+        helpers_1.default.handleException = jest.fn();
+        helpers_1.default.createStreamProxy = jest
+            .fn()
+            .mockImplementation((stream, _generator) => stream);
+        helpers_1.default.buildInputMessages = jest
+            .fn()
+            .mockReturnValue('[{"role":"user","parts":[{"type":"text","content":"Test"}]}]');
+        helpers_1.default.buildOutputMessages = jest
+            .fn()
+            .mockReturnValue('[{"role":"assistant","parts":[{"type":"text","content":"Response"}],"finish_reason":"stop"}]');
+        helpers_1.default.buildSystemInstructionsFromMessages = jest
+            .fn()
+            .mockImplementation((messages) => {
+            const sys = (messages || []).find((m) => m?.role === 'system');
+            return sys ? JSON.stringify([{ type: 'text', content: String(sys.content) }]) : undefined;
+        });
+        helpers_1.default.buildToolDefinitions = jest.fn().mockReturnValue(undefined);
+        helpers_1.default.emitInferenceEvent = jest.fn();
+        helpers_1.default.computeAgentVersionHash = jest
+            .fn()
+            .mockReturnValue('ts-test-version-hash');
+        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);
+            }
+            if (attrs.serverAddress) {
+                span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, attrs.serverAddress);
+            }
+            if (attrs.serverPort !== undefined) {
+                span.setAttribute(semantic_convention_1.default.SERVER_PORT, attrs.serverPort);
+            }
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, '1.9.0');
+        });
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    // AI21 ChatCompletionResponse shape: no `model`, OpenAI-compatible choices/usage.
+    const mockResponse = () => ({
+        id: 'ai21-test-id',
+        choices: [
+            { index: 0, finish_reason: 'stop', message: { role: 'assistant', content: 'Jamba says hi' } },
+        ],
+        usage: { prompt_tokens: 10, completion_tokens: 20, total_tokens: 30 },
+    });
+    describe('Chat Completion Trace Consistency', () => {
+        it('should set the same core attributes as the Python SDK', async () => {
+            const mockArgs = [
+                {
+                    messages: [{ role: 'user', content: 'What is LLM Observability?' }],
+                    model: 'jamba-large',
+                    max_tokens: 100,
+                    temperature: 0.7,
+                    stream: false,
+                },
+            ];
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'ai21.chat.completions',
+                response: mockResponse(),
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL, 'ai21');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, 'jamba-large');
+            // AI21 responses carry no `model`, so it falls back to the request model.
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, 'jamba-large');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_ID, 'ai21-test-id');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 10);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 20);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, 'text');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'api.ai21.com');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 443);
+        });
+        it('stamps openlit.agent.version_hash on the chat span', async () => {
+            const mockArgs = [
+                { messages: [{ role: 'user', content: 'Hash me' }], model: 'jamba-mini', stream: false },
+            ];
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'ai21.chat.completions',
+                response: mockResponse(),
+                span: mockSpan,
+            });
+            expect(helpers_1.default.computeAgentVersionHash).toHaveBeenCalled();
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.OPENLIT_AGENT_VERSION_HASH, 'ts-test-version-hash');
+        });
+        it('should NOT set total_tokens or client.token.usage on the span', async () => {
+            const mockArgs = [
+                { messages: [{ role: 'user', content: 'Test' }], model: 'jamba-mini', stream: false },
+            ];
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'ai21.chat.completions',
+                response: mockResponse(),
+                span: mockSpan,
+            });
+            const attributeKeys = mockSpan.setAttribute.mock.calls.map(([key]) => key);
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS);
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE);
+        });
+        it('should never emit seed / penalty attrs and omits unset optionals', async () => {
+            const mockArgs = [
+                { messages: [{ role: 'user', content: 'Test' }], model: 'jamba-mini', stream: false },
+            ];
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'ai21.chat.completions',
+                response: mockResponse(),
+                span: mockSpan,
+            });
+            const attributeKeys = mockSpan.setAttribute.mock.calls.map(([key]) => key);
+            // AI21 has no seed / frequency_penalty / presence_penalty.
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_SEED);
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY);
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY);
+            // Optionals not supplied in this request.
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS);
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES);
+            expect(attributeKeys).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_CHOICE_COUNT);
+        });
+        it('should set max_tokens, stop and choice_count only when explicitly provided', async () => {
+            const mockArgs = [
+                {
+                    messages: [{ role: 'user', content: 'Test' }],
+                    model: 'jamba-mini',
+                    max_tokens: 200,
+                    stop: ['END'],
+                    n: 2,
+                    stream: false,
+                },
+            ];
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'ai21.chat.completions',
+                response: mockResponse(),
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 200);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, ['END']);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_CHOICE_COUNT, 2);
+        });
+        it('should emit an inference event via the LoggerProvider', async () => {
+            const mockArgs = [
+                { messages: [{ role: 'user', content: 'Test' }], model: 'jamba-mini', stream: false },
+            ];
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'ai21.chat.completions',
+                response: mockResponse(),
+                span: mockSpan,
+            });
+            expect(helpers_1.default.emitInferenceEvent).toHaveBeenCalled();
+        });
+    });
+});
+//# sourceMappingURL=ai21-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ai21-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/ai21-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;;;AAEH,8DAA0C;AAC1C,0DAAyC;AACzC,4DAA0C;AAC1C,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,sCAAsC,EAAE,GAAG,EAAE;IACpD,IAAI,QAAa,CAAC;IAElB,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;QAED,gBAAqB,CAAC,WAAW,GAAG,iBAAiB,CAAC;QACtD,gBAAqB,CAAC,eAAe,GAAG,cAAc,CAAC;QACvD,gBAAqB,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACnD,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;QACvC,gBAAqB,CAAC,aAAa,GAAG,KAAK,CAAC;QAE5C,iBAAqB,CAAC,gBAAgB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC1E,iBAAqB,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAClE,iBAAqB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAClD,iBAAqB,CAAC,iBAAiB,GAAG,IAAI;aAC5C,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;QACrD,iBAAqB,CAAC,kBAAkB,GAAG,IAAI;aAC7C,EAAE,EAAE;aACJ,eAAe,CAAC,8DAA8D,CAAC,CAAC;QAClF,iBAAqB,CAAC,mBAAmB,GAAG,IAAI;aAC9C,EAAE,EAAE;aACJ,eAAe,CACd,8FAA8F,CAC/F,CAAC;QACH,iBAAqB,CAAC,mCAAmC,GAAG,IAAI;aAC9D,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,QAAe,EAAE,EAAE;YACtC,MAAM,GAAG,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC;YACpE,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC5F,CAAC,CAAC,CAAC;QACJ,iBAAqB,CAAC,oBAAoB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAClF,iBAAqB,CAAC,kBAAkB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACrD,iBAAqB,CAAC,uBAAuB,GAAG,IAAI;aAClD,EAAE,EAAE;aACJ,eAAe,CAAC,sBAAsB,CAAC,CAAC;QAE1C,sBAAmB,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9C,sBAAmB,CAAC,qBAAqB,GAAG,IAAI;aAC9C,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAClC,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,EAAE,CAAC;gBAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;YAC5E,CAAC;YACD,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;gBACnC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,kFAAkF;IAClF,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,CAAC;QAC1B,EAAE,EAAE,cAAc;QAClB,OAAO,EAAE;YACP,EAAE,KAAK,EAAE,CAAC,EAAE,aAAa,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE;SAC9F;QACD,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;KACtE,CAAC,CAAC;IAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QACjD,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,QAAQ,GAAG;gBACf;oBACE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,4BAA4B,EAAE,CAAC;oBACnE,KAAK,EAAE,aAAa;oBACpB,UAAU,EAAE,GAAG;oBACf,WAAW,EAAE,GAAG;oBAChB,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,iBAAW,CAAC,eAAe,CAAC;gBAChC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,uBAAuB;gBACtC,QAAQ,EAAE,YAAY,EAAE;gBACxB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,MAAM,CAAC,CAAC;YACzG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,aAAa,CAAC,CAAC;YAC3G,0EAA0E;YAC1E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;YAC5G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,cAAc,CAAC,CAAC;YAC1G,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,0BAA0B,EAAE,EAAE,CAAC,CAAC;YACtG,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,yBAAyB,EAAE,GAAG,CAAC,CAAC;YACtG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YACvG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;YAClG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;YACtG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,QAAQ,GAAG;gBACf,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE;aACzF,CAAC;YAEF,MAAM,iBAAW,CAAC,eAAe,CAAC;gBAChC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,uBAAuB;gBACtC,QAAQ,EAAE,YAAY,EAAE;gBACxB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAE,iBAAqB,CAAC,uBAAuB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC1E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,0BAA0B,EAC7C,sBAAsB,CACvB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,QAAQ,GAAG;gBACf,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE;aACtF,CAAC;YAEF,MAAM,iBAAW,CAAC,eAAe,CAAC;gBAChC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,uBAAuB;gBACtC,QAAQ,EAAE,YAAY,EAAE;gBACxB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,aAAa,GAAI,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAW,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YACpG,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;YAClF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,QAAQ,GAAG;gBACf,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE;aACtF,CAAC;YAEF,MAAM,iBAAW,CAAC,eAAe,CAAC;gBAChC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,uBAAuB;gBACtC,QAAQ,EAAE,YAAY,EAAE;gBACxB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,aAAa,GAAI,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAW,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YACpG,2DAA2D;YAC3D,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC;YAC5E,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,gCAAgC,CAAC,CAAC;YACzF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,+BAA+B,CAAC,CAAC;YACxF,0CAA0C;YAC1C,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;YAClF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,CAAC;YACtF,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;YAC1F,MAAM,QAAQ,GAAG;gBACf;oBACE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;oBAC7C,KAAK,EAAE,YAAY;oBACnB,UAAU,EAAE,GAAG;oBACf,IAAI,EAAE,CAAC,KAAK,CAAC;oBACb,CAAC,EAAE,CAAC;oBACJ,MAAM,EAAE,KAAK;iBACd;aACF,CAAC;YAEF,MAAM,iBAAW,CAAC,eAAe,CAAC;gBAChC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,uBAAuB;gBACtC,QAAQ,EAAE,YAAY,EAAE;gBACxB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YACtG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;YAC9G,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACxG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,QAAQ,GAAG;gBACf,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE;aACtF,CAAC;YAEF,MAAM,iBAAW,CAAC,eAAe,CAAC;gBAChC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,uBAAuB;gBACtC,QAAQ,EAAE,YAAY,EAAE;gBACxB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAE,iBAAqB,CAAC,kBAAkB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,262 @@
+"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("../assemblyai/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('AssemblyAIWrapper', () => {
+    let span;
+    beforeEach(() => {
+        span = mockTracer.startSpan('test-span');
+        span.setAttribute = jest.fn();
+        span.addEvent = jest.fn();
+        jest.clearAllMocks();
+        helpers_1.default.buildInputMessages = jest
+            .fn()
+            .mockImplementation((messages) => JSON.stringify(messages.map((m) => ({
+            role: m.role,
+            parts: [{ type: 'text', content: m.content }],
+        }))));
+        helpers_1.default.buildOutputMessages = jest
+            .fn()
+            .mockImplementation((text, finishReason) => JSON.stringify([
+            {
+                role: 'assistant',
+                parts: [{ type: 'text', content: text }],
+                finish_reason: finishReason,
+            },
+        ]));
+        base_wrapper_1.default.setBaseSpanAttributes = jest.fn();
+    });
+    afterEach(() => {
+        span.end();
+    });
+    describe('_parseAudioArgs', () => {
+        it('should parse options and speechModel (camelCase) from a params object', () => {
+            const parsed = wrapper_1.default._parseAudioArgs([
+                {
+                    audio: 'https://example.com/audio.mp3',
+                    speechModel: 'nano',
+                },
+            ]);
+            expect(parsed.requestModel).toBe('nano');
+            expect(parsed.audioUrl).toBe('https://example.com/audio.mp3');
+        });
+        it('should parse speech_model (snake_case) and default the model to "best"', () => {
+            const withSnake = wrapper_1.default._parseAudioArgs([
+                { audio: 'https://example.com/a.mp3', speech_model: 'best' },
+            ]);
+            expect(withSnake.requestModel).toBe('best');
+            const withDefault = wrapper_1.default._parseAudioArgs([
+                { audio: 'https://example.com/a.mp3' },
+            ]);
+            expect(withDefault.requestModel).toBe('best');
+        });
+        it('should treat a bare string arg as a transcript id (e.g. transcripts.get)', () => {
+            const parsed = wrapper_1.default._parseAudioArgs(['transcript-id-123']);
+            expect(parsed.audioUrl).toBe('transcript-id-123');
+            expect(parsed.requestModel).toBe('best');
+        });
+        it('should return an empty string (never "undefined"/"null") when audio is missing', () => {
+            expect(wrapper_1.default._parseAudioArgs([{ speechModel: 'nano' }]).audioUrl).toBe('');
+            expect(wrapper_1.default._parseAudioArgs([{ audio: null }]).audioUrl).toBe('');
+            expect(wrapper_1.default._parseAudioArgs([{ audio: undefined }]).audioUrl).toBe('');
+            expect(wrapper_1.default._parseAudioArgs([]).audioUrl).toBe('');
+        });
+    });
+    describe('_commonAudioSetter', () => {
+        it('should set span attributes from the transcript response and return metric parameters', () => {
+            const mockArgs = [
+                {
+                    audio: 'https://example.com/audio.mp3',
+                    speechModel: 'best',
+                },
+            ];
+            const mockResponse = {
+                id: 'transcript-abc',
+                text: 'hello transcribed world',
+                audio_url: 'https://example.com/audio.mp3',
+                audio_duration: 42,
+            };
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            config_1.default.captureMessageContent = true;
+            jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0.005);
+            const metricParams = wrapper_1.default._commonAudioSetter({
+                args: mockArgs,
+                genAIEndpoint: 'assemblyai.transcripts.transcribe',
+                response: mockResponse,
+                span,
+                isStream: false,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_AUDIO_DURATION, 42);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, 'text');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_ID, 'transcript-abc');
+            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_IS_STREAM, false);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 0);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 0);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, 0);
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, 'assemblyai');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, 'best');
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, JSON.stringify([
+                { role: 'user', parts: [{ type: 'text', content: 'https://example.com/audio.mp3' }] },
+            ]));
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, JSON.stringify([
+                {
+                    role: 'assistant',
+                    parts: [{ type: 'text', content: 'hello transcribed world' }],
+                    finish_reason: 'stop',
+                },
+            ]));
+            expect(metricParams).toEqual({
+                genAIEndpoint: 'assemblyai.transcripts.transcribe',
+                model: 'best',
+                cost: 0.005,
+                aiSystem: 'assemblyai',
+                serverAddress: 'api.assemblyai.com',
+                serverPort: 443,
+            });
+        });
+        it('should stamp the AssemblyAI package version on the span', () => {
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            config_1.default.captureMessageContent = false;
+            jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0.005);
+            wrapper_1.default._commonAudioSetter({
+                args: [{ audio: 'https://example.com/a.mp3' }],
+                genAIEndpoint: 'assemblyai.transcripts.transcribe',
+                response: { id: 'x', text: '', audio_url: '', audio_duration: 0 },
+                span,
+                isStream: false,
+                sdkVersion: '4.16.1',
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_SDK_VERSION, '4.16.1');
+        });
+        it('should not set message content if captureMessageContent is false', () => {
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            config_1.default.captureMessageContent = false;
+            jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0.005);
+            wrapper_1.default._commonAudioSetter({
+                args: [{ audio: 'https://example.com/a.mp3' }],
+                genAIEndpoint: 'assemblyai.transcripts.transcribe',
+                response: { id: 'x', text: 'hi', audio_url: '', audio_duration: 0 },
+                span,
+                isStream: false,
+            });
+            expect(span.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, expect.any(String));
+            expect(span.setAttribute).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, expect.any(String));
+        });
+        it('should emit inference event when disableEvents is false and captureMessageContent is true', () => {
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = false;
+            config_1.default.captureMessageContent = true;
+            jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0.005);
+            const spyEmit = jest.spyOn(helpers_1.default, 'emitInferenceEvent');
+            wrapper_1.default._commonAudioSetter({
+                args: [{ audio: 'https://example.com/a.mp3', speechModel: 'nano' }],
+                genAIEndpoint: 'assemblyai.transcripts.transcribe',
+                response: { id: 'rid', text: 'hi', audio_url: '', audio_duration: 5 },
+                span,
+                isStream: false,
+            });
+            expect(spyEmit).toHaveBeenCalledWith(span, expect.objectContaining({
+                [semantic_convention_1.default.GEN_AI_OPERATION]: 'audio',
+                [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: 'nano',
+                [semantic_convention_1.default.GEN_AI_RESPONSE_ID]: 'rid',
+            }));
+        });
+        it('should not emit inference event when captureMessageContent is false', () => {
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = false;
+            config_1.default.captureMessageContent = false;
+            jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0.005);
+            const spyEmit = jest.spyOn(helpers_1.default, 'emitInferenceEvent');
+            wrapper_1.default._commonAudioSetter({
+                args: [{ audio: 'https://example.com/a.mp3', speechModel: 'nano' }],
+                genAIEndpoint: 'assemblyai.transcripts.transcribe',
+                response: { id: 'rid', text: 'hi', audio_url: '', audio_duration: 5 },
+                span,
+                isStream: false,
+            });
+            expect(spyEmit).not.toHaveBeenCalled();
+        });
+        it('should resolve the model from the transcript response for transcripts.get polls', () => {
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            config_1.default.captureMessageContent = false;
+            jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0.005);
+            const metricParams = wrapper_1.default._commonAudioSetter({
+                args: ['transcript-id-123'],
+                genAIEndpoint: 'assemblyai.transcripts.get',
+                response: {
+                    id: 'transcript-id-123',
+                    text: 'done',
+                    speech_model: 'nano',
+                    audio_duration: 10,
+                },
+                span,
+                isStream: false,
+            });
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, 'nano');
+            expect(metricParams.model).toBe('nano');
+        });
+        it('should pass audio duration to getAudioModelCost when audio_url is absent', () => {
+            config_1.default.pricingInfo = { audio: { best: 0.001 } };
+            config_1.default.disableEvents = true;
+            config_1.default.captureMessageContent = false;
+            const costSpy = jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0.01);
+            wrapper_1.default._commonAudioSetter({
+                args: ['transcript-id-123'],
+                genAIEndpoint: 'assemblyai.transcripts.get',
+                response: {
+                    id: 'transcript-id-123',
+                    text: 'done',
+                    speech_model: 'best',
+                    audio_duration: 60,
+                },
+                span,
+                isStream: false,
+            });
+            expect(costSpy).toHaveBeenCalledWith('best', { audio: { best: 0.001 } }, '', 60);
+        });
+        it('should never emit the literal string "undefined"/"null" when fields are missing', () => {
+            config_1.default.pricingInfo = {};
+            config_1.default.disableEvents = true;
+            config_1.default.captureMessageContent = true;
+            jest.spyOn(helpers_1.default, 'getAudioModelCost').mockReturnValue(0);
+            // No audio in args, and a response with no id/text/audio_url.
+            wrapper_1.default._commonAudioSetter({
+                args: [{ speechModel: 'best' }],
+                genAIEndpoint: 'assemblyai.transcripts.transcribe',
+                response: {},
+                span,
+                isStream: false,
+            });
+            const calls = span.setAttribute.mock.calls;
+            for (const [, value] of calls) {
+                expect(value).not.toBe('undefined');
+                expect(value).not.toBe('null');
+                if (typeof value === 'string') {
+                    expect(value).not.toContain('"undefined"');
+                    expect(value).not.toContain('"null"');
+                }
+            }
+            // Response id is set as an empty string rather than 'undefined'.
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_ID, '');
+            // Input message content is the empty string, not 'undefined'.
+            expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, JSON.stringify([{ role: 'user', parts: [{ type: 'text', content: '' }] }]));
+        });
+    });
+});
+//# sourceMappingURL=assemblyai-wrapper.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"assemblyai-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/assemblyai-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,oEAAsD;AACtD,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,mBAAmB,EAAE,GAAG,EAAE;IACjC,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;QAEpB,iBAAqB,CAAC,kBAAkB,GAAG,IAAI;aAC7C,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,QAAkD,EAAE,EAAE,CACzE,IAAI,CAAC,SAAS,CACZ,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;SAC9C,CAAC,CAAC,CACJ,CACF,CAAC;QACH,iBAAqB,CAAC,mBAAmB,GAAG,IAAI;aAC9C,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,IAAY,EAAE,YAAoB,EAAE,EAAE,CACzD,IAAI,CAAC,SAAS,CAAC;YACb;gBACE,IAAI,EAAE,WAAW;gBACjB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBACxC,aAAa,EAAE,YAAY;aAC5B;SACF,CAAC,CACH,CAAC;QACH,sBAAmB,CAAC,qBAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IACzD,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,uEAAuE,EAAE,GAAG,EAAE;YAC/E,MAAM,MAAM,GAAG,iBAAiB,CAAC,eAAe,CAAC;gBAC/C;oBACE,KAAK,EAAE,+BAA+B;oBACtC,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,SAAS,GAAG,iBAAiB,CAAC,eAAe,CAAC;gBAClD,EAAE,KAAK,EAAE,2BAA2B,EAAE,YAAY,EAAE,MAAM,EAAE;aAC7D,CAAC,CAAC;YACH,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAE5C,MAAM,WAAW,GAAG,iBAAiB,CAAC,eAAe,CAAC;gBACpD,EAAE,KAAK,EAAE,2BAA2B,EAAE;aACvC,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,MAAM,GAAG,iBAAiB,CAAC,eAAe,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,MAAM,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvF,MAAM,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACpF,MAAM,CAAC,iBAAiB,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,sFAAsF,EAAE,GAAG,EAAE;YAC9F,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,+BAA+B;oBACtC,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YACF,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,gBAAgB;gBACpB,IAAI,EAAE,yBAAyB;gBAC/B,SAAS,EAAE,+BAA+B;gBAC1C,cAAc,EAAE,EAAE;aACnB,CAAC;YAED,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,mBAAmB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEtE,MAAM,YAAY,GAAG,iBAAiB,CAAC,kBAAkB,CAAC;gBACxD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,mCAAmC;gBAClD,QAAQ,EAAE,YAAY;gBACtB,IAAI;gBACJ,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,6BAA6B,EAChD,EAAE,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,kBAAkB,EACrC,MAAM,CACP,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,kBAAkB,EACrC,gBAAgB,CACjB,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,wBAAwB,EAC3C,KAAK,CACN,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,yBAAyB,EAC5C,CAAC,CACF,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,0BAA0B,EAC7C,CAAC,CACF,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,yBAAyB,EAC5C,CAAC,CACF,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,oBAAoB,EACvC,YAAY,CACb,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,qBAAqB,EACxC,MAAM,CACP,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,qBAAqB,EACxC,IAAI,CAAC,SAAS,CAAC;gBACb,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,+BAA+B,EAAE,CAAC,EAAE;aACtF,CAAC,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,sBAAsB,EACzC,IAAI,CAAC,SAAS,CAAC;gBACb;oBACE,IAAI,EAAE,WAAW;oBACjB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,yBAAyB,EAAE,CAAC;oBAC7D,aAAa,EAAE,MAAM;iBACtB;aACF,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBAC3B,aAAa,EAAE,mCAAmC;gBAClD,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE,KAAK;gBACX,QAAQ,EAAE,YAAY;gBACtB,aAAa,EAAE,oBAAoB;gBACnC,UAAU,EAAE,GAAG;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YAChE,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3C,gBAAqB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACrD,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,mBAAmB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEtE,iBAAiB,CAAC,kBAAkB,CAAC;gBACnC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,CAAC;gBAC9C,aAAa,EAAE,mCAAmC;gBAClD,QAAQ,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE;gBACjE,IAAI;gBACJ,QAAQ,EAAE,KAAK;gBACf,UAAU,EAAE,QAAQ;aACrB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,kBAAkB,EACrC,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YACzE,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3C,gBAAqB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACrD,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,mBAAmB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEtE,iBAAiB,CAAC,kBAAkB,CAAC;gBACnC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,CAAC;gBAC9C,aAAa,EAAE,mCAAmC;gBAClD,QAAQ,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE;gBACnE,IAAI;gBACJ,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,qBAAqB,EACxC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,sBAAsB,EACzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2FAA2F,EAAE,GAAG,EAAE;YAClG,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,KAAK,CAAC;YAC5C,gBAAqB,CAAC,qBAAqB,GAAG,IAAI,CAAC;YACpD,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,mBAAmB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEtE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,oBAAoB,CAAC,CAAC;YAEhE,iBAAiB,CAAC,kBAAkB,CAAC;gBACnC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;gBACnE,aAAa,EAAE,mCAAmC;gBAClD,QAAQ,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE;gBACrE,IAAI;gBACJ,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,IAAI,EACJ,MAAM,CAAC,gBAAgB,CAAC;gBACtB,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,OAAO;gBAC9C,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,MAAM;gBACjD,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,KAAK;aAC/C,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC5E,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,KAAK,CAAC;YAC5C,gBAAqB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACrD,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,mBAAmB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEtE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,oBAAoB,CAAC,CAAC;YAEhE,iBAAiB,CAAC,kBAAkB,CAAC;gBACnC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;gBACnE,aAAa,EAAE,mCAAmC;gBAClD,QAAQ,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE;gBACrE,IAAI;gBACJ,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;YACxF,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3C,gBAAqB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACrD,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,mBAAmB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEtE,MAAM,YAAY,GAAG,iBAAiB,CAAC,kBAAkB,CAAC;gBACxD,IAAI,EAAE,CAAC,mBAAmB,CAAC;gBAC3B,aAAa,EAAE,4BAA4B;gBAC3C,QAAQ,EAAE;oBACR,EAAE,EAAE,mBAAmB;oBACvB,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,MAAM;oBACpB,cAAc,EAAE,EAAE;iBACnB;gBACD,IAAI;gBACJ,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,qBAAqB,EACxC,MAAM,CACP,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YACjF,gBAAqB,CAAC,WAAW,GAAG,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAC/D,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3C,gBAAqB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACrD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,mBAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAErF,iBAAiB,CAAC,kBAAkB,CAAC;gBACnC,IAAI,EAAE,CAAC,mBAAmB,CAAC;gBAC3B,aAAa,EAAE,4BAA4B;gBAC3C,QAAQ,EAAE;oBACR,EAAE,EAAE,mBAAmB;oBACvB,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,MAAM;oBACpB,cAAc,EAAE,EAAE;iBACnB;gBACD,IAAI;gBACJ,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;YACxF,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,mBAAmB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAElE,8DAA8D;YAC9D,iBAAiB,CAAC,kBAAkB,CAAC;gBACnC,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;gBAC/B,aAAa,EAAE,mCAAmC;gBAClD,QAAQ,EAAE,EAAE;gBACZ,IAAI;gBACJ,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAI,IAAI,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC;YAC1D,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,KAAK,EAAE,CAAC;gBAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBACpC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC/B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;oBAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBACxC,CAAC;YACH,CAAC;YAED,iEAAiE;YACjE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,kBAAkB,EACrC,EAAE,CACH,CAAC;YACF,8DAA8D;YAC9D,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,qBAAqB,EACxC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAC3E,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,7 @@
+/**
+ * Cross-language trace comparison tests for Astra DB integration.
+ *
+ * Verifies TypeScript SDK trace attributes match the Python SDK's Astra
+ * instrumentation behavior for Collection operations.
+ */
+export {};