@google/adk 0.5.0 → 0.6.0

package/dist/esm/context/token_based_context_compactor.js

@@ -0,0 +1,105 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isCompactedEvent } from "../events/compacted_event.js";
+import { stringifyContent } from "../events/event.js";
+class TokenBasedContextCompactor {
+  constructor(options) {
+    this.tokenThreshold = options.tokenThreshold;
+    this.eventRetentionSize = options.eventRetentionSize;
+    this.summarizer = options.summarizer;
+  }
+  getActiveEvents(events) {
+    let latestCompactedEvent = void 0;
+    for (let i = events.length - 1; i >= 0; i--) {
+      const e = events[i];
+      if (isCompactedEvent(e)) {
+        if (!latestCompactedEvent || e.endTime > latestCompactedEvent.endTime) {
+          latestCompactedEvent = e;
+        }
+      }
+    }
+    if (!latestCompactedEvent) {
+      return events;
+    }
+    const activeRawEvents = events.filter(
+      (e) => !isCompactedEvent(e) && e.timestamp > latestCompactedEvent.endTime
+    );
+    return [latestCompactedEvent, ...activeRawEvents];
+  }
+  shouldCompact(invocationContext) {
+    const events = invocationContext.session.events;
+    const activeEvents = this.getActiveEvents(events);
+    const rawEvents = activeEvents.filter((e) => !isCompactedEvent(e));
+    if (rawEvents.length <= this.eventRetentionSize) {
+      return false;
+    }
+    let totalTokens = 0;
+    for (const event of activeEvents) {
+      totalTokens += getEventTokens(event);
+    }
+    return totalTokens > this.tokenThreshold;
+  }
+  async compact(invocationContext) {
+    const events = invocationContext.session.events;
+    const activeEvents = this.getActiveEvents(events);
+    const rawEvents = activeEvents.filter((e) => !isCompactedEvent(e));
+    if (rawEvents.length <= this.eventRetentionSize) {
+      return;
+    }
+    let retainStartIndex = Math.max(
+      0,
+      rawEvents.length - this.eventRetentionSize
+    );
+    while (retainStartIndex > 0) {
+      const eventToRetain = rawEvents[retainStartIndex];
+      const previousEvent = rawEvents[retainStartIndex - 1];
+      if (hasFunctionResponse(eventToRetain) && hasFunctionCall(previousEvent)) {
+        retainStartIndex--;
+      } else {
+        break;
+      }
+    }
+    if (retainStartIndex === 0) {
+      return;
+    }
+    const rawEventsToCompact = rawEvents.slice(0, retainStartIndex);
+    const compactedEventPresent = activeEvents.find(isCompactedEvent);
+    const eventsToCompact = compactedEventPresent ? [compactedEventPresent, ...rawEventsToCompact] : rawEventsToCompact;
+    const compactedEvent = await this.summarizer.summarize(eventsToCompact);
+    if (!compactedEvent.actions) {
+      compactedEvent.actions = {
+        stateDelta: {},
+        artifactDelta: {},
+        requestedAuthConfigs: [],
+        requestedToolConfirmations: {}
+      };
+    }
+    invocationContext.session.events.push(compactedEvent);
+  }
+}
+function getEventTokens(event) {
+  var _a;
+  if (((_a = event.usageMetadata) == null ? void 0 : _a.promptTokenCount) !== void 0) {
+    return event.usageMetadata.promptTokenCount;
+  }
+  const contentStr = stringifyContent(event);
+  return Math.ceil(contentStr.length / 4);
+}
+function hasFunctionCall(event) {
+  var _a, _b;
+  return !!((_b = (_a = event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.some(
+    (part) => part.functionCall !== void 0
+  ));
+}
+function hasFunctionResponse(event) {
+  var _a, _b;
+  return !!((_b = (_a = event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.some(
+    (part) => part.functionResponse !== void 0
+  ));
+}
+export {
+  TokenBasedContextCompactor
+};

package/dist/esm/context/truncating_context_compactor.js

@@ -0,0 +1,28 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class TruncatingContextCompactor {
+  constructor(options) {
+    var _a;
+    this.threshold = options.threshold;
+    this.preserveLeadingEvents = (_a = options.preserveLeadingEvents) != null ? _a : 0;
+  }
+  shouldCompact(invocationContext) {
+    const eventsLength = invocationContext.session.events.length;
+    return eventsLength > this.threshold + Math.max(0, this.preserveLeadingEvents);
+  }
+  compact(invocationContext) {
+    const events = invocationContext.session.events;
+    const excess = events.length - this.threshold - Math.max(0, this.preserveLeadingEvents);
+    if (excess <= 0) {
+      return;
+    }
+    const startIndexToRemove = Math.max(0, this.preserveLeadingEvents);
+    events.splice(startIndexToRemove, excess);
+  }
+}
+export {
+  TruncatingContextCompactor
+};

package/dist/esm/events/compacted_event.js

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { createEvent } from "./event.js";
+function isCompactedEvent(event) {
+  return "isCompacted" in event && event.isCompacted === true;
+}
+function createCompactedEvent(params = {}) {
+  return {
+    ...createEvent(params),
+    isCompacted: params.isCompacted || true,
+    startTime: params.startTime,
+    endTime: params.endTime,
+    compactedContent: params.compactedContent
+  };
+}
+export {
+  createCompactedEvent,
+  isCompactedEvent
+};