shield360 1.0.0

package/dist/instrumentation/google-adk/index.js

@@ -0,0 +1,371 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const api_1 = require("@opentelemetry/api");
+const constant_1 = require("../../constant");
+const utils_1 = require("./utils");
+const wrapper_1 = require("./wrapper");
+const config_1 = __importDefault(require("../../config"));
+/**
+ * OTel GenAI semantic convention compliant instrumentor for Google ADK.
+ *
+ * Monkey-patches Runner, BaseAgent, and LlmAgent from @google/adk.
+ * Replaces ADK's internal tracers with PassthroughTracer to suppress
+ * duplicate spans. Enriches ADK's LLM and tool spans with OTel attributes.
+ */
+class GoogleADKInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-google-adk`, '1.0.0', config);
+        this._registry = new wrapper_1.AgentCreationRegistry();
+        this._originalTracers = {};
+        this._originalTraceCallLlm = null;
+        this._originalTraceToolCall = null;
+        this._originalTraceMergedToolCalls = null;
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('@google/adk', ['>=1.2.0'], (moduleExports) => {
+            this._patchAll(moduleExports);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatchAll(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(adkModule) {
+        this._patchAll(adkModule);
+    }
+    // ---------------------------------------------------------------------------
+    // Patching
+    // ---------------------------------------------------------------------------
+    _patchAll(moduleExports) {
+        try {
+            this._disableExistingTracers(moduleExports);
+            this._wrapTraceCallLlm(moduleExports);
+            this._wrapTraceToolCall(moduleExports);
+            this._wrapTraceMergedToolCalls(moduleExports);
+            this._patchLlmAgent(moduleExports);
+            this._patchRunner(moduleExports);
+            this._patchBaseAgent(moduleExports);
+        }
+        catch { /* graceful degradation */ }
+    }
+    /**
+     * Replace ADK's internal tracers with PassthroughTracer to suppress
+     * duplicate top-level spans (mirrors Python _disable_existing_tracers).
+     */
+    _disableExistingTracers(moduleExports) {
+        try {
+            const runners = this._resolveSubmodule(moduleExports, 'runners');
+            if (runners && runners.tracer) {
+                this._originalTracers['runners.tracer'] = runners.tracer;
+                runners.tracer = new utils_1.PassthroughTracer(runners.tracer);
+            }
+        }
+        catch { /* ignore */ }
+        try {
+            const baseAgentMod = this._resolveSubmodule(moduleExports, 'agents', 'base_agent') ??
+                this._resolveSubmodule(moduleExports, 'agents');
+            if (baseAgentMod && baseAgentMod.tracer) {
+                this._originalTracers['base_agent.tracer'] = baseAgentMod.tracer;
+                baseAgentMod.tracer = new utils_1.PassthroughTracer(baseAgentMod.tracer);
+            }
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Wrap ADK's trace_call_llm to enrich the current span with OTel GenAI attributes.
+     */
+    _wrapTraceCallLlm(moduleExports) {
+        try {
+            const blfMod = this._resolveSubmodule(moduleExports, 'flows', 'llm_flows', 'base_llm_flow') ??
+                this._resolveSubmodule(moduleExports, 'flows');
+            if (!blfMod)
+                return;
+            const originalFn = blfMod.trace_call_llm ?? blfMod.traceCallLlm;
+            if (!originalFn)
+                return;
+            this._originalTraceCallLlm = originalFn;
+            const captureContent = () => config_1.default.captureMessageContent ?? true;
+            const enriched = (...args) => {
+                const result = originalFn(...args);
+                try {
+                    const span = api_1.trace.getActiveSpan();
+                    if (span) {
+                        const llmRequest = args[2];
+                        const llmResponse = args[3];
+                        (0, utils_1.enrichLlmSpan)(span, llmRequest, llmResponse, captureContent());
+                    }
+                }
+                catch { /* ignore */ }
+                return result;
+            };
+            if (blfMod.trace_call_llm) {
+                blfMod.trace_call_llm = enriched;
+            }
+            else if (blfMod.traceCallLlm) {
+                blfMod.traceCallLlm = enriched;
+            }
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Wrap ADK's trace_tool_call to enrich the current span with tool attributes.
+     */
+    _wrapTraceToolCall(moduleExports) {
+        try {
+            const fnMod = this._resolveSubmodule(moduleExports, 'flows', 'llm_flows', 'functions') ??
+                this._resolveSubmodule(moduleExports, 'flows');
+            if (!fnMod)
+                return;
+            const originalFn = fnMod.trace_tool_call ?? fnMod.traceToolCall;
+            if (!originalFn)
+                return;
+            this._originalTraceToolCall = originalFn;
+            const captureContent = () => config_1.default.captureMessageContent ?? true;
+            const enriched = (...args) => {
+                const result = originalFn(...args);
+                try {
+                    const span = api_1.trace.getActiveSpan();
+                    if (span) {
+                        const tool = args[0];
+                        const functionArgs = args[1];
+                        const functionResponseEvent = args[2];
+                        const toolError = args[3];
+                        (0, utils_1.enrichToolSpan)(span, tool, functionArgs, functionResponseEvent, captureContent(), toolError);
+                    }
+                }
+                catch { /* ignore */ }
+                return result;
+            };
+            if (fnMod.trace_tool_call) {
+                fnMod.trace_tool_call = enriched;
+            }
+            else if (fnMod.traceToolCall) {
+                fnMod.traceToolCall = enriched;
+            }
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Wrap ADK's trace_merged_tool_calls to enrich the current span with merged tool attributes.
+     */
+    _wrapTraceMergedToolCalls(moduleExports) {
+        try {
+            const fnMod = this._resolveSubmodule(moduleExports, 'flows', 'llm_flows', 'functions') ??
+                this._resolveSubmodule(moduleExports, 'flows');
+            if (!fnMod)
+                return;
+            const originalFn = fnMod.trace_merged_tool_calls ?? fnMod.traceMergedToolCalls;
+            if (!originalFn)
+                return;
+            this._originalTraceMergedToolCalls = originalFn;
+            const captureContent = () => config_1.default.captureMessageContent ?? true;
+            const enriched = (...args) => {
+                const result = originalFn(...args);
+                try {
+                    const span = api_1.trace.getActiveSpan();
+                    if (span) {
+                        const responseEventId = args[0];
+                        const functionResponseEvent = args[1];
+                        (0, utils_1.enrichMergedToolSpan)(span, responseEventId, functionResponseEvent, captureContent());
+                    }
+                }
+                catch { /* ignore */ }
+                return result;
+            };
+            if (fnMod.trace_merged_tool_calls) {
+                fnMod.trace_merged_tool_calls = enriched;
+            }
+            else if (fnMod.traceMergedToolCalls) {
+                fnMod.traceMergedToolCalls = enriched;
+            }
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Patch LlmAgent constructor to create create_agent spans.
+     */
+    _patchLlmAgent(moduleExports) {
+        try {
+            const LlmAgent = moduleExports.LlmAgent ?? moduleExports.Agent;
+            if (!LlmAgent?.prototype)
+                return;
+            const originalClass = LlmAgent;
+            const patchFn = (0, wrapper_1.wrapAgentInit)(this.tracer, this._registry);
+            // Wrap any init-like method if available; otherwise wrap the class prototype methods
+            // ADK's LlmAgent may expose initialization through sub_agents, tools setters, etc.
+            // The most reliable approach is patching the constructor via a proxy pattern.
+            if (typeof originalClass === 'function') {
+                const wrappedConstructor = patchFn(function (...args) {
+                    return Reflect.construct(originalClass, args, new.target || originalClass);
+                });
+                // Copy static properties
+                Object.setPrototypeOf(wrappedConstructor, originalClass);
+                wrappedConstructor.prototype = originalClass.prototype;
+                // ESM-to-CJS interop may define exports as getter-only properties;
+                // both defineProperty and assignment may fail for ESM Module Namespace objects.
+                const exportName = moduleExports.LlmAgent ? 'LlmAgent' : 'Agent';
+                try {
+                    Object.defineProperty(moduleExports, exportName, {
+                        enumerable: true,
+                        configurable: true,
+                        writable: true,
+                        value: wrappedConstructor,
+                    });
+                }
+                catch {
+                    try {
+                        moduleExports[exportName] = wrappedConstructor;
+                    }
+                    catch { /* strict mode */ }
+                }
+            }
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Patch Runner.run and Runner.runAsync (or run_async).
+     */
+    _patchRunner(moduleExports) {
+        try {
+            const Runner = moduleExports.Runner;
+            if (!Runner?.prototype)
+                return;
+            // Runner.run (sync)
+            const runMethod = Runner.prototype.run;
+            if (runMethod && !(0, instrumentation_1.isWrapped)(runMethod)) {
+                this._wrap(Runner.prototype, 'run', (0, wrapper_1.wrapRunnerRun)(this.tracer, 'runner_run', this._registry));
+            }
+            // Runner.runAsync or Runner.run_async (async generator)
+            const runAsyncName = Runner.prototype.runAsync ? 'runAsync' : 'run_async';
+            const runAsyncMethod = Runner.prototype[runAsyncName];
+            if (runAsyncMethod) {
+                if ((0, instrumentation_1.isWrapped)(runAsyncMethod))
+                    this._unwrap(Runner.prototype, runAsyncName);
+                this._wrap(Runner.prototype, runAsyncName, (0, wrapper_1.wrapRunnerRunAsync)(this.tracer, 'runner_run_async', this._registry));
+            }
+            // Runner.runLive or Runner.run_live (async generator)
+            const runLiveName = Runner.prototype.runLive ? 'runLive' : 'run_live';
+            const runLiveMethod = Runner.prototype[runLiveName];
+            if (runLiveMethod) {
+                if ((0, instrumentation_1.isWrapped)(runLiveMethod))
+                    this._unwrap(Runner.prototype, runLiveName);
+                this._wrap(Runner.prototype, runLiveName, (0, wrapper_1.wrapRunnerRunAsync)(this.tracer, 'runner_run_live', this._registry));
+            }
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Patch BaseAgent.runAsync (or run_async).
+     */
+    _patchBaseAgent(moduleExports) {
+        try {
+            const BaseAgent = moduleExports.BaseAgent;
+            if (!BaseAgent?.prototype)
+                return;
+            const runAsyncName = BaseAgent.prototype.runAsync ? 'runAsync' : 'run_async';
+            const runAsyncMethod = BaseAgent.prototype[runAsyncName];
+            if (runAsyncMethod) {
+                if ((0, instrumentation_1.isWrapped)(runAsyncMethod))
+                    this._unwrap(BaseAgent.prototype, runAsyncName);
+                this._wrap(BaseAgent.prototype, runAsyncName, (0, wrapper_1.wrapAgentRunAsync)(this.tracer, 'agent_run_async', this._registry));
+            }
+        }
+        catch { /* ignore */ }
+    }
+    // ---------------------------------------------------------------------------
+    // Submodule resolution
+    // ---------------------------------------------------------------------------
+    /**
+     * Attempt to resolve a nested submodule from the ADK exports.
+     * ADK's JS package may organize exports differently from Python.
+     */
+    _resolveSubmodule(moduleExports, ...path) {
+        let current = moduleExports;
+        for (const segment of path) {
+            if (!current)
+                return null;
+            current = current[segment];
+        }
+        return current || null;
+    }
+    // ---------------------------------------------------------------------------
+    // Unpatching
+    // ---------------------------------------------------------------------------
+    _unpatchAll(moduleExports) {
+        try {
+            // Restore original tracers
+            if (this._originalTracers['runners.tracer']) {
+                const runners = this._resolveSubmodule(moduleExports, 'runners');
+                if (runners)
+                    runners.tracer = this._originalTracers['runners.tracer'];
+            }
+            if (this._originalTracers['base_agent.tracer']) {
+                const baseAgentMod = this._resolveSubmodule(moduleExports, 'agents', 'base_agent') ??
+                    this._resolveSubmodule(moduleExports, 'agents');
+                if (baseAgentMod)
+                    baseAgentMod.tracer = this._originalTracers['base_agent.tracer'];
+            }
+            // Restore original tracing functions
+            if (this._originalTraceCallLlm) {
+                const blfMod = this._resolveSubmodule(moduleExports, 'flows', 'llm_flows', 'base_llm_flow') ??
+                    this._resolveSubmodule(moduleExports, 'flows');
+                if (blfMod) {
+                    if (blfMod.trace_call_llm)
+                        blfMod.trace_call_llm = this._originalTraceCallLlm;
+                    else if (blfMod.traceCallLlm)
+                        blfMod.traceCallLlm = this._originalTraceCallLlm;
+                }
+            }
+            if (this._originalTraceToolCall) {
+                const fnMod = this._resolveSubmodule(moduleExports, 'flows', 'llm_flows', 'functions') ??
+                    this._resolveSubmodule(moduleExports, 'flows');
+                if (fnMod) {
+                    if (fnMod.trace_tool_call)
+                        fnMod.trace_tool_call = this._originalTraceToolCall;
+                    else if (fnMod.traceToolCall)
+                        fnMod.traceToolCall = this._originalTraceToolCall;
+                }
+            }
+            if (this._originalTraceMergedToolCalls) {
+                const fnMod = this._resolveSubmodule(moduleExports, 'flows', 'llm_flows', 'functions') ??
+                    this._resolveSubmodule(moduleExports, 'flows');
+                if (fnMod) {
+                    if (fnMod.trace_merged_tool_calls)
+                        fnMod.trace_merged_tool_calls = this._originalTraceMergedToolCalls;
+                    else if (fnMod.traceMergedToolCalls)
+                        fnMod.traceMergedToolCalls = this._originalTraceMergedToolCalls;
+                }
+            }
+            // Unwrap patched methods
+            const Runner = moduleExports.Runner;
+            if (Runner?.prototype) {
+                if ((0, instrumentation_1.isWrapped)(Runner.prototype.run))
+                    this._unwrap(Runner.prototype, 'run');
+                const runAsyncName = Runner.prototype.runAsync ? 'runAsync' : 'run_async';
+                if (Runner.prototype[runAsyncName] && (0, instrumentation_1.isWrapped)(Runner.prototype[runAsyncName])) {
+                    this._unwrap(Runner.prototype, runAsyncName);
+                }
+                const runLiveName = Runner.prototype.runLive ? 'runLive' : 'run_live';
+                if (Runner.prototype[runLiveName] && (0, instrumentation_1.isWrapped)(Runner.prototype[runLiveName])) {
+                    this._unwrap(Runner.prototype, runLiveName);
+                }
+            }
+            const BaseAgent = moduleExports.BaseAgent;
+            if (BaseAgent?.prototype) {
+                const runAsyncName = BaseAgent.prototype.runAsync ? 'runAsync' : 'run_async';
+                if (BaseAgent.prototype[runAsyncName] && (0, instrumentation_1.isWrapped)(BaseAgent.prototype[runAsyncName])) {
+                    this._unwrap(BaseAgent.prototype, runAsyncName);
+                }
+            }
+        }
+        catch { /* ignore */ }
+    }
+}
+exports.default = GoogleADKInstrumentation;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/google-adk/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/google-adk/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAKwC;AAExC,4CAA2C;AAC3C,6CAAwD;AACxD,mCAKiB;AACjB,uCAMmB;AACnB,0DAA2C;AAI3C;;;;;;GAMG;AACH,MAAqB,wBAAyB,SAAQ,qCAAmB;IAOvE,YAAY,SAAyC,EAAE;QACrD,KAAK,CAAC,GAAG,iCAAsB,6BAA6B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAPzE,cAAS,GAAG,IAAI,+BAAqB,EAAE,CAAC;QACxC,qBAAgB,GAAwB,EAAE,CAAC;QAC3C,0BAAqB,GAAQ,IAAI,CAAC;QAClC,2BAAsB,GAAQ,IAAI,CAAC;QACnC,kCAA6B,GAAQ,IAAI,CAAC;IAIlD,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,aAAa,EACb,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAC9B,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CACF,CAAC;QAEF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,SAAc;QAC/B,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC5B,CAAC;IAED,8EAA8E;IAC9E,WAAW;IACX,8EAA8E;IAEtE,SAAS,CAAC,aAAkB;QAClC,IAAI,CAAC;YACH,IAAI,CAAC,uBAAuB,CAAC,aAAa,CAAC,CAAC;YAC5C,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YACtC,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,CAAC;YACvC,IAAI,CAAC,yBAAyB,CAAC,aAAa,CAAC,CAAC;YAC9C,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YACnC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YACjC,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;QACtC,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;IACxC,CAAC;IAED;;;OAGG;IACK,uBAAuB,CAAC,aAAkB;QAChD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;YACjE,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC9B,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;gBACzD,OAAO,CAAC,MAAM,GAAG,IAAI,yBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QAExB,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,QAAQ,EAAE,YAAY,CAAC;gBAChF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YAClD,IAAI,YAAY,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;gBACxC,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,GAAG,YAAY,CAAC,MAAM,CAAC;gBACjE,YAAY,CAAC,MAAM,GAAG,IAAI,yBAAiB,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YACnE,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,aAAkB;QAC1C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,CAAC;gBACzF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;YACjD,IAAI,CAAC,MAAM;gBAAE,OAAO;YAEpB,MAAM,UAAU,GAAG,MAAM,CAAC,cAAc,IAAI,MAAM,CAAC,YAAY,CAAC;YAChE,IAAI,CAAC,UAAU;gBAAE,OAAO;YAExB,IAAI,CAAC,qBAAqB,GAAG,UAAU,CAAC;YACxC,MAAM,cAAc,GAAG,GAAG,EAAE,CAAC,gBAAe,CAAC,qBAAqB,IAAI,IAAI,CAAC;YAE3E,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAW,EAAE,EAAE;gBAClC,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,WAAK,CAAC,aAAa,EAAE,CAAC;oBACnC,IAAI,IAAI,EAAE,CAAC;wBACT,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBAC3B,MAAM,WAAW,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBAC5B,IAAA,qBAAa,EAAC,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC,CAAC;oBACjE,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxB,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YAEF,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;gBAC1B,MAAM,CAAC,cAAc,GAAG,QAAQ,CAAC;YACnC,CAAC;iBAAM,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;gBAC/B,MAAM,CAAC,YAAY,GAAG,QAAQ,CAAC;YACjC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,aAAkB;QAC3C,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC;gBACpF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;YACjD,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,UAAU,GAAG,KAAK,CAAC,eAAe,IAAI,KAAK,CAAC,aAAa,CAAC;YAChE,IAAI,CAAC,UAAU;gBAAE,OAAO;YAExB,IAAI,CAAC,sBAAsB,GAAG,UAAU,CAAC;YACzC,MAAM,cAAc,GAAG,GAAG,EAAE,CAAC,gBAAe,CAAC,qBAAqB,IAAI,IAAI,CAAC;YAE3E,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAW,EAAE,EAAE;gBAClC,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,WAAK,CAAC,aAAa,EAAE,CAAC;oBACnC,IAAI,IAAI,EAAE,CAAC;wBACT,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBACrB,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBAC7B,MAAM,qBAAqB,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBACtC,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBAC1B,IAAA,sBAAc,EAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,qBAAqB,EAAE,cAAc,EAAE,EAAE,SAAS,CAAC,CAAC;oBAC/F,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxB,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YAEF,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;gBAC1B,KAAK,CAAC,eAAe,GAAG,QAAQ,CAAC;YACnC,CAAC;iBAAM,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBAC/B,KAAK,CAAC,aAAa,GAAG,QAAQ,CAAC;YACjC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,yBAAyB,CAAC,aAAkB;QAClD,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC;gBACpF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;YACjD,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,UAAU,GAAG,KAAK,CAAC,uBAAuB,IAAI,KAAK,CAAC,oBAAoB,CAAC;YAC/E,IAAI,CAAC,UAAU;gBAAE,OAAO;YAExB,IAAI,CAAC,6BAA6B,GAAG,UAAU,CAAC;YAChD,MAAM,cAAc,GAAG,GAAG,EAAE,CAAC,gBAAe,CAAC,qBAAqB,IAAI,IAAI,CAAC;YAE3E,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAW,EAAE,EAAE;gBAClC,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,WAAK,CAAC,aAAa,EAAE,CAAC;oBACnC,IAAI,IAAI,EAAE,CAAC;wBACT,MAAM,eAAe,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBAChC,MAAM,qBAAqB,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBACtC,IAAA,4BAAoB,EAAC,IAAI,EAAE,eAAe,EAAE,qBAAqB,EAAE,cAAc,EAAE,CAAC,CAAC;oBACvF,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxB,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YAEF,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;gBAClC,KAAK,CAAC,uBAAuB,GAAG,QAAQ,CAAC;YAC3C,CAAC;iBAAM,IAAI,KAAK,CAAC,oBAAoB,EAAE,CAAC;gBACtC,KAAK,CAAC,oBAAoB,GAAG,QAAQ,CAAC;YACxC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,aAAkB;QACvC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,aAAa,CAAC,QAAQ,IAAI,aAAa,CAAC,KAAK,CAAC;YAC/D,IAAI,CAAC,QAAQ,EAAE,SAAS;gBAAE,OAAO;YAEjC,MAAM,aAAa,GAAG,QAAQ,CAAC;YAC/B,MAAM,OAAO,GAAG,IAAA,uBAAa,EAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAE3D,qFAAqF;YACrF,mFAAmF;YACnF,8EAA8E;YAC9E,IAAI,OAAO,aAAa,KAAK,UAAU,EAAE,CAAC;gBACxC,MAAM,kBAAkB,GAAG,OAAO,CAAC,UAAqB,GAAG,IAAW;oBACpE,OAAO,OAAO,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,IAAI,aAAa,CAAC,CAAC;gBAC7E,CAAC,CAAC,CAAC;gBAEH,yBAAyB;gBACzB,MAAM,CAAC,cAAc,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;gBACzD,kBAAkB,CAAC,SAAS,GAAG,aAAa,CAAC,SAAS,CAAC;gBAEvD,mEAAmE;gBACnE,gFAAgF;gBAChF,MAAM,UAAU,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC;gBACjE,IAAI,CAAC;oBACH,MAAM,CAAC,cAAc,CAAC,aAAa,EAAE,UAAU,EAAE;wBAC/C,UAAU,EAAE,IAAI;wBAChB,YAAY,EAAE,IAAI;wBAClB,QAAQ,EAAE,IAAI;wBACd,KAAK,EAAE,kBAAkB;qBAC1B,CAAC,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACP,IAAI,CAAC;wBAAE,aAAqB,CAAC,UAAU,CAAC,GAAG,kBAAkB,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC;gBAC9F,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,aAAkB;QACrC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;YACpC,IAAI,CAAC,MAAM,EAAE,SAAS;gBAAE,OAAO;YAE/B,oBAAoB;YACpB,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC;YACvC,IAAI,SAAS,IAAI,CAAC,IAAA,2BAAS,EAAC,SAAS,CAAC,EAAE,CAAC;gBACvC,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,SAAS,EAChB,KAAK,EACL,IAAA,uBAAa,EAAC,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,CACzD,CAAC;YACJ,CAAC;YAED,wDAAwD;YACxD,MAAM,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC;YAC1E,MAAM,cAAc,GAAG,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACtD,IAAI,cAAc,EAAE,CAAC;gBACnB,IAAI,IAAA,2BAAS,EAAC,cAAc,CAAC;oBAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;gBAC5E,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,SAAS,EAChB,YAAY,EACZ,IAAA,4BAAkB,EAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,CACpE,CAAC;YACJ,CAAC;YAED,sDAAsD;YACtD,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC;YACtE,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACpD,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,IAAA,2BAAS,EAAC,aAAa,CAAC;oBAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;gBAC1E,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,SAAS,EAChB,WAAW,EACX,IAAA,4BAAkB,EAAC,IAAI,CAAC,MAAM,EAAE,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,CACnE,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,aAAkB;QACxC,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,aAAa,CAAC,SAAS,CAAC;YAC1C,IAAI,CAAC,SAAS,EAAE,SAAS;gBAAE,OAAO;YAElC,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC;YAC7E,MAAM,cAAc,GAAG,SAAS,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACzD,IAAI,cAAc,EAAE,CAAC;gBACnB,IAAI,IAAA,2BAAS,EAAC,cAAc,CAAC;oBAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;gBAC/E,IAAI,CAAC,KAAK,CACR,SAAS,CAAC,SAAS,EACnB,YAAY,EACZ,IAAA,2BAAiB,EAAC,IAAI,CAAC,MAAM,EAAE,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,CAClE,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED,8EAA8E;IAC9E,uBAAuB;IACvB,8EAA8E;IAE9E;;;OAGG;IACK,iBAAiB,CAAC,aAAkB,EAAE,GAAG,IAAc;QAC7D,IAAI,OAAO,GAAG,aAAa,CAAC;QAC5B,KAAK,MAAM,OAAO,IAAI,IAAI,EAAE,CAAC;YAC3B,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC;YAC1B,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,OAAO,IAAI,IAAI,CAAC;IACzB,CAAC;IAED,8EAA8E;IAC9E,aAAa;IACb,8EAA8E;IAEtE,WAAW,CAAC,aAAkB;QACpC,IAAI,CAAC;YACH,2BAA2B;YAC3B,IAAI,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;gBACjE,IAAI,OAAO;oBAAE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;YACxE,CAAC;YACD,IAAI,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,QAAQ,EAAE,YAAY,CAAC;oBAChF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;gBAClD,IAAI,YAAY;oBAAE,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,CAAC;YACrF,CAAC;YAED,qCAAqC;YACrC,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,CAAC;oBACzF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;gBACjD,IAAI,MAAM,EAAE,CAAC;oBACX,IAAI,MAAM,CAAC,cAAc;wBAAE,MAAM,CAAC,cAAc,GAAG,IAAI,CAAC,qBAAqB,CAAC;yBACzE,IAAI,MAAM,CAAC,YAAY;wBAAE,MAAM,CAAC,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC;gBACjF,CAAC;YACH,CAAC;YACD,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBAChC,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC;oBACpF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;gBACjD,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,KAAK,CAAC,eAAe;wBAAE,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,sBAAsB,CAAC;yBAC1E,IAAI,KAAK,CAAC,aAAa;wBAAE,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC,sBAAsB,CAAC;gBAClF,CAAC;YACH,CAAC;YACD,IAAI,IAAI,CAAC,6BAA6B,EAAE,CAAC;gBACvC,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC;oBACpF,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;gBACjD,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,KAAK,CAAC,uBAAuB;wBAAE,KAAK,CAAC,uBAAuB,GAAG,IAAI,CAAC,6BAA6B,CAAC;yBACjG,IAAI,KAAK,CAAC,oBAAoB;wBAAE,KAAK,CAAC,oBAAoB,GAAG,IAAI,CAAC,6BAA6B,CAAC;gBACvG,CAAC;YACH,CAAC;YAED,yBAAyB;YACzB,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;YACpC,IAAI,MAAM,EAAE,SAAS,EAAE,CAAC;gBACtB,IAAI,IAAA,2BAAS,EAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC;oBAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;gBAC3E,MAAM,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC;gBAC1E,IAAI,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,IAAA,2BAAS,EAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;oBAChF,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;gBAC/C,CAAC;gBACD,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC;gBACtE,IAAI,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,IAAA,2BAAS,EAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC;oBAC9E,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YAED,MAAM,SAAS,GAAG,aAAa,CAAC,SAAS,CAAC;YAC1C,IAAI,SAAS,EAAE,SAAS,EAAE,CAAC;gBACzB,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC;gBAC7E,IAAI,SAAS,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,IAAA,2BAAS,EAAC,SAAS,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;oBACtF,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;CACF;AAlXD,2CAkXC"}

package/dist/instrumentation/google-adk/utils.d.ts

@@ -0,0 +1,45 @@
+import { Span, SpanKind } from '@opentelemetry/api';
+import { AsyncLocalStorage } from 'async_hooks';
+/**
+ * Prevents Runner.run_async from creating a second invoke_agent span
+ * when called internally by Runner.run (mirrors Python _ADK_WORKFLOW_ACTIVE).
+ */
+export declare const adkWorkflowActive: AsyncLocalStorage<boolean>;
+export declare function isAdkWorkflowActive(): boolean;
+export declare const OPERATION_MAP: Record<string, string>;
+export declare function getOperationType(endpoint: string): string;
+export declare function getSpanKind(operationType: string): SpanKind;
+export declare function generateSpanName(endpoint: string, instance: any): string;
+/**
+ * Drop-in replacement for ADK's tracer objects. Overrides
+ * `startActiveSpan` to yield the current span instead of creating a new one,
+ * letting Shield360 own top-level spans while ADK's code still runs.
+ */
+export declare class PassthroughTracer {
+    private _wrapped;
+    constructor(wrapped: any);
+    startActiveSpan(...args: any[]): any;
+    startSpan(...args: any[]): any;
+}
+export declare function resolveModelString(modelObj: any): string | null;
+export declare function extractModelName(instance: any): string;
+export declare function resolveServerInfo(instance?: any, modelName?: string | null): [string, number, string];
+export declare function setCommonSpanAttributes(span: Span, operationType: string): void;
+export declare function captureInputMessages(span: Span, llmRequest: any, captureContent: boolean): void;
+export declare function captureOutputMessages(span: Span, llmResponse: any, captureContent: boolean, finishReason?: string): void;
+export declare function captureEventOutput(span: Span, event: any, captureContent: boolean): void;
+export interface TokenUsage {
+    inputTokens?: number;
+    outputTokens?: number;
+    reasoningTokens?: number;
+    cachedTokens?: number;
+    totalTokens?: number;
+}
+export declare function extractTokenUsage(llmResponse: any): TokenUsage;
+export declare function enrichLlmSpan(span: Span, llmRequest: any, llmResponse: any, captureMessageContent: boolean): void;
+export declare function enrichToolSpan(span: Span, tool: any, functionArgs: any, functionResponseEvent: any, captureMessageContent: boolean, error?: any): void;
+export declare function enrichMergedToolSpan(span: Span, responseEventId: any, functionResponseEvent: any, captureMessageContent: boolean): void;
+export declare function setRunnerAgentAttributes(span: Span, instance: any, endpoint: string): void;
+export declare function setAgentAttributes(span: Span, instance: any): void;
+export declare function processGoogleAdkResponse(span: Span, endpoint: string, instance: any, startTime: number, _captureMessageContent: boolean): void;
+export declare function recordGoogleAdkMetrics(operationType: string, duration: number, requestModel: string, serverAddress: string, serverPort: number): void;