@google/adk 0.6.1 → 1.1.0

package/dist/web/models/routed_llm.js

@@ -0,0 +1,122 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : Symbol.for("Symbol." + name);
+var __typeError = (msg) => {
+  throw TypeError(msg);
+};
+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 __await = function(promise, isYieldStar) {
+  this[0] = promise;
+  this[1] = isYieldStar;
+};
+var __asyncGenerator = (__this, __arguments, generator) => {
+  var resume = (k, v, yes, no) => {
+    try {
+      var x = generator[k](v), isAwait = (v = x.value) instanceof __await, done = x.done;
+      Promise.resolve(isAwait ? v[0] : v).then((y) => isAwait ? resume(k === "return" ? k : "next", v[1] ? { done: y.done, value: y.value } : y, yes, no) : yes({ value: y, done })).catch((e) => resume("throw", e, yes, no));
+    } catch (e) {
+      no(e);
+    }
+  }, method = (k) => it[k] = (x) => new Promise((yes, no) => resume(k, x, yes, no)), it = {};
+  return generator = generator.apply(__this, __arguments), it[__knownSymbol("asyncIterator")] = () => it, method("next"), method("throw"), method("return"), it;
+};
+var __yieldStar = (value) => {
+  var obj = value[__knownSymbol("asyncIterator")], isAwait = false, method, it = {};
+  if (obj == null) {
+    obj = value[__knownSymbol("iterator")]();
+    method = (k) => it[k] = (x) => obj[k](x);
+  } else {
+    obj = obj.call(value);
+    method = (k) => it[k] = (v) => {
+      if (isAwait) {
+        isAwait = false;
+        if (k === "throw") throw v;
+        return v;
+      }
+      isAwait = true;
+      return {
+        done: false,
+        value: new __await(new Promise((resolve) => {
+          var x = obj[k](v);
+          if (!(x instanceof Object)) __typeError("Object expected");
+          resolve(x);
+        }), 1)
+      };
+    };
+  }
+  return it[__knownSymbol("iterator")] = () => it, method("next"), "throw" in obj ? method("throw") : it.throw = (x) => {
+    throw x;
+  }, "return" in obj && method("return"), it;
+};
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseLlm } from "./base_llm.js";
+import { experimental } from "../utils/experimental.js";
+import { runWithRouting } from "../utils/failover_utils.js";
+import { logger } from "../utils/logger.js";
+let RoutedLlm = class extends BaseLlm {
+  constructor({
+    models,
+    router,
+    modelName = "routed-llm"
+  }) {
+    const modelsMap = Array.isArray(models) ? Object.fromEntries(models.map((m) => [m.model, m])) : models;
+    const modelNames = Object.entries(modelsMap).map(
+      ([name, model]) => "".concat(name, " (").concat(model.model, ")")
+    );
+    const computedName = "RoutedLlm[".concat(modelNames.join(", "), "]");
+    super({ model: modelName === "routed-llm" ? computedName : modelName });
+    this.models = modelsMap;
+    this.router = router;
+  }
+  /**
+   * Generates content by delegating to the selected model.
+   */
+  generateContentAsync(llmRequest, stream) {
+    return __asyncGenerator(this, null, function* () {
+      logger.info("Routing request via ".concat(this.model));
+      yield* __yieldStar(runWithRouting(
+        this.models,
+        llmRequest,
+        this.router,
+        (model) => model.generateContentAsync(llmRequest, stream)
+      ));
+    });
+  }
+  /**
+   * Creates a live connection to the LLM by delegating to the selected model.
+   * This live connection cannot be switched mid-stream, it is tied to the model
+   * selected at the time of connection.
+   */
+  async connect(llmRequest) {
+    const generator = runWithRouting(
+      this.models,
+      llmRequest,
+      this.router,
+      (model) => model.connect(llmRequest)
+    );
+    const result = await generator.next();
+    if (result.done || result.value === void 0) {
+      throw new Error("Failed to establish connection: No connection yielded.");
+    }
+    return result.value;
+  }
+};
+RoutedLlm = __decorateClass([
+  experimental
+], RoutedLlm);
+export {
+  RoutedLlm
+};

