shield360 1.0.0

package/dist/instrumentation/index.js

@@ -0,0 +1,176 @@
+"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 openai_1 = __importDefault(require("./openai"));
+const anthropic_1 = __importDefault(require("./anthropic"));
+const cohere_1 = __importDefault(require("./cohere"));
+const groq_1 = __importDefault(require("./groq"));
+const ai21_1 = __importDefault(require("./ai21"));
+const gradient_1 = __importDefault(require("./gradient"));
+const browser_use_1 = __importDefault(require("./browser-use"));
+const mistral_1 = __importDefault(require("./mistral"));
+const google_ai_1 = __importDefault(require("./google-ai"));
+const vertexai_1 = __importDefault(require("./vertexai"));
+const together_1 = __importDefault(require("./together"));
+const ollama_1 = __importDefault(require("./ollama"));
+const vllm_1 = __importDefault(require("./vllm"));
+const vercel_ai_1 = __importDefault(require("./vercel-ai"));
+const langchain_1 = __importDefault(require("./langchain"));
+const langgraph_1 = __importDefault(require("./langgraph"));
+const pinecone_1 = __importDefault(require("./pinecone"));
+const bedrock_1 = __importDefault(require("./bedrock"));
+const llamaindex_1 = __importDefault(require("./llamaindex"));
+const huggingface_1 = __importDefault(require("./huggingface"));
+const replicate_1 = __importDefault(require("./replicate"));
+const chroma_1 = __importDefault(require("./chroma"));
+const qdrant_1 = __importDefault(require("./qdrant"));
+const milvus_1 = __importDefault(require("./milvus"));
+const azure_ai_inference_1 = __importDefault(require("./azure-ai-inference"));
+const openai_agents_1 = __importDefault(require("./openai-agents"));
+const strands_1 = __importDefault(require("./strands"));
+const google_adk_1 = __importDefault(require("./google-adk"));
+const claude_agent_sdk_1 = __importDefault(require("./claude-agent-sdk"));
+const cursor_sdk_1 = __importDefault(require("./cursor-sdk"));
+const astra_1 = __importDefault(require("./astra"));
+const mcp_1 = __importDefault(require("./mcp"));
+const mem0_1 = __importDefault(require("./mem0"));
+const elevenlabs_1 = __importDefault(require("./elevenlabs"));
+const assemblyai_1 = __importDefault(require("./assemblyai"));
+const transformers_1 = __importDefault(require("./transformers"));
+const pg_1 = __importDefault(require("./pg"));
+const firecrawl_1 = __importDefault(require("./firecrawl"));
+const letta_1 = __importDefault(require("./letta"));
+/**
+ * OTel community instrumentations loaded dynamically (like Python SDK).
+ * Each entry maps a logical name to the npm package and exported class.
+ * If the package is installed, the instrumentation is registered automatically.
+ * If not installed, it is silently skipped.
+ *
+ * Node.js equivalents of the Python SDK's OTel instrumentations:
+ *   Python: requests, urllib, urllib3  → Node: http  (@opentelemetry/instrumentation-http)
+ *   Python: httpx, aiohttp-client     → Node: undici (@opentelemetry/instrumentation-undici)
+ *   Python: django, flask, fastapi…   → Node: express, fastify, koa, hapi, nestjs-core
+ */
+const OTEL_COMMUNITY_INSTRUMENTATIONS = {
+    // HTTP client instrumentations
+    'http': { pkg: '@opentelemetry/instrumentation-http', cls: 'HttpInstrumentation' },
+    'undici': { pkg: '@opentelemetry/instrumentation-undici', cls: 'UndiciInstrumentation' },
+    // HTTP framework instrumentations
+    'express': { pkg: '@opentelemetry/instrumentation-express', cls: 'ExpressInstrumentation' },
+    'fastify': { pkg: '@opentelemetry/instrumentation-fastify', cls: 'FastifyInstrumentation' },
+    'koa': { pkg: '@opentelemetry/instrumentation-koa', cls: 'KoaInstrumentation' },
+    'hapi': { pkg: '@opentelemetry/instrumentation-hapi', cls: 'HapiInstrumentation' },
+    'nestjs-core': { pkg: '@opentelemetry/instrumentation-nestjs-core', cls: 'NestInstrumentation' },
+};
+function loadOtelCommunityInstrumentations(disabledInstrumentors) {
+    const instances = [];
+    for (const [name, { pkg, cls }] of Object.entries(OTEL_COMMUNITY_INSTRUMENTATIONS)) {
+        if (disabledInstrumentors.includes(name))
+            continue;
+        try {
+            // eslint-disable-next-line @typescript-eslint/no-var-requires
+            const mod = require(pkg);
+            const InstrClass = mod[cls];
+            if (InstrClass) {
+                instances.push(new InstrClass());
+            }
+        }
+        catch {
+            // Package not installed — skip silently (same as Python SDK)
+        }
+    }
+    return instances;
+}
+class Instrumentations {
+    static setup(tracerProvider, disabledInstrumentors = [], instrumentations) {
+        const otelCommunity = loadOtelCommunityInstrumentations(disabledInstrumentors);
+        if (instrumentations === undefined) {
+            const filteredInstrumentations = this.getFilteredInstrumentations(disabledInstrumentors);
+            (0, instrumentation_1.registerInstrumentations)({
+                instrumentations: [
+                    ...filteredInstrumentations.map(([_, instrumentation]) => instrumentation),
+                    ...otelCommunity,
+                ],
+                tracerProvider,
+            });
+        }
+        else {
+            const filteredInstrumentations = this.getFilteredInstrumentations(disabledInstrumentors, instrumentations);
+            filteredInstrumentations.forEach(([k, instrumentation]) => {
+                if (this.availableInstrumentations[k].setTracerProvider) {
+                    this.availableInstrumentations[k].setTracerProvider(tracerProvider);
+                }
+                if (this.availableInstrumentations[k].manualPatch) {
+                    this.availableInstrumentations[k].manualPatch(instrumentation);
+                }
+            });
+            (0, instrumentation_1.registerInstrumentations)({
+                instrumentations: otelCommunity,
+                tracerProvider,
+            });
+        }
+    }
+    static getFilteredInstrumentations(disabledInstrumentors, instrumentations) {
+        const availableInstrumentations = instrumentations || this.availableInstrumentations;
+        return Object.keys(availableInstrumentations)
+            .filter((k) => {
+            if (disabledInstrumentors.includes(k)) {
+                if (typeof availableInstrumentations[k].disable === 'function') {
+                    availableInstrumentations[k].disable();
+                }
+                return false;
+            }
+            if (typeof availableInstrumentations[k].enable === 'function') {
+                availableInstrumentations[k].enable();
+            }
+            return true;
+        })
+            .map((k) => [k, availableInstrumentations[k]]);
+    }
+}
+Instrumentations.availableInstrumentations = {
+    openai: new openai_1.default(),
+    anthropic: new anthropic_1.default(),
+    cohere: new cohere_1.default(),
+    groq: new groq_1.default(),
+    ai21: new ai21_1.default(),
+    gradient: new gradient_1.default(),
+    'browser-use': new browser_use_1.default(),
+    mistral: new mistral_1.default(),
+    'google-ai': new google_ai_1.default(),
+    vertexai: new vertexai_1.default(),
+    together: new together_1.default(),
+    ollama: new ollama_1.default(),
+    vllm: new vllm_1.default(),
+    'vercel-ai': new vercel_ai_1.default(),
+    langchain: new langchain_1.default(),
+    langgraph: new langgraph_1.default(),
+    pinecone: new pinecone_1.default(),
+    bedrock: new bedrock_1.default(),
+    llamaindex: new llamaindex_1.default(),
+    huggingface: new huggingface_1.default(),
+    replicate: new replicate_1.default(),
+    chroma: new chroma_1.default(),
+    qdrant: new qdrant_1.default(),
+    milvus: new milvus_1.default(),
+    'azure-ai-inference': new azure_ai_inference_1.default(),
+    'openai-agents': new openai_agents_1.default(),
+    strands: new strands_1.default(),
+    'google-adk': new google_adk_1.default(),
+    'claude-agent-sdk': new claude_agent_sdk_1.default(),
+    'cursor-sdk': new cursor_sdk_1.default(),
+    'astra': new astra_1.default(),
+    mcp: new mcp_1.default(),
+    mem0: new mem0_1.default(),
+    elevenlabs: new elevenlabs_1.default(),
+    assemblyai: new assemblyai_1.default(),
+    transformers: new transformers_1.default(),
+    pg: new pg_1.default(),
+    firecrawl: new firecrawl_1.default(),
+    letta: new letta_1.default(),
+};
+exports.default = Instrumentations;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/instrumentation/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAA0E;AAI1E,sDAA6C;AAC7C,4DAAmD;AACnD,sDAA6C;AAC7C,kDAAyC;AACzC,kDAAyC;AACzC,0DAAiD;AACjD,gEAAsD;AACtD,wDAA+C;AAC/C,4DAAkD;AAClD,0DAAiD;AACjD,0DAAiD;AACjD,sDAA6C;AAC7C,kDAAyC;AACzC,4DAAkD;AAClD,4DAAmD;AACnD,4DAAmD;AACnD,0DAAiD;AACjD,wDAA+C;AAC/C,8DAAqD;AACrD,gEAAuD;AACvD,4DAAmD;AACnD,sDAA6C;AAC7C,sDAA6C;AAC7C,sDAA6C;AAC7C,8EAAmE;AACnE,oEAA0D;AAC1D,wDAA+C;AAC/C,8DAAoD;AACpD,0EAA+D;AAC/D,8DAAoD;AACpD,oDAA2C;AAC3C,gDAAuC;AACvC,kDAAyC;AACzC,8DAAqD;AACrD,8DAAqD;AACrD,kEAAyD;AACzD,8CAAqC;AACrC,4DAAmD;AACnD,oDAA2C;AAE3C;;;;;;;;;;GAUG;AACH,MAAM,+BAA+B,GAAiD;IACpF,+BAA+B;IAC/B,MAAM,EAAU,EAAE,GAAG,EAAE,qCAAqC,EAAU,GAAG,EAAE,qBAAqB,EAAE;IAClG,QAAQ,EAAQ,EAAE,GAAG,EAAE,uCAAuC,EAAQ,GAAG,EAAE,uBAAuB,EAAE;IACpG,kCAAkC;IAClC,SAAS,EAAO,EAAE,GAAG,EAAE,wCAAwC,EAAO,GAAG,EAAE,wBAAwB,EAAE;IACrG,SAAS,EAAO,EAAE,GAAG,EAAE,wCAAwC,EAAO,GAAG,EAAE,wBAAwB,EAAE;IACrG,KAAK,EAAW,EAAE,GAAG,EAAE,oCAAoC,EAAW,GAAG,EAAE,oBAAoB,EAAE;IACjG,MAAM,EAAU,EAAE,GAAG,EAAE,qCAAqC,EAAU,GAAG,EAAE,qBAAqB,EAAE;IAClG,aAAa,EAAG,EAAE,GAAG,EAAE,4CAA4C,EAAG,GAAG,EAAE,qBAAqB,EAAE;CACnG,CAAC;AAEF,SAAS,iCAAiC,CAAC,qBAA+B;IACxE,MAAM,SAAS,GAAU,EAAE,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,+BAA+B,CAAC,EAAE,CAAC;QACnF,IAAI,qBAAqB,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,SAAS;QACnD,IAAI,CAAC;YACH,8DAA8D;YAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;YACzB,MAAM,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;YAC5B,IAAI,UAAU,EAAE,CAAC;gBACf,SAAS,CAAC,IAAI,CAAC,IAAI,UAAU,EAAE,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,6DAA6D;QAC/D,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAqB,gBAAgB;IA2CnC,MAAM,CAAC,KAAK,CACV,cAA8B,EAC9B,wBAAkC,EAAE,EACpC,gBAA4C;QAE5C,MAAM,aAAa,GAAG,iCAAiC,CAAC,qBAAqB,CAAC,CAAC;QAE/E,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,wBAAwB,GAAG,IAAI,CAAC,2BAA2B,CAAC,qBAAqB,CAAC,CAAC;YACzF,IAAA,0CAAwB,EAAC;gBACvB,gBAAgB,EAAE;oBAChB,GAAG,wBAAwB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;oBAC1E,GAAG,aAAa;iBACjB;gBACD,cAAc;aACf,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,wBAAwB,GAAG,IAAI,CAAC,2BAA2B,CAC/D,qBAAqB,EACrB,gBAAgB,CACjB,CAAC;YACF,wBAAwB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,EAAE,EAAE;gBACxD,IAAI,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;oBACxD,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;gBACtE,CAAC;gBACD,IAAI,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;oBAClD,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC,CAAC,CAAC;YACH,IAAA,0CAAwB,EAAC;gBACvB,gBAAgB,EAAE,aAAa;gBAC/B,cAAc;aACf,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,MAAM,CAAC,2BAA2B,CAChC,qBAA+B,EAC/B,gBAA4C;QAE5C,MAAM,yBAAyB,GAAG,gBAAgB,IAAI,IAAI,CAAC,yBAAyB,CAAC;QACrF,OAAO,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC;aAC1C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;YACZ,IAAI,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gBACtC,IAAI,OAAO,yBAAyB,CAAC,CAAwB,CAAC,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;oBACtF,yBAAyB,CAAC,CAAwB,CAAC,CAAC,OAAO,EAAE,CAAC;gBAChE,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,OAAO,yBAAyB,CAAC,CAAwB,CAAC,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBACrF,yBAAyB,CAAC,CAAwB,CAAC,CAAC,MAAM,EAAE,CAAC;YAC/D,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAwB,EAAE,yBAAyB,CAAC,CAAwB,CAAC,CAAC,CAAC,CAAC;IACjG,CAAC;;AAnGM,0CAAyB,GAA8B;IAC5D,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,IAAI,EAAE,IAAI,cAAmB,EAAE;IAC/B,IAAI,EAAE,IAAI,cAAmB,EAAE;IAC/B,QAAQ,EAAE,IAAI,kBAAuB,EAAE;IACvC,aAAa,EAAE,IAAI,qBAAyB,EAAE;IAC9C,OAAO,EAAE,IAAI,iBAAsB,EAAE;IACrC,WAAW,EAAE,IAAI,mBAAuB,EAAE;IAC1C,QAAQ,EAAE,IAAI,kBAAuB,EAAE;IACvC,QAAQ,EAAE,IAAI,kBAAuB,EAAE;IACvC,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,IAAI,EAAE,IAAI,cAAmB,EAAE;IAC/B,WAAW,EAAE,IAAI,mBAAuB,EAAE;IAC1C,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,QAAQ,EAAE,IAAI,kBAAuB,EAAE;IACvC,OAAO,EAAE,IAAI,iBAAsB,EAAE;IACrC,UAAU,EAAE,IAAI,oBAAyB,EAAE;IAC3C,WAAW,EAAE,IAAI,qBAA0B,EAAE;IAC7C,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,MAAM,EAAE,IAAI,gBAAqB,EAAE;IACnC,oBAAoB,EAAE,IAAI,4BAA+B,EAAE;IAC3D,eAAe,EAAE,IAAI,uBAA2B,EAAE;IAClD,OAAO,EAAE,IAAI,iBAAsB,EAAE;IACrC,YAAY,EAAE,IAAI,oBAAwB,EAAE;IAC5C,kBAAkB,EAAE,IAAI,0BAA6B,EAAE;IACvD,YAAY,EAAE,IAAI,oBAAwB,EAAE;IAC5C,OAAO,EAAE,IAAI,eAAoB,EAAE;IACnC,GAAG,EAAE,IAAI,aAAkB,EAAE;IAC7B,IAAI,EAAE,IAAI,cAAmB,EAAE;IAC/B,UAAU,EAAE,IAAI,oBAAyB,EAAE;IAC3C,UAAU,EAAE,IAAI,oBAAyB,EAAE;IAC3C,YAAY,EAAE,IAAI,sBAA2B,EAAE;IAC/C,EAAE,EAAE,IAAI,YAAiB,EAAE;IAC3B,SAAS,EAAE,IAAI,mBAAwB,EAAE;IACzC,KAAK,EAAE,IAAI,eAAoB,EAAE;CAClC,CAAC;kBAzCiB,gBAAgB"}

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

