shield360 1.0.0

package/dist/instrumentation/vercel-ai/wrapper.js

@@ -0,0 +1,539 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+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 = __importStar(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+function spanCreationAttrs(operationName, requestModel) {
+    return {
+        [semantic_convention_1.default.GEN_AI_OPERATION]: operationName,
+        [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: VercelAIWrapper.aiSystem,
+        [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+        [semantic_convention_1.default.SERVER_ADDRESS]: VercelAIWrapper.serverAddress,
+        [semantic_convention_1.default.SERVER_PORT]: VercelAIWrapper.serverPort,
+    };
+}
+class VercelAIWrapper extends base_wrapper_1.default {
+    /**
+     * Stamp `shield360.agent.version_hash` (auto) and `gen_ai.agent.version`
+     * (user override, if set) on the span and return the same attributes so
+     * the caller can merge them into the inference event extras.
+     */
+    static _stampAgentVersion(span, args) {
+        const out = {};
+        try {
+            const versionHash = helpers_1.default.computeAgentVersionHash({
+                systemInstructions: args.systemInstructionsJson ?? null,
+                toolDefinitions: args.toolDefinitionsJson ?? null,
+                primaryModel: args.primaryModel ?? null,
+                runtimeConfig: {
+                    temperature: args.temperature ?? null,
+                    top_p: args.top_p ?? null,
+                    max_tokens: args.max_tokens ?? null,
+                    provider: semantic_convention_1.default.GEN_AI_SYSTEM_VERCEL_AI,
+                },
+                providers: [semantic_convention_1.default.GEN_AI_SYSTEM_VERCEL_AI],
+            });
+            if (versionHash) {
+                out[semantic_convention_1.default.SHIELD360_AGENT_VERSION_HASH] = versionHash;
+                span.setAttribute(semantic_convention_1.default.SHIELD360_AGENT_VERSION_HASH, versionHash);
+            }
+        }
+        catch {
+            // Hash computation must never fail the wrapped call.
+        }
+        const versionLabel = (0, helpers_1.getCurrentAgentVersion)();
+        if (versionLabel) {
+            out[semantic_convention_1.default.GEN_AI_AGENT_VERSION] = versionLabel;
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_VERSION, versionLabel);
+        }
+        return out;
+    }
+    static _patchGenerateText(tracer) {
+        const genAIEndpoint = 'vercel_ai.generateText';
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, modelId),
+                });
+                return api_1.context
+                    .with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    return originalMethod.apply(this, args);
+                })
+                    .then((response) => {
+                    return VercelAIWrapper._chatComplete({
+                        args,
+                        genAIEndpoint,
+                        response,
+                        span,
+                        outputType: semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT,
+                    });
+                })
+                    .catch((e) => {
+                    helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: modelId,
+                        aiSystem: VercelAIWrapper.aiSystem,
+                        serverAddress: VercelAIWrapper.serverAddress,
+                        serverPort: VercelAIWrapper.serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
+                    span.end();
+                    throw e;
+                });
+            };
+        };
+    }
+    static _patchStreamText(tracer) {
+        const genAIEndpoint = 'vercel_ai.streamText';
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, modelId),
+                });
+                const startTime = Date.now();
+                const chunkTimestamps = [];
+                try {
+                    const response = await api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => originalMethod.apply(this, args));
+                    try {
+                        const originalTextStream = response.textStream;
+                        if (originalTextStream && typeof originalTextStream.getReader === 'function') {
+                            const reader = originalTextStream.getReader();
+                            const wrappedTextStream = new ReadableStream({
+                                async pull(controller) {
+                                    const { done, value } = await reader.read();
+                                    if (done) {
+                                        controller.close();
+                                    }
+                                    else {
+                                        chunkTimestamps.push(Date.now());
+                                        controller.enqueue(value);
+                                    }
+                                },
+                                cancel() {
+                                    reader.cancel();
+                                },
+                            });
+                            Object.defineProperty(response, 'textStream', {
+                                value: wrappedTextStream,
+                                writable: true,
+                                configurable: true,
+                            });
+                        }
+                    }
+                    catch (_) {
+                        // Stream interception is best-effort; TTFT/TBT won't be captured
+                    }
+                    Promise.resolve(response.usage)
+                        .then(async (usage) => {
+                        let metricParams;
+                        try {
+                            const ttft = chunkTimestamps.length > 0 ? (chunkTimestamps[0] - startTime) / 1000 : 0;
+                            let tbt = 0;
+                            if (chunkTimestamps.length > 1) {
+                                const timeDiffs = chunkTimestamps.slice(1).map((t, i) => t - chunkTimestamps[i]);
+                                tbt = timeDiffs.reduce((a, b) => a + b, 0) / timeDiffs.length / 1000;
+                            }
+                            const finishReason = await Promise.resolve(response.finishReason).catch(() => 'stop');
+                            const text = await Promise.resolve(response.text).catch(() => '');
+                            const toolCalls = await Promise.resolve(response.toolCalls).catch(() => undefined);
+                            const responseDetails = await Promise.resolve(response.response).catch(() => undefined);
+                            const result = {
+                                usage: {
+                                    promptTokens: usage?.promptTokens || 0,
+                                    completionTokens: usage?.completionTokens || 0,
+                                },
+                                finishReason: finishReason || 'stop',
+                                text: text || '',
+                                toolCalls,
+                                response: responseDetails,
+                            };
+                            metricParams = await VercelAIWrapper._chatCommonSetter({
+                                args,
+                                genAIEndpoint,
+                                result,
+                                span,
+                                isStream: true,
+                                outputType: semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT,
+                                ttft,
+                                tbt,
+                            });
+                        }
+                        catch (e) {
+                            helpers_1.default.handleException(span, e);
+                        }
+                        finally {
+                            span.end();
+                            if (metricParams) {
+                                base_wrapper_1.default.recordMetrics(span, metricParams);
+                            }
+                        }
+                    })
+                        .catch((e) => {
+                        helpers_1.default.handleException(span, e);
+                        base_wrapper_1.default.recordMetrics(span, {
+                            genAIEndpoint,
+                            model: modelId,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                            errorType: e?.constructor?.name || '_OTHER',
+                        });
+                        span.end();
+                    });
+                    return response;
+                }
+                catch (e) {
+                    helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: modelId,
+                        aiSystem: VercelAIWrapper.aiSystem,
+                        serverAddress: VercelAIWrapper.serverAddress,
+                        serverPort: VercelAIWrapper.serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
+                    span.end();
+                    throw e;
+                }
+            };
+        };
+    }
+    static _patchGenerateObject(tracer) {
+        const genAIEndpoint = 'vercel_ai.generateObject';
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, modelId),
+                });
+                return api_1.context
+                    .with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    return originalMethod.apply(this, args);
+                })
+                    .then((response) => {
+                    const result = {
+                        ...response,
+                        text: JSON.stringify(response.object || {}),
+                    };
+                    return VercelAIWrapper._chatComplete({
+                        args,
+                        genAIEndpoint,
+                        response,
+                        span,
+                        outputType: semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_JSON,
+                        resultOverride: result,
+                    });
+                })
+                    .catch((e) => {
+                    helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: modelId,
+                        aiSystem: VercelAIWrapper.aiSystem,
+                        serverAddress: VercelAIWrapper.serverAddress,
+                        serverPort: VercelAIWrapper.serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
+                    span.end();
+                    throw e;
+                });
+            };
+        };
+    }
+    static _patchEmbed(tracer) {
+        const genAIEndpoint = 'vercel_ai.embed';
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING, modelId),
+                });
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    const captureContent = config_1.default.captureMessageContent;
+                    let metricParams;
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const pricingInfo = config_1.default.pricingInfo || {};
+                        const inputTokens = response.usage?.tokens || 0;
+                        const cost = helpers_1.default.getEmbedModelCost(modelId, pricingInfo, inputTokens);
+                        VercelAIWrapper.setBaseSpanAttributes(span, {
+                            genAIEndpoint,
+                            model: modelId,
+                            cost,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                        });
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+                        if (captureContent && params.value !== undefined) {
+                            const inputStr = typeof params.value === 'string' ? params.value : JSON.stringify(params.value);
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, inputStr);
+                        }
+                        metricParams = {
+                            genAIEndpoint,
+                            model: modelId,
+                            cost,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                        };
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        metricParams = {
+                            genAIEndpoint,
+                            model: modelId,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                            errorType: e?.constructor?.name || '_OTHER',
+                        };
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams)
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                    }
+                });
+            };
+        };
+    }
+    static async _chatComplete({ args, genAIEndpoint, response, span, outputType, resultOverride, }) {
+        let metricParams;
+        try {
+            metricParams = await VercelAIWrapper._chatCommonSetter({
+                args,
+                genAIEndpoint,
+                result: resultOverride || response,
+                span,
+                isStream: false,
+                outputType,
+            });
+            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 _chatCommonSetter({ args, genAIEndpoint, result, span, isStream, outputType, ttft = 0, tbt = 0, }) {
+        const captureContent = config_1.default.captureMessageContent;
+        const params = args[0] || {};
+        const modelId = params.model?.modelId || 'unknown';
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, params.temperature ?? 1);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, params.topP ?? 1);
+        if (params.maxTokens != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, params.maxTokens);
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, isStream);
+        if (params.seed != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_SEED, Number(params.seed));
+        }
+        if (params.frequencyPenalty) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, params.frequencyPenalty);
+        }
+        if (params.presencePenalty) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY, params.presencePenalty);
+        }
+        if (params.stopSequences) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, Array.isArray(params.stopSequences) ? params.stopSequences : [params.stopSequences]);
+        }
+        if (params.topK != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K, params.topK);
+        }
+        const messages = params.messages || (params.prompt ? [{ role: 'user', content: params.prompt }] : []);
+        if (captureContent) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages, params.system));
+        }
+        const responseId = result.response?.id;
+        const responseModel = result.response?.modelId || modelId;
+        const inputTokens = result.usage?.promptTokens || 0;
+        const outputTokens = result.usage?.completionTokens || 0;
+        const pricingInfo = config_1.default.pricingInfo || {};
+        const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, inputTokens, outputTokens);
+        VercelAIWrapper.setBaseSpanAttributes(span, {
+            genAIEndpoint,
+            model: modelId,
+            cost,
+            aiSystem: VercelAIWrapper.aiSystem,
+            serverAddress: VercelAIWrapper.serverAddress,
+            serverPort: VercelAIWrapper.serverPort,
+        });
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, responseModel);
+        if (responseId) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, responseId);
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+        const finishReason = result.finishReason || 'stop';
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [finishReason]);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, outputType);
+        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);
+        }
+        if (result.toolCalls?.length > 0) {
+            const toolNames = result.toolCalls.map((t) => t.toolName || '').filter(Boolean);
+            const toolIds = result.toolCalls.map((t) => t.toolCallId || '').filter(Boolean);
+            const toolArgs = result.toolCalls.map((t) => JSON.stringify(t.args || {})).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_ARGS, toolArgs.join(', '));
+            }
+        }
+        const normalizedToolCalls = result.toolCalls?.map((t) => ({
+            id: t.toolCallId || '',
+            name: t.toolName || '',
+            arguments: t.args || {},
+        }));
+        // Vercel AI `tools` is an object map keyed by tool name: { toolName: { description, parameters } }
+        // Normalize to the flat array shape that buildToolDefinitions understands.
+        let toolsForDefinitions = params.tools;
+        if (toolsForDefinitions && !Array.isArray(toolsForDefinitions) && typeof toolsForDefinitions === 'object') {
+            toolsForDefinitions = Object.entries(toolsForDefinitions).map(([name, def]) => ({
+                name,
+                description: def?.description ?? '',
+                parameters: def?.parameters ?? def?.inputSchema ?? {},
+            }));
+        }
+        const toolDefinitionsJson = helpers_1.default.buildToolDefinitions(toolsForDefinitions);
+        let inputMessagesJson;
+        let outputMessagesJson;
+        // Compute system_instructions JSON regardless of captureContent so the
+        // version hash is stable across runs even when content capture is off.
+        // Vercel AI exposes the system prompt either via a top-level `system`
+        // field or as a `{ role: 'system' }` message in the `messages` array.
+        const systemInstructionsJson = params.system
+            ? JSON.stringify([{ type: 'text', content: String(params.system) }])
+            : helpers_1.default.buildSystemInstructionsFromMessages(messages);
+        const versionExtras = VercelAIWrapper._stampAgentVersion(span, {
+            systemInstructionsJson,
+            toolDefinitionsJson,
+            primaryModel: responseModel || modelId,
+            temperature: typeof params.temperature === 'number' ? params.temperature : null,
+            top_p: typeof params.topP === 'number' ? params.topP : null,
+            max_tokens: typeof params.maxTokens === 'number' ? params.maxTokens : null,
+        });
+        if (captureContent) {
+            outputMessagesJson = helpers_1.default.buildOutputMessages(result.text || '', finishReason, normalizedToolCalls);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
+            inputMessagesJson = helpers_1.default.buildInputMessages(messages, params.system);
+            if (systemInstructionsJson) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS, systemInstructionsJson);
+            }
+        }
+        if (toolDefinitionsJson) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS, toolDefinitionsJson);
+        }
+        if (!config_1.default.disableEvents) {
+            const eventAttrs = {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+                [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: modelId,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: responseModel,
+                [semantic_convention_1.default.SERVER_ADDRESS]: VercelAIWrapper.serverAddress,
+                [semantic_convention_1.default.SERVER_PORT]: VercelAIWrapper.serverPort,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON]: [finishReason],
+                [semantic_convention_1.default.GEN_AI_OUTPUT_TYPE]: outputType,
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: inputTokens,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: outputTokens,
+                ...versionExtras,
+            };
+            if (responseId) {
+                eventAttrs[semantic_convention_1.default.GEN_AI_RESPONSE_ID] = responseId;
+            }
+            if (captureContent) {
+                if (inputMessagesJson)
+                    eventAttrs[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES] = inputMessagesJson;
+                if (outputMessagesJson)
+                    eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES] = outputMessagesJson;
+                if (systemInstructionsJson)
+                    eventAttrs[semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS] = systemInstructionsJson;
+            }
+            if (toolDefinitionsJson)
+                eventAttrs[semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS] = toolDefinitionsJson;
+            helpers_1.default.emitInferenceEvent(span, eventAttrs);
+        }
+        return {
+            genAIEndpoint,
+            model: modelId,
+            cost,
+            aiSystem: VercelAIWrapper.aiSystem,
+            serverAddress: VercelAIWrapper.serverAddress,
+            serverPort: VercelAIWrapper.serverPort,
+        };
+    }
+}
+VercelAIWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_VERCEL_AI;
+VercelAIWrapper.serverAddress = 'vercel.ai';
+VercelAIWrapper.serverPort = 443;
+exports.default = VercelAIWrapper;
+//# sourceMappingURL=wrapper.js.map

