@google/adk 2.2.1 → 2.4.0

package/dist/esm/cli/utils/logs.js

@@ -0,0 +1,111 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import * as util from "node:util";
+import { LogLevel, setLogger, setLogLevel } from "../../utils/logger.js";
+const LOGGING_FORMAT = "%(asctime)s - %(levelname)s - %(filename)s:%(lineno)d - %(message)s";
+const LOG_LEVEL_NAMES = {
+  [LogLevel.DEBUG]: "DEBUG",
+  [LogLevel.INFO]: "INFO",
+  [LogLevel.WARN]: "WARNING",
+  [LogLevel.ERROR]: "ERROR"
+};
+function formatDateTimestamp(date) {
+  const pad = /* @__PURE__ */ __name((n) => String(n).padStart(2, "0"), "pad");
+  return `${date.getFullYear()}${pad(date.getMonth() + 1)}${pad(date.getDate())}_${pad(date.getHours())}${pad(date.getMinutes())}${pad(date.getSeconds())}`;
+}
+__name(formatDateTimestamp, "formatDateTimestamp");
+function formatLogLine(level, messages) {
+  var _a;
+  const iso = (/* @__PURE__ */ new Date()).toISOString();
+  const levelName = (_a = LOG_LEVEL_NAMES[level]) != null ? _a : "INFO";
+  const formattedMessage = messages.map((m) => typeof m === "string" ? m : util.inspect(m, { depth: null })).join(" ");
+  return `${iso} - ${levelName} - ${formattedMessage}`;
+}
+__name(formatLogLine, "formatLogLine");
+function createWriterLogger(initialLevel, writeLine) {
+  let currentLevel = initialLevel;
+  const logAt = /* @__PURE__ */ __name((targetLevel, messages) => {
+    if (currentLevel <= targetLevel) {
+      writeLine(formatLogLine(targetLevel, messages));
+    }
+  }, "logAt");
+  return {
+    setLogLevel(level) {
+      currentLevel = level;
+    },
+    log(level, ...messages) {
+      logAt(level, messages);
+    },
+    debug(...messages) {
+      logAt(LogLevel.DEBUG, messages);
+    },
+    info(...messages) {
+      logAt(LogLevel.INFO, messages);
+    },
+    warn(...messages) {
+      logAt(LogLevel.WARN, messages);
+    },
+    error(...messages) {
+      logAt(LogLevel.ERROR, messages);
+    }
+  };
+}
+__name(createWriterLogger, "createWriterLogger");
+function setupAdkLogger(level = LogLevel.INFO) {
+  setLogLevel(level);
+  setLogger(
+    createWriterLogger(level, (line) => {
+      process.stderr.write(`${line}
+`);
+    })
+  );
+}
+__name(setupAdkLogger, "setupAdkLogger");
+function logToTmpFolder(options = {}) {
+  const {
+    level = LogLevel.INFO,
+    subFolder = "agents_log",
+    logFilePrefix = "agent",
+    logFileTimestamp = formatDateTimestamp(/* @__PURE__ */ new Date())
+  } = options;
+  const logDir = path.join(os.tmpdir(), subFolder);
+  fs.mkdirSync(logDir, { recursive: true });
+  const logFilename = `${logFilePrefix}.${logFileTimestamp}.log`;
+  const logFilepath = path.join(logDir, logFilename);
+  fs.writeFileSync(logFilepath, "", "utf-8");
+  setLogLevel(level);
+  const fileLogger = createWriterLogger(level, (line) => {
+    fs.appendFileSync(logFilepath, `${line}
+`, "utf-8");
+  });
+  setLogger(fileLogger);
+  const latestLogLink = path.join(logDir, `${logFilePrefix}.latest.log`);
+  try {
+    try {
+      fs.lstatSync(latestLogLink);
+      fs.unlinkSync(latestLogLink);
+    } catch {
+    }
+    fs.symlinkSync(logFilepath, latestLogLink);
+  } catch (e) {
+    fileLogger.warn(`Failed to create symlink to latest log: ${String(e)}`);
+  }
+  fileLogger.info(`Log setup complete: ${logFilepath}`);
+  return logFilepath;
+}
+__name(logToTmpFolder, "logToTmpFolder");
+export {
+  LOGGING_FORMAT,
+  logToTmpFolder,
+  setupAdkLogger
+};