@@ -0,0 +1,16 @@
+import { InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+import { InstrumentationConfig } from '@opentelemetry/instrumentation';
+export interface LangChainInstrumentationConfig extends InstrumentationConfig {
+}
+export default class Shield360LangChainInstrumentation extends InstrumentationBase {
+    private _callbackManager;
+    private _ritmHook;
+    constructor(config?: LangChainInstrumentationConfig);
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    enable(): void;
+    disable(): void;
+    manualPatch(callbacksManagerModule: any): void;
+    private _patchFromCache;
+    private _applyPatch;
+    private _unpatch;
+}

package/dist/instrumentation/langchain/index.js

@@ -0,0 +1,91 @@
+"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 Shield360LangChainInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-langchain`, '1.0.0', config);
+        this._callbackManager = null;
+        this._ritmHook = null;
+    }
+    init() {
+        return [];
+    }
+    enable() {
+        super.enable();
+        if (this._ritmHook)
+            return;
+        try {
+            // eslint-disable-next-line @typescript-eslint/no-var-requires
+            const { Hook } = require('require-in-the-middle');
+            this._ritmHook = new Hook(['@langchain/core/language_models/chat_models'], { internals: true }, (exports) => {
+                this._patchFromCache();
+                return exports;
+            });
+        }
+        catch { /* require-in-the-middle not available */ }
+    }
+    disable() {
+        super.disable();
+        this._ritmHook?.unhook?.();
+        this._ritmHook = null;
+        this._unpatch();
+    }
+    manualPatch(callbacksManagerModule) {
+        const CallbackManager = callbacksManagerModule?.CallbackManager;
+        if (CallbackManager) {
+            this._callbackManager = CallbackManager;
+            this._applyPatch(CallbackManager);
+        }
+    }
+    _patchFromCache() {
+        try {
+            const cache = require.cache;
+            for (const filepath of Object.keys(cache)) {
+                if (filepath.includes('@langchain') &&
+                    filepath.includes('callbacks') &&
+                    filepath.includes('manager') &&
+                    !filepath.endsWith('.map')) {
+                    const mod = cache[filepath];
+                    const CallbackManager = mod?.exports?.CallbackManager;
+                    if (CallbackManager && typeof CallbackManager._configureSync === 'function') {
+                        this._callbackManager = CallbackManager;
+                        this._applyPatch(CallbackManager);
+                        return;
+                    }
+                }
+            }
+        }
+        catch (e) {
+            console.error('LangChain: error scanning require.cache for CallbackManager:', e);
+        }
+    }
+    _applyPatch(CallbackManager) {
+        try {
+            if ((0, instrumentation_1.isWrapped)(CallbackManager._configureSync)) {
+                this._unwrap(CallbackManager, '_configureSync');
+            }
+            this._wrap(CallbackManager, '_configureSync', wrapper_1.default._patchConfigure(this.tracer));
+        }
+        catch (e) {
+            console.error('Error in LangChain _patch method:', e);
+        }
+    }
+    _unpatch() {
+        if (!this._callbackManager)
+            return;
+        try {
+            if ((0, instrumentation_1.isWrapped)(this._callbackManager._configureSync)) {
+                this._unwrap(this._callbackManager, '_configureSync');
+            }
+        }
+        catch { /* ignore */ }
+        this._callbackManager = null;
+    }
+}
+exports.default = Shield360LangChainInstrumentation;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/langchain/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAIwC;AAExC,6CAAwD;AACxD,wDAAyC;AAIzC,MAAqB,iCAAkC,SAAQ,qCAAmB;IAIhF,YAAY,SAAyC,EAAE;QACrD,KAAK,CAAC,GAAG,iCAAsB,4BAA4B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAJxE,qBAAgB,GAAQ,IAAI,CAAC;QAC7B,cAAS,GAAQ,IAAI,CAAC;IAI9B,CAAC;IAES,IAAI;QACZ,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,MAAM;QACX,KAAK,CAAC,MAAM,EAAE,CAAC;QACf,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC;YACH,8DAA8D;YAC9D,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,uBAAuB,CAAC,CAAC;YAClD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,CACvB,CAAC,6CAA6C,CAAC,EAC/C,EAAE,SAAS,EAAE,IAAI,EAAE,EACnB,CAAC,OAAY,EAAE,EAAE;gBACf,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC;YACjB,CAAC,CACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC,CAAC,yCAAyC,CAAC,CAAC;IACvD,CAAC;IAEM,OAAO;QACZ,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,sBAA2B;QAC5C,MAAM,eAAe,GAAG,sBAAsB,EAAE,eAAe,CAAC;QAChE,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;YACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC;YACH,MAAM,KAAK,GAAI,OAA2D,CAAC,KAAK,CAAC;YACjF,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,IACE,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;oBAC/B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;oBAC9B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAC5B,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;oBACD,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC;oBAC5B,MAAM,eAAe,GAAG,GAAG,EAAE,OAAO,EAAE,eAAe,CAAC;oBACtD,IAAI,eAAe,IAAI,OAAO,eAAe,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;wBAC5E,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;wBACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;wBAClC,OAAO;oBACT,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,8DAA8D,EAAE,CAAC,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,eAAoB;QACtC,IAAI,CAAC;YACH,IAAI,IAAA,2BAAS,EAAC,eAAe,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC9C,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;YAClD,CAAC;YACD,IAAI,CAAC,KAAK,CACR,eAAe,EACf,gBAAgB,EAChB,iBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAC9C,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QACnC,IAAI,CAAC;YACH,IAAI,IAAA,2BAAS,EAAC,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QACxB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAC/B,CAAC;CACF;AA5FD,oDA4FC"}

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

