shield360 1.0.0

package/dist/instrumentation/azure-ai-inference/wrapper.js

@@ -0,0 +1,580 @@
+"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, serverAddress, serverPort) {
+    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_AZURE_AI_INFERENCE,
+        [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+        [semantic_convention_1.default.SERVER_ADDRESS]: serverAddress,
+        [semantic_convention_1.default.SERVER_PORT]: serverPort,
+    };
+}
+class AzureAIInferenceWrapper 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_AZURE_AI_INFERENCE,
+                },
+                providers: [semantic_convention_1.default.GEN_AI_SYSTEM_AZURE_AI_INFERENCE],
+            });
+            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;
+    }
+    /**
+     * Extracts server address and port from an endpoint URL string.
+     */
+    static parseEndpoint(endpoint) {
+        let serverAddress = AzureAIInferenceWrapper.defaultServerAddress;
+        let serverPort = AzureAIInferenceWrapper.defaultServerPort;
+        try {
+            const url = new URL(endpoint);
+            serverAddress = url.hostname;
+            serverPort = parseInt(url.port) || (url.protocol === 'https:' ? 443 : 80);
+        }
+        catch { /* use defaults */ }
+        return { serverAddress, serverPort };
+    }
+    // ──────────────────── Chat Completions ────────────────────
+    static _patchChatComplete(tracer, serverAddress, serverPort) {
+        const genAIEndpoint = 'az.ai.inference.chat.completions';
+        return (originalMethod) => {
+            return function (...args) {
+                if ((0, helpers_1.isFrameworkLlmActive)())
+                    return originalMethod.apply(this, args);
+                const body = args[0]?.body || {};
+                const requestModel = body.model || 'gpt-4o';
+                const isStream = body.stream === true;
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT} ${requestModel}`;
+                const effectiveCtx = (0, helpers_1.getFrameworkParentContext)() ?? api_1.context.active();
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, requestModel, serverAddress, serverPort),
+                }, effectiveCtx);
+                if (isStream) {
+                    if (args[0]?.body) {
+                        args[0].body.stream_options = { include_usage: true };
+                    }
+                    const pipelineRequest = api_1.context.with(api_1.trace.setSpan(effectiveCtx, span), () => originalMethod.apply(this, args));
+                    const origAsNodeStream = pipelineRequest.asNodeStream?.bind(pipelineRequest);
+                    if (origAsNodeStream) {
+                        pipelineRequest.asNodeStream = async function (...streamArgs) {
+                            try {
+                                const streamResp = await origAsNodeStream(...streamArgs);
+                                const origBody = streamResp.body;
+                                if (!origBody) {
+                                    span.end();
+                                    return streamResp;
+                                }
+                                streamResp.body = AzureAIInferenceWrapper._wrapSseStream(origBody, body, genAIEndpoint, span, serverAddress, serverPort);
+                                return streamResp;
+                            }
+                            catch (e) {
+                                helpers_1.default.handleException(span, e);
+                                base_wrapper_1.default.recordMetrics(span, {
+                                    genAIEndpoint,
+                                    model: requestModel,
+                                    aiSystem: AzureAIInferenceWrapper.aiSystem,
+                                    serverAddress,
+                                    serverPort,
+                                    errorType: e?.constructor?.name || '_OTHER',
+                                });
+                                span.end();
+                                throw e;
+                            }
+                        };
+                    }
+                    return pipelineRequest;
+                }
+                return api_1.context
+                    .with(api_1.trace.setSpan(effectiveCtx, span), async () => {
+                    return originalMethod.apply(this, args);
+                })
+                    .then((httpResponse) => {
+                    return AzureAIInferenceWrapper._chatCompletion({
+                        body,
+                        genAIEndpoint,
+                        httpResponse,
+                        span,
+                        serverAddress,
+                        serverPort,
+                    });
+                })
+                    .catch((e) => {
+                    helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: requestModel,
+                        aiSystem: AzureAIInferenceWrapper.aiSystem,
+                        serverAddress,
+                        serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
+                    span.end();
+                    throw e;
+                });
+            };
+        };
+    }
+    static async _chatCompletion({ body, genAIEndpoint, httpResponse, span, serverAddress, serverPort, }) {
+        let metricParams;
+        try {
+            const result = httpResponse?.body ?? httpResponse;
+            if (result && typeof result === 'object' && result.choices) {
+                metricParams = AzureAIInferenceWrapper._chatCompletionCommonSetter({
+                    body,
+                    genAIEndpoint,
+                    result,
+                    span,
+                    serverAddress,
+                    serverPort,
+                });
+            }
+            return httpResponse;
+        }
+        catch (e) {
+            helpers_1.default.handleException(span, e);
+            throw e;
+        }
+        finally {
+            span.end();
+            if (metricParams) {
+                base_wrapper_1.default.recordMetrics(span, metricParams);
+            }
+        }
+    }
+    /**
+     * Wraps an SSE body stream (Node.js IncomingMessage / ReadableStream) to
+     * aggregate telemetry while passing through chunks to the caller.
+     * Returns an async-iterable that yields the raw SSE buffers/strings so
+     * downstream consumers (e.g. createSseStream) keep working.
+     */
+    static _wrapSseStream(body, requestBody, genAIEndpoint, span, serverAddress, serverPort) {
+        const requestModel = requestBody.model || 'gpt-4o';
+        const startTime = Date.now();
+        const timestamps = [];
+        const aggregated = {
+            id: '',
+            model: '',
+            content: '',
+            finishReason: 'stop',
+            inputTokens: 0,
+            outputTokens: 0,
+            toolCalls: [],
+        };
+        function processSseLine(line) {
+            if (!line.startsWith('data: '))
+                return;
+            const data = line.slice(6).trim();
+            if (data === '[DONE]')
+                return;
+            try {
+                const parsed = JSON.parse(data);
+                if (parsed.id)
+                    aggregated.id = parsed.id;
+                if (parsed.model)
+                    aggregated.model = parsed.model;
+                const choice = parsed.choices?.[0];
+                if (choice) {
+                    if (choice.delta?.content)
+                        aggregated.content += choice.delta.content;
+                    if (choice.finish_reason)
+                        aggregated.finishReason = choice.finish_reason;
+                    if (choice.delta?.tool_calls) {
+                        for (const tc of choice.delta.tool_calls) {
+                            const idx = tc.index ?? 0;
+                            while (aggregated.toolCalls.length <= idx) {
+                                aggregated.toolCalls.push({ id: '', type: 'function', function: { name: '', arguments: '' } });
+                            }
+                            if (tc.id) {
+                                aggregated.toolCalls[idx].id = tc.id;
+                                aggregated.toolCalls[idx].type = tc.type || 'function';
+                                if (tc.function?.name)
+                                    aggregated.toolCalls[idx].function.name = tc.function.name;
+                                if (tc.function?.arguments)
+                                    aggregated.toolCalls[idx].function.arguments = tc.function.arguments;
+                            }
+                            else if (tc.function?.arguments) {
+                                aggregated.toolCalls[idx].function.arguments += tc.function.arguments;
+                            }
+                        }
+                    }
+                }
+                if (parsed.usage) {
+                    aggregated.inputTokens = parsed.usage.prompt_tokens ?? 0;
+                    aggregated.outputTokens = parsed.usage.completion_tokens ?? 0;
+                }
+            }
+            catch { /* ignore parse errors */ }
+        }
+        let pending = '';
+        const readable = body;
+        const _originalPipe = readable.pipe?.bind(readable);
+        const _originalOn = readable.on?.bind(readable);
+        const self = AzureAIInferenceWrapper;
+        let finalized = false;
+        function finalize() {
+            if (finalized)
+                return;
+            finalized = true;
+            const ttft = timestamps.length > 0 ? (timestamps[0] - startTime) / 1000 : 0;
+            let tbt = 0;
+            if (timestamps.length > 1) {
+                const diffs = timestamps.slice(1).map((t, i) => t - timestamps[i]);
+                tbt = diffs.reduce((a, b) => a + b, 0) / diffs.length / 1000;
+            }
+            let inputTokens = aggregated.inputTokens;
+            let outputTokens = aggregated.outputTokens;
+            if (!inputTokens && !outputTokens) {
+                const prompt = JSON.stringify(requestBody.messages || []);
+                inputTokens = Math.ceil(prompt.length / 2);
+                outputTokens = Math.ceil(aggregated.content.length / 2);
+            }
+            const result = {
+                id: aggregated.id,
+                model: aggregated.model || requestModel,
+                choices: [{
+                        finish_reason: aggregated.finishReason,
+                        message: {
+                            role: 'assistant',
+                            content: aggregated.content,
+                            ...(aggregated.toolCalls.length > 0 ? { tool_calls: aggregated.toolCalls } : {}),
+                        },
+                    }],
+                usage: {
+                    prompt_tokens: inputTokens,
+                    completion_tokens: outputTokens,
+                },
+            };
+            const metricParams = self._chatCompletionCommonSetter({
+                body: requestBody,
+                genAIEndpoint,
+                result,
+                span,
+                serverAddress,
+                serverPort,
+                ttft,
+                tbt,
+            });
+            span.end();
+            base_wrapper_1.default.recordMetrics(span, metricParams);
+        }
+        if (typeof readable.on === 'function') {
+            readable.on('data', (chunk) => {
+                timestamps.push(Date.now());
+                const text = typeof chunk === 'string' ? chunk : chunk.toString();
+                pending += text;
+                const lines = pending.split('\n');
+                pending = lines.pop() || '';
+                for (const line of lines) {
+                    processSseLine(line.trim());
+                }
+            });
+            readable.on('end', () => {
+                if (pending.trim())
+                    processSseLine(pending.trim());
+                finalize();
+            });
+            readable.on('close', () => {
+                if (pending.trim())
+                    processSseLine(pending.trim());
+                finalize();
+            });
+            readable.on('error', (err) => {
+                helpers_1.default.handleException(span, err);
+                finalize();
+            });
+        }
+        return readable;
+    }
+    static _chatCompletionCommonSetter({ body, genAIEndpoint, result, span, serverAddress, serverPort, ttft = 0, tbt = 0, }) {
+        const captureContent = config_1.default.captureMessageContent;
+        const requestModel = body.model || 'gpt-4o';
+        const { messages, frequency_penalty = 0, max_tokens = null, n = 1, presence_penalty = 0, seed = null, stop = null, temperature = 1, top_p, user, stream = false, tools: _tools, } = body;
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, top_p || 1);
+        if (max_tokens != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, max_tokens);
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, temperature);
+        if (presence_penalty) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY, presence_penalty);
+        }
+        if (frequency_penalty) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, frequency_penalty);
+        }
+        if (seed != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_SEED, Number(seed));
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, stream);
+        if (stop) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, Array.isArray(stop) ? stop : [stop]);
+        }
+        if (n && n !== 1) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_CHOICE_COUNT, n);
+        }
+        if (captureContent) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages || []));
+        }
+        if (result.id) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, result.id);
+        }
+        const responseModel = result.model || requestModel;
+        const pricingInfo = config_1.default.pricingInfo || {};
+        const inputTokens = result.usage?.prompt_tokens || 0;
+        const outputTokens = result.usage?.completion_tokens || 0;
+        // Azure AI Inference follows OpenAI-style inclusive prompt_tokens.
+        const cacheReadTokens = Number(result.usage?.prompt_tokens_details?.cached_tokens) || 0;
+        const cacheCreationTokens = Number(result.usage?.input_tokens_details?.cache_creation_tokens) ||
+            Number(result.usage?.prompt_tokens_details?.cache_creation_tokens) ||
+            Number(result.usage?.prompt_tokens_details?.cache_write_tokens) ||
+            0;
+        const cost = helpers_1.default.getChatModelCost(requestModel, pricingInfo, inputTokens, outputTokens, cacheReadTokens, cacheCreationTokens, true);
+        AzureAIInferenceWrapper.setBaseSpanAttributes(span, {
+            genAIEndpoint,
+            model: requestModel,
+            user,
+            cost,
+            aiSystem: AzureAIInferenceWrapper.aiSystem,
+            serverAddress,
+            serverPort,
+        });
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, responseModel);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+        if (cacheReadTokens) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, cacheReadTokens);
+        }
+        if (cacheCreationTokens) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, cacheCreationTokens);
+        }
+        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);
+        }
+        const choices = result.choices || [];
+        if (choices[0]?.finish_reason) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [choices[0].finish_reason]);
+        }
+        const outputType = typeof choices[0]?.message?.content === 'string'
+            ? semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT
+            : semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_JSON;
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, outputType);
+        if (choices[0]?.message?.tool_calls) {
+            const toolCalls = choices[0].message.tool_calls;
+            const toolNames = toolCalls.map((t) => t.function?.name || '').filter(Boolean);
+            const toolIds = toolCalls.map((t) => t.id || '').filter(Boolean);
+            const toolArgs = toolCalls.map((t) => t.function?.arguments || '').filter(Boolean);
+            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(', '));
+            }
+        }
+        let inputMessagesJson;
+        let outputMessagesJson;
+        const toolDefinitionsJson = helpers_1.default.buildToolDefinitions(_tools);
+        // Compute system_instructions and version hash regardless of
+        // captureContent so versions still group correctly when content
+        // capture is disabled.
+        const systemInstructionsJson = helpers_1.default.buildSystemInstructionsFromMessages(messages || []);
+        const versionExtras = AzureAIInferenceWrapper._stampAgentVersion(span, {
+            systemInstructionsJson,
+            toolDefinitionsJson,
+            primaryModel: responseModel || requestModel,
+            temperature,
+            top_p,
+            max_tokens,
+        });
+        if (captureContent) {
+            const toolCalls = choices[0]?.message?.tool_calls;
+            outputMessagesJson = helpers_1.default.buildOutputMessages(choices[0]?.message?.content || '', choices[0]?.finish_reason || 'stop', toolCalls);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
+            inputMessagesJson = helpers_1.default.buildInputMessages(messages || []);
+            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]: requestModel,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: responseModel,
+                [semantic_convention_1.default.SERVER_ADDRESS]: serverAddress,
+                [semantic_convention_1.default.SERVER_PORT]: serverPort,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_ID]: result.id,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON]: [choices[0]?.finish_reason || 'stop'],
+                [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 (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: requestModel,
+            user,
+            cost,
+            aiSystem: AzureAIInferenceWrapper.aiSystem,
+        };
+    }
+    // ──────────────────── Embeddings ────────────────────
+    static _patchEmbeddings(tracer, serverAddress, serverPort) {
+        const genAIEndpoint = 'az.ai.inference.embeddings';
+        return (originalMethod) => {
+            return function (...args) {
+                if ((0, helpers_1.isFrameworkLlmActive)())
+                    return originalMethod.apply(this, args);
+                const body = args[0]?.body || {};
+                const requestModel = body.model || 'text-embedding-3-small';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING} ${requestModel}`;
+                const effectiveCtx = (0, helpers_1.getFrameworkParentContext)() ?? api_1.context.active();
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING, requestModel, serverAddress, serverPort),
+                }, effectiveCtx);
+                return api_1.context.with(api_1.trace.setSpan(effectiveCtx, span), async () => {
+                    const captureContent = config_1.default.captureMessageContent;
+                    let metricParams;
+                    try {
+                        const httpResponse = await originalMethod.apply(this, args);
+                        const responseBody = httpResponse?.body ?? httpResponse;
+                        if (responseBody && typeof responseBody === 'object') {
+                            const _responseModel = responseBody.model || requestModel;
+                            const pricingInfo = config_1.default.pricingInfo || {};
+                            const inputTokens = responseBody.usage?.prompt_tokens || 0;
+                            const cost = helpers_1.default.getEmbedModelCost(requestModel, pricingInfo, inputTokens);
+                            const { encoding_format = 'float', input, dimensions, user } = body;
+                            AzureAIInferenceWrapper.setBaseSpanAttributes(span, {
+                                genAIEndpoint,
+                                model: requestModel,
+                                user,
+                                cost,
+                                aiSystem: AzureAIInferenceWrapper.aiSystem,
+                                serverAddress,
+                                serverPort,
+                            });
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_ENCODING_FORMATS, [encoding_format]);
+                            if (dimensions) {
+                                span.setAttribute(semantic_convention_1.default.GEN_AI_EMBEDDINGS_DIMENSION_COUNT, dimensions);
+                            }
+                            if (captureContent) {
+                                const formattedInput = typeof input === 'string' ? input : JSON.stringify(input);
+                                span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, formattedInput);
+                            }
+                            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+                            metricParams = {
+                                genAIEndpoint,
+                                model: requestModel,
+                                user,
+                                cost,
+                                aiSystem: AzureAIInferenceWrapper.aiSystem,
+                            };
+                        }
+                        return httpResponse;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams) {
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                        }
+                    }
+                });
+            };
+        };
+    }
+}
+AzureAIInferenceWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_AZURE_AI_INFERENCE;
+AzureAIInferenceWrapper.defaultServerAddress = 'models.github.ai';
+AzureAIInferenceWrapper.defaultServerPort = 443;
+exports.default = AzureAIInferenceWrapper;
+//# sourceMappingURL=wrapper.js.map

