@google/adk 1.6.0 → 2.1.0

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

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -22,17 +23,10 @@ var __copyProps = (to, from, except, desc) => {
   return to;
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-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--)
-    if (decorator = decorators[i])
-      result = (kind ? decorator(target, key, result) : decorator(result)) || result;
-  if (kind && result) __defProp(target, key, result);
-  return result;
-};
 var schema_exports = {};
 __export(schema_exports, {
   ENTITIES: () => ENTITIES,
+  EVENT_TIMESTAMP_PRECISION: () => EVENT_TIMESTAMP_PRECISION,
   SCHEMA_VERSION_1_JSON: () => SCHEMA_VERSION_1_JSON,
   SCHEMA_VERSION_KEY: () => SCHEMA_VERSION_KEY,
   STORAGE_KEY_COLUMN_LENGTH: () => STORAGE_KEY_COLUMN_LENGTH,
@@ -40,7 +34,12 @@ __export(schema_exports, {
   StorageEvent: () => StorageEvent,
   StorageMetadata: () => StorageMetadata,
   StorageSession: () => StorageSession,
-  StorageUserState: () => StorageUserState
+  StorageUserState: () => StorageUserState,
+  storageAppStateSchema: () => storageAppStateSchema,
+  storageEventSchema: () => storageEventSchema,
+  storageMetadataSchema: () => storageMetadataSchema,
+  storageSessionSchema: () => storageSessionSchema,
+  storageUserStateSchema: () => storageUserStateSchema
 });
 module.exports = __toCommonJS(schema_exports);
 var import_core = require("@mikro-orm/core");
@@ -53,7 +52,8 @@ 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 {
+const EVENT_TIMESTAMP_PRECISION = 6;
+const _CamelCaseToSnakeCaseJsonType = class _CamelCaseToSnakeCaseJsonType extends import_core.JsonType {
   convertToDatabaseValue(value) {
     return JSON.stringify((0, import_event.transformToSnakeCaseEvent)(value));
   }
@@ -63,171 +63,128 @@ class CamelCaseToSnakeCaseJsonType extends import_core.JsonType {
     }
     return (0, import_event.transformToCamelCaseEvent)(value);
   }
-}
-let StorageMetadata = class {
 };
-__decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string" })
-], StorageMetadata.prototype, "key", 2);
-__decorateClass([
-  (0, import_core.Property)({ type: "string" })
-], StorageMetadata.prototype, "value", 2);
-StorageMetadata = __decorateClass([
-  (0, import_core.Entity)({ tableName: "adk_internal_metadata" })
-], StorageMetadata);
-let StorageAppState = class {
+__name(_CamelCaseToSnakeCaseJsonType, "CamelCaseToSnakeCaseJsonType");
+let CamelCaseToSnakeCaseJsonType = _CamelCaseToSnakeCaseJsonType;
+const KEY_COLUMN = {
+  type: "string",
+  length: STORAGE_KEY_COLUMN_LENGTH,
+  primary: true
+};
+const _StorageMetadata = class _StorageMetadata {
+};
+__name(_StorageMetadata, "StorageMetadata");
+let StorageMetadata = _StorageMetadata;
+const storageMetadataSchema = new import_core.EntitySchema({
+  class: StorageMetadata,
+  tableName: "adk_internal_metadata",
+  properties: {
+    key: { type: "string", primary: true },
+    value: { type: "string" }
+  }
+});
+const _StorageAppState = class _StorageAppState {
   constructor() {
     this.updateTime = /* @__PURE__ */ new Date();
   }
 };
-__decorateClass([
-  (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" })
-], StorageAppState.prototype, "state", 2);
-__decorateClass([
-  (0, import_core.Property)({
-    type: "datetime",
-    fieldName: "update_time",
-    onCreate: () => /* @__PURE__ */ new Date(),
-    onUpdate: () => /* @__PURE__ */ new Date()
-  })
-], StorageAppState.prototype, "updateTime", 2);
-StorageAppState = __decorateClass([
-  (0, import_core.Entity)({ tableName: "app_states" })
-], StorageAppState);
-import_core.PrimaryKey.name;
-let StorageUserState = class {
+__name(_StorageAppState, "StorageAppState");
+let StorageAppState = _StorageAppState;
+const storageAppStateSchema = new import_core.EntitySchema({
+  class: StorageAppState,
+  tableName: "app_states",
+  properties: {
+    appName: { ...KEY_COLUMN, fieldName: "app_name" },
+    state: { type: "json" },
+    updateTime: {
+      type: "datetime",
+      fieldName: "update_time",
+      onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate"),
+      onUpdate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onUpdate")
+    }
+  }
+});
+import_core.PrimaryKeyProp;
+const _StorageUserState = class _StorageUserState {
   constructor() {
     this.updateTime = /* @__PURE__ */ new Date();
   }
 };
-__decorateClass([
-  (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",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageUserState.prototype, "userId", 2);
-__decorateClass([
-  (0, import_core.Property)({ type: "json" })
-], StorageUserState.prototype, "state", 2);
-__decorateClass([
-  (0, import_core.Property)({
-    type: "datetime",
-    fieldName: "update_time",
-    onCreate: () => /* @__PURE__ */ new Date(),
-    onUpdate: () => /* @__PURE__ */ new Date()
-  })
-], StorageUserState.prototype, "updateTime", 2);
-StorageUserState = __decorateClass([
-  (0, import_core.Entity)({ tableName: "user_states" })
-], StorageUserState);
-import_core.PrimaryKey.name;
-let StorageSession = class {
+__name(_StorageUserState, "StorageUserState");
+let StorageUserState = _StorageUserState;
+const storageUserStateSchema = new import_core.EntitySchema({
+  class: StorageUserState,
+  tableName: "user_states",
+  properties: {
+    appName: { ...KEY_COLUMN, fieldName: "app_name" },
+    userId: { ...KEY_COLUMN, fieldName: "user_id" },
+    state: { type: "json" },
+    updateTime: {
+      type: "datetime",
+      fieldName: "update_time",
+      onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate"),
+      onUpdate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onUpdate")
+    }
+  }
+});
+import_core.PrimaryKeyProp;
+const _StorageSession = class _StorageSession {
   constructor() {
     this.createTime = /* @__PURE__ */ new Date();
     this.updateTime = /* @__PURE__ */ new Date();
   }
 };
-__decorateClass([
-  (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",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageSession.prototype, "appName", 2);
-__decorateClass([
-  (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" })
-], StorageSession.prototype, "state", 2);
-__decorateClass([
-  (0, import_core.Property)({
-    type: "datetime",
-    fieldName: "create_time",
-    onCreate: () => /* @__PURE__ */ new Date()
-  })
-], StorageSession.prototype, "createTime", 2);
-__decorateClass([
-  (0, import_core.Property)({
-    type: "datetime",
-    fieldName: "update_time",
-    onCreate: () => /* @__PURE__ */ new Date()
-  })
-], StorageSession.prototype, "updateTime", 2);
-StorageSession = __decorateClass([
-  (0, import_core.Entity)({ tableName: "sessions" })
-], StorageSession);
-import_core.PrimaryKey.name;
-let StorageEvent = class {
+__name(_StorageSession, "StorageSession");
+let StorageSession = _StorageSession;
+const storageSessionSchema = new import_core.EntitySchema({
+  class: StorageSession,
+  tableName: "sessions",
+  properties: {
+    id: KEY_COLUMN,
+    appName: { ...KEY_COLUMN, fieldName: "app_name" },
+    userId: { ...KEY_COLUMN, fieldName: "user_id" },
+    state: { type: "json" },
+    createTime: {
+      type: "datetime",
+      fieldName: "create_time",
+      onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate")
+    },
+    updateTime: {
+      type: "datetime",
+      fieldName: "update_time",
+      onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate")
+    }
+  }
+});
+import_core.PrimaryKeyProp;
+const _StorageEvent = class _StorageEvent {
 };
-__decorateClass([
-  (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",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageEvent.prototype, "appName", 2);
-__decorateClass([
-  (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",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageEvent.prototype, "sessionId", 2);
-__decorateClass([
-  (0, import_core.Property)({ type: "string", fieldName: "invocation_id" })
-], StorageEvent.prototype, "invocationId", 2);
-__decorateClass([
-  (0, import_core.Property)({ type: "datetime" })
-], StorageEvent.prototype, "timestamp", 2);
-__decorateClass([
-  (0, import_core.Property)({ type: CamelCaseToSnakeCaseJsonType, fieldName: "event_data" })
-], StorageEvent.prototype, "eventData", 2);
-StorageEvent = __decorateClass([
-  (0, import_core.Entity)({ tableName: "events" })
-], StorageEvent);
+__name(_StorageEvent, "StorageEvent");
+let StorageEvent = _StorageEvent;
+const storageEventSchema = new import_core.EntitySchema({
+  class: StorageEvent,
+  tableName: "events",
+  properties: {
+    id: KEY_COLUMN,
+    appName: { ...KEY_COLUMN, fieldName: "app_name" },
+    userId: { ...KEY_COLUMN, fieldName: "user_id" },
+    sessionId: { ...KEY_COLUMN, fieldName: "session_id" },
+    invocationId: { type: "string", fieldName: "invocation_id" },
+    timestamp: { type: "datetime", length: EVENT_TIMESTAMP_PRECISION },
+    eventData: { type: CamelCaseToSnakeCaseJsonType, fieldName: "event_data" }
+  }
+});
 const ENTITIES = [
-  StorageMetadata,
-  StorageAppState,
-  StorageUserState,
-  StorageSession,
-  StorageEvent
+  storageMetadataSchema,
+  storageAppStateSchema,
+  storageUserStateSchema,
+  storageSessionSchema,
+  storageEventSchema
 ];
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ENTITIES,
+  EVENT_TIMESTAMP_PRECISION,
   SCHEMA_VERSION_1_JSON,
   SCHEMA_VERSION_KEY,
   STORAGE_KEY_COLUMN_LENGTH,
@@ -235,5 +192,10 @@ const ENTITIES = [
   StorageEvent,
   StorageMetadata,
   StorageSession,
-  StorageUserState
+  StorageUserState,
+  storageAppStateSchema,
+  storageEventSchema,
+  storageMetadataSchema,
+  storageSessionSchema,
+  storageUserStateSchema
 });

package/dist/cjs/sessions/in_memory_session_service.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -42,7 +43,8 @@ var import_state = require("./state.js");
 function isInMemoryConnectionString(uri) {
   return uri === "memory://";
 }
-class InMemorySessionService extends import_base_session_service.BaseSessionService {
+__name(isInMemoryConnectionString, "isInMemoryConnectionString");
+const _InMemorySessionService = class _InMemorySessionService extends import_base_session_service.BaseSessionService {
   constructor() {
     super(...arguments);
     /**
@@ -136,7 +138,9 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     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;
@@ -156,15 +160,17 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
         totalPages: 0
       });
     }
-    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
-      })
+    const all = sessionsByUser.flatMap(
+      (sessionsById) => Object.values(sessionsById).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(
@@ -226,9 +232,9 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     const appName = session.appName;
     const userId = session.userId;
     const sessionId = session.id;
-    const warning = (message) => {
+    const warning = /* @__PURE__ */ __name((message) => {
       import_logger.logger.warn(`Failed to append event to session ${sessionId}: ${message}`);
-    };
+    }, "warning");
     if (!this.sessions[appName]) {
       warning(`appName ${appName} not in sessions`);
       return event;
@@ -259,7 +265,9 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     storageSession.lastUpdateTime = event.timestamp;
     return event;
   }
-}
+};
+__name(_InMemorySessionService, "InMemorySessionService");
+let InMemorySessionService = _InMemorySessionService;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   InMemorySessionService,

package/dist/cjs/sessions/registry.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -27,6 +28,7 @@ __export(registry_exports, {
   getSessionServiceFromUri: () => getSessionServiceFromUri
 });
 module.exports = __toCommonJS(registry_exports);
+var import_redact_uri = require("../utils/redact_uri.js");
 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");
@@ -43,10 +45,15 @@ function getSessionServiceFromUri(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({});
+    return new import_vertex_ai_session_service.VertexAiSessionService({
+      projectId: process.env.GOOGLE_CLOUD_PROJECT,
+      location: process.env.GOOGLE_CLOUD_LOCATION,
+      agentEngineId: process.env.GOOGLE_CLOUD_AGENT_ENGINE_ID
+    });
   }
-  throw new Error(`Unsupported session service URI: ${uri}`);
+  throw new Error(`Unsupported session service URI: ${(0, import_redact_uri.redactUriPassword)(uri)}`);
 }
+__name(getSessionServiceFromUri, "getSessionServiceFromUri");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   getSessionServiceFromUri

package/dist/cjs/sessions/session.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -42,6 +43,7 @@ function createSession(params) {
     lastUpdateTime: params.lastUpdateTime || 0
   };
 }
+__name(createSession, "createSession");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   createSession

package/dist/cjs/sessions/state.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -24,18 +25,79 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var state_exports = {};
 __export(state_exports, {
-  State: () => State
+  State: () => State,
+  StateSchemaError: () => StateSchemaError,
+  isStateSchemaError: () => isStateSchemaError
 });
 module.exports = __toCommonJS(state_exports);
+var import_schema = require("../utils/schema.js");
+var import_state_write_order = require("./state_write_order.js");
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class State {
-  constructor(value = {}, delta = {}) {
+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 = (0, import_schema.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.
@@ -61,8 +123,10 @@ class State {
    * @param value The value to set.
    */
   set(key, value) {
+    this.validate(key, value);
     this.value[key] = value;
     this.delta[key] = value;
+    (0, import_state_write_order.recordStateWrite)(this.value, this.delta, key);
   }
   /**
    * Whether the state has pending delta.
@@ -82,8 +146,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)) {
+      (0, import_state_write_order.recordStateWrite)(this.value, this.delta, key);
+    }
   }
   /**
    * Returns the state as a plain JSON object.
@@ -91,11 +159,15 @@ 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;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  State
+  State,
+  StateSchemaError,
+  isStateSchemaError
 });