shield360 1.0.0

package/dist/instrumentation/milvus/wrapper.js

@@ -0,0 +1,219 @@
+"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 config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+class MilvusWrapper extends base_wrapper_1.default {
+    static _setCommonAttributes(span, dbOperation, collectionName) {
+        const applicationName = config_1.default.applicationName || '';
+        const environment = config_1.default.environment || '';
+        span.setAttribute(semantic_convention_1.default.DB_SYSTEM_NAME, MilvusWrapper.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, MilvusWrapper.serverAddress);
+        span.setAttribute(semantic_convention_1.default.SERVER_PORT, MilvusWrapper.serverPort);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_ENVIRONMENT, environment);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_APPLICATION_NAME, applicationName);
+    }
+    // Milvus SDK uses `collection_name` in params for most methods
+    static _getCollectionName(params) {
+        return params?.collection_name || params?.collectionName || 'unknown';
+    }
+    static _patchSearch(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_SEARCH;
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const collectionName = MilvusWrapper._getCollectionName(params);
+                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;
+                        MilvusWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.DB_VECTOR_QUERY_TOP_K, params.topk || params.limit || 10);
+                        if (params.expr || params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, params.expr || params.filter);
+                        }
+                        if (params.output_fields) {
+                            span.setAttribute(semantic_convention_1.default.DB_OUTPUT_FIELDS, JSON.stringify(params.output_fields));
+                        }
+                        const resultCount = response?.results?.length || 0;
+                        span.setAttribute(semantic_convention_1.default.DB_N_RESULTS, resultCount);
+                        if (config_1.default.captureMessageContent && params.vectors) {
+                            const vectors = Array.isArray(params.vectors) ? params.vectors : [];
+                            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(vectors.slice(0, 1).map((v) => v?.slice(0, 10))));
+                        }
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} topk=${params.topk || params.limit || 10}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchInsert(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_INSERT;
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const collectionName = MilvusWrapper._getCollectionName(params);
+                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;
+                        MilvusWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        const data = params.data || params.fields_data || [];
+                        const rowCount = Array.isArray(data) ? data.length : 0;
+                        span.setAttribute(semantic_convention_1.default.DB_VECTOR_COUNT, rowCount);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} rows=${rowCount}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchUpsert(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_UPSERT;
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const collectionName = MilvusWrapper._getCollectionName(params);
+                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;
+                        MilvusWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        const data = params.data || params.fields_data || [];
+                        const rowCount = Array.isArray(data) ? data.length : 0;
+                        span.setAttribute(semantic_convention_1.default.DB_VECTOR_COUNT, rowCount);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} rows=${rowCount}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchDelete(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_DELETE;
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const collectionName = MilvusWrapper._getCollectionName(params);
+                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;
+                        MilvusWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        if (params.expr || params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, params.expr || params.filter);
+                        }
+                        const ids = params.ids || [];
+                        if (Array.isArray(ids) && ids.length > 0) {
+                            span.setAttribute(semantic_convention_1.default.DB_ID_COUNT, ids.length);
+                        }
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchQuery(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_QUERY;
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const collectionName = MilvusWrapper._getCollectionName(params);
+                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;
+                        MilvusWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        if (params.expr || params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, params.expr || params.filter);
+                        }
+                        if (params.limit) {
+                            span.setAttribute(semantic_convention_1.default.DB_QUERY_LIMIT, params.limit);
+                        }
+                        if (params.output_fields) {
+                            span.setAttribute(semantic_convention_1.default.DB_OUTPUT_FIELDS, JSON.stringify(params.output_fields));
+                        }
+                        const resultCount = response?.data?.length || 0;
+                        span.setAttribute(semantic_convention_1.default.DB_N_RESULTS, resultCount);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} results=${resultCount}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+}
+MilvusWrapper.dbSystem = semantic_convention_1.default.DB_SYSTEM_MILVUS;
+MilvusWrapper.serverAddress = 'localhost';
+MilvusWrapper.serverPort = 19530;
+exports.default = MilvusWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/milvus/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAAsE;AACtE,0DAA2C;AAC3C,4DAA4C;AAC5C,oFAA2D;AAC3D,mEAA0C;AAE1C,MAAM,aAAc,SAAQ,sBAAW;IAKrC,MAAM,CAAC,oBAAoB,CAAC,IAAS,EAAE,WAAmB,EAAE,cAAsB;QAChF,MAAM,eAAe,GAAG,gBAAe,CAAC,eAAe,IAAI,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QAEtD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC7E,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,aAAa,CAAC,CAAC;QAClF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC;QAC5E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;QACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,eAAe,CAAC,CAAC;IACjF,CAAC;IAED,+DAA+D;IAC/D,MAAM,CAAC,kBAAkB,CAAC,MAAW;QACnC,OAAO,MAAM,EAAE,eAAe,IAAI,MAAM,EAAE,cAAc,IAAI,SAAS,CAAC;IACxE,CAAC;IAED,MAAM,CAAC,YAAY,CAAC,MAAc;QAChC,MAAM,WAAW,GAAG,6BAAkB,CAAC,mBAAmB,CAAC;QAC3D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,cAAc,GAAG,aAAa,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAChE,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;wBAEjD,aAAa,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAC7E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;wBAE/F,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;4BACjC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;wBAChF,CAAC;wBACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;4BACzB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;wBAC/F,CAAC;wBAED,MAAM,WAAW,GAAG,QAAQ,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;wBACnD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;wBAEhE,IAAI,gBAAe,CAAC,qBAAqB,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BAC5D,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;4BACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAChD,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAW,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;wBAC/E,CAAC;wBAED,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EACnD,GAAG,WAAW,IAAI,cAAc,SAAS,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC,CAAC;wBAEhF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;wBAC5B,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,MAAM,CAAC,YAAY,CAAC,MAAc;QAChC,MAAM,WAAW,GAAG,6BAAkB,CAAC,mBAAmB,CAAC;QAC3D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,cAAc,GAAG,aAAa,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAChE,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;wBAEjD,aAAa,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAE7E,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;wBACrD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;wBACvD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;wBAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EACnD,GAAG,WAAW,IAAI,cAAc,SAAS,QAAQ,EAAE,CAAC,CAAC;wBAEvD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;wBAC5B,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,MAAM,CAAC,YAAY,CAAC,MAAc;QAChC,MAAM,WAAW,GAAG,6BAAkB,CAAC,mBAAmB,CAAC;QAC3D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,cAAc,GAAG,aAAa,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAChE,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;wBAEjD,aAAa,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAE7E,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;wBACrD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;wBACvD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;wBAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EACnD,GAAG,WAAW,IAAI,cAAc,SAAS,QAAQ,EAAE,CAAC,CAAC;wBAEvD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;wBAC5B,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,MAAM,CAAC,YAAY,CAAC,MAAc;QAChC,MAAM,WAAW,GAAG,6BAAkB,CAAC,mBAAmB,CAAC;QAC3D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,cAAc,GAAG,aAAa,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAChE,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;wBAEjD,aAAa,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAE7E,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;4BACjC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;wBAChF,CAAC;wBACD,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC;wBAC7B,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BACzC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;wBAChE,CAAC;wBACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EACnD,GAAG,WAAW,IAAI,cAAc,EAAE,CAAC,CAAC;wBAEtC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;wBAC5B,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,MAAM,CAAC,WAAW,CAAC,MAAc;QAC/B,MAAM,WAAW,GAAG,6BAAkB,CAAC,kBAAkB,CAAC;QAC1D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,cAAc,GAAG,aAAa,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAChE,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;wBAEjD,aAAa,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAE7E,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;4BACjC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;wBAChF,CAAC;wBACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;4BACjB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;wBACrE,CAAC;wBACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;4BACzB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;wBAC/F,CAAC;wBAED,MAAM,WAAW,GAAG,QAAQ,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC;wBAChD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;wBAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EACnD,GAAG,WAAW,IAAI,cAAc,YAAY,WAAW,EAAE,CAAC,CAAC;wBAE7D,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;wBAC5B,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;;AAxOM,sBAAQ,GAAG,6BAAkB,CAAC,gBAAgB,CAAC;AAC/C,2BAAa,GAAG,WAAW,CAAC;AAC5B,wBAAU,GAAG,KAAK,CAAC;AAyO5B,kBAAe,aAAa,CAAC"}

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

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