package/dist/esm/code_executors/agent_engine_sandbox_code_executor.js

@@ -27,6 +27,7 @@ import { BaseCodeExecutor } from "./base_code_executor.js";
 import {
   CodeExecutionLanguage
 } from "./code_execution_utils.js";
+import { CONTEXT_KEY } from "./code_executor_context.js";
 const DEFAULT_MAX_ATTEMPTS = 180;
 const DEFAULT_SANDBOX_TTL = "31536000s";
 const DEFAULT_SANDBOX_DISPLAY_NAME = "default_sandbox";
@@ -208,12 +209,14 @@ let AgentEngineSandboxCodeExecutor = class extends BaseCodeExecutor {
     return this.agentEngineCreationPromise;
   }
   async getOrCreateSandbox(invocationContext, agentEngineName, language) {
-    var _a, _b;
+    var _a, _b, _c, _d, _e;
     if (this.sandboxResourceName) {
       return this.sandboxResourceName;
     }
     const stateKey = `sandbox_name_${language.toLowerCase()}`;
-    let sandboxName = (_b = (_a = invocationContext.session) == null ? void 0 : _a.state) == null ? void 0 : _b[stateKey];
+    const state = (_a = invocationContext.session) == null ? void 0 : _a.state;
+    const codeExecutionContext = state == null ? void 0 : state[CONTEXT_KEY];
+    let sandboxName = (_b = state == null ? void 0 : state[stateKey]) != null ? _b : codeExecutionContext == null ? void 0 : codeExecutionContext[stateKey];
     let createNewSandbox = false;
     if (!sandboxName) {
       createNewSandbox = true;
@@ -270,6 +273,8 @@ let AgentEngineSandboxCodeExecutor = class extends BaseCodeExecutor {
           invocationContext.session.state = {};
         }
         invocationContext.session.state[stateKey] = sandboxName;
+        const context = (_e = (_c = invocationContext.session.state)[_d = CONTEXT_KEY]) != null ? _e : _c[_d] = {};
+        context[stateKey] = sandboxName;
       }
     }
     return sandboxName;

package/dist/esm/code_executors/code_execution_utils.js

@@ -29,33 +29,62 @@ function getEncodedFileContent(data) {
   return isBase64Encoded(data) ? data : base64Encode(data);
 }
 __name(getEncodedFileContent, "getEncodedFileContent");
