@google/adk 0.4.0 → 0.5.0

package/dist/esm/a2a/a2a_event.js

@@ -0,0 +1,243 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { randomUUID } from "../utils/env_aware_utils.js";
+var MessageRole = /* @__PURE__ */ ((MessageRole2) => {
+  MessageRole2["USER"] = "user";
+  MessageRole2["AGENT"] = "agent";
+  return MessageRole2;
+})(MessageRole || {});
+function isTaskStatusUpdateEvent(event) {
+  return (event == null ? void 0 : event.kind) === "status-update";
+}
+function isTaskArtifactUpdateEvent(event) {
+  return (event == null ? void 0 : event.kind) === "artifact-update";
+}
+function isMessage(event) {
+  return (event == null ? void 0 : event.kind) === "message";
+}
+function isTask(event) {
+  return (event == null ? void 0 : event.kind) === "task";
+}
+function getEventMetadata(event) {
+  if (isTaskArtifactUpdateEvent(event)) {
+    return event.artifact.metadata || {};
+  }
+  if (isTaskStatusUpdateEvent(event) || isTask(event) || isMessage(event)) {
+    return event.metadata || {};
+  }
+  return {};
+}
+function isFailedTaskStatusUpdateEvent(event) {
+  return (isTaskStatusUpdateEvent(event) || isTask(event)) && event.status.state === "failed";
+}
+function isTerminalTaskStatusUpdateEvent(event) {
+  return (isTaskStatusUpdateEvent(event) || isTask(event)) && ["completed", "failed", "canceled", "rejected"].includes(event.status.state);
+}
+function isInputRequiredTaskStatusUpdateEvent(event) {
+  return (isTaskStatusUpdateEvent(event) || isTask(event)) && event.status.state === "input-required";
+}
+function getFailedTaskStatusUpdateEventError(event) {
+  var _a;
+  if (!isFailedTaskStatusUpdateEvent(event)) {
+    return void 0;
+  }
+  const parts = ((_a = event.status.message) == null ? void 0 : _a.parts) || [];
+  if (parts.length === 0) {
+    return void 0;
+  }
+  if (parts[0].kind !== "text") {
+    return void 0;
+  }
+  return parts[0].text;
+}
+function createTaskSubmittedEvent({
+  taskId,
+  contextId,
+  message
+}) {
+  return {
+    kind: "status-update",
+    taskId,
+    contextId,
+    final: false,
+    status: {
+      state: "submitted",
+      message,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    }
+  };
+}
+function createTask({
+  contextId,
+  message,
+  taskId
+}) {
+  return {
+    kind: "task",
+    id: taskId || randomUUID(),
+    contextId,
+    history: [message],
+    status: {
+      state: "submitted",
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    }
+  };
+}
+function createTaskWorkingEvent({
+  taskId,
+  contextId,
+  message
+}) {
+  return {
+    kind: "status-update",
+    taskId,
+    contextId,
+    final: false,
+    status: {
+      state: "working",
+      message,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    }
+  };
+}
+function createTaskCompletedEvent({
+  taskId,
+  contextId,
+  metadata = {}
+}) {
+  return {
+    kind: "status-update",
+    taskId,
+    contextId,
+    final: true,
+    status: {
+      state: "completed",
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    },
+    metadata
+  };
+}
+function createTaskArtifactUpdateEvent({
+  taskId,
+  contextId,
+  artifactId,
+  parts = [],
+  metadata,
+  append,
+  lastChunk
+}) {
+  return {
+    kind: "artifact-update",
+    taskId,
+    contextId,
+    append,
+    lastChunk,
+    artifact: {
+      artifactId: artifactId || randomUUID(),
+      parts,
+      metadata
+    }
+  };
+}
+function createTaskFailedEvent({
+  taskId,
+  contextId,
+  error,
+  metadata
+}) {
+  return {
+    kind: "status-update",
+    taskId,
+    contextId,
+    status: {
+      state: "failed",
+      message: {
+        kind: "message",
+        messageId: randomUUID(),
+        role: "agent",
+        taskId,
+        contextId,
+        parts: [
+          {
+            kind: "text",
+            text: error.message
+          }
+        ]
+      },
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    },
+    metadata,
+    final: true
+  };
+}
+function createTaskInputRequiredEvent({
+  taskId,
+  contextId,
+  parts
+}) {
+  return {
+    kind: "status-update",
+    taskId,
+    contextId,
+    final: true,
+    status: {
+      state: "input-required",
+      message: {
+        kind: "message",
+        messageId: randomUUID(),
+        role: "agent",
+        taskId,
+        contextId,
+        parts
+      },
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    }
+  };
+}
+function createInputMissingErrorEvent({
+  taskId,
+  contextId,
+  parts
+}) {
+  return {
+    kind: "status-update",
+    taskId,
+    contextId,
+    final: true,
+    status: {
+      state: "input-required",
+      message: {
+        kind: "message",
+        messageId: randomUUID(),
+        role: "agent",
+        taskId,
+        contextId,
+        parts
+      },
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    }
+  };
+}
+export {
+  MessageRole,
+  createInputMissingErrorEvent,
+  createTask,
+  createTaskArtifactUpdateEvent,
+  createTaskCompletedEvent,
+  createTaskFailedEvent,
+  createTaskInputRequiredEvent,
+  createTaskSubmittedEvent,
+  createTaskWorkingEvent,
+  getEventMetadata,
+  getFailedTaskStatusUpdateEventError,
+  isFailedTaskStatusUpdateEvent,
+  isInputRequiredTaskStatusUpdateEvent,
+  isMessage,
+  isTask,
+  isTaskArtifactUpdateEvent,
+  isTaskStatusUpdateEvent,
+  isTerminalTaskStatusUpdateEvent
+};

