@google/adk 0.2.5 → 0.3.0

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

@@ -4,13 +4,18 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
-import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import {
+  StdioClientTransport
+} from "@modelcontextprotocol/sdk/client/stdio.js";
+import {
+  StreamableHTTPClientTransport
+} from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 class MCPSessionManager {
   constructor(connectionParams) {
     this.connectionParams = connectionParams;
   }
   async createSession() {
+    var _a;
     const client = new Client({ name: "MCPClient", version: "1.0.0" });
     switch (this.connectionParams.type) {
       case "StdioConnectionParams":
@@ -18,20 +23,25 @@ class MCPSessionManager {
           new StdioClientTransport(this.connectionParams.serverParams)
         );
         break;
-      case "StreamableHTTPConnectionParams":
-        const transportOptions = this.connectionParams.header ? {
-          requestInit: {
+      case "StreamableHTTPConnectionParams": {
+        const options = (_a = this.connectionParams.transportOptions) != null ? _a : {};
+        if (!options.requestInit && this.connectionParams.header !== void 0) {
+          options.requestInit = {
             headers: this.connectionParams.header
-          }
-        } : void 0;
-        await client.connect(new StreamableHTTPClientTransport(
-          new URL(this.connectionParams.url),
-          transportOptions
-        ));
+          };
+        }
+        await client.connect(
+          new StreamableHTTPClientTransport(
+            new URL(this.connectionParams.url),
+            options
+          )
+        );
         break;
-      default:
+      }
+      default: {
         const _exhaustiveCheck = this.connectionParams;
         break;
+      }
     }
     return client;
   }

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

@@ -12,8 +12,7 @@ class MCPTool extends BaseTool {
     this.mcpSessionManager = mcpSessionManager;
   }
   _getDeclaration() {
-    let declaration;
-    declaration = {
+    return {
       name: this.mcpTool.name,
       description: this.mcpTool.description,
       parameters: toGeminiSchema(this.mcpTool.inputSchema),
@@ -21,7 +20,6 @@ class MCPTool extends BaseTool {
       // https://modelcontextprotocol.io/specification/2025-06-18/server/tools#tool-result
       response: toGeminiSchema(this.mcpTool.outputSchema)
     };
-    return declaration;
   }
   async runAsync(request) {
     const session = await this.mcpSessionManager.createSession();

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

@@ -12,7 +12,7 @@ class MCPToolset extends BaseToolset {
     super(toolFilter);
     this.mcpSessionManager = new MCPSessionManager(connectionParams);
   }
-  async getTools(context) {
+  async getTools() {
     const session = await this.mcpSessionManager.createSession();
     const listResult = await session.listTools();
     logger.debug(`number of tools: ${listResult.tools.length}`);

package/dist/esm/utils/env_aware_utils.js

@@ -36,7 +36,7 @@ function base64Decode(data) {
 function isBase64Encoded(data) {
   try {
     return base64Encode(base64Decode(data)) === data;
-  } catch (e) {
+  } catch (_e) {
     return false;
   }
 }

package/dist/esm/utils/gemini_schema_util.js

@@ -5,9 +5,9 @@
  */
 import { Type } from "@google/genai";
 import { z } from "zod";
-const MCPToolSchema = z.object({
+const MCPToolSchemaObject = z.object({
   type: z.literal("object"),
-  properties: z.record(z.unknown()).optional(),
+  properties: z.record(z.string(), z.unknown()).optional(),
   required: z.string().array().optional()
 });
 function toGeminiType(mcpType) {
@@ -52,12 +52,17 @@ function toGeminiSchema(mcpSchema) {
       }
     }
     const geminiType = toGeminiType(mcp.type);
-    const geminiSchema = { type: geminiType, description: mcp.description };
+    const geminiSchema = {
+      type: geminiType,
+      description: mcp.description
+    };
     if (geminiType === Type.OBJECT) {
       geminiSchema.properties = {};
       if (mcp.properties) {
         for (const name in mcp.properties) {
-          geminiSchema.properties[name] = recursiveConvert(mcp.properties[name]);
+          geminiSchema.properties[name] = recursiveConvert(
+            mcp.properties[name]
+          );
         }
       }
       geminiSchema.required = mcp.required;

package/dist/esm/utils/logger.js

@@ -61,6 +61,18 @@ class SimpleLogger {
     console.error(getColoredPrefix(3 /* ERROR */), ...args);
   }
 }
+class NoOpLogger {
+  log(_level, ..._args) {
+  }
+  debug(..._args) {
+  }
+  info(..._args) {
+  }
+  warn(..._args) {
+  }
+  error(..._args) {
+  }
+}
 const LOG_LEVEL_STR = {
   [0 /* DEBUG */]: "DEBUG",
   [1 /* INFO */]: "INFO",
@@ -81,9 +93,38 @@ const RESET_COLOR = "\x1B[0m";
 function getColoredPrefix(level) {
   return `${CONSOLE_COLOR_MAP[level]}[ADK ${LOG_LEVEL_STR[level]}]:${RESET_COLOR}`;
 }
-const logger = new SimpleLogger();
+let currentLogger = new SimpleLogger();
+function setLogger(customLogger) {
+  currentLogger = customLogger != null ? customLogger : new NoOpLogger();
+}
+function getLogger() {
+  return currentLogger;
+}
+function resetLogger() {
+  currentLogger = new SimpleLogger();
+}
+const logger = {
+  log(level, ...args) {
+    currentLogger.log(level, ...args);
+  },
+  debug(...args) {
+    currentLogger.debug(...args);
+  },
+  info(...args) {
+    currentLogger.info(...args);
+  },
+  warn(...args) {
+    currentLogger.warn(...args);
+  },
+  error(...args) {
+    currentLogger.error(...args);
+  }
+};
 export {
   LogLevel,
+  getLogger,
   logger,
-  setLogLevel
+  resetLogger,
+  setLogLevel,
+  setLogger
 };

package/dist/esm/utils/simple_zod_to_json.js

@@ -4,155 +4,116 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { Type } from "@google/genai";
-import { z } from "zod";
-function isZodObject(obj) {
-  var _a;
-  return obj !== null && typeof obj === "object" && ((_a = obj._def) == null ? void 0 : _a.typeName) === "ZodObject";
+import {
+  zodToJsonSchema as toJSONSchemaV3
+} from "zod-to-json-schema";
+import { toJSONSchema as toJSONSchemaV4 } from "zod/v4";
+function isZodSchema(obj) {
+  return obj !== null && typeof obj === "object" && "parse" in obj && typeof obj.parse === "function" && "safeParse" in obj && typeof obj.safeParse === "function";
+}
+function isZodV3Schema(obj) {
+  return isZodSchema(obj) && !("_zod" in obj);
+}
+function isZodV4Schema(obj) {
+  return isZodSchema(obj) && "_zod" in obj;
 }
-function parseZodType(zodType) {
-  const def = zodType._def;
-  if (!def) {
-    return {};
+function getZodTypeName(schema) {
+  var _a, _b;
+  const schemaAny = schema;
+  if ((_a = schemaAny._def) == null ? void 0 : _a.typeName) {
+    return schemaAny._def.typeName;
   }
-  const description = def.description;
-  const result = {};
-  if (description) result.description = description;
-  const returnResult = (result2) => {
-    if (result2.description === void 0) {
-      delete result2.description;
-    }
-    return result2;
-  };
-  switch (def.typeName) {
-    case z.ZodFirstPartyTypeKind.ZodString:
-      result.type = Type.STRING;
-      for (const check of def.checks || []) {
-        if (check.kind === "min")
-          result.minLength = check.value.toString();
-        else if (check.kind === "max")
-          result.maxLength = check.value.toString();
-        else if (check.kind === "email")
-          result.format = "email";
-        else if (check.kind === "uuid")
-          result.format = "uuid";
-        else if (check.kind === "url")
-          result.format = "uri";
-        else if (check.kind === "regex")
-          result.pattern = check.regex.source;
-      }
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodNumber:
-      result.type = Type.NUMBER;
-      for (const check of def.checks || []) {
-        if (check.kind === "min")
-          result.minimum = check.value;
-        else if (check.kind === "max")
-          result.maximum = check.value;
-        else if (check.kind === "int")
-          result.type = Type.INTEGER;
-      }
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodBoolean:
-      result.type = Type.BOOLEAN;
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodArray:
-      result.type = Type.ARRAY;
-      result.items = parseZodType(def.type);
-      if (def.minLength) result.minItems = def.minLength.value.toString();
-      if (def.maxLength) result.maxItems = def.maxLength.value.toString();
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodObject: {
-      const nestedSchema = zodObjectToSchema(zodType);
-      return nestedSchema;
-    }
-    case z.ZodFirstPartyTypeKind.ZodLiteral:
-      const literalType = typeof def.value;
-      result.enum = [def.value.toString()];
-      if (literalType === "string") {
-        result.type = Type.STRING;
-      } else if (literalType === "number") {
-        result.type = Type.NUMBER;
-      } else if (literalType === "boolean") {
-        result.type = Type.BOOLEAN;
-      } else if (def.value === null) {
-        result.type = Type.NULL;
-      } else {
-        throw new Error(`Unsupported ZodLiteral value type: ${literalType}`);
-      }
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodEnum:
-      result.type = Type.STRING;
-      result.enum = def.values;
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodNativeEnum:
-      result.type = Type.STRING;
-      result.enum = Object.values(def.values);
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodUnion:
-      result.anyOf = def.options.map(parseZodType);
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodOptional:
-      return parseZodType(def.innerType);
-    case z.ZodFirstPartyTypeKind.ZodNullable:
-      const nullableInner = parseZodType(def.innerType);
-      return nullableInner ? returnResult({
-        anyOf: [nullableInner, { type: Type.NULL }],
-        ...description && { description }
-      }) : returnResult({ type: Type.NULL, ...description && { description } });
-    case z.ZodFirstPartyTypeKind.ZodDefault:
-      const defaultInner = parseZodType(def.innerType);
-      if (defaultInner) defaultInner.default = def.defaultValue();
-      return defaultInner;
-    case z.ZodFirstPartyTypeKind.ZodBranded:
-      return parseZodType(def.type);
-    case z.ZodFirstPartyTypeKind.ZodReadonly:
-      return parseZodType(def.innerType);
-    case z.ZodFirstPartyTypeKind.ZodNull:
-      result.type = Type.NULL;
-      return returnResult(result);
-    case z.ZodFirstPartyTypeKind.ZodAny:
-    case z.ZodFirstPartyTypeKind.ZodUnknown:
-      return returnResult({ ...description && { description } });
-    default:
-      throw new Error(`Unsupported Zod type: ${def.typeName}`);
+  const zod4Type = (_b = schemaAny._def) == null ? void 0 : _b.type;
+  if (typeof zod4Type === "string" && zod4Type) {
+    return "Zod" + zod4Type.charAt(0).toUpperCase() + zod4Type.slice(1);
   }
+  return void 0;
+}
+function isZodObject(obj) {
+  return isZodSchema(obj) && getZodTypeName(obj) === "ZodObject";
 }
 function zodObjectToSchema(schema) {
-  if (schema._def.typeName !== z.ZodFirstPartyTypeKind.ZodObject) {
-    throw new Error("Expected a ZodObject");
+  if (!isZodObject(schema)) {
+    throw new Error("Expected a Zod Object");
   }
-  const shape = schema.shape;
-  const properties = {};
-  const required = [];
-  for (const key in shape) {
-    const fieldSchema = shape[key];
-    const parsedField = parseZodType(fieldSchema);
-    if (parsedField) {
-      properties[key] = parsedField;
-    }
-    let currentSchema = fieldSchema;
-    let isOptional = false;
-    while (currentSchema._def.typeName === z.ZodFirstPartyTypeKind.ZodOptional || currentSchema._def.typeName === z.ZodFirstPartyTypeKind.ZodDefault) {
-      isOptional = true;
-      currentSchema = currentSchema._def.innerType;
-    }
-    if (!isOptional) {
-      required.push(key);
-    }
+  if (isZodV4Schema(schema)) {
+    return toJSONSchemaV4(schema, {
+      target: "openapi-3.0",
+      io: "input",
+      override: (ctx) => {
+        var _a;
+        const { jsonSchema } = ctx;
+        if (jsonSchema.additionalProperties !== void 0) {
+          delete jsonSchema.additionalProperties;
+        }
+        if (jsonSchema.readOnly !== void 0) {
+          delete jsonSchema.readOnly;
+        }
+        if (jsonSchema.maxItems !== void 0) {
+          jsonSchema.maxItems = jsonSchema.maxItems.toString();
+        }
+        if (jsonSchema.format === "email" || jsonSchema.format === "uuid") {
+          delete jsonSchema.pattern;
+        }
+        if (jsonSchema.minItems !== void 0) {
+          jsonSchema.minItems = jsonSchema.minItems.toString();
+        }
+        if (jsonSchema.minLength !== void 0) {
+          jsonSchema.minLength = jsonSchema.minLength.toString();
+        }
+        if (jsonSchema.maxLength !== void 0) {
+          jsonSchema.maxLength = jsonSchema.maxLength.toString();
+        }
+        if (((_a = jsonSchema.enum) == null ? void 0 : _a.length) === 1 && jsonSchema.enum[0] === null) {
+          jsonSchema.type = Type.NULL;
+          delete jsonSchema.enum;
+        }
+        if (jsonSchema.type !== void 0) {
+          jsonSchema.type = jsonSchema.type.toUpperCase();
+        }
+      }
+    });
   }
-  const catchall = schema._def.catchall;
-  let additionalProperties = false;
-  if (catchall && catchall._def.typeName !== z.ZodFirstPartyTypeKind.ZodNever) {
-    additionalProperties = parseZodType(catchall) || true;
-  } else {
-    additionalProperties = schema._def.unknownKeys === "passthrough";
+  if (isZodV3Schema(schema)) {
+    return toJSONSchemaV3(schema, {
+      target: "openApi3",
+      emailStrategy: "format:email",
+      postProcess: (jsonSchema) => {
+        var _a, _b, _c, _d, _e, _f, _g;
+        if (!jsonSchema) {
+          return;
+        }
+        if (jsonSchema.additionalProperties !== void 0) {
+          delete jsonSchema.additionalProperties;
+        }
+        if (jsonSchema.maxItems !== void 0) {
+          jsonSchema.maxItems = (_a = jsonSchema.maxItems) == null ? void 0 : _a.toString();
+        }
+        if (jsonSchema.minItems !== void 0) {
+          jsonSchema.minItems = (_b = jsonSchema.minItems) == null ? void 0 : _b.toString();
+        }
+        if (jsonSchema.minLength !== void 0) {
+          jsonSchema.minLength = (_c = jsonSchema.minLength) == null ? void 0 : _c.toString();
+        }
+        if (jsonSchema.maxLength !== void 0) {
+          jsonSchema.maxLength = (_d = jsonSchema.maxLength) == null ? void 0 : _d.toString();
+        }
+        if (((_e = jsonSchema.enum) == null ? void 0 : _e.length) === 1 && jsonSchema.enum[0] === "null") {
+          jsonSchema.type = Type.NULL;
+          delete jsonSchema.enum;
+        }
+        if (jsonSchema.type === "integer" && jsonSchema.format !== "int64") {
+          (_f = jsonSchema.minimum) != null ? _f : jsonSchema.minimum = Number.MIN_SAFE_INTEGER;
+          (_g = jsonSchema.maximum) != null ? _g : jsonSchema.maximum = Number.MAX_SAFE_INTEGER;
+        }
+        if (jsonSchema.type !== void 0) {
+          jsonSchema.type = jsonSchema.type.toUpperCase();
+        }
+        return jsonSchema;
+      }
+    });
   }
-  return {
-    type: Type.OBJECT,
-    properties,
-    required: required.length > 0 ? required : [],
-    ...schema._def.description ? { description: schema._def.description } : {}
-  };
+  throw new Error("Unsupported Zod schema version.");
 }
 export {
   isZodObject,

package/dist/esm/utils/variant_utils.js

@@ -16,7 +16,7 @@ function getBooleanEnvVar(envVar) {
     return false;
   }
   const envVarValue = (process.env[envVar] || "").toLowerCase();
-  return ["true", "1"].includes(envVar.toLowerCase());
+  return ["true", "1"].includes(envVarValue);
 }
 export {
   GoogleLLMVariant,

package/dist/esm/version.js

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

package/dist/types/agents/base_agent.d.ts

@@ -66,8 +66,9 @@ export declare abstract class BaseAgent {
     readonly description?: string;
     /**
      * Root agent of this agent.
+     * Computed dynamically by traversing up the parent chain.
      */
-    readonly rootAgent: BaseAgent;
+    get rootAgent(): BaseAgent;
     /**
      * The parent agent of this agent.
      *

package/dist/types/agents/callback_context.d.ts

@@ -14,7 +14,7 @@ import { ReadonlyContext } from './readonly_context.js';
 export declare class CallbackContext extends ReadonlyContext {
     private readonly _state;
     readonly eventActions: EventActions;
-    constructor({ invocationContext, eventActions }: {
+    constructor({ invocationContext, eventActions, }: {
         invocationContext: InvocationContext;
         eventActions?: EventActions;
     });

package/dist/types/agents/llm_agent.d.ts

@@ -207,7 +207,7 @@ export interface LlmAgentConfig extends BaseAgentConfig {
 }
 declare class RequestConfirmationLlmRequestProcessor extends BaseLlmRequestProcessor {
     /** Handles tool confirmation information to build the LLM request. */
-    runAsync(invocationContext: InvocationContext, llmRequest: LlmRequest): AsyncGenerator<Event, void, void>;
+    runAsync(invocationContext: InvocationContext): AsyncGenerator<Event, void, void>;
 }
 export declare const REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR: RequestConfirmationLlmRequestProcessor;
 /**

package/dist/types/agents/loop_agent.d.ts

@@ -42,6 +42,6 @@ export declare class LoopAgent extends BaseAgent {
     private readonly maxIterations;
     constructor(config: LoopAgentConfig);
     protected runAsyncImpl(context: InvocationContext): AsyncGenerator<Event, void, void>;
-    protected runLiveImpl(context: InvocationContext): AsyncGenerator<Event, void, void>;
+    protected runLiveImpl(_context: InvocationContext): AsyncGenerator<Event, void, void>;
 }
 export {};

package/dist/types/agents/parallel_agent.d.ts

@@ -32,6 +32,6 @@ export declare class ParallelAgent extends BaseAgent {
      */
     readonly [PARALLEL_AGENT_SIGNATURE_SYMBOL] = true;
     protected runAsyncImpl(context: InvocationContext): AsyncGenerator<Event, void, void>;
-    protected runLiveImpl(context: InvocationContext): AsyncGenerator<Event, void, void>;
+    protected runLiveImpl(_context: InvocationContext): AsyncGenerator<Event, void, void>;
 }
 export {};

package/dist/types/artifacts/in_memory_artifact_service.d.ts

@@ -12,7 +12,7 @@ export declare class InMemoryArtifactService implements BaseArtifactService {
     private readonly artifacts;
     saveArtifact({ appName, userId, sessionId, filename, artifact, }: SaveArtifactRequest): Promise<number>;
     loadArtifact({ appName, userId, sessionId, filename, version, }: LoadArtifactRequest): Promise<Part | undefined>;
-    listArtifactKeys({ appName, userId, sessionId }: ListArtifactKeysRequest): Promise<string[]>;
-    deleteArtifact({ appName, userId, sessionId, filename }: DeleteArtifactRequest): Promise<void>;
-    listVersions({ appName, userId, sessionId, filename }: ListVersionsRequest): Promise<number[]>;
+    listArtifactKeys({ appName, userId, sessionId, }: ListArtifactKeysRequest): Promise<string[]>;
+    deleteArtifact({ appName, userId, sessionId, filename, }: DeleteArtifactRequest): Promise<void>;
+    listVersions({ appName, userId, sessionId, filename, }: ListVersionsRequest): Promise<number[]>;
 }

package/dist/types/code_executors/built_in_code_executor.d.ts

@@ -26,7 +26,7 @@ export declare function isBuiltInCodeExecutor(obj: unknown): obj is BuiltInCodeE
 export declare class BuiltInCodeExecutor extends BaseCodeExecutor {
     /** A unique symbol to identify BuiltInCodeExecutor class. */
     readonly [BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL] = true;
-    executeCode(params: ExecuteCodeParams): Promise<CodeExecutionResult>;
+    executeCode(_params: ExecuteCodeParams): Promise<CodeExecutionResult>;
     processLlmRequest(llmRequest: LlmRequest): void;
 }
 export {};

package/dist/types/code_executors/code_executor_context.d.ts

@@ -78,10 +78,8 @@ export declare class CodeExecutorContext {
      */
     updateCodeExecutionResult({ invocationId, code, resultStdout, resultStderr, }: UpdateCodeExecutionResultParams): void;
     /**
-     * Gets the code executor context from the session state.
-     * @param invocationId The session state to get the code executor context
-     *     from.
+     * Gets the code execution context from the session state.
      * @return The code execution context for the given invocation ID.
      */
-    getCodeExecutionContext(invocationId: string): Record<string, unknown>;
+    getCodeExecutionContext(): Record<string, unknown>;
 }

package/dist/types/common.d.ts

@@ -73,7 +73,8 @@ export { GOOGLE_SEARCH, GoogleSearchTool } from './tools/google_search_tool.js';
 export { LongRunningFunctionTool } from './tools/long_running_tool.js';
 export { ToolConfirmation } from './tools/tool_confirmation.js';
 export { ToolContext } from './tools/tool_context.js';
-export { LogLevel, setLogLevel } from './utils/logger.js';
+export { LogLevel, getLogger, setLogLevel, setLogger } from './utils/logger.js';
+export type { Logger } from './utils/logger.js';
 export { isGemini2OrAbove } from './utils/model_name.js';
 export { zodObjectToSchema } from './utils/simple_zod_to_json.js';
 export { GoogleLLMVariant } from './utils/variant_utils.js';

package/dist/types/index.d.ts

@@ -3,10 +3,10 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+export * from './artifacts/gcs_artifact_service.js';
 export * from './common.js';
+export * from './telemetry/google_cloud.js';
+export * from './telemetry/setup.js';
 export * from './tools/mcp/mcp_session_manager.js';
 export * from './tools/mcp/mcp_tool.js';
 export * from './tools/mcp/mcp_toolset.js';
-export * from './artifacts/gcs_artifact_service.js';
-export * from './telemetry/setup.js';
-export * from './telemetry/google_cloud.js';

package/dist/types/models/base_llm_connection.d.ts

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { Content, Blob } from '@google/genai';
+import { Blob, Content } from '@google/genai';
 import { LlmResponse } from './llm_response.js';
 /**
  * The base class for a live model connection.

package/dist/types/models/llm_response.d.ts

@@ -46,7 +46,7 @@ export interface LlmResponse {
      * NOTE: the entire object must be JSON serializable.
      */
     customMetadata?: {
-        [key: string]: any;
+        [key: string]: unknown;
     };
     /**
      * The usage metadata of the LlmResponse.