@google/adk 0.4.0 → 0.6.0

package/dist/cjs/a2a/event_processor_utils.js

@@ -0,0 +1,180 @@
+/**
+  * @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 event_processor_utils_exports = {};
+__export(event_processor_utils_exports, {
+  getFinalTaskStatusUpdate: () => getFinalTaskStatusUpdate,
+  getTaskInputRequiredEvent: () => getTaskInputRequiredEvent
+});
+module.exports = __toCommonJS(event_processor_utils_exports);
+var import_event_actions = require("../events/event_actions.js");
+var import_a2a_event = require("./a2a_event.js");
+var import_metadata_converter_utils = require("./metadata_converter_utils.js");
+var import_part_converter_utils = require("./part_converter_utils.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function getFinalTaskStatusUpdate(adkEvents, context) {
+  var _a, _b;
+  const finalEventActions = (0, import_event_actions.createEventActions)();
+  for (const adkEvent of adkEvents) {
+    if (adkEvent.errorCode || adkEvent.errorMessage) {
+      return (0, import_a2a_event.createTaskFailedEvent)({
+        taskId: context.requestContext.taskId,
+        contextId: context.requestContext.contextId,
+        error: new Error(adkEvent.errorMessage || adkEvent.errorCode),
+        metadata: {
+          ...(0, import_metadata_converter_utils.getA2AEventMetadata)(adkEvent, context),
+          ...(0, import_metadata_converter_utils.getA2AEventMetadataFromActions)(finalEventActions)
+        }
+      });
+    }
+    finalEventActions.escalate = finalEventActions.escalate || ((_a = adkEvent.actions) == null ? void 0 : _a.escalate);
+    if ((_b = adkEvent.actions) == null ? void 0 : _b.transferToAgent) {
+      finalEventActions.transferToAgent = adkEvent.actions.transferToAgent;
+    }
+  }
+  const inputRequiredEvent = scanForInputRequiredEvents(adkEvents, context);
+  if (inputRequiredEvent) {
+    return {
+      ...inputRequiredEvent,
+      metadata: {
+        ...inputRequiredEvent.metadata,
+        ...(0, import_metadata_converter_utils.getA2AEventMetadataFromActions)(finalEventActions)
+      }
+    };
+  }
+  return (0, import_a2a_event.createTaskCompletedEvent)({
+    taskId: context.requestContext.taskId,
+    contextId: context.requestContext.contextId,
+    metadata: {
+      ...(0, import_metadata_converter_utils.getA2ASessionMetadata)(context),
+      ...(0, import_metadata_converter_utils.getA2AEventMetadataFromActions)(finalEventActions)
+    }
+  });
+}
+function scanForInputRequiredEvents(adkEvents, context) {
+  var _a, _b;
+  const inputRequiredParts = [];
+  const inputRequiredFunctionCallIds = /* @__PURE__ */ new Set();
+  for (const adkEvent of adkEvents) {
+    if (!((_b = (_a = adkEvent.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.length)) {
+      continue;
+    }
+    for (const genAIPart of adkEvent.content.parts) {
+      const longRunningFunctionCallId = getLongRunnningFunctionCallId(
+        genAIPart,
+        adkEvent.longRunningToolIds,
+        inputRequiredParts
+      );
+      if (!longRunningFunctionCallId) {
+        continue;
+      }
+      const isAlreadyAdded = inputRequiredFunctionCallIds.has(
+        longRunningFunctionCallId
+      );
+      if (isAlreadyAdded) {
+        continue;
+      }
+      inputRequiredParts.push(genAIPart);
+      inputRequiredFunctionCallIds.add(longRunningFunctionCallId);
+    }
+  }
+  if (inputRequiredParts.length > 0) {
+    return (0, import_a2a_event.createTaskInputRequiredEvent)({
+      taskId: context.requestContext.taskId,
+      contextId: context.requestContext.contextId,
+      parts: (0, import_part_converter_utils.toA2AParts)(inputRequiredParts, [...inputRequiredFunctionCallIds]),
+      metadata: (0, import_metadata_converter_utils.getA2ASessionMetadata)(context)
+    });
+  }
+  return void 0;
+}
+function getLongRunnningFunctionCallId(genAIPart, longRunningToolIds = [], inputRequiredParts = []) {
+  var _a, _b, _c;
+  const functionCallId = (_a = genAIPart.functionCall) == null ? void 0 : _a.id;
+  const functionResponseId = (_b = genAIPart.functionResponse) == null ? void 0 : _b.id;
+  if (!functionCallId && !functionResponseId) {
+    return;
+  }
+  if (functionCallId && longRunningToolIds.includes(functionCallId)) {
+    return functionCallId;
+  }
+  if (functionResponseId && longRunningToolIds.includes(functionResponseId)) {
+    return functionResponseId;
+  }
+  for (const part of inputRequiredParts) {
+    if (((_c = part.functionCall) == null ? void 0 : _c.id) === functionResponseId) {
+      return functionResponseId;
+    }
+  }
+  return;
+}
+function getTaskInputRequiredEvent(task, genAIContent) {
+  var _a;
+  if (!task || !(0, import_a2a_event.isInputRequiredTaskStatusUpdateEvent)(task) || !task.status.message) {
+    return void 0;
+  }
+  const statusMsg = task.status.message;
+  const taskParts = (0, import_part_converter_utils.toGenAIParts)(statusMsg.parts);
+  for (const taskPart of taskParts) {
+    const functionCallId = (_a = taskPart.functionCall) == null ? void 0 : _a.id;
+    if (!functionCallId) {
+      continue;
+    }
+    const hasMatchingResponse = ((genAIContent == null ? void 0 : genAIContent.parts) || []).some(
+      (p) => {
+        var _a2;
+        return ((_a2 = p.functionResponse) == null ? void 0 : _a2.id) === functionCallId;
+      }
+    );
+    if (!hasMatchingResponse) {
+      return (0, import_a2a_event.createInputMissingErrorEvent)({
+        taskId: task.id,
+        contextId: task.contextId,
+        parts: [
+          ...statusMsg.parts.filter((p) => {
+            var _a2;
+            return !((_a2 = p.metadata) == null ? void 0 : _a2.validation_error);
+          }),
+          {
+            kind: "text",
+            text: `No input provided for function call id ${functionCallId}`,
+            metadata: {
+              validation_error: true
+            }
+          }
+        ]
+      });
+    }
+  }
+  return void 0;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  getFinalTaskStatusUpdate,
+  getTaskInputRequiredEvent
+});

