@google/adk 1.2.0 → 1.4.0

package/dist/cjs/utils/case_utils.js

@@ -0,0 +1,53 @@
+/**
+  * @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 case_utils_exports = {};
+__export(case_utils_exports, {
+  camelCaseKeys: () => camelCaseKeys
+});
+module.exports = __toCommonJS(case_utils_exports);
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function camelCaseKeys(val) {
+  if (Array.isArray(val)) {
+    return val.map(camelCaseKeys);
+  }
+  if (val !== null && typeof val === "object" && val.constructor === Object) {
+    const obj = val;
+    const newObj = {};
+    for (const key of Object.keys(obj)) {
+      const camelKey = key.replace(/_([a-z])/g, (_, g) => g.toUpperCase());
+      newObj[camelKey] = camelCaseKeys(obj[key]);
+    }
+    return newObj;
+  }
+  return val;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  camelCaseKeys
+});

package/dist/cjs/utils/client_labels.js

@@ -24,9 +24,12 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var client_labels_exports = {};
 __export(client_labels_exports, {
-  getClientLabels: () => getClientLabels
+  getClientLabels: () => getClientLabels,
+  parseUserAgent: () => parseUserAgent,
+  runWithClientLabel: () => runWithClientLabel
 });
 module.exports = __toCommonJS(client_labels_exports);
+var import_node_async_hooks = require("node:async_hooks");
 var import_version = require("../version.js");
 var import_env_aware_utils = require("./env_aware_utils.js");
 /**
@@ -38,19 +41,54 @@ const ADK_LABEL = "google-adk";
 const LANGUAGE_LABEL = "gl-typescript";
 const AGENT_ENGINE_TELEMETRY_TAG = "remote_reasoning_engine";
 const AGENT_ENGINE_TELEMETRY_ENV_VARIABLE_NAME = "GOOGLE_CLOUD_AGENT_ENGINE_ID";
+const clientLabelLocalStorage = new import_node_async_hooks.AsyncLocalStorage();
+const USER_AGENT_PATTERNS = [
+  ["Edge", /(?:Edg|Edge|EdgA)\/([0-9.]+)/i],
+  ["Firefox", /(?:Firefox|FxiOS)\/([0-9.]+)/i],
+  ["Chrome", /(?:Chrome|CriOS)\/([0-9.]+)/i],
+  ["Safari", /Version\/([0-9.]+).*Safari/i]
+];
+function parseUserAgent(userAgent) {
+  if (!userAgent) {
+    return "Browser";
+  }
+  for (const [name, regex] of USER_AGENT_PATTERNS) {
+    const match = userAgent.match(regex);
+    if (match) {
+      return `${name}/${match[1]}`;
+    }
+  }
+  return "Browser";
+}
 function _getDefaultLabels() {
   let frameworkLabel = `${ADK_LABEL}/${import_version.version}`;
   if (!(0, import_env_aware_utils.isBrowser)() && process.env[AGENT_ENGINE_TELEMETRY_ENV_VARIABLE_NAME]) {
     frameworkLabel = `${frameworkLabel}+${AGENT_ENGINE_TELEMETRY_TAG}`;
   }
-  const languageLabel = `${LANGUAGE_LABEL}/${(0, import_env_aware_utils.isBrowser)() ? window.navigator.userAgent : process.version}`;
+  const languageLabelDetail = (0, import_env_aware_utils.isBrowser)() ? (
+    // eslint-disable-next-line no-undef
+    parseUserAgent(window.navigator.userAgent)
+  ) : process.version;
+  const languageLabel = `${LANGUAGE_LABEL}/${languageLabelDetail}`;
   return [frameworkLabel, languageLabel];
 }
+function runWithClientLabel(clientLabel, callback) {
+  if (typeof clientLabel !== "string" || clientLabel.trim() === "") {
+    throw new Error("Client label must be a non-empty string.");
+  }
+  return clientLabelLocalStorage.run(clientLabel, callback);
+}
 function getClientLabels() {
   const labels = _getDefaultLabels();
+  const contextLabel = clientLabelLocalStorage.getStore();
+  if (contextLabel) {
+    labels.push(contextLabel);
+  }
   return labels;
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  getClientLabels
+  getClientLabels,
+  parseUserAgent,
+  runWithClientLabel
 });

package/dist/cjs/utils/env_aware_utils.js

@@ -42,6 +42,10 @@ function isBrowser() {
 }
 const UUID_MASK = "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx";
 function randomUUID() {
+  var _a;
+  if ((_a = globalThis.crypto) == null ? void 0 : _a.randomUUID) {
+    return globalThis.crypto.randomUUID();
+  }
   let uuid = "";
   for (let i = 0; i < UUID_MASK.length; i++) {
     const randomValue = Math.random() * 16 | 0;
@@ -57,7 +61,16 @@ function randomUUID() {
 }
 function base64Encode(data) {
   if (isBrowser()) {
-    return window.btoa(data);
+    let strData = "";
+    if (typeof data === "string") {
+      strData = data;
+    } else {
+      const len = data.byteLength;
+      for (let i = 0; i < len; i++) {
+        strData += String.fromCharCode(data[i]);
+      }
+    }
+    return window.btoa(strData);
   }
   return Buffer.from(data).toString("base64");
 }

package/dist/cjs/utils/live_connection_utils.js

@@ -0,0 +1,284 @@
+/**
+  * @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 live_connection_utils_exports = {};
+__export(live_connection_utils_exports, {
+  LiveResponseAggregator: () => LiveResponseAggregator
+});
+module.exports = __toCommonJS(live_connection_utils_exports);
+var import_model_name = require("./model_name.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class LiveResponseAggregator {
+  constructor(modelVersion) {
+    this.modelVersion = modelVersion;
+    this.text = "";
+    this.isThought = false;
+    this.toolCallParts = [];
+    this.pendingGroundingMetadata = void 0;
+    this.inputTranscriptionText = "";
+    this.outputTranscriptionText = "";
+  }
+  *processMessage(message) {
+    if (message.usageMetadata) {
+      yield {
+        usageMetadata: message.usageMetadata,
+        ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+      };
+    }
+    if (message.serverContent) {
+      const serverContent = message.serverContent;
+      const content = serverContent.modelTurn;
+      if (serverContent.groundingMetadata) {
+        this.pendingGroundingMetadata = serverContent.groundingMetadata;
+      }
+      if (!(content && content.parts) && serverContent.groundingMetadata && !serverContent.turnComplete) {
+        yield {
+          groundingMetadata: serverContent.groundingMetadata,
+          ...serverContent.interrupted !== void 0 ? { interrupted: serverContent.interrupted } : {},
+          ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+        };
+      }
+      if (content && content.parts) {
+        const llmResponse = {
+          content,
+          ...serverContent.interrupted !== void 0 ? { interrupted: serverContent.interrupted } : {},
+          ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+        };
+        if (!serverContent.turnComplete && serverContent.groundingMetadata) {
+          llmResponse.groundingMetadata = serverContent.groundingMetadata;
+        }
+        const hasInlineData = content.parts.some((p) => p.inlineData);
+        for (const part of content.parts) {
+          if (part.text) {
+            const currentIsThought = !!part.thought;
+            if (this.text && currentIsThought !== this.isThought) {
+              yield this.buildFullTextResponse(this.text, this.isThought);
+              this.text = "";
+              this.isThought = false;
+            }
+            this.text += part.text;
+            this.isThought = currentIsThought;
+            llmResponse.partial = true;
+          }
+        }
+        if (this.text && !content.parts.some((p) => p.text) && !hasInlineData) {
+          yield this.buildFullTextResponse(this.text, this.isThought);
+          this.text = "";
+          this.isThought = false;
+        }
+        yield llmResponse;
+      }
+      if (serverContent.inputTranscription) {
+        if (serverContent.inputTranscription.text) {
+          this.inputTranscriptionText += serverContent.inputTranscription.text;
+          yield {
+            inputTranscription: {
+              text: serverContent.inputTranscription.text,
+              finished: false
+            },
+            partial: true,
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+        }
+        if (serverContent.inputTranscription.finished) {
+          yield {
+            inputTranscription: {
+              text: this.inputTranscriptionText,
+              finished: true
+            },
+            partial: false,
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+          this.inputTranscriptionText = "";
+        }
+      }
+      if (serverContent.outputTranscription) {
+        if (serverContent.outputTranscription.text) {
+          this.outputTranscriptionText += serverContent.outputTranscription.text;
+          yield {
+            outputTranscription: {
+              text: serverContent.outputTranscription.text,
+              finished: false
+            },
+            partial: true,
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+        }
+        if (serverContent.outputTranscription.finished) {
+          yield {
+            outputTranscription: {
+              text: this.outputTranscriptionText,
+              finished: true
+            },
+            partial: false,
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+          this.outputTranscriptionText = "";
+        }
+      }
+      if (serverContent.interrupted || serverContent.turnComplete || serverContent.generationComplete) {
+        if (this.inputTranscriptionText) {
+          yield {
+            inputTranscription: {
+              text: this.inputTranscriptionText,
+              finished: true
+            },
+            partial: false,
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+          this.inputTranscriptionText = "";
+        }
+        if (this.outputTranscriptionText) {
+          yield {
+            outputTranscription: {
+              text: this.outputTranscriptionText,
+              finished: true
+            },
+            partial: false,
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+          this.outputTranscriptionText = "";
+        }
+      }
+      if (serverContent.turnComplete) {
+        let gMetadataToYield = this.pendingGroundingMetadata;
+        if (this.text) {
+          yield this.buildFullTextResponse(
+            this.text,
+            this.isThought,
+            gMetadataToYield
+          );
+          this.text = "";
+          this.isThought = false;
+          gMetadataToYield = void 0;
+        }
+        if (this.toolCallParts.length > 0) {
+          yield {
+            content: { role: "model", parts: this.toolCallParts },
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+          this.toolCallParts = [];
+        }
+        const finalResponse = {
+          turnComplete: true,
+          ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+        };
+        if (serverContent.interrupted !== void 0) {
+          finalResponse.interrupted = serverContent.interrupted;
+        }
+        const finalGrounding = serverContent.groundingMetadata || gMetadataToYield;
+        if (finalGrounding !== void 0 && finalGrounding !== null) {
+          finalResponse.groundingMetadata = finalGrounding;
+        }
+        yield finalResponse;
+      }
+      if (serverContent.interrupted) {
+        if (this.text) {
+          yield this.buildFullTextResponse(this.text, this.isThought);
+          this.text = "";
+          this.isThought = false;
+        } else {
+          yield {
+            interrupted: serverContent.interrupted,
+            ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+          };
+        }
+      }
+    }
+    if (message.toolCall) {
+      if (this.text) {
+        yield this.buildFullTextResponse(this.text, this.isThought);
+        this.text = "";
+        this.isThought = false;
+      }
+      if (message.toolCall.functionCalls) {
+        this.toolCallParts.push(
+          ...message.toolCall.functionCalls.map((fc) => ({
+            functionCall: fc
+          }))
+        );
+      }
+      const isGemini3x = (0, import_model_name.isGemini3xFlashLive)(this.modelVersion);
+      if (isGemini3x && this.toolCallParts.length > 0) {
+        yield {
+          content: { role: "model", parts: this.toolCallParts },
+          ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+        };
+        this.toolCallParts = [];
+      }
+    }
+    if (message.sessionResumptionUpdate) {
+      yield {
+        liveSessionResumptionUpdate: message.sessionResumptionUpdate,
+        ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+      };
+    }
+    if (message.goAway) {
+      yield {
+        goAway: message.goAway,
+        ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+      };
+    }
+  }
+  /**
+   * Flushes any remaining aggregated components when the connection is closed.
+   */
+  *close() {
+    if (this.toolCallParts.length > 0) {
+      yield {
+        content: { role: "model", parts: this.toolCallParts },
+        ...this.modelVersion ? { modelVersion: this.modelVersion } : {}
+      };
+      this.toolCallParts = [];
+    }
+  }
+  buildFullTextResponse(text, isThought, groundingMetadata) {
+    const part = { text };
+    if (isThought) {
+      part.thought = true;
+    }
+    const response = {
+      content: {
+        role: "model",
+        parts: [part]
+      },
+      partial: false
+    };
+    if (groundingMetadata !== void 0 && groundingMetadata !== null) {
+      response.groundingMetadata = groundingMetadata;
+    }
+    if (this.modelVersion) {
+      response.modelVersion = this.modelVersion;
+    }
+    return response;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  LiveResponseAggregator
+});

