openlit 1.7.1 → 1.9.0

package/dist/instrumentation/llamaindex/wrapper.js

@@ -0,0 +1,252 @@
+"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"));
+class LlamaIndexWrapper extends base_wrapper_1.default {
+    // ---- Helpers ---------------------------------------------------------------
+    static _extractServerInfo(instance) {
+        const baseUrl = instance?.session?.openai?.baseURL ||
+            instance?._client?.baseURL ||
+            instance?.clientOptions?.baseURL ||
+            instance?.llm?.session?.openai?.baseURL ||
+            instance?._llm?.session?.openai?.baseURL ||
+            '';
+        if (baseUrl) {
+            try {
+                const url = new URL(baseUrl);
+                const port = url.port ? parseInt(url.port) : (url.protocol === 'https:' ? 443 : 80);
+                return { address: url.hostname, port };
+            }
+            catch { /* ignore */ }
+        }
+        return { address: 'api.openai.com', port: 443 };
+    }
+    // ---- LLM patches -----------------------------------------------------------
+    static _patchLLMChat(tracer) {
+        const genAIEndpoint = 'llamaindex.llm.chat';
+        return (originalMethod) => {
+            return async function (...args) {
+                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    let metricParams;
+                    const startTime = Date.now();
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const duration = (Date.now() - startTime) / 1000;
+                        const params = args[0] || {};
+                        const messages = params.messages || [];
+                        const modelId = this.model || this.modelName || this.metadata?.model || 'unknown';
+                        const aiSystem = LlamaIndexWrapper.aiSystem;
+                        const rawUsage = response?.raw?.usage || response?.usage || {};
+                        const promptTokens = rawUsage.prompt_tokens || rawUsage.input_tokens || 0;
+                        const completionTokens = rawUsage.completion_tokens || rawUsage.output_tokens || 0;
+                        const totalTokens = rawUsage.total_tokens || promptTokens + completionTokens;
+                        const finishReason = response?.raw?.choices?.[0]?.finish_reason || 'stop';
+                        const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
+                        const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, promptTokens, completionTokens);
+                        const { address, port } = LlamaIndexWrapper._extractServerInfo(this);
+                        LlamaIndexWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, cost, aiSystem });
+                        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_REQUEST_IS_STREAM, false);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, modelId);
+                        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_RESPONSE_FINISH_REASON, [finishReason]);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, address);
+                        span.setAttribute(semantic_convention_1.default.SERVER_PORT, port);
+                        if (config_1.default.traceContent) {
+                            const formattedMessages = messages.map((m) => ({
+                                role: 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(formattedMessages));
+                            const outputContent = response?.message?.content || response?.text || '';
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(outputContent, finishReason));
+                        }
+                        metricParams = { genAIEndpoint, model: modelId, cost, aiSystem };
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams)
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                    }
+                });
+            };
+        };
+    }
+    static _patchLLMComplete(tracer) {
+        const genAIEndpoint = 'llamaindex.llm.complete';
+        return (originalMethod) => {
+            return async function (...args) {
+                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    let metricParams;
+                    const startTime = Date.now();
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const duration = (Date.now() - startTime) / 1000;
+                        const prompt = typeof args[0] === 'string' ? args[0] : args[0]?.prompt || '';
+                        const modelId = this.model || this.modelName || this.metadata?.model || 'unknown';
+                        const aiSystem = LlamaIndexWrapper.aiSystem;
+                        const rawUsage = response?.raw?.usage || response?.usage || {};
+                        const promptTokens = rawUsage.prompt_tokens || rawUsage.input_tokens || 0;
+                        const completionTokens = rawUsage.completion_tokens || rawUsage.output_tokens || 0;
+                        const totalTokens = rawUsage.total_tokens || promptTokens + completionTokens;
+                        const finishReason = response?.raw?.choices?.[0]?.finish_reason || 'stop';
+                        const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
+                        const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, promptTokens, completionTokens);
+                        const { address, port } = LlamaIndexWrapper._extractServerInfo(this);
+                        LlamaIndexWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, cost, aiSystem });
+                        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_REQUEST_IS_STREAM, false);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, modelId);
+                        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_RESPONSE_FINISH_REASON, [finishReason]);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, address);
+                        span.setAttribute(semantic_convention_1.default.SERVER_PORT, port);
+                        if (config_1.default.traceContent) {
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages([{ role: 'user', content: prompt }]));
+                            const outputContent = response?.text || '';
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(outputContent, finishReason));
+                        }
+                        metricParams = { genAIEndpoint, model: modelId, cost, aiSystem };
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams)
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                    }
+                });
+            };
+        };
+    }
+    // ---- Query engine patch ----------------------------------------------------
+    static _patchQueryEngineQuery(tracer) {
+        const genAIEndpoint = 'llamaindex.query_engine.query';
+        return (originalMethod) => {
+            return async function (...args) {
+                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    let metricParams;
+                    const startTime = Date.now();
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const duration = (Date.now() - startTime) / 1000;
+                        const queryStr = typeof args[0] === 'string' ? args[0] : args[0]?.query || '';
+                        const llm = this.llm || this._llm;
+                        const modelId = llm?.model || llm?.modelName || llm?.metadata?.model || 'unknown';
+                        const aiSystem = LlamaIndexWrapper.aiSystem;
+                        const { address, port } = LlamaIndexWrapper._extractServerInfo(llm || this);
+                        LlamaIndexWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, aiSystem });
+                        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_REQUEST_IS_STREAM, false);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, address);
+                        span.setAttribute(semantic_convention_1.default.SERVER_PORT, port);
+                        const sourceNodes = response.sourceNodes || response.source_nodes || [];
+                        if (sourceNodes.length > 0) {
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_RETRIEVAL_SOURCE, JSON.stringify(sourceNodes.slice(0, 5).map((n) => ({
+                                id: n.node?.id_ || n.id_ || '',
+                                score: n.score,
+                                text: n.node?.text?.slice(0, 200) || '',
+                            }))));
+                        }
+                        if (config_1.default.traceContent) {
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages([{ role: 'user', content: queryStr }]));
+                            const responseText = typeof response.response === 'string'
+                                ? response.response
+                                : response.toString?.() || '';
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(responseText, 'stop'));
+                        }
+                        metricParams = { genAIEndpoint, model: modelId, aiSystem };
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams)
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                    }
+                });
+            };
+        };
+    }
+    // ---- Chat engine patch -----------------------------------------------------
+    static _patchChatEngineChat(tracer) {
+        const genAIEndpoint = 'llamaindex.chat_engine.chat';
+        return (originalMethod) => {
+            return async function (...args) {
+                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    let metricParams;
+                    const startTime = Date.now();
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const duration = (Date.now() - startTime) / 1000;
+                        const messageInput = args[0];
+                        const message = typeof messageInput === 'string'
+                            ? messageInput
+                            : messageInput?.message || messageInput?.content || '';
+                        const llm = this.llm || this._llm;
+                        const modelId = llm?.model || llm?.modelName || llm?.metadata?.model || 'unknown';
+                        const aiSystem = LlamaIndexWrapper.aiSystem;
+                        const { address, port } = LlamaIndexWrapper._extractServerInfo(llm || this);
+                        LlamaIndexWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, aiSystem });
+                        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_REQUEST_IS_STREAM, false);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, address);
+                        span.setAttribute(semantic_convention_1.default.SERVER_PORT, port);
+                        if (config_1.default.traceContent) {
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages([{ role: 'user', content: message }]));
+                            const responseContent = response?.message?.content ||
+                                (typeof response?.response === 'string' ? response.response : '') ||
+                                response?.toString?.() || '';
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(responseContent, 'stop'));
+                        }
+                        metricParams = { genAIEndpoint, model: modelId, aiSystem };
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams)
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                    }
+                });
+            };
+        };
+    }
+}
+LlamaIndexWrapper.aiSystem = 'llamaindex';
+exports.default = LlamaIndexWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/llamaindex/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAAsE;AACtE,0DAAyC;AACzC,4DAA0C;AAC1C,oFAA2D;AAC3D,mEAAkE;AAElE,MAAM,iBAAkB,SAAQ,sBAAW;IAGzC,+EAA+E;IAEvE,MAAM,CAAC,kBAAkB,CAAC,QAAa;QAC7C,MAAM,OAAO,GACX,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO;YAClC,QAAQ,EAAE,OAAO,EAAE,OAAO;YAC1B,QAAQ,EAAE,aAAa,EAAE,OAAO;YAChC,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO;YACvC,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO;YACxC,EAAE,CAAC;QACL,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC7B,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpF,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;YACzC,CAAC;YAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QAC1B,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;IAClD,CAAC;IAED,+EAA+E;IAE/E,MAAM,CAAC,aAAa,CAAC,MAAc;QACjC,MAAM,aAAa,GAAG,qBAAqB,CAAC;QAC5C,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACxE,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACpE,IAAI,YAA4C,CAAC;oBACjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAC7B,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;wBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,EAAE,KAAK,IAAI,SAAS,CAAC;wBAClF,MAAM,QAAQ,GAAG,iBAAiB,CAAC,QAAQ,CAAC;wBAE5C,MAAM,QAAQ,GAAG,QAAQ,EAAE,GAAG,EAAE,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC;wBAC/D,MAAM,YAAY,GAAG,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC;wBAC1E,MAAM,gBAAgB,GAAG,QAAQ,CAAC,iBAAiB,IAAI,QAAQ,CAAC,aAAa,IAAI,CAAC,CAAC;wBACnF,MAAM,WAAW,GAAG,QAAQ,CAAC,YAAY,IAAI,YAAY,GAAG,gBAAgB,CAAC;wBAC7E,MAAM,YAAY,GAAG,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,MAAM,CAAC;wBAE1E,MAAM,WAAW,GAAG,MAAM,gBAAa,CAAC,iBAAiB,CAAC,gBAAa,CAAC,YAAY,CAAC,CAAC;wBACtF,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;wBAElG,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,iBAAiB,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;wBAErE,iBAAiB,CAAC,qBAAqB,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;wBAEjG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;wBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;wBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;wBAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,gBAAgB,CAAC,CAAC;wBACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;wBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;wBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;wBACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;wBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;wBAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;wBAExD,IAAI,gBAAa,CAAC,YAAY,EAAE,CAAC;4BAC/B,MAAM,iBAAiB,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;gCAClD,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,MAAM;gCACtB,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;6BAC/E,CAAC,CAAC,CAAC;4BACJ,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,iBAAa,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC;4BAEjH,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,EAAE,OAAO,IAAI,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC;4BACzE,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,aAAa,EAAE,YAAY,CAAC,CAC/D,CAAC;wBACJ,CAAC;wBAED,YAAY,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;wBACjE,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACvC,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAI,YAAY;4BAAE,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,iBAAiB,CAAC,MAAc;QACrC,MAAM,aAAa,GAAG,yBAAyB,CAAC;QAChD,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACxE,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACpE,IAAI,YAA4C,CAAC;oBACjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,MAAM,MAAM,GAAG,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,EAAE,CAAC;wBAC7E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,EAAE,KAAK,IAAI,SAAS,CAAC;wBAClF,MAAM,QAAQ,GAAG,iBAAiB,CAAC,QAAQ,CAAC;wBAE5C,MAAM,QAAQ,GAAG,QAAQ,EAAE,GAAG,EAAE,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC;wBAC/D,MAAM,YAAY,GAAG,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC;wBAC1E,MAAM,gBAAgB,GAAG,QAAQ,CAAC,iBAAiB,IAAI,QAAQ,CAAC,aAAa,IAAI,CAAC,CAAC;wBACnF,MAAM,WAAW,GAAG,QAAQ,CAAC,YAAY,IAAI,YAAY,GAAG,gBAAgB,CAAC;wBAC7E,MAAM,YAAY,GAAG,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,MAAM,CAAC;wBAE1E,MAAM,WAAW,GAAG,MAAM,gBAAa,CAAC,iBAAiB,CAAC,gBAAa,CAAC,YAAY,CAAC,CAAC;wBACtF,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;wBAElG,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,iBAAiB,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;wBAErE,iBAAiB,CAAC,qBAAqB,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;wBAEjG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;wBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;wBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;wBAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,gBAAgB,CAAC,CAAC;wBACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;wBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;wBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;wBACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;wBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;wBAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;wBAExD,IAAI,gBAAa,CAAC,YAAY,EAAE,CAAC;4BAC/B,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAa,CAAC,kBAAkB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CACtE,CAAC;4BACF,MAAM,aAAa,GAAG,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC;4BAC3C,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,aAAa,EAAE,YAAY,CAAC,CAC/D,CAAC;wBACJ,CAAC;wBAED,YAAY,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;wBACjE,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACvC,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAI,YAAY;4BAAE,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,+EAA+E;IAE/E,MAAM,CAAC,sBAAsB,CAAC,MAAc;QAC1C,MAAM,aAAa,GAAG,+BAA+B,CAAC;QACtD,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACxE,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACpE,IAAI,YAA4C,CAAC;oBACjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC;wBAE9E,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC;wBAClC,MAAM,OAAO,GAAG,GAAG,EAAE,KAAK,IAAI,GAAG,EAAE,SAAS,IAAI,GAAG,EAAE,QAAQ,EAAE,KAAK,IAAI,SAAS,CAAC;wBAClF,MAAM,QAAQ,GAAG,iBAAiB,CAAC,QAAQ,CAAC;wBAE5C,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,iBAAiB,CAAC,kBAAkB,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;wBAE5E,iBAAiB,CAAC,qBAAqB,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;wBAE3F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,+BAA+B,CAAC,CAAC;wBAC3G,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;wBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;wBAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;wBAExD,MAAM,WAAW,GAAG,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAAC,YAAY,IAAI,EAAE,CAAC;wBACxE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC3B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,IAAI,CAAC,SAAS,CAC1E,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;gCACvC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,EAAE;gCAC9B,KAAK,EAAE,CAAC,CAAC,KAAK;gCACd,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE;6BACxC,CAAC,CAAC,CACJ,CAAC,CAAC;wBACL,CAAC;wBAED,IAAI,gBAAa,CAAC,YAAY,EAAE,CAAC;4BAC/B,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAa,CAAC,kBAAkB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,CACxE,CAAC;4BACF,MAAM,YAAY,GAAG,OAAO,QAAQ,CAAC,QAAQ,KAAK,QAAQ;gCACxD,CAAC,CAAC,QAAQ,CAAC,QAAQ;gCACnB,CAAC,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC;4BAChC,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,YAAY,EAAE,MAAM,CAAC,CACxD,CAAC;wBACJ,CAAC;wBAED,YAAY,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;wBAC3D,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACvC,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAI,YAAY;4BAAE,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,+EAA+E;IAE/E,MAAM,CAAC,oBAAoB,CAAC,MAAc;QACxC,MAAM,aAAa,GAAG,6BAA6B,CAAC;QACpD,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACxE,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACpE,IAAI,YAA4C,CAAC;oBACjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBAC7B,MAAM,OAAO,GAAG,OAAO,YAAY,KAAK,QAAQ;4BAC9C,CAAC,CAAC,YAAY;4BACd,CAAC,CAAC,YAAY,EAAE,OAAO,IAAI,YAAY,EAAE,OAAO,IAAI,EAAE,CAAC;wBAEzD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC;wBAClC,MAAM,OAAO,GAAG,GAAG,EAAE,KAAK,IAAI,GAAG,EAAE,SAAS,IAAI,GAAG,EAAE,QAAQ,EAAE,KAAK,IAAI,SAAS,CAAC;wBAClF,MAAM,QAAQ,GAAG,iBAAiB,CAAC,QAAQ,CAAC;wBAE5C,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,iBAAiB,CAAC,kBAAkB,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;wBAE5E,iBAAiB,CAAC,qBAAqB,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;wBAE3F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,+BAA+B,CAAC,CAAC;wBAC3G,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;wBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;wBAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;wBAExD,IAAI,gBAAa,CAAC,YAAY,EAAE,CAAC;4BAC/B,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAa,CAAC,kBAAkB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CACvE,CAAC;4BAEF,MAAM,eAAe,GACnB,QAAQ,EAAE,OAAO,EAAE,OAAO;gCAC1B,CAAC,OAAO,QAAQ,EAAE,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gCACjE,QAAQ,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC;4BAC/B,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,eAAe,EAAE,MAAM,CAAC,CAC3D,CAAC;wBACJ,CAAC;wBAED,YAAY,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;wBAC3D,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACvC,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAI,YAAY;4BAAE,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;;AAnSM,0BAAQ,GAAG,YAAY,CAAC;AAsSjC,kBAAe,iBAAiB,CAAC"}

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

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

