openlit 1.7.1 → 1.9.0

package/dist/instrumentation/groq/wrapper.js

@@ -0,0 +1,289 @@
+"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 GroqWrapper extends base_wrapper_1.default {
+    static _patchChatCompletionCreate(tracer) {
+        const genAIEndpoint = 'groq.chat.completions';
+        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 () => {
+                    return originalMethod.apply(this, args);
+                })
+                    .then((response) => {
+                    const { stream = false } = args[0];
+                    if (stream) {
+                        return helpers_1.default.createStreamProxy(response, GroqWrapper._chatCompletionGenerator({
+                            args,
+                            genAIEndpoint,
+                            response,
+                            span,
+                        }));
+                    }
+                    return GroqWrapper._chatCompletion({ args, genAIEndpoint, response, span });
+                })
+                    .catch((e) => {
+                    helpers_1.default.handleException(span, e);
+                    span.end();
+                    throw e;
+                });
+            };
+        };
+    }
+    static async _chatCompletion({ args, genAIEndpoint, response, span, }) {
+        let metricParams;
+        try {
+            metricParams = await GroqWrapper._chatCompletionCommonSetter({
+                args,
+                genAIEndpoint,
+                result: response,
+                span,
+            });
+            return response;
+        }
+        catch (e) {
+            helpers_1.default.handleException(span, e);
+            throw e;
+        }
+        finally {
+            span.end();
+            // Record metrics after span has ended if parameters are available
+            if (metricParams) {
+                base_wrapper_1.default.recordMetrics(span, metricParams);
+            }
+        }
+    }
+    static async *_chatCompletionGenerator({ args, genAIEndpoint, response, span, }) {
+        let metricParams;
+        const timestamps = [];
+        const startTime = Date.now();
+        try {
+            const { messages } = args[0];
+            let { tools } = args[0];
+            const result = {
+                id: '0',
+                created: -1,
+                model: '',
+                system_fingerprint: '',
+                choices: [
+                    {
+                        index: 0,
+                        logprobs: null,
+                        finish_reason: 'stop',
+                        message: { role: 'assistant', content: '' },
+                    },
+                ],
+                usage: {
+                    prompt_tokens: 0,
+                    completion_tokens: 0,
+                    total_tokens: 0,
+                },
+            };
+            let toolCalls = [];
+            for await (const chunk of response) {
+                timestamps.push(Date.now());
+                result.id = chunk.id;
+                result.created = chunk.created;
+                result.model = chunk.model;
+                if (chunk.system_fingerprint) {
+                    result.system_fingerprint = chunk.system_fingerprint;
+                }
+                if (chunk.choices[0]?.finish_reason) {
+                    result.choices[0].finish_reason = chunk.choices[0].finish_reason;
+                }
+                if (chunk.choices[0]?.logprobs) {
+                    result.choices[0].logprobs = chunk.choices[0].logprobs;
+                }
+                if (chunk.choices[0]?.delta.content) {
+                    result.choices[0].message.content += chunk.choices[0].delta.content;
+                }
+                // Handle tool calls for streaming
+                if (chunk.choices[0]?.delta.tool_calls) {
+                    const deltaTools = chunk.choices[0].delta.tool_calls;
+                    for (const tool of deltaTools) {
+                        const idx = tool.index || 0;
+                        // Extend array if needed
+                        while (toolCalls.length <= idx) {
+                            toolCalls.push({
+                                id: '',
+                                type: 'function',
+                                function: { name: '', arguments: '' }
+                            });
+                        }
+                        if (tool.id) {
+                            // New tool call
+                            toolCalls[idx].id = tool.id;
+                            toolCalls[idx].type = tool.type || 'function';
+                            if (tool.function?.name) {
+                                toolCalls[idx].function.name = tool.function.name;
+                            }
+                            if (tool.function?.arguments) {
+                                toolCalls[idx].function.arguments = tool.function.arguments;
+                            }
+                        }
+                        else if (tool.function?.arguments) {
+                            // Append arguments to existing tool call
+                            toolCalls[idx].function.arguments += tool.function.arguments;
+                        }
+                    }
+                    tools = true;
+                }
+                yield chunk;
+            }
+            if (toolCalls.length > 0) {
+                result.choices[0].message = {
+                    ...result.choices[0].message,
+                    tool_calls: toolCalls
+                };
+            }
+            // Estimate token usage if not provided
+            let promptTokens = 0;
+            for (const message of messages || []) {
+                promptTokens += helpers_1.default.openaiTokens(message.content, result.model) ?? 0;
+            }
+            const completionTokens = helpers_1.default.openaiTokens(result.choices[0].message.content ?? '', result.model);
+            if (completionTokens) {
+                result.usage = {
+                    prompt_tokens: promptTokens,
+                    completion_tokens: completionTokens,
+                    total_tokens: promptTokens + completionTokens,
+                };
+            }
+            args[0].tools = tools;
+            // Calculate TTFT and TBT
+            const ttft = timestamps.length > 0 ? (timestamps[0] - startTime) / 1000 : 0;
+            let tbt = 0;
+            if (timestamps.length > 1) {
+                const timeDiffs = timestamps.slice(1).map((t, i) => t - timestamps[i]);
+                tbt = timeDiffs.reduce((a, b) => a + b, 0) / timeDiffs.length / 1000;
+            }
+            metricParams = await GroqWrapper._chatCompletionCommonSetter({
+                args,
+                genAIEndpoint,
+                result,
+                span,
+                ttft,
+                tbt,
+            });
+            return result;
+        }
+        catch (e) {
+            helpers_1.default.handleException(span, e);
+            throw e;
+        }
+        finally {
+            span.end();
+            // Record metrics after span has ended if parameters are available
+            if (metricParams) {
+                base_wrapper_1.default.recordMetrics(span, metricParams);
+            }
+        }
+    }
+    static async _chatCompletionCommonSetter({ args, genAIEndpoint, result, span, ttft = 0, tbt = 0, }) {
+        const traceContent = config_1.default.traceContent;
+        const { messages, frequency_penalty = 0, max_tokens = null, presence_penalty = 0, seed = null, stop = null, temperature = 1, top_p, user, stream = false, } = args[0];
+        // Request Params attributes : Start
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, top_p || 1);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, max_tokens || -1);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, temperature);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY, presence_penalty);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, frequency_penalty);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_SEED, seed ? String(seed) : '');
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, stream);
+        if (stop) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, Array.isArray(stop) ? stop : [stop]);
+        }
+        if (user) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_USER, user);
+        }
+        if (traceContent) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages || []));
+        }
+        // Request Params attributes : End
+        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_RESPONSE_ID, result.id);
+        const model = result.model || 'llama-3.1-8b-instant';
+        const responseModel = result.model || model;
+        const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
+        // Calculate cost of the operation
+        const cost = helpers_1.default.getChatModelCost(model, pricingInfo, result.usage.prompt_tokens, result.usage.completion_tokens);
+        GroqWrapper.setBaseSpanAttributes(span, {
+            genAIEndpoint,
+            model,
+            user,
+            cost,
+            aiSystem: GroqWrapper.aiSystem,
+        });
+        // Response model
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, responseModel);
+        // Groq-specific attributes
+        if (result.system_fingerprint) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_SYSTEM_FINGERPRINT, result.system_fingerprint);
+        }
+        // Token usage
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, result.usage.prompt_tokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, result.usage.completion_tokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, result.usage.total_tokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, result.usage.total_tokens);
+        // TTFT and TBT metrics
+        if (ttft > 0) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+        }
+        if (tbt > 0) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TBT, tbt);
+        }
+        // Finish reason
+        if (result.choices[0].finish_reason) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [result.choices[0].finish_reason]);
+        }
+        // Output type
+        const outputType = typeof result.choices[0].message.content === 'string'
+            ? semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT
+            : semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_JSON;
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, outputType);
+        // Tool calls handling
+        if (result.choices[0].message.tool_calls) {
+            const toolCalls = result.choices[0].message.tool_calls;
+            const toolNames = toolCalls.map((t) => t.function?.name || '').filter(Boolean);
+            const toolIds = toolCalls.map((t) => t.id || '').filter(Boolean);
+            const toolArgs = toolCalls.map((t) => t.function?.arguments || '').filter(Boolean);
+            const toolTypes = toolCalls.map((t) => t.type || '').filter(Boolean);
+            if (toolNames.length > 0) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_NAME, toolNames.join(', '));
+            }
+            if (toolIds.length > 0) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_CALL_ID, toolIds.join(', '));
+            }
+            if (toolArgs.length > 0) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_CALL_ARGUMENTS, toolArgs);
+            }
+            if (toolTypes.length > 0) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_TYPE, toolTypes.join(', '));
+            }
+        }
+        // Content
+        if (traceContent) {
+            const completionContent = result.choices[0].message.content || '';
+            const toolCalls = result.choices[0].message.tool_calls;
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(completionContent, result.choices[0].finish_reason || 'stop', toolCalls));
+        }
+        return {
+            genAIEndpoint,
+            model,
+            user,
+            cost,
+            aiSystem: GroqWrapper.aiSystem,
+        };
+    }
+}
+GroqWrapper.aiSystem = 'groq';
+exports.default = GroqWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/groq/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAA4E;AAC5E,0DAAyC;AACzC,4DAA0C;AAC1C,oFAA2D;AAC3D,mEAA0C;AAE1C,MAAM,WAAY,SAAQ,sBAAW;IAGnC,MAAM,CAAC,0BAA0B,CAAC,MAAc;QAC9C,MAAM,aAAa,GAAG,uBAAuB,CAAC;QAC9C,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;qBACX,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACtD,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC1C,CAAC,CAAC;qBACD,IAAI,CAAC,CAAC,QAAa,EAAE,EAAE;oBACtB,MAAM,EAAE,MAAM,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;oBAEnC,IAAI,MAAM,EAAE,CAAC;wBACX,OAAO,iBAAa,CAAC,iBAAiB,CACpC,QAAQ,EACR,WAAW,CAAC,wBAAwB,CAAC;4BACnC,IAAI;4BACJ,aAAa;4BACb,QAAQ;4BACR,IAAI;yBACL,CAAC,CACH,CAAC;oBACJ,CAAC;oBAED,OAAO,WAAW,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC9E,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE;oBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACvC,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,MAAM,CAAC,CAAC;gBACV,CAAC,CAAC,CAAC;YACP,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,EAC3B,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,IAAI,GAML;QACC,IAAI,YAAY,CAAC;QACjB,IAAI,CAAC;YACH,YAAY,GAAG,MAAM,WAAW,CAAC,2BAA2B,CAAC;gBAC3D,IAAI;gBACJ,aAAa;gBACb,MAAM,EAAE,QAAQ;gBAChB,IAAI;aACL,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,kEAAkE;YAClE,IAAI,YAAY,EAAE,CAAC;gBACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,CAAC,wBAAwB,CAAC,EACrC,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,IAAI,GAML;QACC,IAAI,YAAY,CAAC;QACjB,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACxB,MAAM,MAAM,GAAG;gBACb,EAAE,EAAE,GAAG;gBACP,OAAO,EAAE,CAAC,CAAC;gBACX,KAAK,EAAE,EAAE;gBACT,kBAAkB,EAAE,EAAE;gBACtB,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC;wBACR,QAAQ,EAAE,IAAI;wBACd,aAAa,EAAE,MAAM;wBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE;qBAC5C;iBACF;gBACD,KAAK,EAAE;oBACL,aAAa,EAAE,CAAC;oBAChB,iBAAiB,EAAE,CAAC;oBACpB,YAAY,EAAE,CAAC;iBAChB;aACF,CAAC;YAEF,IAAI,SAAS,GAAU,EAAE,CAAC;YAE1B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACnC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBAE5B,MAAM,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;gBACrB,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;gBAC/B,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;gBAE3B,IAAI,KAAK,CAAC,kBAAkB,EAAE,CAAC;oBAC7B,MAAM,CAAC,kBAAkB,GAAG,KAAK,CAAC,kBAAkB,CAAC;gBACvD,CAAC;gBAED,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC;oBACpC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBACnE,CAAC;gBACD,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC;oBAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;gBACzD,CAAC;gBACD,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;oBACpC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;gBACtE,CAAC;gBAED,kCAAkC;gBAClC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC;oBACvC,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC;oBAErD,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;wBAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC;wBAE5B,yBAAyB;wBACzB,OAAO,SAAS,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;4BAC/B,SAAS,CAAC,IAAI,CAAC;gCACb,EAAE,EAAE,EAAE;gCACN,IAAI,EAAE,UAAU;gCAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;6BACtC,CAAC,CAAC;wBACL,CAAC;wBAED,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;4BACZ,gBAAgB;4BAChB,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;4BAC5B,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,UAAU,CAAC;4BAC9C,IAAI,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;gCACxB,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;4BACpD,CAAC;4BACD,IAAI,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;gCAC7B,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;4BAC9D,CAAC;wBACH,CAAC;6BAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;4BACpC,yCAAyC;4BACzC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;wBAC/D,CAAC;oBACH,CAAC;oBAED,KAAK,GAAG,IAAI,CAAC;gBACf,CAAC;gBAED,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,GAAG;oBAC1B,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO;oBAC5B,UAAU,EAAE,SAAS;iBACf,CAAC;YACX,CAAC;YAED,uCAAuC;YACvC,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,KAAK,MAAM,OAAO,IAAI,QAAQ,IAAI,EAAE,EAAE,CAAC;gBACrC,YAAY,IAAI,iBAAa,CAAC,YAAY,CAAC,OAAO,CAAC,OAAiB,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC3F,CAAC;YAED,MAAM,gBAAgB,GAAG,iBAAa,CAAC,YAAY,CACjD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,EACvC,MAAM,CAAC,KAAK,CACb,CAAC;YAEF,IAAI,gBAAgB,EAAE,CAAC;gBACrB,MAAM,CAAC,KAAK,GAAG;oBACb,aAAa,EAAE,YAAY;oBAC3B,iBAAiB,EAAE,gBAAgB;oBACnC,YAAY,EAAE,YAAY,GAAG,gBAAgB;iBAC9C,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;YAEtB,yBAAyB;YACzB,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5E,IAAI,GAAG,GAAG,CAAC,CAAC;YACZ,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;gBACvE,GAAG,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC;YACvE,CAAC;YAED,YAAY,GAAG,MAAM,WAAW,CAAC,2BAA2B,CAAC;gBAC3D,IAAI;gBACJ,aAAa;gBACb,MAAM;gBACN,IAAI;gBACJ,IAAI;gBACJ,GAAG;aACJ,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,kEAAkE;YAClE,IAAI,YAAY,EAAE,CAAC;gBACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,EACvC,IAAI,EACJ,aAAa,EACb,MAAM,EACN,IAAI,EACJ,IAAI,GAAG,CAAC,EACR,GAAG,GAAG,CAAC,GAQR;QACC,MAAM,YAAY,GAAG,gBAAa,CAAC,YAAY,CAAC;QAChD,MAAM,EACJ,QAAQ,EACR,iBAAiB,GAAG,CAAC,EACrB,UAAU,GAAG,IAAI,EACjB,gBAAgB,GAAG,CAAC,EACpB,IAAI,GAAG,IAAI,EACX,IAAI,GAAG,IAAI,EACX,WAAW,GAAG,CAAC,EACf,KAAK,EACL,IAAI,EACJ,MAAM,GAAG,KAAK,GACf,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAEZ,oCAAoC;QACpC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;QACvE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC;QAClF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,WAAW,CAAC,CAAC;QAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,gBAAgB,CAAC,CAAC;QACxF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,iBAAiB,CAAC,CAAC;QAC1F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,MAAM,CAAC,CAAC;QACvE,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3G,CAAC;QACD,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,iBAAa,CAAC,kBAAkB,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC;QAChH,CAAC;QACD,kCAAkC;QAElC,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,6BAAkB,CAAC,0BAA0B,CAC9C,CAAC;QAEF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;QAEpE,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,sBAAsB,CAAC;QACrD,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC;QAE5C,MAAM,WAAW,GAAG,MAAM,gBAAa,CAAC,iBAAiB,CAAC,gBAAa,CAAC,YAAY,CAAC,CAAC;QAEtF,kCAAkC;QAClC,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CACzC,KAAK,EACL,WAAW,EACX,MAAM,CAAC,KAAK,CAAC,aAAa,EAC1B,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAC/B,CAAC;QAEF,WAAW,CAAC,qBAAqB,CAAC,IAAI,EAAE;YACtC,aAAa;YACb,KAAK;YACL,IAAI;YACJ,IAAI;YACJ,QAAQ,EAAE,WAAW,CAAC,QAAQ;SAC/B,CAAC,CAAC;QAEH,iBAAiB;QACjB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;QAE3E,2BAA2B;QAC3B,IAAI,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAC9B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kCAAkC,EAAE,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACtG,CAAC;QAED,cAAc;QACd,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC5F,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,0BAA0B,EAC7C,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAC/B,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC3F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAE3F,uBAAuB;QACvB,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC;YACZ,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;QAC/D,CAAC;QAED,gBAAgB;QAChB,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YACpC,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAClC,CAAC;QACJ,CAAC;QAED,cAAc;QACd,MAAM,UAAU,GAAG,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,KAAK,QAAQ;YACtE,CAAC,CAAC,6BAAkB,CAAC,uBAAuB;YAC5C,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC;QAC/C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QAErE,sBAAsB;QACtB,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC;YACvD,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACpF,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACtE,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACxF,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE1E,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC/E,CAAC;YACD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAChF,CAAC;YACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,QAAQ,CAAC,CAAC;YAC7E,CAAC;YACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC/E,CAAC;QACH,CAAC;QAED,UAAU;QACV,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,iBAAiB,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;YAClE,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC;YACvD,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,IAAI,MAAM,EAAE,SAAS,CAAC,CAC3G,CAAC;QACJ,CAAC;QAED,OAAO;YACL,aAAa;YACb,KAAK;YACL,IAAI;YACJ,IAAI;YACJ,QAAQ,EAAE,WAAW,CAAC,QAAQ;SAC/B,CAAC;IACJ,CAAC;;AA7XM,oBAAQ,GAAG,MAAM,CAAC;AAgY3B,kBAAe,WAAW,CAAC"}

