shield360 1.0.0

package/dist/instrumentation/astra/wrapper.js

@@ -0,0 +1,349 @@
+"use strict";
+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 semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const config_1 = __importDefault(require("../../config"));
+const constant_1 = require("../../constant");
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+const DEFAULT_SERVER = {
+    serverAddress: 'astra.datastax.com',
+    serverPort: 443,
+};
+let astraDbSdkVersion = 'unknown';
+try {
+    // Optional runtime dependency: keep this guarded so Shield360 can load without Astra installed.
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    astraDbSdkVersion = require('@datastax/astra-db-ts/package.json')?.version || 'unknown';
+}
+catch {
+    astraDbSdkVersion = 'unknown';
+}
+class AstraWrapper extends base_wrapper_1.default {
+    static _setCommonAttributes(span, dbOperation, collectionName, instance) {
+        const applicationName = config_1.default.applicationName || '';
+        const environment = config_1.default.environment || '';
+        const { serverAddress, serverPort } = AstraWrapper._getServerAddressAndPort(instance);
+        span.setAttribute(semantic_convention_1.default.DB_SYSTEM_NAME, AstraWrapper.dbSystem);
+        span.setAttribute(semantic_convention_1.default.DB_OPERATION_NAME, dbOperation);
+        span.setAttribute(semantic_convention_1.default.DB_COLLECTION_NAME, collectionName);
+        span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, serverAddress);
+        span.setAttribute(semantic_convention_1.default.SERVER_PORT, serverPort);
+        span.setAttribute(semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME, constant_1.SDK_NAME);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_VECTORDB);
+        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.DB_SDK_VERSION, astraDbSdkVersion);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_ENVIRONMENT, environment);
+    }
+    /**
+     * Async wrapper for methods that return Promises (all Collection CRUD methods except `find`).
+     */
+    static _patchCollectionMethod(tracer, dbOperation) {
+        return (originalMethod) => {
+            return async function (...args) {
+                const collectionName = this?.name || 'unknown';
+                const spanName = `${dbOperation} ${collectionName}`;
+                const span = tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT });
+                const startTime = Date.now();
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    try {
+                        const response = await originalMethod.apply(this, args);
+                        const duration = (Date.now() - startTime) / 1000;
+                        const actualDbOperation = AstraWrapper._getActualDbOperation(dbOperation, args);
+                        AstraWrapper._setCommonAttributes(span, actualDbOperation, collectionName, this);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        AstraWrapper._setOperationAttributes(span, {
+                            dbOperation: actualDbOperation,
+                            collectionName,
+                            args,
+                            response,
+                        });
+                        span.setStatus({ code: api_1.SpanStatusCode.OK });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    /**
+     * 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) {
+        return (originalMethod) => {
+            return function (...args) {
+                const collectionName = this?.name || 'unknown';
+                const spanName = `${semantic_convention_1.default.DB_OPERATION_SELECT} ${collectionName}`;
+                const span = tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT });
+                const startTime = Date.now();
+                try {
+                    const cursor = originalMethod.apply(this, args);
+                    const duration = (Date.now() - startTime) / 1000;
+                    AstraWrapper._setCommonAttributes(span, semantic_convention_1.default.DB_OPERATION_SELECT, collectionName, this);
+                    span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                    AstraWrapper._setOperationAttributes(span, {
+                        dbOperation: semantic_convention_1.default.DB_OPERATION_SELECT,
+                        collectionName,
+                        args,
+                        response: cursor,
+                    });
+                    span.setStatus({ code: api_1.SpanStatusCode.OK });
+                    return cursor;
+                }
+                catch (e) {
+                    helpers_1.default.handleException(span, e);
+                    throw e;
+                }
+                finally {
+                    span.end();
+                }
+            };
+        };
+    }
+    static _setOperationAttributes(span, context) {
+        switch (context.dbOperation) {
+            case semantic_convention_1.default.DB_OPERATION_INSERT:
+                AstraWrapper._setInsertAttributes(span, context);
+                break;
+            case semantic_convention_1.default.DB_OPERATION_UPDATE:
+                AstraWrapper._setUpdateAttributes(span, context);
+                break;
+            case semantic_convention_1.default.DB_OPERATION_REPLACE:
+            case semantic_convention_1.default.DB_OPERATION_UPSERT:
+                AstraWrapper._setReplaceAttributes(span, context);
+                break;
+            case semantic_convention_1.default.DB_OPERATION_SELECT:
+                AstraWrapper._setSelectAttributes(span, context);
+                break;
+            case semantic_convention_1.default.DB_OPERATION_DELETE:
+            case semantic_convention_1.default.DB_OPERATION_FIND_AND_DELETE:
+                AstraWrapper._setDeleteAttributes(span, context);
+                break;
+            default:
+                span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${context.dbOperation} ${context.collectionName}`);
+        }
+    }
+    static _setInsertAttributes(span, { dbOperation, collectionName, args, response }) {
+        const documents = args[0];
+        const documentsCount = AstraWrapper._objectCount(documents);
+        span.setAttribute(semantic_convention_1.default.DB_DOCUMENTS_COUNT, documentsCount);
+        if (config_1.default.captureMessageContent) {
+            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, AstraWrapper._safeStringify(documents));
+        }
+        const returnedRows = AstraWrapper._getInsertReturnedRows(response);
+        if (returnedRows !== undefined) {
+            span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, returnedRows);
+        }
+        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} documents_count=${documentsCount}`);
+    }
+    static _setUpdateAttributes(span, { dbOperation, collectionName, args, response }) {
+        const filter = args[0] || {};
+        const update = args[1] || {};
+        AstraWrapper._setFilterAttribute(span, filter);
+        if (config_1.default.captureMessageContent) {
+            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, AstraWrapper._safeStringify(update));
+        }
+        const returnedRows = AstraWrapper._getUpdateReturnedRows(response);
+        if (returnedRows !== undefined) {
+            span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, returnedRows);
+        }
+        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} filter=${AstraWrapper._summaryValue(filter)} update=${AstraWrapper._summaryValue(update)}`);
+    }
+    static _setReplaceAttributes(span, { dbOperation, collectionName, args }) {
+        const filter = args[0] || {};
+        const upsert = AstraWrapper._hasUpsertOption(args);
+        AstraWrapper._setFilterAttribute(span, filter);
+        if (config_1.default.captureMessageContent) {
+            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, AstraWrapper._safeStringify(filter));
+        }
+        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} filter=${AstraWrapper._summaryValue(filter)} upsert=${upsert}`);
+    }
+    static _setSelectAttributes(span, { dbOperation, collectionName, args, response }) {
+        const filter = args[0] || {};
+        AstraWrapper._setFilterAttribute(span, filter);
+        if (config_1.default.captureMessageContent) {
+            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, AstraWrapper._safeStringify(filter));
+        }
+        const returnedRows = AstraWrapper._getSelectReturnedRows(response);
+        if (returnedRows !== undefined) {
+            span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, returnedRows);
+        }
+        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} filter=${AstraWrapper._summaryValue(filter)}`);
+    }
+    static _setDeleteAttributes(span, { dbOperation, collectionName, args, response }) {
+        const filter = args[0] || {};
+        AstraWrapper._setFilterAttribute(span, filter);
+        if (config_1.default.captureMessageContent) {
+            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, AstraWrapper._safeStringify(filter));
+        }
+        const returnedRows = AstraWrapper._getDeleteReturnedRows(response, dbOperation);
+        if (returnedRows !== undefined) {
+            span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, returnedRows);
+        }
+        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} filter=${AstraWrapper._summaryValue(filter)}`);
+    }
+    static _getActualDbOperation(dbOperation, args) {
+        if (dbOperation === semantic_convention_1.default.DB_OPERATION_REPLACE && AstraWrapper._hasUpsertOption(args)) {
+            return semantic_convention_1.default.DB_OPERATION_UPSERT;
+        }
+        return dbOperation;
+    }
+    static _hasUpsertOption(args) {
+        return args.slice(2).some((arg) => arg && typeof arg === 'object' && arg.upsert === true);
+    }
+    static _setFilterAttribute(span, filter) {
+        if (filter && typeof filter === 'object' && Object.keys(filter).length > 0) {
+            span.setAttribute(semantic_convention_1.default.DB_FILTER, AstraWrapper._safeStringify(filter));
+        }
+    }
+    static _getInsertReturnedRows(response) {
+        if (!response)
+            return undefined;
+        if (typeof response.insertedCount === 'number')
+            return response.insertedCount;
+        if (typeof response.inserted_count === 'number')
+            return response.inserted_count;
+        if (Array.isArray(response.insertedIds))
+            return response.insertedIds.length;
+        if (Array.isArray(response.inserted_ids))
+            return response.inserted_ids.length;
+        if (response.insertedIds && typeof response.insertedIds === 'object')
+            return Object.keys(response.insertedIds).length;
+        if (response.inserted_ids && typeof response.inserted_ids === 'object')
+            return Object.keys(response.inserted_ids).length;
+        if (response.insertedId || response.inserted_id)
+            return 1;
+        return undefined;
+    }
+    static _getUpdateReturnedRows(response) {
+        if (!response)
+            return undefined;
+        if (typeof response.modifiedCount === 'number')
+            return response.modifiedCount;
+        if (typeof response.modified_count === 'number')
+            return response.modified_count;
+        if (typeof response.matchedCount === 'number')
+            return response.matchedCount;
+        if (typeof response.matched_count === 'number')
+            return response.matched_count;
+        if (typeof response.update_info?.nModified === 'number')
+            return response.update_info.nModified;
+        if (typeof response.updateInfo?.nModified === 'number')
+            return response.updateInfo.nModified;
+        return undefined;
+    }
+    static _getSelectReturnedRows(response) {
+        if (!response || AstraWrapper._isCursorLike(response))
+            return undefined;
+        if (Array.isArray(response))
+            return response.length;
+        if (typeof response.length === 'number')
+            return response.length;
+        return 1;
+    }
+    static _getDeleteReturnedRows(response, dbOperation) {
+        if (!response)
+            return dbOperation === semantic_convention_1.default.DB_OPERATION_FIND_AND_DELETE ? 0 : undefined;
+        if (typeof response.deletedCount === 'number')
+            return response.deletedCount;
+        if (typeof response.deleted_count === 'number')
+            return response.deleted_count;
+        if (dbOperation === semantic_convention_1.default.DB_OPERATION_FIND_AND_DELETE)
+            return 1;
+        return undefined;
+    }
+    static _objectCount(obj) {
+        if (Array.isArray(obj))
+            return obj.length;
+        if (obj !== undefined && obj !== null)
+            return 1;
+        return 0;
+    }
+    static _isCursorLike(response) {
+        return !!response && (typeof response[Symbol.asyncIterator] === 'function'
+            || typeof response[Symbol.iterator] === 'function'
+            || typeof response.toArray === 'function'
+            || typeof response.next === 'function');
+    }
+    static _summaryValue(value) {
+        return AstraWrapper._truncate(AstraWrapper._safeStringify(value));
+    }
+    static _safeStringify(value) {
+        if (typeof value === 'string')
+            return value;
+        try {
+            return JSON.stringify(value);
+        }
+        catch {
+            return String(value);
+        }
+    }
+    static _truncate(value, maxLength = 200) {
+        return value.length > maxLength ? `${value.slice(0, maxLength)}...` : value;
+    }
+    static _getServerAddressAndPort(instance) {
+        const endpoint = AstraWrapper._findEndpoint(instance);
+        if (!endpoint)
+            return DEFAULT_SERVER;
+        return AstraWrapper._parseEndpoint(endpoint);
+    }
+    static _findEndpoint(instance) {
+        const candidates = [
+            instance,
+            instance?.database,
+            instance?.db,
+            instance?.client,
+            instance?._client,
+            instance?.database?.client,
+            instance?.db?.client,
+        ];
+        const keys = ['apiEndpoint', 'api_endpoint', 'endpoint', 'baseUrl', 'base_url'];
+        for (const candidate of candidates) {
+            if (!candidate || typeof candidate !== 'object')
+                continue;
+            for (const key of keys) {
+                const value = candidate[key];
+                if (typeof value === 'string' && value.length > 0)
+                    return value;
+            }
+        }
+        return undefined;
+    }
+    static _parseEndpoint(endpoint) {
+        const fallbackPort = endpoint.startsWith('http://') ? 80 : DEFAULT_SERVER.serverPort;
+        try {
+            if (endpoint.startsWith('http://') || endpoint.startsWith('https://')) {
+                const parsed = new URL(endpoint);
+                return {
+                    serverAddress: parsed.hostname || DEFAULT_SERVER.serverAddress,
+                    serverPort: parsed.port ? Number(parsed.port) : fallbackPort,
+                };
+            }
+        }
+        catch {
+            return DEFAULT_SERVER;
+        }
+        const [serverAddress, port] = endpoint.split(':');
+        return {
+            serverAddress: serverAddress || DEFAULT_SERVER.serverAddress,
+            serverPort: port ? Number(port) || DEFAULT_SERVER.serverPort : DEFAULT_SERVER.serverPort,
+        };
+    }
+}
+AstraWrapper.dbSystem = semantic_convention_1.default.DB_SYSTEM_ASTRA;
+AstraWrapper.serverAddress = DEFAULT_SERVER.serverAddress;
+AstraWrapper.serverPort = DEFAULT_SERVER.serverPort;
+exports.default = AstraWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/astra/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAAsF;AACtF,8EAAiG;AACjG,0DAA2C;AAC3C,6CAA0C;AAC1C,4DAA4C;AAC5C,oFAA2D;AAC3D,mEAA0C;AAc1C,MAAM,cAAc,GAAyB;IAC3C,aAAa,EAAE,oBAAoB;IACnC,UAAU,EAAE,GAAG;CAChB,CAAC;AAEF,IAAI,iBAAiB,GAAG,SAAS,CAAC;AAElC,IAAI,CAAC;IACH,gGAAgG;IAChG,8DAA8D;IAC9D,iBAAiB,GAAG,OAAO,CAAC,oCAAoC,CAAC,EAAE,OAAO,IAAI,SAAS,CAAC;AAC1F,CAAC;AAAC,MAAM,CAAC;IACP,iBAAiB,GAAG,SAAS,CAAC;AAChC,CAAC;AAED,MAAM,YAAa,SAAQ,sBAAW;IAKpC,MAAM,CAAC,oBAAoB,CAAC,IAAS,EAAE,WAAmB,EAAE,cAAsB,EAAE,QAAc;QAChG,MAAM,eAAe,GAAG,gBAAe,CAAC,eAAe,IAAI,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QACtD,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,GAAG,YAAY,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAEtF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC5E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,WAAW,CAAC,CAAC;QACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,cAAc,CAAC,CAAC;QACzE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,YAAY,CAAC,8CAAuB,EAAE,mBAAQ,CAAC,CAAC;QACrD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,8BAA8B,CAAC,CAAC;QAC1G,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,cAAc,EAAE,iBAAiB,CAAC,CAAC;QACxE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;IACxE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,sBAAsB,CAAC,MAAc,EAAE,WAAmB;QAC/D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,cAAc,GAAW,IAAI,EAAE,IAAI,IAAI,SAAS,CAAC;gBACvD,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,cAAc,EAAE,CAAC;gBACpD,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE7B,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,KAAK,IAAI,EAAE;oBACpE,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACxD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBACjD,MAAM,iBAAiB,GAAG,YAAY,CAAC,qBAAqB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;wBAEhF,YAAY,CAAC,oBAAoB,CAAC,IAAI,EAAE,iBAAiB,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;wBACjF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAC7E,YAAY,CAAC,uBAAuB,CAAC,IAAI,EAAE;4BACzC,WAAW,EAAE,iBAAiB;4BAC9B,cAAc;4BACd,IAAI;4BACJ,QAAQ;yBACT,CAAC,CAAC;wBAEH,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC5C,OAAO,QAAQ,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACzC,MAAM,CAAC,CAAC;oBACV,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;oBACb,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,oBAAoB,CAAC,MAAc;QACxC,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,UAAqB,GAAG,IAAW;gBACxC,MAAM,cAAc,GAAW,IAAI,EAAE,IAAI,IAAI,SAAS,CAAC;gBACvD,MAAM,QAAQ,GAAG,GAAG,6BAAkB,CAAC,mBAAmB,IAAI,cAAc,EAAE,CAAC;gBAC/E,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE7B,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBAChD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;oBAEjD,YAAY,CAAC,oBAAoB,CAAC,IAAI,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;oBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;oBAC7E,YAAY,CAAC,uBAAuB,CAAC,IAAI,EAAE;wBACzC,WAAW,EAAE,6BAAkB,CAAC,mBAAmB;wBACnD,cAAc;wBACd,IAAI;wBACJ,QAAQ,EAAE,MAAM;qBACjB,CAAC,CAAC;oBAEH,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC5C,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,CAAM,EAAE,CAAC;oBAChB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACzC,MAAM,CAAC,CAAC;gBACV,CAAC;wBAAS,CAAC;oBACT,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;YACH,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,IAAS,EAAE,OAA8B;QAC9E,QAAQ,OAAO,CAAC,WAAW,EAAE,CAAC;YAC5B,KAAK,6BAAkB,CAAC,mBAAmB;gBACzC,YAAY,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBACjD,MAAM;YACR,KAAK,6BAAkB,CAAC,mBAAmB;gBACzC,YAAY,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBACjD,MAAM;YACR,KAAK,6BAAkB,CAAC,oBAAoB,CAAC;YAC7C,KAAK,6BAAkB,CAAC,mBAAmB;gBACzC,YAAY,CAAC,qBAAqB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAClD,MAAM;YACR,KAAK,6BAAkB,CAAC,mBAAmB;gBACzC,YAAY,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBACjD,MAAM;YACR,KAAK,6BAAkB,CAAC,mBAAmB,CAAC;YAC5C,KAAK,6BAAkB,CAAC,4BAA4B;gBAClD,YAAY,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBACjD,MAAM;YACR;gBACE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QAC/G,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,IAAS,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,QAAQ,EAAyB;QACnH,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,cAAc,GAAG,YAAY,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAE5D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,cAAc,CAAC,CAAC;QACzE,IAAI,gBAAe,CAAC,qBAAqB,EAAE,CAAC;YAC1C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,YAAY,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC;QAC9F,CAAC;QAED,MAAM,YAAY,GAAG,YAAY,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACnE,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,cAAc,oBAAoB,cAAc,EAAE,CACrE,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,IAAS,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,QAAQ,EAAyB;QACnH,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE7B,YAAY,CAAC,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/C,IAAI,gBAAe,CAAC,qBAAqB,EAAE,CAAC;YAC1C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,YAAY,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3F,CAAC;QAED,MAAM,YAAY,GAAG,YAAY,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACnE,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,cAAc,WAAW,YAAY,CAAC,aAAa,CAAC,MAAM,CAAC,WAAW,YAAY,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAC7H,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAAC,IAAS,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAyB;QAC1G,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAEnD,YAAY,CAAC,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/C,IAAI,gBAAe,CAAC,qBAAqB,EAAE,CAAC;YAC1C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,YAAY,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3F,CAAC;QAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,cAAc,WAAW,YAAY,CAAC,aAAa,CAAC,MAAM,CAAC,WAAW,MAAM,EAAE,CACjG,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,IAAS,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,QAAQ,EAAyB;QACnH,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE7B,YAAY,CAAC,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/C,IAAI,gBAAe,CAAC,qBAAqB,EAAE,CAAC;YAC1C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,YAAY,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3F,CAAC;QAED,MAAM,YAAY,GAAG,YAAY,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACnE,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,cAAc,WAAW,YAAY,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAChF,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,IAAS,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,QAAQ,EAAyB;QACnH,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE7B,YAAY,CAAC,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/C,IAAI,gBAAe,CAAC,qBAAqB,EAAE,CAAC;YAC1C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,YAAY,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3F,CAAC;QAED,MAAM,YAAY,GAAG,YAAY,CAAC,sBAAsB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAChF,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,cAAc,WAAW,YAAY,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAChF,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAAC,WAAmB,EAAE,IAAW;QACnE,IAAI,WAAW,KAAK,6BAAkB,CAAC,oBAAoB,IAAI,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YACnG,OAAO,6BAAkB,CAAC,mBAAmB,CAAC;QAChD,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,MAAM,CAAC,gBAAgB,CAAC,IAAW;QACzC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,IAAI,CAAC,CAAC;IAC5F,CAAC;IAEO,MAAM,CAAC,mBAAmB,CAAC,IAAS,EAAE,MAAW;QACvD,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,SAAS,EAAE,YAAY,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;QACvF,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,sBAAsB,CAAC,QAAa;QACjD,IAAI,CAAC,QAAQ;YAAE,OAAO,SAAS,CAAC;QAChC,IAAI,OAAO,QAAQ,CAAC,aAAa,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,aAAa,CAAC;QAC9E,IAAI,OAAO,QAAQ,CAAC,cAAc,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,cAAc,CAAC;QAChF,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;YAAE,OAAO,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC;QAC5E,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;YAAE,OAAO,QAAQ,CAAC,YAAY,CAAC,MAAM,CAAC;QAC9E,IAAI,QAAQ,CAAC,WAAW,IAAI,OAAO,QAAQ,CAAC,WAAW,KAAK,QAAQ;YAAE,OAAO,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC;QACtH,IAAI,QAAQ,CAAC,YAAY,IAAI,OAAO,QAAQ,CAAC,YAAY,KAAK,QAAQ;YAAE,OAAO,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC;QACzH,IAAI,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,WAAW;YAAE,OAAO,CAAC,CAAC;QAC1D,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,sBAAsB,CAAC,QAAa;QACjD,IAAI,CAAC,QAAQ;YAAE,OAAO,SAAS,CAAC;QAChC,IAAI,OAAO,QAAQ,CAAC,aAAa,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,aAAa,CAAC;QAC9E,IAAI,OAAO,QAAQ,CAAC,cAAc,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,cAAc,CAAC;QAChF,IAAI,OAAO,QAAQ,CAAC,YAAY,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,YAAY,CAAC;QAC5E,IAAI,OAAO,QAAQ,CAAC,aAAa,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,aAAa,CAAC;QAC9E,IAAI,OAAO,QAAQ,CAAC,WAAW,EAAE,SAAS,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC;QAC/F,IAAI,OAAO,QAAQ,CAAC,UAAU,EAAE,SAAS,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC;QAC7F,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,sBAAsB,CAAC,QAAa;QACjD,IAAI,CAAC,QAAQ,IAAI,YAAY,CAAC,aAAa,CAAC,QAAQ,CAAC;YAAE,OAAO,SAAS,CAAC;QACxE,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC,MAAM,CAAC;QACpD,IAAI,OAAO,QAAQ,CAAC,MAAM,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,MAAM,CAAC;QAChE,OAAO,CAAC,CAAC;IACX,CAAC;IAEO,MAAM,CAAC,sBAAsB,CAAC,QAAa,EAAE,WAAmB;QACtE,IAAI,CAAC,QAAQ;YAAE,OAAO,WAAW,KAAK,6BAAkB,CAAC,4BAA4B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACtG,IAAI,OAAO,QAAQ,CAAC,YAAY,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,YAAY,CAAC;QAC5E,IAAI,OAAO,QAAQ,CAAC,aAAa,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC,aAAa,CAAC;QAC9E,IAAI,WAAW,KAAK,6BAAkB,CAAC,4BAA4B;YAAE,OAAO,CAAC,CAAC;QAC9E,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,GAAQ;QAClC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;YAAE,OAAO,GAAG,CAAC,MAAM,CAAC;QAC1C,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,KAAK,IAAI;YAAE,OAAO,CAAC,CAAC;QAChD,OAAO,CAAC,CAAC;IACX,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,QAAa;QACxC,OAAO,CAAC,CAAC,QAAQ,IAAI,CACnB,OAAO,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU;eACjD,OAAO,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,UAAU;eAC/C,OAAO,QAAQ,CAAC,OAAO,KAAK,UAAU;eACtC,OAAO,QAAQ,CAAC,IAAI,KAAK,UAAU,CACvC,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,KAAU;QACrC,OAAO,YAAY,CAAC,SAAS,CAAC,YAAY,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;IACpE,CAAC;IAEO,MAAM,CAAC,cAAc,CAAC,KAAU;QACtC,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC5C,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,SAAS,CAAC,KAAa,EAAE,SAAS,GAAG,GAAG;QACrD,OAAO,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;IAC9E,CAAC;IAEO,MAAM,CAAC,wBAAwB,CAAC,QAAa;QACnD,MAAM,QAAQ,GAAG,YAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ;YAAE,OAAO,cAAc,CAAC;QAErC,OAAO,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,QAAa;QACxC,MAAM,UAAU,GAAG;YACjB,QAAQ;YACR,QAAQ,EAAE,QAAQ;YAClB,QAAQ,EAAE,EAAE;YACZ,QAAQ,EAAE,MAAM;YAChB,QAAQ,EAAE,OAAO;YACjB,QAAQ,EAAE,QAAQ,EAAE,MAAM;YAC1B,QAAQ,EAAE,EAAE,EAAE,MAAM;SACrB,CAAC;QACF,MAAM,IAAI,GAAG,CAAC,aAAa,EAAE,cAAc,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QAEhF,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ;gBAAE,SAAS;YAC1D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;gBAC7B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;oBAAE,OAAO,KAAK,CAAC;YAClE,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,cAAc,CAAC,QAAgB;QAC5C,MAAM,YAAY,GAAG,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC;QAErF,IAAI,CAAC;YACH,IAAI,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBACtE,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACjC,OAAO;oBACL,aAAa,EAAE,MAAM,CAAC,QAAQ,IAAI,cAAc,CAAC,aAAa;oBAC9D,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,YAAY;iBAC7D,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,cAAc,CAAC;QACxB,CAAC;QAED,MAAM,CAAC,aAAa,EAAE,IAAI,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAClD,OAAO;YACL,aAAa,EAAE,aAAa,IAAI,cAAc,CAAC,aAAa;YAC5D,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU;SACzF,CAAC;IACJ,CAAC;;AAhWM,qBAAQ,GAAG,6BAAkB,CAAC,eAAe,CAAC;AAC9C,0BAAa,GAAG,cAAc,CAAC,aAAa,CAAC;AAC7C,uBAAU,GAAG,cAAc,CAAC,UAAU,CAAC;AAiWhD,kBAAe,YAAY,CAAC"}

package/dist/instrumentation/azure-ai-inference/index.d.ts

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

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

@@ -0,0 +1,76 @@
+"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 Shield360AzureAIInferenceInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-azure-ai-inference`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@azure-rest/ai-inference', ['>=1.0.0-beta.1'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(azureAIInference) {
+        this._patch(azureAIInference);
+    }
+    _patch(moduleExports) {
+        try {
+            if (!moduleExports.default || typeof moduleExports.default !== 'function') {
+                return;
+            }
+            if ((0, instrumentation_1.isWrapped)(moduleExports.default)) {
+                this._unwrap(moduleExports, 'default');
+            }
+            const tracer = this.tracer;
+            this._wrap(moduleExports, 'default', (original) => {
+                return function (...args) {
+                    const client = original.apply(this, args);
+                    const endpoint = typeof args[0] === 'string' ? args[0] : '';
+                    const { serverAddress, serverPort } = wrapper_1.default.parseEndpoint(endpoint);
+                    if (client && typeof client.path === 'function') {
+                        const originalPath = client.path;
+                        client.path = function (...pathArgs) {
+                            const route = pathArgs[0];
+                            const handler = originalPath.apply(client, pathArgs);
+                            if (route === '/chat/completions' &&
+                                handler &&
+                                typeof handler.post === 'function') {
+                                const origPost = handler.post;
+                                handler.post = wrapper_1.default._patchChatComplete(tracer, serverAddress, serverPort)(origPost);
+                            }
+                            else if (route === '/embeddings' &&
+                                handler &&
+                                typeof handler.post === 'function') {
+                                const origPost = handler.post;
+                                handler.post = wrapper_1.default._patchEmbeddings(tracer, serverAddress, serverPort)(origPost);
+                            }
+                            return handler;
+                        };
+                    }
+                    return client;
+                };
+            });
+        }
+        catch (e) {
+            console.error('Error in _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        if (moduleExports.default) {
+            this._unwrap(moduleExports, 'default');
+        }
+    }
+}
+exports.default = Shield360AzureAIInferenceInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/azure-ai-inference/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAgD;AAIhD,MAAqB,wCAAyC,SAAQ,qCAAmB;IACvF,YAAY,SAAgD,EAAE;QAC5D,KAAK,CAAC,GAAG,iCAAsB,qCAAqC,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACzF,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,0BAA0B,EAC1B,CAAC,gBAAgB,CAAC,EAClB,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,gBAAqB;QACtC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;IAChC,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,IAAI,CAAC,aAAa,CAAC,OAAO,IAAI,OAAO,aAAa,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;gBAC1E,OAAO;YACT,CAAC;YAED,IAAI,IAAA,2BAAS,EAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;YACzC,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAE3B,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,CAAC,QAAiC,EAAE,EAAE;gBACzE,OAAO,UAAqB,GAAG,IAAW;oBACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBAE1C,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5D,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,GACjC,iBAAuB,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;oBAElD,IAAI,MAAM,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAChD,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC;wBACjC,MAAM,CAAC,IAAI,GAAG,UAAU,GAAG,QAAe;4BACxC,MAAM,KAAK,GAAW,QAAQ,CAAC,CAAC,CAAC,CAAC;4BAClC,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;4BAErD,IACE,KAAK,KAAK,mBAAmB;gCAC7B,OAAO;gCACP,OAAO,OAAO,CAAC,IAAI,KAAK,UAAU,EAClC,CAAC;gCACD,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;gCAC9B,OAAO,CAAC,IAAI,GAAG,iBAAuB,CAAC,kBAAkB,CACvD,MAAM,EACN,aAAa,EACb,UAAU,CACX,CAAC,QAAQ,CAAC,CAAC;4BACd,CAAC;iCAAM,IACL,KAAK,KAAK,aAAa;gCACvB,OAAO;gCACP,OAAO,OAAO,CAAC,IAAI,KAAK,UAAU,EAClC,CAAC;gCACD,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;gCAC9B,OAAO,CAAC,IAAI,GAAG,iBAAuB,CAAC,gBAAgB,CACrD,MAAM,EACN,aAAa,EACb,UAAU,CACX,CAAC,QAAQ,CAAC,CAAC;4BACd,CAAC;4BAED,OAAO,OAAO,CAAC;wBACjB,CAAC,CAAC;oBACJ,CAAC;oBAED,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,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,aAAa,CAAC,OAAO,EAAE,CAAC;YAC1B,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;CACF;AA7FD,2DA6FC"}

package/dist/instrumentation/azure-ai-inference/wrapper.d.ts

@@ -0,0 +1,55 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper, { BaseSpanAttributes } from '../base-wrapper';
+declare class AzureAIInferenceWrapper extends BaseWrapper {
+    static aiSystem: string;
+    static defaultServerAddress: string;
+    static defaultServerPort: 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>;
+    /**
+     * Extracts server address and port from an endpoint URL string.
+     */
+    static parseEndpoint(endpoint: string): {
+        serverAddress: string;
+        serverPort: number;
+    };
+    static _patchChatComplete(tracer: Tracer, serverAddress: string, serverPort: number): any;
+    static _chatCompletion({ body, genAIEndpoint, httpResponse, span, serverAddress, serverPort, }: {
+        body: any;
+        genAIEndpoint: string;
+        httpResponse: any;
+        span: Span;
+        serverAddress: string;
+        serverPort: number;
+    }): Promise<any>;
+    /**
+     * Wraps an SSE body stream (Node.js IncomingMessage / ReadableStream) to
+     * aggregate telemetry while passing through chunks to the caller.
+     * Returns an async-iterable that yields the raw SSE buffers/strings so
+     * downstream consumers (e.g. createSseStream) keep working.
+     */
+    static _wrapSseStream(body: any, requestBody: any, genAIEndpoint: string, span: Span, serverAddress: string, serverPort: number): any;
+    static _chatCompletionCommonSetter({ body, genAIEndpoint, result, span, serverAddress, serverPort, ttft, tbt, }: {
+        body: any;
+        genAIEndpoint: string;
+        result: any;
+        span: Span;
+        serverAddress: string;
+        serverPort: number;
+        ttft?: number;
+        tbt?: number;
+    }): BaseSpanAttributes;
+    static _patchEmbeddings(tracer: Tracer, serverAddress: string, serverPort: number): any;
+}
+export default AzureAIInferenceWrapper;