shield360 1.0.0

package/dist/instrumentation/base-wrapper.js

@@ -0,0 +1,139 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const config_1 = __importDefault(require("../config"));
+const constant_1 = require("../constant");
+const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const semantic_convention_1 = __importDefault(require("../semantic-convention"));
+const api_1 = require("@opentelemetry/api");
+const metrics_1 = __importDefault(require("../otel/metrics"));
+const helpers_1 = require("../helpers");
+class BaseWrapper {
+    static setBaseSpanAttributes(span, { genAIEndpoint: _genAIEndpoint, model, user, cost, aiSystem, serverAddress, serverPort }) {
+        const applicationName = config_1.default.applicationName;
+        const environment = config_1.default.environment;
+        if (!applicationName) {
+            throw new Error("[Shield360] Shield360Config.applicationName is not set. Please check your configuration.");
+        }
+        if (!environment) {
+            throw new Error("[Shield360] Shield360Config.environment is not set. Please check your configuration.");
+        }
+        span.setAttribute(semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME, constant_1.SDK_NAME);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL, aiSystem);
+        span.setAttribute(semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT, environment);
+        span.setAttribute(semantic_conventions_1.ATTR_SERVICE_NAME, applicationName);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, model);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, constant_1.SDK_VERSION);
+        if (serverAddress) {
+            span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, serverAddress);
+        }
+        if (serverPort !== undefined) {
+            span.setAttribute(semantic_convention_1.default.SERVER_PORT, serverPort);
+        }
+        if (typeof user === 'string' || typeof user === 'number') {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_USER, user);
+        }
+        if (cost !== undefined) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_COST, cost);
+        }
+        (0, helpers_1.applyCustomSpanAttributes)(span);
+        span.setStatus({ code: api_1.SpanStatusCode.OK });
+    }
+    static recordMetrics(span, baseAttributes) {
+        const applicationName = config_1.default.applicationName;
+        const environment = config_1.default.environment;
+        const { model, aiSystem, cost, errorType } = baseAttributes;
+        const inputTokens = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS);
+        const outputTokens = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS);
+        const duration = BaseWrapper.getSpanAttribute(span, 'duration') ?? BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.GEN_AI_DURATION_LEGACY);
+        const operationName = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.GEN_AI_OPERATION);
+        const responseModel = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.GEN_AI_RESPONSE_MODEL);
+        const serverAddress = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.SERVER_ADDRESS);
+        const serverPort = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.SERVER_PORT);
+        const attributes = {
+            [semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]: constant_1.SDK_NAME,
+            [semantic_conventions_1.ATTR_SERVICE_NAME]: applicationName,
+            [semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]: environment,
+            [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: aiSystem,
+            [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: model,
+        };
+        if (operationName)
+            attributes[semantic_convention_1.default.GEN_AI_OPERATION] = operationName;
+        if (responseModel)
+            attributes[semantic_convention_1.default.GEN_AI_RESPONSE_MODEL] = responseModel;
+        if (serverAddress)
+            attributes[semantic_convention_1.default.SERVER_ADDRESS] = serverAddress;
+        if (serverPort !== undefined)
+            attributes[semantic_convention_1.default.SERVER_PORT] = serverPort;
+        if (errorType)
+            attributes[semantic_convention_1.default.ERROR_TYPE] = errorType;
+        if (Number.isFinite(inputTokens)) {
+            metrics_1.default.genaiClientUsageTokens?.record(inputTokens, {
+                ...attributes,
+                [semantic_convention_1.default.GEN_AI_TOKEN_TYPE]: semantic_convention_1.default.GEN_AI_TOKEN_TYPE_INPUT,
+            });
+        }
+        if (Number.isFinite(outputTokens)) {
+            metrics_1.default.genaiClientUsageTokens?.record(outputTokens, {
+                ...attributes,
+                [semantic_convention_1.default.GEN_AI_TOKEN_TYPE]: semantic_convention_1.default.GEN_AI_TOKEN_TYPE_OUTPUT,
+            });
+        }
+        if (Number.isFinite(duration)) {
+            metrics_1.default.genaiClientOperationDuration?.record(duration / 1e9, attributes);
+        }
+        const tbt = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.GEN_AI_SERVER_TBT);
+        if (Number.isFinite(tbt))
+            metrics_1.default.genaiServerTbt?.record(tbt, attributes);
+        const ttft = BaseWrapper.getSpanAttribute(span, semantic_convention_1.default.GEN_AI_SERVER_TTFT);
+        if (Number.isFinite(ttft))
+            metrics_1.default.genaiServerTtft?.record(ttft, attributes);
+        if (Number.isFinite(ttft) && ttft > 0) {
+            metrics_1.default.genaiClientTimeToFirstChunk?.record(ttft, attributes);
+        }
+        if (Number.isFinite(tbt) && tbt > 0) {
+            metrics_1.default.genaiClientTimePerOutputChunk?.record(tbt, attributes);
+            const outputTokensVal = Number.isFinite(outputTokens) ? outputTokens : 0;
+            const serverRequestDuration = ttft + tbt * Math.max(outputTokensVal - 1, 0);
+            metrics_1.default.genaiServerRequestDuration?.record(serverRequestDuration, attributes);
+        }
+        if (cost !== undefined) {
+            const numericCost = typeof cost === 'number' ? cost : Number(cost);
+            if (Number.isFinite(numericCost)) {
+                metrics_1.default.genaiCost?.record(numericCost, attributes);
+            }
+        }
+    }
+    static getSpanAttribute(span, key) {
+        if (key === 'duration') {
+            // Use duration if present, even if 0
+            const s = span;
+            if (s.attributes && typeof s.attributes.duration !== 'undefined') {
+                const attrDuration = s.attributes.duration;
+                if (typeof attrDuration === 'number' && !isNaN(attrDuration)) {
+                    return attrDuration;
+                }
+            }
+            if (typeof s.duration === 'number' && !isNaN(s.duration))
+                return s.duration;
+            if (typeof s._duration === 'number' && !isNaN(s._duration))
+                return s._duration;
+            if (s.endTime && s.startTime) {
+                const [endSec, endNano] = s.endTime;
+                const [startSec, startNano] = s.startTime;
+                const end = endSec * 1e9 + endNano;
+                const start = startSec * 1e9 + startNano;
+                if (end > start) {
+                    return end - start;
+                }
+            }
+            return undefined;
+        }
+        // @ts-expect-error: OpenTelemetry Span may have attributes property in some implementations
+        return typeof span.attributes === 'object' ? span.attributes[key] : undefined;
+    }
+}
+exports.default = BaseWrapper;
+//# sourceMappingURL=base-wrapper.js.map