package/dist/instrumentation/vercel-ai/wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/vercel-ai/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAwF;AACxF,0DAA2C;AAC3C,yDAAwE;AACxE,oFAA2D;AAC3D,mEAAkE;AAElE,SAAS,iBAAiB,CACxB,aAAqB,EACrB,YAAoB;IAEpB,OAAO;QACL,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;QACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,eAAe,CAAC,QAAQ;QACxE,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;QACvD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,eAAe,CAAC,aAAa;QAClE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,eAAe,CAAC,UAAU;KAC7D,CAAC;AACJ,CAAC;AAED,MAAM,eAAgB,SAAQ,sBAAW;IAKvC;;;;OAIG;IACH,MAAM,CAAC,kBAAkB,CACvB,IAAU,EACV,IAOC;QAED,MAAM,GAAG,GAA2B,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,iBAAe,CAAC,uBAAuB,CAAC;gBAC1D,kBAAkB,EAAE,IAAI,CAAC,sBAAsB,IAAI,IAAI;gBACvD,eAAe,EAAE,IAAI,CAAC,mBAAmB,IAAI,IAAI;gBACjD,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI;gBACvC,aAAa,EAAE;oBACb,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI;oBACrC,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI;oBACzB,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI;oBACnC,QAAQ,EAAE,6BAAkB,CAAC,uBAAuB;iBACrD;gBACD,SAAS,EAAE,CAAC,6BAAkB,CAAC,uBAAuB,CAAC;aACxD,CAAC,CAAC;YACH,IAAI,WAAW,EAAE,CAAC;gBAChB,GAAG,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,GAAG,WAAW,CAAC;gBACnE,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,4BAA4B,EAC/C,WAAW,CACZ,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,qDAAqD;QACvD,CAAC;QACD,MAAM,YAAY,GAAG,IAAA,gCAAsB,GAAE,CAAC;QAC9C,IAAI,YAAY,EAAE,CAAC;YACjB,GAAG,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,GAAG,YAAY,CAAC;YAC5D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,CAAC,kBAAkB,CAAC,MAAc;QACtC,MAAM,aAAa,GAAG,wBAAwB,CAAC;QAC/C,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,SAAS,CAAC;gBACnD,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,0BAA0B,IAAI,OAAO,EAAE,CAAC;gBAC/E,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,OAAO,CAAC;iBACtF,CAAC,CAAC;gBACH,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,OAAO,eAAe,CAAC,aAAa,CAAC;wBACnC,IAAI;wBACJ,aAAa;wBACb,QAAQ;wBACR,IAAI;wBACJ,UAAU,EAAE,6BAAkB,CAAC,uBAAuB;qBACvD,CAAC,CAAC;gBACL,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE;oBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;wBAC9B,aAAa;wBACb,KAAK,EAAE,OAAO;wBACd,QAAQ,EAAE,eAAe,CAAC,QAAQ;wBAClC,aAAa,EAAE,eAAe,CAAC,aAAa;wBAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;wBACtC,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;qBAC5C,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,MAAM,CAAC,CAAC;gBACV,CAAC,CAAC,CAAC;YACP,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,gBAAgB,CAAC,MAAc;QACpC,MAAM,aAAa,GAAG,sBAAsB,CAAC;QAC7C,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,SAAS,CAAC;gBACnD,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,0BAA0B,IAAI,OAAO,EAAE,CAAC;gBAC/E,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,OAAO,CAAC;iBACtF,CAAC,CAAC;gBACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,eAAe,GAAa,EAAE,CAAC;gBAErC,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,aAAO,CAAC,IAAI,CACjC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EACrC,KAAK,IAAI,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAC7C,CAAC;oBAEF,IAAI,CAAC;wBACH,MAAM,kBAAkB,GAAG,QAAQ,CAAC,UAAoC,CAAC;wBACzE,IAAI,kBAAkB,IAAI,OAAO,kBAAkB,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;4BAC7E,MAAM,MAAM,GAAG,kBAAkB,CAAC,SAAS,EAAE,CAAC;4BAC9C,MAAM,iBAAiB,GAAG,IAAI,cAAc,CAAS;gCACnD,KAAK,CAAC,IAAI,CAAC,UAAU;oCACnB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;oCAC5C,IAAI,IAAI,EAAE,CAAC;wCACT,UAAU,CAAC,KAAK,EAAE,CAAC;oCACrB,CAAC;yCAAM,CAAC;wCACN,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;wCACjC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oCAC5B,CAAC;gCACH,CAAC;gCACD,MAAM;oCACJ,MAAM,CAAC,MAAM,EAAE,CAAC;gCAClB,CAAC;6BACF,CAAC,CAAC;4BACH,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,YAAY,EAAE;gCAC5C,KAAK,EAAE,iBAAiB;gCACxB,QAAQ,EAAE,IAAI;gCACd,YAAY,EAAE,IAAI;6BACnB,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,iEAAiE;oBACnE,CAAC;oBAED,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;yBAC5B,IAAI,CAAC,KAAK,EAAE,KAAU,EAAE,EAAE;wBACzB,IAAI,YAAY,CAAC;wBACjB,IAAI,CAAC;4BACH,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;4BACtF,IAAI,GAAG,GAAG,CAAC,CAAC;4BACZ,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gCAC/B,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;gCACjF,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;4BACvE,CAAC;4BAED,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC;4BACtF,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;4BAClE,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;4BACnF,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;4BAExF,MAAM,MAAM,GAAG;gCACb,KAAK,EAAE;oCACL,YAAY,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC;oCACtC,gBAAgB,EAAE,KAAK,EAAE,gBAAgB,IAAI,CAAC;iCAC/C;gCACD,YAAY,EAAE,YAAY,IAAI,MAAM;gCACpC,IAAI,EAAE,IAAI,IAAI,EAAE;gCAChB,SAAS;gCACT,QAAQ,EAAE,eAAe;6BAC1B,CAAC;4BAEF,YAAY,GAAG,MAAM,eAAe,CAAC,iBAAiB,CAAC;gCACrD,IAAI;gCACJ,aAAa;gCACb,MAAM;gCACN,IAAI;gCACJ,QAAQ,EAAE,IAAI;gCACd,UAAU,EAAE,6BAAkB,CAAC,uBAAuB;gCACtD,IAAI;gCACJ,GAAG;6BACJ,CAAC,CAAC;wBACL,CAAC;wBAAC,OAAO,CAAM,EAAE,CAAC;4BAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBAC3C,CAAC;gCAAS,CAAC;4BACT,IAAI,CAAC,GAAG,EAAE,CAAC;4BACX,IAAI,YAAY,EAAE,CAAC;gCACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;4BAChD,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC;yBACD,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE;wBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;4BAC9B,aAAa;4BACb,KAAK,EAAE,OAAO;4BACd,QAAQ,EAAE,eAAe,CAAC,QAAQ;4BAClC,aAAa,EAAE,eAAe,CAAC,aAAa;4BAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;4BACtC,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;yBAC5C,CAAC,CAAC;wBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACb,CAAC,CAAC,CAAC;oBAEL,OAAO,QAAQ,CAAC;gBAClB,CAAC;gBAAC,OAAO,CAAM,EAAE,CAAC;oBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;wBAC9B,aAAa;wBACb,KAAK,EAAE,OAAO;wBACd,QAAQ,EAAE,eAAe,CAAC,QAAQ;wBAClC,aAAa,EAAE,eAAe,CAAC,aAAa;wBAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;wBACtC,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;qBAC5C,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,MAAM,CAAC,CAAC;gBACV,CAAC;YACH,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,oBAAoB,CAAC,MAAc;QACxC,MAAM,aAAa,GAAG,0BAA0B,CAAC;QACjD,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,SAAS,CAAC;gBACnD,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,0BAA0B,IAAI,OAAO,EAAE,CAAC;gBAC/E,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,OAAO,CAAC;iBACtF,CAAC,CAAC;gBACH,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,MAAM,GAAG;wBACb,GAAG,QAAQ;wBACX,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC;qBAC5C,CAAC;oBACF,OAAO,eAAe,CAAC,aAAa,CAAC;wBACnC,IAAI;wBACJ,aAAa;wBACb,QAAQ;wBACR,IAAI;wBACJ,UAAU,EAAE,6BAAkB,CAAC,uBAAuB;wBACtD,cAAc,EAAE,MAAM;qBACvB,CAAC,CAAC;gBACL,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE;oBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;wBAC9B,aAAa;wBACb,KAAK,EAAE,OAAO;wBACd,QAAQ,EAAE,eAAe,CAAC,QAAQ;wBAClC,aAAa,EAAE,eAAe,CAAC,aAAa;wBAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;wBACtC,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;qBAC5C,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,MAAM,CAAC,CAAC;gBACV,CAAC,CAAC,CAAC;YACP,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,MAAc;QAC/B,MAAM,aAAa,GAAG,iBAAiB,CAAC;QACxC,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,SAAS,CAAC;gBACnD,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,+BAA+B,IAAI,OAAO,EAAE,CAAC;gBACpF,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,OAAO,CAAC;iBAC3F,CAAC,CAAC;gBACH,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACpE,MAAM,cAAc,GAAG,gBAAe,CAAC,qBAAqB,CAAC;oBAC7D,IAAI,YAA4C,CAAC;oBACjD,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBAExD,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;wBACtD,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,EAAE,MAAM,IAAI,CAAC,CAAC;wBAChD,MAAM,IAAI,GAAG,iBAAe,CAAC,iBAAiB,CAAC,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;wBAElF,eAAe,CAAC,qBAAqB,CAAC,IAAI,EAAE;4BAC1C,aAAa;4BACb,KAAK,EAAE,OAAO;4BACd,IAAI;4BACJ,QAAQ,EAAE,eAAe,CAAC,QAAQ;4BAClC,aAAa,EAAE,eAAe,CAAC,aAAa;4BAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;yBACvC,CAAC,CAAC;wBAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;wBAE7E,IAAI,cAAc,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;4BACjD,MAAM,QAAQ,GAAG,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;4BAChG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC;wBACxE,CAAC;wBAED,YAAY,GAAG;4BACb,aAAa;4BACb,KAAK,EAAE,OAAO;4BACd,IAAI;4BACJ,QAAQ,EAAE,eAAe,CAAC,QAAQ;4BAClC,aAAa,EAAE,eAAe,CAAC,aAAa;4BAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;yBACvC,CAAC;wBACF,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACzC,YAAY,GAAG;4BACb,aAAa;4BACb,KAAK,EAAE,OAAO;4BACd,QAAQ,EAAE,eAAe,CAAC,QAAQ;4BAClC,aAAa,EAAE,eAAe,CAAC,aAAa;4BAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;4BACtC,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;yBAC5C,CAAC;wBACF,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAI,YAAY;4BAAE,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,EACzB,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,IAAI,EACJ,UAAU,EACV,cAAc,GAQf;QACC,IAAI,YAAY,CAAC;QACjB,IAAI,CAAC;YACH,YAAY,GAAG,MAAM,eAAe,CAAC,iBAAiB,CAAC;gBACrD,IAAI;gBACJ,aAAa;gBACb,MAAM,EAAE,cAAc,IAAI,QAAQ;gBAClC,IAAI;gBACJ,QAAQ,EAAE,KAAK;gBACf,UAAU;aACX,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,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,iBAAiB,CAAC,EAC7B,IAAI,EACJ,aAAa,EACb,MAAM,EACN,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,IAAI,GAAG,CAAC,EACR,GAAG,GAAG,CAAC,GAUR;QACC,MAAM,cAAc,GAAG,gBAAe,CAAC,qBAAqB,CAAC;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,SAAS,CAAC;QAEnD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,MAAM,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;QAC1F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC7E,IAAI,MAAM,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;YAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QACpF,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC;QACzE,IAAI,MAAM,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC5B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAClG,CAAC;QACD,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC;QAChG,CAAC;QACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,6BAA6B,EAChD,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CACpF,CAAC;QACJ,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAEtG,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAe,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAC5D,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;QACvC,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,EAAE,OAAO,IAAI,OAAO,CAAC;QAC1D,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;QACpD,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,EAAE,gBAAgB,IAAI,CAAC,CAAC;QAEzD,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QACtD,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAE/F,eAAe,CAAC,qBAAqB,CAAC,IAAI,EAAE;YAC1C,aAAa;YACb,KAAK,EAAE,OAAO;YACd,IAAI;YACJ,QAAQ,EAAE,eAAe,CAAC,QAAQ;YAClC,aAAa,EAAE,eAAe,CAAC,aAAa;YAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;SACvC,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;QAC3E,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;QAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,YAAY,CAAC,CAAC;QAE/E,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC;QACnD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QAErE,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,IAAI,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACrF,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACrF,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAEhG,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,gBAAgB,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;QAED,MAAM,mBAAmB,GAAG,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;YAC7D,EAAE,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE;YACtB,IAAI,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;YACtB,SAAS,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE;SACxB,CAAC,CAAC,CAAC;QAEJ,mGAAmG;QACnG,2EAA2E;QAC3E,IAAI,mBAAmB,GAAQ,MAAM,CAAC,KAAK,CAAC;QAC5C,IAAI,mBAAmB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,OAAO,mBAAmB,KAAK,QAAQ,EAAE,CAAC;YAC1G,mBAAmB,GAAG,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAgB,EAAE,EAAE,CAAC,CAAC;gBAC7F,IAAI;gBACJ,WAAW,EAAE,GAAG,EAAE,WAAW,IAAI,EAAE;gBACnC,UAAU,EAAE,GAAG,EAAE,UAAU,IAAI,GAAG,EAAE,WAAW,IAAI,EAAE;aACtD,CAAC,CAAC,CAAC;QACN,CAAC;QACD,MAAM,mBAAmB,GAAG,iBAAe,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,CAAC;QAEtF,IAAI,iBAAqC,CAAC;QAC1C,IAAI,kBAAsC,CAAC;QAC3C,uEAAuE;QACvE,uEAAuE;QACvE,sEAAsE;QACtE,sEAAsE;QACtE,MAAM,sBAAsB,GAAuB,MAAM,CAAC,MAAM;YAC9D,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACpE,CAAC,CAAC,iBAAe,CAAC,mCAAmC,CAAC,QAAQ,CAAC,CAAC;QAClE,MAAM,aAAa,GAAG,eAAe,CAAC,kBAAkB,CAAC,IAAI,EAAE;YAC7D,sBAAsB;YACtB,mBAAmB;YACnB,YAAY,EAAE,aAAa,IAAI,OAAO;YACtC,WAAW,EAAE,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI;YAC/E,KAAK,EAAE,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI;YAC3D,UAAU,EAAE,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI;SAC3E,CAAC,CAAC;QACH,IAAI,cAAc,EAAE,CAAC;YACnB,kBAAkB,GAAG,iBAAe,CAAC,mBAAmB,CACtD,MAAM,CAAC,IAAI,IAAI,EAAE,EACjB,YAAY,EACZ,mBAAmB,CACpB,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;YACjF,iBAAiB,GAAG,iBAAe,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YAChF,IAAI,sBAAsB,EAAE,CAAC;gBAC3B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,sBAAsB,CAAC,CAAC;YAC3F,CAAC;QACH,CAAC;QACD,IAAI,mBAAmB,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,mBAAmB,CAAC,CAAC;QACrF,CAAC;QAED,IAAI,CAAC,gBAAe,CAAC,aAAa,EAAE,CAAC;YACnC,MAAM,UAAU,GAAe;gBAC7B,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;gBACpF,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,OAAO;gBAClD,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,aAAa;gBACzD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,eAAe,CAAC,aAAa;gBAClE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,eAAe,CAAC,UAAU;gBAC5D,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,EAAE,CAAC,YAAY,CAAC;gBAClE,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,UAAU;gBACnD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,WAAW;gBAC3D,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,YAAY;gBAC7D,GAAG,aAAa;aACjB,CAAC;YACF,IAAI,UAAU,EAAE,CAAC;gBACf,UAAU,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,GAAG,UAAU,CAAC;YACjE,CAAC;YACD,IAAI,cAAc,EAAE,CAAC;gBACnB,IAAI,iBAAiB;oBAAE,UAAU,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,GAAG,iBAAiB,CAAC;gBAChG,IAAI,kBAAkB;oBAAE,UAAU,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,GAAG,kBAAkB,CAAC;gBACnG,IAAI,sBAAsB;oBAAE,UAAU,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,GAAG,sBAAsB,CAAC;YACjH,CAAC;YACD,IAAI,mBAAmB;gBAAE,UAAU,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,GAAG,mBAAmB,CAAC;YACtG,iBAAe,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;QAED,OAAO;YACL,aAAa;YACb,KAAK,EAAE,OAAO;YACd,IAAI;YACJ,QAAQ,EAAE,eAAe,CAAC,QAAQ;YAClC,aAAa,EAAE,eAAe,CAAC,aAAa;YAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;SACvC,CAAC;IACJ,CAAC;;AAhjBM,wBAAQ,GAAG,6BAAkB,CAAC,uBAAuB,CAAC;AACtD,6BAAa,GAAG,WAAW,CAAC;AAC5B,0BAAU,GAAG,GAAG,CAAC;AAijB1B,kBAAe,eAAe,CAAC"}

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

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

