@google/adk 1.2.0 → 1.3.0

package/dist/cjs/code_executors/code_executor_context.js

@@ -97,7 +97,7 @@ class CodeExecutorContext {
    * @return A list of input files in the code executor context.
    */
   getInputFiles() {
-    if (!(INPUT_FILE_KEY in this.sessionState)) {
+    if (!this.sessionState.has(INPUT_FILE_KEY)) {
       return [];
     }
     return this.sessionState.get(INPUT_FILE_KEY);
@@ -107,13 +107,13 @@ class CodeExecutorContext {
    * @param inputFiles The input files to add to the session state.
    */
   addInputFiles(inputFiles) {
-    if (!(INPUT_FILE_KEY in this.sessionState)) {
+    if (!this.sessionState.has(INPUT_FILE_KEY)) {
       this.sessionState.set(INPUT_FILE_KEY, []);
     }
     this.sessionState.get(INPUT_FILE_KEY).push(...inputFiles);
   }
   clearInputFiles() {
-    if (INPUT_FILE_KEY in this.sessionState) {
+    if (this.sessionState.has(INPUT_FILE_KEY)) {
       this.sessionState.set(INPUT_FILE_KEY, []);
     }
     if (PROCESSED_FILE_NAMES_KEY in this.context) {
@@ -126,7 +126,7 @@ class CodeExecutorContext {
    * @return The error count for the given invocation ID.
    */
   getErrorCount(invocationId) {
-    if (!(ERROR_COUNT_KEY in this.sessionState)) {
+    if (!this.sessionState.has(ERROR_COUNT_KEY)) {
       return 0;
     }
     return this.sessionState.get(ERROR_COUNT_KEY)[invocationId] || 0;
@@ -136,7 +136,7 @@ class CodeExecutorContext {
    * @param invocationId The invocation ID to increment the error count for.
    */
   incrementErrorCount(invocationId) {
-    if (!(ERROR_COUNT_KEY in this.sessionState)) {
+    if (!this.sessionState.has(ERROR_COUNT_KEY)) {
       this.sessionState.set(ERROR_COUNT_KEY, {});
     }
     this.sessionState.get(ERROR_COUNT_KEY)[invocationId] = this.getErrorCount(invocationId) + 1;
@@ -146,7 +146,7 @@ class CodeExecutorContext {
    * @param invocationId The invocation ID to reset the error count for.
    */
   resetErrorCount(invocationId) {
-    if (!(ERROR_COUNT_KEY in this.sessionState)) {
+    if (!this.sessionState.has(ERROR_COUNT_KEY)) {
       return;
     }
     const errorCounts = this.sessionState.get(ERROR_COUNT_KEY);
@@ -168,7 +168,7 @@ class CodeExecutorContext {
     resultStdout,
     resultStderr
   }) {
-    if (!(CODE_EXECUTION_RESULTS_KEY in this.sessionState)) {
+    if (!this.sessionState.has(CODE_EXECUTION_RESULTS_KEY)) {
       this.sessionState.set(CODE_EXECUTION_RESULTS_KEY, {});
     }
     const codeExecutionResults = this.sessionState.get(

package/dist/cjs/common.js

@@ -25,6 +25,7 @@ var __reExport = (target, mod, secondTarget) => (__copyProps(target, mod, "defau
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var common_exports = {};
 __export(common_exports, {
+  AUTH_PREPROCESSOR: () => import_auth_preprocessor.AUTH_PREPROCESSOR,
   ActiveStreamingTool: () => import_active_streaming_tool.ActiveStreamingTool,
   Agent: () => import_llm_agent.LlmAgent,
   AgentTool: () => import_agent_tool.AgentTool,
@@ -32,6 +33,7 @@ __export(common_exports, {
   ApigeeLlm: () => import_apigee_llm.ApigeeLlm,
   AuthCredentialTypes: () => import_auth_credential.AuthCredentialTypes,
   AuthHandler: () => import_auth_handler.AuthHandler,
+  AuthPreprocessor: () => import_auth_preprocessor.AuthPreprocessor,
   AuthProviderRegistry: () => import_auth_provider_registry.AuthProviderRegistry,
   BaseAgent: () => import_base_agent.BaseAgent,
   BaseCodeExecutor: () => import_base_code_executor.BaseCodeExecutor,
@@ -57,6 +59,7 @@ __export(common_exports, {
   ExitLoopTool: () => import_exit_loop_tool.ExitLoopTool,
   FileContentEncoding: () => import_code_execution_utils.FileContentEncoding,
   FunctionTool: () => import_function_tool.FunctionTool,
+  GCPSkillRegistry: () => import_gcp_skill_registry.GCPSkillRegistry,
   GOOGLE_MAPS_GROUNDING: () => import_google_maps_grounding_tool.GOOGLE_MAPS_GROUNDING,
   GOOGLE_SEARCH: () => import_google_search_tool.GOOGLE_SEARCH,
   Gemini: () => import_google_llm.Gemini,
@@ -88,6 +91,9 @@ __export(common_exports, {
   OAuth2CredentialExchanger: () => import_oauth2_credential_exchanger.OAuth2CredentialExchanger,
   OAuth2DiscoveryManager: () => import_oauth2_discovery.OAuth2DiscoveryManager,
   OAuthGrantType: () => import_auth_schemes.OAuthGrantType,
+  OpenAPIToolset: () => import_openapi_toolset.OpenAPIToolset,
+  OpenApiSpecParser: () => import_openapi_spec_parser.OpenApiSpecParser,
+  OperationParser: () => import_operation_parser.OperationParser,
   PRELOAD_MEMORY: () => import_preload_memory_tool.PRELOAD_MEMORY,
   ParallelAgent: () => import_parallel_agent.ParallelAgent,
   PluginManager: () => import_plugin_manager.PluginManager,
@@ -95,9 +101,11 @@ __export(common_exports, {
   PreloadMemoryTool: () => import_preload_memory_tool.PreloadMemoryTool,
   REQUEST_CONFIRMATION_FUNCTION_CALL_NAME: () => import_security_plugin.REQUEST_CONFIRMATION_FUNCTION_CALL_NAME,
   ReadonlyContext: () => import_readonly_context.ReadonlyContext,
+  RestApiTool: () => import_rest_api_tool.RestApiTool,
   RoutedAgent: () => import_routed_agent.RoutedAgent,
   RoutedLlm: () => import_routed_llm.RoutedLlm,
   Runner: () => import_runner.Runner,
+  SearchSkillsTool: () => import_search_skills_tool.SearchSkillsTool,
   SecurityPlugin: () => import_security_plugin.SecurityPlugin,
   SequentialAgent: () => import_sequential_agent.SequentialAgent,
   SessionStateCredentialService: () => import_session_state_credential_service.SessionStateCredentialService,
@@ -105,6 +113,7 @@ __export(common_exports, {
   State: () => import_state.State,
   StreamingMode: () => import_run_config.StreamingMode,
   TokenBasedContextCompactor: () => import_token_based_context_compactor.TokenBasedContextCompactor,
+  ToolAuthHandler: () => import_tool_auth_handler.ToolAuthHandler,
   ToolConfirmation: () => import_tool_confirmation.ToolConfirmation,
   TruncatingContextCompactor: () => import_truncating_context_compactor.TruncatingContextCompactor,
   URL_CONTEXT: () => import_url_context_tool.URL_CONTEXT,
@@ -114,6 +123,7 @@ __export(common_exports, {
   VertexRagRetrievalTool: () => import_vertex_rag_retrieval_tool.VertexRagRetrievalTool,
   createEvent: () => import_event.createEvent,
   createEventActions: () => import_event_actions.createEventActions,
+  createRestApiTool: () => import_rest_api_tool.createRestApiTool,
   createSession: () => import_session.createSession,
   functionsExportedForTestingOnly: () => import_functions.functionsExportedForTestingOnly,
   geminiInitParams: () => import_google_llm.geminiInitParams,
@@ -132,16 +142,21 @@ __export(common_exports, {
   isFinalResponse: () => import_event.isFinalResponse,
   isFunctionTool: () => import_function_tool.isFunctionTool,
   isGemini2OrAbove: () => import_model_name.isGemini2OrAbove,
+  isGemini3xFlashLive: () => import_model_name.isGemini3xFlashLive,
   isLlmAgent: () => import_llm_agent.isLlmAgent,
   isLoopAgent: () => import_loop_agent.isLoopAgent,
   isParallelAgent: () => import_parallel_agent.isParallelAgent,
   isRoutedAgent: () => import_routed_agent.isRoutedAgent,
   isRunner: () => import_runner.isRunner,
   isSequentialAgent: () => import_sequential_agent.isSequentialAgent,
+  loadAllSkillsInDir: () => import_loader.loadAllSkillsInDir,
+  loadSkillFromDir: () => import_loader.loadSkillFromDir,
+  loadSkillFromZipBuffer: () => import_loader.loadSkillFromZipBuffer,
   setLogLevel: () => import_logger.setLogLevel,
   setLogger: () => import_logger.setLogger,
   stringifyContent: () => import_event.stringifyContent,
   toStructuredEvents: () => import_structured_events.toStructuredEvents,
+  validateSkillDir: () => import_loader.validateSkillDir,
   version: () => import_version.version,
   zodObjectToSchema: () => import_simple_zod_to_json.zodObjectToSchema
 });
@@ -166,6 +181,7 @@ var import_sequential_agent = require("./agents/sequential_agent.js");
 var import_in_memory_artifact_service = require("./artifacts/in_memory_artifact_service.js");
 var import_auth_credential = require("./auth/auth_credential.js");
 var import_auth_handler = require("./auth/auth_handler.js");
+var import_auth_preprocessor = require("./auth/auth_preprocessor.js");
 var import_auth_provider_registry = require("./auth/auth_provider_registry.js");
 var import_auth_schemes = require("./auth/auth_schemes.js");
 var import_in_memory_credential_service = require("./auth/credential_service/in_memory_credential_service.js");
@@ -222,10 +238,18 @@ var import_model_name = require("./utils/model_name.js");
 var import_simple_zod_to_json = require("./utils/simple_zod_to_json.js");
 var import_variant_utils = require("./utils/variant_utils.js");
 var import_version = require("./version.js");
+var import_gcp_skill_registry = require("./skills/gcp_skill_registry.js");
+var import_loader = require("./skills/loader.js");
 var import_list_skills_tool = require("./tools/skill/list_skills_tool.js");
 var import_load_skill_resource_tool = require("./tools/skill/load_skill_resource_tool.js");
 var import_load_skill_tool = require("./tools/skill/load_skill_tool.js");
+var import_search_skills_tool = require("./tools/skill/search_skills_tool.js");
 var import_skill_toolset = require("./tools/skill/skill_toolset.js");
+var import_openapi_spec_parser = require("./tools/openapi_tool/openapi_spec_parser/openapi_spec_parser.js");
+var import_operation_parser = require("./tools/openapi_tool/openapi_spec_parser/operation_parser.js");
+var import_tool_auth_handler = require("./tools/openapi_tool/openapi_spec_parser/tool_auth_handler.js");
+var import_openapi_toolset = require("./tools/openapi_tool/openapi_toolset.js");
+var import_rest_api_tool = require("./tools/openapi_tool/rest_api_tool.js");
 __reExport(common_exports, require("./artifacts/base_artifact_service.js"), module.exports);
 __reExport(common_exports, require("./features/feature_registry.js"), module.exports);
 __reExport(common_exports, require("./memory/base_memory_service.js"), module.exports);
@@ -238,6 +262,7 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
  */
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AUTH_PREPROCESSOR,
   ActiveStreamingTool,
   Agent,
   AgentTool,
@@ -245,6 +270,7 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   ApigeeLlm,
   AuthCredentialTypes,
   AuthHandler,
+  AuthPreprocessor,
   AuthProviderRegistry,
   BaseAgent,
   BaseCodeExecutor,
@@ -270,6 +296,7 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   ExitLoopTool,
   FileContentEncoding,
   FunctionTool,
+  GCPSkillRegistry,
   GOOGLE_MAPS_GROUNDING,
   GOOGLE_SEARCH,
   Gemini,
@@ -301,6 +328,9 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   OAuth2CredentialExchanger,
   OAuth2DiscoveryManager,
   OAuthGrantType,
+  OpenAPIToolset,
+  OpenApiSpecParser,
+  OperationParser,
   PRELOAD_MEMORY,
   ParallelAgent,
   PluginManager,
@@ -308,9 +338,11 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   PreloadMemoryTool,
   REQUEST_CONFIRMATION_FUNCTION_CALL_NAME,
   ReadonlyContext,
+  RestApiTool,
   RoutedAgent,
   RoutedLlm,
   Runner,
+  SearchSkillsTool,
   SecurityPlugin,
   SequentialAgent,
   SessionStateCredentialService,
@@ -318,6 +350,7 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   State,
   StreamingMode,
   TokenBasedContextCompactor,
+  ToolAuthHandler,
   ToolConfirmation,
   TruncatingContextCompactor,
   URL_CONTEXT,
@@ -327,6 +360,7 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   VertexRagRetrievalTool,
   createEvent,
   createEventActions,
+  createRestApiTool,
   createSession,
   functionsExportedForTestingOnly,
   geminiInitParams,
@@ -345,16 +379,21 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   isFinalResponse,
   isFunctionTool,
   isGemini2OrAbove,
+  isGemini3xFlashLive,
   isLlmAgent,
   isLoopAgent,
   isParallelAgent,
   isRoutedAgent,
   isRunner,
   isSequentialAgent,
+  loadAllSkillsInDir,
+  loadSkillFromDir,
+  loadSkillFromZipBuffer,
   setLogLevel,
   setLogger,
   stringifyContent,
   toStructuredEvents,
+  validateSkillDir,
   version,
   zodObjectToSchema,
   ...require("./artifacts/base_artifact_service.js"),

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

@@ -36,10 +36,21 @@ var import_event = require("../../events/event.js");
  */
 const DEFAULT_PROMPT = "The following is a conversation history between a user and an AI agent. Please summarize the conversation, focusing on key information and decisions made, as well as any unresolved questions or tasks. The summary should be concise and capture the essence of the interaction.";
 class LlmSummarizer {
+  /**
+   * @param options - Configuration specifying the LLM and optional prompt.
+   */
   constructor(options) {
     this.llm = options.llm;
     this.prompt = options.prompt || DEFAULT_PROMPT;
   }
+  /**
+   * Summarizes a list of events into a single {@link CompactedEvent} using the
+   * configured LLM.
+   *
+   * @param events - The events to summarize. Must be non-empty.
+   * @returns A promise resolving to the compacted representation.
+   * @throws {Error} If `events` is empty or the LLM returns no content.
+   */
   async summarize(events) {
     var _a, _b, _c, _d, _e, _f;
     if (events.length === 0) {

package/dist/cjs/context/token_based_context_compactor.js

@@ -102,7 +102,7 @@ class TokenBasedContextCompactor {
       compactedEvent.actions = {
         stateDelta: {},
         artifactDelta: {},
-        requestedAuthConfigs: [],
+        requestedAuthConfigs: {},
         requestedToolConfirmations: {}
       };
     }

package/dist/cjs/events/event.js

@@ -55,7 +55,7 @@ function createEvent(params = {}) {
   };
 }
 function isFinalResponse(event) {
-  if (event.actions.skipSummarization || event.longRunningToolIds && event.longRunningToolIds.length > 0) {
+  if (event.actions.skipSummarization || event.longRunningToolIds && event.longRunningToolIds.length > 0 || event.actions.requestedAuthConfigs && Object.keys(event.actions.requestedAuthConfigs).length > 0) {
     return true;
   }
   return getFunctionCalls(event).length === 0 && getFunctionResponses(event).length === 0 && !event.partial && !hasTrailingCodeExecutionResult(event);

package/dist/cjs/integrations/agent_registry/agent_registry_mcp_toolset.js

@@ -37,6 +37,19 @@ var import_types2 = require("./types.js");
  * SPDX-License-Identifier: Apache-2.0
  */
 class AgentRegistrySingleMCPToolset extends import_base_toolset.BaseToolset {
+  /**
+   * @param options - Configuration for the MCP toolset.
+   * @param options.destinationResourceId - Telemetry identifier injected as
+   *   `gcp.mcp.server.destination.id` into each resolved tool's custom metadata.
+   * @param options.connectionParams - HTTP connection parameters for the MCP server.
+   * @param options.toolFilter - Optional predicate or list of tool names to include.
+   *   When omitted, all tools from the server are returned.
+   * @param options.prefix - Optional prefix prepended to each tool name (e.g. `myServer_toolName`).
+   * @param options.headerProvider - Optional async function called immediately before each
+   *   {@link getTools} invocation to supply or refresh request headers (e.g. GCP auth tokens).
+   * @param options.authScheme - Optional auth scheme forwarded to each resolved tool.
+   * @param options.authCredential - Optional credential forwarded to each resolved tool.
+   */
   constructor(options) {
     super(options.toolFilter || [], options.prefix);
     this.destinationResourceId = options.destinationResourceId;
@@ -46,8 +59,14 @@ class AgentRegistrySingleMCPToolset extends import_base_toolset.BaseToolset {
     this.authCredential = options.authCredential;
   }
   /**
-   * Connects to the underlying MCP server, retrieves tool definitions, prefixes tool names,
-   * and injects destination telemetry metadata.
+   * Connects to the underlying MCP server, retrieves tool definitions, prefixes
+   * tool names, and injects destination telemetry metadata into each tool.
+   *
+   * The `headerProvider`, if configured, is invoked immediately before the
+   * connection is established so that tokens are always fresh.
+   *
+   * @param context - Optional readonly agent context passed to the header provider.
+   * @returns The resolved and optionally filtered list of {@link MCPTool} instances.
    */
   async getTools(context) {
     var _a, _b, _c;

package/dist/cjs/models/gemini_llm_connection.js

@@ -27,15 +27,19 @@ __export(gemini_llm_connection_exports, {
   GeminiLlmConnection: () => GeminiLlmConnection
 });
 module.exports = __toCommonJS(gemini_llm_connection_exports);
+var import_live_connection_utils = require("../utils/live_connection_utils.js");
 var import_logger = require("../utils/logger.js");
+var import_model_name = require("../utils/model_name.js");
 /**
  * @license
- * Copyright 2025 Google LLC
+ * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 class GeminiLlmConnection {
-  constructor(geminiSession) {
+  constructor(geminiSession, modelVersion, messageQueue) {
     this.geminiSession = geminiSession;
+    this.modelVersion = modelVersion;
+    this.messageQueue = messageQueue;
   }
   /**
    * Sends the conversation history to the gemini model.
@@ -54,9 +58,10 @@ class GeminiLlmConnection {
       }
     );
     if (contents.length > 0) {
+      const isGemini3x = (0, import_model_name.isGemini3xFlashLive)(this.modelVersion);
       this.geminiSession.sendClientContent({
         turns: contents,
-        turnComplete: contents[contents.length - 1].role === "user"
+        turnComplete: isGemini3x ? true : contents[contents.length - 1].role === "user"
       });
     } else {
       import_logger.logger.info("no content is sent");
@@ -83,10 +88,16 @@ class GeminiLlmConnection {
       });
     } else {
       import_logger.logger.debug("Sending LLM new content", content);
-      this.geminiSession.sendClientContent({
-        turns: [content],
-        turnComplete: true
-      });
+      const isGemini3x = (0, import_model_name.isGemini3xFlashLive)(this.modelVersion);
+      if (isGemini3x && content.parts.length === 1 && content.parts[0].text) {
+        import_logger.logger.debug("Using sendRealtimeInput for Gemini 3.x text input");
+        this.geminiSession.sendRealtimeInput({ text: content.parts[0].text });
+      } else {
+        this.geminiSession.sendClientContent({
+          turns: [content],
+          turnComplete: true
+        });
+      }
     }
   }
   /**
@@ -95,8 +106,36 @@ class GeminiLlmConnection {
    * @param blob The blob to send to the model.
    */
   async sendRealtime(blob) {
+    var _a, _b, _c;
     import_logger.logger.debug("Sending LLM Blob:", blob);
-    this.geminiSession.sendRealtimeInput({ media: blob });
+    const isGemini3x = (0, import_model_name.isGemini3xFlashLive)(this.modelVersion);
+    const isNativeAudio = (_a = this.modelVersion) == null ? void 0 : _a.includes("native-audio");
+    if (isGemini3x || isNativeAudio) {
+      if ((_b = blob.mimeType) == null ? void 0 : _b.startsWith("audio/")) {
+        this.geminiSession.sendRealtimeInput({ audio: blob });
+      } else if ((_c = blob.mimeType) == null ? void 0 : _c.startsWith("image/")) {
+        this.geminiSession.sendRealtimeInput({ video: blob });
+      } else {
+        import_logger.logger.warn(
+          "Blob not sent. Unknown or empty mime type for sendRealtimeInput:",
+          blob.mimeType
+        );
+      }
+    } else {
+      this.geminiSession.sendRealtimeInput({ media: blob });
+    }
+  }
+  /**
+   * Sends an activity start signal to the model.
+   */
+  async sendActivityStart() {
+    this.geminiSession.sendRealtimeInput({ activityStart: {} });
+  }
+  /**
+   * Sends an activity end signal to the model.
+   */
+  async sendActivityEnd() {
+    this.geminiSession.sendRealtimeInput({ activityEnd: {} });
   }
   /**
    * Builds a full text response.
@@ -105,20 +144,24 @@ class GeminiLlmConnection {
    * partial.
    *
    * @param text The text to be included in the response.
+   * @param isThought Whether the text is a thought.
+   * @param groundingMetadata The grounding metadata to include.
    * @returns An LlmResponse containing the full text.
    */
-  buildFullTextResponse(text) {
-    return {
-      content: {
-        role: "model",
-        parts: [{ text }]
-      }
-    };
-  }
-  // TODO(b/425992518): GenAI SDK inconsistent API, missing methods.
-  // eslint-disable-next-line require-yield
   async *receive() {
-    throw new Error("Not Implemented.");
+    if (!this.messageQueue) {
+      throw new Error("Message queue is not initialized.");
+    }
+    const aggregator = new import_live_connection_utils.LiveResponseAggregator(this.modelVersion);
+    for await (const message of this.messageQueue) {
+      import_logger.logger.debug("Got LLM Live message:", message);
+      for (const response of aggregator.processMessage(message)) {
+        yield response;
+      }
+    }
+    for (const response of aggregator.close()) {
+      yield response;
+    }
   }
   /**
    * Closes the llm server connection.

package/dist/cjs/models/google_llm.js

@@ -32,6 +32,7 @@ var import_genai = require("@google/genai");
 var import_env_aware_utils = require("../utils/env_aware_utils.js");
 var import_logger = require("../utils/logger.js");
 var import_variant_utils = require("../utils/variant_utils.js");
+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");
@@ -168,10 +169,19 @@ class Gemini extends import_base_llm.BaseLlm {
   }
   get liveApiClient() {
     if (!this._liveApiClient) {
-      this._liveApiClient = new import_genai.GoogleGenAI({
-        apiKey: this.apiKey,
-        httpOptions: this.getLiveHttpOptions()
-      });
+      if (this.vertexai) {
+        this._liveApiClient = new import_genai.GoogleGenAI({
+          vertexai: this.vertexai,
+          project: this.project,
+          location: this.location || "global",
+          httpOptions: this.getLiveHttpOptions()
+        });
+      } else {
+        this._liveApiClient = new import_genai.GoogleGenAI({
+          apiKey: this.apiKey,
+          httpOptions: this.getLiveHttpOptions()
+        });
+      }
     }
     return this._liveApiClient;
   }
@@ -203,16 +213,24 @@ class Gemini extends import_base_llm.BaseLlm {
       };
     }
     llmRequest.liveConnectConfig.tools = (_c = llmRequest.config) == null ? void 0 : _c.tools;
+    const modelVersion = (_d = llmRequest.model) != null ? _d : this.model;
+    const messageQueue = new import_async_queue.AsyncQueue();
     const liveSession = await this.liveApiClient.live.connect({
-      model: (_d = llmRequest.model) != null ? _d : this.model,
+      model: modelVersion,
       config: llmRequest.liveConnectConfig,
       callbacks: {
-        // TODO - b/425992518: GenAI SDK inconsistent API, missing methods.
-        onmessage: () => {
+        onmessage: (message) => {
+          messageQueue.push(message);
+        },
+        onerror: (error) => {
+          messageQueue.error(error);
+        },
+        onclose: () => {
+          messageQueue.close();
         }
       }
     });
-    return new import_gemini_llm_connection.GeminiLlmConnection(liveSession);
+    return new import_gemini_llm_connection.GeminiLlmConnection(liveSession, modelVersion, messageQueue);
   }
   preprocessRequest(llmRequest) {
     if (this.apiBackend === import_variant_utils.GoogleLLMVariant.GEMINI_API) {

package/dist/cjs/plugins/security_plugin.js

@@ -47,6 +47,11 @@ var PolicyOutcome = /* @__PURE__ */ ((PolicyOutcome2) => {
   return PolicyOutcome2;
 })(PolicyOutcome || {});
 class InMemoryPolicyEngine {
+  /**
+   * Always returns {@link PolicyOutcome.ALLOW} for every tool call.
+   *
+   * @returns A promise resolving to an ALLOW result.
+   */
   async evaluate() {
     return Promise.resolve({
       outcome: "ALLOW" /* ALLOW */,
@@ -55,11 +60,22 @@ class InMemoryPolicyEngine {
   }
 }
 class SecurityPlugin extends import_base_plugin.BasePlugin {
+  /**
+   * @param params - Optional configuration. Defaults to {@link InMemoryPolicyEngine}
+   *   when no policy engine is provided.
+   */
   constructor(params) {
     var _a;
     super("security_plugin");
     this.policyEngine = (_a = params == null ? void 0 : params.policyEngine) != null ? _a : new InMemoryPolicyEngine();
   }
+  /**
+   * Intercepts tool calls, evaluating them against the policy engine before
+   * execution and handling confirmation flows for calls that require it.
+   *
+   * @returns A partial or error response object if the call is blocked or
+   *   awaiting confirmation, or `undefined` to allow execution to proceed.
+   */
   async beforeToolCallback({
     tool,
     toolArgs,

package/dist/cjs/sessions/base_session_service.js

@@ -26,7 +26,8 @@ var base_session_service_exports = {};
 __export(base_session_service_exports, {
   BaseSessionService: () => BaseSessionService,
   mergeStates: () => mergeStates,
-  trimTempDeltaState: () => trimTempDeltaState
+  trimTempDeltaState: () => trimTempDeltaState,
+  trimTempState: () => trimTempState
 });
 module.exports = __toCommonJS(base_session_service_exports);
 var import_lodash_es = require("lodash-es");
@@ -103,6 +104,15 @@ function trimTempDeltaState(event) {
   event.actions.stateDelta = filteredStateDelta;
   return event;
 }
+function trimTempState(state) {
+  const filteredState = {};
+  for (const [key, value] of Object.entries(state)) {
+    if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
+      filteredState[key] = value;
+    }
+  }
+  return filteredState;
+}
 function mergeStates(appState = {}, userState = {}, sessionState = {}) {
   const merged = (0, import_lodash_es.cloneDeep)(sessionState);
   for (const [k, v] of Object.entries(appState)) {
@@ -117,5 +127,6 @@ function mergeStates(appState = {}, userState = {}, sessionState = {}) {
 0 && (module.exports = {
   BaseSessionService,
   mergeStates,
-  trimTempDeltaState
+  trimTempDeltaState,
+  trimTempState
 });

package/dist/cjs/sessions/database_session_service.js

@@ -119,7 +119,7 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
           appStateDelta[key.replace(import_state.State.APP_PREFIX, "")] = value;
         } else if (key.startsWith(import_state.State.USER_PREFIX)) {
           userStateDelta[key.replace(import_state.State.USER_PREFIX, "")] = value;
-        } else {
+        } else if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
           sessionState[key] = value;
         }
       }
@@ -368,7 +368,7 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
             appDelta[key.replace(import_state.State.APP_PREFIX, "")] = value;
           } else if (key.startsWith(import_state.State.USER_PREFIX)) {
             userDelta[key.replace(import_state.State.USER_PREFIX, "")] = value;
-          } else {
+          } else if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
             sessionDelta[key] = value;
           }
         }