@google/adk 1.5.0 → 2.0.0

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

@@ -2,6 +2,7 @@ import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __decorateClass = (decorators, target, key, kind) => {
   var result = kind > 1 ? void 0 : kind ? __getOwnPropDesc(target, key) : target;
   for (var i = decorators.length - 1, decorator; i >= 0; i--)
@@ -23,7 +24,7 @@ import {
 const SCHEMA_VERSION_KEY = "schema_version";
 const SCHEMA_VERSION_1_JSON = "1";
 const STORAGE_KEY_COLUMN_LENGTH = 191;
-class CamelCaseToSnakeCaseJsonType extends JsonType {
+const _CamelCaseToSnakeCaseJsonType = class _CamelCaseToSnakeCaseJsonType extends JsonType {
   convertToDatabaseValue(value) {
     return JSON.stringify(transformToSnakeCaseEvent(value));
   }
@@ -33,9 +34,12 @@ class CamelCaseToSnakeCaseJsonType extends JsonType {
     }
     return transformToCamelCaseEvent(value);
   }
-}
+};
+__name(_CamelCaseToSnakeCaseJsonType, "CamelCaseToSnakeCaseJsonType");
+let CamelCaseToSnakeCaseJsonType = _CamelCaseToSnakeCaseJsonType;
 let StorageMetadata = class {
 };
+__name(StorageMetadata, "StorageMetadata");
 __decorateClass([
   PrimaryKey({ type: "string" })
 ], StorageMetadata.prototype, "key", 2);
@@ -50,6 +54,7 @@ let StorageAppState = class {
     this.updateTime = /* @__PURE__ */ new Date();
   }
 };
