shield360 1.0.0

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

@@ -0,0 +1,34 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+export default class AnthropicWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    static _patchMessageCreate(tracer: Tracer): any;
+    static _messageCreate({ args, genAIEndpoint, response, span, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+    }): Promise<any>;
+    static _messageCreateGenerator({ args, genAIEndpoint, response, span, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+    }): AsyncGenerator<unknown, any, unknown>;
+    static _messageCreateCommonSetter({ args, genAIEndpoint, result, span, ttft, tbt, }: {
+        args: any[];
+        genAIEndpoint: string;
+        result: any;
+        span: Span;
+        ttft?: number;
+        tbt?: number;
+    }): Promise<{
+        genAIEndpoint: string;
+        model: any;
+        user: any;
+        cost: number;
+        aiSystem: string;
+    }>;
+}

package/dist/instrumentation/anthropic/wrapper.js

@@ -0,0 +1,426 @@
+"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 FINISH_REASON_MAP = {
+    end_turn: 'stop',
+    max_tokens: 'length',
+    stop_sequence: 'stop',
+    tool_use: 'tool_call',
+};
+function mapFinishReason(reason) {
+    return FINISH_REASON_MAP[reason] || reason || 'stop';
+}
+function spanCreationAttrs(operationName, requestModel) {
+    return {
+        [semantic_convention_1.default.GEN_AI_OPERATION]: operationName,
+        [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: AnthropicWrapper.aiSystem,
+        [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+        [semantic_convention_1.default.SERVER_ADDRESS]: AnthropicWrapper.serverAddress,
+        [semantic_convention_1.default.SERVER_PORT]: AnthropicWrapper.serverPort,
+    };
+}
+class AnthropicWrapper extends base_wrapper_1.default {
+    static _patchMessageCreate(tracer) {
+        const genAIEndpoint = 'anthropic.resources.messages';
+        return (originalMethod) => {
+            return async function (...args) {
+                if ((0, helpers_1.isFrameworkLlmActive)())
+                    return originalMethod.apply(this, args);
+                const requestModel = args[0]?.model || 'claude-3-5-sonnet-latest';
+                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),
+                }, effectiveCtx);
+                return api_1.context
+                    .with(api_1.trace.setSpan(effectiveCtx, span), async () => {
+                    return originalMethod.apply(this, args);
+                })
+                    .then((response) => {
+                    const { stream = false } = args[0];
+                    if (stream) {
+                        return helpers_1.default.createStreamProxy(response, AnthropicWrapper._messageCreateGenerator({
+                            args,
+                            genAIEndpoint,
+                            response,
+                            span,
+                        }));
+                    }
+                    return AnthropicWrapper._messageCreate({ args, genAIEndpoint, response, span });
+                })
+                    .catch((e) => {
+                    helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: requestModel,
+                        aiSystem: AnthropicWrapper.aiSystem,
+                        serverAddress: AnthropicWrapper.serverAddress,
+                        serverPort: AnthropicWrapper.serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
+                    span.end();
+                    throw e;
+                });
+            };
+        };
+    }
+    static async _messageCreate({ args, genAIEndpoint, response, span, }) {
+        let metricParams;
+        try {
+            metricParams = await AnthropicWrapper._messageCreateCommonSetter({
+                args,
+                genAIEndpoint,
+                result: response,
+                span,
+            });
+            return response;
+        }
+        catch (e) {
+            helpers_1.default.handleException(span, e);
+            throw e;
+        }
+        finally {
+            span.end();
+            if (metricParams) {
+                base_wrapper_1.default.recordMetrics(span, metricParams);
+            }
+        }
+    }
+    static async *_messageCreateGenerator({ args, genAIEndpoint, response, span, }) {
+        let metricParams;
+        const timestamps = [];
+        const startTime = Date.now();
+        try {
+            const result = {
+                id: '',
+                model: '',
+                stop_reason: '',
+                content: [],
+                usage: {
+                    input_tokens: 0,
+                    output_tokens: 0,
+                    cache_creation_input_tokens: 0,
+                    cache_read_input_tokens: 0,
+                },
+            };
+            let llmResponse = '';
+            let toolId = '';
+            let toolName = '';
+            let toolArguments = '';
+            for await (const chunk of response) {
+                timestamps.push(Date.now());
+                switch (chunk.type) {
+                    case 'message_start':
+                        if (chunk.message) {
+                            result.id = chunk.message.id;
+                            result.model = chunk.message.model;
+                            result.usage.input_tokens = Number(chunk.message.usage?.input_tokens) || 0;
+                            result.usage.output_tokens += Number(chunk.message.usage?.output_tokens) || 0;
+                            result.usage.cache_creation_input_tokens =
+                                Number(chunk.message.usage?.cache_creation_input_tokens) || 0;
+                            result.usage.cache_read_input_tokens =
+                                Number(chunk.message.usage?.cache_read_input_tokens) || 0;
+                            result.stop_reason = chunk.message.stop_reason || '';
+                        }
+                        break;
+                    case 'content_block_start':
+                        if (chunk.content_block?.type === 'tool_use') {
+                            toolId = chunk.content_block.id || '';
+                            toolName = chunk.content_block.name || '';
+                            toolArguments = '';
+                        }
+                        break;
+                    case 'content_block_delta':
+                        if (chunk.delta?.text) {
+                            llmResponse += chunk.delta.text;
+                        }
+                        if (chunk.delta?.partial_json) {
+                            toolArguments += chunk.delta.partial_json;
+                        }
+                        break;
+                    case 'content_block_stop':
+                        break;
+                    case 'message_delta':
+                        result.stop_reason = chunk.delta?.stop_reason || result.stop_reason;
+                        result.usage.output_tokens += Number(chunk.usage?.output_tokens) || 0;
+                        if (chunk.usage?.cache_creation_input_tokens != null) {
+                            result.usage.cache_creation_input_tokens =
+                                Number(chunk.usage.cache_creation_input_tokens) || 0;
+                        }
+                        if (chunk.usage?.cache_read_input_tokens != null) {
+                            result.usage.cache_read_input_tokens =
+                                Number(chunk.usage.cache_read_input_tokens) || 0;
+                        }
+                        break;
+                    case 'message_stop':
+                        break;
+                }
+                yield chunk;
+            }
+            if (llmResponse) {
+                result.content.push({ type: 'text', text: llmResponse });
+            }
+            if (toolId) {
+                let parsedInput = {};
+                try {
+                    parsedInput = JSON.parse(toolArguments);
+                }
+                catch { /* keep empty */ }
+                result.content.push({
+                    type: 'tool_use',
+                    id: toolId,
+                    name: toolName,
+                    input: parsedInput,
+                });
+            }
+            const ttft = timestamps.length > 0 ? (timestamps[0] - startTime) / 1000 : 0;
+            let tbt = 0;
+            if (timestamps.length > 1) {
+                const timeDiffs = timestamps.slice(1).map((t, i) => t - timestamps[i]);
+                tbt = timeDiffs.reduce((a, b) => a + b, 0) / timeDiffs.length / 1000;
+            }
+            metricParams = await AnthropicWrapper._messageCreateCommonSetter({
+                args,
+                genAIEndpoint,
+                result,
+                span,
+                ttft,
+                tbt,
+            });
+            return result;
+        }
+        catch (e) {
+            helpers_1.default.handleException(span, e);
+            throw e;
+        }
+        finally {
+            span.end();
+            if (metricParams) {
+                base_wrapper_1.default.recordMetrics(span, metricParams);
+            }
+        }
+    }
+    static async _messageCreateCommonSetter({ args, genAIEndpoint, result, span, ttft = 0, tbt = 0, }) {
+        const captureContent = config_1.default.captureMessageContent;
+        const requestModel = args[0]?.model || 'claude-3-5-sonnet-latest';
+        const { messages, system, max_tokens = null, seed = null, temperature = 1, top_p, top_k, stop_sequences = null, stream = false, user, tools: _tools, } = args[0];
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, temperature);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, top_p || 1);
+        if (top_k != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K, top_k);
+        }
+        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_IS_STREAM, stream);
+        if (seed != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_SEED, Number(seed));
+        }
+        if (stop_sequences && Array.isArray(stop_sequences) && stop_sequences.length > 0) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, stop_sequences);
+        }
+        if (captureContent) {
+            const systemStr = typeof system === 'string' ? system : undefined;
+            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages || [], systemStr));
+            if (system) {
+                const sysAttr = typeof system === 'string' ? system : JSON.stringify(system);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS, sysAttr);
+            }
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, result.id);
+        const responseModel = result.model || requestModel;
+        const pricingInfo = config_1.default.pricingInfo || {};
+        // Anthropic reports input_tokens exclusive of cache read / creation tokens,
+        // so cache tokens are added on top (promptTokensIncludeCache stays false).
+        const cost = helpers_1.default.getChatModelCost(requestModel, pricingInfo, result.usage.input_tokens, result.usage.output_tokens, Number(result.usage.cache_read_input_tokens) || 0, Number(result.usage.cache_creation_input_tokens) || 0);
+        AnthropicWrapper.setBaseSpanAttributes(span, {
+            genAIEndpoint,
+            model: requestModel,
+            user,
+            cost,
+            aiSystem: AnthropicWrapper.aiSystem,
+            serverAddress: AnthropicWrapper.serverAddress,
+            serverPort: AnthropicWrapper.serverPort,
+        });
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, responseModel);
+        const inputTokens = result.usage.input_tokens;
+        const outputTokens = result.usage.output_tokens;
+        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 (result.usage.cache_creation_input_tokens) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, result.usage.cache_creation_input_tokens);
+        }
+        if (result.usage.cache_read_input_tokens) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, result.usage.cache_read_input_tokens);
+        }
+        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 finishReason = mapFinishReason(result.stop_reason);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [finishReason]);
+        const toolUseBlocks = (result.content || []).filter((b) => b.type === 'tool_use');
+        const outputType = toolUseBlocks.length > 0
+            ? semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_JSON
+            : semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT;
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, outputType);
+        if (toolUseBlocks.length > 0) {
+            const toolNames = toolUseBlocks.map((b) => b.name || '').filter(Boolean);
+            const toolIds = toolUseBlocks.map((b) => b.id || '').filter(Boolean);
+            const toolArgs = toolUseBlocks.map((b) => JSON.stringify(b.input || {}));
+            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;
+        // Anthropic tool schema uses `input_schema` instead of `parameters`; the
+        // shared helper already handles both shapes.
+        const toolDefinitionsJson = helpers_1.default.buildToolDefinitions(_tools);
+        const systemInstructionsJson = (() => {
+            if (!system)
+                return undefined;
+            if (typeof system === 'string') {
+                return JSON.stringify([{ type: 'text', content: system }]);
+            }
+            try {
+                return JSON.stringify(system);
+            }
+            catch {
+                return undefined;
+            }
+        })();
+        // Stamp `shield360.agent.version_hash` (auto) and `gen_ai.agent.version`
+        // (user override) regardless of content capture so versions still group
+        // correctly when capture_message_content=false.
+        const versionExtras = {};
+        try {
+            const versionHash = helpers_1.default.computeAgentVersionHash({
+                systemInstructions: systemInstructionsJson ?? null,
+                toolDefinitions: toolDefinitionsJson ?? null,
+                primaryModel: responseModel || requestModel,
+                runtimeConfig: {
+                    temperature: temperature ?? null,
+                    top_p: top_p ?? null,
+                    max_tokens: max_tokens ?? null,
+                    provider: semantic_convention_1.default.GEN_AI_SYSTEM_ANTHROPIC,
+                },
+                providers: [semantic_convention_1.default.GEN_AI_SYSTEM_ANTHROPIC],
+            });
+            if (versionHash) {
+                versionExtras[semantic_convention_1.default.SHIELD360_AGENT_VERSION_HASH] = versionHash;
+                span.setAttribute(semantic_convention_1.default.SHIELD360_AGENT_VERSION_HASH, versionHash);
+            }
+        }
+        catch {
+            // Never fail the wrapped call on hash issues.
+        }
+        const versionLabel = (0, helpers_1.getCurrentAgentVersion)();
+        if (versionLabel) {
+            versionExtras[semantic_convention_1.default.GEN_AI_AGENT_VERSION] = versionLabel;
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_VERSION, versionLabel);
+        }
+        if (captureContent) {
+            const textContent = (result.content || [])
+                .filter((b) => b.type === 'text')
+                .map((b) => b.text || '')
+                .join('');
+            const toolCallsForOutput = toolUseBlocks.length > 0
+                ? toolUseBlocks.map((b) => ({
+                    id: b.id || '',
+                    name: b.name || '',
+                    arguments: b.input || {},
+                }))
+                : undefined;
+            outputMessagesJson = helpers_1.default.buildOutputMessages(textContent, finishReason, toolCallsForOutput);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
+            const systemStr = typeof system === 'string' ? system : undefined;
+            inputMessagesJson = helpers_1.default.buildInputMessages(messages || [], systemStr);
+        }
+        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]: AnthropicWrapper.serverAddress,
+                [semantic_convention_1.default.SERVER_PORT]: AnthropicWrapper.serverPort,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_ID]: result.id,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON]: [finishReason],
+                [semantic_convention_1.default.GEN_AI_OUTPUT_TYPE]: outputType,
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: inputTokens,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: outputTokens,
+                ...versionExtras,
+            };
+            if (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 (toolDefinitionsJson)
+                eventAttrs[semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS] = toolDefinitionsJson;
+            helpers_1.default.emitInferenceEvent(span, eventAttrs);
+        }
+        return {
+            genAIEndpoint,
+            model: requestModel,
+            user,
+            cost,
+            aiSystem: AnthropicWrapper.aiSystem,
+        };
+    }
+}
+AnthropicWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_ANTHROPIC;
+AnthropicWrapper.serverAddress = 'api.anthropic.com';
+AnthropicWrapper.serverPort = 443;
+exports.default = AnthropicWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/anthropic/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAwF;AACxF,0DAA2C;AAC3C,yDAIuB;AACvB,oFAA2D;AAC3D,mEAA0C;AAE1C,MAAM,iBAAiB,GAA2B;IAChD,QAAQ,EAAE,MAAM;IAChB,UAAU,EAAE,QAAQ;IACpB,aAAa,EAAE,MAAM;IACrB,QAAQ,EAAE,WAAW;CACtB,CAAC;AAEF,SAAS,eAAe,CAAC,MAAc;IACrC,OAAO,iBAAiB,CAAC,MAAM,CAAC,IAAI,MAAM,IAAI,MAAM,CAAC;AACvD,CAAC;AAED,SAAS,iBAAiB,CACxB,aAAqB,EACrB,YAAoB;IAEpB,OAAO;QACL,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;QACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,gBAAgB,CAAC,QAAQ;QACzE,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;QACvD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,gBAAgB,CAAC,aAAa;QACnE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,gBAAgB,CAAC,UAAU;KAC9D,CAAC;AACJ,CAAC;AAED,MAAqB,gBAAiB,SAAQ,sBAAW;IAKvD,MAAM,CAAC,mBAAmB,CAAC,MAAc;QACvC,MAAM,aAAa,GAAG,8BAA8B,CAAC;QACrD,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;gBACpE,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,0BAA0B,CAAC;gBAClE,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,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,YAAY,CAAC;iBAC3F,EAAE,YAAY,CAAC,CAAC;gBACjB,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,QAAa,EAAE,EAAE;oBACtB,MAAM,EAAE,MAAM,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;oBAEnC,IAAI,MAAM,EAAE,CAAC;wBACX,OAAO,iBAAe,CAAC,iBAAiB,CACtC,QAAQ,EACR,gBAAgB,CAAC,uBAAuB,CAAC;4BACvC,IAAI;4BACJ,aAAa;4BACb,QAAQ;4BACR,IAAI;yBACL,CAAC,CACH,CAAC;oBACJ,CAAC;oBAED,OAAO,gBAAgB,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClF,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,gBAAgB,CAAC,QAAQ;wBACnC,aAAa,EAAE,gBAAgB,CAAC,aAAa;wBAC7C,UAAU,EAAE,gBAAgB,CAAC,UAAU;wBACvC,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,cAAc,CAAC,EAC1B,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,IAAI,GAML;QACC,IAAI,YAAY,CAAC;QACjB,IAAI,CAAC;YACH,YAAY,GAAG,MAAM,gBAAgB,CAAC,0BAA0B,CAAC;gBAC/D,IAAI;gBACJ,aAAa;gBACb,MAAM,EAAE,QAAQ;gBAChB,IAAI;aACL,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,YAAY,EAAE,CAAC;gBACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,CAAC,uBAAuB,CAAC,EACpC,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,IAAI,GAML;QACC,IAAI,YAAY,CAAC;QACjB,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG;gBACb,EAAE,EAAE,EAAE;gBACN,KAAK,EAAE,EAAE;gBACT,WAAW,EAAE,EAAE;gBACf,OAAO,EAAE,EAAW;gBACpB,KAAK,EAAE;oBACL,YAAY,EAAE,CAAC;oBACf,aAAa,EAAE,CAAC;oBAChB,2BAA2B,EAAE,CAAC;oBAC9B,uBAAuB,EAAE,CAAC;iBAC3B;aACF,CAAC;YAEF,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,QAAQ,GAAG,EAAE,CAAC;YAClB,IAAI,aAAa,GAAG,EAAE,CAAC;YAEvB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACnC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBAE5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACnB,KAAK,eAAe;wBAClB,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;4BAClB,MAAM,CAAC,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;4BAC7B,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;4BACnC,MAAM,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;4BAC3E,MAAM,CAAC,KAAK,CAAC,aAAa,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;4BAC9E,MAAM,CAAC,KAAK,CAAC,2BAA2B;gCACtC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,2BAA2B,CAAC,IAAI,CAAC,CAAC;4BAChE,MAAM,CAAC,KAAK,CAAC,uBAAuB;gCAClC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,uBAAuB,CAAC,IAAI,CAAC,CAAC;4BAC5D,MAAM,CAAC,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,IAAI,EAAE,CAAC;wBACvD,CAAC;wBACD,MAAM;oBAER,KAAK,qBAAqB;wBACxB,IAAI,KAAK,CAAC,aAAa,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;4BAC7C,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC;4BACtC,QAAQ,GAAG,KAAK,CAAC,aAAa,CAAC,IAAI,IAAI,EAAE,CAAC;4BAC1C,aAAa,GAAG,EAAE,CAAC;wBACrB,CAAC;wBACD,MAAM;oBAER,KAAK,qBAAqB;wBACxB,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC;4BACtB,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;wBAClC,CAAC;wBACD,IAAI,KAAK,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC;4BAC9B,aAAa,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;wBAC5C,CAAC;wBACD,MAAM;oBAER,KAAK,oBAAoB;wBACvB,MAAM;oBAER,KAAK,eAAe;wBAClB,MAAM,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK,EAAE,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;wBACpE,MAAM,CAAC,KAAK,CAAC,aAAa,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;wBACtE,IAAI,KAAK,CAAC,KAAK,EAAE,2BAA2B,IAAI,IAAI,EAAE,CAAC;4BACrD,MAAM,CAAC,KAAK,CAAC,2BAA2B;gCACtC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC;wBACzD,CAAC;wBACD,IAAI,KAAK,CAAC,KAAK,EAAE,uBAAuB,IAAI,IAAI,EAAE,CAAC;4BACjD,MAAM,CAAC,KAAK,CAAC,uBAAuB;gCAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;wBACrD,CAAC;wBACD,MAAM;oBAER,KAAK,cAAc;wBACjB,MAAM;gBACV,CAAC;gBAED,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YAC3D,CAAC;YACD,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,WAAW,GAAG,EAAE,CAAC;gBACrB,IAAI,CAAC;oBAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBAAC,CAAC;gBAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,CAAC;gBAC3E,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;oBAClB,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,MAAM;oBACV,IAAI,EAAE,QAAQ;oBACd,KAAK,EAAE,WAAW;iBACnB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5E,IAAI,GAAG,GAAG,CAAC,CAAC;YACZ,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;gBACvE,GAAG,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC;YACvE,CAAC;YAED,YAAY,GAAG,MAAM,gBAAgB,CAAC,0BAA0B,CAAC;gBAC/D,IAAI;gBACJ,aAAa;gBACb,MAAM;gBACN,IAAI;gBACJ,IAAI;gBACJ,GAAG;aACJ,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,YAAY,EAAE,CAAC;gBACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,0BAA0B,CAAC,EACtC,IAAI,EACJ,aAAa,EACb,MAAM,EACN,IAAI,EACJ,IAAI,GAAG,CAAC,EACR,GAAG,GAAG,CAAC,GAQR;QACC,MAAM,cAAc,GAAG,gBAAe,CAAC,qBAAqB,CAAC;QAC7D,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,0BAA0B,CAAC;QAClE,MAAM,EACJ,QAAQ,EACR,MAAM,EACN,UAAU,GAAG,IAAI,EACjB,IAAI,GAAG,IAAI,EACX,WAAW,GAAG,CAAC,EACf,KAAK,EACL,KAAK,EACL,cAAc,GAAG,IAAI,EACrB,MAAM,GAAG,KAAK,EACd,IAAI,EACJ,KAAK,EAAE,MAAM,GACd,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAEZ,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,WAAW,CAAC,CAAC;QAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;QACvE,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QACpE,CAAC;QACD,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,wBAAwB,EAAE,MAAM,CAAC,CAAC;QACvE,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,IAAI,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,cAAc,CAAC,CAAC;QACtF,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,SAAS,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAClE,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,qBAAqB,EACxC,iBAAe,CAAC,kBAAkB,CAAC,QAAQ,IAAI,EAAE,EAAE,SAAS,CAAC,CAC9D,CAAC;YACF,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;gBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;QAEpE,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,IAAI,YAAY,CAAC;QAEnD,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QACtD,4EAA4E;QAC5E,2EAA2E;QAC3E,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,YAAY,EACZ,WAAW,EACX,MAAM,CAAC,KAAK,CAAC,YAAY,EACzB,MAAM,CAAC,KAAK,CAAC,aAAa,EAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,CAAC,EACjD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,IAAI,CAAC,CACtD,CAAC;QAEF,gBAAgB,CAAC,qBAAqB,CAAC,IAAI,EAAE;YAC3C,aAAa;YACb,KAAK,EAAE,YAAY;YACnB,IAAI;YACJ,IAAI;YACJ,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;YACnC,aAAa,EAAE,gBAAgB,CAAC,aAAa;YAC7C,UAAU,EAAE,gBAAgB,CAAC,UAAU;SACxC,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;QAE3E,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC;QAC9C,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC;QAChD,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,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,CAAC;YAC7C,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,wCAAwC,EAC3D,MAAM,CAAC,KAAK,CAAC,2BAA2B,CACzC,CAAC;QACJ,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,CAAC;YACzC,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,oCAAoC,EACvD,MAAM,CAAC,KAAK,CAAC,uBAAuB,CACrC,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,YAAY,GAAG,eAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACzD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;QAEpF,MAAM,aAAa,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QACvF,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,GAAG,CAAC;YACzC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB;YAC5C,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC;QAC/C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QAErE,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC9E,MAAM,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC1E,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;YAC9E,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACvG,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACtG,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACvG,CAAC;QAED,IAAI,iBAAqC,CAAC;QAC1C,IAAI,kBAAsC,CAAC;QAC3C,yEAAyE;QACzE,6CAA6C;QAC7C,MAAM,mBAAmB,GAAG,iBAAe,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACzE,MAAM,sBAAsB,GAAG,CAAC,GAAG,EAAE;YACnC,IAAI,CAAC,MAAM;gBAAE,OAAO,SAAS,CAAC;YAC9B,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAC7D,CAAC;YACD,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAChC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,yEAAyE;QACzE,wEAAwE;QACxE,gDAAgD;QAChD,MAAM,aAAa,GAA2B,EAAE,CAAC;QACjD,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,iBAAe,CAAC,uBAAuB,CAAC;gBAC1D,kBAAkB,EAAE,sBAAsB,IAAI,IAAI;gBAClD,eAAe,EAAE,mBAAmB,IAAI,IAAI;gBAC5C,YAAY,EAAE,aAAa,IAAI,YAAY;gBAC3C,aAAa,EAAE;oBACb,WAAW,EAAE,WAAW,IAAI,IAAI;oBAChC,KAAK,EAAE,KAAK,IAAI,IAAI;oBACpB,UAAU,EAAE,UAAU,IAAI,IAAI;oBAC9B,QAAQ,EAAE,6BAAkB,CAAC,uBAAuB;iBACrD;gBACD,SAAS,EAAE,CAAC,6BAAkB,CAAC,uBAAuB,CAAC;aACxD,CAAC,CAAC;YACH,IAAI,WAAW,EAAE,CAAC;gBAChB,aAAa,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,GAAG,WAAW,CAAC;gBAC7E,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,4BAA4B,EAC/C,WAAW,CACZ,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,8CAA8C;QAChD,CAAC;QACD,MAAM,YAAY,GAAG,IAAA,gCAAsB,GAAE,CAAC;QAC9C,IAAI,YAAY,EAAE,CAAC;YACjB,aAAa,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,GAAG,YAAY,CAAC;YACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,WAAW,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;iBACvC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBACrC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;iBAC7B,IAAI,CAAC,EAAE,CAAC,CAAC;YACZ,MAAM,kBAAkB,GAAG,aAAa,CAAC,MAAM,GAAG,CAAC;gBACjD,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;oBAC7B,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE;oBACd,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE;oBAClB,SAAS,EAAE,CAAC,CAAC,KAAK,IAAI,EAAE;iBACzB,CAAC,CAAC;gBACL,CAAC,CAAC,SAAS,CAAC;YACd,kBAAkB,GAAG,iBAAe,CAAC,mBAAmB,CACtD,WAAW,EACX,YAAY,EACZ,kBAAkB,CACnB,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;YACjF,MAAM,SAAS,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAClE,iBAAiB,GAAG,iBAAe,CAAC,kBAAkB,CAAC,QAAQ,IAAI,EAAE,EAAE,SAAS,CAAC,CAAC;QACpF,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,gBAAgB,CAAC,aAAa;gBACnE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,gBAAgB,CAAC,UAAU;gBAC7D,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,EAAE;gBAClD,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,EAAE,CAAC,YAAY,CAAC;gBAClE,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,UAAU;gBACnD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,WAAW;gBAC3D,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,YAAY;gBAC7D,GAAG,aAAa;aACjB,CAAC;YACF,IAAI,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,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,gBAAgB,CAAC,QAAQ;SACpC,CAAC;IACJ,CAAC;;AArcM,yBAAQ,GAAG,6BAAkB,CAAC,uBAAuB,CAAC;AACtD,8BAAa,GAAG,mBAAmB,CAAC;AACpC,2BAAU,GAAG,GAAG,CAAC;kBAHL,gBAAgB"}

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