package/dist/instrumentation/azure-ai-inference/wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/azure-ai-inference/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAwF;AACxF,0DAA2C;AAC3C,yDAIuB;AACvB,oFAA2D;AAC3D,mEAAkE;AAElE,SAAS,iBAAiB,CACxB,aAAqB,EACrB,YAAoB,EACpB,aAAqB,EACrB,UAAkB;IAElB,OAAO;QACL,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;QACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,6BAAkB,CAAC,gCAAgC;QACnG,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;QACvD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,aAAa;QAClD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,UAAU;KAC7C,CAAC;AACJ,CAAC;AAED,MAAM,uBAAwB,SAAQ,sBAAW;IAK/C;;;;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,gCAAgC;iBAC9D;gBACD,SAAS,EAAE,CAAC,6BAAkB,CAAC,gCAAgC,CAAC;aACjE,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;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,QAAgB;QACnC,IAAI,aAAa,GAAG,uBAAuB,CAAC,oBAAoB,CAAC;QACjE,IAAI,UAAU,GAAG,uBAAuB,CAAC,iBAAiB,CAAC;QAC3D,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC9B,aAAa,GAAG,GAAG,CAAC,QAAQ,CAAC;YAC7B,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC5E,CAAC;QAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;QAC9B,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC;IACvC,CAAC;IAED,6DAA6D;IAE7D,MAAM,CAAC,kBAAkB,CACvB,MAAc,EACd,aAAqB,EACrB,UAAkB;QAElB,MAAM,aAAa,GAAG,kCAAkC,CAAC;QACzD,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,UAAqB,GAAG,IAAW;gBACxC,IAAI,IAAA,8BAAoB,GAAE;oBAAE,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACpE,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;gBACjC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC;gBAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC;gBAEtC,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,0BAA0B,IAAI,YAAY,EAAE,CAAC;gBACpF,MAAM,YAAY,GAAG,IAAA,mCAAyB,GAAE,IAAI,aAAO,CAAC,MAAM,EAAE,CAAC;gBACrE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAC3B,6BAAkB,CAAC,0BAA0B,EAC7C,YAAY,EACZ,aAAa,EACb,UAAU,CACX;iBACF,EAAE,YAAY,CAAC,CAAC;gBAEjB,IAAI,QAAQ,EAAE,CAAC;oBACb,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;wBAClB,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;oBACxD,CAAC;oBACD,MAAM,eAAe,GAAG,aAAO,CAAC,IAAI,CAClC,WAAK,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,EACjC,GAAG,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CACvC,CAAC;oBACF,MAAM,gBAAgB,GAAG,eAAe,CAAC,YAAY,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;oBAC7E,IAAI,gBAAgB,EAAE,CAAC;wBACrB,eAAe,CAAC,YAAY,GAAG,KAAK,WAAW,GAAG,UAAiB;4BACjE,IAAI,CAAC;gCACH,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,GAAG,UAAU,CAAC,CAAC;gCACzD,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC;gCACjC,IAAI,CAAC,QAAQ,EAAE,CAAC;oCACd,IAAI,CAAC,GAAG,EAAE,CAAC;oCACX,OAAO,UAAU,CAAC;gCACpB,CAAC;gCACD,UAAU,CAAC,IAAI,GAAG,uBAAuB,CAAC,cAAc,CACtD,QAAQ,EACR,IAAI,EACJ,aAAa,EACb,IAAI,EACJ,aAAa,EACb,UAAU,CACX,CAAC;gCACF,OAAO,UAAU,CAAC;4BACpB,CAAC;4BAAC,OAAO,CAAM,EAAE,CAAC;gCAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gCACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;oCAC9B,aAAa;oCACb,KAAK,EAAE,YAAY;oCACnB,QAAQ,EAAE,uBAAuB,CAAC,QAAQ;oCAC1C,aAAa;oCACb,UAAU;oCACV,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;iCAC5C,CAAC,CAAC;gCACH,IAAI,CAAC,GAAG,EAAE,CAAC;gCACX,MAAM,CAAC,CAAC;4BACV,CAAC;wBACH,CAAC,CAAC;oBACJ,CAAC;oBACD,OAAO,eAAe,CAAC;gBACzB,CAAC;gBAED,OAAO,aAAO;qBACX,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBAClD,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC1C,CAAC,CAAC;qBACD,IAAI,CAAC,CAAC,YAAiB,EAAE,EAAE;oBAC1B,OAAO,uBAAuB,CAAC,eAAe,CAAC;wBAC7C,IAAI;wBACJ,aAAa;wBACb,YAAY;wBACZ,IAAI;wBACJ,aAAa;wBACb,UAAU;qBACX,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,YAAY;wBACnB,QAAQ,EAAE,uBAAuB,CAAC,QAAQ;wBAC1C,aAAa;wBACb,UAAU;wBACV,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,KAAK,CAAC,eAAe,CAAC,EAC3B,IAAI,EACJ,aAAa,EACb,YAAY,EACZ,IAAI,EACJ,aAAa,EACb,UAAU,GAQX;QACC,IAAI,YAAY,CAAC;QACjB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,YAAY,EAAE,IAAI,IAAI,YAAY,CAAC;YAClD,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC3D,YAAY,GAAG,uBAAuB,CAAC,2BAA2B,CAAC;oBACjE,IAAI;oBACJ,aAAa;oBACb,MAAM;oBACN,IAAI;oBACJ,aAAa;oBACb,UAAU;iBACX,CAAC,CAAC;YACL,CAAC;YACD,OAAO,YAAY,CAAC;QACtB,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;;;;;OAKG;IACH,MAAM,CAAC,cAAc,CACnB,IAAS,EACT,WAAgB,EAChB,aAAqB,EACrB,IAAU,EACV,aAAqB,EACrB,UAAkB;QAElB,MAAM,YAAY,GAAG,WAAW,CAAC,KAAK,IAAI,QAAQ,CAAC;QACnD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAa,EAAE,CAAC;QAEhC,MAAM,UAAU,GAAG;YACjB,EAAE,EAAE,EAAE;YACN,KAAK,EAAE,EAAE;YACT,OAAO,EAAE,EAAE;YACX,YAAY,EAAE,MAAM;YACpB,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,EAAW;SACvB,CAAC;QAEF,SAAS,cAAc,CAAC,IAAY;YAClC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;gBAAE,OAAO;YACvC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAClC,IAAI,IAAI,KAAK,QAAQ;gBAAE,OAAO;YAC9B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAChC,IAAI,MAAM,CAAC,EAAE;oBAAE,UAAU,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;gBACzC,IAAI,MAAM,CAAC,KAAK;oBAAE,UAAU,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBAClD,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,MAAM,EAAE,CAAC;oBACX,IAAI,MAAM,CAAC,KAAK,EAAE,OAAO;wBAAE,UAAU,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;oBACtE,IAAI,MAAM,CAAC,aAAa;wBAAE,UAAU,CAAC,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;oBACzE,IAAI,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC;wBAC7B,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;4BACzC,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,CAAC;4BAC1B,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;gCAC1C,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;4BACjG,CAAC;4BACD,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gCACV,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;gCACrC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,IAAI,IAAI,UAAU,CAAC;gCACvD,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;oCAAE,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;gCAClF,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;oCAAE,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,SAAS,GAAG,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;4BACnG,CAAC;iCAAM,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;gCAClC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;4BACxE,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBACjB,UAAU,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;oBACzD,UAAU,CAAC,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC;YAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,OAAO,GAAG,EAAE,CAAC;QAEjB,MAAM,QAAQ,GAAG,IAAI,CAAC;QACtB,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpD,MAAM,WAAW,GAAG,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEhD,MAAM,IAAI,GAAG,uBAAuB,CAAC;QACrC,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,SAAS,QAAQ;YACf,IAAI,SAAS;gBAAE,OAAO;YACtB,SAAS,GAAG,IAAI,CAAC;YACjB,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5E,IAAI,GAAG,GAAG,CAAC,CAAC;YACZ,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnE,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC;YAC/D,CAAC;YAED,IAAI,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;YACzC,IAAI,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC;YAC3C,IAAI,CAAC,WAAW,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;gBAC1D,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAC3C,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC1D,CAAC;YAED,MAAM,MAAM,GAAG;gBACb,EAAE,EAAE,UAAU,CAAC,EAAE;gBACjB,KAAK,EAAE,UAAU,CAAC,KAAK,IAAI,YAAY;gBACvC,OAAO,EAAE,CAAC;wBACR,aAAa,EAAE,UAAU,CAAC,YAAY;wBACtC,OAAO,EAAE;4BACP,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE,UAAU,CAAC,OAAO;4BAC3B,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;yBACjF;qBACF,CAAC;gBACF,KAAK,EAAE;oBACL,aAAa,EAAE,WAAW;oBAC1B,iBAAiB,EAAE,YAAY;iBAChC;aACF,CAAC;YAEF,MAAM,YAAY,GAAG,IAAI,CAAC,2BAA2B,CAAC;gBACpD,IAAI,EAAE,WAAW;gBACjB,aAAa;gBACb,MAAM;gBACN,IAAI;gBACJ,aAAa;gBACb,UAAU;gBACV,IAAI;gBACJ,GAAG;aACJ,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,OAAO,QAAQ,CAAC,EAAE,KAAK,UAAU,EAAE,CAAC;YACtC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAsB,EAAE,EAAE;gBAC7C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC5B,MAAM,IAAI,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAClE,OAAO,IAAI,IAAI,CAAC;gBAChB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAClC,OAAO,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAC5B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACtB,IAAI,OAAO,CAAC,IAAI,EAAE;oBAAE,cAAc,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBACnD,QAAQ,EAAE,CAAC;YACb,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACxB,IAAI,OAAO,CAAC,IAAI,EAAE;oBAAE,cAAc,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBACnD,QAAQ,EAAE,CAAC;YACb,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;gBAClC,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBAC3C,QAAQ,EAAE,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,2BAA2B,CAAC,EACjC,IAAI,EACJ,aAAa,EACb,MAAM,EACN,IAAI,EACJ,aAAa,EACb,UAAU,EACV,IAAI,GAAG,CAAC,EACR,GAAG,GAAG,CAAC,GAUR;QACC,MAAM,cAAc,GAAG,gBAAe,CAAC,qBAAqB,CAAC;QAC7D,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC;QAC5C,MAAM,EACJ,QAAQ,EACR,iBAAiB,GAAG,CAAC,EACrB,UAAU,GAAG,IAAI,EACjB,CAAC,GAAG,CAAC,EACL,gBAAgB,GAAG,CAAC,EACpB,IAAI,GAAG,IAAI,EACX,IAAI,GAAG,IAAI,EACX,WAAW,GAAG,CAAC,EACf,KAAK,EACL,IAAI,EACJ,MAAM,GAAG,KAAK,EACd,KAAK,EAAE,MAAM,GACd,GAAG,IAAI,CAAC;QAET,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;QACvE,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,UAAU,CAAC,CAAC;QAC9E,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,WAAW,CAAC,CAAC;QAC9E,IAAI,gBAAgB,EAAE,CAAC;YACrB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,gBAAgB,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,iBAAiB,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,iBAAiB,CAAC,CAAC;QAC5F,CAAC;QACD,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,MAAM,CAAC,CAAC;QACvE,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,6BAA6B,EAChD,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CACpC,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAe,CAAC,kBAAkB,CAAC,QAAQ,IAAI,EAAE,CAAC,CACnD,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;YACd,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,IAAI,YAAY,CAAC;QACnD,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QACtD,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;QACrD,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;QAC1D,mEAAmE;QACnE,MAAM,eAAe,GACnB,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,qBAAqB,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,mBAAmB,GACvB,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,oBAAoB,EAAE,qBAAqB,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,qBAAqB,EAAE,qBAAqB,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,qBAAqB,EAAE,kBAAkB,CAAC;YAC/D,CAAC,CAAC;QACJ,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,YAAY,EACZ,WAAW,EACX,WAAW,EACX,YAAY,EACZ,eAAe,EACf,mBAAmB,EACnB,IAAI,CACL,CAAC;QAEF,uBAAuB,CAAC,qBAAqB,CAAC,IAAI,EAAE;YAClD,aAAa;YACb,KAAK,EAAE,YAAY;YACnB,IAAI;YACJ,IAAI;YACJ,QAAQ,EAAE,uBAAuB,CAAC,QAAQ;YAC1C,aAAa;YACb,UAAU;SACX,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;QAC3E,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,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,oCAAoC,EACvD,eAAe,CAChB,CAAC;QACJ,CAAC;QACD,IAAI,mBAAmB,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,wCAAwC,EAC3D,mBAAmB,CACpB,CAAC;QACJ,CAAC;QAED,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,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;QACrC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC;YAC9B,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAC3B,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GACd,OAAO,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,KAAK,QAAQ;YAC9C,CAAC,CAAC,6BAAkB,CAAC,uBAAuB;YAC5C,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC;QACjD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QAErE,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;YACpC,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC;YAChD,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACpF,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACtE,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExF,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,IAAI,iBAAqC,CAAC;QAC1C,IAAI,kBAAsC,CAAC;QAC3C,MAAM,mBAAmB,GAAG,iBAAe,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACzE,6DAA6D;QAC7D,gEAAgE;QAChE,uBAAuB;QACvB,MAAM,sBAAsB,GAAG,iBAAe,CAAC,mCAAmC,CAChF,QAAQ,IAAI,EAAE,CACf,CAAC;QACF,MAAM,aAAa,GAAG,uBAAuB,CAAC,kBAAkB,CAAC,IAAI,EAAE;YACrE,sBAAsB;YACtB,mBAAmB;YACnB,YAAY,EAAE,aAAa,IAAI,YAAY;YAC3C,WAAW;YACX,KAAK;YACL,UAAU;SACX,CAAC,CAAC;QACH,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC;YAClD,kBAAkB,GAAG,iBAAe,CAAC,mBAAmB,CACtD,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,EAClC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,MAAM,EACnC,SAAS,CACV,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;YACjF,iBAAiB,GAAG,iBAAe,CAAC,kBAAkB,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;YACvE,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,YAAY;gBACvD,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,aAAa;gBACzD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,aAAa;gBAClD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,UAAU;gBAC5C,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,EAAE;gBAClD,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,MAAM,CAAC;gBACzF,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,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,YAAY;YACnB,IAAI;YACJ,IAAI;YACJ,QAAQ,EAAE,uBAAuB,CAAC,QAAQ;SAC3C,CAAC;IACJ,CAAC;IAED,uDAAuD;IAEvD,MAAM,CAAC,gBAAgB,CACrB,MAAc,EACd,aAAqB,EACrB,UAAkB;QAElB,MAAM,aAAa,GAAG,4BAA4B,CAAC;QACnD,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,UAAqB,GAAG,IAAW;gBACxC,IAAI,IAAA,8BAAoB,GAAE;oBAAE,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACpE,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;gBACjC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,IAAI,wBAAwB,CAAC;gBAC5D,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,+BAA+B,IAAI,YAAY,EAAE,CAAC;gBACzF,MAAM,YAAY,GAAG,IAAA,mCAAyB,GAAE,IAAI,aAAO,CAAC,MAAM,EAAE,CAAC;gBACrE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE;oBACtC,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAC3B,6BAAkB,CAAC,+BAA+B,EAClD,YAAY,EACZ,aAAa,EACb,UAAU,CACX;iBACF,EAAE,YAAY,CAAC,CAAC;gBAEjB,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBAChE,MAAM,cAAc,GAAG,gBAAe,CAAC,qBAAqB,CAAC;oBAC7D,IAAI,YAA4C,CAAC;oBACjD,IAAI,CAAC;wBACH,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBAC5D,MAAM,YAAY,GAAG,YAAY,EAAE,IAAI,IAAI,YAAY,CAAC;wBAExD,IAAI,YAAY,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;4BACrD,MAAM,cAAc,GAAG,YAAY,CAAC,KAAK,IAAI,YAAY,CAAC;4BAC1D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;4BACtD,MAAM,WAAW,GAAG,YAAY,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;4BAC3D,MAAM,IAAI,GAAG,iBAAe,CAAC,iBAAiB,CAAC,YAAY,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;4BAEvF,MAAM,EAAE,eAAe,GAAG,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;4BAEpE,uBAAuB,CAAC,qBAAqB,CAAC,IAAI,EAAE;gCAClD,aAAa;gCACb,KAAK,EAAE,YAAY;gCACnB,IAAI;gCACJ,IAAI;gCACJ,QAAQ,EAAE,uBAAuB,CAAC,QAAQ;gCAC1C,aAAa;gCACb,UAAU;6BACX,CAAC,CAAC;4BAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;4BACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;4BACzF,IAAI,UAAU,EAAE,CAAC;gCACf,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iCAAiC,EAAE,UAAU,CAAC,CAAC;4BACtF,CAAC;4BACD,IAAI,cAAc,EAAE,CAAC;gCACnB,MAAM,cAAc,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;gCACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;4BAC9E,CAAC;4BAED,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;4BAE7E,YAAY,GAAG;gCACb,aAAa;gCACb,KAAK,EAAE,YAAY;gCACnB,IAAI;gCACJ,IAAI;gCACJ,QAAQ,EAAE,uBAAuB,CAAC,QAAQ;6BAC3C,CAAC;wBACJ,CAAC;wBAED,OAAO,YAAY,CAAC;oBACtB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACzC,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;;AA5pBM,gCAAQ,GAAG,6BAAkB,CAAC,gCAAgC,CAAC;AAC/D,4CAAoB,GAAG,kBAAkB,CAAC;AAC1C,yCAAiB,GAAG,GAAG,CAAC;AA6pBjC,kBAAe,uBAAuB,CAAC"}

package/dist/instrumentation/base-wrapper.d.ts

@@ -0,0 +1,16 @@
+import { Span } from '@opentelemetry/api';
+export type BaseSpanAttributes = {
+    genAIEndpoint: string;
+    model: string;
+    user?: unknown;
+    cost?: number | string;
+    aiSystem: string;
+    serverAddress?: string;
+    serverPort?: number;
+    errorType?: string;
+};
+export default class BaseWrapper {
+    static setBaseSpanAttributes(span: Span, { genAIEndpoint: _genAIEndpoint, model, user, cost, aiSystem, serverAddress, serverPort }: BaseSpanAttributes): void;
+    static recordMetrics(span: Span, baseAttributes: BaseSpanAttributes): void;
+    static getSpanAttribute(span: Span, key: string): number | undefined;
+}