@google/adk 0.2.5 → 0.4.0

package/dist/esm/tools/base_tool.js

@@ -49,7 +49,7 @@ class BaseTool {
    *
    * @param request The request to process the LLM request.
    */
-  async processLlmRequest({ toolContext, llmRequest }) {
+  async processLlmRequest({ llmRequest }) {
     const functionDeclaration = this._getDeclaration();
     if (!functionDeclaration) {
       return;
@@ -78,7 +78,9 @@ class BaseTool {
 }
 function findToolWithFunctionDeclarations(llmRequest) {
   var _a2;
-  return (((_a2 = llmRequest.config) == null ? void 0 : _a2.tools) || []).find((tool) => "functionDeclarations" in tool);
+  return (((_a2 = llmRequest.config) == null ? void 0 : _a2.tools) || []).find(
+    (tool) => "functionDeclarations" in tool
+  );
 }
 export {
   BaseTool,

package/dist/esm/tools/forwarding_artifact_service.js

@@ -16,7 +16,7 @@ class ForwardingArtifactService {
   async loadArtifact(request) {
     return this.toolContext.loadArtifact(request.filename, request.version);
   }
-  async listArtifactKeys(request) {
+  async listArtifactKeys() {
     return this.toolContext.listArtifacts();
   }
   async deleteArtifact(request) {
@@ -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/function_tool.js

@@ -5,7 +5,6 @@
  */
 var _a, _b;
 import { Type } from "@google/genai";
-import { ZodObject } from "zod";
 import { isZodObject, zodObjectToSchema } from "../utils/simple_zod_to_json.js";
 import { BaseTool } from "./base_tool.js";
 function toSchema(parameters) {
@@ -60,7 +59,7 @@ class FunctionTool extends (_b = BaseTool, _a = FUNCTION_TOOL_SIGNATURE_SYMBOL,
   async runAsync(req) {
     try {
       let validatedArgs = req.args;
-      if (this.parameters instanceof ZodObject) {
+      if (isZodObject(this.parameters)) {
         validatedArgs = this.parameters.parse(req.args);
       }
       return await this.execute(

package/dist/esm/tools/google_search_tool.js

@@ -4,18 +4,15 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { isGemini1Model, isGeminiModel } from "../utils/model_name.js";
-import {
-  BaseTool
-} from "./base_tool.js";
+import { BaseTool } from "./base_tool.js";
 class GoogleSearchTool extends BaseTool {
   constructor() {
     super({ name: "google_search", description: "Google Search Tool" });
   }
-  runAsync(request) {
+  runAsync() {
     return Promise.resolve();
   }
   async processLlmRequest({
-    toolContext,
     llmRequest
   }) {
     if (!llmRequest.model) {

package/dist/esm/tools/long_running_tool.js

@@ -3,7 +3,9 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { FunctionTool } from "./function_tool.js";
+import {
+  FunctionTool
+} from "./function_tool.js";
 const LONG_RUNNING_INSTRUCTION = `
 
 NOTE: This is a long-running operation. Do not call this tool again if it has already returned some intermediate or pending status.`;

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,13 +36,21 @@ function base64Decode(data) {
 function isBase64Encoded(data) {
   try {
     return base64Encode(base64Decode(data)) === data;
-  } catch (e) {
+  } catch (_e) {
     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/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/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/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

@@ -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(envVar.toLowerCase());
-}
 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.2.5";
+const version = "0.4.0";
 export {
   version
 };