@google/adk 0.4.0 → 0.6.0

package/dist/web/a2a/metadata_converter_utils.js

@@ -0,0 +1,108 @@
+var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isTask } from "./a2a_event.js";
+const ADK_METADATA_KEY_PREFIX = "adk_";
+const A2A_METADATA_KEY_PREFIX = "a2a:";
+var A2AMetadataKeys = ((A2AMetadataKeys2) => {
+  A2AMetadataKeys2[A2AMetadataKeys2["APP_NAME"] = "".concat(ADK_METADATA_KEY_PREFIX, "app_name")] = "APP_NAME";
+  A2AMetadataKeys2[A2AMetadataKeys2["USER_ID"] = "".concat(ADK_METADATA_KEY_PREFIX, "user_id")] = "USER_ID";
+  A2AMetadataKeys2[A2AMetadataKeys2["SESSION_ID"] = "".concat(ADK_METADATA_KEY_PREFIX, "session_id")] = "SESSION_ID";
+  A2AMetadataKeys2[A2AMetadataKeys2["INVOCATION_ID"] = "".concat(ADK_METADATA_KEY_PREFIX, "invocation_id")] = "INVOCATION_ID";
+  A2AMetadataKeys2[A2AMetadataKeys2["AUTHOR"] = "".concat(ADK_METADATA_KEY_PREFIX, "author")] = "AUTHOR";
+  A2AMetadataKeys2[A2AMetadataKeys2["BRANCH"] = "".concat(ADK_METADATA_KEY_PREFIX, "branch")] = "BRANCH";
+  A2AMetadataKeys2[A2AMetadataKeys2["DATA_PART_TYPE"] = "".concat(ADK_METADATA_KEY_PREFIX, "type")] = "DATA_PART_TYPE";
+  A2AMetadataKeys2[A2AMetadataKeys2["PARTIAL"] = "".concat(ADK_METADATA_KEY_PREFIX, "partial")] = "PARTIAL";
+  A2AMetadataKeys2[A2AMetadataKeys2["ESCALATE"] = "".concat(ADK_METADATA_KEY_PREFIX, "escalate")] = "ESCALATE";
+  A2AMetadataKeys2[A2AMetadataKeys2["TRANSFER_TO_AGENT"] = "".concat(ADK_METADATA_KEY_PREFIX, "transfer_to_agent")] = "TRANSFER_TO_AGENT";
+  A2AMetadataKeys2[A2AMetadataKeys2["IS_LONG_RUNNING"] = "".concat(ADK_METADATA_KEY_PREFIX, "is_long_running")] = "IS_LONG_RUNNING";
+  A2AMetadataKeys2[A2AMetadataKeys2["THOUGHT"] = "".concat(ADK_METADATA_KEY_PREFIX, "thought")] = "THOUGHT";
+  A2AMetadataKeys2[A2AMetadataKeys2["ERROR_CODE"] = "".concat(ADK_METADATA_KEY_PREFIX, "error_code")] = "ERROR_CODE";
+  A2AMetadataKeys2[A2AMetadataKeys2["ERROR_MESSAGE"] = "".concat(ADK_METADATA_KEY_PREFIX, "error_message")] = "ERROR_MESSAGE";
+  A2AMetadataKeys2[A2AMetadataKeys2["CITATION_METADATA"] = "".concat(ADK_METADATA_KEY_PREFIX, "citation_metadata")] = "CITATION_METADATA";
+  A2AMetadataKeys2[A2AMetadataKeys2["GROUNDING_METADATA"] = "".concat(ADK_METADATA_KEY_PREFIX, "grounding_metadata")] = "GROUNDING_METADATA";
+  A2AMetadataKeys2[A2AMetadataKeys2["USAGE_METADATA"] = "".concat(ADK_METADATA_KEY_PREFIX, "usage_metadata")] = "USAGE_METADATA";
+  A2AMetadataKeys2[A2AMetadataKeys2["CUSTOM_METADATA"] = "".concat(ADK_METADATA_KEY_PREFIX, "custom_metadata")] = "CUSTOM_METADATA";
+  A2AMetadataKeys2[A2AMetadataKeys2["VIDEO_METADATA"] = "".concat(ADK_METADATA_KEY_PREFIX, "video_metadata")] = "VIDEO_METADATA";
+  return A2AMetadataKeys2;
+})(A2AMetadataKeys || {});
+var AdkMetadataKeys = ((AdkMetadataKeys2) => {
+  AdkMetadataKeys2[AdkMetadataKeys2["TASK_ID"] = "".concat(A2A_METADATA_KEY_PREFIX, "task_id")] = "TASK_ID";
+  AdkMetadataKeys2[AdkMetadataKeys2["CONTEXT_ID"] = "".concat(A2A_METADATA_KEY_PREFIX, "context_id")] = "CONTEXT_ID";
+  return AdkMetadataKeys2;
+})(AdkMetadataKeys || {});
+function getAdkEventMetadata(a2aEvent) {
+  return {
+    [AdkMetadataKeys.TASK_ID]: isTask(a2aEvent) ? a2aEvent.id : a2aEvent.taskId,
+    [AdkMetadataKeys.CONTEXT_ID]: a2aEvent.contextId
+  };
+}
+function getA2AEventMetadata(adkEvent, {
+  appName,
+  userId,
+  sessionId
+}) {
+  return __spreadProps(__spreadValues(__spreadValues({}, getA2AEventMetadataFromActions(adkEvent.actions)), getA2ASessionMetadata({
+    appName,
+    userId,
+    sessionId
+  })), {
+    [A2AMetadataKeys.INVOCATION_ID]: adkEvent.invocationId,
+    [A2AMetadataKeys.AUTHOR]: adkEvent.author,
+    [A2AMetadataKeys.BRANCH]: adkEvent.branch,
+    [A2AMetadataKeys.ERROR_CODE]: adkEvent.errorMessage,
+    [A2AMetadataKeys.ERROR_MESSAGE]: adkEvent.errorMessage,
+    [A2AMetadataKeys.CITATION_METADATA]: adkEvent.citationMetadata,
+    [A2AMetadataKeys.GROUNDING_METADATA]: adkEvent.groundingMetadata,
+    [A2AMetadataKeys.USAGE_METADATA]: adkEvent.usageMetadata,
+    [A2AMetadataKeys.CUSTOM_METADATA]: adkEvent.customMetadata,
+    [A2AMetadataKeys.PARTIAL]: adkEvent.partial,
+    [A2AMetadataKeys.IS_LONG_RUNNING]: (adkEvent.longRunningToolIds || []).length > 0
+  });
+}
+function getA2ASessionMetadata({
+  appName,
+  userId,
+  sessionId
+}) {
+  return {
+    [A2AMetadataKeys.APP_NAME]: appName,
+    [A2AMetadataKeys.USER_ID]: userId,
+    [A2AMetadataKeys.SESSION_ID]: sessionId
+  };
+}
+function getA2AEventMetadataFromActions(actions) {
+  return {
+    [A2AMetadataKeys.ESCALATE]: actions.escalate,
+    [A2AMetadataKeys.TRANSFER_TO_AGENT]: actions.transferToAgent
+  };
+}
+export {
+  A2AMetadataKeys,
+  AdkMetadataKeys,
+  getA2AEventMetadata,
+  getA2AEventMetadataFromActions,
+  getA2ASessionMetadata,
+  getAdkEventMetadata
+};