package/dist/cjs/a2a/executor_context.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 executor_context_exports = {};
+__export(executor_context_exports, {
+  createExecutorContext: () => createExecutorContext
+});
+module.exports = __toCommonJS(executor_context_exports);
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function createExecutorContext({
+  session,
+  userContent,
+  requestContext
+}) {
+  return {
+    userId: session.userId,
+    sessionId: session.id,
+    appName: session.appName,
+    readonlyState: session.state,
+    events: session.events,
+    userContent,
+    requestContext
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  createExecutorContext
+});

package/dist/cjs/a2a/metadata_converter_utils.js

@@ -0,0 +1,126 @@
+/**
+  * @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 metadata_converter_utils_exports = {};
+__export(metadata_converter_utils_exports, {
+  A2AMetadataKeys: () => A2AMetadataKeys,
+  AdkMetadataKeys: () => AdkMetadataKeys,
+  getA2AEventMetadata: () => getA2AEventMetadata,
+  getA2AEventMetadataFromActions: () => getA2AEventMetadataFromActions,
+  getA2ASessionMetadata: () => getA2ASessionMetadata,
+  getAdkEventMetadata: () => getAdkEventMetadata
+});
+module.exports = __toCommonJS(metadata_converter_utils_exports);
+var import_a2a_event = require("./a2a_event.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const ADK_METADATA_KEY_PREFIX = "adk_";
+const A2A_METADATA_KEY_PREFIX = "a2a:";
+var A2AMetadataKeys = ((A2AMetadataKeys2) => {
+  A2AMetadataKeys2["APP_NAME"] = `${ADK_METADATA_KEY_PREFIX}app_name`;
+  A2AMetadataKeys2["USER_ID"] = `${ADK_METADATA_KEY_PREFIX}user_id`;
+  A2AMetadataKeys2["SESSION_ID"] = `${ADK_METADATA_KEY_PREFIX}session_id`;
+  A2AMetadataKeys2["INVOCATION_ID"] = `${ADK_METADATA_KEY_PREFIX}invocation_id`;
+  A2AMetadataKeys2["AUTHOR"] = `${ADK_METADATA_KEY_PREFIX}author`;
+  A2AMetadataKeys2["BRANCH"] = `${ADK_METADATA_KEY_PREFIX}branch`;
+  A2AMetadataKeys2["DATA_PART_TYPE"] = `${ADK_METADATA_KEY_PREFIX}type`;
+  A2AMetadataKeys2["PARTIAL"] = `${ADK_METADATA_KEY_PREFIX}partial`;
+  A2AMetadataKeys2["ESCALATE"] = `${ADK_METADATA_KEY_PREFIX}escalate`;
+  A2AMetadataKeys2["TRANSFER_TO_AGENT"] = `${ADK_METADATA_KEY_PREFIX}transfer_to_agent`;
+  A2AMetadataKeys2["IS_LONG_RUNNING"] = `${ADK_METADATA_KEY_PREFIX}is_long_running`;
+  A2AMetadataKeys2["THOUGHT"] = `${ADK_METADATA_KEY_PREFIX}thought`;
+  A2AMetadataKeys2["ERROR_CODE"] = `${ADK_METADATA_KEY_PREFIX}error_code`;
+  A2AMetadataKeys2["ERROR_MESSAGE"] = `${ADK_METADATA_KEY_PREFIX}error_message`;
+  A2AMetadataKeys2["CITATION_METADATA"] = `${ADK_METADATA_KEY_PREFIX}citation_metadata`;
+  A2AMetadataKeys2["GROUNDING_METADATA"] = `${ADK_METADATA_KEY_PREFIX}grounding_metadata`;
+  A2AMetadataKeys2["USAGE_METADATA"] = `${ADK_METADATA_KEY_PREFIX}usage_metadata`;
+  A2AMetadataKeys2["CUSTOM_METADATA"] = `${ADK_METADATA_KEY_PREFIX}custom_metadata`;
+  A2AMetadataKeys2["VIDEO_METADATA"] = `${ADK_METADATA_KEY_PREFIX}video_metadata`;
+  return A2AMetadataKeys2;
+})(A2AMetadataKeys || {});
+var AdkMetadataKeys = ((AdkMetadataKeys2) => {
+  AdkMetadataKeys2["TASK_ID"] = `${A2A_METADATA_KEY_PREFIX}task_id`;
+  AdkMetadataKeys2["CONTEXT_ID"] = `${A2A_METADATA_KEY_PREFIX}context_id`;
+  return AdkMetadataKeys2;
+})(AdkMetadataKeys || {});
+function getAdkEventMetadata(a2aEvent) {
+  return {
+    [AdkMetadataKeys.TASK_ID]: (0, import_a2a_event.isTask)(a2aEvent) ? a2aEvent.id : a2aEvent.taskId,
+    [AdkMetadataKeys.CONTEXT_ID]: a2aEvent.contextId
+  };
+}
+function getA2AEventMetadata(adkEvent, {
+  appName,
+  userId,
+  sessionId
+}) {
+  return {
+    ...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
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  A2AMetadataKeys,
+  AdkMetadataKeys,
+  getA2AEventMetadata,
+  getA2AEventMetadataFromActions,
+  getA2ASessionMetadata,
+  getAdkEventMetadata
+});

package/dist/cjs/a2a/part_converter_utils.js

@@ -29,6 +29,7 @@ __export(part_converter_utils_exports, {
   toA2APart: () => toA2APart,
   toA2AParts: () => toA2AParts,
   toA2ATextPart: () => toA2ATextPart,
+  toGenAIContent: () => toGenAIContent,
   toGenAIPart: () => toGenAIPart,
   toGenAIPartData: () => toGenAIPartData,
   toGenAIPartFile: () => toGenAIPartFile,
@@ -36,17 +37,13 @@ __export(part_converter_utils_exports, {
   toGenAIParts: () => toGenAIParts
 });
 module.exports = __toCommonJS(part_converter_utils_exports);
+var import_genai = require("@google/genai");
+var import_metadata_converter_utils = require("./metadata_converter_utils.js");
 /**
  * @license
  * 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 || {});
 var DataPartType = /* @__PURE__ */ ((DataPartType2) => {
   DataPartType2["FUNCTION_CALL"] = "function_call";
   DataPartType2["FUNCTION_RESPONSE"] = "function_response";
@@ -54,7 +51,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) {
@@ -68,14 +65,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
+      [import_metadata_converter_utils.A2AMetadataKeys.THOUGHT]: true
     };
   }
   return a2aPart;
 }
 function toA2AFilePart(part) {
+  const metadata = {};
+  if (part.videoMetadata) {
+    metadata[import_metadata_converter_utils.A2AMetadataKeys.VIDEO_METADATA] = part.videoMetadata;
+  }
   if (part.fileData) {
     return {
       kind: "file",
@@ -83,7 +84,7 @@ function toA2AFilePart(part) {
         uri: part.fileData.fileUri || "",
         mimeType: part.fileData.mimeType
       },
-      metadata: {}
+      metadata
     };
   }
   if (part.inlineData) {
@@ -93,37 +94,41 @@ function toA2AFilePart(part) {
         bytes: part.inlineData.data || "",
         mimeType: part.inlineData.mimeType
       },
-      metadata: {}
+      metadata
     };
   }
   throw new Error(`Not a file part: ${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: ${JSON.stringify(part)}`);