package/dist/cjs/utils/model_name.js

@@ -27,6 +27,7 @@ __export(model_name_exports, {
   extractModelName: () => extractModelName,
   isGemini1Model: () => isGemini1Model,
   isGemini2OrAbove: () => isGemini2OrAbove,
+  isGemini3xFlashLive: () => isGemini3xFlashLive,
   isGeminiModel: () => isGeminiModel,
   isGeminiModelIdCheckDisabled: () => isGeminiModelIdCheckDisabled
 });
@@ -77,6 +78,13 @@ function isGemini2OrAbove(modelString) {
   const parsedVersion = parseVersion(versionString);
   return parsedVersion.valid && parsedVersion.major >= 2;
 }
+function isGemini3xFlashLive(modelString) {
+  if (!modelString) {
+    return false;
+  }
+  const modelName = extractModelName(modelString);
+  return modelName.startsWith("gemini-3.") && modelName.includes("-flash-live");
+}
 function isGeminiModelIdCheckDisabled() {
   return (0, import_env_aware_utils.getBooleanEnvVar)("ADK_DISABLE_GEMINI_MODEL_ID_CHECK");
 }
@@ -85,6 +93,7 @@ function isGeminiModelIdCheckDisabled() {
   extractModelName,
   isGemini1Model,
   isGemini2OrAbove,
+  isGemini3xFlashLive,
   isGeminiModel,
   isGeminiModelIdCheckDisabled
 });

