shield360 1.0.0

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

@@ -0,0 +1,208 @@
+"use strict";
+/**
+ * Cross-Language Trace Comparison Tests for the mem0 Integration
+ *
+ * These verify that the TypeScript mem0 instrumentation emits the same spans /
+ * attributes as the Python SDK reference
+ * (sdk/python/src/shield360/instrumentation/mem0). mem0 is a memory layer, not an LLM
+ * provider: each operation (add / search / get / getAll / update / delete /
+ * deleteAll / history) becomes one CLIENT span named `memory <op>` carrying
+ * `gen_ai.*` attributes. There are no tokens, model, cost, or metrics. The same
+ * wrapper serves both the hosted `MemoryClient` and the OSS `Memory` clients, whose
+ * method surfaces are identical.
+ */
+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 semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const wrapper_1 = __importDefault(require("../mem0/wrapper"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+jest.mock('../../config');
+jest.mock('../../helpers', () => ({
+    __esModule: true,
+    default: { handleException: jest.fn() },
+    applyCustomSpanAttributes: jest.fn(),
+}));
+describe('mem0 Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            setStatus: jest.fn(),
+            recordException: jest.fn(),
+            end: jest.fn(),
+        };
+        mockTracer = { startSpan: jest.fn(() => mockSpan) };
+        config_1.default.environment = 'shield360-testing';
+        config_1.default.applicationName = 'shield360-test';
+        config_1.default.captureMessageContent = true;
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    // Wrap `impl` as the mem0 operation `spanName`, then invoke it with `args`.
+    async function runOp(spanName, impl, args = []) {
+        const wrapped = wrapper_1.default._patchMemoryOperation(mockTracer, spanName)(impl);
+        return wrapped.apply({}, args);
+    }
+    // Snapshot of all setAttribute(key, value) calls as a plain object.
+    function attrs() {
+        return Object.fromEntries(mockSpan.setAttribute.mock.calls);
+    }
+    describe('Core span shape', () => {
+        it('opens a CLIENT span named after the Python endpoint string', async () => {
+            await runOp('memory add', async () => [{ id: 'm1' }], [[{ role: 'user', content: 'hi' }], {}]);
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('memory add', { kind: api_1.SpanKind.CLIENT });
+            expect(mockSpan.setStatus).toHaveBeenCalledWith({ code: 1 }); // SpanStatusCode.OK
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        });
+        it('sets the same core attributes as the Python SDK', async () => {
+            await runOp('memory add', async () => [{ id: 'm1' }], [[{ role: 'user', content: 'hi' }], {}]);
+            const a = attrs();
+            expect(a[semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]).toBe('shield360');
+            // Provider name is stamped on BOTH keys (gen_ai.system + gen_ai.provider.name).
+            expect(a[semantic_convention_1.default.GEN_AI_PROVIDER_NAME]).toBe('mem0');
+            expect(a[semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]).toBe('mem0');
+            expect(a[semantic_convention_1.default.GEN_AI_ENDPOINT]).toBe('memory add');
+            expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_MEMORY);
+            expect(a[semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]).toBe('shield360-testing');
+            expect(a[semantic_conventions_1.ATTR_SERVICE_NAME]).toBe('shield360-test');
+            // gen_ai.sdk.version is the mem0ai package version (a string); its exact value
+            // comes from the version OTel passes to the patch hook — asserted in the next test.
+            expect(typeof a[semantic_convention_1.default.GEN_AI_SDK_VERSION]).toBe('string');
+            expect(typeof a[semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION]).toBe('number');
+        });
+        it('stamps the mem0ai version OTel hands to the patch hook', async () => {
+            const wrapped = wrapper_1.default._patchMemoryOperation(mockTracer, 'memory add', '3.0.7')(async () => [{ id: 'm1' }]);
+            await wrapped.apply({}, [[{ role: 'user', content: 'hi' }], {}]);
+            expect(attrs()[semantic_convention_1.default.GEN_AI_SDK_VERSION]).toBe('3.0.7');
+        });
+        it('applies custom span attributes on every memory span', async () => {
+            const { applyCustomSpanAttributes } = jest.requireMock('../../helpers');
+            await runOp('memory get', async () => ({ id: 'm1' }), ['mem-123']);
+            expect(applyCustomSpanAttributes).toHaveBeenCalledWith(mockSpan);
+        });
+    });
+    describe('Operation: add', () => {
+        it('records message count, infer flag, and input content', async () => {
+            const messages = [
+                { role: 'user', content: 'I love coffee' },
+                { role: 'assistant', content: 'Noted' },
+            ];
+            await runOp('memory add', async () => ({ results: [{ id: 'm1' }] }), [
+                messages,
+                { userId: 'alice', infer: false },
+            ]);
+            const a = attrs();
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_COUNT]).toBe(2);
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_INFER]).toBe(false);
+            expect(a[semantic_convention_1.default.GEN_AI_USER_ID]).toBe('alice');
+            expect(a[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES]).toBe(JSON.stringify(messages));
+        });
+        it('captures user / agent / run scope from the config object', async () => {
+            await runOp('memory add', async () => [{ id: 'm1' }], [
+                [{ role: 'user', content: 'hi' }],
+                { userId: 'u1', agentId: 'a1', runId: 'r1', metadata: { topic: 'food' } },
+            ]);
+            const a = attrs();
+            expect(a[semantic_convention_1.default.GEN_AI_USER_ID]).toBe('u1');
+            expect(a[semantic_convention_1.default.GEN_AI_AGENT_ID]).toBe('a1');
+            expect(a[semantic_convention_1.default.GEN_AI_RUN_ID]).toBe('r1');
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_METADATA]).toBe(JSON.stringify({ topic: 'food' }));
+        });
+    });
+    describe('Operation: search', () => {
+        it('records query / limit (topK) / threshold and output content', async () => {
+            const response = { results: [{ id: 'm1', memory: 'likes coffee' }, { id: 'm2', memory: 'likes tea' }] };
+            await runOp('memory search', async () => response, [
+                'what drinks?',
+                { userId: 'alice', topK: 5, threshold: 0.7 },
+            ]);
+            const a = attrs();
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_SEARCH_QUERY]).toBe('what drinks?');
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_SEARCH_LIMIT]).toBe(5);
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_SEARCH_THRESHOLD]).toBe(0.7);
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_OPERATION_RESULT_COUNT]).toBe(2);
+            expect(a[semantic_convention_1.default.GEN_AI_DATA_SOURCES]).toBe(2);
+            expect(a[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES]).toBe(JSON.stringify(response));
+        });
+        it('reads scope from a nested filters object (camel or snake case)', async () => {
+            await runOp('memory search', async () => ({ results: [] }), [
+                'q',
+                { filters: { user_id: 'snake-user' } },
+            ]);
+            expect(attrs()[semantic_convention_1.default.GEN_AI_USER_ID]).toBe('snake-user');
+        });
+        it('reports result_count 0 for an empty results array (not 1)', async () => {
+            await runOp('memory search', async () => ({ results: [] }), ['q', { filters: { user_id: 'x' } }]);
+            const a = attrs();
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_OPERATION_RESULT_COUNT]).toBe(0);
+            expect(a[semantic_convention_1.default.GEN_AI_DATA_SOURCES]).toBe(0);
+        });
+    });
+    describe('Operations carrying a positional memory id', () => {
+        it('get -> db.operation.id', async () => {
+            await runOp('memory get', async () => ({ id: 'mem-1' }), ['mem-1']);
+            expect(attrs()[semantic_convention_1.default.DB_OPERATION_ID]).toBe('mem-1');
+        });
+        it('update -> db.update.id', async () => {
+            await runOp('memory update', async () => ({ message: 'ok' }), ['mem-2', 'new text']);
+            expect(attrs()[semantic_convention_1.default.DB_UPDATE_ID]).toBe('mem-2');
+        });
+        it('delete -> db.delete.id', async () => {
+            await runOp('memory delete', async () => ({ message: 'ok' }), ['mem-3']);
+            expect(attrs()[semantic_convention_1.default.DB_DELETE_ID]).toBe('mem-3');
+        });
+    });
+    describe('Operation: getAll / deleteAll', () => {
+        it('getAll uses the snake_case span name and reads scope from its first-arg config', async () => {
+            await runOp('memory get_all', async () => ({ results: [{ id: 'm1' }, { id: 'm2' }, { id: 'm3' }] }), [
+                { userId: 'bob' },
+            ]);
+            const a = attrs();
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('memory get_all', { kind: api_1.SpanKind.CLIENT });
+            expect(a[semantic_convention_1.default.GEN_AI_USER_ID]).toBe('bob');
+            expect(a[semantic_convention_1.default.GEN_AI_MEMORY_OPERATION_RESULT_COUNT]).toBe(3);
+        });
+        it('deleteAll uses the snake_case span name', async () => {
+            await runOp('memory delete_all', async () => ({ message: 'ok' }), [{ userId: 'bob' }]);
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('memory delete_all', { kind: api_1.SpanKind.CLIENT });
+        });
+        it('reset uses the Python endpoint span name', async () => {
+            await runOp('memory reset', async () => ({ message: 'ok' }), []);
+            expect(mockTracer.startSpan).toHaveBeenCalledWith('memory reset', { kind: api_1.SpanKind.CLIENT });
+            expect(attrs()[semantic_convention_1.default.GEN_AI_ENDPOINT]).toBe('memory reset');
+            expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_MEMORY);
+        });
+    });
+    describe('Content capture toggle', () => {
+        it('omits input/output content when captureMessageContent is false', async () => {
+            config_1.default.captureMessageContent = false;
+            await runOp('memory add', async () => ({ results: [] }), [
+                [{ role: 'user', content: 'secret' }],
+                {},
+            ]);
+            const keys = mockSpan.setAttribute.mock.calls.map(([k]) => k);
+            expect(keys).not.toContain(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES);
+        });
+    });
+    describe('Error path', () => {
+        it('records the exception, still ends the span, and rethrows', async () => {
+            const boom = new Error('mem0 down');
+            await expect(runOp('memory search', async () => {
+                throw boom;
+            }, ['q', { topK: 3 }])).rejects.toThrow('mem0 down');
+            expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, boom);
+            expect(mockSpan.end).toHaveBeenCalledTimes(1);
+            // Core attributes are still emitted on the error path (best-effort).
+            expect(attrs()[semantic_convention_1.default.GEN_AI_PROVIDER_NAME]).toBe('mem0');
+            expect(typeof attrs()[semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION]).toBe('number');
+        });
+    });
+});
+//# sourceMappingURL=mem0-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"mem0-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/mem0-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;GAWG;;;;;AAEH,4CAA8C;AAC9C,8EAAiG;AACjG,8DAA0C;AAC1C,0DAA2C;AAC3C,4DAA4C;AAC5C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,UAAU,EAAE,IAAI;IAChB,OAAO,EAAE,EAAE,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE;IACvC,yBAAyB,EAAE,IAAI,CAAC,EAAE,EAAE;CACrC,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE;YAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;SACf,CAAC;QACF,UAAU,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAEnD,gBAAuB,CAAC,WAAW,GAAG,mBAAmB,CAAC;QAC1D,gBAAuB,CAAC,eAAe,GAAG,gBAAgB,CAAC;QAC3D,gBAAuB,CAAC,qBAAqB,GAAG,IAAI,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,KAAK,UAAU,KAAK,CAAC,QAAgB,EAAE,IAA0B,EAAE,OAAc,EAAE;QACjF,MAAM,OAAO,GAAG,iBAAW,CAAC,qBAAqB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC;QAC9E,OAAO,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,oEAAoE;IACpE,SAAS,KAAK;QACZ,OAAO,MAAM,CAAC,WAAW,CAAE,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7E,CAAC;IAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAE/F,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC3F,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB;YAClF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAC/F,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAElB,MAAM,CAAC,CAAC,CAAC,8CAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACrD,gFAAgF;YAChF,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACrE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,CAAC;YACrG,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YACpF,MAAM,CAAC,CAAC,CAAC,wCAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpD,+EAA+E;YAC/E,oFAAoF;YACpF,MAAM,CAAC,OAAO,CAAC,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvE,MAAM,CAAC,OAAO,CAAC,CAAC,6BAAkB,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,OAAO,GAAG,iBAAW,CAAC,qBAAqB,CAAC,UAAU,EAAE,YAAY,EAAE,OAAO,CAAC,CAClF,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAC3B,CAAC;YACF,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,EAAE,yBAAyB,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;YACxE,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,yBAAyB,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,QAAQ,GAAG;gBACf,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE;gBAC1C,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;aACxC,CAAC;YACF,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE;gBACnE,QAAQ;gBACR,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE;aAClC,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAElB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACrF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;gBACpD,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBACjC,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;aAC1E,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAElB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACxG,MAAM,KAAK,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,QAAQ,EAAE;gBACjD,cAAc;gBACd,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE;aAC7C,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAElB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC9E,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,8BAA8B,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oCAAoC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3E,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,KAAK,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE;gBAC1D,GAAG;gBACH,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE;aACvC,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,KAAK,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;YAClG,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAClB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oCAAoC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3E,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4CAA4C,EAAE,GAAG,EAAE;QAC1D,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,KAAK,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC;YACrF,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,KAAK,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YACzE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;YAC9F,MAAM,KAAK,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE;gBACnG,EAAE,MAAM,EAAE,KAAK,EAAE;aAClB,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAElB,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/F,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oCAAoC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,KAAK,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YACvF,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACpG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,KAAK,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7F,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACzE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,CAAC;QAC7G,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC7E,gBAAuB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACvD,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE;gBACvD,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;gBACrC,EAAE;aACH,CAAC,CAAC;YACH,MAAM,IAAI,GAAI,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;YACvF,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,IAAI,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,MAAM,CACV,KAAK,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;gBAChC,MAAM,IAAI,CAAC;YACb,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CACvB,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAE/B,MAAM,CAAE,iBAAuB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACtF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,qEAAqE;YACrE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtE,MAAM,CAAC,OAAO,KAAK,EAAE,CAAC,6BAAkB,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,17 @@
+/**
+ * Cross-Language Trace Comparison Tests for Milvus Integration
+ *
+ * Verifies TypeScript SDK trace attributes match Python SDK.
+ *
+ * Python SDK reference: sdk/python/src/shield360/instrumentation/milvus/milvus.py
+ *
+ * Key alignment:
+ *   - db.system.name = 'milvus'
+ *   - db.operation.name: SEARCH, INSERT, UPSERT, DELETE, QUERY
+ *   - server.address, server.port (default localhost:19530)
+ *   - db.vector.query.top_k for search
+ *   - db.filter for expr/filter params
+ *   - db.vector.count for insert/upsert
+ *   - db.query.summary
+ */
+export {};

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