+__name(StorageAppState, "StorageAppState");
 __decorateClass([
   PrimaryKey({
     type: "string",
@@ -64,8 +69,8 @@ __decorateClass([
   Property({
     type: "datetime",
     fieldName: "update_time",
-    onCreate: () => /* @__PURE__ */ new Date(),
-    onUpdate: () => /* @__PURE__ */ new Date()
+    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate"),
+    onUpdate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onUpdate")
   })
 ], StorageAppState.prototype, "updateTime", 2);
 StorageAppState = __decorateClass([
@@ -77,6 +82,7 @@ let StorageUserState = class {
     this.updateTime = /* @__PURE__ */ new Date();
   }
 };
+__name(StorageUserState, "StorageUserState");
 __decorateClass([
   PrimaryKey({
     type: "string",
@@ -98,8 +104,8 @@ __decorateClass([
   Property({
     type: "datetime",
     fieldName: "update_time",
-    onCreate: () => /* @__PURE__ */ new Date(),
-    onUpdate: () => /* @__PURE__ */ new Date()
+    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate"),
+    onUpdate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onUpdate")
   })
 ], StorageUserState.prototype, "updateTime", 2);
 StorageUserState = __decorateClass([
@@ -112,6 +118,7 @@ let StorageSession = class {
     this.updateTime = /* @__PURE__ */ new Date();
   }
 };
+__name(StorageSession, "StorageSession");
 __decorateClass([
   PrimaryKey({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
 ], StorageSession.prototype, "id", 2);
@@ -136,14 +143,14 @@ __decorateClass([
   Property({
     type: "datetime",
     fieldName: "create_time",
-    onCreate: () => /* @__PURE__ */ new Date()
+    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate")
   })
 ], StorageSession.prototype, "createTime", 2);
 __decorateClass([
   Property({
     type: "datetime",
     fieldName: "update_time",
-    onCreate: () => /* @__PURE__ */ new Date()
+    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate")
   })
 ], StorageSession.prototype, "updateTime", 2);
 StorageSession = __decorateClass([
@@ -152,6 +159,7 @@ StorageSession = __decorateClass([
 PrimaryKey.name;
 let StorageEvent = class {
 };
+__name(StorageEvent, "StorageEvent");
 __decorateClass([
   PrimaryKey({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
 ], StorageEvent.prototype, "id", 2);

package/dist/esm/sessions/in_memory_session_service.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -18,22 +20,23 @@ import { State } from "./state.js";
 function isInMemoryConnectionString(uri) {
   return uri === "memory://";
 }
-class InMemorySessionService extends BaseSessionService {
+__name(isInMemoryConnectionString, "isInMemoryConnectionString");
+const _InMemorySessionService = class _InMemorySessionService extends BaseSessionService {
   constructor() {
     super(...arguments);
     /**
      * A map from app name to a map from user ID to a map from session ID to
      * session.
      */
-    this.sessions = {};
+    this.sessions = /* @__PURE__ */ Object.create(null);
     /**
      * A map from app name to a map from user ID to a map from key to the value.
      */
-    this.userState = {};
+    this.userState = /* @__PURE__ */ Object.create(null);
     /**
      * A map from app name to a map from key to the value.
      */
-    this.appState = {};
+    this.appState = /* @__PURE__ */ Object.create(null);
   }
   async createSession({
     appName,
@@ -52,10 +55,10 @@ class InMemorySessionService extends BaseSessionService {
       lastUpdateTime: Date.now()
     });
     if (!this.sessions[appName]) {
-      this.sessions[appName] = {};
+      this.sessions[appName] = /* @__PURE__ */ Object.create(null);
     }
     if (!this.sessions[appName][userId]) {
-      this.sessions[appName][userId] = {};
+      this.sessions[appName][userId] = /* @__PURE__ */ Object.create(null);
     }
     this.sessions[appName][userId][session.id] = session;
     const copiedSession = cloneDeep(session);
@@ -112,7 +115,9 @@ class InMemorySessionService extends BaseSessionService {
     page,
     order
   }) {
-    if (!this.sessions[appName] || !this.sessions[appName][userId]) {
+    const appSessions = this.sessions[appName];
+    const sessionsByUser = appSessions === void 0 ? [] : userId === void 0 ? Object.values(appSessions) : appSessions[userId] ? [appSessions[userId]] : [];
+    if (sessionsByUser.length === 0) {
       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;
@@ -132,15 +137,17 @@ class InMemorySessionService extends BaseSessionService {
         totalPages: 0
       });
     }
-    const all = Object.values(this.sessions[appName][userId]).map(
-      (session) => createSession({
-        id: session.id,
-        appName: session.appName,
-        userId: session.userId,
-        state: {},
-        events: [],
-        lastUpdateTime: session.lastUpdateTime
-      })
+    const all = sessionsByUser.flatMap(
+      (sessionsById) => Object.values(sessionsById).map(
+        (session) => createSession({
+          id: session.id,
+          appName: session.appName,
+          userId: session.userId,
+          state: {},
+          events: [],
+          lastUpdateTime: session.lastUpdateTime
+        })
+      )
     );
     if (order === "asc") {
       all.sort(
@@ -202,9 +209,9 @@ class InMemorySessionService extends BaseSessionService {
     const appName = session.appName;
     const userId = session.userId;
     const sessionId = session.id;
-    const warning = (message) => {
+    const warning = /* @__PURE__ */ __name((message) => {
       logger.warn(`Failed to append event to session ${sessionId}: ${message}`);
-    };
+    }, "warning");
     if (!this.sessions[appName]) {
       warning(`appName ${appName} not in sessions`);
       return event;
@@ -220,12 +227,12 @@ class InMemorySessionService extends BaseSessionService {
     if (event.actions && event.actions.stateDelta) {
       for (const key of Object.keys(event.actions.stateDelta)) {
         if (key.startsWith(State.APP_PREFIX)) {
-          this.appState[appName] = this.appState[appName] || {};
+          this.appState[appName] = this.appState[appName] || /* @__PURE__ */ Object.create(null);
           this.appState[appName][key.replace(State.APP_PREFIX, "")] = event.actions.stateDelta[key];
         }
         if (key.startsWith(State.USER_PREFIX)) {
-          this.userState[appName] = this.userState[appName] || {};
-          this.userState[appName][userId] = this.userState[appName][userId] || {};
+          this.userState[appName] = this.userState[appName] || /* @__PURE__ */ Object.create(null);
+          this.userState[appName][userId] = this.userState[appName][userId] || /* @__PURE__ */ Object.create(null);
           this.userState[appName][userId][key.replace(State.USER_PREFIX, "")] = event.actions.stateDelta[key];
         }
       }
@@ -235,7 +242,9 @@ class InMemorySessionService extends BaseSessionService {
     storageSession.lastUpdateTime = event.timestamp;
     return event;
   }
-}
+};
+__name(_InMemorySessionService, "InMemorySessionService");
+let InMemorySessionService = _InMemorySessionService;
 export {
   InMemorySessionService,
   isInMemoryConnectionString

package/dist/esm/sessions/registry.js

@@ -1,10 +1,13 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { redactUriPassword } from "../utils/redact_uri.js";
 import {
   DatabaseSessionService,
   isDatabaseConnectionString
@@ -27,8 +30,9 @@ function getSessionServiceFromUri(uri) {
   if (isVertexAiConnectionString(uri)) {
     return new VertexAiSessionService({});
   }
-  throw new Error(`Unsupported session service URI: ${uri}`);
+  throw new Error(`Unsupported session service URI: ${redactUriPassword(uri)}`);
 }
+__name(getSessionServiceFromUri, "getSessionServiceFromUri");
 export {
   getSessionServiceFromUri
 };

package/dist/esm/sessions/session.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -15,6 +17,7 @@ function createSession(params) {
     lastUpdateTime: params.lastUpdateTime || 0
   };
 }
+__name(createSession, "createSession");
 export {
   createSession
 };

package/dist/esm/sessions/state.js

@@ -1,14 +1,75 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class State {
-  constructor(value = {}, delta = {}) {
+import { objectSchemaFields } from "../utils/schema.js";
+import { recordStateWrite } from "./state_write_order.js";
+const _StateSchemaError = class _StateSchemaError extends TypeError {
+  constructor(message) {
+    super(message);
+    this.name = "StateSchemaError";
+    Object.setPrototypeOf(this, _StateSchemaError.prototype);
+  }
+};
+__name(_StateSchemaError, "StateSchemaError");
+let StateSchemaError = _StateSchemaError;
+function isStateSchemaError(e) {
+  return e instanceof Error && e.name === "StateSchemaError";
+}
+__name(isStateSchemaError, "isStateSchemaError");
+const _State = class _State {
+  constructor(value = {}, delta = {}, schema) {
     this.value = value;
     this.delta = delta;
+    this.schema = schema;
+  }
+  /**
+   * Checks a whole delta against {@link schema}, for writes that reach the
+   * session without passing through this object — a node emitting an event
+   * that carries its own `stateDelta`.
+   */
+  validateDelta(delta) {
+    for (const [key, value] of Object.entries(delta)) {
+      this.validate(key, value);
+    }
+  }
+  /**
+   * Checks one key/value pair against {@link schema}, throwing
+   * {@link StateSchemaError} when the key is undeclared or its value does not
+   * match the declared type.
+   *
+   * Prefixed keys (`app:`, `user:`, `temp:`, or any other namespace) belong to
+   * a scope wider than this state and are never validated — the same carve-out
+   * adk-python makes.
+   */
+  validate(key, value) {
+    if (this.schema === void 0 || key.includes(":")) {
+      return;
+    }
+    const fields = objectSchemaFields(this.schema);
+    if (!fields) {
+      return;
+    }
+    if (!fields.has(key)) {
+      throw new StateSchemaError(
+        `Key '${key}' is not declared in the state schema. Declared fields: ` + JSON.stringify([...fields.keys()].sort())
+      );
+    }
+    const field = fields.get(key);
+    if (!field) {
+      return;
+    }
+    const parsed = field.safeParse(value);
+    if (!parsed.success) {
+      throw new StateSchemaError(
+        `Value for '${key}' does not match the type declared in the state schema: ${parsed.error.message}`
+      );
+    }
   }
   /**
    * Returns the value of the state dict for the given key.
@@ -34,8 +95,10 @@ class State {
    * @param value The value to set.
    */
   set(key, value) {
+    this.validate(key, value);
     this.value[key] = value;
     this.delta[key] = value;
+    recordStateWrite(this.value, this.delta, key);
   }
   /**
    * Whether the state has pending delta.
@@ -55,8 +118,12 @@ class State {
    * @param delta The delta to update the state with.
    */
   update(delta) {
+    this.validateDelta(delta);
     Object.assign(this.delta, delta);
     Object.assign(this.value, delta);
+    for (const key of Object.keys(delta)) {
+      recordStateWrite(this.value, this.delta, key);
+    }
   }
   /**
    * Returns the state as a plain JSON object.
@@ -64,10 +131,14 @@ class State {
   toRecord() {
     return { ...this.value, ...this.delta };
   }
-}
-State.APP_PREFIX = "app:";
-State.USER_PREFIX = "user:";
-State.TEMP_PREFIX = "temp:";
+};
+__name(_State, "State");
+_State.APP_PREFIX = "app:";
+_State.USER_PREFIX = "user:";
+_State.TEMP_PREFIX = "temp:";
+let State = _State;
 export {
-  State
+  State,
+  StateSchemaError,
+  isStateSchemaError
 };

package/dist/esm/sessions/state_write_order.js

@@ -0,0 +1,74 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+let writeClock = 0;
+const stateStamps = /* @__PURE__ */ new WeakMap();
+const deltaStamps = /* @__PURE__ */ new WeakMap();
+function stampsFor(registry, target) {
+  let stamps = registry.get(target);
+  if (!stamps) {
+    stamps = /* @__PURE__ */ new Map();
+    registry.set(target, stamps);
+  }
+  return stamps;
+}
+__name(stampsFor, "stampsFor");
+function recordStateWrite(state, delta, key) {
+  const stamp = ++writeClock;
+  stampsFor(stateStamps, state).set(key, stamp);
+  if (delta && delta !== state) {
+    stampsFor(deltaStamps, delta).set(key, stamp);
+  }
+}
+__name(recordStateWrite, "recordStateWrite");
+function adoptDeltaStamp(state, delta, key) {
+  var _a;
+  const stamp = (_a = deltaStamps.get(delta)) == null ? void 0 : _a.get(key);
+  if (stamp !== void 0) {
+    stampsFor(stateStamps, state).set(key, stamp);
+  }
+}
+__name(adoptDeltaStamp, "adoptDeltaStamp");
+function shouldApplyDeltaWrite(state, delta, key) {
+  var _a, _b;
+  const written = (_a = deltaStamps.get(delta)) == null ? void 0 : _a.get(key);
+  const current = (_b = stateStamps.get(state)) == null ? void 0 : _b.get(key);
+  if (written !== void 0 && current !== void 0 && current > written) {
+    return false;
+  }
+  stampsFor(stateStamps, state).set(key, written != null ? written : ++writeClock);
+  return true;
+}
+__name(shouldApplyDeltaWrite, "shouldApplyDeltaWrite");
+function carryDeltaStamp(from, to, key) {
+  var _a;
+  const stamp = (_a = deltaStamps.get(from)) == null ? void 0 : _a.get(key);
+  if (stamp !== void 0) {
+    stampsFor(deltaStamps, to).set(key, stamp);
+  }
+}
+__name(carryDeltaStamp, "carryDeltaStamp");
+function carryDeltaStamps(from, to) {
+  const stamps = deltaStamps.get(from);
+  if (!stamps) {
+    return;
+  }
+  const carried = stampsFor(deltaStamps, to);
+  for (const [key, stamp] of stamps) {
+    carried.set(key, stamp);
+  }
+}
+__name(carryDeltaStamps, "carryDeltaStamps");
+export {
+  adoptDeltaStamp,
+  carryDeltaStamp,
+  carryDeltaStamps,
+  recordStateWrite,
+  shouldApplyDeltaWrite
+};