openlit 1.7.1 → 1.9.0

package/dist/instrumentation/langchain/wrapper.js

@@ -0,0 +1,318 @@
+"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 config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+// Singleton handler — injected once into every CallbackManager created
+let handlerInstance = null;
+class OpenLITCallbackHandler {
+    constructor(tracer) {
+        this.name = 'openlit_callback_handler';
+        this.lc_serializable = false;
+        // LangChain checks this to decide if it should be copied to child runs
+        this.awaitHandlers = false;
+        this.spans = new Map();
+        this.tracer = tracer;
+    }
+    // ---- Model name helpers --------------------------------------------------
+    _extractModelName(llm) {
+        if (!llm)
+            return 'unknown';
+        const id = llm.id || [];
+        const className = id[id.length - 1] || '';
+        // Try various well-known invocation_params / kwargs paths
+        const kw = llm.kwargs || {};
+        const ip = kw.invocation_params || {};
+        return (ip.model_name || ip.model || ip.model_id ||
+            kw.model_name || kw.model || kw.model_id ||
+            className || 'unknown');
+    }
+    _detectProvider(llm) {
+        if (!llm)
+            return semantic_convention_1.default.GEN_AI_SYSTEM_LANGCHAIN;
+        const classPath = (llm.id || []).join('.').toLowerCase();
+        const providerMap = {
+            openai: semantic_convention_1.default.GEN_AI_SYSTEM_OPENAI,
+            anthropic: semantic_convention_1.default.GEN_AI_SYSTEM_ANTHROPIC,
+            bedrock: semantic_convention_1.default.GEN_AI_SYSTEM_AWS_BEDROCK,
+            google: semantic_convention_1.default.GEN_AI_SYSTEM_VERTEXAI,
+            cohere: semantic_convention_1.default.GEN_AI_SYSTEM_COHERE,
+            mistral: semantic_convention_1.default.GEN_AI_SYSTEM_MISTRAL,
+        };
+        for (const [key, val] of Object.entries(providerMap)) {
+            if (classPath.includes(key))
+                return val;
+        }
+        return semantic_convention_1.default.GEN_AI_SYSTEM_LANGCHAIN;
+    }
+    // ---- Parent context -------------------------------------------------------
+    _getParentContext(parentRunId) {
+        if (!parentRunId)
+            return undefined;
+        const holder = this.spans.get(parentRunId);
+        if (!holder)
+            return undefined;
+        return api_1.trace.setSpan(api_1.context.active(), holder.span);
+    }
+    // ---- LLM callbacks -------------------------------------------------------
+    handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, _tags, metadata, name) {
+        try {
+            const modelName = this._extractModelName(llm);
+            const provider = this._detectProvider(llm);
+            const spanName = `chat ${modelName}`;
+            const parentCtx = this._getParentContext(parentRunId);
+            const span = this.tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT }, parentCtx);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, provider);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_ENVIRONMENT, config_1.default.environment || '');
+            span.setAttribute(semantic_convention_1.default.GEN_AI_APPLICATION_NAME, config_1.default.applicationName || '');
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, modelName);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+            // Model parameters from invocation_params
+            const ip = llm?.kwargs?.invocation_params || {};
+            if (ip.temperature != null)
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, ip.temperature);
+            if (ip.max_tokens != null)
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, ip.max_tokens);
+            if (ip.top_p != null)
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, ip.top_p);
+            if (config_1.default.traceContent && messages?.length > 0) {
+                const flatMessages = messages.flat().map((m) => ({
+                    role: m._getType?.() || m.type || m.role || 'user',
+                    content: typeof m.content === 'string' ? m.content : JSON.stringify(m.content),
+                }));
+                span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(flatMessages));
+            }
+            this.spans.set(runId, {
+                span,
+                startTime: Date.now(),
+                modelName,
+                parentRunId,
+                streamingContent: [],
+                tokenTimestamps: [],
+                promptTokens: 0,
+                completionTokens: 0,
+            });
+        }
+        catch { /* non-blocking */ }
+    }
+    handleLLMNewToken(token, _idx, runId) {
+        try {
+            const holder = this.spans.get(runId);
+            if (!holder)
+                return;
+            const now = Date.now();
+            if (!holder.firstTokenTime)
+                holder.firstTokenTime = now;
+            holder.tokenTimestamps.push(now);
+            if (token)
+                holder.streamingContent.push(token);
+            // Mark as streaming
+            holder.span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, true);
+        }
+        catch { /* non-blocking */ }
+    }
+    handleLLMEnd(output, runId) {
+        this._finalizeLLMSpan(runId, output, undefined);
+    }
+    handleLLMError(error, runId) {
+        try {
+            const holder = this.spans.get(runId);
+            if (!holder)
+                return;
+            helpers_1.default.handleException(holder.span, error);
+            holder.span.end();
+            this.spans.delete(runId);
+        }
+        catch { /* non-blocking */ }
+    }
+    async _finalizeLLMSpan(runId, output, _error) {
+        try {
+            const holder = this.spans.get(runId);
+            if (!holder)
+                return;
+            const { span, startTime, modelName, streamingContent, tokenTimestamps, firstTokenTime } = holder;
+            const endTime = Date.now();
+            const isStreaming = streamingContent.length > 0;
+            // Extract tokens from output
+            let promptTokens = 0;
+            let completionTokens = 0;
+            let completionContent = '';
+            let finishReason = 'stop';
+            let responseModel = modelName;
+            if (output?.llm_output) {
+                const lu = output.llm_output;
+                const tu = lu.token_usage || lu.usage || {};
+                promptTokens = tu.prompt_tokens || tu.input_tokens || 0;
+                completionTokens = tu.completion_tokens || tu.output_tokens || 0;
+                responseModel = lu.model_name || lu.model || modelName;
+            }
+            const generations = output?.generations || [];
+            for (const genList of generations) {
+                for (const gen of (Array.isArray(genList) ? genList : [genList])) {
+                    const msg = gen?.message || gen;
+                    // From usage_metadata (standard LangChain)
+                    const um = msg?.usage_metadata;
+                    if (um) {
+                        if (!promptTokens)
+                            promptTokens = um.input_tokens || um.prompt_tokens || 0;
+                        if (!completionTokens)
+                            completionTokens = um.output_tokens || um.completion_tokens || 0;
+                    }
+                    // Content
+                    const content = gen?.text || msg?.content || '';
+                    if (content && typeof content === 'string' && content.length > completionContent.length) {
+                        completionContent = content;
+                    }
+                    // Finish reason
+                    const fr = gen?.generationInfo?.finish_reason || msg?.response_metadata?.finish_reason;
+                    if (fr)
+                        finishReason = fr;
+                }
+            }
+            if (isStreaming) {
+                completionContent = streamingContent.join('');
+            }
+            const totalTokens = promptTokens + completionTokens;
+            const duration = (endTime - startTime) / 1000;
+            const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
+            const cost = helpers_1.default.getChatModelCost(responseModel, pricingInfo, promptTokens, completionTokens);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, responseModel);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, promptTokens);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, completionTokens);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, totalTokens);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, totalTokens);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_COST, cost);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [finishReason]);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+            if (isStreaming && firstTokenTime) {
+                const ttft = (firstTokenTime - startTime) / 1000;
+                span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+                if (tokenTimestamps.length > 1) {
+                    const diffs = tokenTimestamps.slice(1).map((t, i) => t - tokenTimestamps[i]);
+                    const tbt = diffs.reduce((a, b) => a + b, 0) / diffs.length / 1000;
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TBT, tbt);
+                }
+            }
+            if (config_1.default.traceContent && completionContent) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(completionContent, finishReason));
+            }
+            // base attributes (system, endpoint, etc.)
+            base_wrapper_1.default.setBaseSpanAttributes(span, {
+                genAIEndpoint: `langchain.chat_model`,
+                model: responseModel,
+                cost,
+                aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_LANGCHAIN,
+            });
+            const metricParams = {
+                genAIEndpoint: 'langchain.chat_model',
+                model: responseModel,
+                cost,
+                aiSystem: semantic_convention_1.default.GEN_AI_SYSTEM_LANGCHAIN,
+            };
+            span.setStatus({ code: 1 });
+            span.end();
+            base_wrapper_1.default.recordMetrics(span, metricParams);
+            this.spans.delete(runId);
+        }
+        catch { /* non-blocking */ }
+    }
+    // ---- Chain callbacks -------------------------------------------------------
+    handleChainStart(chain, inputs, runId, parentRunId) {
+        try {
+            const id = chain?.id || [];
+            const name = id[id.length - 1] || 'chain';
+            const spanName = `workflow ${name}`;
+            const parentCtx = this._getParentContext(parentRunId);
+            const span = this.tracer.startSpan(spanName, { kind: api_1.SpanKind.INTERNAL }, parentCtx);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, semantic_convention_1.default.GEN_AI_SYSTEM_LANGCHAIN);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_FRAMEWORK);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_ENVIRONMENT, config_1.default.environment || '');
+            span.setAttribute(semantic_convention_1.default.GEN_AI_APPLICATION_NAME, config_1.default.applicationName || '');
+            if (config_1.default.traceContent && inputs) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, JSON.stringify(inputs).slice(0, 2000));
+            }
+            this.spans.set(runId, {
+                span,
+                startTime: Date.now(),
+                modelName: name,
+                parentRunId,
+                streamingContent: [],
+                tokenTimestamps: [],
+                promptTokens: 0,
+                completionTokens: 0,
+            });
+        }
+        catch { /* non-blocking */ }
+    }
+    handleChainEnd(outputs, runId) {
+        try {
+            const holder = this.spans.get(runId);
+            if (!holder)
+                return;
+            const duration = (Date.now() - holder.startTime) / 1000;
+            holder.span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+            if (config_1.default.traceContent && outputs) {
+                holder.span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, JSON.stringify(outputs).slice(0, 2000));
+            }
+            holder.span.setStatus({ code: 1 });
+            holder.span.end();
+            this.spans.delete(runId);
+        }
+        catch { /* non-blocking */ }
+    }
+    handleChainError(error, runId) {
+        try {
+            const holder = this.spans.get(runId);
+            if (!holder)
+                return;
+            helpers_1.default.handleException(holder.span, error);
+            holder.span.end();
+            this.spans.delete(runId);
+        }
+        catch { /* non-blocking */ }
+    }
+}
+// ----------------------------------------------------------------------------
+// Wrapper factory
+// ----------------------------------------------------------------------------
+class LangChainWrapper extends base_wrapper_1.default {
+    static _patchConfigure(tracer) {
+        // Ensure singleton handler
+        if (!handlerInstance) {
+            handlerInstance = new OpenLITCallbackHandler(tracer);
+        }
+        const handler = handlerInstance;
+        return (originalConfigure) => {
+            return function (inheritableHandlers, ...rest) {
+                // inheritableHandlers can be:
+                //   - undefined  : no handlers passed
+                //   - Array      : list of handler objects (chat model calls)
+                //   - CallbackManager instance: already-built manager (Runnable/chain calls)
+                if (Array.isArray(inheritableHandlers) || !inheritableHandlers) {
+                    const handlers = inheritableHandlers ? [...inheritableHandlers] : [];
+                    if (!handlers.some((h) => h?.name === 'openlit_callback_handler')) {
+                        handlers.unshift(handler);
+                    }
+                    return originalConfigure.call(this, handlers, ...rest);
+                }
+                else {
+                    // Existing CallbackManager — add our handler directly
+                    const cbManager = inheritableHandlers;
+                    if (cbManager?.handlers && !cbManager.handlers.some((h) => h?.name === 'openlit_callback_handler')) {
+                        cbManager.addHandler(handler, true);
+                    }
+                    return originalConfigure.call(this, cbManager, ...rest);
+                }
+            };
+        };
+    }
+}
+exports.default = LangChainWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/langchain/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAAsE;AACtE,0DAAyC;AACzC,4DAA0C;AAC1C,oFAA2D;AAC3D,mEAAkE;AAiBlE,uEAAuE;AACvE,IAAI,eAAe,GAAkC,IAAI,CAAC;AAE1D,MAAM,sBAAsB;IAS1B,YAAY,MAAc;QAR1B,SAAI,GAAG,0BAA0B,CAAC;QAClC,oBAAe,GAAG,KAAK,CAAC;QACxB,uEAAuE;QACvE,kBAAa,GAAG,KAAK,CAAC;QAGd,UAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;QAG5C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,6EAA6E;IAErE,iBAAiB,CAAC,GAAQ;QAChC,IAAI,CAAC,GAAG;YAAE,OAAO,SAAS,CAAC;QAC3B,MAAM,EAAE,GAAa,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAE1C,0DAA0D;QAC1D,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC;QAC5B,MAAM,EAAE,GAAG,EAAE,CAAC,iBAAiB,IAAI,EAAE,CAAC;QAEtC,OAAO,CACL,EAAE,CAAC,UAAU,IAAI,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,QAAQ;YACxC,EAAE,CAAC,UAAU,IAAI,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,QAAQ;YACxC,SAAS,IAAI,SAAS,CACvB,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,GAAQ;QAC9B,IAAI,CAAC,GAAG;YAAE,OAAO,6BAAkB,CAAC,uBAAuB,CAAC;QAC5D,MAAM,SAAS,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;QACzD,MAAM,WAAW,GAA2B;YAC1C,MAAM,EAAE,6BAAkB,CAAC,oBAAoB;YAC/C,SAAS,EAAE,6BAAkB,CAAC,uBAAuB;YACrD,OAAO,EAAE,6BAAkB,CAAC,yBAAyB;YACrD,MAAM,EAAE,6BAAkB,CAAC,sBAAsB;YACjD,MAAM,EAAE,6BAAkB,CAAC,oBAAoB;YAC/C,OAAO,EAAE,6BAAkB,CAAC,qBAAqB;SAClD,CAAC;QACF,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;YACrD,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC;gBAAE,OAAO,GAAG,CAAC;QAC1C,CAAC;QACD,OAAO,6BAAkB,CAAC,uBAAuB,CAAC;IACpD,CAAC;IAED,8EAA8E;IAEtE,iBAAiB,CAAC,WAAoB;QAC5C,IAAI,CAAC,WAAW;YAAE,OAAO,SAAS,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QAC9B,OAAO,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;IAED,6EAA6E;IAE7E,oBAAoB,CAClB,GAAQ,EACR,QAAiB,EACjB,KAAa,EACb,WAAoB,EACpB,YAAkB,EAClB,KAAgB,EAChB,QAAc,EACd,IAAa;QAEb,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAC3C,MAAM,QAAQ,GAAG,QAAQ,SAAS,EAAE,CAAC;YAErC,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAChC,QAAQ,EACR,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,EACzB,SAAS,CACV,CAAC;YAEF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAC;YACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;YACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,gBAAa,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;YAC1F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,gBAAa,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;YACnG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,SAAS,CAAC,CAAC;YACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAEtE,0CAA0C;YAC1C,MAAM,EAAE,GAAG,GAAG,EAAE,MAAM,EAAE,iBAAiB,IAAI,EAAE,CAAC;YAChD,IAAI,EAAE,CAAC,WAAW,IAAI,IAAI;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC;YAC7G,IAAI,EAAE,CAAC,UAAU,IAAI,IAAI;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC;YAC1G,IAAI,EAAE,CAAC,KAAK,IAAI,IAAI;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;YAE3F,IAAI,gBAAa,CAAC,YAAY,IAAI,QAAQ,EAAE,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvD,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;oBACpD,IAAI,EAAE,CAAC,CAAC,QAAQ,EAAE,EAAE,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,IAAI,MAAM;oBAClD,OAAO,EAAE,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;iBAC/E,CAAC,CAAC,CAAC;gBACJ,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,iBAAa,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC;YAC9G,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE;gBACpB,IAAI;gBACJ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,SAAS;gBACT,WAAW;gBACX,gBAAgB,EAAE,EAAE;gBACpB,eAAe,EAAE,EAAE;gBACnB,YAAY,EAAE,CAAC;gBACf,gBAAgB,EAAE,CAAC;aACpB,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAChC,CAAC;IAED,iBAAiB,CAAC,KAAa,EAAE,IAAS,EAAE,KAAa;QACvD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACrC,IAAI,CAAC,MAAM;gBAAE,OAAO;YACpB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,cAAc;gBAAE,MAAM,CAAC,cAAc,GAAG,GAAG,CAAC;YACxD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,KAAK;gBAAE,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/C,oBAAoB;YACpB,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,IAAI,CAAC,CAAC;QAC9E,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAChC,CAAC;IAED,YAAY,CAAC,MAAW,EAAE,KAAa;QACrC,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAClD,CAAC;IAED,cAAc,CAAC,KAAU,EAAE,KAAa;QACtC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACrC,IAAI,CAAC,MAAM;gBAAE,OAAO;YACpB,iBAAa,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAChC,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,KAAa,EAAE,MAAW,EAAE,MAAW;QACpE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACrC,IAAI,CAAC,MAAM;gBAAE,OAAO;YACpB,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,gBAAgB,EAAE,eAAe,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;YACjG,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,WAAW,GAAG,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC;YAEhD,6BAA6B;YAC7B,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,IAAI,gBAAgB,GAAG,CAAC,CAAC;YACzB,IAAI,iBAAiB,GAAG,EAAE,CAAC;YAC3B,IAAI,YAAY,GAAG,MAAM,CAAC;YAC1B,IAAI,aAAa,GAAG,SAAS,CAAC;YAE9B,IAAI,MAAM,EAAE,UAAU,EAAE,CAAC;gBACvB,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,CAAC;gBAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,WAAW,IAAI,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC;gBAC5C,YAAY,GAAG,EAAE,CAAC,aAAa,IAAI,EAAE,CAAC,YAAY,IAAI,CAAC,CAAC;gBACxD,gBAAgB,GAAG,EAAE,CAAC,iBAAiB,IAAI,EAAE,CAAC,aAAa,IAAI,CAAC,CAAC;gBACjE,aAAa,GAAG,EAAE,CAAC,UAAU,IAAI,EAAE,CAAC,KAAK,IAAI,SAAS,CAAC;YACzD,CAAC;YAED,MAAM,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;gBAClC,KAAK,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;oBACjE,MAAM,GAAG,GAAG,GAAG,EAAE,OAAO,IAAI,GAAG,CAAC;oBAChC,2CAA2C;oBAC3C,MAAM,EAAE,GAAG,GAAG,EAAE,cAAc,CAAC;oBAC/B,IAAI,EAAE,EAAE,CAAC;wBACP,IAAI,CAAC,YAAY;4BAAE,YAAY,GAAG,EAAE,CAAC,YAAY,IAAI,EAAE,CAAC,aAAa,IAAI,CAAC,CAAC;wBAC3E,IAAI,CAAC,gBAAgB;4BAAE,gBAAgB,GAAG,EAAE,CAAC,aAAa,IAAI,EAAE,CAAC,iBAAiB,IAAI,CAAC,CAAC;oBAC1F,CAAC;oBACD,UAAU;oBACV,MAAM,OAAO,GAAG,GAAG,EAAE,IAAI,IAAI,GAAG,EAAE,OAAO,IAAI,EAAE,CAAC;oBAChD,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,CAAC;wBACxF,iBAAiB,GAAG,OAAO,CAAC;oBAC9B,CAAC;oBACD,gBAAgB;oBAChB,MAAM,EAAE,GAAG,GAAG,EAAE,cAAc,EAAE,aAAa,IAAI,GAAG,EAAE,iBAAiB,EAAE,aAAa,CAAC;oBACvF,IAAI,EAAE;wBAAE,YAAY,GAAG,EAAE,CAAC;gBAC5B,CAAC;YACH,CAAC;YAED,IAAI,WAAW,EAAE,CAAC;gBAChB,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChD,CAAC;YAED,MAAM,WAAW,GAAG,YAAY,GAAG,gBAAgB,CAAC;YACpD,MAAM,QAAQ,GAAG,CAAC,OAAO,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;YAE9C,MAAM,WAAW,GAAG,MAAM,gBAAa,CAAC,iBAAiB,CAAC,gBAAa,CAAC,YAAY,CAAC,CAAC;YACtF,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,aAAa,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;YAExG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;YAC3E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;YAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,gBAAgB,CAAC,CAAC;YACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;YAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;YAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;YAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;YACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,6BAAkB,CAAC,uBAAuB,CAAC,CAAC;YACrG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;YAEjF,IAAI,WAAW,IAAI,cAAc,EAAE,CAAC;gBAClC,MAAM,IAAI,GAAG,CAAC,cAAc,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;gBACjD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;gBAC/D,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7E,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC;oBACnE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;YAED,IAAI,gBAAa,CAAC,YAAY,IAAI,iBAAiB,EAAE,CAAC;gBACpD,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,YAAY,CAAC,CACnE,CAAC;YACJ,CAAC;YAED,2CAA2C;YAC3C,sBAAW,CAAC,qBAAqB,CAAC,IAAI,EAAE;gBACtC,aAAa,EAAE,sBAAsB;gBACrC,KAAK,EAAE,aAAa;gBACpB,IAAI;gBACJ,QAAQ,EAAE,6BAAkB,CAAC,uBAAuB;aACrD,CAAC,CAAC;YAEH,MAAM,YAAY,GAAuB;gBACvC,aAAa,EAAE,sBAAsB;gBACrC,KAAK,EAAE,aAAa;gBACpB,IAAI;gBACJ,QAAQ,EAAE,6BAAkB,CAAC,uBAAuB;aACrD,CAAC;YACF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAC5B,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAChC,CAAC;IAED,+EAA+E;IAE/E,gBAAgB,CAAC,KAAU,EAAE,MAAW,EAAE,KAAa,EAAE,WAAoB;QAC3E,IAAI,CAAC;YACH,MAAM,EAAE,GAAa,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,EAAE,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,OAAO,CAAC;YAC1C,MAAM,QAAQ,GAAG,YAAY,IAAI,EAAE,CAAC;YAEpC,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAChC,QAAQ,EACR,EAAE,IAAI,EAAE,cAAQ,CAAC,QAAQ,EAAE,EAC3B,SAAS,CACV,CAAC;YAEF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,6BAAkB,CAAC,uBAAuB,CAAC,CAAC;YACvG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,+BAA+B,CAAC,CAAC;YAC3G,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,gBAAa,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;YAC1F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,gBAAa,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;YAEnG,IAAI,gBAAa,CAAC,YAAY,IAAI,MAAM,EAAE,CAAC;gBACzC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YACrG,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE;gBACpB,IAAI;gBACJ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,SAAS,EAAE,IAAI;gBACf,WAAW;gBACX,gBAAgB,EAAE,EAAE;gBACpB,eAAe,EAAE,EAAE;gBACnB,YAAY,EAAE,CAAC;gBACf,gBAAgB,EAAE,CAAC;aACpB,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAChC,CAAC;IAED,cAAc,CAAC,OAAY,EAAE,KAAa;QACxC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACrC,IAAI,CAAC,MAAM;gBAAE,OAAO;YACpB,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;YACxD,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;YACxF,IAAI,gBAAa,CAAC,YAAY,IAAI,OAAO,EAAE,CAAC;gBAC1C,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAC9G,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAChC,CAAC;IAED,gBAAgB,CAAC,KAAU,EAAE,KAAa;QACxC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACrC,IAAI,CAAC,MAAM;gBAAE,OAAO;YACpB,iBAAa,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAChC,CAAC;CACF;AAED,+EAA+E;AAC/E,kBAAkB;AAClB,+EAA+E;AAE/E,MAAM,gBAAiB,SAAQ,sBAAW;IACxC,MAAM,CAAC,eAAe,CAAC,MAAc;QACnC,2BAA2B;QAC3B,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,eAAe,GAAG,IAAI,sBAAsB,CAAC,MAAM,CAAC,CAAC;QACvD,CAAC;QACD,MAAM,OAAO,GAAG,eAAe,CAAC;QAEhC,OAAO,CAAC,iBAA0C,EAAE,EAAE;YACpD,OAAO,UAEL,mBAAyB,EACzB,GAAG,IAAW;gBAEd,8BAA8B;gBAC9B,sCAAsC;gBACtC,8DAA8D;gBAC9D,6EAA6E;gBAC7E,IAAI,KAAK,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;oBAC/D,MAAM,QAAQ,GAAU,mBAAmB,CAAC,CAAC,CAAC,CAAC,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5E,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,0BAA0B,CAAC,EAAE,CAAC;wBACvE,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBAC5B,CAAC;oBACD,OAAO,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,CAAC;gBACzD,CAAC;qBAAM,CAAC;oBACN,sDAAsD;oBACtD,MAAM,SAAS,GAAG,mBAAmB,CAAC;oBACtC,IAAI,SAAS,EAAE,QAAQ,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,0BAA0B,CAAC,EAAE,CAAC;wBACxG,SAAS,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;oBACtC,CAAC;oBACD,OAAO,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;CACF;AAED,kBAAe,gBAAgB,CAAC"}

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

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

