shield360 1.0.0

package/dist/instrumentation/pinecone/index.js

@@ -0,0 +1,79 @@
+"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 Shield360PineconeInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-pinecone`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@pinecone-database/pinecone', ['>=1.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(pinecone) {
+        this._patch(pinecone);
+    }
+    _patch(moduleExports) {
+        try {
+            // Patch Index prototype directly
+            const IndexClass = moduleExports.Index;
+            if (IndexClass?.prototype) {
+                const methods = [
+                    ['query', 'query'],
+                    ['upsert', 'upsert'],
+                    ['deleteOne', 'one'],
+                    ['deleteMany', 'many'],
+                    ['update', 'update'],
+                ];
+                for (const [method, opSuffix] of methods) {
+                    if (typeof IndexClass.prototype[method] === 'function') {
+                        if ((0, instrumentation_1.isWrapped)(IndexClass.prototype[method])) {
+                            this._unwrap(IndexClass.prototype, method);
+                        }
+                        if (method === 'query') {
+                            this._wrap(IndexClass.prototype, method, wrapper_1.default._patchQuery(this.tracer));
+                        }
+                        else if (method === 'upsert') {
+                            this._wrap(IndexClass.prototype, method, wrapper_1.default._patchUpsert(this.tracer));
+                        }
+                        else if (method === 'update') {
+                            this._wrap(IndexClass.prototype, method, wrapper_1.default._patchUpdate(this.tracer));
+                        }
+                        else {
+                            this._wrap(IndexClass.prototype, method, wrapper_1.default._patchDelete(this.tracer, opSuffix));
+                        }
+                    }
+                }
+            }
+        }
+        catch (e) {
+            console.error('Error in Pinecone _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const IndexClass = moduleExports.Index;
+            if (IndexClass?.prototype) {
+                ['query', 'upsert', 'deleteOne', 'deleteMany', 'update'].forEach((method) => {
+                    if (typeof IndexClass.prototype[method] === 'function') {
+                        this._unwrap(IndexClass.prototype, method);
+                    }
+                });
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = Shield360PineconeInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/pinecone/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAwC;AAIxC,MAAqB,gCAAiC,SAAQ,qCAAmB;IAC/E,YAAY,SAAwC,EAAE;QACpD,KAAK,CAAC,GAAG,iCAAsB,2BAA2B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,6BAA6B,EAC7B,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,QAAa;QAC9B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,iCAAiC;YACjC,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC;YACvC,IAAI,UAAU,EAAE,SAAS,EAAE,CAAC;gBAC1B,MAAM,OAAO,GAA4B;oBACvC,CAAC,OAAO,EAAE,OAAO,CAAC;oBAClB,CAAC,QAAQ,EAAE,QAAQ,CAAC;oBACpB,CAAC,WAAW,EAAE,KAAK,CAAC;oBACpB,CAAC,YAAY,EAAE,MAAM,CAAC;oBACtB,CAAC,QAAQ,EAAE,QAAQ,CAAC;iBACrB,CAAC;gBAEF,KAAK,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,OAAO,EAAE,CAAC;oBACzC,IAAI,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;wBACvD,IAAI,IAAA,2BAAS,EAAC,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;4BAC5C,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;wBAC7C,CAAC;wBACD,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;4BACvB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,iBAAe,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;wBACrF,CAAC;6BAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;4BAC/B,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,iBAAe,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;wBACtF,CAAC;6BAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;4BAC/B,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,iBAAe,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;wBACtF,CAAC;6BAAM,CAAC;4BACN,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,iBAAe,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;wBAChG,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC;YACvC,IAAI,UAAU,EAAE,SAAS,EAAE,CAAC;gBAC1B,CAAC,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;oBAC1E,IAAI,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;wBACvD,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;oBAC7C,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AAzED,mDAyEC"}

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

@@ -0,0 +1,14 @@
+import { Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class PineconeWrapper extends BaseWrapper {
+    static dbSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    static _resolveNamespace(indexInstance: any, paramsNamespace?: string): string;
+    static _setCommonAttributes(span: any, dbOperation: string, namespace: string): void;
+    static _patchQuery(tracer: Tracer): any;
+    static _patchUpsert(tracer: Tracer): any;
+    static _patchDelete(tracer: Tracer, operationName: string): any;
+    static _patchUpdate(tracer: Tracer): any;
+}
+export default PineconeWrapper;

package/dist/instrumentation/pinecone/wrapper.js

@@ -0,0 +1,198 @@
+"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 PineconeWrapper extends base_wrapper_1.default {
+    // Safely resolve the namespace string from the Index instance.
+    // `this.namespace` on a Pinecone Index is a METHOD that creates namespace-scoped
+    // sub-indices, not a string property. The actual namespace string may be stored
+    // internally in `this.target.namespace` depending on the SDK version.
+    static _resolveNamespace(indexInstance, paramsNamespace) {
+        if (paramsNamespace && typeof paramsNamespace === 'string')
+            return paramsNamespace;
+        const ns = indexInstance?.target?.namespace ?? indexInstance?._namespace ?? '';
+        return typeof ns === 'string' && ns ? ns : 'default';
+    }
+    static _setCommonAttributes(span, dbOperation, namespace) {
+        const applicationName = config_1.default.applicationName || '';
+        const environment = config_1.default.environment || '';
+        span.setAttribute(semantic_convention_1.default.DB_SYSTEM_NAME, PineconeWrapper.dbSystem);
+        span.setAttribute(semantic_convention_1.default.DB_OPERATION_NAME, dbOperation);
+        span.setAttribute(semantic_convention_1.default.DB_NAMESPACE, namespace);
+        span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, PineconeWrapper.serverAddress);
+        span.setAttribute(semantic_convention_1.default.SERVER_PORT, PineconeWrapper.serverPort);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_ENVIRONMENT, environment);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_APPLICATION_NAME, applicationName);
+    }
+    static _patchQuery(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_QUERY;
+        return (originalMethod) => {
+            return async function (...args) {
+                const params = args[0] || {};
+                const namespace = PineconeWrapper._resolveNamespace(this, params.namespace);
+                const spanName = `${dbOperation} ${namespace}`;
+                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;
+                        PineconeWrapper._setCommonAttributes(span, dbOperation, namespace);
+                        span.setAttribute(semantic_convention_1.default.DB_VECTOR_QUERY_TOP_K, params.topK || 0);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        if (params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, JSON.stringify(params.filter));
+                        }
+                        const matchCount = response?.matches?.length || 0;
+                        span.setAttribute(semantic_convention_1.default.DB_N_RESULTS, matchCount);
+                        if (config_1.default.captureMessageContent && params.vector) {
+                            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(params.vector?.slice(0, 10)));
+                        }
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${namespace} top_k=${params.topK || 0} filtered=${params.filter ? 'true' : 'false'}`);
+                        span.setStatus({ code: 1 }); // SpanStatusCode.OK
+                        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 namespace = PineconeWrapper._resolveNamespace(this);
+                const spanName = `${dbOperation} ${namespace}`;
+                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 records = Array.isArray(args[0]) ? args[0] : [];
+                        const duration = (Date.now() - startTime) / 1000;
+                        PineconeWrapper._setCommonAttributes(span, dbOperation, namespace);
+                        span.setAttribute(semantic_convention_1.default.DB_VECTOR_COUNT, records.length);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${namespace} vectors_count=${records.length}`);
+                        if (config_1.default.captureMessageContent && records.length > 0) {
+                            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, JSON.stringify(records.map((r) => r.id)));
+                        }
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchDelete(tracer, operationName) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_DELETE;
+        return (originalMethod) => {
+            return async function (...args) {
+                const namespace = PineconeWrapper._resolveNamespace(this);
+                const spanName = `${dbOperation} ${namespace}`;
+                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 params = args[0] || {};
+                        const duration = (Date.now() - startTime) / 1000;
+                        PineconeWrapper._setCommonAttributes(span, `${dbOperation}.${operationName}`, namespace);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        // deleteOne passes an id string; deleteMany passes an array of ids
+                        let ids = [];
+                        if (typeof params === 'string') {
+                            ids = [params];
+                            span.setAttribute(semantic_convention_1.default.DB_ID_COUNT, 1);
+                        }
+                        else if (Array.isArray(params)) {
+                            ids = params;
+                            span.setAttribute(semantic_convention_1.default.DB_ID_COUNT, params.length);
+                        }
+                        else if (params.ids) {
+                            ids = Array.isArray(params.ids) ? params.ids : [params.ids];
+                            span.setAttribute(semantic_convention_1.default.DB_ID_COUNT, ids.length);
+                        }
+                        if (params.deleteAll) {
+                            span.setAttribute(semantic_convention_1.default.DB_DELETE_ALL, true);
+                        }
+                        if (params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, JSON.stringify(params.filter));
+                        }
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${namespace} ids=${JSON.stringify(ids)} delete_all=${params.deleteAll || false}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchUpdate(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_UPDATE;
+        return (originalMethod) => {
+            return async function (...args) {
+                const namespace = PineconeWrapper._resolveNamespace(this);
+                const spanName = `${dbOperation} ${namespace}`;
+                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 params = args[0] || {};
+                        const duration = (Date.now() - startTime) / 1000;
+                        PineconeWrapper._setCommonAttributes(span, dbOperation, namespace);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        if (params.id)
+                            span.setAttribute(semantic_convention_1.default.DB_UPDATE_ID, params.id);
+                        if (params.values)
+                            span.setAttribute(semantic_convention_1.default.DB_UPDATE_VALUES, String(params.values?.length || 0));
+                        if (params.metadata) {
+                            span.setAttribute(semantic_convention_1.default.DB_UPDATE_METADATA, JSON.stringify(params.metadata));
+                        }
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${namespace} id=${params.id || ''} values=${params.values?.length || 0} set_metadata=${params.metadata ? JSON.stringify(params.metadata) : ''}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+}
+PineconeWrapper.dbSystem = semantic_convention_1.default.DB_SYSTEM_PINECONE;
+PineconeWrapper.serverAddress = 'pinecone.io';
+PineconeWrapper.serverPort = 443;
+exports.default = PineconeWrapper;
+//# sourceMappingURL=wrapper.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/pinecone/wrapper.ts"],"names":[],"mappings":";;;;;AAAA,4CAAsE;AACtE,0DAA2C;AAC3C,4DAA4C;AAC5C,oFAA2D;AAC3D,mEAA0C;AAE1C,MAAM,eAAgB,SAAQ,sBAAW;IAKvC,+DAA+D;IAC/D,iFAAiF;IACjF,gFAAgF;IAChF,sEAAsE;IACtE,MAAM,CAAC,iBAAiB,CAAC,aAAkB,EAAE,eAAwB;QACnE,IAAI,eAAe,IAAI,OAAO,eAAe,KAAK,QAAQ;YAAE,OAAO,eAAe,CAAC;QACnF,MAAM,EAAE,GAAG,aAAa,EAAE,MAAM,EAAE,SAAS,IAAI,aAAa,EAAE,UAAU,IAAI,EAAE,CAAC;QAC/E,OAAO,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACvD,CAAC;IAED,MAAM,CAAC,oBAAoB,CAAC,IAAS,EAAE,WAAmB,EAAE,SAAiB;QAC3E,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,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC/E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,WAAW,CAAC,CAAC;QACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,eAAe,CAAC,aAAa,CAAC,CAAC;QACpF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;QAC9E,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,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,SAAS,GAAG,eAAe,CAAC,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC5E,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;gBAC/C,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,eAAe,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;wBACnE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;wBAC9E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAE7E,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;4BAClB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;wBACjF,CAAC;wBAED,MAAM,UAAU,GAAG,QAAQ,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;wBAClD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;wBAE/D,IAAI,gBAAe,CAAC,qBAAqB,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;4BAC3D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;wBACnG,CAAC;wBAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,SAAS,UAAU,MAAM,CAAC,IAAI,IAAI,CAAC,aAAa,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CACrG,CAAC;wBAEF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB;wBACjD,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,SAAS,GAAG,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBAC1D,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;gBAC/C,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,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACtD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,eAAe,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;wBACnE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;wBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAC7E,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,SAAS,kBAAkB,OAAO,CAAC,MAAM,EAAE,CAC9D,CAAC;wBAEF,IAAI,gBAAe,CAAC,qBAAqB,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;wBACrG,CAAC;wBAED,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,EAAE,aAAqB;QACvD,MAAM,WAAW,GAAG,6BAAkB,CAAC,mBAAmB,CAAC;QAC3D,OAAO,CAAC,cAAuC,EAAE,EAAE;YACjD,OAAO,KAAK,WAAsB,GAAG,IAAW;gBAC9C,MAAM,SAAS,GAAG,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBAC1D,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;gBAC/C,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,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAC7B,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,eAAe,CAAC,oBAAoB,CAAC,IAAI,EAAE,GAAG,WAAW,IAAI,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;wBACzF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAE7E,mEAAmE;wBACnE,IAAI,GAAG,GAAa,EAAE,CAAC;wBACvB,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;4BAC/B,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC;4BACf,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;wBACvD,CAAC;6BAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;4BACjC,GAAG,GAAG,MAAM,CAAC;4BACb,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;wBACnE,CAAC;6BAAM,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC;4BACtB,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;4BAC5D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;wBAChE,CAAC;wBACD,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;4BACrB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;wBAC5D,CAAC;wBACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;4BAClB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;wBACjF,CAAC;wBAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,SAAS,QAAQ,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,SAAS,IAAI,KAAK,EAAE,CACjG,CAAC;wBAEF,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,SAAS,GAAG,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBAC1D,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;gBAC/C,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,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAC7B,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBAEjD,eAAe,CAAC,oBAAoB,CAAC,IAAI,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;wBACnE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;wBAE7E,IAAI,MAAM,CAAC,EAAE;4BAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,YAAY,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;wBAC7E,IAAI,MAAM,CAAC,MAAM;4BAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC;wBAC9G,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;4BACpB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;wBAC5F,CAAC;wBAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,GAAG,WAAW,IAAI,SAAS,OAAO,MAAM,CAAC,EAAE,IAAI,EAAE,WAAW,MAAM,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,iBAAiB,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAChK,CAAC;wBAEF,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;;AAlNM,wBAAQ,GAAG,6BAAkB,CAAC,kBAAkB,CAAC;AACjD,6BAAa,GAAG,aAAa,CAAC;AAC9B,0BAAU,GAAG,GAAG,CAAC;AAmN1B,kBAAe,eAAe,CAAC"}

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

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

package/dist/instrumentation/qdrant/index.js

@@ -0,0 +1,67 @@
+"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 Shield360QdrantInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-qdrant`, '1.0.0', config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@qdrant/js-client-rest', ['>=1.0.0'], (moduleExports) => {
+            this._patch(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(qdrant) {
+        this._patch(qdrant);
+    }
+    _patch(moduleExports) {
+        try {
+            const QdrantClient = moduleExports.QdrantClient;
+            if (!QdrantClient?.prototype)
+                return;
+            const patchMap = [
+                ['search', wrapper_1.default._patchSearch.bind(wrapper_1.default)],
+                ['upsert', wrapper_1.default._patchUpsert.bind(wrapper_1.default)],
+                ['delete', wrapper_1.default._patchDelete.bind(wrapper_1.default)],
+                ['retrieve', wrapper_1.default._patchRetrieve.bind(wrapper_1.default)],
+                ['scroll', wrapper_1.default._patchScroll.bind(wrapper_1.default)],
+            ];
+            for (const [method, patchFn] of patchMap) {
+                if (typeof QdrantClient.prototype[method] === 'function') {
+                    if ((0, instrumentation_1.isWrapped)(QdrantClient.prototype[method])) {
+                        this._unwrap(QdrantClient.prototype, method);
+                    }
+                    this._wrap(QdrantClient.prototype, method, patchFn(this.tracer));
+                }
+            }
+        }
+        catch (e) {
+            console.error('Error in Qdrant _patch method:', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            const QdrantClient = moduleExports.QdrantClient;
+            if (!QdrantClient?.prototype)
+                return;
+            for (const method of ['search', 'upsert', 'delete', 'retrieve', 'scroll']) {
+                if (typeof QdrantClient.prototype[method] === 'function') {
+                    this._unwrap(QdrantClient.prototype, method);
+                }
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = Shield360QdrantInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/qdrant/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,6CAAwD;AACxD,wDAAsC;AAItC,MAAqB,8BAA+B,SAAQ,qCAAmB;IAC7E,YAAY,SAAsC,EAAE;QAClD,KAAK,CAAC,GAAG,iCAAsB,yBAAyB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC7E,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,wBAAwB,EACxB,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,MAAW;QAC5B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACtB,CAAC;IAES,MAAM,CAAC,aAAkB;QACjC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAAC,YAAY,CAAC;YAChD,IAAI,CAAC,YAAY,EAAE,SAAS;gBAAE,OAAO;YAErC,MAAM,QAAQ,GAA0C;gBACtD,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBAC1D,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBAC1D,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBAC1D,CAAC,UAAU,EAAE,iBAAa,CAAC,cAAc,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;gBAC9D,CAAC,QAAQ,EAAE,iBAAa,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAa,CAAC,CAAC;aAC3D,CAAC;YAEF,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzC,IAAI,OAAO,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;oBACzD,IAAI,IAAA,2BAAS,EAAC,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;wBAC9C,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;oBAC/C,CAAC;oBACD,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAkB;QACnC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAAC,YAAY,CAAC;YAChD,IAAI,CAAC,YAAY,EAAE,SAAS;gBAAE,OAAO;YACrC,KAAK,MAAM,MAAM,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,CAAC;gBAC1E,IAAI,OAAO,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE,CAAC;oBACzD,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AA/DD,iDA+DC"}

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

@@ -0,0 +1,14 @@
+import { Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+declare class QdrantWrapper extends BaseWrapper {
+    static dbSystem: string;
+    static serverAddress: string;
+    static serverPort: number;
+    static _setCommonAttributes(span: any, dbOperation: string, collectionName: string): void;
+    static _patchSearch(tracer: Tracer): any;
+    static _patchUpsert(tracer: Tracer): any;
+    static _patchDelete(tracer: Tracer): any;
+    static _patchRetrieve(tracer: Tracer): any;
+    static _patchScroll(tracer: Tracer): any;
+}
+export default QdrantWrapper;

package/dist/instrumentation/qdrant/wrapper.js

@@ -0,0 +1,213 @@
+"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 QdrantWrapper 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, QdrantWrapper.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, QdrantWrapper.serverAddress);
+        span.setAttribute(semantic_convention_1.default.SERVER_PORT, QdrantWrapper.serverPort);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_ENVIRONMENT, environment);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_APPLICATION_NAME, applicationName);
+    }
+    static _patchSearch(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_SEARCH;
+        return (originalMethod) => {
+            return async function (...args) {
+                // args[0] = collectionName, args[1] = searchParams
+                const collectionName = typeof args[0] === 'string' ? args[0] : 'unknown';
+                const params = args[1] || {};
+                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;
+                        QdrantWrapper._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.limit || 10);
+                        if (params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, JSON.stringify(params.filter));
+                        }
+                        if (params.with_payload !== undefined) {
+                            span.setAttribute(semantic_convention_1.default.DB_WITH_PAYLOAD, String(params.with_payload));
+                        }
+                        const resultCount = Array.isArray(response) ? response.length : 0;
+                        span.setAttribute(semantic_convention_1.default.DB_N_RESULTS, resultCount);
+                        if (config_1.default.captureMessageContent && params.vector) {
+                            const vectorPreview = Array.isArray(params.vector)
+                                ? JSON.stringify(params.vector.slice(0, 10))
+                                : String(params.vector);
+                            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, vectorPreview);
+                        }
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} limit=${params.limit || 10} filtered=${params.filter ? 'true' : 'false'}`);
+                        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 collectionName = typeof args[0] === 'string' ? args[0] : 'unknown';
+                const params = args[1] || {};
+                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 points = params.points || [];
+                        QdrantWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.DB_VECTOR_COUNT, points.length);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} points=${points.length}`);
+                        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 collectionName = typeof args[0] === 'string' ? args[0] : 'unknown';
+                const params = args[1] || {};
+                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;
+                        QdrantWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        const points = params.points || [];
+                        if (points.length > 0) {
+                            span.setAttribute(semantic_convention_1.default.DB_ID_COUNT, points.length);
+                        }
+                        if (params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, JSON.stringify(params.filter));
+                        }
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} points=${points.length}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchRetrieve(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_GET;
+        return (originalMethod) => {
+            return async function (...args) {
+                const collectionName = typeof args[0] === 'string' ? args[0] : 'unknown';
+                const params = args[1] || {};
+                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;
+                        QdrantWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        const ids = Array.isArray(params.ids) ? params.ids : [];
+                        span.setAttribute(semantic_convention_1.default.DB_ID_COUNT, ids.length);
+                        const resultCount = Array.isArray(response) ? response.length : 0;
+                        span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, resultCount);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} ids=${ids.length}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+    static _patchScroll(tracer) {
+        const dbOperation = semantic_convention_1.default.DB_OPERATION_GET;
+        return (originalMethod) => {
+            return async function (...args) {
+                const collectionName = typeof args[0] === 'string' ? args[0] : 'unknown';
+                const params = args[1] || {};
+                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;
+                        QdrantWrapper._setCommonAttributes(span, dbOperation, collectionName);
+                        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, duration);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_LIMIT, params.limit || 10);
+                        if (params.filter) {
+                            span.setAttribute(semantic_convention_1.default.DB_FILTER, JSON.stringify(params.filter));
+                        }
+                        const resultCount = response?.points?.length || 0;
+                        span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, resultCount);
+                        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, `${dbOperation} ${collectionName} limit=${params.limit || 10}`);
+                        span.setStatus({ code: 1 });
+                        return response;
+                    }
+                    catch (e) {
+                        helpers_1.default.handleException(span, e);
+                        throw e;
+                    }
+                    finally {
+                        span.end();
+                    }
+                });
+            };
+        };
+    }
+}
+QdrantWrapper.dbSystem = semantic_convention_1.default.DB_SYSTEM_QDRANT;
+QdrantWrapper.serverAddress = 'localhost';
+QdrantWrapper.serverPort = 6333;
+exports.default = QdrantWrapper;
+//# sourceMappingURL=wrapper.js.map