shield360 1.0.0

package/dist/instrumentation/strands/utils.d.ts

@@ -0,0 +1,24 @@
+/**
+ * Strands Agents instrumentation utilities.
+ *
+ * Provides model-to-provider mapping, server address inference, content
+ * extraction from Strands native span events, inference event emission,
+ * and metrics recording.
+ *
+ * Mirrors: sdk/python/src/shield360/instrumentation/strands/utils.py
+ */
+export declare function inferServerAddress(modelName: string): [string, number];
+export declare function inferProviderName(modelName: string): string;
+/**
+ * Extract message content from Strands span events.
+ *
+ * Handles both legacy named events (gen_ai.user.message, gen_ai.choice, etc.)
+ * and the gen_ai.client.inference.operation.details event convention.
+ *
+ * Returns [inputMessages, outputMessages, systemInstructions, toolDefinitions].
+ */
+export declare function extractContentFromEvents(span: any, operation: string): [any[], any[], string | null, string | null];
+export declare function truncateContent(content: string): string;
+export declare function truncateMessageContent(messages: any[]): void;
+export declare function recordStrandsMetrics(operation: string, duration: number, modelName: string, serverAddress: string, serverPort: number): void;
+export declare function emitStrandsInferenceEvent(span: any, requestModel: string, serverAddress: string, serverPort: number, extra?: Record<string, any>): void;

package/dist/instrumentation/strands/utils.js

