@google/adk 2.0.0 → 2.1.0

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

@@ -1,216 +0,0 @@
-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--)
-    if (decorator = decorators[i])
-      result = (kind ? decorator(target, key, result) : decorator(result)) || result;
-  if (kind && result) __defProp(target, key, result);
-  return result;
-};
-/**
- * @license
- * Copyright 2026 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { Entity, JsonType, PrimaryKey, Property } from "@mikro-orm/core";
-import {
-  transformToCamelCaseEvent,
-  transformToSnakeCaseEvent
-} from "../../events/event.js";
-const SCHEMA_VERSION_KEY = "schema_version";
-const SCHEMA_VERSION_1_JSON = "1";
-const STORAGE_KEY_COLUMN_LENGTH = 191;
-const _CamelCaseToSnakeCaseJsonType = class _CamelCaseToSnakeCaseJsonType extends JsonType {
-  convertToDatabaseValue(value) {
-    return JSON.stringify(transformToSnakeCaseEvent(value));
-  }
-  convertToJSValue(value) {
-    if (typeof value === "string") {
-      return transformToCamelCaseEvent(JSON.parse(value));
-    }
-    return transformToCamelCaseEvent(value);
-  }
-};
-__name(_CamelCaseToSnakeCaseJsonType, "CamelCaseToSnakeCaseJsonType");
-let CamelCaseToSnakeCaseJsonType = _CamelCaseToSnakeCaseJsonType;
-let StorageMetadata = class {
-};
-__name(StorageMetadata, "StorageMetadata");
-__decorateClass([
-  PrimaryKey({ type: "string" })
-], StorageMetadata.prototype, "key", 2);
-__decorateClass([
-  Property({ type: "string" })
-], StorageMetadata.prototype, "value", 2);
-StorageMetadata = __decorateClass([
-  Entity({ tableName: "adk_internal_metadata" })
-], StorageMetadata);
-let StorageAppState = class {
-  constructor() {
-    this.updateTime = /* @__PURE__ */ new Date();
-  }
-};
-__name(StorageAppState, "StorageAppState");
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "app_name",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageAppState.prototype, "appName", 2);
-__decorateClass([
-  Property({ type: "json" })
-], StorageAppState.prototype, "state", 2);
-__decorateClass([
-  Property({
-    type: "datetime",
-    fieldName: "update_time",
-    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate"),
-    onUpdate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onUpdate")
-  })
-], StorageAppState.prototype, "updateTime", 2);
-StorageAppState = __decorateClass([
-  Entity({ tableName: "app_states" })
-], StorageAppState);
-PrimaryKey.name;
-let StorageUserState = class {
-  constructor() {
-    this.updateTime = /* @__PURE__ */ new Date();
-  }
-};
-__name(StorageUserState, "StorageUserState");
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "app_name",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageUserState.prototype, "appName", 2);
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "user_id",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageUserState.prototype, "userId", 2);
-__decorateClass([
-  Property({ type: "json" })
-], StorageUserState.prototype, "state", 2);
-__decorateClass([
-  Property({
-    type: "datetime",
-    fieldName: "update_time",
-    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate"),
-    onUpdate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onUpdate")
-  })
-], StorageUserState.prototype, "updateTime", 2);
-StorageUserState = __decorateClass([
-  Entity({ tableName: "user_states" })
-], StorageUserState);
-PrimaryKey.name;
-let StorageSession = class {
-  constructor() {
-    this.createTime = /* @__PURE__ */ new Date();
-    this.updateTime = /* @__PURE__ */ new Date();
-  }
-};
-__name(StorageSession, "StorageSession");
-__decorateClass([
-  PrimaryKey({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
-], StorageSession.prototype, "id", 2);
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "app_name",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageSession.prototype, "appName", 2);
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "user_id",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageSession.prototype, "userId", 2);
-__decorateClass([
-  Property({ type: "json" })
-], StorageSession.prototype, "state", 2);
-__decorateClass([
-  Property({
-    type: "datetime",
-    fieldName: "create_time",
-    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate")
-  })
-], StorageSession.prototype, "createTime", 2);
-__decorateClass([
-  Property({
-    type: "datetime",
-    fieldName: "update_time",
-    onCreate: /* @__PURE__ */ __name(() => /* @__PURE__ */ new Date(), "onCreate")
-  })
-], StorageSession.prototype, "updateTime", 2);
-StorageSession = __decorateClass([
-  Entity({ tableName: "sessions" })
-], StorageSession);
-PrimaryKey.name;
-let StorageEvent = class {
-};
-__name(StorageEvent, "StorageEvent");
-__decorateClass([
-  PrimaryKey({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
-], StorageEvent.prototype, "id", 2);
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "app_name",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageEvent.prototype, "appName", 2);
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "user_id",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageEvent.prototype, "userId", 2);
-__decorateClass([
-  PrimaryKey({
-    type: "string",
-    fieldName: "session_id",
-    length: STORAGE_KEY_COLUMN_LENGTH
-  })
-], StorageEvent.prototype, "sessionId", 2);
-__decorateClass([
-  Property({ type: "string", fieldName: "invocation_id" })
-], StorageEvent.prototype, "invocationId", 2);
-__decorateClass([
-  Property({ type: "datetime" })
-], StorageEvent.prototype, "timestamp", 2);
-__decorateClass([
-  Property({ type: CamelCaseToSnakeCaseJsonType, fieldName: "event_data" })
-], StorageEvent.prototype, "eventData", 2);
-StorageEvent = __decorateClass([
-  Entity({ tableName: "events" })
-], StorageEvent);
-const ENTITIES = [
-  StorageMetadata,
-  StorageAppState,
-  StorageUserState,
-  StorageSession,
-  StorageEvent
-];
-export {
-  ENTITIES,
-  SCHEMA_VERSION_1_JSON,
-  SCHEMA_VERSION_KEY,
-  STORAGE_KEY_COLUMN_LENGTH,
-  StorageAppState,
-  StorageEvent,
-  StorageMetadata,
-  StorageSession,
-  StorageUserState
-};

package/dist/web/sessions/in_memory_session_service.js

@@ -1,251 +0,0 @@
-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
- */
-import { cloneDeep } from "lodash-es";
-import { randomUUID } from "../utils/env_aware_utils.js";
-import { logger } from "../utils/logger.js";
-import {
-  BaseSessionService,
-  mergeStates,
-  trimTempState
-} from "./base_session_service.js";
-import { createSession } from "./session.js";
-import { State } from "./state.js";
-function isInMemoryConnectionString(uri) {
-  return uri === "memory://";
-}
-__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 = /* @__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 = /* @__PURE__ */ Object.create(null);
-    /**
-     * A map from app name to a map from key to the value.
-     */
-    this.appState = /* @__PURE__ */ Object.create(null);
-  }
-  async createSession({
-    appName,
-    userId,
-    state,
-    sessionId
-  }) {
-    var _a;
-    const filteredState = state ? trimTempState(state) : void 0;
-    const session = createSession({
-      id: sessionId || randomUUID(),
-      appName,
-      userId,
-      state: filteredState,
-      events: [],
-      lastUpdateTime: Date.now()
-    });
-    if (!this.sessions[appName]) {
-      this.sessions[appName] = /* @__PURE__ */ Object.create(null);
-    }
-    if (!this.sessions[appName][userId]) {
-      this.sessions[appName][userId] = /* @__PURE__ */ Object.create(null);
-    }
-    this.sessions[appName][userId][session.id] = session;
-    const copiedSession = cloneDeep(session);
-    copiedSession.state = mergeStates(
-      this.appState[appName],
-      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
-      copiedSession.state
-    );
-    return copiedSession;
-  }
-  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);
-    }
-    const session = this.sessions[appName][userId][sessionId];
-    const copiedSession = cloneDeep(session);
-    if (config) {
-      if (config.numRecentEvents) {
-        copiedSession.events = copiedSession.events.slice(
-          -config.numRecentEvents
-        );
-      }
-      if (config.afterTimestamp) {
-        let i = copiedSession.events.length - 1;
-        while (i >= 0) {
-          if (copiedSession.events[i].timestamp < config.afterTimestamp) {
-            break;
-          }
-          i--;
-        }
-        if (i >= 0) {
-          copiedSession.events = copiedSession.events.slice(i + 1);
-        }
-      }
-    }
-    copiedSession.state = mergeStates(
-      this.appState[appName],
-      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
-      copiedSession.state
-    );
-    return copiedSession;
-  }
-  listSessions({
-    appName,
-    userId,
-    limit,
-    offset,
-    page,
-    order
-  }) {
-    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;
-        return Promise.resolve({
-          sessions: [],
-          page: effectivePage2,
-          limit,
-          totalItems: 0,
-          totalPages: 0
-        });
-      }
-      return Promise.resolve({
-        sessions: [],
-        page: 1,
-        limit: 0,
-        totalItems: 0,
-        totalPages: 0
-      });
-    }
-    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(
-        (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)
-      );
-    }
-    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,
-    userId,
-    sessionId
-  }) {
-    const session = await this.getSession({ appName, userId, sessionId });
-    if (!session) {
-      return;
-    }
-    delete this.sessions[appName][userId][sessionId];
-  }
-  async appendEvent({
-    session,
-    event
-  }) {
-    await super.appendEvent({ session, event });
-    session.lastUpdateTime = event.timestamp;
-    const appName = session.appName;
-    const userId = session.userId;
-    const sessionId = session.id;
-    const warning = /* @__PURE__ */ __name((message) => {
-      logger.warn("Failed to append event to session ".concat(sessionId, ": ").concat(message));
-    }, "warning");
-    if (!this.sessions[appName]) {
-      warning("appName ".concat(appName, " not in sessions"));
-      return event;
-    }
-    if (!this.sessions[appName][userId]) {
-      warning("userId ".concat(userId, " not in sessions[appName]"));
-      return event;
-    }
-    if (!this.sessions[appName][userId][sessionId]) {
-      warning("sessionId ".concat(sessionId, " not in sessions[appName][userId]"));
-      return event;
-    }
-    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] || /* @__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] || /* @__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];
-        }
-      }
-    }
-    const storageSession = this.sessions[appName][userId][sessionId];
-    await super.appendEvent({ session: storageSession, event });
-    storageSession.lastUpdateTime = event.timestamp;
-    return event;
-  }
-};
-__name(_InMemorySessionService, "InMemorySessionService");
-let InMemorySessionService = _InMemorySessionService;
-export {
-  InMemorySessionService,
-  isInMemoryConnectionString
-};

package/dist/web/sessions/registry.js

@@ -1,38 +0,0 @@
-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
-} from "./database_session_service.js";
-import {
-  InMemorySessionService,
-  isInMemoryConnectionString
-} from "./in_memory_session_service.js";
-import {
-  VertexAiSessionService,
-  isVertexAiConnectionString
-} from "./vertex_ai_session_service.js";
-function getSessionServiceFromUri(uri) {
-  if (isInMemoryConnectionString(uri)) {
-    return new InMemorySessionService();
-  }
-  if (isDatabaseConnectionString(uri)) {
-    return new DatabaseSessionService(uri);
-  }
-  if (isVertexAiConnectionString(uri)) {
-    return new VertexAiSessionService({});
-  }
-  throw new Error("Unsupported session service URI: ".concat(redactUriPassword(uri)));
-}
-__name(getSessionServiceFromUri, "getSessionServiceFromUri");
-export {
-  getSessionServiceFromUri
-};

package/dist/web/sessions/session.js

@@ -1,23 +0,0 @@
-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
- */
-function createSession(params) {
-  return {
-    id: params.id,
-    appName: params.appName,
-    userId: params.userId || "",
-    state: params.state || {},
-    events: params.events || [],
-    lastUpdateTime: params.lastUpdateTime || 0
-  };
-}
-__name(createSession, "createSession");
-export {
-  createSession
-};

