openlit 1.10.0 → 1.12.0

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

@@ -1,11 +1,36 @@
+/**
+ * OpenLIT LlamaIndex Instrumentation
+ *
+ * Monkey-patches LlamaIndex JS classes to emit OTel-compliant telemetry.
+ * Mirrors the Python SDK: sdk/python/src/openlit/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 { InstrumentationConfig } from '@opentelemetry/instrumentation';
-export interface LlamaIndexInstrumentationConfig extends InstrumentationConfig {
-}
+import type { InstrumentationConfig } from '@opentelemetry/instrumentation';
 export default class OpenlitLlamaIndexInstrumentation extends InstrumentationBase {
-    constructor(config?: LlamaIndexInstrumentationConfig);
+    constructor(config?: InstrumentationConfig);
     protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
     manualPatch(llamaindex: any): void;
-    protected _patch(moduleExports: any): void;
-    protected _unpatch(moduleExports: 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

@@ -1,4 +1,14 @@
 "use strict";
+/**
+ * OpenLIT LlamaIndex Instrumentation
+ *
+ * Monkey-patches LlamaIndex JS classes to emit OTel-compliant telemetry.
+ * Mirrors the Python SDK: sdk/python/src/openlit/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 };
 };
@@ -11,89 +21,198 @@ class OpenlitLlamaIndexInstrumentation extends instrumentation_1.Instrumentation
         super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-llamaindex`, '1.0.0', config);
     }
     init() {
-        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('llamaindex', ['>=0.3.0'], (moduleExports) => {
+        const mainModule = new instrumentation_1.InstrumentationNodeModuleDefinition('llamaindex', ['>=0.3.0'], (moduleExports) => {
             this._patch(moduleExports);
             return moduleExports;
         }, (moduleExports) => {
-            if (moduleExports !== undefined) {
+            if (moduleExports)
                 this._unpatch(moduleExports);
-            }
         });
-        return [module];
+        return [mainModule];
     }
     manualPatch(llamaindex) {
         this._patch(llamaindex);
     }
-    _patch(moduleExports) {
+    // ---------------------------------------------------------------------------
+    // 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 {
-            // Patch query engines
+            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',
-                'BaseQueryEngine',
-                'QueryEngine',
+                'RetrieverQueryEngine', 'TransformQueryEngine',
+                'SubQuestionQueryEngine', '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
+            this._patchProto(m, queryEngineClasses, 'query', wrapper_1.default._patchQueryEngineQuery(tracer));
+            // === CHAT ENGINE OPERATIONS (invoke_workflow spans) ===
             const chatEngineClasses = [
-                'ContextChatEngine',
-                'SimpleChatEngine',
-                'BaseChatEngine',
+                'ContextChatEngine', 'SimpleChatEngine',
+                'CondenseQuestionChatEngine', 'BaseChatEngine', 'ChatEngine',
             ];
-            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));
+            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 (e) {
-            console.error('Error in LlamaIndex _patch method:', e);
-        }
+        catch { /* graceful degradation — do not break the application */ }
     }
-    _unpatch(moduleExports) {
+    // ---------------------------------------------------------------------------
+    // Unpatch
+    // ---------------------------------------------------------------------------
+    _unpatch(m) {
         try {
-            const allClasses = [
-                'RetrieverQueryEngine', 'BaseQueryEngine', 'QueryEngine',
-                'ContextChatEngine', 'SimpleChatEngine', 'BaseChatEngine',
-                'OpenAI', 'Anthropic', 'Ollama', 'LLM', 'BaseLLM',
+            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',
             ];
-            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');
+            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 */ }
     }

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

@@ -1 +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"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/llamaindex/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;AAEH,oEAKwC;AAExC,6CAAwD;AACxD,wDAA0C;AAE1C,MAAqB,gCAAiC,SAAQ,qCAAmB;IAC/E,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,mDAyQC"}

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

@@ -1,11 +1,23 @@
+/**
+ * OpenLIT LlamaIndex Wrapper
+ *
+ * Mirrors Python SDK: sdk/python/src/openlit/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 } from '@opentelemetry/api';
-import BaseWrapper from '../base-wrapper';
-declare class LlamaIndexWrapper extends BaseWrapper {
+export default class LlamaIndexWrapper {
     static aiSystem: 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;
 }
-export default LlamaIndexWrapper;