shield360 1.0.0

package/dist/instrumentation/assemblyai/wrapper.js

@@ -0,0 +1,238 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const api_1 = require("@opentelemetry/api");
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importStar(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+function spanCreationAttrs(operationName, requestModel, serverAddress, serverPort) {
+    return {
+        [semantic_convention_1.default.GEN_AI_OPERATION]: operationName,
+        [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: AssemblyAIWrapper.aiSystem,
+        [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+        [semantic_convention_1.default.SERVER_ADDRESS]: serverAddress,
+        [semantic_convention_1.default.SERVER_PORT]: serverPort,
+    };
+}
+class AssemblyAIWrapper extends base_wrapper_1.default {
+    /**
+     * 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) {
+        const options = typeof args[0] === 'object' && args[0] !== null ? args[0] : {};
+        const requestModel = options.speechModel || options.speech_model || 'best';
+        const rawAudioUrl = typeof args[0] === 'string'
+            ? args[0]
+            : options.audio ?? options.audioUrl ?? options.audio_url;
+        // Only stringify when a real value is present, so undefined/null never
+        // becomes the literal string 'undefined'/'null'.
+        const audioUrl = rawAudioUrl === undefined || rawAudioUrl === null ? '' : String(rawAudioUrl);
+        return { options, requestModel, audioUrl };
+    }
+    /**
+     * 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, response) {
+        const { requestModel } = AssemblyAIWrapper._parseAudioArgs(args);
+        const responseModel = response?.speech_model ??
+            response?.speechModel ??
+            response?.config?.speech_model ??
+            response?.config?.speechModel;
+        if (typeof responseModel === 'string' && responseModel.length > 0) {
+            return responseModel;
+        }
+        return requestModel;
+    }
+    static _patchTranscribe(tracer, methodName, sdkVersion) {
+        const genAIEndpoint = `assemblyai.transcripts.${methodName}`;
+        return (originalMethod) => {
+            return async function (...args) {
+                if ((0, helpers_1.isFrameworkLlmActive)())
+                    return originalMethod.apply(this, args);
+                const { requestModel } = AssemblyAIWrapper._parseAudioArgs(args);
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AUDIO} ${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_AUDIO, requestModel, AssemblyAIWrapper.serverAddress, AssemblyAIWrapper.serverPort),
+                }, effectiveCtx);
+                return api_1.context.with(api_1.trace.setSpan(effectiveCtx, span), async () => {
+                    let metricParams;
+                    const startTime = Date.now();
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const ttft = (Date.now() - startTime) / 1000;
+                        metricParams = AssemblyAIWrapper._commonAudioSetter({
+                            args,
+                            genAIEndpoint,
+                            response,
+                            span,
+                            ttft,
+                            tbt: 0,
+                            isStream: false,
+                            sdkVersion,
+                        });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        base_wrapper_1.default.recordMetrics(span, {
+                            genAIEndpoint,
+                            model: requestModel,
+                            aiSystem: AssemblyAIWrapper.aiSystem,
+                            serverAddress: AssemblyAIWrapper.serverAddress,
+                            serverPort: AssemblyAIWrapper.serverPort,
+                            errorType: e?.constructor?.name || '_OTHER',
+                        });
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                        if (metricParams) {
+                            base_wrapper_1.default.recordMetrics(span, metricParams);
+                        }
+                    }
+                });
+            };
+        };
+    }
+    static _commonAudioSetter({ args, genAIEndpoint, response, span, ttft = 0, tbt = 0, isStream = false, sdkVersion, }) {
+        const captureContent = config_1.default.captureMessageContent;
+        const { audioUrl } = AssemblyAIWrapper._parseAudioArgs(args);
+        const requestModel = AssemblyAIWrapper._resolveRequestModel(args, response);
+        // Coerce to a string only when a real value is present, so undefined/null
+        // never becomes the literal string 'undefined'/'null'.
+        const toStr = (v) => v === undefined || v === null ? '' : String(v);
+        // Prefer values from the returned Transcript (matches the Python instrumentor,
+        // which reads audio_url / audio_duration / id / text off the response).
+        // Do not treat a bare transcript id (transcripts.get) as an audio URL for
+        // cost or message content — Python uses response.audio_url only.
+        const argsAudioPrompt = typeof args[0] === 'object' && args[0] !== null ? audioUrl : '';
+        const prompt = toStr(response?.audio_url ?? response?.audioUrl ?? argsAudioPrompt);
+        const audioDuration = response?.audio_duration ?? response?.audioDuration ?? 0;
+        const responseId = toStr(response?.id);
+        const responseText = toStr(response?.text);
+        // Request parameters
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_AUDIO_DURATION, audioDuration);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, isStream);
+        // Response parameters
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, 'text');
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, responseId);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+        // Token usage (AssemblyAI transcription does not report token usage)
+        const usage = response?.usage;
+        const inputTokens = (usage && (usage.input_tokens ?? usage.inputTokens)) || 0;
+        const outputTokens = (usage && (usage.output_tokens ?? usage.outputTokens)) || 0;
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, inputTokens + outputTokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, 0);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, 0);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TBT, tbt);
+        const pricingInfo = config_1.default.pricingInfo || {};
+        // AssemblyAI bills per second of audio (pricing.json best/nano rates).
+        // Do not pass audio_url as the prompt — getAudioModelCost treats a
+        // truthy prompt as character-length pricing (TTS-style), which made
+        // gen_ai.usage.cost track URL length instead of duration. Match Python
+        // assemblyai/utils.py: pass an empty prompt and the duration.
+        const cost = helpers_1.default.getAudioModelCost(requestModel, pricingInfo, '', audioDuration);
+        AssemblyAIWrapper.setBaseSpanAttributes(span, {
+            genAIEndpoint,
+            model: requestModel,
+            cost,
+            aiSystem: AssemblyAIWrapper.aiSystem,
+            serverAddress: AssemblyAIWrapper.serverAddress,
+            serverPort: AssemblyAIWrapper.serverPort,
+        });
+        // Python stamps gen_ai.system and the AssemblyAI package version (not Shield360's).
+        span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, AssemblyAIWrapper.aiSystem);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, requestModel);
+        if (sdkVersion) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, sdkVersion);
+        }
+        let inputMessagesJson;
+        let outputMessagesJson;
+        if (captureContent) {
+            inputMessagesJson = helpers_1.default.buildInputMessages([{ role: 'user', content: prompt }]);
+            outputMessagesJson = helpers_1.default.buildOutputMessages(responseText, 'stop');
+            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, inputMessagesJson);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
+        }
+        // Match Python assemblyai: inference events are emitted only when content
+        // capture is enabled (messages are part of the event payload).
+        if (!config_1.default.disableEvents && captureContent) {
+            const eventAttrs = {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AUDIO,
+                [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: requestModel,
+                [semantic_convention_1.default.SERVER_ADDRESS]: AssemblyAIWrapper.serverAddress,
+                [semantic_convention_1.default.SERVER_PORT]: AssemblyAIWrapper.serverPort,
+                [semantic_convention_1.default.GEN_AI_OUTPUT_TYPE]: 'text',
+                [semantic_convention_1.default.GEN_AI_RESPONSE_ID]: responseId,
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: inputTokens,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: outputTokens,
+            };
+            if (inputMessagesJson) {
+                eventAttrs[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES] = inputMessagesJson;
+            }
+            if (outputMessagesJson) {
+                eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES] = outputMessagesJson;
+            }
+            helpers_1.default.emitInferenceEvent(span, eventAttrs);
+        }
+        return {
+            genAIEndpoint,
+            model: requestModel,
+            cost,
+            aiSystem: AssemblyAIWrapper.aiSystem,
+            serverAddress: AssemblyAIWrapper.serverAddress,
+            serverPort: AssemblyAIWrapper.serverPort,
+        };
+    }
+}
+AssemblyAIWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_ASSEMBLYAI;
+AssemblyAIWrapper.serverAddress = 'api.assemblyai.com';
+AssemblyAIWrapper.serverPort = 443;
+exports.default = AssemblyAIWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/assemblyai/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAwF;AACxF,0DAA2C;AAC3C,yDAGuB;AACvB,oFAA2D;AAC3D,mEAAkE;AAElE,SAAS,iBAAiB,CACxB,aAAqB,EACrB,YAAoB,EACpB,aAAqB,EACrB,UAAkB;IAElB,OAAO;QACL,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,aAAa;QACpD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,iBAAiB,CAAC,QAAQ;QAC1E,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;QACvD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,aAAa;QAClD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,UAAU;KAC7C,CAAC;AACJ,CAAC;AAED,MAAM,iBAAkB,SAAQ,sBAAW;IAKzC;;;;;OAKG;IACH,MAAM,CAAC,eAAe,CAAC,IAAW;QAKhC,MAAM,OAAO,GACX,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACjE,MAAM,YAAY,GAChB,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,YAAY,IAAI,MAAM,CAAC;QACxD,MAAM,WAAW,GACf,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ;YACzB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACT,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,SAAS,CAAC;QAE7D,uEAAuE;QACvE,iDAAiD;QACjD,MAAM,QAAQ,GACZ,WAAW,KAAK,SAAS,IAAI,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAE/E,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,oBAAoB,CAAC,IAAW,EAAE,QAAa;QACpD,MAAM,EAAE,YAAY,EAAE,GAAG,iBAAiB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,aAAa,GACjB,QAAQ,EAAE,YAAY;YACtB,QAAQ,EAAE,WAAW;YACrB,QAAQ,EAAE,MAAM,EAAE,YAAY;YAC9B,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC;QAChC,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClE,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,CAAC,gBAAgB,CAAC,MAAc,EAAE,UAAkB,EAAE,UAAmB;QAC7E,MAAM,aAAa,GAAG,0BAA0B,UAAU,EAAE,CAAC;QAC7D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,IAAI,IAAA,8BAAoB,GAAE;oBAAE,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAEpE,MAAM,EAAE,YAAY,EAAE,GAAG,iBAAiB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;gBAEjE,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,2BAA2B,IAAI,YAAY,EAAE,CAAC;gBACrF,MAAM,YAAY,GAAG,IAAA,mCAAyB,GAAE,IAAI,aAAO,CAAC,MAAM,EAAE,CAAC;gBACrE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAC3B,QAAQ,EACR;oBACE,IAAI,EAAE,cAAQ,CAAC,MAAM;oBACrB,UAAU,EAAE,iBAAiB,CAC3B,6BAAkB,CAAC,2BAA2B,EAC9C,YAAY,EACZ,iBAAiB,CAAC,aAAa,EAC/B,iBAAiB,CAAC,UAAU,CAC7B;iBACF,EACD,YAAY,CACb,CAAC;gBAEF,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBAChE,IAAI,YAAY,CAAC;oBACjB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAE7C,YAAY,GAAG,iBAAiB,CAAC,kBAAkB,CAAC;4BAClD,IAAI;4BACJ,aAAa;4BACb,QAAQ;4BACR,IAAI;4BACJ,IAAI;4BACJ,GAAG,EAAE,CAAC;4BACN,QAAQ,EAAE,KAAK;4BACf,UAAU;yBACX,CAAC,CAAC;wBAEH,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACzC,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE;4BAC9B,aAAa;4BACb,KAAK,EAAE,YAAY;4BACnB,QAAQ,EAAE,iBAAiB,CAAC,QAAQ;4BACpC,aAAa,EAAE,iBAAiB,CAAC,aAAa;4BAC9C,UAAU,EAAE,iBAAiB,CAAC,UAAU;4BACxC,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,IAAI,QAAQ;yBAC5C,CAAC,CAAC;wBACH,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAI,YAAY,EAAE,CAAC;4BACjB,sBAAW,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;wBAChD,CAAC;oBACH,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,kBAAkB,CAAC,EACxB,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,IAAI,EACJ,IAAI,GAAG,CAAC,EACR,GAAG,GAAG,CAAC,EACP,QAAQ,GAAG,KAAK,EAChB,UAAU,GAUX;QACC,MAAM,cAAc,GAAG,gBAAe,CAAC,qBAAqB,CAAC;QAE7D,MAAM,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC7D,MAAM,YAAY,GAAG,iBAAiB,CAAC,oBAAoB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAE5E,0EAA0E;QAC1E,uDAAuD;QACvD,MAAM,KAAK,GAAG,CAAC,CAAU,EAAU,EAAE,CACnC,CAAC,KAAK,SAAS,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAEjD,+EAA+E;QAC/E,wEAAwE;QACxE,0EAA0E;QAC1E,iEAAiE;QACjE,MAAM,eAAe,GACnB,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;QAClE,MAAM,MAAM,GAAG,KAAK,CAClB,QAAQ,EAAE,SAAS,IAAI,QAAQ,EAAE,QAAQ,IAAI,eAAe,CAC7D,CAAC;QACF,MAAM,aAAa,GACjB,QAAQ,EAAE,cAAc,IAAI,QAAQ,EAAE,aAAa,IAAI,CAAC,CAAC;QAC3D,MAAM,UAAU,GAAG,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACvC,MAAM,YAAY,GAAG,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE3C,qBAAqB;QACrB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,aAAa,CAAC,CAAC;QACnF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC;QAEzE,sBAAsB;QACtB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QAE9E,qEAAqE;QACrE,MAAM,KAAK,GAAG,QAAQ,EAAE,KAAK,CAAC;QAC9B,MAAM,WAAW,GAAG,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC;QAC9E,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC;QACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;QAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,YAAY,CAAC,CAAC;QAC/E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,WAAW,GAAG,YAAY,CAAC,CAAC;QAC5F,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAC;QAClF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;QAC/D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;QAE7D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QACtD,uEAAuE;QACvE,mEAAmE;QACnE,oEAAoE;QACpE,uEAAuE;QACvE,8DAA8D;QAC9D,MAAM,IAAI,GAAG,iBAAe,CAAC,iBAAiB,CAC5C,YAAY,EACZ,WAAW,EACX,EAAE,EACF,aAAa,CACd,CAAC;QAEF,iBAAiB,CAAC,qBAAqB,CAAC,IAAI,EAAE;YAC5C,aAAa;YACb,KAAK,EAAE,YAAY;YACnB,IAAI;YACJ,QAAQ,EAAE,iBAAiB,CAAC,QAAQ;YACpC,aAAa,EAAE,iBAAiB,CAAC,aAAa;YAC9C,UAAU,EAAE,iBAAiB,CAAC,UAAU;SACzC,CAAC,CAAC;QAEH,oFAAoF;QACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACvF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QAC1E,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,iBAAqC,CAAC;QAC1C,IAAI,kBAAsC,CAAC;QAE3C,IAAI,cAAc,EAAE,CAAC;YACnB,iBAAiB,GAAG,iBAAe,CAAC,kBAAkB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAC5F,kBAAkB,GAAG,iBAAe,CAAC,mBAAmB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAE/E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,iBAAiB,CAAC,CAAC;YAC/E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;QACnF,CAAC;QAED,0EAA0E;QAC1E,+DAA+D;QAC/D,IAAI,CAAC,gBAAe,CAAC,aAAa,IAAI,cAAc,EAAE,CAAC;YACrD,MAAM,UAAU,GAAe;gBAC7B,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,2BAA2B;gBACrF,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,YAAY;gBACvD,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,YAAY;gBACxD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,iBAAiB,CAAC,aAAa;gBACpE,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,iBAAiB,CAAC,UAAU;gBAC9D,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,MAAM;gBAC/C,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,UAAU;gBACnD,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,WAAW;gBAC3D,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,YAAY;aAC9D,CAAC;YACF,IAAI,iBAAiB,EAAE,CAAC;gBACtB,UAAU,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,GAAG,iBAAiB,CAAC;YAC3E,CAAC;YACD,IAAI,kBAAkB,EAAE,CAAC;gBACvB,UAAU,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,GAAG,kBAAkB,CAAC;YAC7E,CAAC;YACD,iBAAe,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;QAED,OAAO;YACL,aAAa;YACb,KAAK,EAAE,YAAY;YACnB,IAAI;YACJ,QAAQ,EAAE,iBAAiB,CAAC,QAAQ;YACpC,aAAa,EAAE,iBAAiB,CAAC,aAAa;YAC9C,UAAU,EAAE,iBAAiB,CAAC,UAAU;SACzC,CAAC;IACJ,CAAC;;AAzPM,0BAAQ,GAAG,6BAAkB,CAAC,wBAAwB,CAAC;AACvD,+BAAa,GAAG,oBAAoB,CAAC;AACrC,4BAAU,GAAG,GAAG,CAAC;AA0P1B,kBAAe,iBAAiB,CAAC"}

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

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

package/dist/instrumentation/astra/index.js

@@ -0,0 +1,87 @@
+"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 semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const wrapper_1 = __importDefault(require("./wrapper"));
+class Shield360AstraInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-astra`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@datastax/astra-db-ts', ['>=1.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(astra) {
+        this._patch(astra);
+    }
+    _patch(moduleExports) {
+        try {
+            const Collection = moduleExports.Collection;
+            if (!Collection?.prototype)
+                return;
+            // `find` returns a cursor synchronously — must not be wrapped as async
+            if (typeof Collection.prototype['find'] === 'function') {
+                if ((0, instrumentation_1.isWrapped)(Collection.prototype['find'])) {
+                    this._unwrap(Collection.prototype, 'find');
+                }
+                this._wrap(Collection.prototype, 'find', wrapper_1.default._patchSyncFindMethod(this.tracer));
+            }
+            // All other methods return Promises and use the async wrapper
+            const asyncMethods = [
+                ['insertOne', semantic_convention_1.default.DB_OPERATION_INSERT],
+                ['insertMany', semantic_convention_1.default.DB_OPERATION_INSERT],
+                ['updateOne', semantic_convention_1.default.DB_OPERATION_UPDATE],
+                ['updateMany', semantic_convention_1.default.DB_OPERATION_UPDATE],
+                ['replaceOne', semantic_convention_1.default.DB_OPERATION_REPLACE],
+                ['findOne', semantic_convention_1.default.DB_OPERATION_SELECT],
+                ['findOneAndUpdate', semantic_convention_1.default.DB_OPERATION_REPLACE],
+                ['findOneAndReplace', semantic_convention_1.default.DB_OPERATION_REPLACE],
+                ['findOneAndDelete', semantic_convention_1.default.DB_OPERATION_FIND_AND_DELETE],
+                ['deleteOne', semantic_convention_1.default.DB_OPERATION_DELETE],
+                ['deleteMany', semantic_convention_1.default.DB_OPERATION_DELETE],
+            ];
+            for (const [method, dbOp] of asyncMethods) {
+                if (typeof Collection.prototype[method] === 'function') {
+                    if ((0, instrumentation_1.isWrapped)(Collection.prototype[method])) {
+                        this._unwrap(Collection.prototype, method);
+                    }
+                    this._wrap(Collection.prototype, method, wrapper_1.default._patchCollectionMethod(this.tracer, dbOp));
+                }
+            }
+        }
+        catch (e) {
+            api_1.diag.error('Astra instrumentation: error in _patch method', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const Collection = moduleExports.Collection;
+            if (!Collection?.prototype)
+                return;
+            for (const method of [
+                'insertOne', 'insertMany', 'updateOne', 'updateMany', 'replaceOne',
+                'find', 'findOne', 'findOneAndUpdate', 'findOneAndReplace',
+                'findOneAndDelete', 'deleteOne', 'deleteMany',
+            ]) {
+                if (typeof Collection.prototype[method] === 'function') {
+                    this._unwrap(Collection.prototype, method);
+                }
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = Shield360AstraInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/astra/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,4CAA0C;AAC1C,6CAAwD;AACxD,oFAA2D;AAC3D,wDAAqC;AAIrC,MAAqB,6BAA8B,SAAQ,qCAAmB;IAC5E,YAAY,SAAqC,EAAE;QACjD,KAAK,CAAC,GAAG,iCAAsB,wBAAwB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,uBAAuB,EACvB,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,KAAU;QAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC;YAC5C,IAAI,CAAC,UAAU,EAAE,SAAS;gBAAE,OAAO;YAEnC,uEAAuE;YACvE,IAAI,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;gBACvD,IAAI,IAAA,2BAAS,EAAC,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;oBAC5C,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;gBAC7C,CAAC;gBACD,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,iBAAY,CAAC,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3F,CAAC;YAED,8DAA8D;YAC9D,MAAM,YAAY,GAA4B;gBAC5C,CAAC,WAAW,EAAE,6BAAkB,CAAC,mBAAmB,CAAC;gBACrD,CAAC,YAAY,EAAE,6BAAkB,CAAC,mBAAmB,CAAC;gBACtD,CAAC,WAAW,EAAE,6BAAkB,CAAC,mBAAmB,CAAC;gBACrD,CAAC,YAAY,EAAE,6BAAkB,CAAC,mBAAmB,CAAC;gBACtD,CAAC,YAAY,EAAE,6BAAkB,CAAC,oBAAoB,CAAC;gBACvD,CAAC,SAAS,EAAE,6BAAkB,CAAC,mBAAmB,CAAC;gBACnD,CAAC,kBAAkB,EAAE,6BAAkB,CAAC,oBAAoB,CAAC;gBAC7D,CAAC,mBAAmB,EAAE,6BAAkB,CAAC,oBAAoB,CAAC;gBAC9D,CAAC,kBAAkB,EAAE,6BAAkB,CAAC,4BAA4B,CAAC;gBACrE,CAAC,WAAW,EAAE,6BAAkB,CAAC,mBAAmB,CAAC;gBACrD,CAAC,YAAY,EAAE,6BAAkB,CAAC,mBAAmB,CAAC;aACvD,CAAC;YAEF,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,YAAY,EAAE,CAAC;gBAC1C,IAAI,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;oBACvD,IAAI,IAAA,2BAAS,EAAC,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;wBAC5C,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;oBAC7C,CAAC;oBACD,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,iBAAY,CAAC,sBAAsB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;gBACnG,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,UAAI,CAAC,KAAK,CAAC,+CAA+C,EAAE,CAAC,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC;YAC5C,IAAI,CAAC,UAAU,EAAE,SAAS;gBAAE,OAAO;YACnC,KAAK,MAAM,MAAM,IAAI;gBACnB,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY;gBAClE,MAAM,EAAE,SAAS,EAAE,kBAAkB,EAAE,mBAAmB;gBAC1D,kBAAkB,EAAE,WAAW,EAAE,YAAY;aAC9C,EAAE,CAAC;gBACF,IAAI,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;oBACvD,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;gBAC7C,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AAlFD,gDAkFC"}

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

@@ -0,0 +1,39 @@
+import { Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class AstraWrapper extends BaseWrapper {
+    static dbSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    static _setCommonAttributes(span: any, dbOperation: string, collectionName: string, instance?: any): void;
+    /**
+     * Async wrapper for methods that return Promises (all Collection CRUD methods except `find`).
+     */
+    static _patchCollectionMethod(tracer: Tracer, dbOperation: string): any;
+    /**
+     * Synchronous wrapper for `find()` which returns a cursor synchronously.
+     * Wrapping it as async would break the cursor API (e.g. `collection.find({}).toArray()`).
+     */
+    static _patchSyncFindMethod(tracer: Tracer): any;
+    private static _setOperationAttributes;
+    private static _setInsertAttributes;
+    private static _setUpdateAttributes;
+    private static _setReplaceAttributes;
+    private static _setSelectAttributes;
+    private static _setDeleteAttributes;
+    private static _getActualDbOperation;
+    private static _hasUpsertOption;
+    private static _setFilterAttribute;
+    private static _getInsertReturnedRows;
+    private static _getUpdateReturnedRows;
+    private static _getSelectReturnedRows;
+    private static _getDeleteReturnedRows;
+    private static _objectCount;
+    private static _isCursorLike;
+    private static _summaryValue;
+    private static _safeStringify;
+    private static _truncate;
+    private static _getServerAddressAndPort;
+    private static _findEndpoint;
+    private static _parseEndpoint;
+}
+export default AstraWrapper;