package/dist/web/a2a/part_converter_utils.js

@@ -3,12 +3,11 @@
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-var MetadataKeys = /* @__PURE__ */ ((MetadataKeys2) => {
-  MetadataKeys2["TYPE"] = "adk_type";
-  MetadataKeys2["LONG_RUNNING"] = "adk_is_long_running";
-  MetadataKeys2["THOUGHT"] = "adk_thought";
-  return MetadataKeys2;
-})(MetadataKeys || {});
+import {
+  createModelContent,
+  createUserContent
+} from "@google/genai";
+import { A2AMetadataKeys } from "./metadata_converter_utils.js";
 var DataPartType = /* @__PURE__ */ ((DataPartType2) => {
   DataPartType2["FUNCTION_CALL"] = "function_call";
   DataPartType2["FUNCTION_RESPONSE"] = "function_response";
@@ -16,7 +15,7 @@ var DataPartType = /* @__PURE__ */ ((DataPartType2) => {
   DataPartType2["CODE_EXECUTABLE_CODE"] = "executable_code";
   return DataPartType2;
 })(DataPartType || {});
-function toA2AParts(parts, longRunningToolIDs = []) {
+function toA2AParts(parts = [], longRunningToolIDs = []) {
   return parts.map((part) => toA2APart(part, longRunningToolIDs));
 }
 function toA2APart(part, longRunningToolIDs) {
@@ -30,14 +29,18 @@ function toA2APart(part, longRunningToolIDs) {
 }
 function toA2ATextPart(part) {
   const a2aPart = { kind: "text", text: part.text || "" };
-  if ("thought" in part && part["thought"]) {
+  if (part.thought) {
     a2aPart.metadata = {
-      ["adk_thought" /* THOUGHT */]: true
+      [A2AMetadataKeys.THOUGHT]: true
     };
   }
   return a2aPart;
 }
 function toA2AFilePart(part) {
+  const metadata = {};
+  if (part.videoMetadata) {
+    metadata[A2AMetadataKeys.VIDEO_METADATA] = part.videoMetadata;
+  }
   if (part.fileData) {
     return {
       kind: "file",
@@ -45,7 +48,7 @@ function toA2AFilePart(part) {
         uri: part.fileData.fileUri || "",
         mimeType: part.fileData.mimeType
       },
-      metadata: {}
+      metadata
     };
   }
   if (part.inlineData) {
@@ -55,37 +58,41 @@ function toA2AFilePart(part) {
         bytes: part.inlineData.data || "",
         mimeType: part.inlineData.mimeType
       },
-      metadata: {}
+      metadata
     };
   }
   throw new Error("Not a file part: ".concat(JSON.stringify(part)));
 }
 function toA2ADataPart(part, longRunningToolIDs = []) {
-  let type;
+  let dataPartType;
   let data;
   if (part.functionCall) {
-    type = "function_call" /* FUNCTION_CALL */;
+    dataPartType = "function_call" /* FUNCTION_CALL */;
     data = part.functionCall;
   } else if (part.functionResponse) {
-    type = "function_response" /* FUNCTION_RESPONSE */;
+    dataPartType = "function_response" /* FUNCTION_RESPONSE */;
     data = part.functionResponse;
   } else if (part.executableCode) {
-    type = "executable_code" /* CODE_EXECUTABLE_CODE */;
+    dataPartType = "executable_code" /* CODE_EXECUTABLE_CODE */;
     data = part.executableCode;
   } else if (part.codeExecutionResult) {
-    type = "code_execution_result" /* CODE_EXEC_RESULT */;
+    dataPartType = "code_execution_result" /* CODE_EXEC_RESULT */;
     data = part.codeExecutionResult;
   } else {
-    throw new Error("Unknown part type: ".concat(JSON.stringify(part)));
+    return {
+      kind: "data",
+      data: {},
+      metadata: {}
+    };
   }
   const metadata = {
-    ["adk_type" /* TYPE */]: type
+    [A2AMetadataKeys.DATA_PART_TYPE]: dataPartType
   };
-  if (part.functionCall && part.functionCall.name && longRunningToolIDs.includes(part.functionCall.name)) {
-    metadata["adk_is_long_running" /* LONG_RUNNING */] = true;
+  if (part.functionCall && part.functionCall.id && longRunningToolIDs.includes(part.functionCall.id)) {
+    metadata[A2AMetadataKeys.IS_LONG_RUNNING] = true;
   }
-  if (part.functionResponse && part.functionResponse.name && longRunningToolIDs.includes(part.functionResponse.name)) {
-    metadata["adk_is_long_running" /* LONG_RUNNING */] = true;
+  if (part.functionResponse && part.functionResponse.id && longRunningToolIDs.includes(part.functionResponse.id)) {
+    metadata[A2AMetadataKeys.IS_LONG_RUNNING] = true;
   }
   return {
     kind: "data",
@@ -93,6 +100,10 @@ function toA2ADataPart(part, longRunningToolIDs = []) {
     metadata
   };
 }
+function toGenAIContent(a2aMessage) {
+  const parts = toGenAIParts(a2aMessage.parts);
+  return a2aMessage.role === "user" ? createUserContent(parts) : createModelContent(parts);
+}
 function toGenAIParts(a2aParts) {
   return a2aParts.map((a2aPart) => toGenAIPart(a2aPart));
 }
@@ -112,25 +123,28 @@ function toGenAIPartText(a2aPart) {
   var _a;
   return {
     text: a2aPart.text,
-    thought: !!((_a = a2aPart.metadata) == null ? void 0 : _a["adk_thought" /* THOUGHT */])
+    thought: !!((_a = a2aPart.metadata) == null ? void 0 : _a[A2AMetadataKeys.THOUGHT])
   };
 }
 function toGenAIPartFile(a2aPart) {
+  var _a;
+  const part = {};
+  if ((_a = a2aPart.metadata) == null ? void 0 : _a[A2AMetadataKeys.VIDEO_METADATA]) {
+    part.videoMetadata = a2aPart.metadata[A2AMetadataKeys.VIDEO_METADATA];
+  }
   if ("bytes" in a2aPart.file) {
-    return {
-      inlineData: {
-        data: a2aPart.file.bytes,
-        mimeType: a2aPart.file.mimeType || ""
-      }
+    part.inlineData = {
+      data: a2aPart.file.bytes,
+      mimeType: a2aPart.file.mimeType || ""
     };
+    return part;
   }
   if ("uri" in a2aPart.file) {
-    return {
-      fileData: {
-        fileUri: a2aPart.file.uri,
-        mimeType: a2aPart.file.mimeType || ""
-      }
+    part.fileData = {
+      fileUri: a2aPart.file.uri,
+      mimeType: a2aPart.file.mimeType || ""
     };
+    return part;
   }
   throw new Error("Not a file part: ".concat(JSON.stringify(a2aPart)));
 }
@@ -140,7 +154,7 @@ function toGenAIPartData(a2aPart) {
     throw new Error("No data in part: ".concat(JSON.stringify(a2aPart)));
   }
   const data = a2aPart.data;
-  const type = (_a = a2aPart.metadata) == null ? void 0 : _a["adk_type" /* TYPE */];
+  const type = (_a = a2aPart.metadata) == null ? void 0 : _a[A2AMetadataKeys.DATA_PART_TYPE];
   if (type === "function_call" /* FUNCTION_CALL */) {
     return { functionCall: data };
   }
@@ -163,6 +177,7 @@ export {
   toA2APart,
   toA2AParts,
   toA2ATextPart,
+  toGenAIContent,
   toGenAIPart,
   toGenAIPartData,
   toGenAIPartFile,

package/dist/web/agents/base_agent.js

@@ -78,7 +78,7 @@ import {
   traceAgentInvocation,
   tracer
 } from "../telemetry/tracing.js";
-import { CallbackContext } from "./callback_context.js";
+import { Context } from "./context.js";
 import { InvocationContext } from "./invocation_context.js";
 const BASE_AGENT_SIGNATURE_SYMBOL = Symbol.for("google.adk.baseAgent");
 function isBaseAgent(obj) {
@@ -239,7 +239,7 @@ class BaseAgent {
     if (this.beforeAgentCallback.length === 0) {
       return void 0;
     }
-    const callbackContext = new CallbackContext({ invocationContext });
+    const callbackContext = new Context({ invocationContext });
     for (const callback of this.beforeAgentCallback) {
       const content = await callback(callbackContext);
       if (content) {
@@ -274,7 +274,7 @@ class BaseAgent {
     if (this.afterAgentCallback.length === 0) {
       return void 0;
     }
-    const callbackContext = new CallbackContext({ invocationContext });
+    const callbackContext = new Context({ invocationContext });
     for (const callback of this.afterAgentCallback) {
       const content = await callback(callbackContext);
       if (content) {

package/dist/{esm/tools/tool_context.js → web/agents/context.js}

@@ -3,29 +3,84 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { CallbackContext } from "../agents/callback_context.js";
 import { AuthHandler } from "../auth/auth_handler.js";
+import { createEventActions } from "../events/event_actions.js";
+import { State } from "../sessions/state.js";
 import { ToolConfirmation } from "../tools/tool_confirmation.js";
-class ToolContext extends CallbackContext {
+import { ReadonlyContext } from "./readonly_context.js";
+class Context extends ReadonlyContext {
   /**
-   * @param params.invocationContext The invocation context of the tool.
-   * @param params.eventActions The event actions of the current tool call.
-   * @param params.functionCallId The function call id of the current tool call.
+   * @param options The configuration options for the Context.
+   * @param options.invocationContext The invocation context.
+   * @param options.eventActions The event actions of the current call.
+   * @param options.functionCallId The function call id of the current tool call.
    *     This id was returned in the function call event from LLM to identify a
    *     function call. If LLM didn't return this id, ADK will assign one to it.
    *     This id is used to map function call response to the original function
    *     call.
-   * @param params.toolConfirmation The tool confirmation of the current tool
+   * @param options.toolConfirmation The tool confirmation of the current tool
    *     call.
    */
-  constructor(params) {
-    super(params);
-    this.functionCallId = params.functionCallId;
-    this.toolConfirmation = params.toolConfirmation;
+  constructor(options) {
+    super(options.invocationContext);
+    this.eventActions = options.eventActions || createEventActions();
+    this._state = new State(
+      options.invocationContext.session.state,
+      this.eventActions.stateDelta
+    );
+    this.functionCallId = options.functionCallId;
+    this.toolConfirmation = options.toolConfirmation;
+  }
+  /**
+   * The delta-aware state of the current session.
+   */
+  get state() {
+    return this._state;
   }
   get actions() {
     return this.eventActions;
   }
+  /**
+   * Loads an artifact attached to the current session.
+   *
+   * @param filename The filename of the artifact.
+   * @param version The version of the artifact. If not provided, the latest
+   *     version will be used.
+   * @return A promise that resolves to the loaded artifact.
+   */
+  loadArtifact(filename, version) {
+    if (!this.invocationContext.artifactService) {
+      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
+    });
+  }
+  /**
+   * Saves an artifact attached to the current session.
+   *
+   * @param filename The filename of the artifact.
+   * @param artifact The artifact to save.
+   * @return A promise that resolves to the version of the saved artifact.
+   */
+  async saveArtifact(filename, artifact) {
+    if (!this.invocationContext.artifactService) {
+      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
+    });
+    this.eventActions.artifactDelta[filename] = version;
+    return version;
+  }
   requestCredential(authConfig) {
     if (!this.functionCallId) {
       throw new Error("functionCallId is not set.");
@@ -90,5 +145,5 @@ class ToolContext extends CallbackContext {
   }
 }
 export {
-  ToolContext
+  Context
 };

package/dist/web/agents/functions.js

@@ -7,9 +7,9 @@ import { createUserContent } from "@google/genai";
 import { isEmpty } from "lodash-es";
 import { createEvent, getFunctionCalls } from "../events/event.js";
 import { mergeEventActions } from "../events/event_actions.js";
-import { ToolContext } from "../tools/tool_context.js";
 import { randomUUID } from "../utils/env_aware_utils.js";
 import { logger } from "../utils/logger.js";
+import { Context } from "./context.js";
 import {
   traceMergedToolCalls,
   tracer,
@@ -352,7 +352,7 @@ function getToolAndContext({
       "Function ".concat(functionCall.name, " is not found in the toolsDict.")
     );
   }
-  const toolContext = new ToolContext({
+  const toolContext = new Context({
     invocationContext,
     functionCallId: functionCall.id || void 0,
     toolConfirmation