openlit 1.8.0 → 1.9.0

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

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

package/dist/instrumentation/bedrock/index.js

@@ -0,0 +1,52 @@
+"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 OpenlitBedrockInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-bedrock`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@aws-sdk/client-bedrock-runtime', ['>=3.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(bedrock) {
+        this._patch(bedrock);
+    }
+    _patch(moduleExports) {
+        try {
+            const BedrockRuntimeClient = moduleExports.BedrockRuntimeClient;
+            if (!BedrockRuntimeClient?.prototype)
+                return;
+            if ((0, instrumentation_1.isWrapped)(BedrockRuntimeClient.prototype.send)) {
+                this._unwrap(BedrockRuntimeClient.prototype, 'send');
+            }
+            this._wrap(BedrockRuntimeClient.prototype, 'send', wrapper_1.default._patchSend(this.tracer));
+        }
+        catch (e) {
+            console.error('Error in Bedrock _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const BedrockRuntimeClient = moduleExports.BedrockRuntimeClient;
+            if (BedrockRuntimeClient?.prototype?.send) {
+                this._unwrap(BedrockRuntimeClient.prototype, 'send');
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = OpenlitBedrockInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/bedrock/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,iCAAiC,EACjC,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,MAAM,oBAAoB,GAAG,aAAa,CAAC,oBAAoB,CAAC;YAChE,IAAI,CAAC,oBAAoB,EAAE,SAAS;gBAAE,OAAO;YAE7C,IAAI,IAAA,2BAAS,EAAC,oBAAoB,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnD,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,CAAC,KAAK,CACR,oBAAoB,CAAC,SAAS,EAC9B,MAAM,EACN,iBAAc,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CACvC,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,oBAAoB,GAAG,aAAa,CAAC,oBAAoB,CAAC;YAChE,IAAI,oBAAoB,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;gBAC1C,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AArDD,gDAqDC"}

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

@@ -0,0 +1,10 @@
+import { Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class BedrockWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static _patchSend(tracer: Tracer): any;
+    static _handleConverseCommand(tracer: Tracer, originalMethod: any, instance: any, args: any[]): Promise<any>;
+    static _handleConverseStreamCommand(tracer: Tracer, originalMethod: any, instance: any, args: any[]): Promise<any>;
+    static _handleInvokeModelCommand(tracer: Tracer, originalMethod: any, instance: any, args: any[], commandName: string): Promise<any>;
+}
+export default BedrockWrapper;

package/dist/instrumentation/bedrock/wrapper.js

@@ -0,0 +1,345 @@
+"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"));
+const BEDROCK_SERVER_ADDRESS = 'bedrock-runtime.amazonaws.com';
+const BEDROCK_SERVER_PORT = 443;
+function mapFinishReason(stopReason) {
+    const map = {
+        end_turn: 'stop',
+        max_tokens: 'max_tokens',
+        stop_sequence: 'stop',
+        tool_use: 'tool_calls',
+        content_filtered: 'content_filter',
+        guardrail_intervention: 'content_filter',
+    };
+    return map[stopReason] || stopReason;
+}
+function applyInferenceConfigAttributes(span, inferenceConfig) {
+    if (inferenceConfig.maxTokens !== undefined)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, inferenceConfig.maxTokens);
+    if (inferenceConfig.temperature !== undefined)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, inferenceConfig.temperature);
+    if (inferenceConfig.topP !== undefined)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, inferenceConfig.topP);
+    if (inferenceConfig.topK !== undefined)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K, inferenceConfig.topK);
+    if (inferenceConfig.stopSequences !== undefined)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, inferenceConfig.stopSequences);
+}
+class BedrockWrapper extends base_wrapper_1.default {
+    static _patchSend(tracer) {
+        return (originalMethod) => {
+            return async function (...args) {
+                const command = args[0];
+                if (!command)
+                    return originalMethod.apply(this, args);
+                const commandName = command.constructor?.name || '';
+                if (commandName === 'ConverseCommand') {
+                    return BedrockWrapper._handleConverseCommand(tracer, originalMethod, this, args);
+                }
+                if (commandName === 'ConverseStreamCommand') {
+                    return BedrockWrapper._handleConverseStreamCommand(tracer, originalMethod, this, args);
+                }
+                if (commandName === 'InvokeModelCommand' || commandName === 'InvokeModelWithResponseStreamCommand') {
+                    return BedrockWrapper._handleInvokeModelCommand(tracer, originalMethod, this, args, commandName);
+                }
+                return originalMethod.apply(this, args);
+            };
+        };
+    }
+    static async _handleConverseCommand(tracer, originalMethod, instance, args) {
+        const command = args[0];
+        const input = command.input || {};
+        const modelId = input.modelId || 'unknown';
+        const genAIEndpoint = 'bedrock.converse';
+        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 () => {
+            const startTime = Date.now();
+            let metricParams;
+            try {
+                const response = await originalMethod.apply(instance, args);
+                const duration = (Date.now() - startTime) / 1000;
+                const usage = response.usage || {};
+                const promptTokens = usage.inputTokens || 0;
+                const completionTokens = usage.outputTokens || 0;
+                const totalTokens = usage.totalTokens || promptTokens + completionTokens;
+                const cacheReadTokens = usage.cacheReadInputTokens || 0;
+                const cacheWriteTokens = usage.cacheWriteInputTokens || 0;
+                const finishReason = mapFinishReason(response.stopReason || 'stop');
+                const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
+                const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, promptTokens, completionTokens);
+                BedrockWrapper.setBaseSpanAttributes(span, {
+                    genAIEndpoint,
+                    model: modelId,
+                    cost,
+                    aiSystem: BedrockWrapper.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_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, BEDROCK_SERVER_ADDRESS);
+                span.setAttribute(semantic_convention_1.default.SERVER_PORT, BEDROCK_SERVER_PORT);
+                if (cacheReadTokens > 0)
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_PROMPT_TOKENS_DETAILS_CACHE_READ, cacheReadTokens);
+                if (cacheWriteTokens > 0)
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_PROMPT_TOKENS_DETAILS_CACHE_WRITE, cacheWriteTokens);
+                const requestId = response.$metadata?.requestId;
+                if (requestId)
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, requestId);
+                applyInferenceConfigAttributes(span, input.inferenceConfig || {});
+                if (config_1.default.traceContent) {
+                    const messages = (input.messages || []).map((m) => ({
+                        role: m.role,
+                        content: m.content?.map((c) => c.text || '').join('') || '',
+                    }));
+                    const systemText = input.system?.[0]?.text;
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages, systemText));
+                    if (systemText) {
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS, JSON.stringify([{ type: 'text', content: systemText }]));
+                    }
+                    const outputText = response.output?.message?.content?.map((c) => c.text || '').join('') || '';
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(outputText, finishReason));
+                }
+                metricParams = { genAIEndpoint, model: modelId, cost, aiSystem: BedrockWrapper.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 async _handleConverseStreamCommand(tracer, originalMethod, instance, args) {
+        const command = args[0];
+        const input = command.input || {};
+        const modelId = input.modelId || 'unknown';
+        const genAIEndpoint = 'bedrock.converse_stream';
+        const startTime = Date.now();
+        const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
+        // Call the original method to get the response object (with .stream async iterable)
+        const response = await api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), () => originalMethod.apply(instance, args));
+        // Accumulated state from stream events
+        let llmResponse = '';
+        let finishReason = 'stop';
+        let inputTokens = 0;
+        let outputTokens = 0;
+        let cacheReadTokens = 0;
+        let cacheWriteTokens = 0;
+        let firstChunkTime = null;
+        const originalStream = response.stream;
+        async function* wrappedStream() {
+            try {
+                for await (const event of originalStream) {
+                    if (firstChunkTime === null)
+                        firstChunkTime = Date.now();
+                    if (event.contentBlockDelta?.delta?.text)
+                        llmResponse += event.contentBlockDelta.delta.text;
+                    if (event.messageStop?.stopReason)
+                        finishReason = mapFinishReason(event.messageStop.stopReason);
+                    if (event.metadata?.usage) {
+                        inputTokens = event.metadata.usage.inputTokens || 0;
+                        outputTokens = event.metadata.usage.outputTokens || 0;
+                        cacheReadTokens = event.metadata.usage.cacheReadInputTokens || 0;
+                        cacheWriteTokens = event.metadata.usage.cacheWriteInputTokens || 0;
+                    }
+                    yield event;
+                }
+            }
+            finally {
+                // Record telemetry once the stream is fully consumed
+                try {
+                    const duration = (Date.now() - startTime) / 1000;
+                    const ttft = firstChunkTime !== null ? (firstChunkTime - startTime) / 1000 : 0;
+                    const totalTokens = inputTokens + outputTokens;
+                    const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
+                    const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, inputTokens, outputTokens);
+                    BedrockWrapper.setBaseSpanAttributes(span, {
+                        genAIEndpoint,
+                        model: modelId,
+                        cost,
+                        aiSystem: BedrockWrapper.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, true);
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, modelId);
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+                    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_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+                    span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, BEDROCK_SERVER_ADDRESS);
+                    span.setAttribute(semantic_convention_1.default.SERVER_PORT, BEDROCK_SERVER_PORT);
+                    if (cacheReadTokens > 0)
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_PROMPT_TOKENS_DETAILS_CACHE_READ, cacheReadTokens);
+                    if (cacheWriteTokens > 0)
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_PROMPT_TOKENS_DETAILS_CACHE_WRITE, cacheWriteTokens);
+                    const requestId = response.$metadata?.requestId;
+                    if (requestId)
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, requestId);
+                    applyInferenceConfigAttributes(span, input.inferenceConfig || {});
+                    if (config_1.default.traceContent) {
+                        const messages = (input.messages || []).map((m) => ({
+                            role: m.role,
+                            content: m.content?.map((c) => c.text || '').join('') || '',
+                        }));
+                        const systemText = input.system?.[0]?.text;
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages, systemText));
+                        if (systemText) {
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS, JSON.stringify([{ type: 'text', content: systemText }]));
+                        }
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(llmResponse, finishReason));
+                    }
+                    const metricParams = { genAIEndpoint, model: modelId, cost, aiSystem: BedrockWrapper.aiSystem };
+                    base_wrapper_1.default.recordMetrics(span, metricParams);
+                }
+                catch { /* ignore telemetry errors in finally */ }
+                finally {
+                    span.end();
+                }
+            }
+        }
+        return { ...response, stream: wrappedStream() };
+    }
+    static async _handleInvokeModelCommand(tracer, originalMethod, instance, args, commandName) {
+        const command = args[0];
+        const input = command.input || {};
+        const modelId = input.modelId || 'unknown';
+        const isStream = commandName === 'InvokeModelWithResponseStreamCommand';
+        const genAIEndpoint = isStream ? 'bedrock.invoke_model_stream' : 'bedrock.invoke_model';
+        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 () => {
+            const startTime = Date.now();
+            let metricParams;
+            try {
+                const response = await originalMethod.apply(instance, args);
+                const duration = (Date.now() - startTime) / 1000;
+                // Parse response body
+                let parsedBody = {};
+                let promptTokens = 0;
+                let completionTokens = 0;
+                let outputText = '';
+                let rawFinishReason = 'stop';
+                try {
+                    const bodyBytes = response.body;
+                    if (bodyBytes) {
+                        const bodyStr = typeof bodyBytes === 'string'
+                            ? bodyBytes
+                            : Buffer.from(bodyBytes).toString('utf-8');
+                        parsedBody = JSON.parse(bodyStr);
+                    }
+                }
+                catch { /* ignore parse errors */ }
+                // Handle different provider response formats
+                if (modelId.startsWith('anthropic')) {
+                    promptTokens = parsedBody.usage?.input_tokens || 0;
+                    completionTokens = parsedBody.usage?.output_tokens || 0;
+                    outputText = parsedBody.content?.[0]?.text || '';
+                    rawFinishReason = parsedBody.stop_reason || 'stop';
+                }
+                else if (modelId.startsWith('amazon')) {
+                    promptTokens = parsedBody.inputTextTokenCount || 0;
+                    completionTokens = parsedBody.results?.[0]?.tokenCount || 0;
+                    outputText = parsedBody.results?.[0]?.outputText || '';
+                    rawFinishReason = parsedBody.results?.[0]?.completionReason || 'stop';
+                }
+                else if (modelId.startsWith('meta')) {
+                    promptTokens = parsedBody.prompt_token_count || 0;
+                    completionTokens = parsedBody.generation_token_count || 0;
+                    outputText = parsedBody.generation || '';
+                    rawFinishReason = parsedBody.stop_reason || 'stop';
+                }
+                else if (modelId.startsWith('mistral') || modelId.startsWith('mixtral')) {
+                    promptTokens = parsedBody.usage?.prompt_tokens || 0;
+                    completionTokens = parsedBody.usage?.completion_tokens || 0;
+                    outputText = parsedBody.outputs?.[0]?.text || '';
+                    rawFinishReason = parsedBody.outputs?.[0]?.stop_reason || 'stop';
+                }
+                else if (modelId.startsWith('ai21')) {
+                    promptTokens = parsedBody.prompt?.tokens?.length || 0;
+                    completionTokens = parsedBody.completions?.[0]?.data?.tokens?.length || 0;
+                    outputText = parsedBody.completions?.[0]?.data?.text || '';
+                    rawFinishReason = parsedBody.completions?.[0]?.finishReason?.reason || 'stop';
+                }
+                else {
+                    outputText = parsedBody.output || parsedBody.generation || parsedBody.text || '';
+                }
+                const finishReason = mapFinishReason(rawFinishReason);
+                const totalTokens = promptTokens + completionTokens;
+                const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
+                const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, promptTokens, completionTokens);
+                BedrockWrapper.setBaseSpanAttributes(span, {
+                    genAIEndpoint,
+                    model: modelId,
+                    cost,
+                    aiSystem: BedrockWrapper.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, isStream);
+                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_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, duration);
+                span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, BEDROCK_SERVER_ADDRESS);
+                span.setAttribute(semantic_convention_1.default.SERVER_PORT, BEDROCK_SERVER_PORT);
+                const requestId = response.$metadata?.requestId;
+                if (requestId)
+                    span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, requestId);
+                if (config_1.default.traceContent) {
+                    try {
+                        const reqBody = input.body
+                            ? JSON.parse(typeof input.body === 'string' ? input.body : Buffer.from(input.body).toString())
+                            : {};
+                        const prompt = reqBody.prompt || reqBody.inputText || '';
+                        if (prompt) {
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages([{ role: 'user', content: prompt }]));
+                        }
+                    }
+                    catch { /* ignore */ }
+                    if (outputText) {
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(outputText, finishReason));
+                    }
+                }
+                metricParams = { genAIEndpoint, model: modelId, cost, aiSystem: BedrockWrapper.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);
+            }
+        });
+    }
+}
+BedrockWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_AWS_BEDROCK;
+exports.default = BedrockWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/bedrock/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAAsE;AACtE,0DAAyC;AACzC,4DAA0C;AAC1C,oFAA2D;AAC3D,mEAAkE;AAElE,MAAM,sBAAsB,GAAG,+BAA+B,CAAC;AAC/D,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAEhC,SAAS,eAAe,CAAC,UAAkB;IACzC,MAAM,GAAG,GAA2B;QAClC,QAAQ,EAAE,MAAM;QAChB,UAAU,EAAE,YAAY;QACxB,aAAa,EAAE,MAAM;QACrB,QAAQ,EAAE,YAAY;QACtB,gBAAgB,EAAE,gBAAgB;QAClC,sBAAsB,EAAE,gBAAgB;KACzC,CAAC;IACF,OAAO,GAAG,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC;AACvC,CAAC;AAED,SAAS,8BAA8B,CAAC,IAAS,EAAE,eAAoB;IACrE,IAAI,eAAe,CAAC,SAAS,KAAK,SAAS;QACzC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,eAAe,CAAC,SAAS,CAAC,CAAC;IAC7F,IAAI,eAAe,CAAC,WAAW,KAAK,SAAS;QAC3C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,eAAe,CAAC,WAAW,CAAC,CAAC;IAChG,IAAI,eAAe,CAAC,IAAI,KAAK,SAAS;QACpC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;IACnF,IAAI,eAAe,CAAC,IAAI,KAAK,SAAS;QACpC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;IACnF,IAAI,eAAe,CAAC,aAAa,KAAK,SAAS;QAC7C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,eAAe,CAAC,aAAa,CAAC,CAAC;AACvG,CAAC;AAED,MAAM,cAAe,SAAQ,sBAAW;IAGtC,MAAM,CAAC,UAAU,CAAC,MAAc;QAC9B,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACxB,IAAI,CAAC,OAAO;oBAAE,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAEtD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC;gBAEpD,IAAI,WAAW,KAAK,iBAAiB,EAAE,CAAC;oBACtC,OAAO,cAAc,CAAC,sBAAsB,CAAC,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;gBACnF,CAAC;gBACD,IAAI,WAAW,KAAK,uBAAuB,EAAE,CAAC;oBAC5C,OAAO,cAAc,CAAC,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;gBACzF,CAAC;gBACD,IAAI,WAAW,KAAK,oBAAoB,IAAI,WAAW,KAAK,sCAAsC,EAAE,CAAC;oBACnG,OAAO,cAAc,CAAC,yBAAyB,CAAC,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;gBACnG,CAAC;gBAED,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC1C,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,sBAAsB,CACjC,MAAc,EACd,cAAmB,EACnB,QAAa,EACb,IAAW;QAEX,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC;QAC3C,MAAM,aAAa,GAAG,kBAAkB,CAAC;QAEzC,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACxE,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,YAA4C,CAAC;YACjD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAC5D,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;gBAEjD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;gBACnC,MAAM,YAAY,GAAG,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;gBAC5C,MAAM,gBAAgB,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;gBACjD,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,YAAY,GAAG,gBAAgB,CAAC;gBACzE,MAAM,eAAe,GAAG,KAAK,CAAC,oBAAoB,IAAI,CAAC,CAAC;gBACxD,MAAM,gBAAgB,GAAG,KAAK,CAAC,qBAAqB,IAAI,CAAC,CAAC;gBAE1D,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,UAAU,IAAI,MAAM,CAAC,CAAC;gBAEpE,MAAM,WAAW,GAAG,MAAM,gBAAa,CAAC,iBAAiB,CAAC,gBAAa,CAAC,YAAY,CAAC,CAAC;gBACtF,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;gBAElG,cAAc,CAAC,qBAAqB,CAAC,IAAI,EAAE;oBACzC,aAAa;oBACb,KAAK,EAAE,OAAO;oBACd,IAAI;oBACJ,QAAQ,EAAE,cAAc,CAAC,QAAQ;iBAClC,CAAC,CAAC;gBAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;gBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;gBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;gBAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,gBAAgB,CAAC,CAAC;gBACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;gBACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,6BAAkB,CAAC,uBAAuB,CAAC,CAAC;gBACrG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;gBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,sBAAsB,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;gBAEvE,IAAI,eAAe,GAAG,CAAC;oBACrB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6CAA6C,EAAE,eAAe,CAAC,CAAC;gBACvG,IAAI,gBAAgB,GAAG,CAAC;oBACtB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,8CAA8C,EAAE,gBAAgB,CAAC,CAAC;gBAEzG,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;gBAChD,IAAI,SAAS;oBACX,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;gBAEtE,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;gBAElE,IAAI,gBAAa,CAAC,YAAY,EAAE,CAAC;oBAC/B,MAAM,QAAQ,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;wBACvD,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE;qBACjE,CAAC,CAAC,CAAC;oBACJ,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;oBAC3C,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAa,CAAC,kBAAkB,CAAC,QAAQ,EAAE,UAAU,CAAC,CACvD,CAAC;oBACF,IAAI,UAAU,EAAE,CAAC;wBACf,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,0BAA0B,EAC7C,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CACxD,CAAC;oBACJ,CAAC;oBACD,MAAM,UAAU,GACd,QAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;oBAClF,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,UAAU,EAAE,YAAY,CAAC,CAC5D,CAAC;gBACJ,CAAC;gBAED,YAAY,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE,CAAC;gBAC1F,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBACvC,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,IAAI,YAAY;oBAAE,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAClE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,4BAA4B,CACvC,MAAc,EACd,cAAmB,EACnB,QAAa,EACb,IAAW;QAEX,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC;QAC3C,MAAM,aAAa,GAAG,yBAAyB,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAExE,oFAAoF;QACpF,MAAM,QAAQ,GAAG,MAAM,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,CAC9E,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CACrC,CAAC;QAEF,uCAAuC;QACvC,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,YAAY,GAAG,MAAM,CAAC;QAC1B,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,IAAI,gBAAgB,GAAG,CAAC,CAAC;QACzB,IAAI,cAAc,GAAkB,IAAI,CAAC;QAEzC,MAAM,cAAc,GAAuB,QAAQ,CAAC,MAAM,CAAC;QAE3D,KAAK,SAAS,CAAC,CAAC,aAAa;YAC3B,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;oBACzC,IAAI,cAAc,KAAK,IAAI;wBAAE,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAEzD,IAAI,KAAK,CAAC,iBAAiB,EAAE,KAAK,EAAE,IAAI;wBACtC,WAAW,IAAI,KAAK,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC;oBACpD,IAAI,KAAK,CAAC,WAAW,EAAE,UAAU;wBAC/B,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;oBAC/D,IAAI,KAAK,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC;wBAC1B,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;wBACpD,YAAY,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;wBACtD,eAAe,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,CAAC;wBACjE,gBAAgB,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,CAAC;oBACrE,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,qDAAqD;gBACrD,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;oBACjD,MAAM,IAAI,GAAG,cAAc,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC/E,MAAM,WAAW,GAAG,WAAW,GAAG,YAAY,CAAC;oBAE/C,MAAM,WAAW,GAAG,MAAM,gBAAa,CAAC,iBAAiB,CAAC,gBAAa,CAAC,YAAY,CAAC,CAAC;oBACtF,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;oBAE7F,cAAc,CAAC,qBAAqB,CAAC,IAAI,EAAE;wBACzC,aAAa;wBACb,KAAK,EAAE,OAAO;wBACd,IAAI;wBACJ,QAAQ,EAAE,cAAc,CAAC,QAAQ;qBAClC,CAAC,CAAC;oBAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;oBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,IAAI,CAAC,CAAC;oBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;oBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;oBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,YAAY,CAAC,CAAC;oBAC/E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;oBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;oBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;oBACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,6BAAkB,CAAC,uBAAuB,CAAC,CAAC;oBACrG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;oBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;oBAC/D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,sBAAsB,CAAC,CAAC;oBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;oBAEvE,IAAI,eAAe,GAAG,CAAC;wBACrB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6CAA6C,EAAE,eAAe,CAAC,CAAC;oBACvG,IAAI,gBAAgB,GAAG,CAAC;wBACtB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,8CAA8C,EAAE,gBAAgB,CAAC,CAAC;oBAEzG,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;oBAChD,IAAI,SAAS;wBACX,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;oBAEtE,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;oBAElE,IAAI,gBAAa,CAAC,YAAY,EAAE,CAAC;wBAC/B,MAAM,QAAQ,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;4BACvD,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE;yBACjE,CAAC,CAAC,CAAC;wBACJ,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;wBAC3C,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAa,CAAC,kBAAkB,CAAC,QAAQ,EAAE,UAAU,CAAC,CACvD,CAAC;wBACF,IAAI,UAAU,EAAE,CAAC;4BACf,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,0BAA0B,EAC7C,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CACxD,CAAC;wBACJ,CAAC;wBACD,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;oBACJ,CAAC;oBAED,MAAM,YAAY,GAAuB,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE,CAAC;oBACpH,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;gBAChD,CAAC;gBAAC,MAAM,CAAC,CAAC,wCAAwC,CAAC,CAAC;wBAAS,CAAC;oBAC5D,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,GAAG,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,yBAAyB,CACpC,MAAc,EACd,cAAmB,EACnB,QAAa,EACb,IAAW,EACX,WAAmB;QAEnB,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC;QAC3C,MAAM,QAAQ,GAAG,WAAW,KAAK,sCAAsC,CAAC;QACxE,MAAM,aAAa,GAAG,QAAQ,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,sBAAsB,CAAC;QAExF,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACxE,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,YAA4C,CAAC;YACjD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAC5D,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;gBAEjD,sBAAsB;gBACtB,IAAI,UAAU,GAAQ,EAAE,CAAC;gBACzB,IAAI,YAAY,GAAG,CAAC,CAAC;gBACrB,IAAI,gBAAgB,GAAG,CAAC,CAAC;gBACzB,IAAI,UAAU,GAAG,EAAE,CAAC;gBACpB,IAAI,eAAe,GAAG,MAAM,CAAC;gBAE7B,IAAI,CAAC;oBACH,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC;oBAChC,IAAI,SAAS,EAAE,CAAC;wBACd,MAAM,OAAO,GAAG,OAAO,SAAS,KAAK,QAAQ;4BAC3C,CAAC,CAAC,SAAS;4BACX,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;wBAC7C,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACnC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;gBAErC,6CAA6C;gBAC7C,IAAI,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;oBACpC,YAAY,GAAG,UAAU,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;oBACnD,gBAAgB,GAAG,UAAU,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;oBACxD,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;oBACjD,eAAe,GAAG,UAAU,CAAC,WAAW,IAAI,MAAM,CAAC;gBACrD,CAAC;qBAAM,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACxC,YAAY,GAAG,UAAU,CAAC,mBAAmB,IAAI,CAAC,CAAC;oBACnD,gBAAgB,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,IAAI,CAAC,CAAC;oBAC5D,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,IAAI,EAAE,CAAC;oBACvD,eAAe,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,IAAI,MAAM,CAAC;gBACxE,CAAC;qBAAM,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;oBACtC,YAAY,GAAG,UAAU,CAAC,kBAAkB,IAAI,CAAC,CAAC;oBAClD,gBAAgB,GAAG,UAAU,CAAC,sBAAsB,IAAI,CAAC,CAAC;oBAC1D,UAAU,GAAG,UAAU,CAAC,UAAU,IAAI,EAAE,CAAC;oBACzC,eAAe,GAAG,UAAU,CAAC,WAAW,IAAI,MAAM,CAAC;gBACrD,CAAC;qBAAM,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC1E,YAAY,GAAG,UAAU,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;oBACpD,gBAAgB,GAAG,UAAU,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;oBAC5D,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;oBACjD,eAAe,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,IAAI,MAAM,CAAC;gBACnE,CAAC;qBAAM,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;oBACtC,YAAY,GAAG,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC;oBACtD,gBAAgB,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC;oBAC1E,UAAU,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC;oBAC3D,eAAe,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,IAAI,MAAM,CAAC;gBAChF,CAAC;qBAAM,CAAC;oBACN,UAAU,GAAG,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,IAAI,IAAI,EAAE,CAAC;gBACnF,CAAC;gBAED,MAAM,YAAY,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;gBACtD,MAAM,WAAW,GAAG,YAAY,GAAG,gBAAgB,CAAC;gBACpD,MAAM,WAAW,GAAG,MAAM,gBAAa,CAAC,iBAAiB,CAAC,gBAAa,CAAC,YAAY,CAAC,CAAC;gBACtF,MAAM,IAAI,GAAG,iBAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;gBAElG,cAAc,CAAC,qBAAqB,CAAC,IAAI,EAAE;oBACzC,aAAa;oBACb,KAAK,EAAE,OAAO;oBACd,IAAI;oBACJ,QAAQ,EAAE,cAAc,CAAC,QAAQ;iBAClC,CAAC,CAAC;gBAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;gBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC;gBACzE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;gBAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,gBAAgB,CAAC,CAAC;gBACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;gBACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,6BAAkB,CAAC,uBAAuB,CAAC,CAAC;gBACrG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,QAAQ,CAAC,CAAC;gBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,sBAAsB,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;gBAEvE,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;gBAChD,IAAI,SAAS;oBACX,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;gBAEtE,IAAI,gBAAa,CAAC,YAAY,EAAE,CAAC;oBAC/B,IAAI,CAAC;wBACH,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI;4BACxB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;4BAC9F,CAAC,CAAC,EAAE,CAAC;wBACP,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC;wBACzD,IAAI,MAAM,EAAE,CAAC;4BACX,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;wBACJ,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;oBAExB,IAAI,UAAU,EAAE,CAAC;wBACf,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,sBAAsB,EACzC,iBAAa,CAAC,mBAAmB,CAAC,UAAU,EAAE,YAAY,CAAC,CAC5D,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,YAAY,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE,CAAC;gBAC1F,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBACvC,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,IAAI,YAAY;oBAAE,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAClE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;;AAtXM,uBAAQ,GAAG,6BAAkB,CAAC,yBAAyB,CAAC;AAyXjE,kBAAe,cAAc,CAAC"}

package/dist/instrumentation/index.js

@@ -12,6 +12,11 @@ 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) {
@@ -63,6 +68,11 @@ Instrumentations.availableInstrumentations = {
     '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;AAC7C,kDAAyC;AACzC,wDAA+C;AAC/C,4DAAkD;AAClD,0DAAiD;AACjD,sDAA6C;AAE7C,MAAqB,gBAAgB;IAYnC,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;;AA9DM,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;CACpC,CAAC;kBAViB,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"}