shield360 1.0.0

package/dist/instrumentation/mem0/index.js

@@ -0,0 +1,113 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const api_1 = require("@opentelemetry/api");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+/**
+ * mem0 method name -> span name. Span names reuse the Python endpoint strings
+ * (snake_case `get_all` / `delete_all`) so TS and Python emit identical span names,
+ * even though the JS methods are camelCase (`getAll` / `deleteAll`).
+ */
+const MEM0_METHODS = [
+    ['add', 'memory add'],
+    ['search', 'memory search'],
+    ['get', 'memory get'],
+    ['getAll', 'memory get_all'],
+    ['update', 'memory update'],
+    ['delete', 'memory delete'],
+    ['deleteAll', 'memory delete_all'],
+    ['history', 'memory history'],
+    ['reset', 'memory reset'],
+];
+class Shield360Mem0Instrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-mem0`, '1.0.0', config);
+    }
+    init() {
+        // mem0ai ships two clients at two specifiers:
+        //   - `mem0ai`       -> hosted MemoryClient (the package entry, patched below)
+        //   - `mem0ai/oss`   -> self-hosted Memory (an internal file)
+        // require-in-the-middle reaches internal files via the `files` array, so the OSS
+        // entry is registered as InstrumentationNodeModuleFile(s) on the main definition.
+        // A few candidate paths are listed to stay robust across build layouts.
+        const ossFiles = ['mem0ai/dist/oss/index.js', 'mem0ai/dist/oss/index.cjs', 'mem0ai/oss/index.js'].map((filePath) => new instrumentation_1.InstrumentationNodeModuleFile(filePath, ['>=0.1.32'], (moduleExports, moduleVersion) => {
+            this._patchOss(moduleExports, moduleVersion);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined)
+                this._unpatchOss(moduleExports);
+        }));
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('mem0ai', ['>=0.1.32'], (moduleExports, moduleVersion) => {
+            this._patch(moduleExports, moduleVersion);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined)
+                this._unpatch(moduleExports);
+        }, ossFiles);
+        return [module];
+    }
+    manualPatch(mem0) {
+        // Manual entry point (shield360.init({ instrumentations: { mem0 } })). The caller may
+        // pass either the hosted module/class or the OSS one, so try both resolvers — the
+        // isWrapped guard in _patchClass keeps it idempotent.
+        this._patch(mem0);
+        this._patchOss(mem0);
+    }
+    /** Patch the hosted MemoryClient (default export of `mem0ai`). */
+    _patch(moduleExports, moduleVersion) {
+        const ClientClass = moduleExports?.MemoryClient ?? moduleExports?.default ?? moduleExports;
+        this._patchClass(ClientClass, moduleVersion);
+    }
+    /** Patch the self-hosted Memory class (`mem0ai/oss`). */
+    _patchOss(moduleExports, moduleVersion) {
+        const MemoryClass = moduleExports?.Memory ?? moduleExports?.default ?? moduleExports;
+        this._patchClass(MemoryClass, moduleVersion);
+    }
+    _patchClass(target, moduleVersion) {
+        try {
+            const proto = target?.prototype;
+            if (!proto)
+                return;
+            for (const [method, spanName] of MEM0_METHODS) {
+                if (typeof proto[method] !== 'function')
+                    continue;
+                if ((0, instrumentation_1.isWrapped)(proto[method])) {
+                    this._unwrap(proto, method);
+                }
+                this._wrap(proto, method, wrapper_1.default._patchMemoryOperation(this.tracer, spanName, moduleVersion));
+            }
+        }
+        catch (e) {
+            api_1.diag.error('mem0 instrumentation: error in _patch method', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        const ClientClass = moduleExports?.MemoryClient ?? moduleExports?.default ?? moduleExports;
+        this._unpatchClass(ClientClass);
+    }
+    _unpatchOss(moduleExports) {
+        const MemoryClass = moduleExports?.Memory ?? moduleExports?.default ?? moduleExports;
+        this._unpatchClass(MemoryClass);
+    }
+    _unpatchClass(target) {
+        try {
+            const proto = target?.prototype;
+            if (!proto)
+                return;
+            for (const [method] of MEM0_METHODS) {
+                if (typeof proto[method] === 'function' && (0, instrumentation_1.isWrapped)(proto[method])) {
+                    this._unwrap(proto, method);
+                }
+            }
+        }
+        catch {
+            /* ignore */
+        }
+    }
+}
+exports.default = Shield360Mem0Instrumentation;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/mem0/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/mem0/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAMwC;AAExC,4CAA0C;AAC1C,6CAAwD;AACxD,wDAAoC;AAIpC;;;;GAIG;AACH,MAAM,YAAY,GAA4B;IAC5C,CAAC,KAAK,EAAE,YAAY,CAAC;IACrB,CAAC,QAAQ,EAAE,eAAe,CAAC;IAC3B,CAAC,KAAK,EAAE,YAAY,CAAC;IACrB,CAAC,QAAQ,EAAE,gBAAgB,CAAC;IAC5B,CAAC,QAAQ,EAAE,eAAe,CAAC;IAC3B,CAAC,QAAQ,EAAE,eAAe,CAAC;IAC3B,CAAC,WAAW,EAAE,mBAAmB,CAAC;IAClC,CAAC,SAAS,EAAE,gBAAgB,CAAC;IAC7B,CAAC,OAAO,EAAE,cAAc,CAAC;CAC1B,CAAC;AAEF,MAAqB,4BAA6B,SAAQ,qCAAmB;IAC3E,YAAY,SAAoC,EAAE;QAChD,KAAK,CAAC,GAAG,iCAAsB,uBAAuB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAES,IAAI;QACZ,8CAA8C;QAC9C,+EAA+E;QAC/E,8DAA8D;QAC9D,iFAAiF;QACjF,kFAAkF;QAClF,wEAAwE;QACxE,MAAM,QAAQ,GAAG,CAAC,0BAA0B,EAAE,2BAA2B,EAAE,qBAAqB,CAAC,CAAC,GAAG,CACnG,CAAC,QAAQ,EAAE,EAAE,CACX,IAAI,+CAA6B,CAC/B,QAAQ,EACR,CAAC,UAAU,CAAC,EACZ,CAAC,aAAkB,EAAE,aAAsB,EAAE,EAAE;YAC7C,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;YAC7C,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAkB,EAAE,EAAE;YACrB,IAAI,aAAa,KAAK,SAAS;gBAAE,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;QACnE,CAAC,CACF,CACJ,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,QAAQ,EACR,CAAC,UAAU,CAAC,EACZ,CAAC,aAAkB,EAAE,aAAsB,EAAE,EAAE;YAC7C,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;YAC1C,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAkB,EAAE,EAAE;YACrB,IAAI,aAAa,KAAK,SAAS;gBAAE,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAChE,CAAC,EACD,QAAQ,CACT,CAAC;QAEF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,IAAS;QAC1B,sFAAsF;QACtF,kFAAkF;QAClF,sDAAsD;QACtD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAClB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAED,kEAAkE;IACxD,MAAM,CAAC,aAAkB,EAAE,aAAsB;QACzD,MAAM,WAAW,GAAG,aAAa,EAAE,YAAY,IAAI,aAAa,EAAE,OAAO,IAAI,aAAa,CAAC;QAC3F,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED,yDAAyD;IAC/C,SAAS,CAAC,aAAkB,EAAE,aAAsB;QAC5D,MAAM,WAAW,GAAG,aAAa,EAAE,MAAM,IAAI,aAAa,EAAE,OAAO,IAAI,aAAa,CAAC;QACrF,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAC/C,CAAC;IAEO,WAAW,CAAC,MAAW,EAAE,aAAsB;QACrD,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,EAAE,SAAS,CAAC;YAChC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,KAAK,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,YAAY,EAAE,CAAC;gBAC9C,IAAI,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,UAAU;oBAAE,SAAS;gBAClD,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;oBAC7B,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC9B,CAAC;gBACD,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,iBAAW,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC;YACrG,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,UAAI,CAAC,KAAK,CAAC,8CAA8C,EAAE,CAAC,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,MAAM,WAAW,GAAG,aAAa,EAAE,YAAY,IAAI,aAAa,EAAE,OAAO,IAAI,aAAa,CAAC;QAC3F,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;IAClC,CAAC;IAES,WAAW,CAAC,aAAkB;QACtC,MAAM,WAAW,GAAG,aAAa,EAAE,MAAM,IAAI,aAAa,EAAE,OAAO,IAAI,aAAa,CAAC;QACrF,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;IAClC,CAAC;IAEO,aAAa,CAAC,MAAW;QAC/B,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,EAAE,SAAS,CAAC;YAChC,IAAI,CAAC,KAAK;gBAAE,OAAO;YACnB,KAAK,MAAM,CAAC,MAAM,CAAC,IAAI,YAAY,EAAE,CAAC;gBACpC,IAAI,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,UAAU,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;oBACpE,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,YAAY;QACd,CAAC;IACH,CAAC;CACF;AAvGD,+CAuGC"}

package/dist/instrumentation/mem0/wrapper.d.ts

@@ -0,0 +1,37 @@
+import { Tracer } from '@opentelemetry/api';
+/**
+ * Wrapper for mem0 memory-layer operations. mem0 exposes the same method surface on
+ * the hosted `MemoryClient` (`mem0ai`) and the self-hosted `Memory` (`mem0ai/oss`):
+ * add / search / get / getAll / update / delete / deleteAll / history. Every method
+ * returns a Promise, so a single async wrapper handles them all. The emitted spans
+ * mirror the Python reference (sdk/python/src/shield360/instrumentation/mem0): one
+ * CLIENT span per call named `memory <op>`, with `gen_ai.*` attributes. No tokens,
+ * model, cost, or metrics are involved (matching Python) so this does not use the
+ * chat-oriented BaseWrapper.
+ */
+declare class Mem0Wrapper {
+    static aiSystem: string;
+    /**
+     * Returns a wrapper (over an original Promise-returning method) that emits one
+     * `memory <op>` CLIENT span. `spanName` is the Python endpoint string, e.g.
+     * `memory add` or `memory get_all`.
+     */
+    static _patchMemoryOperation(tracer: Tracer, spanName: string, version?: string): any;
+    /** Sets every non-content attribute (core + scope + memory + operation + response). */
+    static _setSpanAttributes(span: any, spanName: string, args: any[], response?: any, sdkVersion?: string): void;
+    private static _setOperationAttributes;
+    private static _setResponseAttributes;
+    /** Captures input (add) / output (search) content when capture is enabled. */
+    static _setContentAttributes(span: any, spanName: string, args: any[], response: any): void;
+    /**
+     * Returns the options/config object for an operation, accounting for the fact that
+     * its argument position differs by method: add/search take it as the 2nd arg, while
+     * getAll/deleteAll take it as the 1st. get/update/history take a positional id and
+     * carry no scope config.
+     */
+    private static _extractConfig;
+    private static _scopeValue;
+    private static _hasMessages;
+    private static _safeStringify;
+}
+export default Mem0Wrapper;

package/dist/instrumentation/mem0/wrapper.js

@@ -0,0 +1,298 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+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 config_1 = __importDefault(require("../../config"));
+const constant_1 = require("../../constant");
+const helpers_1 = __importStar(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+// `gen_ai.sdk.version` reports the *instrumented* package version (mem0ai), matching
+// the Python SDK (importlib.metadata.version("mem0ai")) rather than Shield360's version.
+// The reliable source is the version OTel passes to the patch hook, threaded in as the
+// `version` arg of _patchMemoryOperation. This module-level require is only a fallback
+// for manualPatch — and note mem0ai's "exports" map blocks `mem0ai/package.json`, so it
+// resolves to 'unknown' for mem0ai itself; that's fine, the OTel-provided version wins.
+let mem0SdkVersion = 'unknown';
+try {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    mem0SdkVersion = require('mem0ai/package.json')?.version || 'unknown';
+}
+catch {
+    mem0SdkVersion = 'unknown';
+}
+/**
+ * Wrapper for mem0 memory-layer operations. mem0 exposes the same method surface on
+ * the hosted `MemoryClient` (`mem0ai`) and the self-hosted `Memory` (`mem0ai/oss`):
+ * add / search / get / getAll / update / delete / deleteAll / history. Every method
+ * returns a Promise, so a single async wrapper handles them all. The emitted spans
+ * mirror the Python reference (sdk/python/src/shield360/instrumentation/mem0): one
+ * CLIENT span per call named `memory <op>`, with `gen_ai.*` attributes. No tokens,
+ * model, cost, or metrics are involved (matching Python) so this does not use the
+ * chat-oriented BaseWrapper.
+ */
+class Mem0Wrapper {
+    /**
+     * Returns a wrapper (over an original Promise-returning method) that emits one
+     * `memory <op>` CLIENT span. `spanName` is the Python endpoint string, e.g.
+     * `memory add` or `memory get_all`.
+     */
+    static _patchMemoryOperation(tracer, spanName, version) {
+        const sdkVersion = version || mem0SdkVersion;
+        return (originalMethod) => {
+            return async function (...args) {
+                const span = tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT });
+                const startTime = Date.now();
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const duration = (Date.now() - startTime) / 1000;
+                        Mem0Wrapper._setSpanAttributes(span, spanName, args, response, sdkVersion);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                        Mem0Wrapper._setContentAttributes(span, spanName, args, response);
+                        span.setStatus({ code: api_1.SpanStatusCode.OK });
+                        return response;
+                    }
+                    catch (e) {
+                        const duration = (Date.now() - startTime) / 1000;
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                        try {
+                            Mem0Wrapper._setSpanAttributes(span, spanName, args, undefined, sdkVersion);
+                        }
+                        catch {
+                            /* best-effort attributes on the error path */
+                        }
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    /** Sets every non-content attribute (core + scope + memory + operation + response). */
+    static _setSpanAttributes(span, spanName, args, response, sdkVersion = mem0SdkVersion) {
+        const applicationName = config_1.default.applicationName || '';
+        const environment = config_1.default.environment || '';
+        // Core attributes (set on every memory span).
+        span.setAttribute(semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME, constant_1.SDK_NAME);
+        // Set both provider-name keys to 'mem0': gen_ai.system (the key Python mem0 uses)
+        // and gen_ai.provider.name (the OTel framework convention). Same value either way.
+        span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, Mem0Wrapper.aiSystem);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL, Mem0Wrapper.aiSystem);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_ENDPOINT, spanName);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_MEMORY);
+        span.setAttribute(semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT, environment);
+        span.setAttribute(semantic_conventions_1.ATTR_SERVICE_NAME, applicationName);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, sdkVersion);
+        const config = Mem0Wrapper._extractConfig(spanName, args);
+        // Session scope — present either at the top level of the config object or inside a
+        // nested `filters` object, in camelCase (v3 SDK) or snake_case (filters / hosted API).
+        const userId = Mem0Wrapper._scopeValue(config, 'userId', 'user_id');
+        const agentId = Mem0Wrapper._scopeValue(config, 'agentId', 'agent_id');
+        const runId = Mem0Wrapper._scopeValue(config, 'runId', 'run_id');
+        if (userId !== undefined)
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USER_ID, userId);
+        if (agentId !== undefined)
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_ID, agentId);
+        if (runId !== undefined)
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RUN_ID, runId);
+        // Memory type / metadata.
+        const memoryType = config?.memoryType ?? config?.memory_type;
+        if (memoryType !== undefined) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_TYPE, memoryType);
+        }
+        if (config?.metadata !== undefined && config?.metadata !== null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_METADATA, Mem0Wrapper._safeStringify(config.metadata));
+        }
+        Mem0Wrapper._setOperationAttributes(span, spanName, args, config);
+        Mem0Wrapper._setResponseAttributes(span, response);
+        (0, helpers_1.applyCustomSpanAttributes)(span);
+    }
+    static _setOperationAttributes(span, spanName, args, config) {
+        switch (spanName) {
+            case 'memory add': {
+                const messages = args[0];
+                if (config && Object.prototype.hasOwnProperty.call(config, 'infer')) {
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_INFER, config.infer);
+                }
+                if (Mem0Wrapper._hasMessages(messages)) {
+                    const count = Array.isArray(messages) ? messages.length : 1;
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_COUNT, count);
+                }
+                break;
+            }
+            case 'memory search': {
+                const query = args[0];
+                if (query !== undefined) {
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_SEARCH_QUERY, query);
+                }
+                // v3 SDK uses `topK`; older / hosted shapes use `limit`.
+                const limit = config?.topK ?? config?.limit;
+                if (limit !== undefined) {
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_SEARCH_LIMIT, limit);
+                }
+                if (config?.threshold !== undefined) {
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_SEARCH_THRESHOLD, config.threshold);
+                }
+                break;
+            }
+            case 'memory update': {
+                const memoryId = args[0];
+                if (memoryId !== undefined)
+                    span.setAttribute(semantic_convention_1.default.DB_UPDATE_ID, memoryId);
+                break;
+            }
+            case 'memory delete': {
+                const memoryId = args[0];
+                if (memoryId !== undefined)
+                    span.setAttribute(semantic_convention_1.default.DB_DELETE_ID, memoryId);
+                break;
+            }
+            case 'memory get': {
+                const memoryId = args[0];
+                if (memoryId !== undefined)
+                    span.setAttribute(semantic_convention_1.default.DB_OPERATION_ID, memoryId);
+                break;
+            }
+            default:
+                // get_all, delete_all, history: no extra operation-specific id attribute.
+                break;
+        }
+    }
+    static _setResponseAttributes(span, response) {
+        if (!response)
+            return;
+        let count;
+        if (Array.isArray(response)) {
+            count = response.length;
+        }
+        else if (typeof response === 'object') {
+            // A results array reflects the true count — an empty list is 0; a single-object
+            // response (e.g. get / update) counts as 1. NOTE: the Python reference reports 1
+            // for an empty results list (its `if results` is falsy on []); we intentionally
+            // report the accurate 0 here (per a Sourcery review; Python sync tracked separately).
+            count = Array.isArray(response.results) ? response.results.length : 1;
+        }
+        else {
+            count = 1;
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_MEMORY_OPERATION_RESULT_COUNT, count);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_DATA_SOURCES, count);
+    }
+    /** Captures input (add) / output (search) content when capture is enabled. */
+    static _setContentAttributes(span, spanName, args, response) {
+        if (!config_1.default.captureMessageContent)
+            return;
+        try {
+            if (spanName === 'memory add') {
+                const messages = args[0];
+                if (Mem0Wrapper._hasMessages(messages)) {
+                    const content = typeof messages === 'object' ? Mem0Wrapper._safeStringify(messages) : String(messages);
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, content);
+                }
+            }
+            if (spanName === 'memory search' && response) {
+                const content = typeof response === 'object' ? Mem0Wrapper._safeStringify(response) : String(response);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, content);
+            }
+        }
+        catch {
+            /* best-effort content capture */
+        }
+    }
+    /**
+     * Returns the options/config object for an operation, accounting for the fact that
+     * its argument position differs by method: add/search take it as the 2nd arg, while
+     * getAll/deleteAll take it as the 1st. get/update/history take a positional id and
+     * carry no scope config.
+     */
+    static _extractConfig(spanName, args) {
+        switch (spanName) {
+            case 'memory add':
+            case 'memory search':
+            case 'memory delete':
+                return args[1] && typeof args[1] === 'object' ? args[1] : {};
+            case 'memory get_all':
+            case 'memory delete_all':
+                return args[0] && typeof args[0] === 'object' ? args[0] : {};
+            default:
+                return {};
+        }
+    }
+    static _scopeValue(config, camelKey, snakeKey) {
+        if (!config || typeof config !== 'object')
+            return undefined;
+        if (config[camelKey] !== undefined)
+            return config[camelKey];
+        if (config[snakeKey] !== undefined)
+            return config[snakeKey];
+        const filters = config.filters;
+        if (filters && typeof filters === 'object') {
+            if (filters[camelKey] !== undefined)
+                return filters[camelKey];
+            if (filters[snakeKey] !== undefined)
+                return filters[snakeKey];
+        }
+        return undefined;
+    }
+    static _hasMessages(messages) {
+        if (messages === undefined || messages === null)
+            return false;
+        if (Array.isArray(messages))
+            return messages.length > 0;
+        if (typeof messages === 'string')
+            return messages.length > 0;
+        return true;
+    }
+    static _safeStringify(value) {
+        if (typeof value === 'string')
+            return value;
+        try {
+            return JSON.stringify(value);
+        }
+        catch {
+            return String(value);
+        }
+    }
+}
+Mem0Wrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_MEM0;
+exports.default = Mem0Wrapper;
+//# sourceMappingURL=wrapper.js.map

package/dist/instrumentation/mem0/wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/mem0/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAsF;AACtF,8EAAiG;AACjG,0DAA2C;AAC3C,6CAA0C;AAC1C,yDAA2E;AAC3E,oFAA2D;AAE3D,qFAAqF;AACrF,yFAAyF;AACzF,uFAAuF;AACvF,uFAAuF;AACvF,wFAAwF;AACxF,wFAAwF;AACxF,IAAI,cAAc,GAAG,SAAS,CAAC;AAC/B,IAAI,CAAC;IACH,8DAA8D;IAC9D,cAAc,GAAG,OAAO,CAAC,qBAAqB,CAAC,EAAE,OAAO,IAAI,SAAS,CAAC;AACxE,CAAC;AAAC,MAAM,CAAC;IACP,cAAc,GAAG,SAAS,CAAC;AAC7B,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW;IAGf;;;;OAIG;IACH,MAAM,CAAC,qBAAqB,CAAC,MAAc,EAAE,QAAgB,EAAE,OAAgB;QAC7E,MAAM,UAAU,GAAG,OAAO,IAAI,cAAc,CAAC;QAC7C,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE7B,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACpE,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,WAAW,CAAC,kBAAkB,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;wBAC3E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;wBACjF,WAAW,CAAC,qBAAqB,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;wBAElE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC5C,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBACjD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;wBACjF,IAAI,CAAC;4BACH,WAAW,CAAC,kBAAkB,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;wBAC9E,CAAC;wBAAC,MAAM,CAAC;4BACP,8CAA8C;wBAChD,CAAC;wBACD,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACzC,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;oBACb,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,uFAAuF;IACvF,MAAM,CAAC,kBAAkB,CACvB,IAAS,EACT,QAAgB,EAChB,IAAW,EACX,QAAc,EACd,aAAqB,cAAc;QAEnC,MAAM,eAAe,GAAG,gBAAe,CAAC,eAAe,IAAI,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QAEtD,8CAA8C;QAC9C,IAAI,CAAC,YAAY,CAAC,8CAAuB,EAAE,mBAAQ,CAAC,CAAC;QACrD,kFAAkF;QAClF,mFAAmF;QACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC;QACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC;QACtF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;QAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,4BAA4B,CAAC,CAAC;QACxG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,WAAW,CAAC,CAAC;QAC/E,IAAI,CAAC,YAAY,CAAC,wCAAiB,EAAE,eAAe,CAAC,CAAC;QACtD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,WAAW,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE1D,mFAAmF;QACnF,uFAAuF;QACvF,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QACpE,MAAM,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,KAAK,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QACjE,IAAI,MAAM,KAAK,SAAS;YAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QACvF,IAAI,OAAO,KAAK,SAAS;YAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QAC1F,IAAI,KAAK,KAAK,SAAS;YAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;QAEpF,0BAA0B;QAC1B,MAAM,UAAU,GAAG,MAAM,EAAE,UAAU,IAAI,MAAM,EAAE,WAAW,CAAC;QAC7D,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,MAAM,EAAE,QAAQ,KAAK,SAAS,IAAI,MAAM,EAAE,QAAQ,KAAK,IAAI,EAAE,CAAC;YAChE,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,WAAW,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAC5C,CAAC;QACJ,CAAC;QAED,WAAW,CAAC,uBAAuB,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QAClE,WAAW,CAAC,sBAAsB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAEnD,IAAA,mCAAyB,EAAC,IAAI,CAAC,CAAC;IAClC,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,IAAS,EAAE,QAAgB,EAAE,IAAW,EAAE,MAAW;QAC1F,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACzB,IAAI,MAAM,IAAI,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;oBACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC1E,CAAC;gBACD,IAAI,WAAW,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACvC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC5D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;gBACnE,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACtB,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;gBAC1E,CAAC;gBACD,yDAAyD;gBACzD,MAAM,KAAK,GAAG,MAAM,EAAE,IAAI,IAAI,MAAM,EAAE,KAAK,CAAC;gBAC5C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;gBAC1E,CAAC;gBACD,IAAI,MAAM,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;oBACpC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,8BAA8B,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;gBACzF,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACzB,IAAI,QAAQ,KAAK,SAAS;oBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;gBACzF,MAAM;YACR,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACzB,IAAI,QAAQ,KAAK,SAAS;oBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;gBACzF,MAAM;YACR,CAAC;YACD,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACzB,IAAI,QAAQ,KAAK,SAAS;oBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;gBAC5F,MAAM;YACR,CAAC;YACD;gBACE,0EAA0E;gBAC1E,MAAM;QACV,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,sBAAsB,CAAC,IAAS,EAAE,QAAa;QAC5D,IAAI,CAAC,QAAQ;YAAE,OAAO;QAEtB,IAAI,KAAa,CAAC;QAClB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC;QAC1B,CAAC;aAAM,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACxC,gFAAgF;YAChF,iFAAiF;YACjF,gFAAgF;YAChF,sFAAsF;YACtF,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACxE,CAAC;aAAM,CAAC;YACN,KAAK,GAAG,CAAC,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;QAClF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;IACnE,CAAC;IAED,8EAA8E;IAC9E,MAAM,CAAC,qBAAqB,CAAC,IAAS,EAAE,QAAgB,EAAE,IAAW,EAAE,QAAa;QAClF,IAAI,CAAC,gBAAe,CAAC,qBAAqB;YAAE,OAAO;QAEnD,IAAI,CAAC;YACH,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;gBAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACzB,IAAI,WAAW,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACvC,MAAM,OAAO,GACX,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;oBACzF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;gBACvE,CAAC;YACH,CAAC;YACD,IAAI,QAAQ,KAAK,eAAe,IAAI,QAAQ,EAAE,CAAC;gBAC7C,MAAM,OAAO,GACX,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACzF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;YACxE,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,iCAAiC;QACnC,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,MAAM,CAAC,cAAc,CAAC,QAAgB,EAAE,IAAW;QACzD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,YAAY,CAAC;YAClB,KAAK,eAAe,CAAC;YACrB,KAAK,eAAe;gBAClB,OAAO,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/D,KAAK,gBAAgB,CAAC;YACtB,KAAK,mBAAmB;gBACtB,OAAO,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/D;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,WAAW,CAAC,MAAW,EAAE,QAAgB,EAAE,QAAgB;QACxE,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,SAAS,CAAC;QAC5D,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,SAAS;YAAE,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5D,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,SAAS;YAAE,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC/B,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC3C,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,SAAS;gBAAE,OAAO,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC9D,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,SAAS;gBAAE,OAAO,OAAO,CAAC,QAAQ,CAAC,CAAC;QAChE,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,QAAa;QACvC,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI;YAAE,OAAO,KAAK,CAAC;QAC9D,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACxD,IAAI,OAAO,QAAQ,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QAC7D,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,cAAc,CAAC,KAAU;QACtC,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC5C,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;;AAzOM,oBAAQ,GAAG,6BAAkB,CAAC,kBAAkB,CAAC;AA4O1D,kBAAe,WAAW,CAAC"}

package/dist/instrumentation/milvus/index.d.ts

@@ -0,0 +1,11 @@
+import { InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+import { InstrumentationConfig } from '@opentelemetry/instrumentation';
+export interface MilvusInstrumentationConfig extends InstrumentationConfig {
+}
+export default class Shield360MilvusInstrumentation extends InstrumentationBase {
+    constructor(config?: MilvusInstrumentationConfig);
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    manualPatch(milvus: any): void;
+    protected _patch(moduleExports: any): void;
+    protected _unpatch(moduleExports: any): void;
+}

package/dist/instrumentation/milvus/index.js

@@ -0,0 +1,67 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+class Shield360MilvusInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-milvus`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@zilliz/milvus2-sdk-node', ['>=2.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(milvus) {
+        this._patch(milvus);
+    }
+    _patch(moduleExports) {
+        try {
+            const MilvusClient = moduleExports.MilvusClient;
+            if (!MilvusClient?.prototype)
+                return;
+            const patchMap = [
+                ['search', wrapper_1.default._patchSearch.bind(wrapper_1.default)],
+                ['insert', wrapper_1.default._patchInsert.bind(wrapper_1.default)],
+                ['delete', wrapper_1.default._patchDelete.bind(wrapper_1.default)],
+                ['query', wrapper_1.default._patchQuery.bind(wrapper_1.default)],
+                ['upsert', wrapper_1.default._patchUpsert.bind(wrapper_1.default)],
+            ];
+            for (const [method, patchFn] of patchMap) {
+                if (typeof MilvusClient.prototype[method] === 'function') {
+                    if ((0, instrumentation_1.isWrapped)(MilvusClient.prototype[method])) {
+                        this._unwrap(MilvusClient.prototype, method);
+                    }
+                    this._wrap(MilvusClient.prototype, method, patchFn(this.tracer));
+                }
+            }
+        }
+        catch (e) {
+            console.error('Error in Milvus _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const MilvusClient = moduleExports.MilvusClient;
+            if (!MilvusClient?.prototype)
+                return;
+            for (const method of ['search', 'insert', 'delete', 'query', 'upsert']) {
+                if (typeof MilvusClient.prototype[method] === 'function') {
+                    this._unwrap(MilvusClient.prototype, method);
+                }
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = Shield360MilvusInstrumentation;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/milvus/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/milvus/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAsC;AAItC,MAAqB,8BAA+B,SAAQ,qCAAmB;IAC7E,YAAY,SAAsC,EAAE;QAClD,KAAK,CAAC,GAAG,iCAAsB,yBAAyB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC7E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,0BAA0B,EAC1B,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,MAAW;QAC5B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACtB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAAC,YAAY,CAAC;YAChD,IAAI,CAAC,YAAY,EAAE,SAAS;gBAAE,OAAO;YAErC,MAAM,QAAQ,GAA0C;gBACtD,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBAC1D,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBAC1D,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBAC1D,CAAC,OAAO,EAAE,iBAAa,CAAC,WAAW,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBACxD,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;aAC3D,CAAC;YAEF,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzC,IAAI,OAAO,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;oBACzD,IAAI,IAAA,2BAAS,EAAC,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;wBAC9C,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;oBAC/C,CAAC;oBACD,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAAC,YAAY,CAAC;YAChD,IAAI,CAAC,YAAY,EAAE,SAAS;gBAAE,OAAO;YACrC,KAAK,MAAM,MAAM,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,CAAC;gBACvE,IAAI,OAAO,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;oBACzD,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AA/DD,iDA+DC"}

package/dist/instrumentation/milvus/wrapper.d.ts

@@ -0,0 +1,15 @@
+import { Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class MilvusWrapper extends BaseWrapper {
+    static dbSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    static _setCommonAttributes(span: any, dbOperation: string, collectionName: string): void;
+    static _getCollectionName(params: any): string;
+    static _patchSearch(tracer: Tracer): any;
+    static _patchInsert(tracer: Tracer): any;
+    static _patchUpsert(tracer: Tracer): any;
+    static _patchDelete(tracer: Tracer): any;
+    static _patchQuery(tracer: Tracer): any;
+}
+export default MilvusWrapper;