@google/adk 0.6.1 → 1.1.0

package/dist/esm/models/google_llm.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -5,12 +7,12 @@
  */
 import {
   createPartFromText,
-  FinishReason,
   GoogleGenAI
 } from "@google/genai";
 import { getBooleanEnvVar, isBrowser } from "../utils/env_aware_utils.js";
 import { logger } from "../utils/logger.js";
 import { GoogleLLMVariant } from "../utils/variant_utils.js";
+import { StreamingResponseAggregator } from "../utils/streaming_utils.js";
 import { BaseLlm } from "./base_llm.js";
 import { GeminiLlmConnection } from "./gemini_llm_connection.js";
 import { createLlmResponse } from "./llm_response.js";
@@ -55,80 +57,44 @@ class Gemini extends BaseLlm {
    * @param stream bool = false, whether to do streaming call.
    * @yields LlmResponse: The model response.
    */
-  async *generateContentAsync(llmRequest, stream = false) {
-    var _a, _b, _c, _d, _e, _f, _g;
+  async *generateContentAsync(llmRequest, stream = false, abortSignal) {
+    var _a, _b, _c;
     this.preprocessRequest(llmRequest);
     this.maybeAppendUserContent(llmRequest);
     logger.info(
-      `Sending out request, model: ${llmRequest.model}, backend: ${this.apiBackend}, stream: ${stream}`
+      `Sending out request, model: ${(_a = llmRequest.model) != null ? _a : this.model}, backend: ${this.apiBackend}, stream: ${stream}`
     );
-    if ((_a = llmRequest.config) == null ? void 0 : _a.httpOptions) {
+    if (!llmRequest.config) {
+      llmRequest.config = {};
+    }
+    if (llmRequest.config.httpOptions) {
       llmRequest.config.httpOptions.headers = {
         ...llmRequest.config.httpOptions.headers,
         ...this.trackingHeaders
       };
     }
+    if (abortSignal) {
+      llmRequest.config.abortSignal = abortSignal;
+    }
     if (stream) {
       const streamResult = await this.apiClient.models.generateContentStream({
         model: (_b = llmRequest.model) != null ? _b : this.model,
         contents: llmRequest.contents,
         config: llmRequest.config
       });
-      let thoughtText = "";
-      let text = "";
-      let usageMetadata;
-      let lastResponse;
+      const aggregator = new StreamingResponseAggregator();
       for await (const response of streamResult) {
-        lastResponse = response;
-        const llmResponse = createLlmResponse(response);
-        usageMetadata = llmResponse.usageMetadata;
-        const firstPart = (_d = (_c = llmResponse.content) == null ? void 0 : _c.parts) == null ? void 0 : _d[0];
-        if (firstPart == null ? void 0 : firstPart.text) {
-          if ("thought" in firstPart && firstPart.thought) {
-            thoughtText += firstPart.text;
-          } else {
-            text += firstPart.text;
-          }
-          llmResponse.partial = true;
-        } else if ((thoughtText || text) && (!firstPart || !firstPart.inlineData)) {
-          const parts = [];
-          if (thoughtText) {
-            parts.push({ text: thoughtText, thought: true });
-          }
-          if (text) {
-            parts.push(createPartFromText(text));
-          }
-          yield {
-            content: {
-              role: "model",
-              parts
-            },
-            usageMetadata: llmResponse.usageMetadata
-          };
-          thoughtText = "";
-          text = "";
+        for await (const llmResponse of aggregator.processResponse(response)) {
+          yield llmResponse;
         }
-        yield llmResponse;
       }
-      if ((text || thoughtText) && ((_f = (_e = lastResponse == null ? void 0 : lastResponse.candidates) == null ? void 0 : _e[0]) == null ? void 0 : _f.finishReason) === FinishReason.STOP) {
-        const parts = [];
-        if (thoughtText) {
-          parts.push({ text: thoughtText, thought: true });
-        }
-        if (text) {
-          parts.push({ text });
-        }
-        yield {
-          content: {
-            role: "model",
-            parts
-          },
-          usageMetadata
-        };
+      const finalResponse = aggregator.close();
+      if (finalResponse) {
+        yield finalResponse;
       }
     } else {
       const response = await this.apiClient.models.generateContent({
-        model: (_g = llmRequest.model) != null ? _g : this.model,
+        model: (_c = llmRequest.model) != null ? _c : this.model,
         contents: llmRequest.contents,
         config: llmRequest.config
       });

package/dist/esm/models/llm_request.js

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

package/dist/esm/models/llm_response.js

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

package/dist/esm/models/registry.js

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

package/dist/esm/models/routed_llm.js

@@ -0,0 +1,73 @@
+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 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]) => `${name} (${model.model})`
+    );
+    const computedName = `RoutedLlm[${modelNames.join(", ")}]`;
+    super({ model: modelName === "routed-llm" ? computedName : modelName });
+    this.models = modelsMap;
+    this.router = router;
+  }
+  /**
+   * Generates content by delegating to the selected model.
+   */
+  async *generateContentAsync(llmRequest, stream) {
+    logger.info(`Routing request via ${this.model}`);
+    yield* 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/esm/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/esm/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/esm/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/esm/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/esm/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/esm/runner/runner.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -96,12 +98,15 @@ class Runner {
         ctx,
         this,
         async function* () {
-          var _a2;
+          var _a2, _b, _c, _d, _e, _f, _g, _h, _i, _j;
           const session = 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(
@@ -131,17 +136,21 @@ class Runner {
             session,
             userContent: newMessage,
             runConfig,
-            pluginManager: this.pluginManager
+            pluginManager: this.pluginManager,
+            abortSignal: params.abortSignal
           });
           const pluginUserMessage = 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) {
@@ -151,6 +160,9 @@ class Runner {
                 session.id,
                 newMessage
               );
+              if ((_d = params.abortSignal) == null ? void 0 : _d.aborted) {
+                return;
+              }
             }
             await this.sessionService.appendEvent({
               session,
@@ -161,6 +173,9 @@ class Runner {
                 content: newMessage
               })
             });
+            if ((_e = params.abortSignal) == null ? void 0 : _e.aborted) {
+              return;
+            }
           }
           invocationContext.agent = this.determineAgentForResumption(
             session,
@@ -170,6 +185,9 @@ class Runner {
             const beforeRunCallbackResponse = await this.pluginManager.runBeforeRunCallback({
               invocationContext
             });
+            if ((_f = params.abortSignal) == null ? void 0 : _f.aborted) {
+              return;
+            }
             if (beforeRunCallbackResponse) {
               const earlyExitEvent = createEvent({
                 invocationId: invocationContext.invocationId,
@@ -180,11 +198,17 @@ class Runner {
                 session,
                 event: earlyExitEvent
               });
+              if ((_g = params.abortSignal) == null ? void 0 : _g.aborted) {
+                return;
+              }
               yield earlyExitEvent;
             } else {
               for await (const event of invocationContext.agent.runAsync(
                 invocationContext
               )) {
+                if ((_h = params.abortSignal) == null ? void 0 : _h.aborted) {
+                  return;
+                }
                 if (!event.partial) {
                   await this.sessionService.appendEvent({ session, event });
                 }
@@ -192,6 +216,9 @@ class Runner {
                   invocationContext,
                   event
                 });
+                if ((_i = params.abortSignal) == null ? void 0 : _i.aborted) {
+                  return;
+                }
                 if (modifiedEvent) {
                   yield modifiedEvent;
                 } else {
@@ -199,6 +226,9 @@ class Runner {
                 }
               }
               await this.pluginManager.runAfterRunCallback({ invocationContext });
+              if ((_j = params.abortSignal) == null ? void 0 : _j.aborted) {
+                return;
+              }
             }
           }
         }

package/dist/esm/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/esm/sessions/database_session_service.js

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

package/dist/esm/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: ${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/esm/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/esm/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/esm/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/esm/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/esm/sessions/state.js

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