package/dist/instrumentation/base-wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base-wrapper.js","sourceRoot":"","sources":["../../src/instrumentation/base-wrapper.ts"],"names":[],"mappings":";;;;;AAAA,uDAAwC;AACxC,0CAAoD;AACpD,8EAAiG;AACjG,iFAAwD;AACxD,4CAAsE;AACtE,8DAAsC;AACtC,wCAAuD;AAavD,MAAqB,WAAW;IAC9B,MAAM,CAAC,qBAAqB,CAC1B,IAAU,EACV,EAAE,aAAa,EAAE,cAAc,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,UAAU,EAAsB;QAE7G,MAAM,eAAe,GAAG,gBAAe,CAAC,eAAe,CAAC;QACxD,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,CAAC;QAEhD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,0FAA0F,CAAC,CAAC;QAC9G,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,sFAAsF,CAAC,CAAC;QAC1G,CAAC;QAGD,IAAI,CAAC,YAAY,CAAC,8CAAuB,EAAE,mBAAQ,CAAC,CAAC;QACrD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,QAAQ,CAAC,CAAC;QAC1E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,WAAW,CAAC,CAAC;QAC/E,IAAI,CAAC,YAAY,CAAC,wCAAiB,EAAE,eAAe,CAAC,CAAC;QACtD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,sBAAW,CAAC,CAAC;QACtE,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YACzD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAChE,CAAC;QACD,IAAA,mCAAyB,EAAC,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,IAAU,EAAE,cAAkC;QACjE,MAAM,eAAe,GAAG,gBAAe,CAAC,eAAgB,CAAC;QACzD,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAY,CAAC;QACjD,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,cAAc,CAAC;QAE5D,MAAM,WAAW,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;QACrG,MAAM,YAAY,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;QACvG,MAAM,QAAQ,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,sBAAsB,CAAC,CAAC;QACjJ,MAAM,aAAa,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,gBAAgB,CAAsB,CAAC;QACnH,MAAM,aAAa,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,qBAAqB,CAAsB,CAAC;QACxH,MAAM,aAAa,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,cAAc,CAAsB,CAAC;QACjH,MAAM,UAAU,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,WAAW,CAAsB,CAAC;QAC3G,MAAM,UAAU,GAAe;YAC7B,CAAC,8CAAuB,CAAC,EAAE,mBAAQ;YACnC,CAAC,wCAAiB,CAAC,EAAE,eAAe;YACpC,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,EAAE,WAAW;YAC7D,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,QAAQ;YACxD,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,KAAK;SACjD,CAAC;QACF,IAAI,aAAa;YAAE,UAAU,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,GAAG,aAAa,CAAC;QACnF,IAAI,aAAa;YAAE,UAAU,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,GAAG,aAAa,CAAC;QACxF,IAAI,aAAa;YAAE,UAAU,CAAC,6BAAkB,CAAC,cAAc,CAAC,GAAG,aAAa,CAAC;QACjF,IAAI,UAAU,KAAK,SAAS;YAAE,UAAU,CAAC,6BAAkB,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC;QACtF,IAAI,SAAS;YAAE,UAAU,CAAC,6BAAkB,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC;QAErE,IAAI,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACjC,iBAAO,CAAC,sBAAsB,EAAE,MAAM,CAAC,WAAqB,EAAE;gBAC5D,GAAG,UAAU;gBACb,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,6BAAkB,CAAC,uBAAuB;aACnF,CAAC,CAAC;QACL,CAAC;QACD,IAAI,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAClC,iBAAO,CAAC,sBAAsB,EAAE,MAAM,CAAC,YAAsB,EAAE;gBAC7D,GAAG,UAAU;gBACb,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,6BAAkB,CAAC,wBAAwB;aACpF,CAAC,CAAC;QACL,CAAC;QACD,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9B,iBAAO,CAAC,4BAA4B,EAAE,MAAM,CAAE,QAAmB,GAAG,GAAG,EAAE,UAAU,CAAC,CAAC;QACvF,CAAC;QACD,MAAM,GAAG,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,iBAAiB,CAAC,CAAC;QACrF,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,iBAAO,CAAC,cAAc,EAAE,MAAM,CAAC,GAAa,EAAE,UAAU,CAAC,CAAC;QACpF,MAAM,IAAI,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,EAAE,6BAAkB,CAAC,kBAAkB,CAAC,CAAC;QACvF,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,iBAAO,CAAC,eAAe,EAAE,MAAM,CAAC,IAAc,EAAE,UAAU,CAAC,CAAC;QACvF,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAK,IAAe,GAAG,CAAC,EAAE,CAAC;YAClD,iBAAO,CAAC,2BAA2B,EAAE,MAAM,CAAC,IAAc,EAAE,UAAU,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAK,GAAc,GAAG,CAAC,EAAE,CAAC;YAChD,iBAAO,CAAC,6BAA6B,EAAE,MAAM,CAAC,GAAa,EAAE,UAAU,CAAC,CAAC;YACzE,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAE,YAAuB,CAAC,CAAC,CAAC,CAAC,CAAC;YACrF,MAAM,qBAAqB,GAAI,IAAe,GAAI,GAAc,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACpG,iBAAO,CAAC,0BAA0B,EAAE,MAAM,CAAC,qBAAqB,EAAE,UAAU,CAAC,CAAC;QAChF,CAAC;QACD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,WAAW,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACnE,IAAI,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBACjC,iBAAO,CAAC,SAAS,EAAE,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,CAAC,gBAAgB,CAAC,IAAU,EAAE,GAAW;QAC7C,IAAI,GAAG,KAAK,UAAU,EAAE,CAAC;YACvB,qCAAqC;YACrC,MAAM,CAAC,GAAG,IAMT,CAAC;YAEF,IAAI,CAAC,CAAC,UAAU,IAAI,OAAO,CAAC,CAAC,UAAU,CAAC,QAAQ,KAAK,WAAW,EAAE,CAAC;gBACjE,MAAM,YAAY,GAAG,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC;gBAC3C,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;oBAC7D,OAAO,YAAY,CAAC;gBACtB,CAAC;YACH,CAAC;YAED,IAAI,OAAO,CAAC,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;gBAAE,OAAO,CAAC,CAAC,QAAQ,CAAC;YAC5E,IAAI,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;gBAAE,OAAO,CAAC,CAAC,SAAS,CAAC;YAE/E,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;gBAC7B,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;gBACpC,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC;gBAC1C,MAAM,GAAG,GAAG,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC;gBACnC,MAAM,KAAK,GAAG,QAAQ,GAAG,GAAG,GAAG,SAAS,CAAC;gBACzC,IAAI,GAAG,GAAG,KAAK,EAAE,CAAC;oBAChB,OAAO,GAAG,GAAG,KAAK,CAAC;gBACrB,CAAC;YACH,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,4FAA4F;QAC5F,OAAO,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChF,CAAC;CACF;AAtID,8BAsIC"}

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

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