@@ -0,0 +1,41 @@
+import { Tracer } from '@opentelemetry/api';
+import { runWithFrameworkLlm } from '../../helpers';
+import BaseWrapper from '../base-wrapper';
+declare class Shield360CallbackHandler {
+    name: string;
+    lc_serializable: boolean;
+    awaitHandlers: boolean;
+    private tracer;
+    private spans;
+    private skippedRuns;
+    constructor(tracer: Tracer);
+    private _getNameFromCallback;
+    private _resolveParentRunId;
+    private _getParentContext;
+    private _createSpan;
+    private _newHolder;
+    private _setCommonAttributes;
+    private _setModelParameters;
+    private _endSpan;
+    handleChainStart(chain: any, inputs: any, runId: string, parentRunId?: string, _tags?: string[], metadata?: any, _runType?: string, name?: string): void;
+    handleChainEnd(outputs: any, runId: string): void;
+    handleChainError(error: any, runId: string): void;
+    handleChatModelStart(llm: any, messages: any[][], runId: string, parentRunId?: string, _extraParams?: any, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleLLMStart(serialized: any, prompts: string[], runId: string, parentRunId?: string, _extraParams?: any, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleLLMNewToken(token: string, _idx: any, runId: string, _chunk?: any): void;
+    handleLLMEnd(output: any, runId: string): void;
+    handleLLMError(error: any, runId: string): void;
+    handleToolStart(tool: any, input: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleToolEnd(output: any, runId: string): void;
+    handleToolError(error: any, runId: string): void;
+    handleRetrieverStart(retriever: any, query: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleRetrieverEnd(documents: any[], runId: string): void;
+    handleRetrieverError(error: any, runId: string): void;
+    handleAgentAction(action: any, runId: string): void;
+    handleAgentFinish(finish: any, runId: string): void;
+}
+declare class LangChainWrapper extends BaseWrapper {
+    static _patchConfigure(tracer: Tracer): any;
+}
+export { Shield360CallbackHandler, runWithFrameworkLlm };
+export default LangChainWrapper;