+const FENCE_TAG_LANGUAGE_MAP = {
+  tool_code: "python" /* PYTHON */,
+  python: "python" /* PYTHON */,
+  py: "python" /* PYTHON */,
+  javascript: "javascript" /* JAVASCRIPT */,
+  js: "javascript" /* JAVASCRIPT */,
+  typescript: "typescript" /* TYPESCRIPT */,
+  ts: "typescript" /* TYPESCRIPT */,
+  bash: "shell" /* SHELL */,
+  sh: "shell" /* SHELL */,
+  shell: "shell" /* SHELL */
+};
+const NO_CODE_BLOCK = Object.freeze({
+  code: "",
+  language: "unspecified" /* UNSPECIFIED */
+});
+function getCodeBlockLanguage(leadingDelimiter) {
+  var _a;
+  const tag = leadingDelimiter.trim().replace(/^`+/, "").toLowerCase();
+  return (_a = FENCE_TAG_LANGUAGE_MAP[tag]) != null ? _a : "unspecified" /* UNSPECIFIED */;
+}
+__name(getCodeBlockLanguage, "getCodeBlockLanguage");
 function extractCodeAndTruncateContent(content, codeBlockDelimiters) {
+  return extractCodeBlockAndTruncateContent(content, codeBlockDelimiters).code;
+}
+__name(extractCodeAndTruncateContent, "extractCodeAndTruncateContent");
+function extractCodeBlockAndTruncateContent(content, codeBlockDelimiters) {
   var _a;
   if (!((_a = content.parts) == null ? void 0 : _a.length)) {
-    return "";
+    return { ...NO_CODE_BLOCK };
   }
   for (let i = 0; i < content.parts.length; i++) {
     const part = content.parts[i];
     if (part.executableCode && (i === content.parts.length - 1 || !content.parts[i + 1].codeExecutionResult)) {
       content.parts = content.parts.slice(0, i + 1);
-      return part.executableCode.code || "";
+      return {
+        code: part.executableCode.code || "",
+        language: "python" /* PYTHON */
+      };
     }
   }
   const textParts = content.parts.filter((part) => part.text);
   if (!textParts.length) {
-    return "";
+    return { ...NO_CODE_BLOCK };
   }
   const firstTextPart = cloneDeep(textParts[0]);
   const responseText = textParts.map((part) => part.text).join("\n");
   const leadingDelimiterPattern = codeBlockDelimiters.map((d) => d[0]).join("|");
   const trailingDelimiterPattern = codeBlockDelimiters.map((d) => d[1]).join("|");
   const match = new RegExp(
-    `(?<prefix>.*?)(${leadingDelimiterPattern})(?<codeStr>.*?)(${trailingDelimiterPattern})(?<suffix>.*?)$`,
+    `(?<prefix>.*?)(?<leading>${leadingDelimiterPattern})(?<codeStr>.*?)(${trailingDelimiterPattern})(?<suffix>.*?)$`,
     "s"
   ).exec(responseText);
-  const { prefix, codeStr } = (match == null ? void 0 : match.groups) || {};
+  const { prefix, leading, codeStr } = (match == null ? void 0 : match.groups) || {};
   if (!codeStr) {
-    return "";
+    return { ...NO_CODE_BLOCK };
   }
   content.parts = [];
   if (prefix) {
@@ -63,12 +92,11 @@ function extractCodeAndTruncateContent(content, codeBlockDelimiters) {
     content.parts.push(firstTextPart);
   }
   content.parts.push(buildExecutableCodePart(codeStr));
-  return codeStr;
+  return { code: codeStr, language: getCodeBlockLanguage(leading != null ? leading : "") };
 }
-__name(extractCodeAndTruncateContent, "extractCodeAndTruncateContent");
+__name(extractCodeBlockAndTruncateContent, "extractCodeBlockAndTruncateContent");
 function buildExecutableCodePart(code) {
   return {
-    text: code,
     executableCode: {
       code,
       language: Language.PYTHON
@@ -79,28 +107,28 @@ __name(buildExecutableCodePart, "buildExecutableCodePart");
 function buildCodeExecutionResultPart(codeExecutionResult) {
   if (codeExecutionResult.stderr) {
     return {
-      text: codeExecutionResult.stderr,
       codeExecutionResult: {
-        outcome: Outcome.OUTCOME_FAILED
+        outcome: Outcome.OUTCOME_FAILED,
+        output: codeExecutionResult.stderr
       }
     };
   }
   const finalResult = [];
-  if (codeExecutionResult.stdout || !codeExecutionResult.outputFiles) {
+  if (codeExecutionResult.stdout || !codeExecutionResult.outputFiles.length) {
     finalResult.push(`Code execution result:
 ${codeExecutionResult.stdout}
 `);
   }
-  if (codeExecutionResult.outputFiles) {
+  if (codeExecutionResult.outputFiles.length) {
     finalResult.push(
       `Saved artifacts:
-` + codeExecutionResult.outputFiles.map((f) => f.name).join(", ")
+` + codeExecutionResult.outputFiles.map((f) => `\`${f.name}\``).join(",")
     );
   }
   return {
-    text: finalResult.join("\n\n"),
     codeExecutionResult: {
-      outcome: Outcome.OUTCOME_OK
+      outcome: Outcome.OUTCOME_OK,
+      output: finalResult.join("\n\n")
     }
   };
 }
@@ -130,5 +158,7 @@ export {
   buildExecutableCodePart,
   convertCodeExecutionParts,
   extractCodeAndTruncateContent,
+  extractCodeBlockAndTruncateContent,
+  getCodeBlockLanguage,
   getEncodedFileContent
 };

package/dist/esm/code_executors/code_executor_context.js

