openlit 1.13.0 → 1.14.0

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 OpenlitAstraInstrumentation 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 = OpenlitAstraInstrumentation;
+//# 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,2BAA4B,SAAQ,qCAAmB;IAC1E,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,8CAkFC"}

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;

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 OpenLIT 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,0DAAyC;AACzC,6CAA0C;AAC1C,4DAA0C;AAC1C,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,8FAA8F;IAC9F,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,gBAAa,CAAC,eAAe,IAAI,EAAE,CAAC;QAC5D,MAAM,WAAW,GAAG,gBAAa,CAAC,WAAW,IAAI,EAAE,CAAC;QACpD,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,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBACvC,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,iBAAa,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACvC,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,gBAAa,CAAC,qBAAqB,EAAE,CAAC;YACxC,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,gBAAa,CAAC,qBAAqB,EAAE,CAAC;YACxC,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,gBAAa,CAAC,qBAAqB,EAAE,CAAC;YACxC,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,gBAAa,CAAC,qBAAqB,EAAE,CAAC;YACxC,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,gBAAa,CAAC,qBAAqB,EAAE,CAAC;YACxC,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/wrapper.d.ts

@@ -4,6 +4,19 @@ declare class AzureAIInferenceWrapper extends BaseWrapper {
     static aiSystem: string;
     static defaultServerAddress: string;
     static defaultServerPort: number;
+    /**
+     * Stamp `openlit.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.
      */

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

@@ -51,6 +51,41 @@ function spanCreationAttrs(operationName, requestModel, serverAddress, serverPor
     };
 }
 class AzureAIInferenceWrapper extends base_wrapper_1.default {
+    /**
+     * Stamp `openlit.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, args) {
+        const out = {};
+        try {
+            const versionHash = helpers_1.default.computeAgentVersionHash({
+                systemInstructions: args.systemInstructionsJson ?? null,
+                toolDefinitions: args.toolDefinitionsJson ?? null,
+                primaryModel: args.primaryModel ?? null,
+                runtimeConfig: {
+                    temperature: args.temperature ?? null,
+                    top_p: args.top_p ?? null,
+                    max_tokens: args.max_tokens ?? null,
+                    provider: semantic_convention_1.default.GEN_AI_SYSTEM_AZURE_AI_INFERENCE,
+                },
+                providers: [semantic_convention_1.default.GEN_AI_SYSTEM_AZURE_AI_INFERENCE],
+            });
+            if (versionHash) {
+                out[semantic_convention_1.default.OPENLIT_AGENT_VERSION_HASH] = versionHash;
+                span.setAttribute(semantic_convention_1.default.OPENLIT_AGENT_VERSION_HASH, versionHash);
+            }
+        }
+        catch {
+            // Hash computation must never fail the wrapped call.
+        }
+        const versionLabel = (0, helpers_1.getCurrentAgentVersion)();
+        if (versionLabel) {
+            out[semantic_convention_1.default.GEN_AI_AGENT_VERSION] = versionLabel;
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_VERSION, versionLabel);
+        }
+        return out;
+    }
     /**
      * Extracts server address and port from an endpoint URL string.
      */
@@ -320,7 +355,7 @@ class AzureAIInferenceWrapper extends base_wrapper_1.default {
     static _chatCompletionCommonSetter({ body, genAIEndpoint, result, span, serverAddress, serverPort, ttft = 0, tbt = 0, }) {
         const captureContent = config_1.default.captureMessageContent;
         const requestModel = body.model || 'gpt-4o';
-        const { messages, frequency_penalty = 0, max_tokens = null, n = 1, presence_penalty = 0, seed = null, stop = null, temperature = 1, top_p, user, stream = false, } = body;
+        const { messages, frequency_penalty = 0, max_tokens = null, n = 1, presence_penalty = 0, seed = null, stop = null, temperature = 1, top_p, user, stream = false, tools: _tools, } = body;
         span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, top_p || 1);
         if (max_tokens != null) {
             span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, max_tokens);
@@ -402,11 +437,30 @@ class AzureAIInferenceWrapper extends base_wrapper_1.default {
         }
         let inputMessagesJson;
         let outputMessagesJson;
+        const toolDefinitionsJson = helpers_1.default.buildToolDefinitions(_tools);
+        // Compute system_instructions and version hash regardless of
+        // captureContent so versions still group correctly when content
+        // capture is disabled.
+        const systemInstructionsJson = helpers_1.default.buildSystemInstructionsFromMessages(messages || []);
+        const versionExtras = AzureAIInferenceWrapper._stampAgentVersion(span, {
+            systemInstructionsJson,
+            toolDefinitionsJson,
+            primaryModel: responseModel || requestModel,
+            temperature,
+            top_p,
+            max_tokens,
+        });
         if (captureContent) {
             const toolCalls = choices[0]?.message?.tool_calls;
             outputMessagesJson = helpers_1.default.buildOutputMessages(choices[0]?.message?.content || '', choices[0]?.finish_reason || 'stop', toolCalls);
             span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
             inputMessagesJson = helpers_1.default.buildInputMessages(messages || []);
+            if (systemInstructionsJson) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS, systemInstructionsJson);
+            }
+        }
+        if (toolDefinitionsJson) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS, toolDefinitionsJson);
         }
         if (!config_1.default.disableEvents) {
             const eventAttrs = {
@@ -420,13 +474,18 @@ class AzureAIInferenceWrapper extends base_wrapper_1.default {
                 [semantic_convention_1.default.GEN_AI_OUTPUT_TYPE]: outputType,
                 [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: inputTokens,
                 [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: outputTokens,
+                ...versionExtras,
             };
             if (captureContent) {
                 if (inputMessagesJson)
                     eventAttrs[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES] = inputMessagesJson;
                 if (outputMessagesJson)
                     eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES] = outputMessagesJson;
+                if (systemInstructionsJson)
+                    eventAttrs[semantic_convention_1.default.GEN_AI_SYSTEM_INSTRUCTIONS] = systemInstructionsJson;
             }
+            if (toolDefinitionsJson)
+                eventAttrs[semantic_convention_1.default.GEN_AI_TOOL_DEFINITIONS] = toolDefinitionsJson;
             helpers_1.default.emitInferenceEvent(span, eventAttrs);
         }
         return {