package/dist/instrumentation/index.js

@@ -7,6 +7,16 @@ const instrumentation_1 = require("@opentelemetry/instrumentation");
 const openai_1 = __importDefault(require("./openai"));
 const anthropic_1 = __importDefault(require("./anthropic"));
 const cohere_1 = __importDefault(require("./cohere"));
+const groq_1 = __importDefault(require("./groq"));
+const mistral_1 = __importDefault(require("./mistral"));
+const google_ai_1 = __importDefault(require("./google-ai"));
+const together_1 = __importDefault(require("./together"));
+const ollama_1 = __importDefault(require("./ollama"));
+const vercel_ai_1 = __importDefault(require("./vercel-ai"));
+const langchain_1 = __importDefault(require("./langchain"));
+const pinecone_1 = __importDefault(require("./pinecone"));
+const bedrock_1 = __importDefault(require("./bedrock"));
+const llamaindex_1 = __importDefault(require("./llamaindex"));
 class Instrumentations {
     static setup(tracerProvider, disabledInstrumentations = [], instrumentations) {
         if (instrumentations === undefined) {
@@ -53,6 +63,16 @@ Instrumentations.availableInstrumentations = {
     openai: new openai_1.default(),
     anthropic: new anthropic_1.default(),
     cohere: new cohere_1.default(),
+    groq: new groq_1.default(),
+    mistral: new mistral_1.default(),
+    'google-ai': new google_ai_1.default(),
+    together: new together_1.default(),
+    ollama: new ollama_1.default(),
+    'vercel-ai': new vercel_ai_1.default(),
+    langchain: new langchain_1.default(),
+    pinecone: new pinecone_1.default(),
+    bedrock: new bedrock_1.default(),
+    llamaindex: new llamaindex_1.default(),
 };
 exports.default = Instrumentations;
 //# sourceMappingURL=index.js.map

package/dist/instrumentation/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/instrumentation/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAA0E;AAI1E,sDAA6C;AAC7C,4DAAmD;AACnD,sDAA6C;AAE7C,MAAqB,gBAAgB;IAOnC,MAAM,CAAC,KAAK,CACV,cAA8B,EAC9B,2BAAqC,EAAE,EACvC,gBAA0C;QAE1C,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,wBAAwB,GAAG,IAAI,CAAC,2BAA2B,CAAC,wBAAwB,CAAC,CAAC;YAC5F,IAAA,0CAAwB,EAAC;gBACvB,gBAAgB,EAAE,wBAAwB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACzF,cAAc;aACf,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,wBAAwB,GAAG,IAAI,CAAC,2BAA2B,CAC/D,wBAAwB,EACxB,gBAAgB,CACjB,CAAC;YACF,wBAAwB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,EAAE;gBACxD,IAAI,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;oBACxD,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;gBACtE,CAAC;gBACD,IAAI,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;oBAClD,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC,CAAC,CAAC;YACH,IAAA,0CAAwB,EAAC;gBACvB,cAAc;aACf,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,MAAM,CAAC,2BAA2B,CAChC,wBAAkC,EAClC,gBAA0C;QAE1C,MAAM,yBAAyB,GAAG,gBAAgB,IAAI,IAAI,CAAC,yBAAyB,CAAC;QACrF,OAAO,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC;aAC1C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;YACZ,IAAI,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzC,IAAI,OAAO,yBAAyB,CAAC,CAAwB,CAAC,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;oBACtF,yBAAyB,CAAC,CAAwB,CAAC,CAAC,OAAO,EAAE,CAAC;gBAChE,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,OAAO,yBAAyB,CAAC,CAAwB,CAAC,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBACrF,yBAAyB,CAAC,CAAwB,CAAC,CAAC,MAAM,EAAE,CAAC;YAC/D,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAwB,EAAE,yBAAyB,CAAC,CAAwB,CAAC,CAAC,CAAC,CAAC;IACjG,CAAC;;AAzDM,0CAAyB,GAA4B;IAC1D,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,MAAM,EAAE,IAAI,gBAAqB,EAAE;CACpC,CAAC;kBALiB,gBAAgB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/instrumentation/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAA0E;AAI1E,sDAA6C;AAC7C,4DAAmD;AACnD,sDAA6C;AAC7C,kDAAyC;AACzC,wDAA+C;AAC/C,4DAAkD;AAClD,0DAAiD;AACjD,sDAA6C;AAC7C,4DAAkD;AAClD,4DAAmD;AACnD,0DAAiD;AACjD,wDAA+C;AAC/C,8DAAqD;AAErD,MAAqB,gBAAgB;IAiBnC,MAAM,CAAC,KAAK,CACV,cAA8B,EAC9B,2BAAqC,EAAE,EACvC,gBAA0C;QAE1C,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,wBAAwB,GAAG,IAAI,CAAC,2BAA2B,CAAC,wBAAwB,CAAC,CAAC;YAC5F,IAAA,0CAAwB,EAAC;gBACvB,gBAAgB,EAAE,wBAAwB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACzF,cAAc;aACf,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,wBAAwB,GAAG,IAAI,CAAC,2BAA2B,CAC/D,wBAAwB,EACxB,gBAAgB,CACjB,CAAC;YACF,wBAAwB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,EAAE;gBACxD,IAAI,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;oBACxD,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;gBACtE,CAAC;gBACD,IAAI,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;oBAClD,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC,CAAC,CAAC;YACH,IAAA,0CAAwB,EAAC;gBACvB,cAAc;aACf,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,MAAM,CAAC,2BAA2B,CAChC,wBAAkC,EAClC,gBAA0C;QAE1C,MAAM,yBAAyB,GAAG,gBAAgB,IAAI,IAAI,CAAC,yBAAyB,CAAC;QACrF,OAAO,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC;aAC1C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;YACZ,IAAI,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzC,IAAI,OAAO,yBAAyB,CAAC,CAAwB,CAAC,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;oBACtF,yBAAyB,CAAC,CAAwB,CAAC,CAAC,OAAO,EAAE,CAAC;gBAChE,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,OAAO,yBAAyB,CAAC,CAAwB,CAAC,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBACrF,yBAAyB,CAAC,CAAwB,CAAC,CAAC,MAAM,EAAE,CAAC;YAC/D,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAwB,EAAE,yBAAyB,CAAC,CAAwB,CAAC,CAAC,CAAC,CAAC;IACjG,CAAC;;AAnEM,0CAAyB,GAA4B;IAC1D,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,IAAI,EAAE,IAAI,cAAmB,EAAE;IAC/B,OAAO,EAAE,IAAI,iBAAsB,EAAE;IACrC,WAAW,EAAE,IAAI,mBAAuB,EAAE;IAC1C,QAAQ,EAAE,IAAI,kBAAuB,EAAE;IACvC,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,WAAW,EAAE,IAAI,mBAAuB,EAAE;IAC1C,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,QAAQ,EAAE,IAAI,kBAAuB,EAAE;IACvC,OAAO,EAAE,IAAI,iBAAsB,EAAE;IACrC,UAAU,EAAE,IAAI,oBAAyB,EAAE;CAC5C,CAAC;kBAfiB,gBAAgB"}

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

@@ -0,0 +1,23 @@
+import { InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+import { InstrumentationConfig } from '@opentelemetry/instrumentation';
+export interface LangChainInstrumentationConfig extends InstrumentationConfig {
+}
+export default class OpenlitLangChainInstrumentation extends InstrumentationBase {
+    private _callbackManager;
+    private _ritmHook;
+    constructor(config?: LangChainInstrumentationConfig);
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    /** Override enable() to install a direct RITM Hook rather than going through the singleton. */
+    enable(): void;
+    /** Override disable() to remove our direct RITM Hook. */
+    disable(): void;
+    /** Called from tests / manual usage — pass the @langchain/core/callbacks/manager exports. */
+    manualPatch(callbacksManagerModule: any): void;
+    /**
+     * Scan require.cache for @langchain/core's dist/callbacks/manager.cjs.
+     * It is always loaded as a relative dep before the chat_models hook fires.
+     */
+    private _patchFromCache;
+    private _applyPatch;
+    private _unpatch;
+}

package/dist/instrumentation/langchain/index.js

@@ -0,0 +1,109 @@
+"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 OpenlitLangChainInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-langchain`, '1.0.0', config);
+        this._callbackManager = null;
+        this._ritmHook = null;
+    }
+    init() {
+        // We bypass OTel's RequireInTheMiddleSingleton here because it uses a null-whitelist
+        // RITM instance, which causes RITM to report `name` as the `fullModuleName` (including
+        // the .cjs extension), e.g. '@langchain/core/language_models/chat_models.cjs'.
+        // The trie search then fails to match our registered bare name (without .cjs).
+        // Instead, we set up a direct require-in-the-middle Hook in enable() with a proper
+        // whitelist, which causes RITM to preserve the original specifier as `name`.
+        return [];
+    }
+    /** Override enable() to install a direct RITM Hook rather than going through the singleton. */
+    enable() {
+        super.enable(); // sets this._enabled = true, initialises tracer
+        if (this._ritmHook)
+            return;
+        try {
+            // Require RITM directly — it is a transitive dep of @opentelemetry/instrumentation.
+            // Using a whitelist ['@langchain/core/language_models/chat_models'] causes RITM to
+            // preserve the original module specifier as `name` (avoiding the .cjs issue).
+            // eslint-disable-next-line @typescript-eslint/no-var-requires
+            const { Hook } = require('require-in-the-middle');
+            this._ritmHook = new Hook(['@langchain/core/language_models/chat_models'], { internals: true }, (exports) => {
+                this._patchFromCache();
+                return exports;
+            });
+        }
+        catch { /* require-in-the-middle not available — no-op */ }
+    }
+    /** Override disable() to remove our direct RITM Hook. */
+    disable() {
+        super.disable();
+        this._ritmHook?.unhook?.();
+        this._ritmHook = null;
+        this._unpatch();
+    }
+    /** Called from tests / manual usage — pass the @langchain/core/callbacks/manager exports. */
+    manualPatch(callbacksManagerModule) {
+        const CallbackManager = callbacksManagerModule?.CallbackManager;
+        if (CallbackManager) {
+            this._callbackManager = CallbackManager;
+            this._applyPatch(CallbackManager);
+        }
+    }
+    /**
+     * Scan require.cache for @langchain/core's dist/callbacks/manager.cjs.
+     * It is always loaded as a relative dep before the chat_models hook fires.
+     */
+    _patchFromCache() {
+        try {
+            const cache = require.cache;
+            for (const filepath of Object.keys(cache)) {
+                if (filepath.includes('@langchain') &&
+                    filepath.includes('callbacks') &&
+                    filepath.includes('manager') &&
+                    !filepath.endsWith('.map')) {
+                    const mod = cache[filepath];
+                    const CallbackManager = mod?.exports?.CallbackManager;
+                    if (CallbackManager && typeof CallbackManager._configureSync === 'function') {
+                        this._callbackManager = CallbackManager;
+                        this._applyPatch(CallbackManager);
+                        return;
+                    }
+                }
+            }
+        }
+        catch (e) {
+            console.error('LangChain: error scanning require.cache for CallbackManager:', e);
+        }
+    }
+    _applyPatch(CallbackManager) {
+        try {
+            // Patch _configureSync — the single low-level entry point called by both
+            // CallbackManager.configure and getCallbackManagerForConfig (Runnables/chains).
+            if ((0, instrumentation_1.isWrapped)(CallbackManager._configureSync)) {
+                this._unwrap(CallbackManager, '_configureSync');
+            }
+            this._wrap(CallbackManager, '_configureSync', wrapper_1.default._patchConfigure(this.tracer));
+        }
+        catch (e) {
+            console.error('Error in LangChain _patch method:', e);
+        }
+    }
+    _unpatch() {
+        if (!this._callbackManager)
+            return;
+        try {
+            if ((0, instrumentation_1.isWrapped)(this._callbackManager._configureSync)) {
+                this._unwrap(this._callbackManager, '_configureSync');
+            }
+        }
+        catch { /* ignore */ }
+        this._callbackManager = null;
+    }
+}
+exports.default = OpenlitLangChainInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/langchain/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAIwC;AAExC,6CAAwD;AACxD,wDAAyC;AAIzC,MAAqB,+BAAgC,SAAQ,qCAAmB;IAI9E,YAAY,SAAyC,EAAE;QACrD,KAAK,CAAC,GAAG,iCAAsB,4BAA4B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAJxE,qBAAgB,GAAQ,IAAI,CAAC;QAC7B,cAAS,GAAQ,IAAI,CAAC;IAI9B,CAAC;IAES,IAAI;QACZ,qFAAqF;QACrF,uFAAuF;QACvF,+EAA+E;QAC/E,+EAA+E;QAC/E,mFAAmF;QACnF,6EAA6E;QAC7E,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,+FAA+F;IACxF,MAAM;QACX,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,gDAAgD;QAChE,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC;YACH,oFAAoF;YACpF,mFAAmF;YACnF,8EAA8E;YAC9E,8DAA8D;YAC9D,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,uBAAuB,CAAC,CAAC;YAClD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,CACvB,CAAC,6CAA6C,CAAC,EAC/C,EAAE,SAAS,EAAE,IAAI,EAAE,EACnB,CAAC,OAAY,EAAE,EAAE;gBACf,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC;YACjB,CAAC,CACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC,CAAC,iDAAiD,CAAC,CAAC;IAC/D,CAAC;IAED,yDAAyD;IAClD,OAAO;QACZ,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;IAED,6FAA6F;IACtF,WAAW,CAAC,sBAA2B;QAC5C,MAAM,eAAe,GAAG,sBAAsB,EAAE,eAAe,CAAC;QAChE,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;YACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,eAAe;QACrB,IAAI,CAAC;YACH,MAAM,KAAK,GAAI,OAA2D,CAAC,KAAK,CAAC;YACjF,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,IACE,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;oBAC/B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;oBAC9B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAC5B,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;oBACD,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC;oBAC5B,MAAM,eAAe,GAAG,GAAG,EAAE,OAAO,EAAE,eAAe,CAAC;oBACtD,IAAI,eAAe,IAAI,OAAO,eAAe,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;wBAC5E,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;wBACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;wBAClC,OAAO;oBACT,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,8DAA8D,EAAE,CAAC,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,eAAoB;QACtC,IAAI,CAAC;YACH,yEAAyE;YACzE,gFAAgF;YAChF,IAAI,IAAA,2BAAS,EAAC,eAAe,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC9C,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;YAClD,CAAC;YACD,IAAI,CAAC,KAAK,CACR,eAAe,EACf,gBAAgB,EAChB,iBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAC9C,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QACnC,IAAI,CAAC;YACH,IAAI,IAAA,2BAAS,EAAC,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QACxB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAC/B,CAAC;CACF;AA9GD,kDA8GC"}

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

@@ -0,0 +1,6 @@
+import { Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class LangChainWrapper extends BaseWrapper {
+    static _patchConfigure(tracer: Tracer): any;
+}
+export default LangChainWrapper;