@google/adk 1.3.0 → 1.4.0

package/dist/cjs/context/token_based_context_compactor.js

@@ -27,13 +27,14 @@ __export(token_based_context_compactor_exports, {
   TokenBasedContextCompactor: () => TokenBasedContextCompactor
 });
 module.exports = __toCommonJS(token_based_context_compactor_exports);
+var import_content_processor_utils = require("../agents/processors/content_processor_utils.js");
 var import_compacted_event = require("../events/compacted_event.js");
-var import_event = require("../events/event.js");
 /**
  * @license
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+const CHARS_PER_TOKEN = 4;
 class TokenBasedContextCompactor {
   constructor(options) {
     this.tokenThreshold = options.tokenThreshold;
@@ -65,11 +66,14 @@ class TokenBasedContextCompactor {
     if (rawEvents.length <= this.eventRetentionSize) {
       return false;
     }
-    let totalTokens = 0;
-    for (const event of activeEvents) {
-      totalTokens += getEventTokens(event);
+    const promptTokenCount = latestPromptTokenCount(
+      activeEvents,
+      invocationContext
+    );
+    if (promptTokenCount === void 0) {
+      return false;
     }
-    return totalTokens > this.tokenThreshold;
+    return promptTokenCount > this.tokenThreshold;
   }
   async compact(invocationContext) {
     const events = invocationContext.session.events;
@@ -109,13 +113,41 @@ class TokenBasedContextCompactor {
     invocationContext.session.events.push(compactedEvent);
   }
 }
-function getEventTokens(event) {
+function latestPromptTokenCount(activeEvents, invocationContext) {
+  var _a;
+  for (let i = activeEvents.length - 1; i >= 0; i--) {
+    const count = (_a = activeEvents[i].usageMetadata) == null ? void 0 : _a.promptTokenCount;
+    if (count !== void 0) {
+      return count;
+    }
+  }
+  return estimatePromptTokenCount(activeEvents, invocationContext);
+}
+function estimatePromptTokenCount(activeEvents, invocationContext) {
   var _a;
-  if (((_a = event.usageMetadata) == null ? void 0 : _a.promptTokenCount) !== void 0) {
-    return event.usageMetadata.promptTokenCount;
+  const contents = (0, import_content_processor_utils.getContents)(
+    activeEvents,
+    invocationContext.agent.name,
+    invocationContext.branch
+  );
+  let totalChars = 0;
+  for (const content of contents) {
+    for (const part of (_a = content.parts) != null ? _a : []) {
+      if (part.text) {
+        totalChars += part.text.length;
+      }
+      if (part.functionCall) {
+        totalChars += JSON.stringify(part.functionCall).length;
+      }
+      if (part.functionResponse) {
+        totalChars += JSON.stringify(part.functionResponse).length;
+      }
+    }
+  }
+  if (totalChars <= 0) {
+    return void 0;
   }
-  const contentStr = (0, import_event.stringifyContent)(event);
-  return Math.ceil(contentStr.length / 4);
+  return Math.ceil(totalChars / CHARS_PER_TOKEN);
 }
 function hasFunctionCall(event) {
   var _a, _b;

package/dist/cjs/context/trajectory_thought_pruning_compactor.js

@@ -0,0 +1,71 @@
+/**
+  * @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 __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 trajectory_thought_pruning_compactor_exports = {};
+__export(trajectory_thought_pruning_compactor_exports, {
+  TrajectoryThoughtPruningCompactor: () => TrajectoryThoughtPruningCompactor
+});
+module.exports = __toCommonJS(trajectory_thought_pruning_compactor_exports);
+var import_event = require("../events/event.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class TrajectoryThoughtPruningCompactor {
+  constructor(options) {
+    if (options.eventRetentionSize < 0) {
+      throw new Error("eventRetentionSize must be a non-negative integer.");
+    }
+    this.eventRetentionSize = options.eventRetentionSize;
+  }
+  shouldCompact(invocationContext) {
+    const events = invocationContext.session.events;
+    if (events.length <= this.eventRetentionSize) {
+      return false;
+    }
+    const olderEvents = events.slice(
+      0,
+      events.length - this.eventRetentionSize
+    );
+    return olderEvents.some((event) => (0, import_event.hasThoughts)(event));
+  }
+  compact(invocationContext) {
+    const events = invocationContext.session.events;
+    if (events.length <= this.eventRetentionSize) {
+      return;
+    }
+    const pruneLimit = events.length - this.eventRetentionSize;
+    for (let i = 0; i < pruneLimit; i++) {
+      const event = events[i];
+      if ((0, import_event.hasThoughts)(event)) {
+        events[i] = (0, import_event.pruneThoughts)(event);
+      }
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  TrajectoryThoughtPruningCompactor
+});

package/dist/cjs/events/event.js

@@ -28,8 +28,10 @@ __export(event_exports, {
   createNewEventId: () => createNewEventId,
   getFunctionCalls: () => getFunctionCalls,
   getFunctionResponses: () => getFunctionResponses,
+  hasThoughts: () => hasThoughts,
   hasTrailingCodeExecutionResult: () => hasTrailingCodeExecutionResult,
   isFinalResponse: () => isFinalResponse,
+  pruneThoughts: () => pruneThoughts,
   stringifyContent: () => stringifyContent,
   transformToCamelCaseEvent: () => transformToCamelCaseEvent,
   transformToSnakeCaseEvent: () => transformToSnakeCaseEvent
@@ -100,6 +102,26 @@ function stringifyContent(event) {
     return (_a2 = part.text) != null ? _a2 : "";
   }).join("");
 }
+function hasThoughts(event) {
+  var _a, _b;
+  return !!((_b = (_a = event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.some((part) => part.thought === true));
+}
+function pruneThoughts(event) {
+  var _a;
+  if (!((_a = event.content) == null ? void 0 : _a.parts)) {
+    return event;
+  }
+  const prunedParts = event.content.parts.filter(
+    (part) => part.thought !== true
+  );
+  return {
+    ...event,
+    content: {
+      ...event.content,
+      parts: prunedParts
+    }
+  };
+}
 const ASCII_LETTERS_AND_NUMBERS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
 function createNewEventId() {
   let id = "";
@@ -140,8 +162,10 @@ function transformToSnakeCaseEvent(event) {
   createNewEventId,
   getFunctionCalls,
   getFunctionResponses,
+  hasThoughts,
   hasTrailingCodeExecutionResult,
   isFinalResponse,
+  pruneThoughts,
   stringifyContent,
   transformToCamelCaseEvent,
   transformToSnakeCaseEvent

package/dist/cjs/models/google_llm.js

@@ -25,7 +25,8 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var google_llm_exports = {};
 __export(google_llm_exports, {
   Gemini: () => Gemini,
-  geminiInitParams: () => geminiInitParams
+  geminiInitParams: () => geminiInitParams,
+  isGemini: () => isGemini
 });
 module.exports = __toCommonJS(google_llm_exports);
 var import_genai = require("@google/genai");
@@ -36,13 +37,19 @@ var import_async_queue = require("../utils/async_queue.js");
 var import_streaming_utils = require("../utils/streaming_utils.js");
 var import_base_llm = require("./base_llm.js");
 var import_gemini_llm_connection = require("./gemini_llm_connection.js");
+var import_interactions_utils = require("./interactions_utils.js");
 var import_llm_response = require("./llm_response.js");
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class Gemini extends import_base_llm.BaseLlm {
+var _a, _b;
+const GEMINI_MODEL_SYMBOL = Symbol.for("google.adk.geminiModel");
+function isGemini(obj) {
+  return typeof obj === "object" && obj !== null && GEMINI_MODEL_SYMBOL in obj && obj[GEMINI_MODEL_SYMBOL] === true;
+}
+class Gemini extends (_b = import_base_llm.BaseLlm, _a = GEMINI_MODEL_SYMBOL, _b) {
   /**
    * @param params The parameters for creating a Gemini instance.
    */