package/dist/esm/a2a/event_converter_utils.js

@@ -0,0 +1,187 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {
+  createModelContent,
+  createUserContent
+} from "@google/genai";
+import { createEvent } from "../events/event.js";
+import { createEventActions } from "../events/event_actions.js";
+import { randomUUID } from "../utils/env_aware_utils.js";
+import {
+  getEventMetadata,
+  getFailedTaskStatusUpdateEventError,
+  isFailedTaskStatusUpdateEvent,
+  isInputRequiredTaskStatusUpdateEvent,
+  isMessage,
+  isTask,
+  isTaskArtifactUpdateEvent,
+  isTaskStatusUpdateEvent,
+  isTerminalTaskStatusUpdateEvent,
+  MessageRole
+} from "./a2a_event.js";
+import {
+  A2AMetadataKeys,
+  getA2AEventMetadata
+} from "./metadata_converter_utils.js";
+import { toA2AParts, toGenAIPart, toGenAIParts } from "./part_converter_utils.js";
+function toA2AMessage(event, {
+  appName,
+  userId,
+  sessionId
+}) {
+  var _a;
+  return {
+    kind: "message",
+    messageId: randomUUID(),
+    role: event.author === MessageRole.USER ? MessageRole.USER : MessageRole.AGENT,
+    parts: toA2AParts(((_a = event.content) == null ? void 0 : _a.parts) || [], event.longRunningToolIds),
+    metadata: getA2AEventMetadata(event, { appName, userId, sessionId })
+  };
+}
+function toAdkEvent(event, invocationId, agentName) {
+  if (isMessage(event)) {
+    return messageToAdkEvent(event, invocationId, agentName);
+  }
+  if (isTask(event)) {
+    return taskToAdkEvent(event, invocationId, agentName);
+  }
+  if (isTaskArtifactUpdateEvent(event)) {
+    return artifactUpdateToAdkEvent(event, invocationId, agentName);
+  }
+  if (isTaskStatusUpdateEvent(event)) {
+    return event.final ? finalTaskStatusUpdateToAdkEvent(event, invocationId, agentName) : taskStatusUpdateToAdkEvent(event, invocationId, agentName);
+  }
+  return void 0;
+}
+function messageToAdkEvent(msg, invocationId, agentName, parentEvent) {
+  const parts = toGenAIParts(msg.parts);
+  return {
+    ...createAdkEventFromMetadata(parentEvent || msg),
+    invocationId,
+    author: msg.role === MessageRole.USER ? MessageRole.USER : agentName,
+    content: msg.role === MessageRole.USER ? createUserContent(parts) : createModelContent(parts),
+    turnComplete: true,
+    partial: false
+  };
+}
+function artifactUpdateToAdkEvent(a2aEvent, invocationId, agentName) {
+  var _a;
+  const partsToConvert = ((_a = a2aEvent.artifact) == null ? void 0 : _a.parts) || [];
+  if (partsToConvert.length === 0) {
+    return void 0;
+  }
+  const partial = !!getEventMetadata(a2aEvent)[A2AMetadataKeys.PARTIAL] || a2aEvent.append || !a2aEvent.lastChunk;
+  return {
+    ...createAdkEventFromMetadata(a2aEvent),
+    invocationId,
+    author: agentName,
+    content: createModelContent(toGenAIParts(partsToConvert)),
+    longRunningToolIds: getLongRunningToolIDs(partsToConvert),
+    partial
+  };
+}
+function finalTaskStatusUpdateToAdkEvent(a2aEvent, invocationId, agentName) {
+  var _a;
+  const partsToConvert = ((_a = a2aEvent.status.message) == null ? void 0 : _a.parts) || [];
+  if (partsToConvert.length === 0) {
+    return void 0;
+  }
+  const parts = toGenAIParts(partsToConvert);
+  const isFailedTask = isFailedTaskStatusUpdateEvent(a2aEvent);
+  const hasContent = !isFailedTask && parts.length > 0;
+  return {
+    ...createAdkEventFromMetadata(a2aEvent),
+    invocationId,
+    author: agentName,
+    errorMessage: isFailedTask ? getFailedTaskStatusUpdateEventError(a2aEvent) : void 0,
+    content: hasContent ? createModelContent(parts) : void 0,
+    longRunningToolIds: getLongRunningToolIDs(partsToConvert),
+    turnComplete: true
+  };
+}
+function taskStatusUpdateToAdkEvent(a2aEvent, invocationId, agentName) {
+  const msg = a2aEvent.status.message;
+  if (!msg) {
+    return void 0;
+  }
+  const parts = toGenAIParts(msg.parts);
+  return {
+    ...createAdkEventFromMetadata(a2aEvent),
+    invocationId,
+    author: agentName,
+    content: createModelContent(parts),
+    turnComplete: false,
+    partial: true
+  };
+}
+function taskToAdkEvent(a2aTask, invocationId, agentName) {
+  var _a, _b;
+  const parts = [];
+  const longRunningToolIds = [];
+  if (a2aTask.artifacts) {
+    for (const artifact of a2aTask.artifacts) {
+      if (((_a = artifact.parts) == null ? void 0 : _a.length) > 0) {
+        const artifactParts = toGenAIParts(artifact.parts);
+        parts.push(...artifactParts);
+        longRunningToolIds.push(...getLongRunningToolIDs(artifact.parts));
+      }
+    }
+  }
+  if ((_b = a2aTask.status) == null ? void 0 : _b.message) {
+    const a2aParts = a2aTask.status.message.parts;
+    const genAIParts = toGenAIParts(a2aParts);
+    parts.push(...genAIParts);
+    longRunningToolIds.push(...getLongRunningToolIDs(a2aParts));
+  }
+  const isTerminal = isTerminalTaskStatusUpdateEvent(a2aTask) || isInputRequiredTaskStatusUpdateEvent(a2aTask);
+  const isFailed = isFailedTaskStatusUpdateEvent(a2aTask);
+  if (parts.length === 0 && !isTerminal) {
+    return void 0;
+  }
+  return {
+    ...createAdkEventFromMetadata(a2aTask),
+    invocationId,
+    author: agentName,
+    content: isFailed ? void 0 : createModelContent(parts),
+    errorMessage: isFailed ? getFailedTaskStatusUpdateEventError(a2aTask) : void 0,
+    longRunningToolIds,
+    turnComplete: isTerminal
+  };
+}
+function createAdkEventFromMetadata(a2aEvent) {
+  const metadata = a2aEvent.metadata || {};
+  return createEvent({
+    branch: metadata[A2AMetadataKeys.BRANCH],
+    author: metadata[A2AMetadataKeys.AUTHOR],
+    partial: metadata[A2AMetadataKeys.PARTIAL],
+    errorCode: metadata[A2AMetadataKeys.ERROR_CODE],
+    errorMessage: metadata[A2AMetadataKeys.ERROR_MESSAGE],
+    citationMetadata: metadata[A2AMetadataKeys.CITATION_METADATA],
+    groundingMetadata: metadata[A2AMetadataKeys.GROUNDING_METADATA],
+    usageMetadata: metadata[A2AMetadataKeys.USAGE_METADATA],
+    customMetadata: metadata[A2AMetadataKeys.CUSTOM_METADATA],
+    actions: createEventActions({
+      escalate: !!metadata[A2AMetadataKeys.ESCALATE],
+      transferToAgent: metadata[A2AMetadataKeys.TRANSFER_TO_AGENT]
+    })
+  });
+}
+function getLongRunningToolIDs(parts) {
+  const ids = [];
+  for (const a2aPart of parts) {
+    if (a2aPart.metadata && a2aPart.metadata[A2AMetadataKeys.IS_LONG_RUNNING]) {
+      const genAIPart = toGenAIPart(a2aPart);
+      if (genAIPart.functionCall && genAIPart.functionCall.id) {
+        ids.push(genAIPart.functionCall.id);
+      }
+    }
+  }
+  return ids;
+}
+export {
+  toA2AMessage,
+  toAdkEvent
+};

