@google/adk 2.4.0 → 2.5.0

package/dist/cjs/telemetry/export/agent_engine.js

@@ -0,0 +1,90 @@
+/**
+  * @license
+  * Copyright 2026 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var agent_engine_exports = {};
+__export(agent_engine_exports, {
+  runWithAgentEngineContext: () => runWithAgentEngineContext
+});
+module.exports = __toCommonJS(agent_engine_exports);
+var import_api = require("@opentelemetry/api");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const AGENT_ENGINE_TRACEPARENT_HEADER = "google-agent-engine-traceparent";
+const TRACEPARENT_HEADER = "traceparent";
+const MAX_SUPPORT_ID_LENGTH = 256;
+const SUPPORT_ID_KEY = (0, import_api.createContextKey)("google.adk.agent_engine.support_id");
+const TRACEPARENT_PATTERN = /^\s?([0-9a-f]{2})-([0-9a-f]{32})-([0-9a-f]{16})-([0-9a-f]{2})(-.*)?\s?$/;
+function parseTraceparent(value) {
+  const match = TRACEPARENT_PATTERN.exec(value);
+  if (!match) {
+    return void 0;
+  }
+  const [, version, traceId, spanId, flags, rest] = match;
+  if (version === "ff" || version === "00" && rest) {
+    return void 0;
+  }
+  const spanContext = {
+    traceId,
+    spanId,
+    traceFlags: parseInt(flags, 16),
+    isRemote: true
+  };
+  return import_api.trace.isSpanContextValid(spanContext) ? spanContext : void 0;
+}
+__name(parseTraceparent, "parseTraceparent");
+function getHeader(headers, name) {
+  for (const [key, value] of Object.entries(headers)) {
+    if (key.toLowerCase() === name) {
+      return Array.isArray(value) ? value[0] : value;
+    }
+  }
+  return void 0;
+}
+__name(getHeader, "getHeader");
+function runWithAgentEngineContext(headers, fn) {
+  var _a;
+  let ctx = import_api.context.active();
+  const agentEngineTraceparent = getHeader(
+    headers,
+    AGENT_ENGINE_TRACEPARENT_HEADER
+  );
+  const remoteParent = agentEngineTraceparent === void 0 ? void 0 : parseTraceparent(agentEngineTraceparent);
+  if (remoteParent && !import_api.trace.getSpan(ctx)) {
+    ctx = import_api.trace.setSpanContext(ctx, remoteParent);
+  }
+  const supportId = (_a = getHeader(headers, TRACEPARENT_HEADER)) == null ? void 0 : _a.trim();
+  if (supportId && supportId.length <= MAX_SUPPORT_ID_LENGTH) {
+    ctx = ctx.setValue(SUPPORT_ID_KEY, supportId);
+  }
+  return import_api.context.with(ctx, fn);
+}
+__name(runWithAgentEngineContext, "runWithAgentEngineContext");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  runWithAgentEngineContext
+});

package/dist/cjs/telemetry/export/agent_runtime.js

@@ -0,0 +1,73 @@
+/**
+  * @license
+  * Copyright 2026 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var agent_runtime_exports = {};
+__export(agent_runtime_exports, {
+  agentRuntimeRequestHooks: () => agentRuntimeRequestHooks
+});
+module.exports = __toCommonJS(agent_runtime_exports);
+var import_error_utils = require("../../utils/error_utils.js");
+var import_logger = require("../../utils/logger.js");
+var import_request_driven_metric_reader = require("./request_driven_metric_reader.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function agentRuntimeRequestHooks(otelHooks) {
+  var _a, _b, _c;
+  const readers = ((_a = otelHooks.metricReaders) != null ? _a : []).filter(
+    import_request_driven_metric_reader.isRequestDrivenMetricReader
+  );
+  const flushables = [
+    ...(_b = otelHooks.spanProcessors) != null ? _b : [],
+    ...(_c = otelHooks.logRecordProcessors) != null ? _c : []
+  ];
+  return {
+    onRequestStart() {
+      for (const reader of readers) {
+        reader.onRequestStart();
+      }
+    },
+    async onRequestEnd() {
+      const results = await Promise.allSettled([
+        ...readers.map(async (reader) => reader.onRequestEnd()),
+        ...flushables.map(async (processor) => processor.forceFlush())
+      ]);
+      for (const result of results) {
+        if (result.status === "rejected") {
+          import_logger.logger.warn(
+            `Telemetry flush on request end failed: ${(0, import_error_utils.formatError)(result.reason)}`
+          );
+        }
+      }
+    }
+  };
+}
+__name(agentRuntimeRequestHooks, "agentRuntimeRequestHooks");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  agentRuntimeRequestHooks
+});