shield360 1.0.0

package/dist/instrumentation/letta/wrapper.js

@@ -0,0 +1,180 @@
+"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 helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const utils_1 = require("./utils");
+let lettaSdkVersion = 'unknown';
+try {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    lettaSdkVersion = require('@letta-ai/letta-client/version').VERSION || 'unknown';
+}
+catch {
+    lettaSdkVersion = 'unknown';
+}
+/**
+ * Splits Letta method arguments into the request body and, for agent-scoped
+ * resources, the leading agent id. A leading string on a top-level resource is
+ * that resource's own id, so it is never treated as an agent id.
+ */
+function resolveArgs(args, agentScoped) {
+    const agentId = agentScoped && typeof args[0] === 'string' ? args[0] : undefined;
+    const body = args.find((arg) => arg && typeof arg === 'object' && !Array.isArray(arg));
+    return { body, agentId };
+}
+function isAsyncIterable(value) {
+    return value != null && typeof value[Symbol.asyncIterator] === 'function';
+}
+/**
+ * Instrumentation for the Letta agent platform (`@letta-ai/letta-client`). Each
+ * patched resource method becomes one CLIENT span whose name and operation type
+ * mirror the Python reference (sdk/python/src/shield360/instrumentation/letta): the
+ * method suffix drives the operation type via OPERATION_TYPE_MAP, and chat
+ * (message) operations stream their response through a traced async iterator.
+ */
+class LettaWrapper {
+    /**
+     * Wraps a Letta resource method. `endpoint` is the Python-style endpoint id
+     * (e.g. `letta.create`, `letta.create_message`) that drives the operation type
+     * and span name.
+     */
+    static _patchOperation(tracer, endpoint, agentScoped, version) {
+        const sdkVersion = version || lettaSdkVersion;
+        const operationType = (0, utils_1.getOperationType)(endpoint);
+        return (originalMethod) => {
+            return function (...args) {
+                if ((0, core_1.isTracingSuppressed)(api_1.context.active())) {
+                    return originalMethod.apply(this, args);
+                }
+                const { body, agentId } = resolveArgs(args, agentScoped);
+                const spanBody = agentId && (!body || body.agent_id == null) ? { ...(body || {}), agent_id: agentId } : body;
+                const spanName = (0, utils_1.getSpanName)(operationType, endpoint, this, spanBody);
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: {
+                        [semantic_convention_1.default.GEN_AI_OPERATION]: operationType,
+                        [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: LettaWrapper.aiSystem,
+                    },
+                });
+                const startTime = Date.now();
+                if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT) {
+                    return LettaWrapper._handleChat(originalMethod, this, args, span, startTime, body, operationType, sdkVersion, agentId);
+                }
+                return LettaWrapper._handleOperation(originalMethod, this, args, span, startTime, body, operationType, sdkVersion, agentId);
+            };
+        };
+    }
+    /**
+     * Non-chat operations return the SDK's own promise untouched. `list` yields a
+     * PagePromise whose async iterator drives auto-pagination, so flattening it here
+     * would break paging; the span is finalized from a completion callback instead.
+     */
+    static _handleOperation(originalMethod, instance, args, span, startTime, body, operationType, sdkVersion, agentId) {
+        return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), () => {
+            let result;
+            try {
+                result = originalMethod.apply(instance, args);
+            }
+            catch (e) {
+                LettaWrapper._finalizeError(span, e, body, operationType, instance, startTime, sdkVersion, agentId);
+                throw e;
+            }
+            Promise.resolve(result).then((response) => {
+                (0, utils_1.processLettaResponse)(span, response, body, operationType, instance, (Date.now() - startTime) / 1000, sdkVersion, agentId);
+                span.setStatus({ code: api_1.SpanStatusCode.OK });
+                span.end();
+            }, (e) => {
+                LettaWrapper._finalizeError(span, e, body, operationType, instance, startTime, sdkVersion, agentId);
+            });
+            return result;
+        });
+    }
+    /**
+     * Chat (message) operations may resolve to a streaming iterator. The resolved
+     * value is awaited so a stream can be wrapped in a traced async iterator; a
+     * non-streaming response is processed inline.
+     */
+    static async _handleChat(originalMethod, instance, args, span, startTime, body, operationType, sdkVersion, agentId) {
+        return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+            try {
+                const resolved = await originalMethod.apply(instance, args);
+                if (isAsyncIterable(resolved)) {
+                    return LettaWrapper._traceStream(resolved, span, startTime, body, operationType, instance, sdkVersion, agentId);
+                }
+                (0, utils_1.processLettaResponse)(span, resolved, body, operationType, instance, (Date.now() - startTime) / 1000, sdkVersion, agentId);
+                span.setStatus({ code: api_1.SpanStatusCode.OK });
+                span.end();
+                return resolved;
+            }
+            catch (e) {
+                LettaWrapper._finalizeError(span, e, body, operationType, instance, startTime, sdkVersion, agentId);
+                throw e;
+            }
+        });
+    }
+    /**
+     * Wraps a Letta message stream so chunks flow through untouched while the span
+     * accumulates chunk count, TTFT, and the collected messages. Mirrors
+     * TracedLettaStream in the Python reference.
+     */
+    static _traceStream(stream, span, startTime, body, operationType, instance, sdkVersion, agentId) {
+        async function* traced() {
+            let chunkCount = 0;
+            let ttft = 0;
+            const messages = [];
+            try {
+                for await (const chunk of stream) {
+                    chunkCount += 1;
+                    if (chunkCount === 1) {
+                        ttft = (Date.now() - startTime) / 1000;
+                    }
+                    messages.push(chunk);
+                    if (span.isRecording()) {
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_STREAMING_CHUNK_COUNT, chunkCount);
+                        if (ttft > 0)
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+                    }
+                    yield chunk;
+                }
+                const synthetic = { messages };
+                for (const msg of messages) {
+                    if (msg?.message_type === 'usage_statistics') {
+                        synthetic.usage = msg;
+                        break;
+                    }
+                }
+                (0, utils_1.processLettaResponse)(span, synthetic, body, operationType, instance, (Date.now() - startTime) / 1000, sdkVersion, agentId);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_STREAMING_CHUNK_COUNT, chunkCount);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, true);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_STREAMING_RESPONSE_COUNT, messages.length);
+                span.setStatus({ code: api_1.SpanStatusCode.OK });
+            }
+            catch (e) {
+                helpers_1.default.handleException(span, e);
+                throw e;
+            }
+            finally {
+                span.end();
+            }
+        }
+        return helpers_1.default.createStreamProxy(stream, traced());
+    }
+    static _finalizeError(span, error, body, operationType, instance, startTime, sdkVersion, agentId) {
+        try {
+            (0, utils_1.processLettaResponse)(span, undefined, body, operationType, instance, (Date.now() - startTime) / 1000, sdkVersion, agentId);
+        }
+        catch {
+            /* best-effort attributes on the error path */
+        }
+        helpers_1.default.handleException(span, error);
+        span.end();
+    }
+}
+LettaWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_LETTA;
+exports.default = LettaWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/letta/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAA4F;AAC5F,8CAA0D;AAC1D,4DAA4C;AAC5C,oFAA2D;AAC3D,mCAA8E;AAE9E,IAAI,eAAe,GAAG,SAAS,CAAC;AAChC,IAAI,CAAC;IACH,8DAA8D;IAC9D,eAAe,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC,OAAO,IAAI,SAAS,CAAC;AACnF,CAAC;AAAC,MAAM,CAAC;IACP,eAAe,GAAG,SAAS,CAAC;AAC9B,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,IAAW,EAAE,WAAoB;IACpD,MAAM,OAAO,GAAG,WAAW,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACjF,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IACvF,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,eAAe,CAAC,KAAU;IACjC,OAAO,KAAK,IAAI,IAAI,IAAI,OAAO,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,CAAC;AAC5E,CAAC;AAED;;;;;;GAMG;AACH,MAAM,YAAY;IAGhB;;;;OAIG;IACH,MAAM,CAAC,eAAe,CACpB,MAAc,EACd,QAAgB,EAChB,WAAoB,EACpB,OAAgB;QAEhB,MAAM,UAAU,GAAG,OAAO,IAAI,eAAe,CAAC;QAC9C,MAAM,aAAa,GAAG,IAAA,wBAAgB,EAAC,QAAQ,CAAC,CAAC;QAEjD,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,UAAqB,GAAG,IAAW;gBACxC,IAAI,IAAA,0BAAmB,EAAC,aAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;oBAC1C,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC1C,CAAC;gBAED,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBACzD,MAAM,QAAQ,GACZ,OAAO,IAAI,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC9F,MAAM,QAAQ,GAAG,IAAA,mBAAW,EAAC,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAEtE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE;wBACV,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;wBACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,YAAY,CAAC,QAAQ;qBACtE;iBACF,CAAC,CAAC;gBACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE7B,IAAI,aAAa,KAAK,6BAAkB,CAAC,0BAA0B,EAAE,CAAC;oBACpE,OAAO,YAAY,CAAC,WAAW,CAC7B,cAAc,EACd,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,SAAS,EACT,IAAI,EACJ,aAAa,EACb,UAAU,EACV,OAAO,CACR,CAAC;gBACJ,CAAC;gBAED,OAAO,YAAY,CAAC,gBAAgB,CAClC,cAAc,EACd,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,SAAS,EACT,IAAI,EACJ,aAAa,EACb,UAAU,EACV,OAAO,CACR,CAAC;YACJ,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACK,MAAM,CAAC,gBAAgB,CAC7B,cAAuC,EACvC,QAAa,EACb,IAAW,EACX,IAAU,EACV,SAAiB,EACjB,IAAS,EACT,aAAqB,EACrB,UAAkB,EAClB,OAAgB;QAEhB,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE;YAC9D,IAAI,MAAW,CAAC;YAChB,IAAI,CAAC;gBACH,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAChD,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,YAAY,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;gBACpG,MAAM,CAAC,CAAC;YACV,CAAC;YAED,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAC1B,CAAC,QAAQ,EAAE,EAAE;gBACX,IAAA,4BAAoB,EAClB,IAAI,EACJ,QAAQ,EACR,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,EAC/B,UAAU,EACV,OAAO,CACR,CAAC;gBACF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC,EACD,CAAC,CAAC,EAAE,EAAE;gBACJ,YAAY,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YACtG,CAAC,CACF,CAAC;YAEF,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,MAAM,CAAC,KAAK,CAAC,WAAW,CAC9B,cAAuC,EACvC,QAAa,EACb,IAAW,EACX,IAAU,EACV,SAAiB,EACjB,IAAS,EACT,aAAqB,EACrB,UAAkB,EAClB,OAAgB;QAEhB,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;YACpE,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAE5D,IAAI,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC9B,OAAO,YAAY,CAAC,YAAY,CAC9B,QAAQ,EACR,IAAI,EACJ,SAAS,EACT,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,UAAU,EACV,OAAO,CACR,CAAC;gBACJ,CAAC;gBAED,IAAA,4BAAoB,EAClB,IAAI,EACJ,QAAQ,EACR,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,EAC/B,UAAU,EACV,OAAO,CACR,CAAC;gBACF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,YAAY,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;gBACpG,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,MAAM,CAAC,YAAY,CACzB,MAAW,EACX,IAAU,EACV,SAAiB,EACjB,IAAS,EACT,aAAqB,EACrB,QAAa,EACb,UAAkB,EAClB,OAAgB;QAEhB,KAAK,SAAS,CAAC,CAAC,MAAM;YACpB,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,MAAM,QAAQ,GAAU,EAAE,CAAC;YAE3B,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBACjC,UAAU,IAAI,CAAC,CAAC;oBAChB,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;wBACrB,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;oBACzC,CAAC;oBACD,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACrB,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;wBACvB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;wBAC/E,IAAI,IAAI,GAAG,CAAC;4BAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;oBAC/E,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,MAAM,SAAS,GAAQ,EAAE,QAAQ,EAAE,CAAC;gBACpC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;oBAC3B,IAAI,GAAG,EAAE,YAAY,KAAK,kBAAkB,EAAE,CAAC;wBAC7C,SAAS,CAAC,KAAK,GAAG,GAAG,CAAC;wBACtB,MAAM;oBACR,CAAC;gBACH,CAAC;gBAED,IAAA,4BAAoB,EAClB,IAAI,EACJ,SAAS,EACT,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,EAC/B,UAAU,EACV,OAAO,CACR,CAAC;gBACF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;gBAC/E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;gBAC/D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,IAAI,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACvF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9C,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBACzC,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC;QAED,OAAO,iBAAe,CAAC,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAC7D,CAAC;IAEO,MAAM,CAAC,cAAc,CAC3B,IAAU,EACV,KAAU,EACV,IAAS,EACT,aAAqB,EACrB,QAAa,EACb,SAAiB,EACjB,UAAkB,EAClB,OAAgB;QAEhB,IAAI,CAAC;YACH,IAAA,4BAAoB,EAClB,IAAI,EACJ,SAAS,EACT,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,EAC/B,UAAU,EACV,OAAO,CACR,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,8CAA8C;QAChD,CAAC;QACD,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC;;AArQM,qBAAQ,GAAG,6BAAkB,CAAC,mBAAmB,CAAC;AAwQ3D,kBAAe,YAAY,CAAC"}

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

