@google/adk 0.2.5 → 0.4.0

package/dist/cjs/sessions/in_memory_session_service.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -24,7 +24,8 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var in_memory_session_service_exports = {};
 __export(in_memory_session_service_exports, {
-  InMemorySessionService: () => InMemorySessionService
+  InMemorySessionService: () => InMemorySessionService,
+  isInMemoryConnectionString: () => isInMemoryConnectionString
 });
 module.exports = __toCommonJS(in_memory_session_service_exports);
 var import_lodash_es = require("lodash-es");
@@ -38,6 +39,9 @@ var import_state = require("./state.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+function isInMemoryConnectionString(uri) {
+  return uri === "memory://";
+}
 class InMemorySessionService extends import_base_session_service.BaseSessionService {
   constructor() {
     super(...arguments);
@@ -55,7 +59,13 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
      */
     this.appState = {};
   }
-  createSession({ appName, userId, state, sessionId }) {
+  async createSession({
+    appName,
+    userId,
+    state,
+    sessionId
+  }) {
+    var _a;
     const session = (0, import_session.createSession)({
       id: sessionId || (0, import_env_aware_utils.randomUUID)(),
       appName,
@@ -71,11 +81,21 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
       this.sessions[appName][userId] = {};
     }
     this.sessions[appName][userId][session.id] = session;
-    return Promise.resolve(
-      this.mergeState(appName, userId, (0, import_lodash_es.cloneDeep)(session))
+    const copiedSession = (0, import_lodash_es.cloneDeep)(session);
+    copiedSession.state = (0, import_base_session_service.mergeStates)(
+      this.appState[appName],
+      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
+      copiedSession.state
     );
+    return copiedSession;
   }
-  getSession({ appName, userId, sessionId, config }) {
+  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 +103,9 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     const copiedSession = (0, import_lodash_es.cloneDeep)(session);
     if (config) {
       if (config.numRecentEvents) {
-        copiedSession.events = copiedSession.events.slice(-config.numRecentEvents);
+        copiedSession.events = copiedSession.events.slice(
+          -config.numRecentEvents
+        );
       }
       if (config.afterTimestamp) {
         let i = copiedSession.events.length - 1;
@@ -98,33 +120,50 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
         }
       }
     }
-    return Promise.resolve(this.mergeState(appName, userId, copiedSession));
+    copiedSession.state = (0, import_base_session_service.mergeStates)(
+      this.appState[appName],
+      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
+      copiedSession.state
+    );
+    return copiedSession;
   }
-  listSessions({ appName, userId }) {
+  listSessions({
+    appName,
+    userId
+  }) {
     if (!this.sessions[appName] || !this.sessions[appName][userId]) {
       return Promise.resolve({ sessions: [] });
     }
     const sessionsWithoutEvents = [];
     for (const session of Object.values(this.sessions[appName][userId])) {
-      sessionsWithoutEvents.push((0, import_session.createSession)({
-        id: session.id,
-        appName: session.appName,
-        userId: session.userId,
-        state: {},
-        events: [],
-        lastUpdateTime: session.lastUpdateTime
-      }));
+      sessionsWithoutEvents.push(
+        (0, import_session.createSession)({
+          id: session.id,
+          appName: session.appName,
+          userId: session.userId,
+          state: {},
+          events: [],
+          lastUpdateTime: session.lastUpdateTime
+        })
+      );
     }
     return Promise.resolve({ sessions: sessionsWithoutEvents });
   }
-  async deleteSession({ appName, userId, sessionId }) {
+  async deleteSession({
+    appName,
+    userId,
+    sessionId
+  }) {
     const session = await this.getSession({ appName, userId, sessionId });
     if (!session) {
       return;
     }
     delete this.sessions[appName][userId][sessionId];
   }
-  async appendEvent({ session, event }) {
+  async appendEvent({
+    session,
+    event
+  }) {
     await super.appendEvent({ session, event });
     session.lastUpdateTime = event.timestamp;
     const appName = session.appName;
@@ -163,22 +202,9 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     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[import_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[import_state.State.USER_PREFIX + key] = this.userState[appName][userId][key];
-    }
-    return copiedSession;
-  }
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  InMemorySessionService
+  InMemorySessionService,
+  isInMemoryConnectionString
 });

package/dist/cjs/sessions/registry.js

@@ -0,0 +1,49 @@
+/**
+  * @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 registry_exports = {};
+__export(registry_exports, {
+  getSessionServiceFromUri: () => getSessionServiceFromUri
+});
+module.exports = __toCommonJS(registry_exports);
+var import_database_session_service = require("./database_session_service.js");
+var import_in_memory_session_service = require("./in_memory_session_service.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function getSessionServiceFromUri(uri) {
+  if ((0, import_in_memory_session_service.isInMemoryConnectionString)(uri)) {
+    return new import_in_memory_session_service.InMemorySessionService();
+  }
+  if ((0, import_database_session_service.isDatabaseConnectionString)(uri)) {
+    return new import_database_session_service.DatabaseSessionService(uri);
+  }
+  throw new Error(`Unsupported session service URI: ${uri}`);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  getSessionServiceFromUri
+});

package/dist/cjs/sessions/session.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/sessions/state.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/telemetry/google_cloud.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -28,13 +28,13 @@ __export(google_cloud_exports, {
   getGcpResource: () => getGcpResource
 });
 module.exports = __toCommonJS(google_cloud_exports);
-var import_google_auth_library = require("google-auth-library");
-var import_sdk_metrics = require("@opentelemetry/sdk-metrics");
-var import_resources = require("@opentelemetry/resources");
-var import_resource_detector_gcp = require("@opentelemetry/resource-detector-gcp");
+var import_opentelemetry_cloud_monitoring_exporter = require("@google-cloud/opentelemetry-cloud-monitoring-exporter");
 var import_opentelemetry_cloud_trace_exporter = require("@google-cloud/opentelemetry-cloud-trace-exporter");
+var import_resource_detector_gcp = require("@opentelemetry/resource-detector-gcp");
+var import_resources = require("@opentelemetry/resources");
+var import_sdk_metrics = require("@opentelemetry/sdk-metrics");
 var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
-var import_opentelemetry_cloud_monitoring_exporter = require("@google-cloud/opentelemetry-cloud-monitoring-exporter");
+var import_google_auth_library = require("google-auth-library");
 var import_logger = require("../utils/logger.js");
 /**
  * @license
@@ -47,7 +47,7 @@ async function getGcpProjectId() {
     const auth = new import_google_auth_library.GoogleAuth();
     const projectId = await auth.getProjectId();
     return projectId || void 0;
-  } catch (error) {
+  } catch (_e) {
     return void 0;
   }
 }
@@ -63,9 +63,7 @@ async function getGcpExporters(config = {}) {
     return {};
   }
   return {
-    spanProcessors: enableTracing ? [
-      new import_sdk_trace_base.BatchSpanProcessor(new import_opentelemetry_cloud_trace_exporter.TraceExporter({ projectId }))
-    ] : [],
+    spanProcessors: enableTracing ? [new import_sdk_trace_base.BatchSpanProcessor(new import_opentelemetry_cloud_trace_exporter.TraceExporter({ projectId }))] : [],
     metricReaders: enableMetrics ? [
       new import_sdk_metrics.PeriodicExportingMetricReader({
         exporter: new import_opentelemetry_cloud_monitoring_exporter.MetricExporter({ projectId }),

package/dist/cjs/telemetry/setup.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -29,14 +29,14 @@ __export(setup_exports, {
 module.exports = __toCommonJS(setup_exports);
 var import_api = require("@opentelemetry/api");
 var import_api_logs = require("@opentelemetry/api-logs");
+var import_exporter_logs_otlp_http = require("@opentelemetry/exporter-logs-otlp-http");
+var import_exporter_metrics_otlp_http = require("@opentelemetry/exporter-metrics-otlp-http");
+var import_exporter_trace_otlp_http = require("@opentelemetry/exporter-trace-otlp-http");
+var import_resources = require("@opentelemetry/resources");
 var import_sdk_logs = require("@opentelemetry/sdk-logs");
 var import_sdk_metrics = require("@opentelemetry/sdk-metrics");
-var import_resources = require("@opentelemetry/resources");
 var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
 var import_sdk_trace_node = require("@opentelemetry/sdk-trace-node");
-var import_exporter_trace_otlp_http = require("@opentelemetry/exporter-trace-otlp-http");
-var import_exporter_metrics_otlp_http = require("@opentelemetry/exporter-metrics-otlp-http");
-var import_exporter_logs_otlp_http = require("@opentelemetry/exporter-logs-otlp-http");
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -45,9 +45,13 @@ var import_exporter_logs_otlp_http = require("@opentelemetry/exporter-logs-otlp-
 function maybeSetOtelProviders(otelHooksToSetup = [], otelResource) {
   const resource = otelResource || getOtelResource();
   const allHooks = [...otelHooksToSetup, getOtelExporters()];
-  const spanProcessors = allHooks.flatMap((hooks) => hooks.spanProcessors || []);
+  const spanProcessors = allHooks.flatMap(
+    (hooks) => hooks.spanProcessors || []
+  );
   const metricReaders = allHooks.flatMap((hooks) => hooks.metricReaders || []);
-  const logRecordProcessors = allHooks.flatMap((hooks) => hooks.logRecordProcessors || []);
+  const logRecordProcessors = allHooks.flatMap(
+    (hooks) => hooks.logRecordProcessors || []
+  );
   if (spanProcessors.length > 0) {
     const tracerProvider = new import_sdk_trace_node.NodeTracerProvider({
       resource,
@@ -87,7 +91,11 @@ function getOtelExporters(config = getOtelExportersConfig()) {
   const { enableTracing, enableMetrics, enableLogging } = config;
   return {
     spanProcessors: enableTracing ? [new import_sdk_trace_base.BatchSpanProcessor(new import_exporter_trace_otlp_http.OTLPTraceExporter())] : [],
-    metricReaders: enableMetrics ? [new import_sdk_metrics.PeriodicExportingMetricReader({ exporter: new import_exporter_metrics_otlp_http.OTLPMetricExporter() })] : [],
+    metricReaders: enableMetrics ? [
+      new import_sdk_metrics.PeriodicExportingMetricReader({
+        exporter: new import_exporter_metrics_otlp_http.OTLPMetricExporter()
+      })
+    ] : [],
     logRecordProcessors: enableLogging ? [new import_sdk_logs.BatchLogRecordProcessor(new import_exporter_logs_otlp_http.OTLPLogExporter())] : []
   };
 }

package/dist/cjs/telemetry/tracing.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -24,7 +24,7 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var tracing_exports = {};
 __export(tracing_exports, {
-  bindAsyncGenerator: () => bindAsyncGenerator,
+  runAsyncGeneratorWithOtelContext: () => runAsyncGeneratorWithOtelContext,
   traceAgentInvocation: () => traceAgentInvocation,
   traceCallLlm: () => traceCallLlm,
   traceMergedToolCalls: () => traceMergedToolCalls,
@@ -48,14 +48,11 @@ const GEN_AI_TOOL_CALL_ID = "gen_ai.tool.call.id";
 const GEN_AI_TOOL_DESCRIPTION = "gen_ai.tool.description";
 const GEN_AI_TOOL_NAME = "gen_ai.tool.name";
 const GEN_AI_TOOL_TYPE = "gen_ai.tool.type";
-const tracer = import_api.trace.getTracer(
-  "gcp.vertex.agent",
-  import_version.version
-);
+const tracer = import_api.trace.getTracer("gcp.vertex.agent", import_version.version);
 function safeJsonSerialize(obj) {
   try {
     return JSON.stringify(obj);
-  } catch (error) {
+  } catch (_e) {
     return "<not serializable>";
   }
 }
@@ -132,7 +129,10 @@ function traceMergedToolCalls({
     "gcp.vertex.agent.llm_request": "{}",
     "gcp.vertex.agent.llm_response": "{}"
   });
-  span.setAttribute("gcp.vertex.agent.tool_response", shouldAddRequestResponseToSpans() ? safeJsonSerialize(functionResponseEvent) : "{}");
+  span.setAttribute(
+    "gcp.vertex.agent.tool_response",
+    shouldAddRequestResponseToSpans() ? safeJsonSerialize(functionResponseEvent) : "{}"
+  );
 }
 function traceCallLlm({
   invocationContext,
@@ -156,14 +156,26 @@ function traceCallLlm({
     span.setAttribute("gen_ai.request.top_p", llmRequest.config.topP);
   }
   if (((_b = llmRequest.config) == null ? void 0 : _b.maxOutputTokens) !== void 0) {
-    span.setAttribute("gen_ai.request.max_tokens", llmRequest.config.maxOutputTokens);
+    span.setAttribute(
+      "gen_ai.request.max_tokens",
+      llmRequest.config.maxOutputTokens
+    );
   }
-  span.setAttribute("gcp.vertex.agent.llm_response", shouldAddRequestResponseToSpans() ? safeJsonSerialize(llmResponse) : "{}");
+  span.setAttribute(
+    "gcp.vertex.agent.llm_response",
+    shouldAddRequestResponseToSpans() ? safeJsonSerialize(llmResponse) : "{}"
+  );
   if (llmResponse.usageMetadata) {
-    span.setAttribute("gen_ai.usage.input_tokens", llmResponse.usageMetadata.promptTokenCount || 0);
+    span.setAttribute(
+      "gen_ai.usage.input_tokens",
+      llmResponse.usageMetadata.promptTokenCount || 0
+    );
   }
   if ((_c = llmResponse.usageMetadata) == null ? void 0 : _c.candidatesTokenCount) {
-    span.setAttribute("gen_ai.usage.output_tokens", llmResponse.usageMetadata.candidatesTokenCount);
+    span.setAttribute(
+      "gen_ai.usage.output_tokens",
+      llmResponse.usageMetadata.candidatesTokenCount
+    );
   }
   if (llmResponse.finishReason) {
     const finishReasonValue = typeof llmResponse.finishReason === "string" ? llmResponse.finishReason.toLowerCase() : String(llmResponse.finishReason).toLowerCase();
@@ -181,7 +193,10 @@ function traceSendData({
     "gcp.vertex.agent.invocation_id": invocationContext.invocationId,
     "gcp.vertex.agent.event_id": eventId
   });
-  span.setAttribute("gcp.vertex.agent.data", shouldAddRequestResponseToSpans() ? safeJsonSerialize(data) : "{}");
+  span.setAttribute(
+    "gcp.vertex.agent.data",
+    shouldAddRequestResponseToSpans() ? safeJsonSerialize(data) : "{}"
+  );
 }
 function buildLlmRequestForTrace(llmRequest) {
   const result = {
@@ -201,7 +216,7 @@ function buildLlmRequestForTrace(llmRequest) {
   });
   return result;
 }
-function bindAsyncGenerator(ctx, generator) {
+function bindOtelContextToAsyncGenerator(ctx, generator) {
   return {
     // Bind the next() method to execute within the provided context
     next: import_api.context.bind(ctx, generator.next.bind(generator)),
@@ -211,17 +226,24 @@ function bindAsyncGenerator(ctx, generator) {
     throw: import_api.context.bind(ctx, generator.throw.bind(generator)),
     // Ensure the async iterator symbol also returns a context-bound generator
     [Symbol.asyncIterator]() {
-      return bindAsyncGenerator(ctx, generator[Symbol.asyncIterator]());
+      return bindOtelContextToAsyncGenerator(
+        ctx,
+        generator[Symbol.asyncIterator]()
+      );
     }
   };
 }
+function runAsyncGeneratorWithOtelContext(otelContext, generatorFnContext, generatorFn) {
+  const generator = generatorFn.call(generatorFnContext);
+  return bindOtelContextToAsyncGenerator(otelContext, generator);
+}
 function shouldAddRequestResponseToSpans() {
   const envValue = process.env.ADK_CAPTURE_MESSAGE_CONTENT_IN_SPANS || "true";
   return envValue === "true" || envValue === "1";
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  bindAsyncGenerator,
+  runAsyncGeneratorWithOtelContext,
   traceAgentInvocation,
   traceCallLlm,
   traceMergedToolCalls,

package/dist/cjs/tools/agent_tool.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -48,9 +48,10 @@ function isAgentTool(obj) {
 }
 class AgentTool extends (_b = import_base_tool.BaseTool, _a = AGENT_TOOL_SIGNATURE_SYMBOL, _b) {
   constructor(config) {
-    super(
-      { name: config.agent.name, description: config.agent.description || "" }
-    );
+    super({
+      name: config.agent.name,
+      description: config.agent.description || ""
+    });
     /** A unique symbol to identify ADK agent tool class. */
     this[_a] = true;
     this.agent = config.agent;
@@ -88,7 +89,10 @@ class AgentTool extends (_b = import_base_tool.BaseTool, _a = AGENT_TOOL_SIGNATU
     }
     return declaration;
   }
-  async runAsync({ args, toolContext }) {
+  async runAsync({
+    args,
+    toolContext
+  }) {
     var _a2, _b2;
     if (this.skipSummarization) {
       toolContext.actions.skipSummarization = true;

package/dist/cjs/tools/base_tool.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -79,7 +79,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;
@@ -108,7 +108,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
+  );
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/cjs/tools/base_toolset.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/tools/forwarding_artifact_service.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -45,7 +45,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) {
@@ -64,6 +64,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
+    );
+  }
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/cjs/tools/function_tool.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -29,7 +29,6 @@ __export(function_tool_exports, {
 });
 module.exports = __toCommonJS(function_tool_exports);
 var import_genai = require("@google/genai");
-var import_zod = require("zod");
 var import_simple_zod_to_json = require("../utils/simple_zod_to_json.js");
 var import_base_tool = require("./base_tool.js");
 /**
@@ -90,7 +89,7 @@ class FunctionTool extends (_b = import_base_tool.BaseTool, _a = FUNCTION_TOOL_S
   async runAsync(req) {
     try {
       let validatedArgs = req.args;
-      if (this.parameters instanceof import_zod.ZodObject) {
+      if ((0, import_simple_zod_to_json.isZodObject)(this.parameters)) {
         validatedArgs = this.parameters.parse(req.args);
       }
       return await this.execute(

package/dist/cjs/tools/google_search_tool.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -39,11 +39,10 @@ class GoogleSearchTool extends import_base_tool.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/cjs/tools/long_running_tool.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

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

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -40,6 +40,7 @@ class MCPSessionManager {
     this.connectionParams = connectionParams;
   }
   async createSession() {
+    var _a;
     const client = new import_client.Client({ name: "MCPClient", version: "1.0.0" });
     switch (this.connectionParams.type) {
       case "StdioConnectionParams":
@@ -47,20 +48,25 @@ class MCPSessionManager {
           new import_stdio.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 import_streamableHttp.StreamableHTTPClientTransport(
-          new URL(this.connectionParams.url),
-          transportOptions
-        ));
+          };
+        }
+        await client.connect(
+          new import_streamableHttp.StreamableHTTPClientTransport(
+            new URL(this.connectionParams.url),
+            options
+          )
+        );
         break;
-      default:
+      }
+      default: {
         const _exhaustiveCheck = this.connectionParams;
         break;
+      }
     }
     return client;
   }