package/dist/web/plugins/base_plugin.js

@@ -1,8 +1,26 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+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 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { experimental } from "../utils/experimental.js";
+var ContextCompactionTrigger = /* @__PURE__ */ ((ContextCompactionTrigger2) => {
+  ContextCompactionTrigger2["Manual"] = "Manual";
+  ContextCompactionTrigger2["Auto"] = "Auto";
+  return ContextCompactionTrigger2;
+})(ContextCompactionTrigger || {});
 class BasePlugin {
   /**
    * Initializes the plugin.
@@ -157,6 +175,18 @@ class BasePlugin {
   async onModelErrorCallback(params) {
     return;
   }
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async beforeToolSelection(params) {
+    return;
+  }
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async beforeContextCompaction(params) {
+    return;
+  }
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async afterContextCompaction(params) {
+    return;
+  }
   /**
    * Callback executed before a tool is called.
    *
@@ -223,6 +253,16 @@ class BasePlugin {
     return;
   }
 }
+__decorateClass([
+  experimental
+], BasePlugin.prototype, "beforeToolSelection", 1);
+__decorateClass([
+  experimental
+], BasePlugin.prototype, "beforeContextCompaction", 1);
+__decorateClass([
+  experimental
+], BasePlugin.prototype, "afterContextCompaction", 1);
 export {
-  BasePlugin
+  BasePlugin,
+  ContextCompactionTrigger
 };

package/dist/web/plugins/logging_plugin.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/web/plugins/plugin_manager.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -151,6 +153,45 @@ class PluginManager {
       "afterAgentCallback"
     );
   }
+  /**
+   * Runs the `beforeToolSelection` for all plugins.
+   */
+  async runBeforeToolSelection({
+    callbackContext,
+    tools
+  }) {
+    return await this.runCallbacks(
+      this.plugins,
+      (plugin) => plugin.beforeToolSelection({ callbackContext, tools }),
+      "beforeToolSelection"
+    );
+  }
+  /**
+   * Runs the `beforeContextCompaction` for all plugins.
+   */
+  async runBeforeContextCompaction({
+    invocationContext,
+    trigger
+  }) {
+    await this.runCallbacks(
+      this.plugins,
+      (plugin) => plugin.beforeContextCompaction({ invocationContext, trigger }),
+      "beforeContextCompaction"
+    );
+  }
+  /**
+   * Runs the `afterContextCompaction` for all plugins.
+   */
+  async runAfterContextCompaction({
+    invocationContext,
+    trigger
+  }) {
+    await this.runCallbacks(
+      this.plugins,
+      (plugin) => plugin.afterContextCompaction({ invocationContext, trigger }),
+      "afterContextCompaction"
+    );
+  }
   /**
    * Runs the `beforeToolCallback` for all plugins.
    */

package/dist/web/plugins/security_plugin.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/web/runner/in_memory_runner.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/web/runner/runner.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : Symbol.for("Symbol." + name);
 var __typeError = (msg) => {
   throw TypeError(msg);
@@ -148,12 +150,15 @@ class Runner {
           this,
           function() {
             return __asyncGenerator(this, null, function* () {
-              var _a2;
+              var _a2, _b, _c, _d, _e, _f, _g, _h, _i, _j;
               const session = yield new __await(this.sessionService.getSession({
                 appName: this.appName,
                 userId,
                 sessionId
               }));
+              if ((_a2 = params.abortSignal) == null ? void 0 : _a2.aborted) {
+                return;
+              }
               if (!session) {
                 if (!this.appName) {
                   throw new Error(
@@ -183,17 +188,21 @@ class Runner {
                 session,
                 userContent: newMessage,
                 runConfig,
-                pluginManager: this.pluginManager
+                pluginManager: this.pluginManager,
+                abortSignal: params.abortSignal
               });
               const pluginUserMessage = yield new __await(this.pluginManager.runOnUserMessageCallback({
                 userMessage: newMessage,
                 invocationContext
               }));
+              if ((_b = params.abortSignal) == null ? void 0 : _b.aborted) {
+                return;
+              }
               if (pluginUserMessage) {
                 newMessage = pluginUserMessage;
               }
               if (newMessage) {
-                if (!((_a2 = newMessage.parts) == null ? void 0 : _a2.length)) {
+                if (!((_c = newMessage.parts) == null ? void 0 : _c.length)) {
                   throw new Error("No parts in the newMessage.");
                 }
                 if (runConfig.saveInputBlobsAsArtifacts) {
@@ -203,6 +212,9 @@ class Runner {
                     session.id,
                     newMessage
                   ));
+                  if ((_d = params.abortSignal) == null ? void 0 : _d.aborted) {
+                    return;
+                  }
                 }
                 yield new __await(this.sessionService.appendEvent({
                   session,
@@ -213,6 +225,9 @@ class Runner {
                     content: newMessage
                   })
                 }));
+                if ((_e = params.abortSignal) == null ? void 0 : _e.aborted) {
+                  return;
+                }
               }
               invocationContext.agent = this.determineAgentForResumption(
                 session,
@@ -222,6 +237,9 @@ class Runner {
                 const beforeRunCallbackResponse = yield new __await(this.pluginManager.runBeforeRunCallback({
                   invocationContext
                 }));
+                if ((_f = params.abortSignal) == null ? void 0 : _f.aborted) {
+                  return;
+                }
                 if (beforeRunCallbackResponse) {
                   const earlyExitEvent = createEvent({
                     invocationId: invocationContext.invocationId,
@@ -232,6 +250,9 @@ class Runner {
                     session,
                     event: earlyExitEvent
                   }));
+                  if ((_g = params.abortSignal) == null ? void 0 : _g.aborted) {
+                    return;
+                  }
                   yield earlyExitEvent;
                 } else {
                   try {
@@ -239,6 +260,9 @@ class Runner {
                       invocationContext
                     )), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
                       const event = temp.value;
+                      if ((_h = params.abortSignal) == null ? void 0 : _h.aborted) {
+                        return;
+                      }
                       if (!event.partial) {
                         yield new __await(this.sessionService.appendEvent({ session, event }));
                       }
@@ -246,6 +270,9 @@ class Runner {
                         invocationContext,
                         event
                       }));
+                      if ((_i = params.abortSignal) == null ? void 0 : _i.aborted) {
+                        return;
+                      }
                       if (modifiedEvent) {
                         yield modifiedEvent;
                       } else {
@@ -263,6 +290,9 @@ class Runner {
                     }
                   }
                   yield new __await(this.pluginManager.runAfterRunCallback({ invocationContext }));
+                  if ((_j = params.abortSignal) == null ? void 0 : _j.aborted) {
+                    return;
+                  }
                 }
               }
             });

package/dist/web/sessions/base_session_service.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/web/sessions/database_session_service.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;

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

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
@@ -51,23 +53,16 @@ async function getConnectionOptionsFromUri(uri) {
   } else {
     throw new Error("Unsupported database URI: ".concat(uri));
   }
-  if (uri === "sqlite://:memory:") {
+  if (uri.startsWith("sqlite://")) {
     return {
       entities: ENTITIES,
-      dbName: ":memory:",
+      dbName: uri === "sqlite://:memory:" ? ":memory:" : uri.substring("sqlite://".length),
       driver
     };
   }
-  const { host, port, username, password, pathname } = new URL(uri);
-  const hostName = host.split(":")[0];
-  const dbName = uri.startsWith("sqlite://") ? uri.substring("sqlite://".length) : pathname.slice(1);
   return {
     entities: ENTITIES,
-    dbName,
-    host: hostName,
-    port: port ? parseInt(port) : void 0,
-    user: username,
-    password,
+    clientUrl: uri,
     driver
   };
 }

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

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __decorateClass = (decorators, target, key, kind) => {

package/dist/web/sessions/in_memory_session_service.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/web/sessions/registry.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/sessions/session.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/web/sessions/state.js

@@ -1,3 +1,5 @@
+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;