@google/adk 1.1.0 → 1.3.0

package/dist/cjs/plugins/security_plugin.js

@@ -47,6 +47,11 @@ var PolicyOutcome = /* @__PURE__ */ ((PolicyOutcome2) => {
   return PolicyOutcome2;
 })(PolicyOutcome || {});
 class InMemoryPolicyEngine {
+  /**
+   * Always returns {@link PolicyOutcome.ALLOW} for every tool call.
+   *
+   * @returns A promise resolving to an ALLOW result.
+   */
   async evaluate() {
     return Promise.resolve({
       outcome: "ALLOW" /* ALLOW */,
@@ -55,11 +60,22 @@ class InMemoryPolicyEngine {
   }
 }
 class SecurityPlugin extends import_base_plugin.BasePlugin {
+  /**
+   * @param params - Optional configuration. Defaults to {@link InMemoryPolicyEngine}
+   *   when no policy engine is provided.
+   */
   constructor(params) {
     var _a;
     super("security_plugin");
     this.policyEngine = (_a = params == null ? void 0 : params.policyEngine) != null ? _a : new InMemoryPolicyEngine();
   }
+  /**
+   * Intercepts tool calls, evaluating them against the policy engine before
+   * execution and handling confirmation flows for calls that require it.
+   *
+   * @returns A partial or error response object if the call is blocked or
+   *   awaiting confirmation, or `undefined` to allow execution to proceed.
+   */
   async beforeToolCallback({
     tool,
     toolArgs,

package/dist/cjs/runner/in_memory_runner.js

@@ -37,11 +37,16 @@ var import_runner = require("./runner.js");
  * SPDX-License-Identifier: Apache-2.0
  */
 class InMemoryRunner extends import_runner.Runner {
-  constructor({
-    agent,
-    appName = "InMemoryRunner",
-    plugins = []
-  }) {
+  /**
+   * Creates a new InMemoryRunner instance.
+   *
+   * @param params The configuration for the runner.
+   * @param params.agent The root agent to run.
+   * @param params.appName The application name. Defaults to `'InMemoryRunner'`.
+   * @param params.plugins An optional list of plugins.
+   */
+  constructor(params) {
+    const { agent, appName = "InMemoryRunner", plugins = [] } = params;
     super({
       appName,
       agent,

package/dist/cjs/runner/runner.js

@@ -38,6 +38,7 @@ var import_event = require("../events/event.js");
 var import_event_actions = require("../events/event_actions.js");
 var import_plugin_manager = require("../plugins/plugin_manager.js");
 var import_tracing = require("../telemetry/tracing.js");
+var import_base_toolset = require("../tools/base_toolset.js");
 var import_logger = require("../utils/logger.js");
 var import_model_name = require("../utils/model_name.js");
 /**
@@ -52,6 +53,11 @@ function isRunner(obj) {
 }
 _a = RUNNER_SIGNATURE_SYMBOL;
 class Runner {
+  /**
+   * Creates a new Runner instance.
+   *
+   * @param input The configuration for the runner.
+   */
   constructor(input) {
     this[_a] = true;
     var _a2;
@@ -84,7 +90,8 @@ class Runner {
         sessionId,
         newMessage: params.newMessage,
         stateDelta: params.stateDelta,
-        runConfig: params.runConfig
+        runConfig: params.runConfig,
+        customMetadata: params.customMetadata
       });
     } finally {
       await this.sessionService.deleteSession({
@@ -189,7 +196,8 @@ class Runner {
                 invocationId: invocationContext.invocationId,
                 author: "user",
                 actions: stateDelta ? (0, import_event_actions.createEventActions)({ stateDelta }) : void 0,
-                content: newMessage
+                content: newMessage,
+                customMetadata: params.customMetadata
               })
             });
             if ((_e = params.abortSignal) == null ? void 0 : _e.aborted) {
@@ -254,6 +262,8 @@ class Runner {
       );
     } finally {
       span.end();
+      const toolsets = getAllToolsets(this.agent);
+      await Promise.allSettled(toolsets.map((t) => t.close()));
     }
   }
   /**
@@ -373,6 +383,26 @@ function findEventByLastFunctionResponseId(events) {
   }
   return null;
 }
+function getAllToolsets(agent) {
+  const toolsets = [];
+  const visited = /* @__PURE__ */ new Set();
+  function traverse(curr) {
+    if (visited.has(curr)) return;
+    visited.add(curr);
+    if ((0, import_llm_agent.isLlmAgent)(curr)) {
+      for (const tool of curr.tools) {
+        if ((0, import_base_toolset.isBaseToolset)(tool)) {
+          toolsets.push(tool);
+        }
+      }
+    }
+    for (const sub of curr.subAgents) {
+      traverse(sub);
+    }
+  }
+  traverse(agent);
+  return toolsets;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Runner,

package/dist/cjs/sessions/base_session_service.js

@@ -26,7 +26,8 @@ var base_session_service_exports = {};
 __export(base_session_service_exports, {
   BaseSessionService: () => BaseSessionService,
   mergeStates: () => mergeStates,
-  trimTempDeltaState: () => trimTempDeltaState
+  trimTempDeltaState: () => trimTempDeltaState,
+  trimTempState: () => trimTempState
 });
 module.exports = __toCommonJS(base_session_service_exports);
 var import_lodash_es = require("lodash-es");
@@ -103,6 +104,15 @@ function trimTempDeltaState(event) {
   event.actions.stateDelta = filteredStateDelta;
   return event;
 }
+function trimTempState(state) {
+  const filteredState = {};
+  for (const [key, value] of Object.entries(state)) {
+    if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
+      filteredState[key] = value;
+    }
+  }
+  return filteredState;
+}
 function mergeStates(appState = {}, userState = {}, sessionState = {}) {
   const merged = (0, import_lodash_es.cloneDeep)(sessionState);
   for (const [k, v] of Object.entries(appState)) {
@@ -117,5 +127,6 @@ function mergeStates(appState = {}, userState = {}, sessionState = {}) {
 0 && (module.exports = {
   BaseSessionService,
   mergeStates,
-  trimTempDeltaState
+  trimTempDeltaState,
+  trimTempState
 });

package/dist/cjs/sessions/database_session_service.js

@@ -119,7 +119,7 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
           appStateDelta[key.replace(import_state.State.APP_PREFIX, "")] = value;
         } else if (key.startsWith(import_state.State.USER_PREFIX)) {
           userStateDelta[key.replace(import_state.State.USER_PREFIX, "")] = value;
-        } else {
+        } else if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
           sessionState[key] = value;
         }
       }
@@ -204,7 +204,11 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
   }
   async listSessions({
     appName,
-    userId
+    userId,
+    limit,
+    offset,
+    page,
+    order
   }) {
     await this.init();
     const em = this.orm.em.fork();
@@ -212,7 +216,49 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
     if (userId) {
       where.userId = userId;
     }
-    const storageSessions = await em.find(import_schema.StorageSession, where);
+    const orderBy = order === "asc" ? { updateTime: "ASC", id: "ASC" } : order === "desc" ? { updateTime: "DESC", id: "ASC" } : void 0;
+    let storageSessions;
+    let paginationMeta;
+    if (limit !== void 0) {
+      const totalItems = await em.count(import_schema.StorageSession, where);
+      const totalPages = limit === 0 ? 0 : Math.ceil(totalItems / limit);
+      let effectiveOffset;
+      let effectivePage;
+      if (page !== void 0) {
+        effectiveOffset = (page - 1) * limit;
+        effectivePage = page;
+      } else {
+        effectiveOffset = offset != null ? offset : 0;
+        effectivePage = limit === 0 ? 1 : Math.floor(effectiveOffset / limit) + 1;
+      }
+      storageSessions = await em.find(import_schema.StorageSession, where, {
+        orderBy,
+        limit,
+        offset: effectiveOffset
+      });
+      paginationMeta = { page: effectivePage, limit, totalItems, totalPages };
+    } else if (offset) {
+      const totalItems = await em.count(import_schema.StorageSession, where);
+      storageSessions = await em.find(import_schema.StorageSession, where, {
+        orderBy,
+        offset
+      });
+      paginationMeta = {
+        page: 1,
+        limit: totalItems,
+        totalItems,
+        totalPages: totalItems === 0 ? 0 : 1
+      };
+    } else {
+      storageSessions = await em.find(import_schema.StorageSession, where, { orderBy });
+      const totalItems = storageSessions.length;
+      paginationMeta = {
+        page: 1,
+        limit: totalItems,
+        totalItems,
+        totalPages: totalItems === 0 ? 0 : 1
+      };
+    }
     const appStateModel = await em.findOne(import_schema.StorageAppState, { appName });
     const appState = (appStateModel == null ? void 0 : appStateModel.state) || {};
     const userStateMap = {};
@@ -237,7 +283,7 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
         lastUpdateTime: ss.updateTime.getTime()
       });
     });
-    return { sessions };
+    return { sessions, ...paginationMeta };
   }
   async deleteSession({
     appName,
@@ -322,7 +368,7 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
             appDelta[key.replace(import_state.State.APP_PREFIX, "")] = value;
           } else if (key.startsWith(import_state.State.USER_PREFIX)) {
             userDelta[key.replace(import_state.State.USER_PREFIX, "")] = value;
-          } else {
+          } else if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
             sessionDelta[key] = value;
           }
         }

package/dist/cjs/sessions/db/schema.js

@@ -35,6 +35,7 @@ __export(schema_exports, {
   ENTITIES: () => ENTITIES,
   SCHEMA_VERSION_1_JSON: () => SCHEMA_VERSION_1_JSON,
   SCHEMA_VERSION_KEY: () => SCHEMA_VERSION_KEY,
+  STORAGE_KEY_COLUMN_LENGTH: () => STORAGE_KEY_COLUMN_LENGTH,
   StorageAppState: () => StorageAppState,
   StorageEvent: () => StorageEvent,
   StorageMetadata: () => StorageMetadata,
@@ -51,6 +52,7 @@ var import_event = require("../../events/event.js");
  */
 const SCHEMA_VERSION_KEY = "schema_version";
 const SCHEMA_VERSION_1_JSON = "1";
+const STORAGE_KEY_COLUMN_LENGTH = 191;
 class CamelCaseToSnakeCaseJsonType extends import_core.JsonType {
   convertToDatabaseValue(value) {
     return JSON.stringify((0, import_event.transformToSnakeCaseEvent)(value));
@@ -79,7 +81,11 @@ let StorageAppState = class {
   }
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageAppState.prototype, "appName", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "json" })
@@ -102,10 +108,18 @@ let StorageUserState = class {
   }
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageUserState.prototype, "appName", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "user_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "user_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageUserState.prototype, "userId", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "json" })
@@ -129,13 +143,21 @@ let StorageSession = class {
   }
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string" })
+  (0, import_core.PrimaryKey)({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
 ], StorageSession.prototype, "id", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageSession.prototype, "appName", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "user_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "user_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageSession.prototype, "userId", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "json" })
@@ -161,16 +183,28 @@ import_core.PrimaryKey.name;
 let StorageEvent = class {
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string" })
+  (0, import_core.PrimaryKey)({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
 ], StorageEvent.prototype, "id", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageEvent.prototype, "appName", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "user_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "user_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageEvent.prototype, "userId", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "session_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "session_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageEvent.prototype, "sessionId", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "string", fieldName: "invocation_id" })
@@ -196,6 +230,7 @@ const ENTITIES = [
   ENTITIES,
   SCHEMA_VERSION_1_JSON,
   SCHEMA_VERSION_KEY,
+  STORAGE_KEY_COLUMN_LENGTH,
   StorageAppState,
   StorageEvent,
   StorageMetadata,

package/dist/cjs/sessions/in_memory_session_service.js

@@ -66,11 +66,12 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     sessionId
   }) {
     var _a;
+    const filteredState = state ? (0, import_base_session_service.trimTempState)(state) : void 0;
     const session = (0, import_session.createSession)({
       id: sessionId || (0, import_env_aware_utils.randomUUID)(),
       appName,
       userId,
-      state,
+      state: filteredState,
       events: [],
       lastUpdateTime: Date.now()
     });
@@ -129,25 +130,81 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
   }
   listSessions({
     appName,
-    userId
+    userId,
+    limit,
+    offset,
+    page,
+    order
   }) {
     if (!this.sessions[appName] || !this.sessions[appName][userId]) {
-      return Promise.resolve({ sessions: [] });
+      if (limit !== void 0) {
+        const effectiveOffset2 = page !== void 0 ? (page - 1) * limit : offset != null ? offset : 0;
+        const effectivePage2 = page !== void 0 ? page : limit === 0 ? 1 : Math.floor(effectiveOffset2 / limit) + 1;
+        return Promise.resolve({
+          sessions: [],
+          page: effectivePage2,
+          limit,
+          totalItems: 0,
+          totalPages: 0
+        });
+      }
+      return Promise.resolve({
+        sessions: [],
+        page: 1,
+        limit: 0,
+        totalItems: 0,
+        totalPages: 0
+      });
     }
-    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
-        })
+    const all = Object.values(this.sessions[appName][userId]).map(
+      (session) => (0, import_session.createSession)({
+        id: session.id,
+        appName: session.appName,
+        userId: session.userId,
+        state: {},
+        events: [],
+        lastUpdateTime: session.lastUpdateTime
+      })
+    );
+    if (order === "asc") {
+      all.sort(
+        (a, b) => a.lastUpdateTime - b.lastUpdateTime || a.id.localeCompare(b.id)
+      );
+    } else if (order === "desc") {
+      all.sort(
+        (a, b) => b.lastUpdateTime - a.lastUpdateTime || a.id.localeCompare(b.id)
       );
     }