@@ -0,0 +1,373 @@
+"use strict";
+/**
+ * Strands Agents instrumentation utilities.
+ *
+ * Provides model-to-provider mapping, server address inference, content
+ * extraction from Strands native span events, inference event emission,
+ * and metrics recording.
+ *
+ * Mirrors: sdk/python/src/shield360/instrumentation/strands/utils.py
+ */
+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 });
+exports.inferServerAddress = inferServerAddress;
+exports.inferProviderName = inferProviderName;
+exports.extractContentFromEvents = extractContentFromEvents;
+exports.truncateContent = truncateContent;
+exports.truncateMessageContent = truncateMessageContent;
+exports.recordStrandsMetrics = recordStrandsMetrics;
+exports.emitStrandsInferenceEvent = emitStrandsInferenceEvent;
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importStar(require("../../helpers"));
+const metrics_1 = __importDefault(require("../../otel/metrics"));
+// -------------------------------------------------------------------------
+// Model prefix → provider mapping (mirrors Python _MODEL_PREFIX_TO_PROVIDER)
+// -------------------------------------------------------------------------
+const MODEL_PREFIX_TO_PROVIDER = [
+    ['anthropic.', 'aws.bedrock'],
+    ['amazon.', 'aws.bedrock'],
+    ['meta.', 'aws.bedrock'],
+    ['us.anthropic.', 'aws.bedrock'],
+    ['us.amazon.', 'aws.bedrock'],
+    ['us.meta.', 'aws.bedrock'],
+    ['eu.anthropic.', 'aws.bedrock'],
+    ['eu.amazon.', 'aws.bedrock'],
+    ['eu.meta.', 'aws.bedrock'],
+    ['gpt-', 'openai'],
+    ['o1', 'openai'],
+    ['o3', 'openai'],
+    ['o4', 'openai'],
+    ['claude', 'anthropic'],
+    ['gemini', 'google'],
+    ['mistral', 'mistral_ai'],
+    ['command', 'cohere'],
+    ['deepseek', 'deepseek'],
+];
+function inferServerAddress(modelName) {
+    if (!modelName)
+        return ['', 0];
+    const lower = modelName.toLowerCase();
+    for (const [prefix, provider] of MODEL_PREFIX_TO_PROVIDER) {
+        if (lower.startsWith(prefix)) {
+            return (0, helpers_1.getServerAddressForProvider)(provider);
+        }
+    }
+    return ['', 0];
+}
+function inferProviderName(modelName) {
+    if (!modelName)
+        return '';
+    const lower = modelName.toLowerCase();
+    for (const [prefix, provider] of MODEL_PREFIX_TO_PROVIDER) {
+        if (lower.startsWith(prefix)) {
+            return provider;
+        }
+    }
+    return '';
+}
+// -------------------------------------------------------------------------
+// Content extraction from Strands span events
+// -------------------------------------------------------------------------
+function safeJsonParse(value) {
+    if (typeof value === 'string') {
+        try {
+            return JSON.parse(value);
+        }
+        catch {
+            return value;
+        }
+    }
+    return value;
+}
+/**
+ * Convert Strands Bedrock-style content blocks to OTel message parts.
+ */
+function convertStrandsContentToParts(content) {
+    let blocks = safeJsonParse(content);
+    if (!Array.isArray(blocks)) {
+        blocks = blocks ? [blocks] : [];
+    }
+    const parts = [];
+    for (const block of blocks) {
+        if (typeof block === 'object' && block !== null) {
+            if ('text' in block) {
+                parts.push({ type: 'text', content: block.text });
+            }
+            else if ('toolUse' in block) {
+                const tu = block.toolUse;
+                parts.push({
+                    type: 'tool_call',
+                    id: tu.toolUseId || '',
+                    name: tu.name || '',
+                    arguments: tu.input || {},
+                });
+            }
+            else if ('toolResult' in block) {
+                const tr = block.toolResult;
+                parts.push({
+                    type: 'tool_call_response',
+                    id: tr.toolUseId || '',
+                    response: tr.content || '',
+                });
+            }
+            else {
+                for (const [key, value] of Object.entries(block)) {
+                    parts.push({ type: key, content: value });
+                }
+            }
+        }
+        else if (typeof block === 'string') {
+            parts.push({ type: 'text', content: block });
+        }
+    }
+    return parts.length > 0 ? parts : [{ type: 'text', content: String(content) }];
+}
+/**
+ * Extract message content from Strands span events.
+ *
+ * Handles both legacy named events (gen_ai.user.message, gen_ai.choice, etc.)
+ * and the gen_ai.client.inference.operation.details event convention.
+ *
+ * Returns [inputMessages, outputMessages, systemInstructions, toolDefinitions].
+ */
+function extractContentFromEvents(span, operation) {
+    const inputMsgs = [];
+    const outputMsgs = [];
+    let systemInstructions = null;
+    let toolDefinitions = null;
+    for (const event of span.events || []) {
+        const ea = event.attributes || {};
+        if (event.name === semantic_convention_1.default.GEN_AI_CLIENT_INFERENCE_OPERATION_DETAILS) {
+            if (ea[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES]) {
+                const raw = safeJsonParse(ea[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES]);
+                if (Array.isArray(raw))
+                    inputMsgs.push(...raw);
+                else if (raw)
+                    inputMsgs.push(raw);
+            }
+            if (ea[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES]) {
+                const raw = safeJsonParse(ea[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES]);
+                if (Array.isArray(raw))
+                    outputMsgs.push(...raw);
+                else if (raw)
+                    outputMsgs.push(raw);
+            }
+            if (ea[semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS]) {
+                systemInstructions = String(ea[semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS]);
+            }
+            if (ea[semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS]) {
+                toolDefinitions = String(ea[semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS]);
+            }
+            continue;
+        }
+        if (event.name === 'gen_ai.system.message') {
+            systemInstructions = String(ea.content || '');
+        }
+        else if (event.name === 'gen_ai.user.message') {
+            const content = ea.content || '';
+            const parts = convertStrandsContentToParts(content);
+            inputMsgs.push({ role: 'user', parts });
+        }
+        else if (event.name === 'gen_ai.assistant.message') {
+            const content = ea.content || '';
+            const parts = convertStrandsContentToParts(content);
+            inputMsgs.push({ role: 'assistant', parts });
+        }
+        else if (event.name === 'gen_ai.tool.message') {
+            const content = ea.content || '';
+            const toolId = ea.id || '';
+            if (operation === 'execute_tool') {
+                inputMsgs.push({
+                    role: 'tool',
+                    parts: [{
+                            type: 'tool_call',
+                            id: toolId,
+                            name: '',
+                            arguments: safeJsonParse(content),
+                        }],
+                });
+            }
+            else {
+                inputMsgs.push({
+                    role: 'tool',
+                    parts: [{
+                            type: 'tool_call_response',
+                            id: toolId,
+                            response: safeJsonParse(content),
+                        }],
+                });
+            }
+        }
+        else if (event.name === 'gen_ai.choice') {
+            const message = ea.message || '';
+            const finishReason = ea.finish_reason || '';
+            if (operation === 'execute_tool') {
+                outputMsgs.push({
+                    role: 'tool',
+                    parts: convertStrandsContentToParts(message),
+                });
+            }
+            else {
+                const parts = convertStrandsContentToParts(message);
+                const entry = { role: 'assistant', parts };
+                if (finishReason)
+                    entry.finish_reason = String(finishReason);
+                outputMsgs.push(entry);
+            }
+        }
+    }
+    return [inputMsgs, outputMsgs, systemInstructions, toolDefinitions];
+}
+// -------------------------------------------------------------------------
+// Content truncation
+// -------------------------------------------------------------------------
+function truncateContent(content) {
+    const maxLen = config_1.default.maxContentLength;
+    if (maxLen && content.length > maxLen) {
+        return content.substring(0, maxLen) + '...';
+    }
+    return content;
+}
+function truncateMessageContent(messages) {
+    for (const msg of messages) {
+        if (msg.parts && Array.isArray(msg.parts)) {
+            for (const part of msg.parts) {
+                if (part.content && typeof part.content === 'string') {
+                    part.content = truncateContent(part.content);
+                }
+                if (part.response && typeof part.response === 'string') {
+                    part.response = truncateContent(part.response);
+                }
+                if (part.arguments && typeof part.arguments === 'string') {
+                    part.arguments = truncateContent(part.arguments);
+                }
+            }
+        }
+    }
+}
+// -------------------------------------------------------------------------
+// Metrics recording (mirrors Python record_strands_metrics)
+// -------------------------------------------------------------------------
+function recordStrandsMetrics(operation, duration, modelName, serverAddress, serverPort) {
+    try {
+        const attributes = {
+            [semantic_convention_1.default.GEN_AI_OPERATION]: operation,
+            [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_STRANDS,
+            'service.name': config_1.default.applicationName || 'default',
+            'deployment.environment': config_1.default.environment || 'default',
+        };
+        if (modelName && modelName !== 'unknown') {
+            attributes[semantic_convention_1.default.GEN_AI_REQUEST_MODEL] = modelName;
+        }
+        if (serverAddress) {
+            attributes[semantic_convention_1.default.SERVER_ADDRESS] = serverAddress;
+        }
+        if (serverPort) {
+            attributes[semantic_convention_1.default.SERVER_PORT] = serverPort;
+        }
+        metrics_1.default.genaiClientOperationDuration?.record(duration, attributes);
+    }
+    catch {
+        // ignore
+    }
+}
+// -------------------------------------------------------------------------
+// Inference event emission (mirrors Python emit_strands_inference_event)
+// -------------------------------------------------------------------------
+function emitStrandsInferenceEvent(span, requestModel, serverAddress, serverPort, extra = {}) {
+    try {
+        if (config_1.default.disableEvents)
+            return;
+        const eventAttrs = {
+            [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+        };
+        if (requestModel) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_REQUEST_MODEL] = requestModel;
+        }
+        if (serverAddress) {
+            eventAttrs[semantic_convention_1.default.SERVER_ADDRESS] = serverAddress;
+        }
+        if (serverPort) {
+            eventAttrs[semantic_convention_1.default.SERVER_PORT] = serverPort;
+        }
+        if (extra.inputTokens != null) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS] = extra.inputTokens;
+        }
+        if (extra.outputTokens != null) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS] = extra.outputTokens;
+        }
+        if (extra.cacheReadInputTokens != null) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS] = extra.cacheReadInputTokens;
+        }
+        if (extra.cacheCreationInputTokens != null) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS] = extra.cacheCreationInputTokens;
+        }
+        if (extra.responseId) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_RESPONSE_ID] = extra.responseId;
+        }
+        if (extra.finishReasons) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON] = extra.finishReasons;
+        }
+        if (extra.systemInstructions) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS] = extra.systemInstructions;
+        }
+        if (extra.toolDefinitions) {
+            eventAttrs[semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS] = extra.toolDefinitions;
+        }
+        if (extra.versionExtras && typeof extra.versionExtras === 'object') {
+            for (const [k, v] of Object.entries(extra.versionExtras)) {
+                if (typeof v === 'string' && v)
+                    eventAttrs[k] = v;
+            }
+        }
+        if (config_1.default.captureMessageContent) {
+            if (extra.inputMessages) {
+                eventAttrs[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES] = JSON.stringify(extra.inputMessages);
+            }
+            if (extra.outputMessages) {
+                eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES] = JSON.stringify(extra.outputMessages);
+            }
+        }
+        helpers_1.default.emitInferenceEvent(span, eventAttrs);
+    }
+    catch {
+        // ignore
+    }
+}
+//# sourceMappingURL=utils.js.map

