shield360 1.0.0

package/dist/instrumentation/transformers/wrapper.js

@@ -0,0 +1,472 @@
+"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"));
+const TASK_OPERATION = {
+    // Generative chat-style (Python parity)
+    'text-generation': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, isEmbedding: false },
+    // Sequence-to-sequence / extractive text producers
+    'text2text-generation': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    summarization: { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    translation: { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    'fill-mask': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    'question-answering': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    'text-classification': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    'token-classification': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    'zero-shot-classification': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION, isEmbedding: false },
+    // Embedding producers
+    'feature-extraction': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING, isEmbedding: true },
+    'sentence-similarity': { operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING, isEmbedding: true },
+};
+/** Map a Pipeline subclass name to its canonical Transformers.js task string. */
+const CLASS_TASK = {
+    TextGenerationPipeline: 'text-generation',
+    Text2TextGenerationPipeline: 'text2text-generation',
+    SummarizationPipeline: 'summarization',
+    TranslationPipeline: 'translation',
+    FillMaskPipeline: 'fill-mask',
+    QuestionAnsweringPipeline: 'question-answering',
+    TextClassificationPipeline: 'text-classification',
+    TokenClassificationPipeline: 'token-classification',
+    ZeroShotClassificationPipeline: 'zero-shot-classification',
+    FeatureExtractionPipeline: 'feature-extraction',
+};
+function resolveTask(instance, className) {
+    return ((typeof instance?.task === 'string' && instance.task) ||
+        CLASS_TASK[className] ||
+        'text-generation');
+}
+function classifyTask(task) {
+    return (TASK_OPERATION[task] || {
+        operation: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION,
+        isEmbedding: false,
+    });
+}
+/**
+ * Resolve the model identifier from a pipeline instance, mirroring Python's
+ * `instance.model.config.name_or_path`. HF configs expose the path under a few
+ * keys across versions, so we probe the common ones before falling back.
+ */
+function resolveModel(instance) {
+    const config = instance?.model?.config ?? {};
+    return (config._name_or_path ||
+        config.name_or_path ||
+        config.model_type ||
+        instance?.model?.name_or_path ||
+        (typeof instance?.task === 'string' ? instance.task : '') ||
+        'unknown');
+}
+/**
+ * Extract the generation parameters for a call, mirroring Python which reads
+ * `instance._forward_params` (set at pipeline construction) merged with the
+ * call-time options object.
+ */
+function resolveGenerationParams(instance, options) {
+    const forward = instance?._forward_params ?? {};
+    const opts = options ?? {};
+    const pick = (key, altKey) => opts[key] ?? (altKey ? opts[altKey] : undefined) ?? forward[key] ?? (altKey ? forward[altKey] : undefined) ?? null;
+    return {
+        temperature: pick('temperature'),
+        topK: pick('top_k'),
+        topP: pick('top_p'),
+        maxTokens: pick('max_new_tokens', 'max_length'),
+    };
+}
+/**
+ * Convert a Transformers.js pipeline result into a flat text string per task,
+ * mirroring the task branches in Python's `process_chat_response`.
+ */
+function extractCompletion(task, response) {
+    const first = Array.isArray(response) ? response[0] : response;
+    const fromEntry = (entry) => {
+        if (entry === null || entry === undefined)
+            return '';
+        if (typeof entry !== 'object')
+            return String(entry);
+        // text-generation may nest a chat-message list under generated_text
+        if (Array.isArray(entry.generated_text)) {
+            const last = entry.generated_text[entry.generated_text.length - 1];
+            return last?.content ?? String(last ?? '');
+        }
+        return (entry.generated_text ??
+            entry.summary_text ??
+            entry.translation_text ??
+            entry.answer ??
+            entry.sequence ??
+            entry.token_str ??
+            entry.label ??
+            entry.text ??
+            '');
+    };
+    switch (task) {
+        case 'automatic-speech-recognition':
+            return typeof response === 'object' && response !== null ? response.text ?? '' : '';
+        case 'feature-extraction':
+        case 'sentence-similarity':
+            return '';
+        default: {
+            const out = fromEntry(first);
+            if (out)
+                return out;
+            // Fall back to a stable serialization for unknown shapes.
+            try {
+                return typeof response === 'string' ? response : JSON.stringify(response);
+            }
+            catch {
+                return String(response ?? '');
+            }
+        }
+    }
+}
+function stringifyInputs(inputs) {
+    if (typeof inputs === 'string')
+        return inputs;
+    if (Array.isArray(inputs)) {
+        return inputs
+            .map((i) => (typeof i === 'string' ? i : i?.content ?? JSON.stringify(i)))
+            .join('\n');
+    }
+    if (inputs && typeof inputs === 'object') {
+        // question-answering style { question, context }
+        if (typeof inputs.question === 'string') {
+            return inputs.context
+                ? `question: ${inputs.question} context: ${inputs.context}`
+                : inputs.question;
+        }
+        try {
+            return JSON.stringify(inputs);
+        }
+        catch {
+            return String(inputs);
+        }
+    }
+    return String(inputs ?? '');
+}
+class TransformersWrapper extends base_wrapper_1.default {
+    /**
+     * Stamp `shield360.agent.version_hash` (auto) and `gen_ai.agent.version`
+     * (user override) on the span and return them 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: 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_HUGGING_FACE,
+                },
+                providers: [semantic_convention_1.default.GEN_AI_SYSTEM_HUGGING_FACE],
+            });
+            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;
+    }
+    /**
+     * Patch a Pipeline subclass `_call` (the method invoked when the pipeline
+     * object is used as a function). `this` is the pipeline instance.
+     * args[0] = inputs, args[1] = generation options.
+     */
+    static _patchPipelineCall(tracer, className, sdkVersion) {
+        return (originalMethod) => {
+            return async function (...args) {
+                if ((0, helpers_1.isFrameworkLlmActive)())
+                    return originalMethod.apply(this, args);
+                const task = resolveTask(this, className);
+                const requestModel = resolveModel(this);
+                const { operation } = classifyTask(task);
+                const genAIEndpoint = `transformers.${task}`;
+                const spanName = `${operation} ${requestModel}`;
+                const effectiveCtx = (0, helpers_1.getFrameworkParentContext)() ?? api_1.context.active();
+                const span = tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT, attributes: spanCreationAttrs(operation, requestModel) }, effectiveCtx);
+                return api_1.context.with(api_1.trace.setSpan(effectiveCtx, span), async () => {
+                    let metricParams;
+                    const startTime = Date.now();
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        metricParams = TransformersWrapper._handleResponse({
+                            instance: this,
+                            args,
+                            response,
+                            span,
+                            requestModel,
+                            task,
+                            operation,
+                            genAIEndpoint,
+                            ttft: (Date.now() - startTime) / 1000,
+                            sdkVersion,
+                        });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        base_wrapper_1.default.recordMetrics(span, {
+                            genAIEndpoint,
+                            model: requestModel,
+                            aiSystem: TransformersWrapper.aiSystem,
+                            serverAddress: TransformersWrapper.serverAddress,
+                            serverPort: TransformersWrapper.serverPort,
+                            errorType: e?.constructor?.name || '_OTHER',
+                        });
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams) {
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                        }
+                    }
+                });
+            };
+        };
+    }
+    /**
+     * Patch the `pipeline()` factory as a fallback when no Pipeline subclass
+     * prototype is exported. Wraps the returned callable so each invocation
+     * emits a span. The original callable is invoked directly (not via the
+     * wrapper) so we never lose its prototype behavior.
+     */
+    static _patchPipelineFactory(tracer, sdkVersion) {
+        return (originalFactory) => {
+            return async function (...factoryArgs) {
+                const pipe = await originalFactory.apply(this, factoryArgs);
+                if (typeof pipe !== 'function')
+                    return pipe;
+                const task = (typeof factoryArgs[0] === 'string' && factoryArgs[0]) ||
+                    (typeof pipe.task === 'string' && pipe.task) ||
+                    'text-generation';
+                const { operation } = classifyTask(task);
+                const genAIEndpoint = `transformers.${task}`;
+                const wrappedPipe = async function (...callArgs) {
+                    if ((0, helpers_1.isFrameworkLlmActive)())
+                        return pipe.apply(this, callArgs);
+                    const requestModel = resolveModel(pipe) ||
+                        (typeof factoryArgs[1] === 'string' ? factoryArgs[1] : 'unknown');
+                    const spanName = `${operation} ${requestModel}`;
+                    const effectiveCtx = (0, helpers_1.getFrameworkParentContext)() ?? api_1.context.active();
+                    const span = tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT, attributes: spanCreationAttrs(operation, requestModel) }, effectiveCtx);
+                    return api_1.context.with(api_1.trace.setSpan(effectiveCtx, span), async () => {
+                        let metricParams;
+                        const startTime = Date.now();
+                        try {
+                            const response = await pipe.apply(this, callArgs);
+                            metricParams = TransformersWrapper._handleResponse({
+                                instance: pipe,
+                                args: callArgs,
+                                response,
+                                span,
+                                requestModel,
+                                task,
+                                operation,
+                                genAIEndpoint,
+                                ttft: (Date.now() - startTime) / 1000,
+                                sdkVersion,
+                            });
+                            return response;
+                        }
+                        catch (e) {
+                            helpers_1.default.handleException(span, e);
+                            base_wrapper_1.default.recordMetrics(span, {
+                                genAIEndpoint,
+                                model: requestModel,
+                                aiSystem: TransformersWrapper.aiSystem,
+                                serverAddress: TransformersWrapper.serverAddress,
+                                serverPort: TransformersWrapper.serverPort,
+                                errorType: e?.constructor?.name || '_OTHER',
+                            });
+                            throw e;
+                        }
+                        finally {
+                            span.end();
+                            if (metricParams) {
+                                base_wrapper_1.default.recordMetrics(span, metricParams);
+                            }
+                        }
+                    });
+                };
+                // Preserve callable identity so the wrapped pipe behaves like the original.
+                Object.setPrototypeOf(wrappedPipe, Object.getPrototypeOf(pipe));
+                Object.assign(wrappedPipe, pipe);
+                return wrappedPipe;
+            };
+        };
+    }
+    /**
+     * Synchronous attribute setter shared by the class- and factory-patch paths.
+     * Returns the metric params so the caller can record metrics in `finally`.
+     */
+    static _handleResponse({ instance, args, response, span, requestModel, task, operation, genAIEndpoint, ttft, sdkVersion, }) {
+        const captureContent = config_1.default.captureMessageContent;
+        const { isEmbedding } = classifyTask(task);
+        const inputs = args[0];
+        const options = args[1] || {};
+        const { temperature, topK, topP, maxTokens } = resolveGenerationParams(instance, options);
+        const inputStr = stringifyInputs(inputs);
+        const completion = extractCompletion(task, response);
+        const inputTokens = helpers_1.default.generalTokens(inputStr) ?? 0;
+        const outputTokens = isEmbedding ? 0 : helpers_1.default.generalTokens(completion) ?? 0;
+        const pricingInfo = config_1.default.pricingInfo || {};
+        const cost = helpers_1.default.getChatModelCost(requestModel, pricingInfo, inputTokens, outputTokens);
+        // Common attributes (telemetry sdk, env, app, request model, cost, server).
+        TransformersWrapper.setBaseSpanAttributes(span, {
+            genAIEndpoint,
+            model: requestModel,
+            cost,
+            aiSystem: TransformersWrapper.aiSystem,
+            serverAddress: TransformersWrapper.serverAddress,
+            serverPort: TransformersWrapper.serverPort,
+        });
+        // Parity: Python stamps gen_ai.system and the transformers package version
+        // (setBaseSpanAttributes uses Shield360's SDK version).
+        span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, TransformersWrapper.aiSystem);
+        if (sdkVersion) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, sdkVersion);
+        }
+        // Request parameters (only when present, matching Python).
+        if (temperature !== null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, temperature);
+        }
+        if (topK !== null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K, topK);
+        }
+        if (topP !== null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, topP);
+        }
+        if (maxTokens !== null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, maxTokens);
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+        // Response parameters.
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, requestModel);
+        if (!isEmbedding) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+        }
+        // Tokens, cost, cache (cache stamped as 0 even when unused, like Python).
+        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_CLIENT_TOKEN_USAGE, inputTokens + outputTokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, 0);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, 0);
+        // Timing (Python always sets these; tbt is 0 for non-streaming pipelines).
+        span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TBT, 0);
+        const versionExtras = TransformersWrapper._stampAgentVersion(span, {
+            systemInstructionsJson: null,
+            primaryModel: requestModel,
+            temperature,
+            top_p: topP,
+            max_tokens: maxTokens,
+        });
+        let inputMessagesJson;
+        let outputMessagesJson;
+        if (captureContent) {
+            inputMessagesJson = helpers_1.default.buildInputMessages([{ role: 'user', content: inputStr }]);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, inputMessagesJson);
+            if (!isEmbedding) {
+                outputMessagesJson = helpers_1.default.buildOutputMessages(completion, 'stop');
+                span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
+            }
+        }
+        if (!config_1.default.disableEvents) {
+            const eventAttrs = {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: operation,
+                [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: requestModel,
+                [semantic_convention_1.default.SERVER_ADDRESS]: TransformersWrapper.serverAddress,
+                [semantic_convention_1.default.SERVER_PORT]: TransformersWrapper.serverPort,
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: inputTokens,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: outputTokens,
+                ...versionExtras,
+            };
+            if (!isEmbedding) {
+                eventAttrs[semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON] = ['stop'];
+                eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_TYPE] = semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT;
+            }
+            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;
+            }
+            helpers_1.default.emitInferenceEvent(span, eventAttrs);
+        }
+        return {
+            genAIEndpoint,
+            model: requestModel,
+            cost,
+            aiSystem: TransformersWrapper.aiSystem,
+            serverAddress: TransformersWrapper.serverAddress,
+            serverPort: TransformersWrapper.serverPort,
+        };
+    }
+}
+TransformersWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_HUGGING_FACE;
+TransformersWrapper.serverAddress = '127.0.0.1';
+TransformersWrapper.serverPort = 80;
+function spanCreationAttrs(operationName, requestModel) {
+    return {
+        [semantic_convention_1.default.GEN_AI_OPERATION]: operationName,
+        [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_HUGGING_FACE,
+        [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+        [semantic_convention_1.default.SERVER_ADDRESS]: TransformersWrapper.serverAddress,
+        [semantic_convention_1.default.SERVER_PORT]: TransformersWrapper.serverPort,
+    };
+}
+exports.default = TransformersWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/transformers/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAwF;AACxF,0DAA2C;AAC3C,yDAIuB;AACvB,oFAA2D;AAC3D,mEAAkE;AAelE,MAAM,cAAc,GAA8B;IAChD,wCAAwC;IACxC,iBAAiB,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,0BAA0B,EAAE,WAAW,EAAE,KAAK,EAAE;IACnG,mDAAmD;IACnD,sBAAsB,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IACnH,aAAa,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IAC1G,WAAW,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IACxG,WAAW,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IACxG,oBAAoB,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IACjH,qBAAqB,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IAClH,sBAAsB,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IACnH,0BAA0B,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,qCAAqC,EAAE,WAAW,EAAE,KAAK,EAAE;IACvH,sBAAsB;IACtB,oBAAoB,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,+BAA+B,EAAE,WAAW,EAAE,IAAI,EAAE;IAC1G,qBAAqB,EAAE,EAAE,SAAS,EAAE,6BAAkB,CAAC,+BAA+B,EAAE,WAAW,EAAE,IAAI,EAAE;CAC5G,CAAC;AAEF,iFAAiF;AACjF,MAAM,UAAU,GAA2B;IACzC,sBAAsB,EAAE,iBAAiB;IACzC,2BAA2B,EAAE,sBAAsB;IACnD,qBAAqB,EAAE,eAAe;IACtC,mBAAmB,EAAE,aAAa;IAClC,gBAAgB,EAAE,WAAW;IAC7B,yBAAyB,EAAE,oBAAoB;IAC/C,0BAA0B,EAAE,qBAAqB;IACjD,2BAA2B,EAAE,sBAAsB;IACnD,8BAA8B,EAAE,0BAA0B;IAC1D,yBAAyB,EAAE,oBAAoB;CAChD,CAAC;AAEF,SAAS,WAAW,CAAC,QAAa,EAAE,SAAiB;IACnD,OAAO,CACL,CAAC,OAAO,QAAQ,EAAE,IAAI,KAAK,QAAQ,IAAI,QAAQ,CAAC,IAAI,CAAC;QACrD,UAAU,CAAC,SAAS,CAAC;QACrB,iBAAiB,CAClB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,IAAY;IAChC,OAAO,CACL,cAAc,CAAC,IAAI,CAAC,IAAI;QACtB,SAAS,EAAE,6BAAkB,CAAC,qCAAqC;QACnE,WAAW,EAAE,KAAK;KACnB,CACF,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,YAAY,CAAC,QAAa;IACjC,MAAM,MAAM,GAAG,QAAQ,EAAE,KAAK,EAAE,MAAM,IAAI,EAAE,CAAC;IAC7C,OAAO,CACL,MAAM,CAAC,aAAa;QACpB,MAAM,CAAC,YAAY;QACnB,MAAM,CAAC,UAAU;QACjB,QAAQ,EAAE,KAAK,EAAE,YAAY;QAC7B,CAAC,OAAO,QAAQ,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,SAAS,CACV,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,uBAAuB,CAAC,QAAa,EAAE,OAAY;IAM1D,MAAM,OAAO,GAAG,QAAQ,EAAE,eAAe,IAAI,EAAE,CAAC;IAChD,MAAM,IAAI,GAAG,OAAO,IAAI,EAAE,CAAC;IAC3B,MAAM,IAAI,GAAG,CAAC,GAAW,EAAE,MAAe,EAAE,EAAE,CAC5C,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC;IAErH,OAAO;QACL,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC;QAChC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC;QACnB,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC;QACnB,SAAS,EAAE,IAAI,CAAC,gBAAgB,EAAE,YAAY,CAAC;KAChD,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAS,iBAAiB,CAAC,IAAY,EAAE,QAAa;IACpD,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAE/D,MAAM,SAAS,GAAG,CAAC,KAAU,EAAU,EAAE;QACvC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,EAAE,CAAC;QACrD,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACpD,oEAAoE;QACpE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACnE,OAAO,IAAI,EAAE,OAAO,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,CACL,KAAK,CAAC,cAAc;YACpB,KAAK,CAAC,YAAY;YAClB,KAAK,CAAC,gBAAgB;YACtB,KAAK,CAAC,MAAM;YACZ,KAAK,CAAC,QAAQ;YACd,KAAK,CAAC,SAAS;YACf,KAAK,CAAC,KAAK;YACX,KAAK,CAAC,IAAI;YACV,EAAE,CACH,CAAC;IACJ,CAAC,CAAC;IAEF,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,8BAA8B;YACjC,OAAO,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACtF,KAAK,oBAAoB,CAAC;QAC1B,KAAK,qBAAqB;YACxB,OAAO,EAAE,CAAC;QACZ,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;YAC7B,IAAI,GAAG;gBAAE,OAAO,GAAG,CAAC;YACpB,0DAA0D;YAC1D,IAAI,CAAC;gBACH,OAAO,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC5E,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,MAAW;IAClC,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC9C,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM;aACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;aACzE,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IACD,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QACzC,iDAAiD;QACjD,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,MAAM,CAAC,OAAO;gBACnB,CAAC,CAAC,aAAa,MAAM,CAAC,QAAQ,aAAa,MAAM,CAAC,OAAO,EAAE;gBAC3D,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;QACtB,CAAC;QACD,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,mBAAoB,SAAQ,sBAAW;IAK3C;;;;OAIG;IACH,MAAM,CAAC,kBAAkB,CACvB,IAAU,EACV,IAMC;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;gBACrB,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,0BAA0B;iBACxD;gBACD,SAAS,EAAE,CAAC,6BAAkB,CAAC,0BAA0B,CAAC;aAC3D,CAAC,CAAC;YACH,IAAI,WAAW,EAAE,CAAC;gBAChB,GAAG,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,GAAG,WAAW,CAAC;gBACnE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,WAAW,CAAC,CAAC;YAClF,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,wDAAwD;QAC1D,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;;;;OAIG;IACH,MAAM,CAAC,kBAAkB,CAAC,MAAc,EAAE,SAAiB,EAAE,UAAmB;QAC9E,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,IAAI,IAAA,8BAAoB,GAAE;oBAAE,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAEpE,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBAC1C,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;gBACxC,MAAM,EAAE,SAAS,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;gBACzC,MAAM,aAAa,GAAG,gBAAgB,IAAI,EAAE,CAAC;gBAC7C,MAAM,QAAQ,GAAG,GAAG,SAAS,IAAI,YAAY,EAAE,CAAC;gBAChD,MAAM,YAAY,GAAG,IAAA,mCAAyB,GAAE,IAAI,aAAO,CAAC,MAAM,EAAE,CAAC;gBACrE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAC3B,QAAQ,EACR,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,UAAU,EAAE,iBAAiB,CAAC,SAAS,EAAE,YAAY,CAAC,EAAE,EACjF,YAAY,CACb,CAAC;gBAEF,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBAChE,IAAI,YAA4C,CAAC;oBACjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,YAAY,GAAG,mBAAmB,CAAC,eAAe,CAAC;4BACjD,QAAQ,EAAE,IAAI;4BACd,IAAI;4BACJ,QAAQ;4BACR,IAAI;4BACJ,YAAY;4BACZ,IAAI;4BACJ,SAAS;4BACT,aAAa;4BACb,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI;4BACrC,UAAU;yBACX,CAAC,CAAC;wBACH,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;4BAC9B,aAAa;4BACb,KAAK,EAAE,YAAY;4BACnB,QAAQ,EAAE,mBAAmB,CAAC,QAAQ;4BACtC,aAAa,EAAE,mBAAmB,CAAC,aAAa;4BAChD,UAAU,EAAE,mBAAmB,CAAC,UAAU;4BAC1C,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;yBAC5C,CAAC,CAAC;wBACH,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAI,YAAY,EAAE,CAAC;4BACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;wBAChD,CAAC;oBACH,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,qBAAqB,CAAC,MAAc,EAAE,UAAmB;QAC9D,OAAO,CAAC,eAAwC,EAAE,EAAE;YAClD,OAAO,KAAK,WAAsB,GAAG,WAAkB;gBACrD,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,KAAK,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBAC5D,IAAI,OAAO,IAAI,KAAK,UAAU;oBAAE,OAAO,IAAI,CAAC;gBAE5C,MAAM,IAAI,GACR,CAAC,OAAO,WAAW,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;oBACtD,CAAC,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC;oBAC5C,iBAAiB,CAAC;gBACpB,MAAM,EAAE,SAAS,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;gBACzC,MAAM,aAAa,GAAG,gBAAgB,IAAI,EAAE,CAAC;gBAE7C,MAAM,WAAW,GAAQ,KAAK,WAAsB,GAAG,QAAe;oBACpE,IAAI,IAAA,8BAAoB,GAAE;wBAAE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;oBAE9D,MAAM,YAAY,GAChB,YAAY,CAAC,IAAI,CAAC;wBAClB,CAAC,OAAO,WAAW,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBACpE,MAAM,QAAQ,GAAG,GAAG,SAAS,IAAI,YAAY,EAAE,CAAC;oBAChD,MAAM,YAAY,GAAG,IAAA,mCAAyB,GAAE,IAAI,aAAO,CAAC,MAAM,EAAE,CAAC;oBACrE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAC3B,QAAQ,EACR,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,UAAU,EAAE,iBAAiB,CAAC,SAAS,EAAE,YAAY,CAAC,EAAE,EACjF,YAAY,CACb,CAAC;oBAEF,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;wBAChE,IAAI,YAA4C,CAAC;wBACjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAC7B,IAAI,CAAC;4BACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;4BAClD,YAAY,GAAG,mBAAmB,CAAC,eAAe,CAAC;gCACjD,QAAQ,EAAE,IAAI;gCACd,IAAI,EAAE,QAAQ;gCACd,QAAQ;gCACR,IAAI;gCACJ,YAAY;gCACZ,IAAI;gCACJ,SAAS;gCACT,aAAa;gCACb,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI;gCACrC,UAAU;6BACX,CAAC,CAAC;4BACH,OAAO,QAAQ,CAAC;wBAClB,CAAC;wBAAC,OAAO,CAAM,EAAE,CAAC;4BAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;4BACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;gCAC9B,aAAa;gCACb,KAAK,EAAE,YAAY;gCACnB,QAAQ,EAAE,mBAAmB,CAAC,QAAQ;gCACtC,aAAa,EAAE,mBAAmB,CAAC,aAAa;gCAChD,UAAU,EAAE,mBAAmB,CAAC,UAAU;gCAC1C,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;6BAC5C,CAAC,CAAC;4BACH,MAAM,CAAC,CAAC;wBACV,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,CAAC;gBACL,CAAC,CAAC;gBAEF,4EAA4E;gBAC5E,MAAM,CAAC,cAAc,CAAC,WAAW,EAAE,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;gBAChE,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;gBACjC,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,eAAe,CAAC,EACrB,QAAQ,EACR,IAAI,EACJ,QAAQ,EACR,IAAI,EACJ,YAAY,EACZ,IAAI,EACJ,SAAS,EACT,aAAa,EACb,IAAI,EACJ,UAAU,GAYX;QACC,MAAM,cAAc,GAAG,gBAAe,CAAC,qBAAqB,CAAC;QAC7D,MAAM,EAAE,WAAW,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QAE3C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACvB,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9B,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,uBAAuB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAE1F,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,UAAU,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAErD,MAAM,WAAW,GAAG,iBAAe,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,YAAY,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAe,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAEtF,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QACtD,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAAC,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAEpG,4EAA4E;QAC5E,mBAAmB,CAAC,qBAAqB,CAAC,IAAI,EAAE;YAC9C,aAAa;YACb,KAAK,EAAE,YAAY;YACnB,IAAI;YACJ,QAAQ,EAAE,mBAAmB,CAAC,QAAQ;YACtC,aAAa,EAAE,mBAAmB,CAAC,aAAa;YAChD,UAAU,EAAE,mBAAmB,CAAC,UAAU;SAC3C,CAAC,CAAC;QAEH,2EAA2E;QAC3E,wDAAwD;QACxD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QACzF,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QACvE,CAAC;QAED,2DAA2D;QAC3D,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;YACzB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,WAAW,CAAC,CAAC;QAChF,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,IAAI,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,IAAI,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YACvB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,SAAS,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QAEtE,uBAAuB;QACvB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QAC1E,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,6BAAkB,CAAC,uBAAuB,CAAC,CAAC;QACvG,CAAC;QAED,0EAA0E;QAC1E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;QAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,YAAY,CAAC,CAAC;QAC/E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,GAAG,YAAY,CAAC,CAAC;QAC5F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAC;QAElF,2EAA2E;QAC3E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;QAC/D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;QAE3D,MAAM,aAAa,GAAG,mBAAmB,CAAC,kBAAkB,CAAC,IAAI,EAAE;YACjE,sBAAsB,EAAE,IAAI;YAC5B,YAAY,EAAE,YAAY;YAC1B,WAAW;YACX,KAAK,EAAE,IAAI;YACX,UAAU,EAAE,SAAS;SACtB,CAAC,CAAC;QAEH,IAAI,iBAAqC,CAAC;QAC1C,IAAI,kBAAsC,CAAC;QAC3C,IAAI,cAAc,EAAE,CAAC;YACnB,iBAAiB,GAAG,iBAAe,CAAC,kBAAkB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;YAC9F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,iBAAiB,CAAC,CAAC;YAC/E,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,kBAAkB,GAAG,iBAAe,CAAC,mBAAmB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;YACnF,CAAC;QACH,CAAC;QAED,IAAI,CAAC,gBAAe,CAAC,aAAa,EAAE,CAAC;YACnC,MAAM,UAAU,GAAe;gBAC7B,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,SAAS;gBAChD,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;gBACvD,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,YAAY;gBACxD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,mBAAmB,CAAC,aAAa;gBACtE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,mBAAmB,CAAC,UAAU;gBAChE,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,WAAW;gBAC3D,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,YAAY;gBAC7D,GAAG,aAAa;aACjB,CAAC;YACF,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,UAAU,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACxE,UAAU,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,GAAG,6BAAkB,CAAC,uBAAuB,CAAC;YACjG,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;YACrG,CAAC;YACD,iBAAe,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;QAED,OAAO;YACL,aAAa;YACb,KAAK,EAAE,YAAY;YACnB,IAAI;YACJ,QAAQ,EAAE,mBAAmB,CAAC,QAAQ;YACtC,aAAa,EAAE,mBAAmB,CAAC,aAAa;YAChD,UAAU,EAAE,mBAAmB,CAAC,UAAU;SAC3C,CAAC;IACJ,CAAC;;AA3UM,4BAAQ,GAAG,6BAAkB,CAAC,0BAA0B,CAAC;AACzD,iCAAa,GAAG,WAAW,CAAC;AAC5B,8BAAU,GAAG,EAAE,CAAC;AA4UzB,SAAS,iBAAiB,CAAC,aAAqB,EAAE,YAAoB;IACpE,OAAO;QACL,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;QACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;QAC7F,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;QACvD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,mBAAmB,CAAC,aAAa;QACtE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,mBAAmB,CAAC,UAAU;KACjE,CAAC;AACJ,CAAC;AAED,kBAAe,mBAAmB,CAAC"}

package/dist/instrumentation/vercel-ai/index.d.ts

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

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

@@ -0,0 +1,59 @@
+"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"));
+// Functions to intercept from the `ai` module
+const PATCHED_FNS = ['generateText', 'streamText', 'generateObject', 'embed'];
+class Shield360VercelAIInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-vercel-ai`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('ai', ['>=3.0.0 <5'], (moduleExports) => this._patch(moduleExports), 
+        // No-op unpatch: Proxy is discarded when the module is re-required
+        () => { });
+        return [module];
+    }
+    manualPatch(ai) {
+        return this._patch(ai);
+    }
+    _patch(moduleExports) {
+        try {
+            const tracer = this.tracer;
+            // The `ai` package exports functions as non-configurable getter properties,
+            // so shimmer's Object.defineProperty-based wrapping fails. Instead, return a
+            // Proxy that intercepts property access and returns wrapped functions.
+            const patchers = {
+                generateText: wrapper_1.default._patchGenerateText,
+                streamText: wrapper_1.default._patchStreamText,
+                generateObject: wrapper_1.default._patchGenerateObject,
+                embed: wrapper_1.default._patchEmbed,
+            };
+            // Pre-build wrapped functions once (avoid re-wrapping on every get)
+            const wrapped = {};
+            for (const name of PATCHED_FNS) {
+                const original = moduleExports[name];
+                if (typeof original === 'function') {
+                    wrapped[name] = patchers[name](tracer)(original);
+                }
+            }
+            return new Proxy(moduleExports, {
+                get(target, prop, receiver) {
+                    if (prop in wrapped)
+                        return wrapped[prop];
+                    return Reflect.get(target, prop, receiver);
+                },
+            });
+        }
+        catch (e) {
+            console.error('Error in VercelAI _patch method:', e);
+            return moduleExports;
+        }
+    }
+}
+exports.default = Shield360VercelAIInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/vercel-ai/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAIwC;AAExC,6CAAwD;AACxD,wDAAwC;AAIxC,8CAA8C;AAC9C,MAAM,WAAW,GAAG,CAAC,cAAc,EAAE,YAAY,EAAE,gBAAgB,EAAE,OAAO,CAAU,CAAC;AAEvF,MAAqB,gCAAiC,SAAQ,qCAAmB;IAC/E,YAAY,SAAwC,EAAE;QACpD,KAAK,CAAC,GAAG,iCAAsB,4BAA4B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,IAAI,EACJ,CAAC,YAAY,CAAC,EACd,CAAC,aAAa,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;QAC7C,mEAAmE;QACnE,GAAG,EAAE,GAAE,CAAC,CACT,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,EAAO;QACxB,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAE3B,4EAA4E;YAC5E,6EAA6E;YAC7E,uEAAuE;YACvE,MAAM,QAAQ,GAA6D;gBACzE,YAAY,EAAE,iBAAe,CAAC,kBAAkB;gBAChD,UAAU,EAAE,iBAAe,CAAC,gBAAgB;gBAC5C,cAAc,EAAE,iBAAe,CAAC,oBAAoB;gBACpD,KAAK,EAAE,iBAAe,CAAC,WAAW;aACnC,CAAC;YAEF,oEAAoE;YACpE,MAAM,OAAO,GAAwB,EAAE,CAAC;YACxC,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;gBAC/B,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;gBACrC,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;oBACnC,OAAO,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;YAED,OAAO,IAAI,KAAK,CAAC,aAAa,EAAE;gBAC9B,GAAG,CAAC,MAAM,EAAE,IAAY,EAAE,QAAQ;oBAChC,IAAI,IAAI,IAAI,OAAO;wBAAE,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;oBAC1C,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAC7C,CAAC;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;YACrD,OAAO,aAAa,CAAC;QACvB,CAAC;IACH,CAAC;CACF;AAtDD,mDAsDC"}

package/dist/instrumentation/vercel-ai/wrapper.d.ts

@@ -0,0 +1,50 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class VercelAIWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    /**
+     * 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: Span, args: {
+        systemInstructionsJson?: string;
+        toolDefinitionsJson?: string;
+        primaryModel?: string;
+        temperature?: number | null;
+        top_p?: number | null;
+        max_tokens?: number | null;
+    }): Record<string, string>;
+    static _patchGenerateText(tracer: Tracer): any;
+    static _patchStreamText(tracer: Tracer): any;
+    static _patchGenerateObject(tracer: Tracer): any;
+    static _patchEmbed(tracer: Tracer): any;
+    static _chatComplete({ args, genAIEndpoint, response, span, outputType, resultOverride, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+        outputType: string;
+        resultOverride?: any;
+    }): Promise<any>;
+    static _chatCommonSetter({ args, genAIEndpoint, result, span, isStream, outputType, ttft, tbt, }: {
+        args: any[];
+        genAIEndpoint: string;
+        result: any;
+        span: Span;
+        isStream: boolean;
+        outputType: string;
+        ttft?: number;
+        tbt?: number;
+    }): Promise<{
+        genAIEndpoint: string;
+        model: any;
+        cost: number;
+        aiSystem: string;
+        serverAddress: string;
+        serverPort: number;
+    }>;
+}
+export default VercelAIWrapper;