@@ -52,12 +59,14 @@ class Gemini extends import_base_llm.BaseLlm {
     vertexai,
     project,
     location,
-    headers
+    headers,
+    useInteractionsApi
   }) {
     if (!model) {
       model = "gemini-2.5-flash";
     }
     super({ model });
+    this[_a] = true;
     const params = geminiInitParams({
       model,
       vertexai,
@@ -75,6 +84,7 @@ class Gemini extends import_base_llm.BaseLlm {
     this.apiKey = params.apiKey;
     this.headers = headers;
     this.vertexai = !!params.vertexai;
+    this.useInteractionsApi = !!useInteractionsApi;
   }
   /**
    * Sends a request to the Gemini model.
@@ -84,11 +94,15 @@ class Gemini extends import_base_llm.BaseLlm {
    * @yields LlmResponse: The model response.
    */
   async *generateContentAsync(llmRequest, stream = false, abortSignal) {
-    var _a, _b, _c;
+    var _a2, _b2, _c;
+    if (this.useInteractionsApi) {
+      yield* (0, import_interactions_utils.generateContentViaInteractions)(this.apiClient, llmRequest, stream);
+      return;
+    }
     this.preprocessRequest(llmRequest);
     this.maybeAppendUserContent(llmRequest);
     import_logger.logger.info(
-      `Sending out request, model: ${(_a = llmRequest.model) != null ? _a : this.model}, backend: ${this.apiBackend}, stream: ${stream}`
+      `Sending out request, model: ${(_a2 = llmRequest.model) != null ? _a2 : this.model}, backend: ${this.apiBackend}, stream: ${stream}`
     );
     if (!llmRequest.config) {
       llmRequest.config = {};
@@ -104,7 +118,7 @@ class Gemini extends import_base_llm.BaseLlm {
     }
     if (stream) {
       const streamResult = await this.apiClient.models.generateContentStream({
-        model: (_b = llmRequest.model) != null ? _b : this.model,
+        model: (_b2 = llmRequest.model) != null ? _b2 : this.model,
         contents: llmRequest.contents,
         config: llmRequest.config
       });
@@ -192,8 +206,8 @@ class Gemini extends import_base_llm.BaseLlm {
    * @returns BaseLlmConnection, the connection to the Gemini model.
    */
   async connect(llmRequest) {
-    var _a, _b, _c, _d;
-    if ((_a = llmRequest.liveConnectConfig) == null ? void 0 : _a.httpOptions) {
+    var _a2, _b2, _c, _d;
+    if ((_a2 = llmRequest.liveConnectConfig) == null ? void 0 : _a2.httpOptions) {
       if (!llmRequest.liveConnectConfig.httpOptions.headers) {
         llmRequest.liveConnectConfig.httpOptions.headers = {};
       }
@@ -203,7 +217,7 @@ class Gemini extends import_base_llm.BaseLlm {
       );
       llmRequest.liveConnectConfig.httpOptions.apiVersion = this.liveApiVersion;
     }
-    if ((_b = llmRequest.config) == null ? void 0 : _b.systemInstruction) {
+    if ((_b2 = llmRequest.config) == null ? void 0 : _b2.systemInstruction) {
       llmRequest.liveConnectConfig.systemInstruction = {
         role: "system",
         // TODO - b/425992518: validate type casting works well.
@@ -305,5 +319,6 @@ function geminiInitParams({
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Gemini,
-  geminiInitParams
+  geminiInitParams,
+  isGemini
 });