package/dist/instrumentation/strands/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/instrumentation/strands/utils.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiCH,gDASC;AAED,8CASC;AAmED,4DAmFC;AAMD,0CAMC;AAED,wDAgBC;AAMD,oDA4BC;AAMD,8DAmEC;AAjVD,oFAA2D;AAC3D,0DAA2C;AAC3C,yDAA6E;AAC7E,iEAAyC;AAEzC,4EAA4E;AAC5E,6EAA6E;AAC7E,4EAA4E;AAE5E,MAAM,wBAAwB,GAAuB;IACnD,CAAC,YAAY,EAAE,aAAa,CAAC;IAC7B,CAAC,SAAS,EAAE,aAAa,CAAC;IAC1B,CAAC,OAAO,EAAE,aAAa,CAAC;IACxB,CAAC,eAAe,EAAE,aAAa,CAAC;IAChC,CAAC,YAAY,EAAE,aAAa,CAAC;IAC7B,CAAC,UAAU,EAAE,aAAa,CAAC;IAC3B,CAAC,eAAe,EAAE,aAAa,CAAC;IAChC,CAAC,YAAY,EAAE,aAAa,CAAC;IAC7B,CAAC,UAAU,EAAE,aAAa,CAAC;IAC3B,CAAC,MAAM,EAAE,QAAQ,CAAC;IAClB,CAAC,IAAI,EAAE,QAAQ,CAAC;IAChB,CAAC,IAAI,EAAE,QAAQ,CAAC;IAChB,CAAC,IAAI,EAAE,QAAQ,CAAC;IAChB,CAAC,QAAQ,EAAE,WAAW,CAAC;IACvB,CAAC,QAAQ,EAAE,QAAQ,CAAC;IACpB,CAAC,SAAS,EAAE,YAAY,CAAC;IACzB,CAAC,SAAS,EAAE,QAAQ,CAAC;IACrB,CAAC,UAAU,EAAE,UAAU,CAAC;CACzB,CAAC;AAEF,SAAgB,kBAAkB,CAAC,SAAiB;IAClD,IAAI,CAAC,SAAS;QAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAC/B,MAAM,KAAK,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IACtC,KAAK,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,wBAAwB,EAAE,CAAC;QAC1D,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7B,OAAO,IAAA,qCAA2B,EAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IACD,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;AACjB,CAAC;AAED,SAAgB,iBAAiB,CAAC,SAAiB;IACjD,IAAI,CAAC,SAAS;QAAE,OAAO,EAAE,CAAC;IAC1B,MAAM,KAAK,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IACtC,KAAK,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,wBAAwB,EAAE,CAAC;QAC1D,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7B,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,4EAA4E;AAC5E,8CAA8C;AAC9C,4EAA4E;AAE5E,SAAS,aAAa,CAAC,KAAU;IAC/B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,4BAA4B,CAAC,OAAY;IAChD,IAAI,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACpC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAClC,CAAC;IAED,MAAM,KAAK,GAAU,EAAE,CAAC;IACxB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAChD,IAAI,MAAM,IAAI,KAAK,EAAE,CAAC;gBACpB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YACpD,CAAC;iBAAM,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;gBAC9B,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC;gBACzB,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,WAAW;oBACjB,EAAE,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE;oBACtB,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,EAAE;oBACnB,SAAS,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE;iBAC1B,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;gBACjC,MAAM,EAAE,GAAG,KAAK,CAAC,UAAU,CAAC;gBAC5B,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,oBAAoB;oBAC1B,EAAE,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE;oBACtB,QAAQ,EAAE,EAAE,CAAC,OAAO,IAAI,EAAE;iBAC3B,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;QACH,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACrC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;AACjF,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,wBAAwB,CACtC,IAAS,EACT,SAAiB;IAEjB,MAAM,SAAS,GAAU,EAAE,CAAC;IAC5B,MAAM,UAAU,GAAU,EAAE,CAAC;IAC7B,IAAI,kBAAkB,GAAkB,IAAI,CAAC;IAC7C,IAAI,eAAe,GAAkB,IAAI,CAAC;IAE1C,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;QACtC,MAAM,EAAE,GAAG,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC;QAElC,IAAI,KAAK,CAAC,IAAI,KAAK,6BAAkB,CAAC,yCAAyC,EAAE,CAAC;YAChF,IAAI,EAAE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBACjD,MAAM,GAAG,GAAG,aAAa,CAAC,EAAE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBACxE,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;oBAAE,SAAS,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC;qBAC1C,IAAI,GAAG;oBAAE,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACpC,CAAC;YACD,IAAI,EAAE,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,EAAE,CAAC;gBAClD,MAAM,GAAG,GAAG,aAAa,CAAC,EAAE,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,CAAC,CAAC;gBACzE,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;oBAAE,UAAU,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC;qBAC3C,IAAI,GAAG;oBAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,CAAC;YACD,IAAI,EAAE,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,CAAC;gBACtD,kBAAkB,GAAG,MAAM,CAAC,EAAE,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC;YACjF,CAAC;YACD,IAAI,EAAE,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,EAAE,CAAC;gBACnD,eAAe,GAAG,MAAM,CAAC,EAAE,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC;YAC3E,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;YAC3C,kBAAkB,GAAG,MAAM,CAAC,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,4BAA4B,CAAC,OAAO,CAAC,CAAC;YACpD,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1C,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;YACrD,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,4BAA4B,CAAC,OAAO,CAAC,CAAC;YACpD,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/C,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC;YAC3B,IAAI,SAAS,KAAK,cAAc,EAAE,CAAC;gBACjC,SAAS,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC;4BACN,IAAI,EAAE,WAAW;4BACjB,EAAE,EAAE,MAAM;4BACV,IAAI,EAAE,EAAE;4BACR,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC;yBAClC,CAAC;iBACH,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC;4BACN,IAAI,EAAE,oBAAoB;4BAC1B,EAAE,EAAE,MAAM;4BACV,QAAQ,EAAE,aAAa,CAAC,OAAO,CAAC;yBACjC,CAAC;iBACH,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YAC1C,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC;YACjC,MAAM,YAAY,GAAG,EAAE,CAAC,aAAa,IAAI,EAAE,CAAC;YAC5C,IAAI,SAAS,KAAK,cAAc,EAAE,CAAC;gBACjC,UAAU,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,4BAA4B,CAAC,OAAO,CAAC;iBAC7C,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,GAAG,4BAA4B,CAAC,OAAO,CAAC,CAAC;gBACpD,MAAM,KAAK,GAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;gBAChD,IAAI,YAAY;oBAAE,KAAK,CAAC,aAAa,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;gBAC7D,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,kBAAkB,EAAE,eAAe,CAAC,CAAC;AACtE,CAAC;AAED,4EAA4E;AAC5E,qBAAqB;AACrB,4EAA4E;AAE5E,SAAgB,eAAe,CAAC,OAAe;IAC7C,MAAM,MAAM,GAAG,gBAAe,CAAC,gBAAgB,CAAC;IAChD,IAAI,MAAM,IAAI,OAAO,CAAC,MAAM,GAAG,MAAM,EAAE,CAAC;QACtC,OAAO,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,KAAK,CAAC;IAC9C,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAgB,sBAAsB,CAAC,QAAe;IACpD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBAC7B,IAAI,IAAI,CAAC,OAAO,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACrD,IAAI,CAAC,OAAO,GAAG,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/C,CAAC;gBACD,IAAI,IAAI,CAAC,QAAQ,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACvD,IAAI,CAAC,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACjD,CAAC;gBACD,IAAI,IAAI,CAAC,SAAS,IAAI,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACzD,IAAI,CAAC,SAAS,GAAG,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,4EAA4E;AAC5E,4DAA4D;AAC5D,4EAA4E;AAE5E,SAAgB,oBAAoB,CAClC,SAAiB,EACjB,QAAgB,EAChB,SAAiB,EACjB,aAAqB,EACrB,UAAkB;IAElB,IAAI,CAAC;QACH,MAAM,UAAU,GAAe;YAC7B,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,SAAS;YAChD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,6BAAkB,CAAC,qBAAqB;YACxF,cAAc,EAAE,gBAAe,CAAC,eAAe,IAAI,SAAS;YAC5D,wBAAwB,EAAE,gBAAe,CAAC,WAAW,IAAI,SAAS;SACnE,CAAC;QACF,IAAI,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACzC,UAAU,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,GAAG,SAAS,CAAC;QAClE,CAAC;QACD,IAAI,aAAa,EAAE,CAAC;YAClB,UAAU,CAAC,6BAAkB,CAAC,cAAc,CAAC,GAAG,aAAa,CAAC;QAChE,CAAC;QACD,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,6BAAkB,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC;QAC1D,CAAC;QAED,iBAAO,CAAC,4BAA4B,EAAE,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IACrE,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;AACH,CAAC;AAED,4EAA4E;AAC5E,yEAAyE;AACzE,4EAA4E;AAE5E,SAAgB,yBAAyB,CACvC,IAAS,EACT,YAAoB,EACpB,aAAqB,EACrB,UAAkB,EAClB,QAA6B,EAAE;IAE/B,IAAI,CAAC;QACH,IAAI,gBAAe,CAAC,aAAa;YAAE,OAAO;QAE1C,MAAM,UAAU,GAAe;YAC7B,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;SACrF,CAAC;QAEF,IAAI,YAAY,EAAE,CAAC;YACjB,UAAU,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,GAAG,YAAY,CAAC;QACrE,CAAC;QACD,IAAI,aAAa,EAAE,CAAC;YAClB,UAAU,CAAC,6BAAkB,CAAC,cAAc,CAAC,GAAG,aAAa,CAAC;QAChE,CAAC;QACD,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,6BAAkB,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC;QAC1D,CAAC;QAED,IAAI,KAAK,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;YAC9B,UAAU,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC;QAC/E,CAAC;QACD,IAAI,KAAK,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;YAC/B,UAAU,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,GAAG,KAAK,CAAC,YAAY,CAAC;QACjF,CAAC;QACD,IAAI,KAAK,CAAC,oBAAoB,IAAI,IAAI,EAAE,CAAC;YACvC,UAAU,CAAC,6BAAkB,CAAC,oCAAoC,CAAC,GAAG,KAAK,CAAC,oBAAoB,CAAC;QACnG,CAAC;QACD,IAAI,KAAK,CAAC,wBAAwB,IAAI,IAAI,EAAE,CAAC;YAC3C,UAAU,CAAC,6BAAkB,CAAC,wCAAwC,CAAC,GAAG,KAAK,CAAC,wBAAwB,CAAC;QAC3G,CAAC;QACD,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACrB,UAAU,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,GAAG,KAAK,CAAC,UAAU,CAAC;QACvE,CAAC;QACD,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;YACxB,UAAU,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,GAAG,KAAK,CAAC,aAAa,CAAC;QACrF,CAAC;QACD,IAAI,KAAK,CAAC,kBAAkB,EAAE,CAAC;YAC7B,UAAU,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,GAAG,KAAK,CAAC,kBAAkB,CAAC;QACvF,CAAC;QACD,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC1B,UAAU,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,GAAG,KAAK,CAAC,eAAe,CAAC;QACjF,CAAC;QACD,IAAI,KAAK,CAAC,aAAa,IAAI,OAAO,KAAK,CAAC,aAAa,KAAK,QAAQ,EAAE,CAAC;YACnE,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzD,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC;oBAAE,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;QAED,IAAI,gBAAe,CAAC,qBAAqB,EAAE,CAAC;YAC1C,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACxB,UAAU,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC7F,CAAC;YACD,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBACzB,UAAU,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YAC/F,CAAC;QACH,CAAC;QAED,iBAAe,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IACvD,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;AACH,CAAC"}

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

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