@@ -0,0 +1,36 @@
+/**
+ * Shield360 LlamaIndex Instrumentation
+ *
+ * Monkey-patches LlamaIndex JS classes to emit OTel-compliant telemetry.
+ * Mirrors the Python SDK: sdk/python/src/shield360/instrumentation/llamaindex/__init__.py
+ *
+ * Targets the `llamaindex` npm package (>=0.3.0).
+ * Patches: LLM classes, query engines, chat engines, retrievers, embeddings,
+ * index construction, document loaders, splitters, and synthesizers.
+ */
+import { InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+import type { InstrumentationConfig } from '@opentelemetry/instrumentation';
+export default class Shield360LlamaIndexInstrumentation extends InstrumentationBase {
+    constructor(config?: InstrumentationConfig);
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    manualPatch(llamaindex: any): void;
+    /**
+     * Wrap an instance method on the first class found that exposes it.
+     * Silently skips classes that aren't exported or don't have the method.
+     */
+    private _patchProto;
+    /**
+     * Wrap a static method on a class.
+     */
+    private _patchStatic;
+    /**
+     * Unwrap an instance method if it's wrapped.
+     */
+    private _unwrapProto;
+    /**
+     * Unwrap a static method if it's wrapped.
+     */
+    private _unwrapStatic;
+    protected _patch(m: any): void;
+    protected _unpatch(m: any): void;
+}

package/dist/instrumentation/llamaindex/index.js

@@ -0,0 +1,221 @@
+"use strict";
+/**
+ * Shield360 LlamaIndex Instrumentation
+ *
+ * Monkey-patches LlamaIndex JS classes to emit OTel-compliant telemetry.
+ * Mirrors the Python SDK: sdk/python/src/shield360/instrumentation/llamaindex/__init__.py
+ *
+ * Targets the `llamaindex` npm package (>=0.3.0).
+ * Patches: LLM classes, query engines, chat engines, retrievers, embeddings,
+ * index construction, document loaders, splitters, and synthesizers.
+ */
+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 Shield360LlamaIndexInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-llamaindex`, '1.0.0', config);
+    }
+    init() {
+        const mainModule = new instrumentation_1.InstrumentationNodeModuleDefinition('llamaindex', ['>=0.3.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports)
+                this._unpatch(moduleExports);
+        });
+        return [mainModule];
+    }
+    manualPatch(llamaindex) {
+        this._patch(llamaindex);
+    }
+    // ---------------------------------------------------------------------------
+    // Patch helpers
+    // ---------------------------------------------------------------------------
+    /**
+     * Wrap an instance method on the first class found that exposes it.
+     * Silently skips classes that aren't exported or don't have the method.
+     */
+    _patchProto(moduleExports, classNames, method, wrapper) {
+        for (const name of classNames) {
+            const Cls = moduleExports[name];
+            if (Cls?.prototype?.[method] && !(0, instrumentation_1.isWrapped)(Cls.prototype[method])) {
+                try {
+                    this._wrap(Cls.prototype, method, wrapper);
+                }
+                catch { /* skip silently */ }
+            }
+        }
+    }
+    /**
+     * Wrap a static method on a class.
+     */
+    _patchStatic(Cls, method, wrapper) {
+        if (typeof Cls?.[method] === 'function' && !(0, instrumentation_1.isWrapped)(Cls[method])) {
+            try {
+                this._wrap(Cls, method, wrapper);
+            }
+            catch { /* skip silently */ }
+        }
+    }
+    /**
+     * Unwrap an instance method if it's wrapped.
+     */
+    _unwrapProto(moduleExports, classNames, method) {
+        for (const name of classNames) {
+            const Cls = moduleExports[name];
+            if (Cls?.prototype?.[method] && (0, instrumentation_1.isWrapped)(Cls.prototype[method])) {
+                try {
+                    this._unwrap(Cls.prototype, method);
+                }
+                catch { /* ignore */ }
+            }
+        }
+    }
+    /**
+     * Unwrap a static method if it's wrapped.
+     */
+    _unwrapStatic(Cls, method) {
+        if (Cls && typeof Cls[method] === 'function' && (0, instrumentation_1.isWrapped)(Cls[method])) {
+            try {
+                this._unwrap(Cls, method);
+            }
+            catch { /* ignore */ }
+        }
+    }
+    // ---------------------------------------------------------------------------
+    // Main patch — mirrors Python _instrument() operation lists
+    // ---------------------------------------------------------------------------
+    _patch(m) {
+        try {
+            const tracer = this.tracer;
+            // === LLM OPERATIONS (chat spans) ===
+            // Mirrors Python: LLM.chat, LLM.complete
+            const llmClasses = [
+                'OpenAI', 'Anthropic', 'Ollama', 'Gemini', 'Groq',
+                'HuggingFaceInference', 'Replicate', 'DeepSeek', 'Portkey',
+                'BaseLLM', 'LLM',
+            ];
+            this._patchProto(m, llmClasses, 'chat', wrapper_1.default._patchLLMChat(tracer));
+            this._patchProto(m, llmClasses, 'complete', wrapper_1.default._patchLLMComplete(tracer));
+            // === QUERY ENGINE OPERATIONS (retrieval spans) ===
+            // Mirrors Python: RetrieverQueryEngine.query, TransformQueryEngine.query
+            const queryEngineClasses = [
+                'RetrieverQueryEngine', 'TransformQueryEngine',
+                'SubQuestionQueryEngine', 'BaseQueryEngine', 'QueryEngine',
+            ];
+            this._patchProto(m, queryEngineClasses, 'query', wrapper_1.default._patchQueryEngineQuery(tracer));
+            // === CHAT ENGINE OPERATIONS (invoke_workflow spans) ===
+            const chatEngineClasses = [
+                'ContextChatEngine', 'SimpleChatEngine',
+                'CondenseQuestionChatEngine', 'BaseChatEngine', 'ChatEngine',
+            ];
+            this._patchProto(m, chatEngineClasses, 'chat', wrapper_1.default._patchChatEngineChat(tracer));
+            // === RETRIEVER OPERATIONS (retrieval spans) ===
+            // Mirrors Python: VectorIndexRetriever.retrieve, BaseRetriever.retrieve
+            const retrieverClasses = [
+                'VectorIndexRetriever', 'BaseRetriever', 'Retriever',
+            ];
+            this._patchProto(m, retrieverClasses, 'retrieve', wrapper_1.default._patchRetrieverRetrieve(tracer));
+            // === EMBEDDING OPERATIONS (embeddings spans) ===
+            // Mirrors Python: BaseEmbedding.get_text_embedding_batch
+            const embeddingClasses = [
+                'OpenAIEmbedding', 'BaseEmbedding', 'HuggingFaceEmbedding',
+            ];
+            this._patchProto(m, embeddingClasses, 'getTextEmbedding', wrapper_1.default._patchEmbedding(tracer, 'embedding_generate'));
+            this._patchProto(m, embeddingClasses, 'getQueryEmbedding', wrapper_1.default._patchEmbedding(tracer, 'embedding_generate'));
+            this._patchProto(m, embeddingClasses, 'getTextEmbeddingsBatch', wrapper_1.default._patchEmbedding(tracer, 'embedding_generate'));
+            // === INDEX OPERATIONS (invoke_workflow spans) ===
+            // Mirrors Python: VectorStoreIndex.from_documents, from_vector_store
+            const indexClasses = ['VectorStoreIndex', 'ListIndex', 'TreeIndex'];
+            for (const name of indexClasses) {
+                const Cls = m[name];
+                if (Cls) {
+                    this._patchStatic(Cls, 'fromDocuments', wrapper_1.default._patchFrameworkMethod(tracer, 'index_construct'));
+                    this._patchStatic(Cls, 'fromVectorStore', wrapper_1.default._patchFrameworkMethod(tracer, 'index_construct'));
+                }
+            }
+            // Mirrors Python: BaseIndex.as_query_engine, BaseIndex.as_retriever
+            const indexProtoClasses = [
+                'VectorStoreIndex', 'ListIndex', 'TreeIndex', 'BaseIndex',
+            ];
+            this._patchProto(m, indexProtoClasses, 'asQueryEngine', wrapper_1.default._patchFrameworkMethod(tracer, 'query_engine_create'));
+            this._patchProto(m, indexProtoClasses, 'asRetriever', wrapper_1.default._patchFrameworkMethod(tracer, 'retriever_create'));
+            // === DOCUMENT OPERATIONS (framework / retrieval spans) ===
+            // Mirrors Python: SimpleDirectoryReader.load_data
+            this._patchProto(m, ['SimpleDirectoryReader'], 'loadData', wrapper_1.default._patchFrameworkMethod(tracer, 'document_load'));
+            // Mirrors Python: SentenceSplitter.get_nodes_from_documents
+            this._patchProto(m, ['SentenceSplitter', 'NodeParser'], 'getNodesFromDocuments', wrapper_1.default._patchFrameworkMethod(tracer, 'document_split'));
+            // Mirrors Python: SentenceSplitter.split_text
+            this._patchProto(m, ['SentenceSplitter'], 'splitText', wrapper_1.default._patchFrameworkMethod(tracer, 'text_splitter_split'));
+            // === SYNTHESIZER OPERATIONS (chat spans) ===
+            // Mirrors Python: BaseSynthesizer.synthesize
+            const synthesizerClasses = [
+                'ResponseSynthesizer', 'CompactAndRefine', 'TreeSummarize',
+                'BaseSynthesizer',
+            ];
+            this._patchProto(m, synthesizerClasses, 'synthesize', wrapper_1.default._patchFrameworkMethod(tracer, 'response_synthesize'));
+            // === POSTPROCESSOR OPERATIONS (framework spans) ===
+            this._patchProto(m, ['BaseNodePostprocessor', 'NodePostprocessor'], 'postprocessNodes', wrapper_1.default._patchFrameworkMethod(tracer, 'postprocessor_process'));
+        }
+        catch { /* graceful degradation — do not break the application */ }
+    }
+    // ---------------------------------------------------------------------------
+    // Unpatch
+    // ---------------------------------------------------------------------------
+    _unpatch(m) {
+        try {
+            const llmClasses = [
+                'OpenAI', 'Anthropic', 'Ollama', 'Gemini', 'Groq',
+                'HuggingFaceInference', 'Replicate', 'DeepSeek', 'Portkey',
+                'BaseLLM', 'LLM',
+            ];
+            this._unwrapProto(m, llmClasses, 'chat');
+            this._unwrapProto(m, llmClasses, 'complete');
+            const queryEngineClasses = [
+                'RetrieverQueryEngine', 'TransformQueryEngine',
+                'SubQuestionQueryEngine', 'BaseQueryEngine', 'QueryEngine',
+            ];
+            this._unwrapProto(m, queryEngineClasses, 'query');
+            const chatEngineClasses = [
+                'ContextChatEngine', 'SimpleChatEngine',
+                'CondenseQuestionChatEngine', 'BaseChatEngine', 'ChatEngine',
+            ];
+            this._unwrapProto(m, chatEngineClasses, 'chat');
+            const retrieverClasses = ['VectorIndexRetriever', 'BaseRetriever', 'Retriever'];
+            this._unwrapProto(m, retrieverClasses, 'retrieve');
+            const embeddingClasses = ['OpenAIEmbedding', 'BaseEmbedding', 'HuggingFaceEmbedding'];
+            this._unwrapProto(m, embeddingClasses, 'getTextEmbedding');
+            this._unwrapProto(m, embeddingClasses, 'getQueryEmbedding');
+            this._unwrapProto(m, embeddingClasses, 'getTextEmbeddingsBatch');
+            const indexClasses = ['VectorStoreIndex', 'ListIndex', 'TreeIndex'];
+            for (const name of indexClasses) {
+                const Cls = m[name];
+                if (Cls) {
+                    this._unwrapStatic(Cls, 'fromDocuments');
+                    this._unwrapStatic(Cls, 'fromVectorStore');
+                }
+            }
+            const indexProtoClasses = [
+                'VectorStoreIndex', 'ListIndex', 'TreeIndex', 'BaseIndex',
+            ];
+            this._unwrapProto(m, indexProtoClasses, 'asQueryEngine');
+            this._unwrapProto(m, indexProtoClasses, 'asRetriever');
+            this._unwrapProto(m, ['SimpleDirectoryReader'], 'loadData');
+            this._unwrapProto(m, ['SentenceSplitter', 'NodeParser'], 'getNodesFromDocuments');
+            this._unwrapProto(m, ['SentenceSplitter'], 'splitText');
+            const synthesizerClasses = [
+                'ResponseSynthesizer', 'CompactAndRefine', 'TreeSummarize', 'BaseSynthesizer',
+            ];
+            this._unwrapProto(m, synthesizerClasses, 'synthesize');
+            this._unwrapProto(m, ['BaseNodePostprocessor', 'NodePostprocessor'], 'postprocessNodes');
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = Shield360LlamaIndexInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/llamaindex/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;AAEH,oEAKwC;AAExC,6CAAwD;AACxD,wDAA0C;AAE1C,MAAqB,kCAAmC,SAAQ,qCAAmB;IACjF,YAAY,SAAgC,EAAE;QAC5C,KAAK,CAAC,GAAG,iCAAsB,6BAA6B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACjF,CAAC;IAES,IAAI;QAIZ,MAAM,UAAU,GAAG,IAAI,qDAAmC,CACxD,YAAY,EACZ,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;gBAAE,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAClD,CAAC,CACF,CAAC;QAEF,OAAO,CAAC,UAAU,CAAC,CAAC;IACtB,CAAC;IAEM,WAAW,CAAC,UAAe;QAChC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC1B,CAAC;IAED,8EAA8E;IAC9E,gBAAgB;IAChB,8EAA8E;IAE9E;;;OAGG;IACK,WAAW,CACjB,aAAkB,EAClB,UAAoB,EACpB,MAAc,EACd,OAAY;QAEZ,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YAChC,IAAI,GAAG,EAAE,SAAS,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAA,2BAAS,EAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;gBAClE,IAAI,CAAC;oBACH,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC7C,CAAC;gBAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,YAAY,CAClB,GAAQ,EACR,MAAc,EACd,OAAY;QAEZ,IAAI,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,UAAU,IAAI,CAAC,IAAA,2BAAS,EAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;YACnE,IAAI,CAAC;gBACH,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YACnC,CAAC;YAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,YAAY,CAClB,aAAkB,EAClB,UAAoB,EACpB,MAAc;QAEd,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YAChC,IAAI,GAAG,EAAE,SAAS,EAAE,CAAC,MAAM,CAAC,IAAI,IAAA,2BAAS,EAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;gBACjE,IAAI,CAAC;oBACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;gBACtC,CAAC;gBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,GAAQ,EAAE,MAAc;QAC5C,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,UAAU,IAAI,IAAA,2BAAS,EAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;YACvE,IAAI,CAAC;gBACH,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC5B,CAAC;YAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,4DAA4D;IAC5D,8EAA8E;IAEpE,MAAM,CAAC,CAAM;QACrB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAE3B,sCAAsC;YACtC,yCAAyC;YACzC,MAAM,UAAU,GAAG;gBACjB,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM;gBACjD,sBAAsB,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS;gBAC1D,SAAS,EAAE,KAAK;aACjB,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,EACpC,iBAAiB,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3C,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,UAAU,EAAE,UAAU,EACxC,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC;YAE/C,oDAAoD;YACpD,yEAAyE;YACzE,MAAM,kBAAkB,GAAG;gBACzB,sBAAsB,EAAE,sBAAsB;gBAC9C,wBAAwB,EAAE,iBAAiB,EAAE,aAAa;aAC3D,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,kBAAkB,EAAE,OAAO,EAC7C,iBAAiB,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC,CAAC;YAEpD,yDAAyD;YACzD,MAAM,iBAAiB,GAAG;gBACxB,mBAAmB,EAAE,kBAAkB;gBACvC,4BAA4B,EAAE,gBAAgB,EAAE,YAAY;aAC7D,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,iBAAiB,EAAE,MAAM,EAC3C,iBAAiB,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC;YAElD,iDAAiD;YACjD,wEAAwE;YACxE,MAAM,gBAAgB,GAAG;gBACvB,sBAAsB,EAAE,eAAe,EAAE,WAAW;aACrD,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAC9C,iBAAiB,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC,CAAC;YAErD,kDAAkD;YAClD,yDAAyD;YACzD,MAAM,gBAAgB,GAAG;gBACvB,iBAAiB,EAAE,eAAe,EAAE,sBAAsB;aAC3D,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,gBAAgB,EAAE,kBAAkB,EACtD,iBAAiB,CAAC,eAAe,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC,CAAC;YACnE,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,gBAAgB,EAAE,mBAAmB,EACvD,iBAAiB,CAAC,eAAe,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC,CAAC;YACnE,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,gBAAgB,EAAE,wBAAwB,EAC5D,iBAAiB,CAAC,eAAe,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC,CAAC;YAEnE,mDAAmD;YACnD,qEAAqE;YACrE,MAAM,YAAY,GAAG,CAAC,kBAAkB,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;YACpE,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;gBAChC,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBACpB,IAAI,GAAG,EAAE,CAAC;oBACR,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,eAAe,EACpC,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;oBACtE,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,iBAAiB,EACtC,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;YAED,oEAAoE;YACpE,MAAM,iBAAiB,GAAG;gBACxB,kBAAkB,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW;aAC1D,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,iBAAiB,EAAE,eAAe,EACpD,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC,CAAC;YAC1E,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,iBAAiB,EAAE,aAAa,EAClD,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC,CAAC;YAEvE,4DAA4D;YAC5D,kDAAkD;YAClD,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,uBAAuB,CAAC,EAAE,UAAU,EACvD,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC,CAAC;YAEpE,4DAA4D;YAC5D,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,kBAAkB,EAAE,YAAY,CAAC,EAAE,uBAAuB,EAC7E,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC;YAErE,8CAA8C;YAC9C,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAC,EAAE,WAAW,EACnD,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC,CAAC;YAE1E,8CAA8C;YAC9C,6CAA6C;YAC7C,MAAM,kBAAkB,GAAG;gBACzB,qBAAqB,EAAE,kBAAkB,EAAE,eAAe;gBAC1D,iBAAiB;aAClB,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,kBAAkB,EAAE,YAAY,EAClD,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC,CAAC;YAE1E,qDAAqD;YACrD,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,uBAAuB,EAAE,mBAAmB,CAAC,EAAE,kBAAkB,EACpF,iBAAiB,CAAC,qBAAqB,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC,CAAC;QAE9E,CAAC;QAAC,MAAM,CAAC,CAAC,yDAAyD,CAAC,CAAC;IACvE,CAAC;IAED,8EAA8E;IAC9E,UAAU;IACV,8EAA8E;IAEpE,QAAQ,CAAC,CAAM;QACvB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG;gBACjB,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM;gBACjD,sBAAsB,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS;gBAC1D,SAAS,EAAE,KAAK;aACjB,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;YACzC,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;YAE7C,MAAM,kBAAkB,GAAG;gBACzB,sBAAsB,EAAE,sBAAsB;gBAC9C,wBAAwB,EAAE,iBAAiB,EAAE,aAAa;aAC3D,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,kBAAkB,EAAE,OAAO,CAAC,CAAC;YAElD,MAAM,iBAAiB,GAAG;gBACxB,mBAAmB,EAAE,kBAAkB;gBACvC,4BAA4B,EAAE,gBAAgB,EAAE,YAAY;aAC7D,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,iBAAiB,EAAE,MAAM,CAAC,CAAC;YAEhD,MAAM,gBAAgB,GAAG,CAAC,sBAAsB,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC;YAChF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,gBAAgB,EAAE,UAAU,CAAC,CAAC;YAEnD,MAAM,gBAAgB,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,sBAAsB,CAAC,CAAC;YACtF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;YAC3D,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;YAC5D,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,gBAAgB,EAAE,wBAAwB,CAAC,CAAC;YAEjE,MAAM,YAAY,GAAG,CAAC,kBAAkB,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;YACpE,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;gBAChC,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBACpB,IAAI,GAAG,EAAE,CAAC;oBACR,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;oBACzC,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;gBAC7C,CAAC;YACH,CAAC;YAED,MAAM,iBAAiB,GAAG;gBACxB,kBAAkB,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW;aAC1D,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,iBAAiB,EAAE,eAAe,CAAC,CAAC;YACzD,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,iBAAiB,EAAE,aAAa,CAAC,CAAC;YAEvD,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,uBAAuB,CAAC,EAAE,UAAU,CAAC,CAAC;YAC5D,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,kBAAkB,EAAE,YAAY,CAAC,EAAE,uBAAuB,CAAC,CAAC;YAClF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAC,EAAE,WAAW,CAAC,CAAC;YAExD,MAAM,kBAAkB,GAAG;gBACzB,qBAAqB,EAAE,kBAAkB,EAAE,eAAe,EAAE,iBAAiB;aAC9E,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,kBAAkB,EAAE,YAAY,CAAC,CAAC;YAEvD,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,uBAAuB,EAAE,mBAAmB,CAAC,EAAE,kBAAkB,CAAC,CAAC;QAC3F,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AAzQD,qDAyQC"}

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

@@ -0,0 +1,36 @@
+/**
+ * Shield360 LlamaIndex Wrapper
+ *
+ * Mirrors Python SDK: sdk/python/src/shield360/instrumentation/llamaindex/
+ * Uses the same OPERATION_MAP and span semantics as the Python implementation.
+ *
+ * LLM operations get full provider-style telemetry (attributes, events, metrics).
+ * Framework operations (query engine, retriever, index, etc.) get framework-level spans.
+ */
+import { Tracer, Span } from '@opentelemetry/api';
+export default class LlamaIndexWrapper {
+    static aiSystem: string;
+    /**
+     * Stamp `shield360.agent.version_hash` (auto) and `gen_ai.agent.version`
+     * (user override, if set) on the span and return the same attributes so
+     * the caller can merge them into the inference event extras.
+     */
+    static _stampAgentVersion(span: Span, args: {
+        systemInstructionsJson?: string;
+        toolDefinitionsJson?: string;
+        primaryModel?: string;
+        temperature?: number | null;
+        top_p?: number | null;
+        max_tokens?: number | null;
+    }): Record<string, string>;
+    private static _extractModel;
+    private static _extractServerInfo;
+    static _patchLLMChat(tracer: Tracer): any;
+    static _patchLLMComplete(tracer: Tracer): any;
+    private static _processLLMResponse;
+    static _patchQueryEngineQuery(tracer: Tracer): any;
+    static _patchChatEngineChat(tracer: Tracer): any;
+    static _patchRetrieverRetrieve(tracer: Tracer): any;
+    static _patchEmbedding(tracer: Tracer, endpoint?: string): any;
+    static _patchFrameworkMethod(tracer: Tracer, endpoint: string): any;
+}