@@ -17,8 +17,10 @@ const CODE_EXECUTION_RESULTS_KEY = "_code_execution_results";
 const _CodeExecutorContext = class _CodeExecutorContext {
   constructor(sessionState) {
     this.sessionState = sessionState;
-    var _a;
-    this.context = (_a = sessionState.get(CONTEXT_KEY)) != null ? _a : {};
+    if (!sessionState.has(CONTEXT_KEY)) {
+      sessionState.set(CONTEXT_KEY, {});
+    }
+    this.context = sessionState.get(CONTEXT_KEY);
     this.sessionState = sessionState;
   }
   /**
@@ -170,5 +172,6 @@ const _CodeExecutorContext = class _CodeExecutorContext {
 __name(_CodeExecutorContext, "CodeExecutorContext");
 let CodeExecutorContext = _CodeExecutorContext;
 export {
+  CONTEXT_KEY,
   CodeExecutorContext
 };

package/dist/esm/code_executors/container_code_executor.js

@@ -40,6 +40,8 @@ let ContainerCodeExecutor = class extends BaseCodeExecutor {
   constructor(options = {}) {
     var _a, _b, _c;
     super();
+    this.containers = /* @__PURE__ */ new Map();
+    this.initPromises = /* @__PURE__ */ new Map();
     if (!options.image && !options.dockerPath) {
       throw new Error(
         "Either image or dockerPath must be set for ContainerCodeExecutor."
@@ -69,8 +71,9 @@ let ContainerCodeExecutor = class extends BaseCodeExecutor {
         `Unsupported language for ContainerCodeExecutor: ${language}. Supported: ${Object.keys(LANGUAGE_RUNTIME_COMMAND_MAP).join(", ")}.`
       );
     }
-    await this.ensureContainer();
-    const { stdout, stderr, exitCode } = await this.container.execute(
+    const key = sessionKey(params.invocationContext);
+    const container = await this.ensureContainer(key);
+    const { stdout, stderr, exitCode } = await container.execute(
       [...TIMEOUT_COMMAND, String(this.timeoutSeconds), ...command, code],
       this.timeoutSeconds
     );
@@ -102,32 +105,35 @@ ${message}` : message;
     return stderr;
   }
   /**
-   * Stops and removes the container. Safe to call when no container has been
-   * started; provided for deterministic teardown in tests and app shutdown.
+   * Stops and removes every session's container. Safe to call when none has
+   * been started; provided for deterministic teardown in tests and app
+   * shutdown.
    */
   async close() {
-    const container = this.container;
-    this.container = void 0;
-    this.initPromise = void 0;
-    await (container == null ? void 0 : container.stop());
+    const containers = [...this.containers.values()];
+    this.containers.clear();
+    this.initPromises.clear();
+    await Promise.all(containers.map((container) => container.stop()));
   }
-  /** Lazily builds/starts the container exactly once. */
-  ensureContainer() {
-    if (!this.initPromise) {
-      this.initPromise = this.initContainer().catch((error) => {
-        this.initPromise = void 0;
+  /** Lazily builds/starts the given session's container exactly once. */
+  ensureContainer(key) {
+    let initPromise = this.initPromises.get(key);
+    if (!initPromise) {
+      initPromise = this.initContainer(key).catch((error) => {
+        this.initPromises.delete(key);
         throw error;
       });
+      this.initPromises.set(key, initPromise);
     }
-    return this.initPromise;
+    return initPromise.then(() => this.containers.get(key));
   }
-  async initContainer() {
+  async initContainer(key) {
     const container = new DockerContainer(this.containerOptions);
     if (this.dockerPath) {
       await container.build(this.dockerPath);
     }
     await container.start();
-    this.container = container;
+    this.containers.set(key, container);
     const { exitCode } = await container.execute(
       ["which", "python3"],
       this.timeoutSeconds
@@ -141,6 +147,11 @@ __name(ContainerCodeExecutor, "ContainerCodeExecutor");
 ContainerCodeExecutor = __decorateClass([
   experimental
 ], ContainerCodeExecutor);
+function sessionKey(invocationContext) {
+  const session = invocationContext.session;
+  return JSON.stringify([session.appName, session.userId, session.id]);
+}
+__name(sessionKey, "sessionKey");
 function defineFrozenFalse(target, key) {
   Object.defineProperty(target, key, {
     value: false,

package/dist/esm/code_executors/unsafe_local_code_executor.js

@@ -216,7 +216,12 @@ Code execution timed out after ${this.timeoutSeconds} seconds.`;
       };
     } finally {
       if (tempDir) {
-        await fs.rm(tempDir, { recursive: true, force: true });
+        await fs.rm(tempDir, {
+          recursive: true,
+          force: true,
+          maxRetries: 10,
+          retryDelay: 500
+        });
       }
     }
   }

