@google/adk 0.3.0 → 0.5.0

package/dist/esm/sessions/in_memory_session_service.js

@@ -7,10 +7,14 @@ import { cloneDeep } from "lodash-es";
 import { randomUUID } from "../utils/env_aware_utils.js";
 import { logger } from "../utils/logger.js";
 import {
-  BaseSessionService
+  BaseSessionService,
+  mergeStates
 } from "./base_session_service.js";
 import { createSession } from "./session.js";
 import { State } from "./state.js";
+function isInMemoryConnectionString(uri) {
+  return uri === "memory://";
+}
 class InMemorySessionService extends BaseSessionService {
   constructor() {
     super(...arguments);
@@ -28,12 +32,13 @@ class InMemorySessionService extends BaseSessionService {
      */
     this.appState = {};
   }
-  createSession({
+  async createSession({
     appName,
     userId,
     state,
     sessionId
   }) {
+    var _a;
     const session = createSession({
       id: sessionId || randomUUID(),
       appName,
@@ -49,16 +54,21 @@ class InMemorySessionService extends BaseSessionService {
       this.sessions[appName][userId] = {};
     }
     this.sessions[appName][userId][session.id] = session;
-    return Promise.resolve(
-      this.mergeState(appName, userId, cloneDeep(session))
+    const copiedSession = cloneDeep(session);
+    copiedSession.state = mergeStates(
+      this.appState[appName],
+      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
+      copiedSession.state
     );
+    return copiedSession;
   }
-  getSession({
+  async getSession({
     appName,
     userId,
     sessionId,
     config
   }) {
+    var _a;
     if (!this.sessions[appName] || !this.sessions[appName][userId] || !this.sessions[appName][userId][sessionId]) {
       return Promise.resolve(void 0);
     }
@@ -83,7 +93,12 @@ class InMemorySessionService extends BaseSessionService {
         }
       }
     }
-    return Promise.resolve(this.mergeState(appName, userId, copiedSession));
+    copiedSession.state = mergeStates(
+      this.appState[appName],
+      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
+      copiedSession.state
+    );
+    return copiedSession;
   }
   listSessions({
     appName,
@@ -160,21 +175,8 @@ class InMemorySessionService extends BaseSessionService {
     storageSession.lastUpdateTime = event.timestamp;
     return event;
   }
-  mergeState(appName, userId, copiedSession) {
-    if (this.appState[appName]) {
-      for (const key of Object.keys(this.appState[appName])) {
-        copiedSession.state[State.APP_PREFIX + key] = this.appState[appName][key];
-      }
-    }
-    if (!this.userState[appName] || !this.userState[appName][userId]) {
-      return copiedSession;
-    }
-    for (const key of Object.keys(this.userState[appName][userId])) {
-      copiedSession.state[State.USER_PREFIX + key] = this.userState[appName][userId][key];
-    }
-    return copiedSession;
-  }
 }
 export {
-  InMemorySessionService
+  InMemorySessionService,
+  isInMemoryConnectionString
 };

package/dist/esm/sessions/registry.js

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {
+  DatabaseSessionService,
+  isDatabaseConnectionString
+} from "./database_session_service.js";
+import {
+  InMemorySessionService,
+  isInMemoryConnectionString
+} from "./in_memory_session_service.js";
+function getSessionServiceFromUri(uri) {
+  if (isInMemoryConnectionString(uri)) {
+    return new InMemorySessionService();
+  }
+  if (isDatabaseConnectionString(uri)) {
+    return new DatabaseSessionService(uri);
+  }
+  throw new Error(`Unsupported session service URI: ${uri}`);
+}
+export {
+  getSessionServiceFromUri
+};

package/dist/esm/tools/base_tool.js

@@ -54,6 +54,9 @@ class BaseTool {
     if (!functionDeclaration) {
       return;
     }
+    if (this.name in llmRequest.toolsDict) {
+      throw new Error(`Duplicate tool name: ${this.name}`);
+    }
     llmRequest.toolsDict[this.name] = this;
     const tool = findToolWithFunctionDeclarations(llmRequest);
     if (tool) {

package/dist/esm/tools/base_toolset.js

@@ -3,9 +3,17 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+var _a;
+const BASE_TOOLSET_SIGNATURE_SYMBOL = Symbol.for("google.adk.baseToolset");
+function isBaseToolset(obj) {
+  return typeof obj === "object" && obj !== null && BASE_TOOLSET_SIGNATURE_SYMBOL in obj && obj[BASE_TOOLSET_SIGNATURE_SYMBOL] === true;
+}
+_a = BASE_TOOLSET_SIGNATURE_SYMBOL;
 class BaseToolset {
-  constructor(toolFilter) {
+  constructor(toolFilter, prefix) {
     this.toolFilter = toolFilter;
+    this.prefix = prefix;
+    this[_a] = true;
   }
   /**
    * Returns whether the tool should be exposed to LLM.
@@ -42,5 +50,6 @@ class BaseToolset {
   }
 }
 export {
-  BaseToolset
+  BaseToolset,
+  isBaseToolset
 };

package/dist/esm/tools/exit_loop_tool.js

@@ -0,0 +1,32 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseTool } from "./base_tool.js";
+class ExitLoopTool extends BaseTool {
+  constructor() {
+    super({
+      name: "exit_loop",
+      description: "Exits the loop.\n\nCall this function only when you are instructed to do so."
+    });
+  }
+  _getDeclaration() {
+    return {
+      name: this.name,
+      description: this.description
+    };
+  }
+  async runAsync({
+    toolContext
+  }) {
+    toolContext.actions.escalate = true;
+    toolContext.actions.skipSummarization = true;
+    return "";
+  }
+}
+const EXIT_LOOP = new ExitLoopTool();
+export {
+  EXIT_LOOP,
+  ExitLoopTool
+};

package/dist/esm/tools/forwarding_artifact_service.js

@@ -35,6 +35,22 @@ class ForwardingArtifactService {
       request
     );
   }
+  listArtifactVersions(request) {
+    if (!this.toolContext.invocationContext.artifactService) {
+      throw new Error("Artifact service is not initialized.");
+    }
+    return this.toolContext.invocationContext.artifactService.listArtifactVersions(
+      request
+    );
+  }
+  getArtifactVersion(request) {
+    if (!this.toolContext.invocationContext.artifactService) {
+      throw new Error("Artifact service is not initialized.");
+    }
+    return this.toolContext.invocationContext.artifactService.getArtifactVersion(
+      request
+    );
+  }
 }
 export {
   ForwardingArtifactService

package/dist/esm/tools/mcp/mcp_toolset.js

@@ -8,8 +8,8 @@ import { BaseToolset } from "../base_toolset.js";
 import { MCPSessionManager } from "./mcp_session_manager.js";
 import { MCPTool } from "./mcp_tool.js";
 class MCPToolset extends BaseToolset {
-  constructor(connectionParams, toolFilter = []) {
-    super(toolFilter);
+  constructor(connectionParams, toolFilter = [], prefix) {
+    super(toolFilter, prefix);
     this.mcpSessionManager = new MCPSessionManager(connectionParams);
   }
   async getTools() {
@@ -19,9 +19,13 @@ class MCPToolset extends BaseToolset {
     for (const tool of listResult.tools) {
       logger.debug(`tool: ${tool.name}`);
     }
-    return listResult.tools.map(
-      (tool) => new MCPTool(tool, this.mcpSessionManager)
-    );
+    return listResult.tools.map((tool) => {
+      const toolWithPrefix = {
+        ...tool,
+        name: this.prefix ? `${this.prefix}_${tool.name}` : tool.name
+      };
+      return new MCPTool(toolWithPrefix, this.mcpSessionManager);
+    });
   }
   async close() {
   }

package/dist/esm/utils/env_aware_utils.js

@@ -40,9 +40,17 @@ function isBase64Encoded(data) {
     return false;
   }
 }
+function getBooleanEnvVar(envVar) {
+  if (!process.env) {
+    return false;
+  }
+  const envVarValue = (process.env[envVar] || "").toLowerCase();
+  return ["true", "1"].includes(envVarValue);
+}
 export {
   base64Decode,
   base64Encode,
+  getBooleanEnvVar,
   isBase64Encoded,
   isBrowser,
   randomUUID

package/dist/esm/utils/logger.js

@@ -3,6 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import * as winston from "winston";
 var LogLevel = /* @__PURE__ */ ((LogLevel2) => {
   LogLevel2[LogLevel2["DEBUG"] = 0] = "DEBUG";
   LogLevel2[LogLevel2["INFO"] = 1] = "INFO";
@@ -10,58 +11,68 @@ var LogLevel = /* @__PURE__ */ ((LogLevel2) => {
   LogLevel2[LogLevel2["ERROR"] = 3] = "ERROR";
   return LogLevel2;
 })(LogLevel || {});
-let logLevel = 1 /* INFO */;
-function setLogLevel(level) {
-  logLevel = level;
-}
 class SimpleLogger {
-  log(level, ...args) {
-    if (level < logLevel) {
+  constructor() {
+    this.logLevel = 1 /* INFO */;
+    this.logger = winston.createLogger({
+      levels: {
+        "debug": 0 /* DEBUG */,
+        "info": 1 /* INFO */,
+        "warn": 2 /* WARN */,
+        "error": 3 /* ERROR */
+      },
+      format: winston.format.combine(
+        winston.format.label({ label: "ADK" }),
+        winston.format((info) => {
+          info.level = info.level.toUpperCase();
+          return info;
+        })(),
+        winston.format.colorize(),
+        winston.format.timestamp(),
+        winston.format.printf((info) => {
+          return `${info.level}: [${info.label}] ${info.timestamp} ${info.message}`;
+        })
+      ),
+      transports: [new winston.transports.Console()]
+    });
+  }
+  setLogLevel(level) {
+    this.logLevel = level;
+  }
+  log(level, ...messages) {
+    if (this.logLevel > level) {
       return;
     }
-    switch (level) {
-      case 0 /* DEBUG */:
-        this.debug(...args);
-        break;
-      case 1 /* INFO */:
-        this.info(...args);
-        break;
-      case 2 /* WARN */:
-        this.warn(...args);
-        break;
-      case 3 /* ERROR */:
-        this.error(...args);
-        break;
-      default:
-        throw new Error(`Unsupported log level: ${level}`);
-    }
+    this.logger.log(level.toString(), messages.join(" "));
   }
-  debug(...args) {
-    if (logLevel > 0 /* DEBUG */) {
+  debug(...messages) {
+    if (this.logLevel > 0 /* DEBUG */) {
       return;
     }
-    console.debug(getColoredPrefix(0 /* DEBUG */), ...args);
+    this.logger.debug(messages.join(" "));
   }
-  info(...args) {
-    if (logLevel > 1 /* INFO */) {
+  info(...messages) {
+    if (this.logLevel > 1 /* INFO */) {
       return;
     }
-    console.info(getColoredPrefix(1 /* INFO */), ...args);
+    this.logger.info(messages.join(" "));
   }
-  warn(...args) {
-    if (logLevel > 2 /* WARN */) {
+  warn(...messages) {
+    if (this.logLevel > 2 /* WARN */) {
       return;
     }
-    console.warn(getColoredPrefix(2 /* WARN */), ...args);
+    this.logger.warn(messages.join(" "));
   }
-  error(...args) {
-    if (logLevel > 3 /* ERROR */) {
+  error(...messages) {
+    if (this.logLevel > 3 /* ERROR */) {
       return;
     }
-    console.error(getColoredPrefix(3 /* ERROR */), ...args);
+    this.logger.error(messages.join(" "));
   }
 }
 class NoOpLogger {
+  setLogLevel(_level) {
+  }
   log(_level, ..._args) {
   }
   debug(..._args) {
@@ -73,26 +84,6 @@ class NoOpLogger {
   error(..._args) {
   }
 }
-const LOG_LEVEL_STR = {
-  [0 /* DEBUG */]: "DEBUG",
-  [1 /* INFO */]: "INFO",
-  [2 /* WARN */]: "WARN",
-  [3 /* ERROR */]: "ERROR"
-};
-const CONSOLE_COLOR_MAP = {
-  [0 /* DEBUG */]: "\x1B[34m",
-  // Blue
-  [1 /* INFO */]: "\x1B[32m",
-  // Green
-  [2 /* WARN */]: "\x1B[33m",
-  // Yellow
-  [3 /* ERROR */]: "\x1B[31m"
-  // Red
-};
-const RESET_COLOR = "\x1B[0m";
-function getColoredPrefix(level) {
-  return `${CONSOLE_COLOR_MAP[level]}[ADK ${LOG_LEVEL_STR[level]}]:${RESET_COLOR}`;
-}
 let currentLogger = new SimpleLogger();
 function setLogger(customLogger) {
   currentLogger = customLogger != null ? customLogger : new NoOpLogger();
@@ -103,7 +94,13 @@ function getLogger() {
 function resetLogger() {
   currentLogger = new SimpleLogger();
 }
+function setLogLevel(level) {
+  logger.setLogLevel(level);
+}
 const logger = {
+  setLogLevel(level) {
+    currentLogger.setLogLevel(level);
+  },
   log(level, ...args) {
     currentLogger.log(level, ...args);
   },

package/dist/esm/utils/object_notation_utils.js

@@ -0,0 +1,47 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function toCamelCase(obj, preserveKeys = []) {
+  return toNotation(obj, toCamelCaseKey, "", preserveKeys);
+}
+function toSnakeCase(obj, preserveKeys = []) {
+  return toNotation(obj, toSnakeCaseKey, "", preserveKeys);
+}
+const toCamelCaseKey = (key) => key.replace(
+  /_([a-z])/g,
+  (_match, letter) => letter.toUpperCase()
+);
+const toSnakeCaseKey = (key) => key.replace(/[A-Z]/g, (g) => "_" + g.toLowerCase());
+function toNotation(obj, converter, parentKey = "", preserveKeys = []) {
+  if (Array.isArray(obj)) {
+    return obj.map(
+      (item) => toNotation(item, converter, parentKey, preserveKeys)
+    );
+  }
+  if (typeof obj === "object" && obj !== null) {
+    const source = obj;
+    const result = {};
+    for (const key of Object.keys(source)) {
+      const convertedKey = converter(key);
+      const fullPath = parentKey !== "" ? parentKey + "." + key : key;
+      if (preserveKeys.includes(fullPath)) {
+        result[convertedKey] = source[key];
+      } else {
+        result[convertedKey] = toNotation(
+          source[key],
+          converter,
+          fullPath,
+          preserveKeys
+        );
+      }
+    }
+    return result;
+  }
+  return obj;
+}
+export {
+  toCamelCase,
+  toSnakeCase
+};

package/dist/esm/utils/variant_utils.js

@@ -3,6 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { getBooleanEnvVar } from "./env_aware_utils.js";
 var GoogleLLMVariant = /* @__PURE__ */ ((GoogleLLMVariant2) => {
   GoogleLLMVariant2["VERTEX_AI"] = "VERTEX_AI";
   GoogleLLMVariant2["GEMINI_API"] = "GEMINI_API";
@@ -11,13 +12,6 @@ var GoogleLLMVariant = /* @__PURE__ */ ((GoogleLLMVariant2) => {
 function getGoogleLlmVariant() {
   return getBooleanEnvVar("GOOGLE_GENAI_USE_VERTEXAI") ? "VERTEX_AI" /* VERTEX_AI */ : "GEMINI_API" /* GEMINI_API */;
 }
-function getBooleanEnvVar(envVar) {
-  if (!process.env) {
-    return false;
-  }
-  const envVarValue = (process.env[envVar] || "").toLowerCase();
-  return ["true", "1"].includes(envVarValue);
-}
 export {
   GoogleLLMVariant,
   getGoogleLlmVariant

package/dist/esm/version.js

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const version = "0.3.0";
+const version = "0.5.0";
 export {
   version
 };

package/dist/types/a2a/a2a_event.d.ts

@@ -0,0 +1,122 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Part as A2APart, Message, Task, TaskArtifactUpdateEvent, TaskStatusUpdateEvent } from '@a2a-js/sdk';
+/**
+ * Message roles.
+ */
+export declare enum MessageRole {
+    USER = "user",
+    AGENT = "agent"
+}
+/**
+ * A2A event.
+ */
+export type A2AEvent = Task | Message | TaskStatusUpdateEvent | TaskArtifactUpdateEvent;
+/**
+ * Checks if the event is an A2A TaskStatusUpdateEvent.
+ */
+export declare function isTaskStatusUpdateEvent(event: unknown): event is TaskStatusUpdateEvent;
+/**
+ * Checks if the event is an A2A TaskArtifactUpdateEvent.
+ */
+export declare function isTaskArtifactUpdateEvent(event: unknown): event is TaskArtifactUpdateEvent;
+/**
+ * Checks if the event is an A2A Message.
+ */
+export declare function isMessage(event: unknown): event is Message;
+/**
+ * Checks if the event is an A2A Task.
+ */
+export declare function isTask(event: unknown): event is Task;
+/**
+ * Gets the metadata from an A2A event.
+ */
+export declare function getEventMetadata(event: A2AEvent): Record<string, unknown>;
+/**
+ * Checks if the event is a failed task status update event.
+ */
+export declare function isFailedTaskStatusUpdateEvent(event: unknown): boolean;
+/**
+ * Checks if the event is a terminal task status update event.
+ */
+export declare function isTerminalTaskStatusUpdateEvent(event: unknown): boolean;
+/**
+ * Checks if the event is an input required task status update event.
+ */
+export declare function isInputRequiredTaskStatusUpdateEvent(event: unknown): boolean;
+/**
+ * Gets the error message from a failed task status update event.
+ */
+export declare function getFailedTaskStatusUpdateEventError(event: TaskStatusUpdateEvent | Task): string | undefined;
+/**
+ * Creates a task submitted event.
+ */
+export declare function createTaskSubmittedEvent({ taskId, contextId, message, }: {
+    taskId: string;
+    contextId: string;
+    message: Message;
+}): TaskStatusUpdateEvent;
+/**
+ * Creates a task with submitted status.
+ */
+export declare function createTask({ contextId, message, taskId, }: {
+    taskId: string;
+    contextId: string;
+    message: Message;
+}): Task;
+/**
+ * Creates a task working event.
+ */
+export declare function createTaskWorkingEvent({ taskId, contextId, message, }: {
+    taskId: string;
+    contextId: string;
+    message: Message;
+}): TaskStatusUpdateEvent;
+/**
+ * Creates a task completed event.
+ */
+export declare function createTaskCompletedEvent({ taskId, contextId, metadata, }: {
+    taskId: string;
+    contextId: string;
+    metadata?: Record<string, unknown>;
+}): TaskStatusUpdateEvent;
+/**
+ * Creates an artifact update event.
+ */
+export declare function createTaskArtifactUpdateEvent({ taskId, contextId, artifactId, parts, metadata, append, lastChunk, }: {
+    taskId: string;
+    contextId: string;
+    artifactId?: string;
+    parts?: A2APart[];
+    metadata?: Record<string, unknown>;
+    append?: boolean;
+    lastChunk?: boolean;
+}): TaskArtifactUpdateEvent;
+/**
+ * Creates an error message for task execution failure.
+ */
+export declare function createTaskFailedEvent({ taskId, contextId, error, metadata, }: {
+    taskId: string;
+    contextId: string;
+    error: Error;
+    metadata?: Record<string, unknown>;
+}): TaskStatusUpdateEvent;
+/**
+ * Creates an input required event.
+ */
+export declare function createTaskInputRequiredEvent({ taskId, contextId, parts, }: {
+    taskId: string;
+    contextId: string;
+    parts: A2APart[];
+}): TaskStatusUpdateEvent;
+/**
+ * Creates an error message for missing input for a function call.
+ */
+export declare function createInputMissingErrorEvent({ taskId, contextId, parts, }: {
+    parts: A2APart[];
+    taskId: string;
+    contextId: string;
+}): TaskStatusUpdateEvent;

package/dist/types/a2a/event_converter_utils.d.ts

@@ -0,0 +1,20 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Message } from '@a2a-js/sdk';
+import { Event as AdkEvent } from '../events/event.js';
+import { A2AEvent } from './a2a_event.js';
+/**
+ * Converts a session Event to an A2A Message.
+ */
+export declare function toA2AMessage(event: AdkEvent, { appName, userId, sessionId, }: {
+    appName: string;
+    userId: string;
+    sessionId: string;
+}): Message;
+/**
+ * Converts an A2A Event to a Session Event.
+ */
+export declare function toAdkEvent(event: A2AEvent, invocationId: string, agentName: string): AdkEvent | undefined;

package/dist/types/a2a/executor_context.d.ts

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { RequestContext } from '@a2a-js/sdk/server';
+import { Content } from '@google/genai';
+import { Event } from '../events/event.js';
+import { Session } from '../sessions/session.js';
+/**
+ * The A2A Agent Executor context.
+ */
+export interface ExecutorContext {
+    userId: string;
+    sessionId: string;
+    agentName: string;
+    readonlyState: Record<string, unknown>;
+    events: Event[];
+    userContent: Content;
+    requestContext: RequestContext;
+}
+/**
+ * Creates an A2A Agent Executor context from the given parameters.
+ * @param session The session.
+ * @param userContent The content of the user.
+ * @param requestContext The request context.
+ * @returns The A2A Agent Executor context.
+ */
+export declare function createExecutorContext({ session, userContent, requestContext, }: {
+    session: Session;
+    userContent: Content;
+    requestContext: RequestContext;
+}): ExecutorContext;