+    return {
+      kind: "data",
+      data: {},
+      metadata: {}
+    };
   }
   const metadata = {
-    ["adk_type" /* TYPE */]: type
+    [import_metadata_converter_utils.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[import_metadata_converter_utils.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[import_metadata_converter_utils.A2AMetadataKeys.IS_LONG_RUNNING] = true;
   }
   return {
     kind: "data",
@@ -131,6 +136,10 @@ function toA2ADataPart(part, longRunningToolIDs = []) {
     metadata
   };
 }
+function toGenAIContent(a2aMessage) {
+  const parts = toGenAIParts(a2aMessage.parts);
+  return a2aMessage.role === "user" ? (0, import_genai.createUserContent)(parts) : (0, import_genai.createModelContent)(parts);
+}
 function toGenAIParts(a2aParts) {
   return a2aParts.map((a2aPart) => toGenAIPart(a2aPart));
 }
@@ -150,25 +159,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[import_metadata_converter_utils.A2AMetadataKeys.THOUGHT])
   };
 }
 function toGenAIPartFile(a2aPart) {
+  var _a;
+  const part = {};
+  if ((_a = a2aPart.metadata) == null ? void 0 : _a[import_metadata_converter_utils.A2AMetadataKeys.VIDEO_METADATA]) {
+    part.videoMetadata = a2aPart.metadata[import_metadata_converter_utils.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: ${JSON.stringify(a2aPart)}`);
 }
@@ -178,7 +190,7 @@ function toGenAIPartData(a2aPart) {
     throw new Error(`No data in part: ${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[import_metadata_converter_utils.A2AMetadataKeys.DATA_PART_TYPE];
   if (type === "function_call" /* FUNCTION_CALL */) {
     return { functionCall: data };
   }
@@ -202,6 +214,7 @@ function toGenAIPartData(a2aPart) {
   toA2APart,
   toA2AParts,
   toA2ATextPart,
+  toGenAIContent,
   toGenAIPart,
   toGenAIPartData,
   toGenAIPartFile,

package/dist/cjs/agents/base_agent.js

@@ -32,7 +32,7 @@ module.exports = __toCommonJS(base_agent_exports);
 var import_api = require("@opentelemetry/api");
 var import_event = require("../events/event.js");
 var import_tracing = require("../telemetry/tracing.js");
-var import_callback_context = require("./callback_context.js");
+var import_context = require("./context.js");
 var import_invocation_context = require("./invocation_context.js");
 /**
  * @license
@@ -180,7 +180,7 @@ class BaseAgent {
     if (this.beforeAgentCallback.length === 0) {
       return void 0;
     }
-    const callbackContext = new import_callback_context.CallbackContext({ invocationContext });
+    const callbackContext = new import_context.Context({ invocationContext });
     for (const callback of this.beforeAgentCallback) {
       const content = await callback(callbackContext);
       if (content) {
@@ -215,7 +215,7 @@ class BaseAgent {
     if (this.afterAgentCallback.length === 0) {
       return void 0;
     }
-    const callbackContext = new import_callback_context.CallbackContext({ invocationContext });
+    const callbackContext = new import_context.Context({ invocationContext });
     for (const callback of this.afterAgentCallback) {
       const content = await callback(callbackContext);
       if (content) {

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

@@ -22,39 +22,94 @@ var __copyProps = (to, from, except, desc) => {
   return to;
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var tool_context_exports = {};
-__export(tool_context_exports, {
-  ToolContext: () => ToolContext
+var context_exports = {};
+__export(context_exports, {
+  Context: () => Context
 });
-module.exports = __toCommonJS(tool_context_exports);
-var import_callback_context = require("../agents/callback_context.js");
+module.exports = __toCommonJS(context_exports);
 var import_auth_handler = require("../auth/auth_handler.js");
+var import_event_actions = require("../events/event_actions.js");
+var import_state = require("../sessions/state.js");
 var import_tool_confirmation = require("../tools/tool_confirmation.js");
+var import_readonly_context = require("./readonly_context.js");
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class ToolContext extends import_callback_context.CallbackContext {
+class Context extends import_readonly_context.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 || (0, import_event_actions.createEventActions)();
+    this._state = new import_state.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.");
@@ -120,5 +175,5 @@ class ToolContext extends import_callback_context.CallbackContext {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  ToolContext
+  Context
 });