package/dist/cjs/utils/streaming_utils.js

@@ -38,6 +38,7 @@ var import_llm_response = require("../models/llm_response.js");
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+const UNSAFE_PROPERTY_KEYS = /* @__PURE__ */ new Set(["__proto__", "constructor", "prototype"]);
 class ProgressiveStrategy {
   constructor() {
     this.partsSequence = [];
@@ -104,10 +105,13 @@ class ProgressiveStrategy {
     const pathParts = import_jsonpath_plus.JSONPath.toPathArray(jsonPath).filter(
       (p) => p !== "$" && p !== "$["
     );
+    if (pathParts.some((part) => UNSAFE_PROPERTY_KEYS.has(String(part)))) {
+      return;
+    }
     let current = this.currentFcArgs;
     for (let i = 0; i < pathParts.length - 1; i++) {
       const part = pathParts[i];
-      if (!(part in current)) {
+      if (!Object.prototype.hasOwnProperty.call(current, part)) {
         current[part] = {};
       }
       current = current[part];
@@ -299,17 +303,13 @@ class StreamingResponseAggregator {
   )) {
     this.isProgressiveMode = isProgressiveMode;
     this.lastThoughtSignature = {};
-    this.sawFunctionCall = false;
     this.strategy = this.isProgressiveMode ? new ProgressiveStrategy() : new NonProgressiveStrategy();
   }
   async *processResponse(response) {
     var _a, _b;
     const llmResponse = (0, import_llm_response.createLlmResponse)(response);
     const parts = (_b = (_a = llmResponse.content) == null ? void 0 : _a.parts) != null ? _b : [];
-    if (parts.some((part) => part.functionCall)) {
-      this.sawFunctionCall = true;
-    }
-    if (this.sawFunctionCall && llmResponse.finishReason === import_genai.FinishReason.STOP && parts.length > 0 && parts.every(isEmptyContentPart)) {
+    if (parts.every(isEmptyContentPart) && (llmResponse.finishReason === void 0 || llmResponse.finishReason === import_genai.FinishReason.STOP)) {
       if (llmResponse.usageMetadata) {
         this.usageMetadata = llmResponse.usageMetadata;
       }
@@ -349,16 +349,23 @@ class StreamingResponseAggregator {
   close() {
     var _a, _b, _c;
     const finalParts = this.strategy.close();
-    if (!finalParts) {
+    const hasMetadata = this.usageMetadata !== void 0 || this.groundingMetadata !== void 0 || this.citationMetadata !== void 0;
+    if (!finalParts && !hasMetadata) {
       return void 0;
     }
     const candidate = (_b = (_a = this.response) == null ? void 0 : _a.candidates) == null ? void 0 : _b[0];
     const finishReason = (_c = this.finishReason) != null ? _c : candidate == null ? void 0 : candidate.finishReason;
     return {
-      content: {
-        role: "model",
-        parts: finalParts
-      },
+      // Only attach a model `content` when there are actual parts to emit. When
+      // the turn accumulated no parts (e.g. a function call that was already
+      // flushed, followed by a trailing STOP chunk that carries only usage
+      // metadata) we still surface that trailing metadata, but MUST NOT emit
+      // `content: { parts: [] }`. An empty-parts model turn corrupts the session
+      // history: on the next request the Vertex AI backend rejects the whole
+      // call with HTTP 400 "Unable to submit request because it must include at
+      // least one parts field", which breaks multi-turn tool-using sessions
+      // right after the first tool call (google/adk-js#21, #22).
+      content: finalParts ? { role: "model", parts: finalParts } : void 0,
       groundingMetadata: this.groundingMetadata,
       citationMetadata: this.citationMetadata,
       errorCode: finishReason === import_genai.FinishReason.STOP ? void 0 : finishReason,

package/dist/cjs/utils/task.js

@@ -0,0 +1,62 @@
+/**
+  * @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 task_exports = {};
+__export(task_exports, {
+  Task: () => Task
+});
+module.exports = __toCommonJS(task_exports);
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class Task {
+  constructor(executable) {
+    this.executable = executable;
+    this.isDone = false;
+    this.abortController = new AbortController();
+    const markDone = () => {
+      this.isDone = true;
+    };
+    this.promise = executable(this.abortController.signal);
+    this.promise.then(markDone).catch(markDone);
+  }
+  /**
+   * Cancels the task execution.
+   */
+  cancel() {
+    this.abortController.abort();
+  }
+  /**
+   * Returns true if the task has completed (either resolved or rejected).
+   */
+  done() {
+    return this.isDone;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Task
+});

package/dist/cjs/version.js

@@ -32,7 +32,7 @@ module.exports = __toCommonJS(version_exports);
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const version = "1.2.0";
+const version = "1.4.0";
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   version

package/dist/esm/agents/active_streaming_tool.js

@@ -5,9 +5,14 @@ const require = topLevelCreateRequire(import.meta.url);
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { Task } from "../utils/task.js";
 class ActiveStreamingTool {
   constructor(params = {}) {
-    this.task = params.task;
+    if (params.task instanceof Promise) {
+      this.task = new Task(() => params.task);
+    } else {
+      this.task = params.task;
+    }
     this.stream = params.stream;
   }
 }

package/dist/esm/agents/base_agent.js

@@ -99,9 +99,27 @@ class BaseAgent {
         ctx,
         this,
         async function* () {
+          var _a2, _b;
+          const context2 = this.createInvocationContext(parentContext);
+          const beforeAgentCallbackEvent = await this.handleBeforeAgentCallback(context2);
+          if (beforeAgentCallbackEvent) {
+            yield beforeAgentCallbackEvent;
+          }
+          if (context2.endInvocation || ((_a2 = parentContext.abortSignal) == null ? void 0 : _a2.aborted)) {
+            return;
+          }
+          for await (const event of this.runLiveImpl(context2)) {
+            yield event;
+          }
+          if (context2.endInvocation || ((_b = parentContext.abortSignal) == null ? void 0 : _b.aborted)) {
+            return;
+          }
+          const afterAgentCallbackEvent = await this.handleAfterAgentCallback(context2);
+          if (afterAgentCallbackEvent) {
+            yield afterAgentCallbackEvent;
+          }
         }
       );
-      throw new Error("Live mode is not implemented yet.");
     } finally {
       span.end();
     }

package/dist/esm/agents/context.js

@@ -56,9 +56,6 @@ class Context extends ReadonlyContext {
       throw new Error("Artifact service is not initialized.");
     }
     return this.invocationContext.artifactService.loadArtifact({
-      appName: this.invocationContext.appName,
-      userId: this.invocationContext.userId,
-      sessionId: this.invocationContext.session.id,
       filename,
       version
     });
@@ -75,9 +72,6 @@ class Context extends ReadonlyContext {
       throw new Error("Artifact service is not initialized.");
     }
     const version = await this.invocationContext.artifactService.saveArtifact({
-      appName: this.invocationContext.appName,
-      userId: this.invocationContext.userId,
-      sessionId: this.invocationContext.session.id,
       filename,
       artifact
     });
@@ -110,11 +104,7 @@ class Context extends ReadonlyContext {
     if (!this.invocationContext.artifactService) {
       throw new Error("Artifact service is not initialized.");
     }
-    return this.invocationContext.artifactService.listArtifactKeys({
-      appName: this.invocationContext.session.appName,
-      userId: this.invocationContext.session.userId,
-      sessionId: this.invocationContext.session.id
-    });
+    return this.invocationContext.artifactService.listArtifactKeys();
   }
   /**
    * Searches the memory of the current user.