package/dist/instrumentation/mistral/index.js

@@ -0,0 +1,66 @@
+"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 OpenlitMistralInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-mistral`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@mistralai/mistralai', ['>=1.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(mistral) {
+        this._patch(mistral);
+    }
+    _patch(moduleExports) {
+        try {
+            // Chat and Embeddings are instance properties in the new SDK — get their
+            // prototypes via a dummy instance (no API calls are made at construction time)
+            const dummy = new moduleExports.Mistral({ apiKey: 'dummy' });
+            const ChatProto = Object.getPrototypeOf(dummy.chat);
+            const EmbeddingsProto = Object.getPrototypeOf(dummy.embeddings);
+            if ((0, instrumentation_1.isWrapped)(ChatProto.complete)) {
+                this._unwrap(ChatProto, 'complete');
+            }
+            if ((0, instrumentation_1.isWrapped)(ChatProto.stream)) {
+                this._unwrap(ChatProto, 'stream');
+            }
+            if ((0, instrumentation_1.isWrapped)(EmbeddingsProto.create)) {
+                this._unwrap(EmbeddingsProto, 'create');
+            }
+            this._wrap(ChatProto, 'complete', wrapper_1.default._patchChatCompletionCreate(this.tracer));
+            this._wrap(ChatProto, 'stream', wrapper_1.default._patchChatCompletionCreate(this.tracer));
+            this._wrap(EmbeddingsProto, 'create', wrapper_1.default._patchEmbedding(this.tracer));
+        }
+        catch (e) {
+            console.error('Error in _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const dummy = new moduleExports.Mistral({ apiKey: 'dummy' });
+            const ChatProto = Object.getPrototypeOf(dummy.chat);
+            const EmbeddingsProto = Object.getPrototypeOf(dummy.embeddings);
+            this._unwrap(ChatProto, 'complete');
+            this._unwrap(ChatProto, 'stream');
+            this._unwrap(EmbeddingsProto, 'create');
+        }
+        catch (e) {
+            console.error('Error in _unpatch method:', e);
+        }
+    }
+}
+exports.default = OpenlitMistralInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/mistral/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAuC;AAIvC,MAAqB,6BAA8B,SAAQ,qCAAmB;IAC5E,YAAY,SAAuC,EAAE;QACnD,KAAK,CAAC,GAAG,iCAAsB,0BAA0B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC9E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,sBAAsB,EACtB,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,OAAY;QAC7B,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,yEAAyE;YACzE,+EAA+E;YAC/E,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAEhE,IAAI,IAAA,2BAAS,EAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAClC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,IAAA,2BAAS,EAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACpC,CAAC;YACD,IAAI,IAAA,2BAAS,EAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,EAAE,iBAAc,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,EAAE,iBAAc,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,KAAK,CAAC,eAAe,EAAE,QAAQ,EAAE,iBAAc,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACrF,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAChE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YACpC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAClC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;CACF;AAhED,gDAgEC"}

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

@@ -0,0 +1,34 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class MistralWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static _patchChatCompletionCreate(tracer: Tracer): any;
+    static _chatCompletion({ args, genAIEndpoint, response, span, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+    }): Promise<any>;
+    static _chatCompletionGenerator({ args, genAIEndpoint, response, span, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+    }): AsyncGenerator<unknown, any, unknown>;
+    static _chatCompletionCommonSetter({ args, genAIEndpoint, result, span, ttft, tbt, }: {
+        args: any[];
+        genAIEndpoint: string;
+        result: any;
+        span: Span;
+        ttft?: number;
+        tbt?: number;
+    }): Promise<{
+        genAIEndpoint: string;
+        model: any;
+        user: any;
+        cost: number;
+        aiSystem: string;
+    }>;
+    static _patchEmbedding(tracer: Tracer): any;
+}
+export default MistralWrapper;