package/dist/instrumentation/llamaindex/index.js

@@ -0,0 +1,102 @@
+"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 OpenlitLlamaIndexInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-llamaindex`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('llamaindex', ['>=0.3.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(llamaindex) {
+        this._patch(llamaindex);
+    }
+    _patch(moduleExports) {
+        try {
+            // Patch query engines
+            const queryEngineClasses = [
+                'RetrieverQueryEngine',
+                'BaseQueryEngine',
+                'QueryEngine',
+            ];
+            for (const className of queryEngineClasses) {
+                const QueryEngineClass = moduleExports[className];
+                if (QueryEngineClass?.prototype?.query) {
+                    if ((0, instrumentation_1.isWrapped)(QueryEngineClass.prototype.query)) {
+                        this._unwrap(QueryEngineClass.prototype, 'query');
+                    }
+                    this._wrap(QueryEngineClass.prototype, 'query', wrapper_1.default._patchQueryEngineQuery(this.tracer));
+                    break; // Only patch the first one found to avoid duplicates
+                }
+            }
+            // Patch chat engines
+            const chatEngineClasses = [
+                'ContextChatEngine',
+                'SimpleChatEngine',
+                'BaseChatEngine',
+            ];
+            for (const className of chatEngineClasses) {
+                const ChatEngineClass = moduleExports[className];
+                if (ChatEngineClass?.prototype?.chat) {
+                    if ((0, instrumentation_1.isWrapped)(ChatEngineClass.prototype.chat)) {
+                        this._unwrap(ChatEngineClass.prototype, 'chat');
+                    }
+                    this._wrap(ChatEngineClass.prototype, 'chat', wrapper_1.default._patchChatEngineChat(this.tracer));
+                    break; // Only patch the first one found
+                }
+            }
+            // Patch base LLM chat and complete methods for lower-level tracing
+            const llmClasses = ['OpenAI', 'Anthropic', 'Ollama', 'LLM', 'BaseLLM'];
+            for (const className of llmClasses) {
+                const LLMClass = moduleExports[className];
+                if (!LLMClass?.prototype)
+                    continue;
+                if (LLMClass.prototype.chat && !(0, instrumentation_1.isWrapped)(LLMClass.prototype.chat)) {
+                    this._wrap(LLMClass.prototype, 'chat', wrapper_1.default._patchLLMChat(this.tracer));
+                }
+                if (LLMClass.prototype.complete && !(0, instrumentation_1.isWrapped)(LLMClass.prototype.complete)) {
+                    this._wrap(LLMClass.prototype, 'complete', wrapper_1.default._patchLLMComplete(this.tracer));
+                }
+            }
+        }
+        catch (e) {
+            console.error('Error in LlamaIndex _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const allClasses = [
+                'RetrieverQueryEngine', 'BaseQueryEngine', 'QueryEngine',
+                'ContextChatEngine', 'SimpleChatEngine', 'BaseChatEngine',
+                'OpenAI', 'Anthropic', 'Ollama', 'LLM', 'BaseLLM',
+            ];
+            for (const className of allClasses) {
+                const Cls = moduleExports[className];
+                if (Cls?.prototype) {
+                    if ((0, instrumentation_1.isWrapped)(Cls.prototype.query))
+                        this._unwrap(Cls.prototype, 'query');
+                    if ((0, instrumentation_1.isWrapped)(Cls.prototype.chat))
+                        this._unwrap(Cls.prototype, 'chat');
+                    if ((0, instrumentation_1.isWrapped)(Cls.prototype.complete))
+                        this._unwrap(Cls.prototype, 'complete');
+                }
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = OpenlitLlamaIndexInstrumentation;
+//# 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,oEAKwC;AAExC,6CAAwD;AACxD,wDAA0C;AAI1C,MAAqB,gCAAiC,SAAQ,qCAAmB;IAC/E,YAAY,SAA0C,EAAE;QACtD,KAAK,CAAC,GAAG,iCAAsB,6BAA6B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACjF,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,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,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,UAAe;QAChC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC1B,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,sBAAsB;YACtB,MAAM,kBAAkB,GAAG;gBACzB,sBAAsB;gBACtB,iBAAiB;gBACjB,aAAa;aACd,CAAC;YAEF,KAAK,MAAM,SAAS,IAAI,kBAAkB,EAAE,CAAC;gBAC3C,MAAM,gBAAgB,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;gBAClD,IAAI,gBAAgB,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;oBACvC,IAAI,IAAA,2BAAS,EAAC,gBAAgB,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;wBAChD,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;oBACpD,CAAC;oBACD,IAAI,CAAC,KAAK,CACR,gBAAgB,CAAC,SAAS,EAC1B,OAAO,EACP,iBAAiB,CAAC,sBAAsB,CAAC,IAAI,CAAC,MAAM,CAAC,CACtD,CAAC;oBACF,MAAM,CAAC,qDAAqD;gBAC9D,CAAC;YACH,CAAC;YAED,qBAAqB;YACrB,MAAM,iBAAiB,GAAG;gBACxB,mBAAmB;gBACnB,kBAAkB;gBAClB,gBAAgB;aACjB,CAAC;YAEF,KAAK,MAAM,SAAS,IAAI,iBAAiB,EAAE,CAAC;gBAC1C,MAAM,eAAe,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;gBACjD,IAAI,eAAe,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;oBACrC,IAAI,IAAA,2BAAS,EAAC,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC9C,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;oBAClD,CAAC;oBACD,IAAI,CAAC,KAAK,CACR,eAAe,CAAC,SAAS,EACzB,MAAM,EACN,iBAAiB,CAAC,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,CACpD,CAAC;oBACF,MAAM,CAAC,iCAAiC;gBAC1C,CAAC;YACH,CAAC;YAED,mEAAmE;YACnE,MAAM,UAAU,GAAG,CAAC,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;YACvE,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,MAAM,QAAQ,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;gBAC1C,IAAI,CAAC,QAAQ,EAAE,SAAS;oBAAE,SAAS;gBAEnC,IAAI,QAAQ,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,IAAA,2BAAS,EAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;oBACnE,IAAI,CAAC,KAAK,CACR,QAAQ,CAAC,SAAS,EAClB,MAAM,EACN,iBAAiB,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAC7C,CAAC;gBACJ,CAAC;gBACD,IAAI,QAAQ,CAAC,SAAS,CAAC,QAAQ,IAAI,CAAC,IAAA,2BAAS,EAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC3E,IAAI,CAAC,KAAK,CACR,QAAQ,CAAC,SAAS,EAClB,UAAU,EACV,iBAAiB,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CACjD,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG;gBACjB,sBAAsB,EAAE,iBAAiB,EAAE,aAAa;gBACxD,mBAAmB,EAAE,kBAAkB,EAAE,gBAAgB;gBACzD,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS;aAClD,CAAC;YAEF,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,MAAM,GAAG,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;gBACrC,IAAI,GAAG,EAAE,SAAS,EAAE,CAAC;oBACnB,IAAI,IAAA,2BAAS,EAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC;wBAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;oBACzE,IAAI,IAAA,2BAAS,EAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;wBAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;oBACvE,IAAI,IAAA,2BAAS,EAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC;wBAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;gBACjF,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AApHD,mDAoHC"}

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

@@ -0,0 +1,11 @@
+import { Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class LlamaIndexWrapper extends BaseWrapper {
+    static aiSystem: string;
+    private static _extractServerInfo;
+    static _patchLLMChat(tracer: Tracer): any;
+    static _patchLLMComplete(tracer: Tracer): any;
+    static _patchQueryEngineQuery(tracer: Tracer): any;
+    static _patchChatEngineChat(tracer: Tracer): any;
+}
+export default LlamaIndexWrapper;