shield360 1.0.0

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

@@ -0,0 +1,282 @@
+"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 = require("../langchain/wrapper");
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+// `buildOutputMessages` is the observable surface for the fix — we check that
+// it receives the normalised toolCalls array (not undefined).
+jest.mock('../../../src/helpers', () => ({
+    __esModule: true,
+    default: {
+        buildInputMessages: jest.fn(() => '[]'),
+        buildOutputMessages: jest.fn(() => '[]'),
+        updatePricingJson: jest.fn(async () => ({})),
+        getChatModelCost: jest.fn(() => 0),
+    },
+}));
+jest.mock('../../../src/config', () => ({
+    __esModule: true,
+    default: {
+        traceContent: true,
+        pricing_json: {},
+        updatePricingJson: jest.fn(async () => ({})),
+    },
+}));
+jest.mock('../../../src/instrumentation/base-wrapper', () => {
+    class MockBaseWrapper {
+    }
+    MockBaseWrapper.setBaseSpanAttributes = jest.fn();
+    MockBaseWrapper.recordMetrics = jest.fn();
+    return {
+        __esModule: true,
+        default: MockBaseWrapper,
+    };
+});
+const mockTracer = api_1.trace.getTracer('test-tracer');
+function makeSpan() {
+    const span = mockTracer.startSpan('test-span');
+    span.setAttribute = jest.fn();
+    return span;
+}
+function seedRun(handler, runId, span, modelName = 'test-model') {
+    // Reach into the private `spans` map so we don't need a real LangChain
+    // callback manager to set up a fake LLM run.
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    handler.spans.set(runId, {
+        span,
+        startTime: Date.now() - 10,
+        modelName,
+        streamingContent: [],
+        tokenTimestamps: [],
+        promptTokens: 0,
+        completionTokens: 0,
+    });
+}
+describe('LangChain wrapper — tool_calls propagation', () => {
+    let handler;
+    beforeEach(() => {
+        jest.clearAllMocks();
+        handler = new wrapper_1.Shield360CallbackHandler(mockTracer);
+    });
+    it('extracts tool_calls from AIMessage and forwards them to buildOutputMessages', async () => {
+        const span = makeSpan();
+        seedRun(handler, 'run-tc-1', span);
+        const output = {
+            generations: [
+                [
+                    {
+                        text: '',
+                        message: {
+                            content: '',
+                            tool_calls: [
+                                { id: 'call_1', name: 'get_weather', args: { city: 'Tokyo' } },
+                                { id: 'call_2', name: 'calculator', args: { expression: '2+2' } },
+                            ],
+                            response_metadata: { finish_reason: 'tool_calls' },
+                        },
+                    },
+                ],
+            ],
+        };
+        handler.handleLLMEnd(output, 'run-tc-1');
+        // handleLLMEnd → _finalizeLLMSpan is async
+        await new Promise((r) => setImmediate(r));
+        expect(helpers_1.default.buildOutputMessages).toHaveBeenCalledTimes(1);
+        const [text, finish, toolCalls] = helpers_1.default.buildOutputMessages.mock.calls[0];
+        expect(text).toBe('');
+        expect(finish).toBe('tool_calls');
+        expect(toolCalls).toEqual([
+            {
+                id: 'call_1',
+                type: 'function',
+                name: 'get_weather',
+                arguments: { city: 'Tokyo' },
+            },
+            {
+                id: 'call_2',
+                type: 'function',
+                name: 'calculator',
+                arguments: { expression: '2+2' },
+            },
+        ]);
+        // Flat attributes for easy filtering
+        const setAttr = span.setAttribute;
+        expect(setAttr).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_NAME, 'get_weather, calculator');
+        expect(setAttr).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_CALL_ID, 'call_1, call_2');
+        const toolArgsCall = setAttr.mock.calls.find((c) => c[0] === semantic_convention_1.default.GEN_AI_TOOL_CALL_ARGUMENTS);
+        expect(toolArgsCall).toBeTruthy();
+        expect(toolArgsCall?.[1]).toEqual(['{"city":"Tokyo"}', '{"expression":"2+2"}']);
+    });
+    it('passes undefined as toolCalls for text-only completions (preserves existing behaviour)', async () => {
+        const span = makeSpan();
+        seedRun(handler, 'run-tc-2', span);
+        const output = {
+            generations: [
+                [
+                    {
+                        text: 'Hello world',
+                        message: {
+                            content: 'Hello world',
+                            response_metadata: { finish_reason: 'stop' },
+                        },
+                    },
+                ],
+            ],
+        };
+        handler.handleLLMEnd(output, 'run-tc-2');
+        await new Promise((r) => setImmediate(r));
+        const calls = helpers_1.default.buildOutputMessages.mock.calls;
+        expect(calls).toHaveLength(1);
+        const [text, finish, toolCalls] = calls[0];
+        expect(text).toBe('Hello world');
+        expect(finish).toBe('stop');
+        expect(toolCalls).toBeUndefined();
+        // Flat tool attributes must NOT be set
+        const setAttr = span.setAttribute;
+        const toolNameCall = setAttr.mock.calls.find((c) => c[0] === semantic_convention_1.default.GEN_AI_TOOL_NAME);
+        expect(toolNameCall).toBeUndefined();
+    });
+    it('still emits output.messages when only tool_calls are present (no assistant text)', async () => {
+        const span = makeSpan();
+        seedRun(handler, 'run-tc-3', span);
+        const output = {
+            generations: [
+                [
+                    {
+                        text: '',
+                        message: {
+                            content: '',
+                            tool_calls: [{ id: 'call_only', name: 'ping', args: {} }],
+                            response_metadata: { finish_reason: 'tool_calls' },
+                        },
+                    },
+                ],
+            ],
+        };
+        handler.handleLLMEnd(output, 'run-tc-3');
+        await new Promise((r) => setImmediate(r));
+        const setAttr = span.setAttribute;
+        const outputMsgCall = setAttr.mock.calls.find((c) => c[0] === semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES);
+        expect(outputMsgCall).toBeTruthy();
+        expect(helpers_1.default.buildOutputMessages).toHaveBeenCalledTimes(1);
+    });
+    it('reads tool_calls from additional_kwargs when not on the top-level message', async () => {
+        const span = makeSpan();
+        seedRun(handler, 'run-tc-4', span);
+        const output = {
+            generations: [
+                [
+                    {
+                        text: '',
+                        message: {
+                            content: '',
+                            additional_kwargs: {
+                                tool_calls: [
+                                    {
+                                        id: 'legacy_1',
+                                        type: 'function',
+                                        function: { name: 'lookup', arguments: '{"k":"v"}' },
+                                    },
+                                ],
+                            },
+                            response_metadata: { finish_reason: 'tool_calls' },
+                        },
+                    },
+                ],
+            ],
+        };
+        handler.handleLLMEnd(output, 'run-tc-4');
+        await new Promise((r) => setImmediate(r));
+        const [, , toolCalls] = helpers_1.default.buildOutputMessages.mock.calls[0];
+        expect(toolCalls).toEqual([
+            { id: 'legacy_1', type: 'function', name: 'lookup', arguments: '{"k":"v"}' },
+        ]);
+    });
+    it('does not merge tool_calls across multiple generations (n > 1)', async () => {
+        // When the provider returns multiple choices, LangChain still collapses to
+        // one assistant message and `finishReason`/`completionContent` take the
+        // last writer. Tool calls must follow the same last-writer-wins rule —
+        // otherwise calls from different choices get flattened into one message
+        // and flat tool-name / id attributes become cross-choice joins.
+        const span = makeSpan();
+        seedRun(handler, 'run-tc-5', span);
+        const output = {
+            generations: [
+                [
+                    {
+                        text: '',
+                        message: {
+                            content: '',
+                            tool_calls: [{ id: 'choice_a_1', name: 'tool_a', args: {} }],
+                            response_metadata: { finish_reason: 'tool_calls' },
+                        },
+                    },
+                    {
+                        text: '',
+                        message: {
+                            content: '',
+                            tool_calls: [{ id: 'choice_b_1', name: 'tool_b', args: {} }],
+                            response_metadata: { finish_reason: 'tool_calls' },
+                        },
+                    },
+                ],
+            ],
+        };
+        handler.handleLLMEnd(output, 'run-tc-5');
+        await new Promise((r) => setImmediate(r));
+        const [, , toolCalls] = helpers_1.default.buildOutputMessages.mock.calls[0];
+        // Expect only the last generation's tool_calls — NOT both merged.
+        expect(toolCalls).toHaveLength(1);
+        expect(toolCalls[0].name).toBe('tool_b');
+        // Flat attributes must not be a cross-choice join like "tool_a, tool_b".
+        const setAttr = span.setAttribute;
+        expect(setAttr).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_NAME, 'tool_b');
+        expect(setAttr).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_CALL_ID, 'choice_b_1');
+    });
+    it('survives unserialisable tool arguments (circular refs / BigInt) and still ends the span', async () => {
+        // If JSON.stringify throws, the outer try/catch in _finalizeLLMSpan would
+        // swallow the error and leave the span un-ended + the `spans` entry
+        // leaked. Guard per-argument serialisation and fall back to a sentinel.
+        const span = makeSpan();
+        seedRun(handler, 'run-tc-6', span);
+        const circular = {};
+        circular.self = circular;
+        const output = {
+            generations: [
+                [
+                    {
+                        text: '',
+                        message: {
+                            content: '',
+                            tool_calls: [
+                                { id: 'call_circ', name: 'bad_args', args: circular },
+                                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                { id: 'call_big', name: 'big_args', args: { n: 10n } },
+                                { id: 'call_ok', name: 'ok_args', args: { x: 1 } },
+                            ],
+                            response_metadata: { finish_reason: 'tool_calls' },
+                        },
+                    },
+                ],
+            ],
+        };
+        expect(() => handler.handleLLMEnd(output, 'run-tc-6')).not.toThrow();
+        await new Promise((r) => setImmediate(r));
+        const setAttr = span.setAttribute;
+        const argsCall = setAttr.mock.calls.find((c) => c[0] === semantic_convention_1.default.GEN_AI_TOOL_CALL_ARGUMENTS);
+        expect(argsCall).toBeTruthy();
+        const [, argsValue] = argsCall;
+        expect(argsValue).toHaveLength(3);
+        expect(argsValue[0]).toBe('[unserializable]'); // circular
+        expect(argsValue[1]).toBe('[unserializable]'); // BigInt
+        expect(argsValue[2]).toBe('{"x":1}'); // ok
+        // Most important: the span was finalised, not leaked.
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        expect(handler.spans.has('run-tc-6')).toBe(false);
+    });
+});
+//# sourceMappingURL=langchain-wrapper.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"langchain-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/langchain-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,kDAAgE;AAChE,4DAA4C;AAC5C,oFAA2D;AAE3D,8EAA8E;AAC9E,8DAA8D;AAC9D,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,UAAU,EAAE,IAAI;IAChB,OAAO,EAAE;QACP,kBAAkB,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;QACvC,mBAAmB,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;QACxC,iBAAiB,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAC5C,gBAAgB,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;KACnC;CACF,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,UAAU,EAAE,IAAI;IAChB,OAAO,EAAE;QACP,YAAY,EAAE,IAAI;QAClB,YAAY,EAAE,EAAE;QAChB,iBAAiB,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;KAC7C;CACF,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE;IAC1D,MAAM,eAAe;;IACZ,qCAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAClC,6BAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAEnC,OAAO;QACL,UAAU,EAAE,IAAI;QAChB,OAAO,EAAE,eAAe;KACzB,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,UAAU,GAAG,WAAK,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;AAElD,SAAS,QAAQ;IACf,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC/C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAC9B,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,OAAO,CACd,OAAiC,EACjC,KAAa,EACb,IAAU,EACV,SAAS,GAAG,YAAY;IAExB,uEAAuE;IACvE,6CAA6C;IAC7C,8DAA8D;IAC7D,OAAe,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE;QAChC,IAAI;QACJ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE;QAC1B,SAAS;QACT,gBAAgB,EAAE,EAAE;QACpB,eAAe,EAAE,EAAE;QACnB,YAAY,EAAE,CAAC;QACf,gBAAgB,EAAE,CAAC;KACpB,CAAC,CAAC;AACL,CAAC;AAED,QAAQ,CAAC,4CAA4C,EAAE,GAAG,EAAE;IAC1D,IAAI,OAAiC,CAAC;IAEtC,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,OAAO,GAAG,IAAI,kCAAwB,CAAC,UAAU,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,MAAM,GAAG;YACb,WAAW,EAAE;gBACX;oBACE;wBACE,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE;4BACP,OAAO,EAAE,EAAE;4BACX,UAAU,EAAE;gCACV,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE;gCAC9D,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE;6BAClE;4BACD,iBAAiB,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;yBACnD;qBACF;iBACF;aACF;SACF,CAAC;QAEF,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACzC,2CAA2C;QAC3C,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,CAAC,iBAAe,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,GAAI,iBAAe,CAAC,mBAAiC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACnG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtB,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC;YACxB;gBACE,EAAE,EAAE,QAAQ;gBACZ,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,aAAa;gBACnB,SAAS,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;aAC7B;YACD;gBACE,EAAE,EAAE,QAAQ;gBACZ,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,YAAY;gBAClB,SAAS,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE;aACjC;SACF,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAyB,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,6BAAkB,CAAC,gBAAgB,EACnC,yBAAyB,CAC1B,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,6BAAkB,CAAC,mBAAmB,EACtC,gBAAgB,CACjB,CAAC;QACF,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAC1C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,6BAAkB,CAAC,0BAA0B,CAC9D,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,kBAAkB,EAAE,sBAAsB,CAAC,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;QACtG,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,MAAM,GAAG;YACb,WAAW,EAAE;gBACX;oBACE;wBACE,IAAI,EAAE,aAAa;wBACnB,OAAO,EAAE;4BACP,OAAO,EAAE,aAAa;4BACtB,iBAAiB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;yBAC7C;qBACF;iBACF;aACF;SACF,CAAC;QAEF,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACzC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,KAAK,GAAI,iBAAe,CAAC,mBAAiC,CAAC,IAAI,CAAC,KAAK,CAAC;QAC5E,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;QAElC,uCAAuC;QACvC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAyB,CAAC;QAC/C,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAC1C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,6BAAkB,CAAC,gBAAgB,CACpD,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,aAAa,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,MAAM,GAAG;YACb,WAAW,EAAE;gBACX;oBACE;wBACE,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE;4BACP,OAAO,EAAE,EAAE;4BACX,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;4BACzD,iBAAiB,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;yBACnD;qBACF;iBACF;aACF;SACF,CAAC;QAEF,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACzC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,OAAO,GAAG,IAAI,CAAC,YAAyB,CAAC;QAC/C,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAC3C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,6BAAkB,CAAC,sBAAsB,CAC1D,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,CAAC;QACnC,MAAM,CAAC,iBAAe,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,MAAM,GAAG;YACb,WAAW,EAAE;gBACX;oBACE;wBACE,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE;4BACP,OAAO,EAAE,EAAE;4BACX,iBAAiB,EAAE;gCACjB,UAAU,EAAE;oCACV;wCACE,EAAE,EAAE,UAAU;wCACd,IAAI,EAAE,UAAU;wCAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE;qCACrD;iCACF;6BACF;4BACD,iBAAiB,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;yBACnD;qBACF;iBACF;aACF;SACF,CAAC;QAEF,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACzC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,CAAC,EAAE,AAAD,EAAG,SAAS,CAAC,GAAI,iBAAe,CAAC,mBAAiC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzF,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC;YACxB,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE;SAC7E,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,2EAA2E;QAC3E,wEAAwE;QACxE,uEAAuE;QACvE,wEAAwE;QACxE,gEAAgE;QAChE,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,MAAM,GAAG;YACb,WAAW,EAAE;gBACX;oBACE;wBACE,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE;4BACP,OAAO,EAAE,EAAE;4BACX,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;4BAC5D,iBAAiB,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;yBACnD;qBACF;oBACD;wBACE,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE;4BACP,OAAO,EAAE,EAAE;4BACX,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;4BAC5D,iBAAiB,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;yBACnD;qBACF;iBACF;aACF;SACF,CAAC;QAEF,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACzC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,CAAC,EAAE,AAAD,EAAG,SAAS,CAAC,GAAI,iBAAe,CAAC,mBAAiC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzF,kEAAkE;QAClE,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEzC,yEAAyE;QACzE,MAAM,OAAO,GAAG,IAAI,CAAC,YAAyB,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAC;QACpF,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,0EAA0E;QAC1E,oEAAoE;QACpE,wEAAwE;QACxE,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,QAAQ,GAA4B,EAAE,CAAC;QAC7C,QAAQ,CAAC,IAAI,GAAG,QAAQ,CAAC;QAEzB,MAAM,MAAM,GAAG;YACb,WAAW,EAAE;gBACX;oBACE;wBACE,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE;4BACP,OAAO,EAAE,EAAE;4BACX,UAAU,EAAE;gCACV,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE;gCACrD,8DAA8D;gCAC9D,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC,EAAG,GAAW,EAAE,EAAE;gCAC/D,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;6BACnD;4BACD,iBAAiB,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;yBACnD;qBACF;iBACF;aACF;SACF,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACrE,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,OAAO,GAAG,IAAI,CAAC,YAAyB,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,6BAAkB,CAAC,0BAA0B,CAC9D,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,UAAU,EAAE,CAAC;QAC9B,MAAM,CAAC,EAAE,SAAS,CAAC,GAAG,QAAS,CAAC;QAChC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,WAAW;QAC1D,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;QACxD,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAW,KAAK;QAErD,sDAAsD;QACtD,8DAA8D;QAC9D,MAAM,CAAE,OAAe,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

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