@@ -0,0 +1,12 @@
+import { InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+import { InstrumentationConfig } from '@opentelemetry/instrumentation';
+export interface AssemblyAIInstrumentationConfig extends InstrumentationConfig {
+}
+export default class Shield360AssemblyAIInstrumentation extends InstrumentationBase {
+    private _transcriptsProto;
+    constructor(config?: AssemblyAIInstrumentationConfig);
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    manualPatch(assemblyai: any): void;
+    protected _patch(moduleExports: any, moduleVersion?: string): void;
+    protected _unpatch(moduleExports: any): void;
+}

package/dist/instrumentation/assemblyai/index.js

@@ -0,0 +1,91 @@
+"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 api_1 = require("@opentelemetry/api");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+const TRANSCRIBE_METHODS = ['transcribe', 'submit', 'get'];
+class Shield360AssemblyAIInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-assemblyai`, '1.0.0', config);
+        this._transcriptsProto = null;
+    }
+    init() {
+        const assemblyaiModule = new instrumentation_1.InstrumentationNodeModuleDefinition('assemblyai', ['>=4.0.0'], (moduleExports, moduleVersion) => {
+            this._patch(moduleExports, moduleVersion);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [assemblyaiModule];
+    }
+    manualPatch(assemblyai) {
+        this._patch(assemblyai);
+    }
+    _patch(moduleExports, moduleVersion) {
+        try {
+            const OriginalAssemblyAIClient = moduleExports.AssemblyAI;
+            if (!OriginalAssemblyAIClient || typeof OriginalAssemblyAIClient !== 'function') {
+                return;
+            }
+            if ((0, instrumentation_1.isWrapped)(OriginalAssemblyAIClient)) {
+                this._unwrap(moduleExports, 'AssemblyAI');
+            }
+            const tracer = this.tracer;
+            const sdkVersion = moduleVersion ? String(moduleVersion) : undefined;
+            // Capture instance helpers in closures so the wrapped constructor (a plain
+            // function expression, required to carry the original's static surface)
+            // never needs to alias `this`.
+            const wrap = this._wrap.bind(this);
+            const rememberTranscriptsProto = (proto) => {
+                this._transcriptsProto = proto;
+            };
+            this._wrap(moduleExports, 'AssemblyAI', (original) => {
+                const WrappedAssemblyAI = function (...clientArgs) {
+                    const client = new original(...clientArgs);
+                    if (client && client.transcripts) {
+                        const transcriptsProto = Object.getPrototypeOf(client.transcripts);
+                        if (transcriptsProto) {
+                            rememberTranscriptsProto(transcriptsProto);
+                            for (const methodName of TRANSCRIBE_METHODS) {
+                                if (typeof transcriptsProto[methodName] === 'function' &&
+                                    !(0, instrumentation_1.isWrapped)(transcriptsProto[methodName])) {
+                                    wrap(transcriptsProto, methodName, wrapper_1.default._patchTranscribe(tracer, methodName, sdkVersion));
+                                }
+                            }
+                        }
+                    }
+                    return client;
+                };
+                // Preserve the original constructor's static surface (static methods,
+                // properties and metadata) so consumers relying on them still work.
+                Object.setPrototypeOf(WrappedAssemblyAI, original);
+                Object.assign(WrappedAssemblyAI, original);
+                WrappedAssemblyAI.prototype = original.prototype;
+                return WrappedAssemblyAI;
+            });
+        }
+        catch (e) {
+            api_1.diag.error('assemblyai instrumentation: error in _patch method', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        if (moduleExports.AssemblyAI) {
+            this._unwrap(moduleExports, 'AssemblyAI');
+        }
+        if (this._transcriptsProto) {
+            for (const methodName of TRANSCRIBE_METHODS) {
+                if ((0, instrumentation_1.isWrapped)(this._transcriptsProto[methodName])) {
+                    this._unwrap(this._transcriptsProto, methodName);
+                }
+            }
+        }
+    }
+}
+exports.default = Shield360AssemblyAIInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/assemblyai/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,4CAA0C;AAC1C,6CAAwD;AACxD,wDAA0C;AAI1C,MAAM,kBAAkB,GAAG,CAAC,YAAY,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;AAE3D,MAAqB,kCAAmC,SAAQ,qCAAmB;IAGjF,YAAY,SAA0C,EAAE;QACtD,KAAK,CAAC,GAAG,iCAAsB,6BAA6B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAHzE,sBAAiB,GAAQ,IAAI,CAAC;IAItC,CAAC;IAES,IAAI;QACZ,MAAM,gBAAgB,GAAG,IAAI,qDAAmC,CAC9D,YAAY,EACZ,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,aAAa,EAAE,EAAE;YAC/B,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;YAC1C,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;QAEF,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC5B,CAAC;IAEM,WAAW,CAAC,UAAe;QAChC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC1B,CAAC;IAES,MAAM,CAAC,aAAkB,EAAE,aAAsB;QACzD,IAAI,CAAC;YACH,MAAM,wBAAwB,GAAG,aAAa,CAAC,UAAU,CAAC;YAC1D,IAAI,CAAC,wBAAwB,IAAI,OAAO,wBAAwB,KAAK,UAAU,EAAE,CAAC;gBAChF,OAAO;YACT,CAAC;YAED,IAAI,IAAA,2BAAS,EAAC,wBAAwB,CAAC,EAAE,CAAC;gBACxC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAC5C,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAC3B,MAAM,UAAU,GAAG,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAErE,2EAA2E;YAC3E,wEAAwE;YACxE,+BAA+B;YAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,wBAAwB,GAAG,CAAC,KAAU,EAAE,EAAE;gBAC9C,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC;YACjC,CAAC,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,YAAY,EAAE,CAAC,QAAiC,EAAE,EAAE;gBAC5E,MAAM,iBAAiB,GAAG,UAAqB,GAAG,UAAiB;oBACjE,MAAM,MAAM,GAAG,IAAK,QAAgB,CAAC,GAAG,UAAU,CAAC,CAAC;oBAEpD,IAAI,MAAM,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;wBACjC,MAAM,gBAAgB,GAAG,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;wBACnE,IAAI,gBAAgB,EAAE,CAAC;4BACrB,wBAAwB,CAAC,gBAAgB,CAAC,CAAC;4BAE3C,KAAK,MAAM,UAAU,IAAI,kBAAkB,EAAE,CAAC;gCAC5C,IACE,OAAO,gBAAgB,CAAC,UAAU,CAAC,KAAK,UAAU;oCAClD,CAAC,IAAA,2BAAS,EAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,EACxC,CAAC;oCACD,IAAI,CACF,gBAAgB,EAChB,UAAU,EACV,iBAAiB,CAAC,gBAAgB,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CACnE,CAAC;gCACJ,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;oBAED,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC;gBAEF,sEAAsE;gBACtE,oEAAoE;gBACpE,MAAM,CAAC,cAAc,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;gBAC3C,iBAAiB,CAAC,SAAS,GAAI,QAAgB,CAAC,SAAS,CAAC;gBAE1D,OAAO,iBAAiB,CAAC;YAC3B,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,UAAI,CAAC,KAAK,CAAC,oDAAoD,EAAE,CAAC,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,KAAK,MAAM,UAAU,IAAI,kBAAkB,EAAE,CAAC;gBAC5C,IAAI,IAAA,2BAAS,EAAC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;oBAClD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,EAAE,UAAU,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAvGD,qDAuGC"}

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

@@ -0,0 +1,35 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper, { BaseSpanAttributes } from '../base-wrapper';
+declare class AssemblyAIWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    /**
+     * Parse the request arguments of `client.transcripts.transcribe` / `submit` / `get`.
+     * AssemblyAI accepts either a params object (`{ audio, speechModel, ... }`) or, for
+     * `get`, a transcript id string. Mirrors the Python instrumentor which reads
+     * `speech_model` from kwargs (default "best").
+     */
+    static _parseAudioArgs(args: any[]): {
+        options: Record<string, any>;
+        requestModel: string;
+        audioUrl: string;
+    };
+    /**
+     * Prefer the model stamped on the Transcript response (e.g. transcripts.get
+     * polls only pass a transcript id). Fall back to request args, then "best".
+     */
+    static _resolveRequestModel(args: any[], response: any): string;
+    static _patchTranscribe(tracer: Tracer, methodName: string, sdkVersion?: string): any;
+    static _commonAudioSetter({ args, genAIEndpoint, response, span, ttft, tbt, isStream, sdkVersion, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+        ttft?: number;
+        tbt?: number;
+        isStream?: boolean;
+        sdkVersion?: string;
+    }): BaseSpanAttributes;
+}
+export default AssemblyAIWrapper;