package/dist/instrumentation/mistral/index.js

@@ -0,0 +1,66 @@
+"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 Shield360MistralInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-mistral`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@mistralai/mistralai', ['>=1.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(mistral) {
+        this._patch(mistral);
+    }
+    _patch(moduleExports) {
+        try {
+            // Chat and Embeddings are instance properties in the new SDK — get their
+            // prototypes via a dummy instance (no API calls are made at construction time)
+            const dummy = new moduleExports.Mistral({ apiKey: 'dummy' });
+            const ChatProto = Object.getPrototypeOf(dummy.chat);
+            const EmbeddingsProto = Object.getPrototypeOf(dummy.embeddings);
+            if ((0, instrumentation_1.isWrapped)(ChatProto.complete)) {
+                this._unwrap(ChatProto, 'complete');
+            }
+            if ((0, instrumentation_1.isWrapped)(ChatProto.stream)) {
+                this._unwrap(ChatProto, 'stream');
+            }
+            if ((0, instrumentation_1.isWrapped)(EmbeddingsProto.create)) {
+                this._unwrap(EmbeddingsProto, 'create');
+            }
+            this._wrap(ChatProto, 'complete', wrapper_1.default._patchChatCompletionCreate(this.tracer));
+            this._wrap(ChatProto, 'stream', wrapper_1.default._patchChatCompletionCreate(this.tracer));
+            this._wrap(EmbeddingsProto, 'create', wrapper_1.default._patchEmbedding(this.tracer));
+        }
+        catch (e) {
+            console.error('Error in _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const dummy = new moduleExports.Mistral({ apiKey: 'dummy' });
+            const ChatProto = Object.getPrototypeOf(dummy.chat);
+            const EmbeddingsProto = Object.getPrototypeOf(dummy.embeddings);
+            this._unwrap(ChatProto, 'complete');
+            this._unwrap(ChatProto, 'stream');
+            this._unwrap(EmbeddingsProto, 'create');
+        }
+        catch (e) {
+            console.error('Error in _unpatch method:', e);
+        }
+    }
+}
+exports.default = Shield360MistralInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/mistral/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAuC;AAIvC,MAAqB,+BAAgC,SAAQ,qCAAmB;IAC9E,YAAY,SAAuC,EAAE;QACnD,KAAK,CAAC,GAAG,iCAAsB,0BAA0B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC9E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,sBAAsB,EACtB,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC3B,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,OAAY;QAC7B,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,yEAAyE;YACzE,+EAA+E;YAC/E,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAEhE,IAAI,IAAA,2BAAS,EAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAClC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,IAAA,2BAAS,EAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACpC,CAAC;YACD,IAAI,IAAA,2BAAS,EAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,EAAE,iBAAc,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,EAAE,iBAAc,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,KAAK,CAAC,eAAe,EAAE,QAAQ,EAAE,iBAAc,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACrF,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAChE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YACpC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAClC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;CACF;AAhED,kDAgEC"}

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

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