package/dist/instrumentation/together/index.js

@@ -0,0 +1,43 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+class Shield360TogetherInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-together`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('together-ai', ['>=0.1.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(together) {
+        this._patch(together);
+    }
+    _patch(moduleExports) {
+        try {
+            if ((0, instrumentation_1.isWrapped)(moduleExports.Together.Chat.Completions.prototype.create)) {
+                this._unwrap(moduleExports.Together.Chat.Completions.prototype, 'create');
+            }
+            this._wrap(moduleExports.Together.Chat.Completions.prototype, 'create', wrapper_1.default._patchChatCompletionCreate(this.tracer));
+        }
+        catch (e) {
+            console.error('Error in _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        this._unwrap(moduleExports.Together.Chat.Completions.prototype, 'create');
+    }
+}
+exports.default = Shield360TogetherInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/together/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAwC;AAIxC,MAAqB,gCAAiC,SAAQ,qCAAmB;IAC/E,YAAY,SAAwC,EAAE;QACpD,KAAK,CAAC,GAAG,iCAAsB,2BAA2B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,aAAa,EACb,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,QAAa;QAC9B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,IAAI,IAAA,2BAAS,EAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;gBACxE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAC5E,CAAC;YAED,IAAI,CAAC,KAAK,CACR,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,EACjD,QAAQ,EACR,iBAAe,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CACxD,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC5E,CAAC;CACF;AA7CD,mDA6CC"}

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

@@ -0,0 +1,48 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class TogetherWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    /**
+     * Stamp `shield360.agent.version_hash` (auto) and `gen_ai.agent.version`
+     * (user override, if set) on the span and return the same attributes so
+     * the caller can merge them into the inference event extras.
+     */
+    static _stampAgentVersion(span: Span, args: {
+        systemInstructionsJson?: string;
+        toolDefinitionsJson?: string;
+        primaryModel?: string;
+        temperature?: number | null;
+        top_p?: number | null;
+        max_tokens?: number | null;
+    }): Record<string, string>;
+    static _patchChatCompletionCreate(tracer: Tracer): any;
+    static _chatCompletion({ args, genAIEndpoint, response, span, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+    }): Promise<any>;
+    static _chatCompletionGenerator({ args, genAIEndpoint, response, span, }: {
+        args: any[];
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+    }): AsyncGenerator<unknown, any, unknown>;
+    static _chatCompletionCommonSetter({ 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;
+    }>;
+}
+export default TogetherWrapper;