@google/adk 1.2.0 → 1.4.0

package/dist/cjs/context/agent_controlled_context_compactor.js

@@ -0,0 +1,109 @@
+/**
+  * @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 agent_controlled_context_compactor_exports = {};
+__export(agent_controlled_context_compactor_exports, {
+  AgentControlledContextCompactor: () => AgentControlledContextCompactor
+});
+module.exports = __toCommonJS(agent_controlled_context_compactor_exports);
+var import_compacted_event = require("../events/compacted_event.js");
+var import_event = require("../events/event.js");
+var import_base_plugin = require("../plugins/base_plugin.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class AgentControlledContextCompactor {
+  constructor(options) {
+    this.trigger = import_base_plugin.ContextCompactionTrigger.AgentControlled;
+    this.summarizer = options.summarizer;
+  }
+  shouldCompact(invocationContext) {
+    return invocationContext.session.state["temp:consolidate_context"] === true;
+  }
+  async compact(invocationContext) {
+    const events = invocationContext.session.events;
+    const activeEvents = this.getActiveEvents(events);
+    const consolidateToolCallIndex = activeEvents.reduce(
+      (acc, e, idx) => {
+        var _a, _b;
+        return ((_b = (_a = e.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.some(
+          (p) => {
+            var _a2;
+            return ((_a2 = p.functionCall) == null ? void 0 : _a2.name) === "consolidate_context";
+          }
+        )) ? idx : acc;
+      },
+      -1
+    );
+    if (consolidateToolCallIndex <= 0) {
+      this.clearFlags(invocationContext);
+      return;
+    }
+    const eventsToCompact = activeEvents.slice(0, consolidateToolCallIndex);
+    const detail = invocationContext.session.state["temp:consolidate_context_detail"];
+    if (detail) {
+      eventsToCompact.push(
+        (0, import_event.createEvent)({
+          author: "system",
+          timestamp: eventsToCompact[eventsToCompact.length - 1].timestamp,
+          content: {
+            role: "user",
+            parts: [
+              {
+                text: `CRITICAL INSTRUCTION FOR SUMMARY: Please summarize the history. Focus especially on: ${detail}`
+              }
+            ]
+          }
+        })
+      );
+    }
+    try {
+      const compactedEvent = await this.summarizer.summarize(eventsToCompact);
+      invocationContext.session.events.push(compactedEvent);
+    } catch (error) {
+      console.error("Compaction failed:", error);
+    } finally {
+      this.clearFlags(invocationContext);
+    }
+  }
+  clearFlags(invocationContext) {
+    delete invocationContext.session.state["temp:consolidate_context"];
+    delete invocationContext.session.state["temp:consolidate_context_detail"];
+  }
+  getActiveEvents(events) {
+    const latest = events.filter(import_compacted_event.isCompactedEvent).pop();
+    return latest ? [
+      latest,
+      ...events.filter(
+        (e) => !(0, import_compacted_event.isCompactedEvent)(e) && e.timestamp > latest.endTime
+      )
+    ] : events;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AgentControlledContextCompactor
+});

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

@@ -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;
@@ -102,20 +106,48 @@ class TokenBasedContextCompactor {
       compactedEvent.actions = {
         stateDelta: {},
         artifactDelta: {},
-        requestedAuthConfigs: [],
+        requestedAuthConfigs: {},
         requestedToolConfirmations: {}
       };
     }
     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
@@ -55,7 +57,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);
@@ -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/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

@@ -25,23 +25,31 @@ 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");
 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");
+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.
    */
@@ -51,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,
@@ -74,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.
@@ -83,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 = {};
@@ -103,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
       });
@@ -168,10 +183,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;
   }
@@ -182,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 = {};
       }
@@ -193,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.
@@ -203,16 +227,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) {
@@ -287,5 +319,6 @@ function geminiInitParams({
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Gemini,
-  geminiInitParams
+  geminiInitParams,
+  isGemini
 });