-    return Promise.resolve({ sessions: sessionsWithoutEvents });
+    if (limit === void 0) {
+      const totalItems2 = all.length;
+      const sliced = offset ? all.slice(offset) : all;
+      return Promise.resolve({
+        sessions: sliced,
+        page: 1,
+        limit: totalItems2,
+        totalItems: totalItems2,
+        totalPages: totalItems2 === 0 ? 0 : 1
+      });
+    }
+    const totalItems = all.length;
+    const totalPages = limit === 0 ? 0 : Math.ceil(totalItems / limit);
+    let effectiveOffset;
+    let effectivePage;
+    if (page !== void 0) {
+      effectiveOffset = (page - 1) * limit;
+      effectivePage = page;
+    } else {
+      effectiveOffset = offset != null ? offset : 0;
+      effectivePage = limit === 0 ? 1 : Math.floor(effectiveOffset / limit) + 1;
+    }
+    const paginated = all.slice(effectiveOffset, effectiveOffset + limit);
+    return Promise.resolve({
+      sessions: paginated,
+      page: effectivePage,
+      limit,
+      totalItems,
+      totalPages
+    });
   }
   async deleteSession({
     appName,

package/dist/cjs/sessions/registry.js

@@ -29,6 +29,7 @@ __export(registry_exports, {
 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");
+var import_vertex_ai_session_service = require("./vertex_ai_session_service.js");
 /**
  * @license
  * Copyright 2026 Google LLC
@@ -41,6 +42,9 @@ function getSessionServiceFromUri(uri) {
   if ((0, import_database_session_service.isDatabaseConnectionString)(uri)) {
     return new import_database_session_service.DatabaseSessionService(uri);
   }
+  if ((0, import_vertex_ai_session_service.isVertexAiConnectionString)(uri)) {
+    return new import_vertex_ai_session_service.VertexAiSessionService({});
+  }
   throw new Error(`Unsupported session service URI: ${uri}`);
 }
 // Annotate the CommonJS export names for ESM import in node: