@google/adk 0.3.0 → 0.4.0

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

@@ -0,0 +1,172 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+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";
+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);
+  }
+}
+let StorageMetadata = class {
+};
+__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();
+  }
+};
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "app_name" })
+], StorageAppState.prototype, "appName", 2);
+__decorateClass([
+  Property({ type: "json" })
+], StorageAppState.prototype, "state", 2);
+__decorateClass([
+  Property({
+    type: "datetime",
+    fieldName: "update_time",
+    onCreate: () => /* @__PURE__ */ new Date(),
+    onUpdate: () => /* @__PURE__ */ new Date()
+  })
+], StorageAppState.prototype, "updateTime", 2);
+StorageAppState = __decorateClass([
+  Entity({ tableName: "app_states" })
+], StorageAppState);
+PrimaryKey.name;
+let StorageUserState = class {
+  constructor() {
+    this.updateTime = /* @__PURE__ */ new Date();
+  }
+};
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "app_name" })
+], StorageUserState.prototype, "appName", 2);
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "user_id" })
+], StorageUserState.prototype, "userId", 2);
+__decorateClass([
+  Property({ type: "json" })
+], StorageUserState.prototype, "state", 2);
+__decorateClass([
+  Property({
+    type: "datetime",
+    fieldName: "update_time",
+    onCreate: () => /* @__PURE__ */ new Date(),
+    onUpdate: () => /* @__PURE__ */ new Date()
+  })
+], 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();
+  }
+};
+__decorateClass([
+  PrimaryKey({ type: "string" })
+], StorageSession.prototype, "id", 2);
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "app_name" })
+], StorageSession.prototype, "appName", 2);
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "user_id" })
+], StorageSession.prototype, "userId", 2);
+__decorateClass([
+  Property({ type: "json" })
+], StorageSession.prototype, "state", 2);
+__decorateClass([
+  Property({
+    type: "datetime",
+    fieldName: "create_time",
+    onCreate: () => /* @__PURE__ */ new Date()
+  })
+], StorageSession.prototype, "createTime", 2);
+__decorateClass([
+  Property({
+    type: "datetime",
+    fieldName: "update_time",
+    onCreate: () => /* @__PURE__ */ new Date()
+  })
+], StorageSession.prototype, "updateTime", 2);
+StorageSession = __decorateClass([
+  Entity({ tableName: "sessions" })
+], StorageSession);
+PrimaryKey.name;
+let StorageEvent = class {
+};
+__decorateClass([
+  PrimaryKey({ type: "string" })
+], StorageEvent.prototype, "id", 2);
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "app_name" })
+], StorageEvent.prototype, "appName", 2);
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "user_id" })
+], StorageEvent.prototype, "userId", 2);
+__decorateClass([
+  PrimaryKey({ type: "string", fieldName: "session_id" })
+], 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,
+  StorageAppState,
+  StorageEvent,
+  StorageMetadata,
+  StorageSession,
+  StorageUserState
+};

package/dist/web/sessions/in_memory_session_service.js

@@ -7,10 +7,14 @@ import { cloneDeep } from "lodash-es";
 import { randomUUID } from "../utils/env_aware_utils.js";
 import { logger } from "../utils/logger.js";
 import {
-  BaseSessionService
+  BaseSessionService,
+  mergeStates
 } from "./base_session_service.js";
 import { createSession } from "./session.js";
 import { State } from "./state.js";
+function isInMemoryConnectionString(uri) {
+  return uri === "memory://";
+}
 class InMemorySessionService extends BaseSessionService {
   constructor() {
     super(...arguments);
@@ -28,12 +32,13 @@ class InMemorySessionService extends BaseSessionService {
      */
     this.appState = {};
   }
-  createSession({
+  async createSession({
     appName,
     userId,
     state,
     sessionId
   }) {
+    var _a;
     const session = createSession({
       id: sessionId || randomUUID(),
       appName,
@@ -49,16 +54,21 @@ class InMemorySessionService extends BaseSessionService {
       this.sessions[appName][userId] = {};
     }
     this.sessions[appName][userId][session.id] = session;
-    return Promise.resolve(
-      this.mergeState(appName, userId, cloneDeep(session))
+    const copiedSession = cloneDeep(session);
+    copiedSession.state = mergeStates(
+      this.appState[appName],
+      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
+      copiedSession.state
     );
+    return copiedSession;
   }
-  getSession({
+  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 +93,12 @@ class InMemorySessionService extends BaseSessionService {
         }
       }
     }
-    return Promise.resolve(this.mergeState(appName, userId, copiedSession));
+    copiedSession.state = mergeStates(
+      this.appState[appName],
+      (_a = this.userState[appName]) == null ? void 0 : _a[userId],
+      copiedSession.state
+    );
+    return copiedSession;
   }
   listSessions({
     appName,
@@ -160,21 +175,8 @@ class InMemorySessionService extends BaseSessionService {
     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[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[State.USER_PREFIX + key] = this.userState[appName][userId][key];
-    }
-    return copiedSession;
-  }
 }
 export {
-  InMemorySessionService
+  InMemorySessionService,
+  isInMemoryConnectionString
 };

package/dist/web/sessions/registry.js

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {
+  DatabaseSessionService,
+  isDatabaseConnectionString
+} from "./database_session_service.js";
+import {
+  InMemorySessionService,
+  isInMemoryConnectionString
+} from "./in_memory_session_service.js";
+function getSessionServiceFromUri(uri) {
+  if (isInMemoryConnectionString(uri)) {
+    return new InMemorySessionService();
+  }
+  if (isDatabaseConnectionString(uri)) {
+    return new DatabaseSessionService(uri);
+  }
+  throw new Error("Unsupported session service URI: ".concat(uri));
+}
+export {
+  getSessionServiceFromUri
+};

package/dist/web/tools/forwarding_artifact_service.js

@@ -35,6 +35,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
+    );
+  }
 }
 export {
   ForwardingArtifactService

package/dist/web/utils/env_aware_utils.js

@@ -40,9 +40,17 @@ function isBase64Encoded(data) {
     return false;
   }
 }
+function getBooleanEnvVar(envVar) {
+  if (!process.env) {
+    return false;
+  }
+  const envVarValue = (process.env[envVar] || "").toLowerCase();
+  return ["true", "1"].includes(envVarValue);
+}
 export {
   base64Decode,
   base64Encode,
+  getBooleanEnvVar,
   isBase64Encoded,
   isBrowser,
   randomUUID

package/dist/web/utils/object_notation_utils.js

@@ -0,0 +1,47 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function toCamelCase(obj, preserveKeys = []) {
+  return toNotation(obj, toCamelCaseKey, "", preserveKeys);
+}
+function toSnakeCase(obj, preserveKeys = []) {
+  return toNotation(obj, toSnakeCaseKey, "", preserveKeys);
+}
+const toCamelCaseKey = (key) => key.replace(
+  /_([a-z])/g,
+  (_match, letter) => letter.toUpperCase()
+);
+const toSnakeCaseKey = (key) => key.replace(/[A-Z]/g, (g) => "_" + g.toLowerCase());
+function toNotation(obj, converter, parentKey = "", preserveKeys = []) {
+  if (Array.isArray(obj)) {
+    return obj.map(
+      (item) => toNotation(item, converter, parentKey, preserveKeys)
+    );
+  }
+  if (typeof obj === "object" && obj !== null) {
+    const source = obj;
+    const result = {};
+    for (const key of Object.keys(source)) {
+      const convertedKey = converter(key);
+      const fullPath = parentKey !== "" ? parentKey + "." + key : key;
+      if (preserveKeys.includes(fullPath)) {
+        result[convertedKey] = source[key];
+      } else {
+        result[convertedKey] = toNotation(
+          source[key],
+          converter,
+          fullPath,
+          preserveKeys
+        );
+      }
+    }
+    return result;
+  }
+  return obj;
+}
+export {
+  toCamelCase,
+  toSnakeCase
+};

package/dist/web/utils/variant_utils.js

@@ -3,6 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { getBooleanEnvVar } from "./env_aware_utils.js";
 var GoogleLLMVariant = /* @__PURE__ */ ((GoogleLLMVariant2) => {
   GoogleLLMVariant2["VERTEX_AI"] = "VERTEX_AI";
   GoogleLLMVariant2["GEMINI_API"] = "GEMINI_API";
@@ -11,13 +12,6 @@ var GoogleLLMVariant = /* @__PURE__ */ ((GoogleLLMVariant2) => {
 function getGoogleLlmVariant() {
   return getBooleanEnvVar("GOOGLE_GENAI_USE_VERTEXAI") ? "VERTEX_AI" /* VERTEX_AI */ : "GEMINI_API" /* GEMINI_API */;
 }
-function getBooleanEnvVar(envVar) {
-  if (!process.env) {
-    return false;
-  }
-  const envVarValue = (process.env[envVar] || "").toLowerCase();
-  return ["true", "1"].includes(envVarValue);
-}
 export {
   GoogleLLMVariant,
   getGoogleLlmVariant

package/dist/web/version.js

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const version = "0.3.0";
+const version = "0.4.0";
 export {
   version
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@google/adk",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Google ADK JS",
   "author": "Google",
   "license": "Apache-2.0",
@@ -37,11 +37,14 @@
     "test": "vitest",
     "lint": "eslint 'src/**/*.ts'",
     "format": "prettier --write 'src/**/*.ts'",
-    "prepublishOnly": "npm run build:bundle"
+    "prepublishOnly": "npm run build"
   },
   "dependencies": {
+    "@a2a-js/sdk": "^0.3.10",
     "@google/genai": "^1.37.0",
-    "@modelcontextprotocol/sdk": "^1.24.0",
+    "@mikro-orm/core": "^6.6.6",
+    "@mikro-orm/reflection": "^6.6.6",
+    "@modelcontextprotocol/sdk": "^1.26.0",
     "google-auth-library": "^10.3.0",
     "lodash-es": "^4.17.23",
     "zod": "^4.2.1",
@@ -49,12 +52,18 @@
   },
   "devDependencies": {
     "@types/lodash-es": "^4.17.12",
-    "openapi-types": "^12.1.3"
+    "openapi-types": "^12.1.3",
+    "@mikro-orm/sqlite": "^6.6.6"
   },
   "peerDependencies": {
     "@google-cloud/opentelemetry-cloud-monitoring-exporter": "^0.21.0",
     "@google-cloud/opentelemetry-cloud-trace-exporter": "^3.0.0",
     "@google-cloud/storage": "^7.17.1",
+    "@mikro-orm/mariadb": "^6.6.6",
+    "@mikro-orm/mssql": "^6.6.6",
+    "@mikro-orm/mysql": "^6.6.6",
+    "@mikro-orm/postgresql": "^6.6.6",
+    "@mikro-orm/sqlite": "^6.6.6",
     "@opentelemetry/api": "1.9.0",
     "@opentelemetry/api-logs": "^0.205.0",
     "@opentelemetry/exporter-logs-otlp-http": "^0.205.0",