package/dist/instrumentation/bedrock/index.js

@@ -0,0 +1,52 @@
+"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 Shield360BedrockInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-bedrock`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@aws-sdk/client-bedrock-runtime', ['>=3.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(bedrock) {
+        this._patch(bedrock);
+    }
+    _patch(moduleExports) {
+        try {
+            const BedrockRuntimeClient = moduleExports.BedrockRuntimeClient;
+            if (!BedrockRuntimeClient?.prototype)
+                return;
+            if ((0, instrumentation_1.isWrapped)(BedrockRuntimeClient.prototype.send)) {
+                this._unwrap(BedrockRuntimeClient.prototype, 'send');
+            }
+            this._wrap(BedrockRuntimeClient.prototype, 'send', wrapper_1.default._patchSend(this.tracer));
+        }
+        catch (e) {
+            console.error('Error in Bedrock _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const BedrockRuntimeClient = moduleExports.BedrockRuntimeClient;
+            if (BedrockRuntimeClient?.prototype?.send) {
+                this._unwrap(BedrockRuntimeClient.prototype, 'send');
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = Shield360BedrockInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/bedrock/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAuC;AAIvC,MAAqB,+BAAgC,SAAQ,qCAAmB;IAC9E,YAAY,SAAuC,EAAE;QACnD,KAAK,CAAC,GAAG,iCAAsB,0BAA0B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC9E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,iCAAiC,EACjC,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,OAAY;QAC7B,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,MAAM,oBAAoB,GAAG,aAAa,CAAC,oBAAoB,CAAC;YAChE,IAAI,CAAC,oBAAoB,EAAE,SAAS;gBAAE,OAAO;YAE7C,IAAI,IAAA,2BAAS,EAAC,oBAAoB,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnD,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACvD,CAAC;YAED,IAAI,CAAC,KAAK,CACR,oBAAoB,CAAC,SAAS,EAC9B,MAAM,EACN,iBAAc,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CACvC,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,oBAAoB,GAAG,aAAa,CAAC,oBAAoB,CAAC;YAChE,IAAI,oBAAoB,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;gBAC1C,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AArDD,kDAqDC"}

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

@@ -0,0 +1,41 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper, { BaseSpanAttributes } from '../base-wrapper';
+declare class BedrockWrapper 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 _patchSend(tracer: Tracer): any;
+    static _handleConverseCommand(tracer: Tracer, originalMethod: any, instance: any, args: any[]): Promise<any>;
+    static _converseComplete({ input, genAIEndpoint, response, span, modelId, }: {
+        input: any;
+        genAIEndpoint: string;
+        response: any;
+        span: Span;
+        modelId: string;
+    }): Promise<any>;
+    static _handleConverseStreamCommand(tracer: Tracer, originalMethod: any, instance: any, args: any[]): Promise<any>;
+    static _converseCommonSetter({ input, genAIEndpoint, result, span, modelId, isStream, ttft, tbt, }: {
+        input: any;
+        genAIEndpoint: string;
+        result: any;
+        span: Span;
+        modelId: string;
+        isStream: boolean;
+        ttft?: number;
+        tbt?: number;
+    }): BaseSpanAttributes;
+}
+export default BedrockWrapper;