package/dist/esm/a2a/executor_context.js

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function createExecutorContext({
+  session,
+  userContent,
+  requestContext
+}) {
+  return {
+    userId: session.userId,
+    sessionId: session.id,
+    agentName: session.appName,
+    readonlyState: session.state,
+    events: session.events,
+    userContent,
+    requestContext
+  };
+}
+export {
+  createExecutorContext
+};

package/dist/esm/a2a/metadata_converter_utils.js

@@ -0,0 +1,90 @@
+/**
+ * @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["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`;
+  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]: 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
+  };
+}
+export {
+  A2AMetadataKeys,
+  AdkMetadataKeys,
+  getA2AEventMetadata,
+  getA2AEventMetadataFromActions,
+  getA2ASessionMetadata,
+  getAdkEventMetadata
+};

package/dist/esm/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,9 +29,9 @@ 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;
@@ -61,31 +60,31 @@ function toA2AFilePart(part) {
   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)}`);
   }
   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 +92,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,7 +115,7 @@ 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) {
@@ -140,7 +143,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[A2AMetadataKeys.DATA_PART_TYPE];
   if (type === "function_call" /* FUNCTION_CALL */) {
     return { functionCall: data };
   }
@@ -163,6 +166,7 @@ export {
   toA2APart,
   toA2AParts,
   toA2ATextPart,
+  toGenAIContent,
   toGenAIPart,
   toGenAIPartData,
   toGenAIPartFile,

package/dist/esm/agents/base_agent.js

@@ -11,7 +11,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) {
@@ -153,7 +153,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) {
@@ -188,7 +188,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) {