package/dist/instrumentation/vertexai/index.js

@@ -0,0 +1,70 @@
+"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 Shield360VertexAIInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-vertexai`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@google-cloud/vertexai', ['>=0.1.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(vertexAI) {
+        this._patch(vertexAI);
+    }
+    _patch(moduleExports) {
+        try {
+            const GenerativeModel = moduleExports.GenerativeModel || moduleExports.GenerativeModelPreview;
+            const ChatSession = moduleExports.ChatSession || moduleExports.ChatSessionPreview;
+            if (GenerativeModel) {
+                if ((0, instrumentation_1.isWrapped)(GenerativeModel.prototype.generateContent)) {
+                    this._unwrap(GenerativeModel.prototype, 'generateContent');
+                }
+                if ((0, instrumentation_1.isWrapped)(GenerativeModel.prototype.generateContentStream)) {
+                    this._unwrap(GenerativeModel.prototype, 'generateContentStream');
+                }
+                this._wrap(GenerativeModel.prototype, 'generateContent', wrapper_1.default._patchGenerateContent(this.tracer));
+                this._wrap(GenerativeModel.prototype, 'generateContentStream', wrapper_1.default._patchGenerateContentStream(this.tracer));
+            }
+            if (ChatSession) {
+                if ((0, instrumentation_1.isWrapped)(ChatSession.prototype.sendMessage)) {
+                    this._unwrap(ChatSession.prototype, 'sendMessage');
+                }
+                if ((0, instrumentation_1.isWrapped)(ChatSession.prototype.sendMessageStream)) {
+                    this._unwrap(ChatSession.prototype, 'sendMessageStream');
+                }
+                this._wrap(ChatSession.prototype, 'sendMessage', wrapper_1.default._patchSendMessage(this.tracer));
+                this._wrap(ChatSession.prototype, 'sendMessageStream', wrapper_1.default._patchSendMessageStream(this.tracer));
+            }
+        }
+        catch (e) {
+            console.error('Error in _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        const GenerativeModel = moduleExports.GenerativeModel || moduleExports.GenerativeModelPreview;
+        const ChatSession = moduleExports.ChatSession || moduleExports.ChatSessionPreview;
+        if (GenerativeModel) {
+            this._unwrap(GenerativeModel.prototype, 'generateContent');
+            this._unwrap(GenerativeModel.prototype, 'generateContentStream');
+        }
+        if (ChatSession) {
+            this._unwrap(ChatSession.prototype, 'sendMessage');
+            this._unwrap(ChatSession.prototype, 'sendMessageStream');
+        }
+    }
+}
+exports.default = Shield360VertexAIInstrumentation;
+//# sourceMappingURL=index.js.map