package/dist/web/sessions/state.js

@@ -1,159 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-var __defProp = Object.defineProperty;
-var __getOwnPropSymbols = Object.getOwnPropertySymbols;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __propIsEnum = Object.prototype.propertyIsEnumerable;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues = (a, b) => {
-  for (var prop in b || (b = {}))
-    if (__hasOwnProp.call(b, prop))
-      __defNormalProp(a, prop, b[prop]);
-  if (__getOwnPropSymbols)
-    for (var prop of __getOwnPropSymbols(b)) {
-      if (__propIsEnum.call(b, prop))
-        __defNormalProp(a, prop, b[prop]);
-    }
-  return a;
-};
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-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 '".concat(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 '".concat(key, "' does not match the type declared in the state ") + "schema: ".concat(parsed.error.message)
-      );
-    }
-  }
-  /**
-   * Returns the value of the state dict for the given key.
-   *
-   * @param key The key to get the value for.
-   * @param defaultValue The default value to return if the key is not found.
-   * @return The value of the state for the given key, or the default value if
-   *     not found.
-   */
-  get(key, defaultValue) {
-    if (key in this.delta) {
-      return this.delta[key];
-    }
-    if (key in this.value) {
-      return this.value[key];
-    }
-    return defaultValue;
-  }
-  /**
-   * Sets the value of the state dict for the given key.
-   *
-   * @param key The key to set the value for.
-   * @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.
-   */
-  has(key) {
-    return key in this.value || key in this.delta;
-  }
-  /**
-   * Whether the state has pending delta.
-   */
-  hasDelta() {
-    return Object.keys(this.delta).length > 0;
-  }
-  /**
-   * Updates the state dict with the given delta.
-   *
-   * @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.
-   */
-  toRecord() {
-    return __spreadValues(__spreadValues({}, this.value), this.delta);
-  }
-};
-__name(_State, "State");
-_State.APP_PREFIX = "app:";
-_State.USER_PREFIX = "user:";
-_State.TEMP_PREFIX = "temp:";
-let State = _State;
-export {
-  State,
-  StateSchemaError,
-  isStateSchemaError
-};