@@ -0,0 +1,206 @@
+"use strict";
+/**
+ * Cross-Language Trace Comparison Tests for Milvus Integration
+ *
+ * Verifies TypeScript SDK trace attributes match Python SDK.
+ *
+ * Python SDK reference: sdk/python/src/shield360/instrumentation/milvus/milvus.py
+ *
+ * Key alignment:
+ *   - db.system.name = 'milvus'
+ *   - db.operation.name: SEARCH, INSERT, UPSERT, DELETE, QUERY
+ *   - server.address, server.port (default localhost:19530)
+ *   - db.vector.query.top_k for search
+ *   - db.filter for expr/filter params
+ *   - db.vector.count for insert/upsert
+ *   - db.query.summary
+ */
+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("../milvus/wrapper"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+jest.mock('../../config');
+jest.mock('../../helpers');
+describe('Milvus Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            addEvent: jest.fn(),
+            end: jest.fn(),
+            setStatus: jest.fn(),
+        };
+        mockTracer = {
+            startSpan: jest.fn().mockReturnValue(mockSpan),
+        };
+        config_1.default.environment = 'shield360-testing';
+        config_1.default.applicationName = 'shield360-test';
+        config_1.default.captureMessageContent = true;
+        helpers_1.default.handleException = jest.fn();
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    // ── Common attributes ────────────────────────────────────────────────────
+    it('should set db.system.name = "milvus" matching Python DB_SYSTEM_MILVUS', async () => {
+        const patchFn = wrapper_1.default._patchSearch(mockTracer);
+        const originalMethod = jest.fn().mockResolvedValue({ results: [] });
+        const wrapped = patchFn(originalMethod);
+        await wrapped.call({}, { collection_name: 'test', vectors: [[0.1]], topk: 5 });
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_SYSTEM_NAME, 'milvus');
+    });
+    it('should set server.address = localhost and server.port = 19530', async () => {
+        const patchFn = wrapper_1.default._patchSearch(mockTracer);
+        const originalMethod = jest.fn().mockResolvedValue({ results: [] });
+        const wrapped = patchFn(originalMethod);
+        await wrapped.call({}, { collection_name: 'test', vectors: [[0.1]], topk: 5 });
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'localhost');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 19530);
+    });
+    // ── search → SEARCH ───────────────────────────────────────────────────────
+    describe('search() → SEARCH', () => {
+        async function runSearch(params, returnValue = { results: [] }) {
+            const patchFn = wrapper_1.default._patchSearch(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue(returnValue);
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, params);
+        }
+        it('should set db.operation.name = "SEARCH"', async () => {
+            await runSearch({ collection_name: 'embeddings', vectors: [[0.1]], topk: 10 });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'SEARCH');
+        });
+        it('should set db.collection.name from collection_name param', async () => {
+            await runSearch({ collection_name: 'my_embeddings', vectors: [[0.1]], topk: 5 });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_COLLECTION_NAME, 'my_embeddings');
+        });
+        it('should set db.vector.query.top_k from topk param', async () => {
+            await runSearch({ collection_name: 'col', vectors: [[0.1]], topk: 20 });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_VECTOR_QUERY_TOP_K, 20);
+        });
+        it('should also accept "limit" as top_k param', async () => {
+            await runSearch({ collection_name: 'col', vectors: [[0.1]], limit: 15 });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_VECTOR_QUERY_TOP_K, 15);
+        });
+        it('should set db.filter when expr provided', async () => {
+            await runSearch({
+                collection_name: 'col',
+                vectors: [[0.1]],
+                topk: 5,
+                expr: 'category == "books"',
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, 'category == "books"');
+        });
+        it('should set db.n_results from results length', async () => {
+            await runSearch({ collection_name: 'col', vectors: [[0.1]], topk: 5 }, { results: [{ id: 1 }, { id: 2 }] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_N_RESULTS, 2);
+        });
+    });
+    // ── insert → INSERT ───────────────────────────────────────────────────────
+    describe('insert() → INSERT (matches Python DB_OPERATION_INSERT)', () => {
+        async function runInsert(params) {
+            const patchFn = wrapper_1.default._patchInsert(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({ insert_cnt: params.data?.length ?? 0 });
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, params);
+        }
+        it('should set db.operation.name = "INSERT"', async () => {
+            await runInsert({ collection_name: 'col', data: [{ id: 1, vector: [0.1] }] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'INSERT');
+        });
+        it('should set db.vector.count from data length', async () => {
+            await runInsert({
+                collection_name: 'col',
+                data: [{ id: 1 }, { id: 2 }, { id: 3 }],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_VECTOR_COUNT, 3);
+        });
+        it('should also work with collectionName (camelCase) param', async () => {
+            const patchFn = wrapper_1.default._patchInsert(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({});
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, { collectionName: 'camelCase', data: [{ id: 1 }] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_COLLECTION_NAME, 'camelCase');
+        });
+    });
+    // ── upsert → UPSERT ───────────────────────────────────────────────────────
+    describe('upsert() → UPSERT', () => {
+        it('should set db.operation.name = "UPSERT" and db.vector.count', async () => {
+            const patchFn = wrapper_1.default._patchUpsert(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({});
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, {
+                collection_name: 'col',
+                data: [{ id: 1, vector: [0.1] }, { id: 2, vector: [0.2] }],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'UPSERT');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_VECTOR_COUNT, 2);
+        });
+    });
+    // ── delete → DELETE ───────────────────────────────────────────────────────
+    describe('delete() → DELETE', () => {
+        it('should set db.operation.name = "DELETE"', async () => {
+            const patchFn = wrapper_1.default._patchDelete(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({ delete_cnt: 2 });
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, {
+                collection_name: 'col',
+                ids: [1, 2],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'DELETE');
+        });
+        it('should set db.filter when expr provided', async () => {
+            const patchFn = wrapper_1.default._patchDelete(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({});
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, {
+                collection_name: 'col',
+                expr: 'id in [1, 2, 3]',
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, 'id in [1, 2, 3]');
+        });
+        it('should set db.ids_count when ids provided', async () => {
+            const patchFn = wrapper_1.default._patchDelete(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({});
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, { collection_name: 'col', ids: [10, 20, 30] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_ID_COUNT, 3);
+        });
+    });
+    // ── query → QUERY ─────────────────────────────────────────────────────────
+    describe('query() → QUERY (scalar filter query)', () => {
+        it('should set db.operation.name = "QUERY"', async () => {
+            const patchFn = wrapper_1.default._patchQuery(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({ data: [{ id: 1 }, { id: 2 }] });
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, {
+                collection_name: 'col',
+                expr: 'category == "books"',
+                limit: 50,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'QUERY');
+        });
+        it('should set db.n_results from data length', async () => {
+            const patchFn = wrapper_1.default._patchQuery(mockTracer);
+            const originalMethod = jest.fn().mockResolvedValue({ data: [{ id: 1 }, { id: 2 }, { id: 3 }] });
+            const wrapped = patchFn(originalMethod);
+            await wrapped.call({}, { collection_name: 'col', expr: 'id > 0' });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_N_RESULTS, 3);
+        });
+    });
+    // ── Error handling ────────────────────────────────────────────────────────
+    it('should call handleException and end span on error', async () => {
+        const patchFn = wrapper_1.default._patchSearch(mockTracer);
+        const error = new Error('Milvus not available');
+        const originalMethod = jest.fn().mockRejectedValue(error);
+        const wrapped = patchFn(originalMethod);
+        await expect(wrapped.call({}, { collection_name: 'col', vectors: [[0.1]], topk: 5 })).rejects.toThrow();
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, error);
+        expect(mockSpan.end).toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=milvus-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"milvus-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/milvus-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;GAeG;;;;;AAEH,gEAA8C;AAC9C,0DAA2C;AAC3C,4DAA4C;AAC5C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAE3B,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;IACtD,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;SACrB,CAAC;QAEF,UAAU,GAAG;YACX,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;SAC/C,CAAC;QAED,gBAAuB,CAAC,WAAW,GAAG,mBAAmB,CAAC;QAC1D,gBAAuB,CAAC,eAAe,GAAG,gBAAgB,CAAC;QAC3D,gBAAuB,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACrD,iBAAuB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAE5E,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAE/E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAE/E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QACnG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IAC5F,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,KAAK,UAAU,SAAS,CAAC,MAAW,EAAE,cAAmB,EAAE,OAAO,EAAE,EAAE,EAAE;YACtE,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACjC,CAAC;QAED,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,SAAS,CAAC,EAAE,eAAe,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAE/E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,SAAS,CAAC,EAAE,eAAe,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAEjF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,eAAe,CAChB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,SAAS,CAAC,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAExE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;QACnG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,SAAS,CAAC,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAEzE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;QACnG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,SAAS,CAAC;gBACd,eAAe,EAAE,KAAK;gBACtB,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;gBAChB,IAAI,EAAE,CAAC;gBACP,IAAI,EAAE,qBAAqB;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,SAAS,EAC5B,qBAAqB,CACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,SAAS,CACb,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EACrD,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CACpC,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACtE,KAAK,UAAU,SAAS,CAAC,MAAW;YAClC,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;YAC7F,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACjC,CAAC;QAED,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,SAAS,CAAC,EAAE,eAAe,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAE9E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,SAAS,CAAC;gBACd,eAAe,EAAE,KAAK;gBACtB,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;aACxC,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAE3E,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;gBACrB,eAAe,EAAE,KAAK;gBACtB,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC;aAC3D,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;YACtE,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;gBACrB,eAAe,EAAE,KAAK;gBACtB,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;aACZ,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;gBACrB,eAAe,EAAE,KAAK;gBACtB,IAAI,EAAE,iBAAiB;aACxB,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,SAAS,EAC5B,iBAAiB,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACvD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YAEtE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;QACrD,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,OAAO,GAAG,iBAAa,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YACtD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACrF,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;gBACrB,eAAe,EAAE,KAAK;gBACtB,IAAI,EAAE,qBAAqB;gBAC3B,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,OAAO,CACR,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,OAAO,GAAG,iBAAa,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YACtD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAChG,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEnE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,OAAO,GAAG,iBAAa,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAChD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QAExC,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACxG,MAAM,CAAC,iBAAe,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC9E,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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