@@ -0,0 +1,285 @@
+"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 core_1 = require("@opentelemetry/core");
+const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const wrapper_1 = __importDefault(require("../letta/wrapper"));
+const letta_1 = __importDefault(require("../letta"));
+jest.mock('../../config');
+jest.mock('../../helpers', () => ({
+    __esModule: true,
+    default: {
+        handleException: jest.fn(),
+        getChatModelCost: jest.fn().mockReturnValue(0.0025),
+        createStreamProxy: async (stream, gen) => new Proxy(stream, {
+            get(target, prop, receiver) {
+                if (prop === Symbol.asyncIterator)
+                    return () => gen;
+                return Reflect.get(target, prop, receiver);
+            },
+        }),
+    },
+    applyCustomSpanAttributes: jest.fn(),
+}));
+jest.mock('@opentelemetry/core', () => ({
+    isTracingSuppressed: jest.fn().mockReturnValue(false),
+}));
+const flush = () => new Promise((resolve) => setImmediate(resolve));
+describe('letta wrapper', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            setStatus: jest.fn(),
+            end: jest.fn(),
+            recordException: jest.fn(),
+            isRecording: jest.fn().mockReturnValue(true),
+        };
+        mockTracer = { startSpan: jest.fn(() => mockSpan) };
+        config_1.default.environment = 'shield360-testing';
+        config_1.default.applicationName = 'shield360-test';
+        config_1.default.captureMessageContent = true;
+        config_1.default.maxContentLength = null;
+        config_1.default.pricingInfo = { chat: {} };
+        core_1.isTracingSuppressed.mockReturnValue(false);
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    function attrs() {
+        return Object.fromEntries(mockSpan.setAttribute.mock.calls);
+    }
+    async function runOp(endpoint, impl, args, agentScoped = true) {
+        const wrapped = wrapper_1.default._patchOperation(mockTracer, endpoint, agentScoped, '1.12.1')(impl);
+        const result = await wrapped.apply({}, args);
+        await flush();
+        return result;
+    }
+    it('opens a create_agent span with agent metadata', async () => {
+        const agent = {
+            id: 'agent-123',
+            name: 'Support Bot',
+            agent_type: 'memgpt_agent',
+            system: 'You are a helpful assistant.',
+            llm_config: { model: 'gpt-4o-mini', provider_name: 'openai', context_window: 8192 },
+        };
+        await runOp('letta.create', async () => agent, [{ name: 'Support Bot', model: 'gpt-4o-mini' }]);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('create_agent support_bot', {
+            kind: api_1.SpanKind.CLIENT,
+            attributes: {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CREATE_AGENT,
+                [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_LETTA,
+            },
+        });
+        const a = attrs();
+        expect(a[semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]).toBe('shield360');
+        expect(a[semantic_conventions_1.ATTR_SERVICE_NAME]).toBe('shield360-test');
+        expect(a[semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]).toBe('shield360-testing');
+        expect(a[semantic_convention_1.default.GEN_AI_PROVIDER_NAME]).toBe('letta');
+        expect(a[semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]).toBe('letta');
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('create_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_SDK_VERSION]).toBe('1.12.1');
+        expect(a[semantic_convention_1.default.SERVER_ADDRESS]).toBe('api.letta.com');
+        expect(a[semantic_convention_1.default.SERVER_PORT]).toBe(443);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_ID]).toBe('agent-123');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_NAME]).toBe('Support Bot');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_TYPE]).toBe('memgpt_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_INSTRUCTIONS]).toBe('You are a helpful assistant.');
+        expect(a[semantic_convention_1.default.GEN_AI_REQUEST_MODEL]).toBe('gpt-4o-mini');
+        expect(a[semantic_convention_1.default.GEN_AI_REQUEST_CONTEXT_WINDOW]).toBe(8192);
+        expect(typeof a[semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION]).toBe('number');
+        expect(mockSpan.setStatus).toHaveBeenCalledWith({ code: 1 });
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('applies custom span attributes on every span', async () => {
+        const { applyCustomSpanAttributes } = jest.requireMock('../../helpers');
+        await runOp('letta.retrieve', async () => ({ id: 'agent-1' }), ['agent-1']);
+        expect(applyCustomSpanAttributes).toHaveBeenCalledWith(mockSpan);
+    });
+    it('names a message create span chat {model} and records usage + cost', async () => {
+        const response = {
+            messages: [{ message_type: 'assistant_message', content: 'Hi there' }],
+            usage: { prompt_tokens: 12, completion_tokens: 4, total_tokens: 16, step_count: 1 },
+        };
+        await runOp('letta.create_message', async () => response, [
+            'agent-1',
+            { messages: [{ role: 'user', content: 'hello' }], model: 'gpt-4o' },
+        ]);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('chat gpt-4o', {
+            kind: api_1.SpanKind.CLIENT,
+            attributes: {
+                [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_LETTA,
+            },
+        });
+        const a = attrs();
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('chat');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_ID]).toBe('agent-1');
+        expect(a[semantic_convention_1.default.GEN_AI_REQUEST_MESSAGE_COUNT]).toBe(1);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]).toBe(12);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]).toBe(4);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS]).toBe(16);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_STEP_COUNT]).toBe(1);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_COST]).toBe(0.0025);
+        expect(JSON.parse(a[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES])).toEqual([
+            { role: 'user', content: 'hello' },
+        ]);
+        expect(JSON.parse(a[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES])).toEqual(response.messages);
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('wraps a streaming message response and records streaming attributes', async () => {
+        async function* stream() {
+            yield { message_type: 'assistant_message', content: 'Hel' };
+            yield { message_type: 'assistant_message', content: 'lo' };
+            yield {
+                message_type: 'usage_statistics',
+                prompt_tokens: 8,
+                completion_tokens: 2,
+                total_tokens: 10,
+            };
+        }
+        const proxy = await runOp('letta.create_stream', () => stream(), [
+            'agent-1',
+            { messages: [{ role: 'user', content: 'hi' }] },
+        ]);
+        const seen = [];
+        for await (const chunk of proxy) {
+            seen.push(chunk);
+        }
+        await flush();
+        expect(seen).toHaveLength(3);
+        const a = attrs();
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('chat');
+        expect(a[semantic_convention_1.default.GEN_AI_STREAMING_CHUNK_COUNT]).toBe(3);
+        expect(a[semantic_convention_1.default.GEN_AI_STREAMING_RESPONSE_COUNT]).toBe(3);
+        expect(a[semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM]).toBe(true);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]).toBe(8);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]).toBe(2);
+        expect(mockSpan.setStatus).toHaveBeenCalledWith({ code: 1 });
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('names list operations workflow {method}', async () => {
+        await runOp('letta.list', async () => [{ id: 'b1' }, { id: 'b2' }], [{}]);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('workflow list', expect.objectContaining({ kind: api_1.SpanKind.CLIENT }));
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('workflow');
+    });
+    it('names attach/detach operations tool {method} with execute_tool op', async () => {
+        await runOp('letta.attach', async () => ({ id: 'agent-1' }), ['tool-1', { agent_id: 'agent-1' }]);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('tool attach', expect.objectContaining({ kind: api_1.SpanKind.CLIENT }));
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('execute_tool');
+    });
+    it('records the exception, ends the span, and rethrows on a chat error', async () => {
+        const boom = new Error('letta down');
+        const wrapped = wrapper_1.default._patchOperation(mockTracer, 'letta.create_message', true)(async () => {
+            throw boom;
+        });
+        await expect(wrapped.apply({}, ['agent-1', { messages: [] }])).rejects.toThrow('letta down');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, boom);
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+        expect(attrs()[semantic_convention_1.default.GEN_AI_PROVIDER_NAME]).toBe('letta');
+    });
+    it('ends the stream span exactly once when the consumer breaks early', async () => {
+        async function* stream() {
+            yield { message_type: 'assistant_message', content: 'one' };
+            yield { message_type: 'assistant_message', content: 'two' };
+        }
+        const proxy = await runOp('letta.create_stream', () => stream(), [
+            'agent-1',
+            { messages: [{ role: 'user', content: 'hi' }] },
+        ]);
+        for await (const _chunk of proxy) {
+            break;
+        }
+        await flush();
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('does not treat a top-level resource id as an agent id', async () => {
+        await runOp('letta.retrieve', async () => ({ label: 'persona' }), ['block-id'], false);
+        expect(attrs()[semantic_convention_1.default.GEN_AI_AGENT_ID]).not.toBe('block-id');
+    });
+    it('maps an agent update to the invoke_agent operation', async () => {
+        await runOp('letta.modify', async () => ({ id: 'agent-1' }), ['agent-1', { name: 'renamed' }]);
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('invoke_agent');
+    });
+    it('rejects and ends the span on a non-chat operation error', async () => {
+        const boom = new Error('list failed');
+        const wrapped = wrapper_1.default._patchOperation(mockTracer, 'letta.list', false)(async () => {
+            throw boom;
+        });
+        await expect(wrapped.apply({}, [{}])).rejects.toThrow('list failed');
+        await flush();
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, boom);
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('bypasses instrumentation when tracing is suppressed', async () => {
+        core_1.isTracingSuppressed.mockReturnValue(true);
+        const original = jest.fn().mockResolvedValue({ id: 'agent-1' });
+        const wrapped = wrapper_1.default._patchOperation(mockTracer, 'letta.create', false)(original);
+        await wrapped.apply({}, [{ name: 'bot' }]);
+        expect(original).toHaveBeenCalled();
+        expect(mockTracer.startSpan).not.toHaveBeenCalled();
+    });
+});
+describe('LettaInstrumentation patch targets', () => {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const { isWrapped } = require('@opentelemetry/instrumentation');
+    it('wraps resource-client prototype methods via manualPatch', () => {
+        class Agents {
+            async create() {
+                return null;
+            }
+            async retrieve() {
+                return null;
+            }
+            async update() {
+                return null;
+            }
+            async delete() {
+                return null;
+            }
+            async list() {
+                return null;
+            }
+        }
+        class Messages {
+            async create() {
+                return null;
+            }
+            async stream() {
+                return null;
+            }
+            async createAsync() {
+                return null;
+            }
+            async list() {
+                return null;
+            }
+            async cancel() {
+                return null;
+            }
+            async reset() {
+                return null;
+            }
+        }
+        const fakeModule = { Agents, Messages };
+        const instrumentation = new letta_1.default();
+        instrumentation.manualPatch(fakeModule);
+        for (const method of ['create', 'retrieve', 'update', 'delete', 'list']) {
+            expect(isWrapped(Agents.prototype[method])).toBe(true);
+        }
+        for (const method of ['create', 'stream', 'createAsync', 'list', 'cancel', 'reset']) {
+            expect(isWrapped(Messages.prototype[method])).toBe(true);
+        }
+        // Idempotent: a second patch must not throw or rewrap.
+        instrumentation.manualPatch(fakeModule);
+        expect(isWrapped(Agents.prototype.create)).toBe(true);
+    });
+});
+//# sourceMappingURL=letta-wrapper.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"letta-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/letta-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAA8C;AAC9C,8CAA0D;AAC1D,8EAAiG;AACjG,0DAA2C;AAC3C,4DAA4C;AAC5C,oFAA2D;AAC3D,+DAA4C;AAC5C,qDAA4C;AAE5C,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;QACP,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE;QAC1B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC;QACnD,iBAAiB,EAAE,KAAK,EAAE,MAAW,EAAE,GAAQ,EAAE,EAAE,CACjD,IAAI,KAAK,CAAC,MAAM,EAAE;YAChB,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;gBACxB,IAAI,IAAI,KAAK,MAAM,CAAC,aAAa;oBAAE,OAAO,GAAG,EAAE,CAAC,GAAG,CAAC;gBACpD,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC;KACL;IACD,yBAAyB,EAAE,IAAI,CAAC,EAAE,EAAE;CACrC,CAAC,CAAC,CAAC;AACJ,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;CACtD,CAAC,CAAC,CAAC;AAEJ,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;AAEpE,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,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,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE;YAC1B,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;SAC7C,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;QACrD,gBAAuB,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAChD,gBAAuB,CAAC,WAAW,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QACnD,0BAAiC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,KAAK;QACZ,OAAO,MAAM,CAAC,WAAW,CAAE,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,UAAU,KAAK,CAClB,QAAgB,EAChB,IAA0B,EAC1B,IAAW,EACX,WAAW,GAAG,IAAI;QAElB,MAAM,OAAO,GAAG,iBAAY,CAAC,eAAe,CAAC,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC;QAChG,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC7C,MAAM,KAAK,EAAE,CAAC;QACd,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,WAAW;YACf,IAAI,EAAE,aAAa;YACnB,UAAU,EAAE,cAAc;YAC1B,MAAM,EAAE,8BAA8B;YACtC,UAAU,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE;SACpF,CAAC;QACF,MAAM,KAAK,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;QAEhG,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,0BAA0B,EAAE;YAC5E,IAAI,EAAE,cAAQ,CAAC,MAAM;YACrB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,kCAAkC;gBAC5F,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,6BAAkB,CAAC,mBAAmB;aACvF;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;QAClB,MAAM,CAAC,CAAC,CAAC,8CAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,CAAC,CAAC,CAAC,wCAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACpD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACpF,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACpE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACrE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAC7F,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACvE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvE,MAAM,CAAC,OAAO,CAAC,CAAC,6BAAkB,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrF,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,yBAAyB,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACxE,MAAM,KAAK,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5E,MAAM,CAAC,yBAAyB,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,CAAC,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;YACtE,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE;SACpF,CAAC;QACF,MAAM,KAAK,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,QAAQ,EAAE;YACxD,SAAS;YACT,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE;SACpE,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,aAAa,EAAE;YAC/D,IAAI,EAAE,cAAQ,CAAC,MAAM;YACrB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;gBACpF,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,6BAAkB,CAAC,mBAAmB;aACvF;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;QAClB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACtE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;SACnC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5F,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,KAAK,SAAS,CAAC,CAAC,MAAM;YACpB,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC5D,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC3D,MAAM;gBACJ,YAAY,EAAE,kBAAkB;gBAChC,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,CAAC;gBACpB,YAAY,EAAE,EAAE;aACjB,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,EAAE;YAC/D,SAAS;YACT,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE;SAChD,CAAC,CAAC;QAEH,MAAM,IAAI,GAAU,EAAE,CAAC;QACvB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnB,CAAC;QACD,MAAM,KAAK,EAAE,CAAC;QAEd,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;QAClB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,+BAA+B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAE1E,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAC/C,eAAe,EACf,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CACnD,CAAC;QACF,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,KAAK,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAElG,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAC/C,aAAa,EACb,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CACnD,CAAC;QACF,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,IAAI,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QACrC,MAAM,OAAO,GAAG,iBAAY,CAAC,eAAe,CAAC,UAAU,EAAE,sBAAsB,EAAE,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE;YAChG,MAAM,IAAI,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAE7F,MAAM,CAAE,iBAAuB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACtF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,KAAK,SAAS,CAAC,CAAC,MAAM;YACpB,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC5D,MAAM,EAAE,YAAY,EAAE,mBAAmB,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC9D,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,EAAE;YAC/D,SAAS;YACT,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE;SAChD,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,KAAK,EAAE,CAAC;YACjC,MAAM;QACR,CAAC;QACD,MAAM,KAAK,EAAE,CAAC;QAEd,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,KAAK,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;QACvF,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,KAAK,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAC/F,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,IAAI,GAAG,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;QACtC,MAAM,OAAO,GAAG,iBAAY,CAAC,eAAe,CAAC,UAAU,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC,KAAK,IAAI,EAAE;YACvF,MAAM,IAAI,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACrE,MAAM,KAAK,EAAE,CAAC;QAEd,MAAM,CAAE,iBAAuB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACtF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QAClE,0BAAiC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,iBAAY,CAAC,eAAe,CAAC,UAAU,EAAE,cAAc,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC;QAE1F,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,8DAA8D;IAC9D,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC;IAEhE,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,MAAM;YACV,KAAK,CAAC,MAAM;gBACV,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,QAAQ;gBACZ,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,MAAM;gBACV,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,MAAM;gBACV,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,IAAI;gBACR,OAAO,IAAI,CAAC;YACd,CAAC;SACF;QACD,MAAM,QAAQ;YACZ,KAAK,CAAC,MAAM;gBACV,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,MAAM;gBACV,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,WAAW;gBACf,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,IAAI;gBACR,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,MAAM;gBACV,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,KAAK;gBACT,OAAO,IAAI,CAAC;YACd,CAAC;SACF;QAED,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAS,CAAC;QAC/C,MAAM,eAAe,GAAG,IAAI,eAAoB,EAAE,CAAC;QACnD,eAAe,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QAExC,KAAK,MAAM,MAAM,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC;YACxE,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,MAAuC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1F,CAAC;QACD,KAAK,MAAM,MAAM,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;YACpF,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAyC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9F,CAAC;QAED,uDAAuD;QACvD,eAAe,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACxC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,13 @@
+/**
+ * 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.
+ */
+export {};