@google/adk 0.2.5 → 0.4.0

package/dist/esm/models/google_llm.js

@@ -3,14 +3,17 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { createPartFromText, FinishReason, GoogleGenAI } from "@google/genai";
+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 { BaseLlm } from "./base_llm.js";
 import { GeminiLlmConnection } from "./gemini_llm_connection.js";
 import { createLlmResponse } from "./llm_response.js";
-const AGENT_ENGINE_TELEMETRY_TAG = "remote_reasoning_engine";
-const AGENT_ENGINE_TELEMETRY_ENV_VARIABLE_NAME = "GOOGLE_CLOUD_AGENT_ENGINE_ID";
 class Gemini extends BaseLlm {
   /**
    * @param params The parameters for creating a Gemini instance.
@@ -27,45 +30,23 @@ class Gemini extends BaseLlm {
       model = "gemini-2.5-flash";
     }
     super({ model });
-    this.project = project;
-    this.location = location;
-    this.apiKey = apiKey;
-    this.headers = headers;
-    const canReadEnv = typeof process === "object";
-    this.vertexai = !!vertexai;
-    if (!this.vertexai && canReadEnv) {
-      const vertexAIfromEnv = process.env["GOOGLE_GENAI_USE_VERTEXAI"];
-      if (vertexAIfromEnv) {
-        this.vertexai = vertexAIfromEnv.toLowerCase() === "true" || vertexAIfromEnv === "1";
-      }
-    }
-    if (this.vertexai) {
-      if (canReadEnv && !this.project) {
-        this.project = process.env["GOOGLE_CLOUD_PROJECT"];
-      }
-      if (canReadEnv && !this.location) {
-        this.location = process.env["GOOGLE_CLOUD_LOCATION"];
-      }
-      if (!this.project) {
-        throw new Error(
-          "VertexAI project must be provided via constructor or GOOGLE_CLOUD_PROJECT environment variable."
-        );
-      }
-      if (!this.location) {
-        throw new Error(
-          "VertexAI location must be provided via constructor or GOOGLE_CLOUD_LOCATION environment variable."
-        );
-      }
-    } else {
-      if (!this.apiKey && canReadEnv) {
-        this.apiKey = process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"];
-      }
-      if (!this.apiKey) {
-        throw new Error(
-          "API key must be provided via constructor or GOOGLE_GENAI_API_KEY or GEMINI_API_KEY environment variable."
-        );
-      }
+    const params = geminiInitParams({
+      model,
+      vertexai,
+      project,
+      location,
+      apiKey
+    });
+    if (!params.vertexai && !params.apiKey) {
+      throw new Error(
+        "API key must be provided via constructor or GOOGLE_GENAI_API_KEY or GEMINI_API_KEY environment variable."
+      );
     }
+    this.project = params.project;
+    this.location = params.location;
+    this.apiKey = params.apiKey;
+    this.headers = headers;
+    this.vertexai = !!params.vertexai;
   }
   /**
    * Sends a request to the Gemini model.
@@ -154,25 +135,24 @@ class Gemini extends BaseLlm {
       yield createLlmResponse(response);
     }
   }
+  getHttpOptions() {
+    return { headers: { ...this.trackingHeaders, ...this.headers } };
+  }
   get apiClient() {
     if (this._apiClient) {
       return this._apiClient;
     }
-    const combinedHeaders = {
-      ...this.trackingHeaders,
-      ...this.headers
-    };
     if (this.vertexai) {
       this._apiClient = new GoogleGenAI({
         vertexai: this.vertexai,
         project: this.project,
         location: this.location,
-        httpOptions: { headers: combinedHeaders }
+        httpOptions: this.getHttpOptions()
       });
     } else {
       this._apiClient = new GoogleGenAI({
         apiKey: this.apiKey,
-        httpOptions: { headers: combinedHeaders }
+        httpOptions: this.getHttpOptions()
       });
     }
     return this._apiClient;
@@ -189,14 +169,17 @@ class Gemini extends BaseLlm {
     }
     return this._liveApiVersion;
   }
+  getLiveHttpOptions() {
+    return {
+      headers: this.trackingHeaders,
+      apiVersion: this.liveApiVersion
+    };
+  }
   get liveApiClient() {
     if (!this._liveApiClient) {
       this._liveApiClient = new GoogleGenAI({
         apiKey: this.apiKey,
-        httpOptions: {
-          headers: this.trackingHeaders,
-          apiVersion: this.liveApiVersion
-        }
+        httpOptions: this.getLiveHttpOptions()
       });
     }
     return this._liveApiClient;
@@ -223,7 +206,9 @@ class Gemini extends BaseLlm {
       llmRequest.liveConnectConfig.systemInstruction = {
         role: "system",
         // TODO - b/425992518: validate type casting works well.
-        parts: [createPartFromText(llmRequest.config.systemInstruction)]
+        parts: [
+          createPartFromText(llmRequest.config.systemInstruction)
+        ]
       };
     }
     llmRequest.liveConnectConfig.tools = (_c = llmRequest.config) == null ? void 0 : _c.tools;
@@ -272,6 +257,43 @@ function removeDisplayNameIfPresent(dataObj) {
     dataObj.displayName = void 0;
   }
 }
+function geminiInitParams({
+  model,
+  vertexai,
+  project,
+  location,
+  apiKey
+}) {
+  const params = { model, vertexai, project, location, apiKey };
+  params.vertexai = !!vertexai;
+  if (!params.vertexai && !isBrowser()) {
+    params.vertexai = getBooleanEnvVar("GOOGLE_GENAI_USE_VERTEXAI");
+  }
+  if (params.vertexai) {
+    if (!isBrowser() && !params.project) {
+      params.project = process.env["GOOGLE_CLOUD_PROJECT"];
+    }
+    if (!isBrowser() && !params.location) {
+      params.location = process.env["GOOGLE_CLOUD_LOCATION"];
+    }
+    if (!params.project) {
+      throw new Error(
+        "VertexAI project must be provided via constructor or GOOGLE_CLOUD_PROJECT environment variable."
+      );
+    }
+    if (!params.location) {
+      throw new Error(
+        "VertexAI location must be provided via constructor or GOOGLE_CLOUD_LOCATION environment variable."
+      );
+    }
+  } else {
+    if (!params.apiKey && !isBrowser()) {
+      params.apiKey = process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"];
+    }
+  }
+  return params;
+}
 export {
-  Gemini
+  Gemini,
+  geminiInitParams
 };

package/dist/esm/models/registry.js

@@ -4,6 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { logger } from "../utils/logger.js";
+import { ApigeeLlm } from "./apigee_llm.js";
 import { Gemini } from "./google_llm.js";
 class LRUCache {
   constructor(maxSize) {
@@ -86,6 +87,7 @@ _LLMRegistry.llmRegistryDict = /* @__PURE__ */ new Map();
 _LLMRegistry.resolveCache = new LRUCache(32);
 let LLMRegistry = _LLMRegistry;
 LLMRegistry.register(Gemini);
+LLMRegistry.register(ApigeeLlm);
 export {
   LLMRegistry
 };

package/dist/esm/plugins/base_plugin.js

@@ -25,6 +25,7 @@ class BasePlugin {
    *     value to replace the user message. Returning `undefined` to proceed
    *     normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onUserMessageCallback(params) {
     return;
   }
@@ -40,6 +41,7 @@ class BasePlugin {
    *     to halt execution of the runner and ends the runner with that event.
    *     Return `undefined` to proceed normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeRunCallback(params) {
     return;
   }
@@ -55,6 +57,7 @@ class BasePlugin {
    *     framework to modify or replace the response. Returning `undefined`
    *     allows the original response to be used.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onEventCallback(params) {
     return;
   }
@@ -67,6 +70,7 @@ class BasePlugin {
    * @param params.invocationContext The context for the entire invocation.
    * @returns undefined
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterRunCallback(params) {
     return;
   }
@@ -82,6 +86,7 @@ class BasePlugin {
    *     bypass the agent's callbacks and its execution, and return this value
    *     directly. Returning `undefined` allows the agent to proceed normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeAgentCallback(params) {
     return;
   }
@@ -97,6 +102,7 @@ class BasePlugin {
    *     replace the agent's original result. Returning `undefined` uses the
    *     original, unmodified result.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterAgentCallback(params) {
     return;
   }
@@ -113,6 +119,7 @@ class BasePlugin {
    *     trigger an early exit and returns the response immediately. Returning
    *     `undefined` allows the LLM request to proceed normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeModelCallback(params) {
     return;
   }
@@ -128,6 +135,7 @@ class BasePlugin {
    *     framework to modify or replace the response. Returning `undefined`
    *     allows the original response to be used.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterModelCallback(params) {
     return;
   }
@@ -145,6 +153,7 @@ class BasePlugin {
    *     used instead of propagating the error. Returning `undefined` allows
    *     the original error to be raised.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onModelErrorCallback(params) {
     return;
   }
@@ -162,6 +171,7 @@ class BasePlugin {
    *     the tool execution and return this response immediately. Returning
    *     `undefined` uses the original, unmodified arguments.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeToolCallback(params) {
     return;
   }
@@ -180,6 +190,7 @@ class BasePlugin {
    *     post-processing or altering tool outputs. Returning `undefined` uses
    *     the original, unmodified result.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterToolCallback(params) {
     return;
   }
@@ -207,6 +218,7 @@ class BasePlugin {
      *     used as the tool response instead of propagating the error. Returning
      *     `undefined` allows the original error to be raised.
      */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onToolErrorCallback(params) {
     return;
   }

package/dist/esm/plugins/logging_plugin.js

@@ -3,7 +3,11 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { getFunctionCalls, getFunctionResponses, isFinalResponse } from "../events/event.js";
+import {
+  getFunctionCalls,
+  getFunctionResponses,
+  isFinalResponse
+} from "../events/event.js";
 import { logger } from "../utils/logger.js";
 import { BasePlugin } from "./base_plugin.js";
 class LoggingPlugin extends BasePlugin {
@@ -15,7 +19,10 @@ class LoggingPlugin extends BasePlugin {
   constructor(name = "logging_plugin") {
     super(name);
   }
-  async onUserMessageCallback({ invocationContext, userMessage }) {
+  async onUserMessageCallback({
+    invocationContext,
+    userMessage
+  }) {
     var _a;
     this.log("\u{1F680} USER MESSAGE RECEIVED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
@@ -29,14 +36,18 @@ class LoggingPlugin extends BasePlugin {
     }
     return void 0;
   }
-  async beforeRunCallback({ invocationContext }) {
+  async beforeRunCallback({
+    invocationContext
+  }) {
     var _a;
     this.log("\u{1F3C3} INVOCATION STARTING");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
     this.log(`   Starting Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
     return void 0;
   }
-  async onEventCallback({ invocationContext, event }) {
+  async onEventCallback({
+    event
+  }) {
     this.log("\u{1F4E2} EVENT YIELDED");
     this.log(`   Event ID: ${event.id}`);
     this.log(`   Author: ${event.author}`);
@@ -57,14 +68,18 @@ class LoggingPlugin extends BasePlugin {
     }
     return void 0;
   }
-  async afterRunCallback({ invocationContext }) {
+  async afterRunCallback({
+    invocationContext
+  }) {
     var _a;
     this.log("\u2705 INVOCATION COMPLETED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
     this.log(`   Final Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
     return void 0;
   }
-  async beforeAgentCallback({ agent, callbackContext }) {
+  async beforeAgentCallback({
+    callbackContext
+  }) {
     this.log("\u{1F916} AGENT STARTING");
     this.log(`   Agent Name: ${callbackContext.agentName}`);
     this.log(`   Invocation ID: ${callbackContext.invocationId}`);
@@ -73,13 +88,18 @@ class LoggingPlugin extends BasePlugin {
     }
     return void 0;
   }
-  async afterAgentCallback({ agent, callbackContext }) {
+  async afterAgentCallback({
+    callbackContext
+  }) {
     this.log("\u{1F916} AGENT COMPLETED");
     this.log(`   Agent Name: ${callbackContext.agentName}`);
     this.log(`   Invocation ID: ${callbackContext.invocationId}`);
     return void 0;
   }
-  async beforeModelCallback({ callbackContext, llmRequest }) {
+  async beforeModelCallback({
+    callbackContext,
+    llmRequest
+  }) {
     var _a;
     this.log("\u{1F9E0} LLM REQUEST");
     this.log(`   Model: ${(_a = llmRequest.model) != null ? _a : "default"}`);
@@ -97,7 +117,10 @@ class LoggingPlugin extends BasePlugin {
     }
     return void 0;
   }
-  async afterModelCallback({ callbackContext, llmResponse }) {
+  async afterModelCallback({
+    callbackContext,
+    llmResponse
+  }) {
     this.log("\u{1F9E0} LLM RESPONSE");
     this.log(`   Agent: ${callbackContext.agentName}`);
     if (llmResponse.errorCode) {
@@ -113,11 +136,17 @@ class LoggingPlugin extends BasePlugin {
       }
     }
     if (llmResponse.usageMetadata) {
-      this.log(`   Token Usage - Input: ${llmResponse.usageMetadata.promptTokenCount}, Output: ${llmResponse.usageMetadata.candidatesTokenCount}`);
+      this.log(
+        `   Token Usage - Input: ${llmResponse.usageMetadata.promptTokenCount}, Output: ${llmResponse.usageMetadata.candidatesTokenCount}`
+      );
     }
     return void 0;
   }
-  async beforeToolCallback({ tool, toolArgs, toolContext }) {
+  async beforeToolCallback({
+    tool,
+    toolArgs,
+    toolContext
+  }) {
     this.log("\u{1F527} TOOL STARTING");
     this.log(`   Tool Name: ${tool.name}`);
     this.log(`   Agent: ${toolContext.agentName}`);
@@ -125,7 +154,11 @@ class LoggingPlugin extends BasePlugin {
     this.log(`   Arguments: ${this.formatArgs(toolArgs)}`);
     return void 0;
   }
-  async afterToolCallback({ tool, toolArgs, toolContext, result }) {
+  async afterToolCallback({
+    tool,
+    toolContext,
+    result
+  }) {
     this.log("\u{1F527} TOOL COMPLETED");
     this.log(`   Tool Name: ${tool.name}`);
     this.log(`   Agent: ${toolContext.agentName}`);
@@ -133,13 +166,21 @@ class LoggingPlugin extends BasePlugin {
     this.log(`   Result: ${this.formatArgs(result)}`);
     return void 0;
   }
-  async onModelErrorCallback({ callbackContext, llmRequest, error }) {
+  async onModelErrorCallback({
+    callbackContext,
+    error
+  }) {
     this.log("\u{1F9E0} LLM ERROR");
     this.log(`   Agent: ${callbackContext.agentName}`);
     this.log(`   Error: ${error}`);
     return void 0;
   }
-  async onToolErrorCallback({ tool, toolArgs, toolContext, error }) {
+  async onToolErrorCallback({
+    tool,
+    toolArgs,
+    toolContext,
+    error
+  }) {
     this.log("\u{1F527} TOOL ERROR");
     this.log(`   Tool Name: ${tool.name}`);
     this.log(`   Agent: ${toolContext.agentName}`);

package/dist/esm/plugins/plugin_manager.js

@@ -78,19 +78,22 @@ class PluginManager {
   /**
    * Runs the `onUserMessageCallback` for all plugins.
    */
-  async runOnUserMessageCallback({ userMessage, invocationContext }) {
+  async runOnUserMessageCallback({
+    userMessage,
+    invocationContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.onUserMessageCallback(
-        { userMessage, invocationContext }
-      ),
+      (plugin) => plugin.onUserMessageCallback({ userMessage, invocationContext }),
       "onUserMessageCallback"
     );
   }
   /**
    * Runs the `beforeRunCallback` for all plugins.
    */
-  async runBeforeRunCallback({ invocationContext }) {
+  async runBeforeRunCallback({
+    invocationContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeRunCallback({ invocationContext }),
@@ -100,7 +103,9 @@ class PluginManager {
   /**
    * Runs the `afterRunCallback` for all plugins.
    */
-  async runAfterRunCallback({ invocationContext }) {
+  async runAfterRunCallback({
+    invocationContext
+  }) {
     await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.afterRunCallback({ invocationContext }),
@@ -110,7 +115,10 @@ class PluginManager {
   /**
    * Runs the `onEventCallback` for all plugins.
    */
-  async runOnEventCallback({ invocationContext, event }) {
+  async runOnEventCallback({
+    invocationContext,
+    event
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.onEventCallback({ invocationContext, event }),
@@ -120,7 +128,10 @@ class PluginManager {
   /**
    * Runs the `beforeAgentCallback` for all plugins.
    */
-  async runBeforeAgentCallback({ agent, callbackContext }) {
+  async runBeforeAgentCallback({
+    agent,
+    callbackContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeAgentCallback({ agent, callbackContext }),
@@ -130,7 +141,10 @@ class PluginManager {
   /**
    * Runs the `afterAgentCallback` for all plugins.
    */
-  async runAfterAgentCallback({ agent, callbackContext }) {
+  async runAfterAgentCallback({
+    agent,
+    callbackContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.afterAgentCallback({ agent, callbackContext }),
@@ -140,7 +154,11 @@ class PluginManager {
   /**
    * Runs the `beforeToolCallback` for all plugins.
    */
-  async runBeforeToolCallback({ tool, toolArgs, toolContext }) {
+  async runBeforeToolCallback({
+    tool,
+    toolArgs,
+    toolContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeToolCallback({ tool, toolArgs, toolContext }),
@@ -150,31 +168,39 @@ class PluginManager {
   /**
    * Runs the `afterToolCallback` for all plugins.
    */
-  async runAfterToolCallback({ tool, toolArgs, toolContext, result }) {
+  async runAfterToolCallback({
+    tool,
+    toolArgs,
+    toolContext,
+    result
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.afterToolCallback(
-        { tool, toolArgs, toolContext, result }
-      ),
+      (plugin) => plugin.afterToolCallback({ tool, toolArgs, toolContext, result }),
       "afterToolCallback"
     );
   }
   /**
    * Runs the `onModelErrorCallback` for all plugins.
    */
-  async runOnModelErrorCallback({ callbackContext, llmRequest, error }) {
+  async runOnModelErrorCallback({
+    callbackContext,
+    llmRequest,
+    error
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.onModelErrorCallback(
-        { callbackContext, llmRequest, error }
-      ),
+      (plugin) => plugin.onModelErrorCallback({ callbackContext, llmRequest, error }),
       "onModelErrorCallback"
     );
   }
   /**
    * Runs the `beforeModelCallback` for all plugins.
    */
-  async runBeforeModelCallback({ callbackContext, llmRequest }) {
+  async runBeforeModelCallback({
+    callbackContext,
+    llmRequest
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeModelCallback({ callbackContext, llmRequest }),
@@ -184,7 +210,10 @@ class PluginManager {
   /**
    * Runs the `afterModelCallback` for all plugins.
    */
-  async runAfterModelCallback({ callbackContext, llmResponse }) {
+  async runAfterModelCallback({
+    callbackContext,
+    llmResponse
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.afterModelCallback({ callbackContext, llmResponse }),
@@ -194,12 +223,15 @@ class PluginManager {
   /**
    * Runs the `onToolErrorCallback` for all plugins.
    */
-  async runOnToolErrorCallback({ tool, toolArgs, toolContext, error }) {
+  async runOnToolErrorCallback({
+    tool,
+    toolArgs,
+    toolContext,
+    error
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.onToolErrorCallback(
-        { tool, toolArgs, toolContext, error }
-      ),
+      (plugin) => plugin.onToolErrorCallback({ tool, toolArgs, toolContext, error }),
       "onToolErrorCallback"
     );
   }

package/dist/esm/plugins/security_plugin.js

@@ -14,7 +14,7 @@ var PolicyOutcome = /* @__PURE__ */ ((PolicyOutcome2) => {
   return PolicyOutcome2;
 })(PolicyOutcome || {});
 class InMemoryPolicyEngine {
-  async evaluate(context) {
+  async evaluate() {
     return Promise.resolve({
       outcome: "ALLOW" /* ALLOW */,
       reason: "For prototyping purpose, all tool calls are allowed."