package/dist/esm/common.js

@@ -15,9 +15,20 @@ import {
   findEventByFunctionCallId,
   findMatchingFunctionCall,
   functionsExportedForTestingOnly,
+  handleFunctionCallsLive,
   isToolNotFound
 } from "./agents/functions.js";
-import { InvocationContext, requireAgent } from "./agents/invocation_context.js";
+import {
+  InvocationContext,
+  LlmCallsLimitExceededError,
+  isLlmCallsLimitExceededError,
+  requireAgent
+} from "./agents/invocation_context.js";
+import {
+  LangGraphAgent,
+  getLastHumanMessages,
+  isLangGraphAgent
+} from "./agents/langgraph_agent.js";
 import { LiveRequestQueue } from "./agents/live_request_queue.js";
 import { LlmAgent, LlmAgent as LlmAgent2, isLlmAgent } from "./agents/llm_agent.js";
 import { LoopAgent, isLoopAgent } from "./agents/loop_agent.js";
@@ -27,6 +38,7 @@ import {
   BaseLlmRequestProcessor,
   BaseLlmResponseProcessor
 } from "./agents/processors/base_llm_processor.js";
+import { CodeExecutionResponseProcessor } from "./agents/processors/code_execution_request_processor.js";
 import {
   CONTENT_REQUEST_PROCESSOR,
   ContentRequestProcessor
@@ -38,7 +50,7 @@ import {
 } from "./agents/processors/interactions_request_processor.js";
 import { ReadonlyContext } from "./agents/readonly_context.js";
 import { RoutedAgent, isRoutedAgent } from "./agents/routed_agent.js";
-import { StreamingMode } from "./agents/run_config.js";
+import { StreamingMode, createRunConfig } from "./agents/run_config.js";
 import { SequentialAgent, isSequentialAgent } from "./agents/sequential_agent.js";
 import {
   getPendingUserInputRequests,
@@ -53,7 +65,11 @@ import { AuthCredentialTypes, isAuthCredential } from "./auth/auth_credential.js
 import { AuthHandler } from "./auth/auth_handler.js";
 import { AUTH_PREPROCESSOR, AuthPreprocessor } from "./auth/auth_preprocessor.js";
 import { AuthProviderRegistry } from "./auth/auth_provider_registry.js";
-import { OAuthGrantType } from "./auth/auth_schemes.js";
+import {
+  AuthSchemeType,
+  OAuthGrantType,
+  getOAuthGrantTypeFromFlow
+} from "./auth/auth_schemes.js";
 import { InMemoryCredentialService } from "./auth/credential_service/in_memory_credential_service.js";
 import { SessionStateCredentialService } from "./auth/credential_service/session_state_credential_service.js";
 import { CredentialExchangeError } from "./auth/exchanger/base_credential_exchanger.js";
@@ -118,6 +134,8 @@ import {
   stripAdkIdentityPreamble
 } from "./models/chrome_prompt_llm.js";
 import { Gemini, geminiInitParams } from "./models/google_llm.js";
+import { appendInstructions } from "./models/llm_request.js";
+import { createLlmResponse } from "./models/llm_response.js";
 import { LLMRegistry } from "./models/registry.js";
 import { RoutedLlm } from "./models/routed_llm.js";
 import { BasePlanner, isBasePlanner } from "./planners/base_planner.js";
@@ -219,6 +237,8 @@ import {
 import { AsyncQueue } from "./utils/async_queue.js";
 import { renameReservedKeywords, toSnakeCaseName } from "./utils/case_utils.js";
 import { getClientLabels, runWithClientLabel } from "./utils/client_labels.js";
+import { toGeminiSchema } from "./utils/gemini_schema_util.js";
+import { genaiSchemaToJsonSchema } from "./utils/genai_schema_to_json.js";
 import {
   LogLevel,
   getLogger,
@@ -233,7 +253,12 @@ import {
   isGemini3xFlashLive,
   isGemini3xLive
 } from "./utils/model_name.js";
-import { zodObjectToSchema } from "./utils/simple_zod_to_json.js";
+import { toJsonSchema } from "./utils/schema.js";
+import {
+  isZodObject,
+  isZodSchema,
+  zodObjectToSchema
+} from "./utils/simple_zod_to_json.js";
 import { Task } from "./utils/task.js";
 import { GoogleLLMVariant } from "./utils/variant_utils.js";
 import { version } from "./version.js";
@@ -288,6 +313,17 @@ export * from "./features/feature_registry.js";
 export * from "./memory/base_memory_service.js";
 export * from "./sessions/base_session_service.js";
 export * from "./tools/base_tool.js";
+import {
+  convertSessionToEvalFormat,
+  createEmptyState
+} from "./cli/utils/evals.js";
+import {
+  buildFunctionDeclarationForLangchain,
+  buildFunctionDeclarationForParamsForCrewai,
+  buildFunctionDeclarationUtil
+} from "./tools/function_tool.js";
+import { isBaseRetrievalTool } from "./tools/retrieval/base_retrieval_tool.js";
+import { ToolboxToolset, isToolboxToolset } from "./tools/toolbox_toolset.js";
 export {
   ADK_HANDLE_MODEL_ERROR_TOOL_NAME,
   AUTH_PREPROCESSOR,
@@ -305,6 +341,7 @@ export {
   AuthHandler,
   AuthPreprocessor,
   AuthProviderRegistry,
+  AuthSchemeType,
   BaseAgent,
   BaseCodeExecutor,
   BaseEnvironment,
@@ -328,6 +365,7 @@ export {
   ChromeBuiltInLlm,
   ChromeModelUnavailableError,
   CodeExecutionLanguage,
+  CodeExecutionResponseProcessor,
   ConsolidateContextTool,
   ContentRequestProcessor,
   Context,
@@ -375,10 +413,12 @@ export {
   LLMRegistry,
   LOAD_ARTIFACTS,
   LOAD_MEMORY,
+  LangGraphAgent,
   ListSkillsTool,
   LiveRequestQueue,
   LlamaIndexRetrieval,
   LlmAgent2 as LlmAgent,
+  LlmCallsLimitExceededError,
   LlmSummarizer,
   LoadArtifactsTool,
   LoadMemoryTool,
@@ -437,6 +477,7 @@ export {
   ToolErrorType,
   ToolExecutionError,
   ToolNode,
+  ToolboxToolset,
   TrackingScope,
   TrajectoryThoughtPruningCompactor,
   TruncatingContextCompactor,
@@ -449,13 +490,21 @@ export {
   WebMCPToolset,
   Workflow,
   WorkflowNode,
+  appendInstructions,
   asRunnableRoot,
+  buildFunctionDeclarationForLangchain,
+  buildFunctionDeclarationForParamsForCrewai,
+  buildFunctionDeclarationUtil,
   commonPrefixOf,
+  convertSessionToEvalFormat,
+  createEmptyState,
   createEvent,
   createEventActions,
+  createLlmResponse,
   createNodeErrorEvent,
   createNodeState,
   createResumabilityConfig,
+  createRunConfig,
   createSession,
   createSubBranch,
   determineAgentForResumption,
@@ -464,16 +513,20 @@ export {
   findMatchingFunctionCall,
   functionsExportedForTestingOnly,
   geminiInitParams,
+  genaiSchemaToJsonSchema,
   generateClientFunctionCallId,
   getAskUserConfirmationFunctionCalls,
   getClientLabels,
   getFunctionCalls,
   getFunctionResponses,
+  getLastHumanMessages,
   getLogger,
   getModelContext,
+  getOAuthGrantTypeFromFlow,
   getPendingUserInputRequests,
   getUserChoiceTool,
   getUserInputRequests,
+  handleFunctionCallsLive,
   hasThoughts,
   hasTrailingCodeExecutionResult,
   isAgentTool,
@@ -483,6 +536,7 @@ export {
   isBaseExampleProvider,
   isBaseLlm,
   isBasePlanner,
+  isBaseRetrievalTool,
   isBaseTool,
   isBaseToolset,
   isBuiltInPlanner,
@@ -494,7 +548,9 @@ export {
   isGemini3xFlashLive,
   isGemini3xLive,
   isIntentMismatchError,
+  isLangGraphAgent,
   isLlmAgent,
+  isLlmCallsLimitExceededError,
   isLoopAgent,
   isNodeErrorEvent,
   isNodeReportedError,
@@ -512,8 +568,11 @@ export {
   isSessionArtifactService,
   isStateSchemaError,
   isToolNotFound,
+  isToolboxToolset,
   isWebMCPSupported,
   isWorkflow,
+  isZodObject,
+  isZodSchema,
   logger,
   markRestored,
   node,
@@ -535,6 +594,8 @@ export {
   setLogger,
   stringifyContent,
   stripAdkIdentityPreamble,
+  toGeminiSchema,
+  toJsonSchema,
   toSnakeCaseName,
   toStructuredEvents,
   validateAppName,

package/dist/esm/context/summarizers/llm_summarizer.js

@@ -29,7 +29,7 @@ const _LlmSummarizer = class _LlmSummarizer {
    * @throws {Error} If `events` is empty or the LLM returns no content.
    */
   async summarize(events) {
-    var _a, _b, _c, _d, _e, _f;
+    var _a, _b;
     if (events.length === 0) {
       throw new Error("Cannot summarize an empty list of events.");
     }
@@ -51,18 +51,20 @@ ${formattedEvents}`;
       toolsDict: {},
       liveConnectConfig: {}
     };
-    const responseGen = this.llm.generateContentAsync(request, false);
     let compactedContent = "";
-    const firstResponse = await responseGen.next();
-    if (firstResponse.done || !((_c = (_b = (_a = firstResponse.value.content) == null ? void 0 : _a.parts) == null ? void 0 : _b[0]) == null ? void 0 : _c.text)) {
-      throw new Error("LLM failed to return a valid summary.");
-    }
-    compactedContent += firstResponse.value.content.parts[0].text;
-    for await (const chunk of responseGen) {
-      if ((_f = (_e = (_d = chunk.content) == null ? void 0 : _d.parts) == null ? void 0 : _e[0]) == null ? void 0 : _f.text) {
-        compactedContent += chunk.content.parts[0].text;
+    for await (const response of this.llm.generateContentAsync(
+      request,
+      false
+    )) {
+      for (const part of (_b = (_a = response.content) == null ? void 0 : _a.parts) != null ? _b : []) {
+        if (part.text && !part.thought) {
+          compactedContent += part.text;
+        }
       }
     }
+    if (!compactedContent) {
+      throw new Error("LLM failed to return a valid summary.");
+    }
     return createCompactedEvent({
       author: "system",
       content: {

package/dist/esm/context/token_based_context_compactor.js

@@ -86,7 +86,8 @@ let TokenBasedContextCompactor = _TokenBasedContextCompactor;
 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;
+    const usage = activeEvents[i].usageMetadata;
+    const count = (_a = usage == null ? void 0 : usage.initialPromptTokenCount) != null ? _a : usage == null ? void 0 : usage.promptTokenCount;
     if (count !== void 0) {
       return count;
     }

package/dist/esm/errors/input_validation_error.js

@@ -10,9 +10,10 @@ var __name = (target, value) => __defProp(target, "name", { value, configurable:
 const _InputValidationError = class _InputValidationError extends Error {
   /**
    * @param message A message describing why the input is invalid.
+   * @param options Standard error options, e.g. `{cause}`.
    */
-  constructor(message = "Invalid input.") {
-    super(message);
+  constructor(message = "Invalid input.", options) {
+    super(message, options);
     this.name = "InputValidationError";
   }
 };