@google/adk 0.3.0 → 0.4.0

package/dist/cjs/models/apigee_llm.js

@@ -0,0 +1,182 @@
+/**
+  * @license
+  * Copyright 2026 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var apigee_llm_exports = {};
+__export(apigee_llm_exports, {
+  ApigeeLlm: () => ApigeeLlm
+});
+module.exports = __toCommonJS(apigee_llm_exports);
+var import_env_aware_utils = require("../utils/env_aware_utils.js");
+var import_logger = require("../utils/logger.js");
+var import_google_llm = require("./google_llm.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const APIGEE_PROXY_URL_ENV_VARIABLE_NAME = "APIGEE_PROXY_URL";
+class ApigeeLlm extends import_google_llm.Gemini {
+  constructor({
+    model,
+    proxyUrl,
+    apiKey,
+    vertexai,
+    location,
+    project,
+    headers
+  }) {
+    var _a;
+    if (!validateModel(model)) {
+      throw new Error(
+        `Model ${model} is not a valid Apigee model, expected apigee/[<provider>/][<version>/]<model_id>`
+      );
+    }
+    super({
+      ...apigeeToGeminiInitParams({ model, vertexai, project, location, apiKey }),
+      headers
+    });
+    this.proxyUrl = proxyUrl != null ? proxyUrl : "";
+    if (!(0, import_env_aware_utils.isBrowser)() && !this.proxyUrl) {
+      this.proxyUrl = (_a = process.env[APIGEE_PROXY_URL_ENV_VARIABLE_NAME]) != null ? _a : "";
+    }
+    if (!this.proxyUrl) {
+      throw new Error(
+        `Proxy URL must be provided via the constructor or ${APIGEE_PROXY_URL_ENV_VARIABLE_NAME} environment variable.`
+      );
+    }
+  }
+  getHttpOptions() {
+    const opts = super.getHttpOptions();
+    opts.baseUrl = this.proxyUrl;
+    return opts;
+  }
+  getLiveHttpOptions() {
+    const opts = super.getLiveHttpOptions();
+    opts.baseUrl = this.proxyUrl;
+    return opts;
+  }
+  identifyApiVersion() {
+    const modelTrimmed = this.model.startsWith("apigee/") ? this.model.substring("apigee/".length) : this.model;
+    const components = modelTrimmed.split("/");
+    if (components.length === 3) {
+      return components[1];
+    }
+    if (components.length === 2) {
+      if (components[0] != "vertex_ai" && components[0] != "gemini" && components[0].startsWith("v")) {
+        return components[0];
+      }
+    }
+    return this.vertexai ? "v1beta1" : "v1alpha";
+  }
+  get liveApiVersion() {
+    if (!this._apigeeLiveApiVersion) {
+      this._apigeeLiveApiVersion = this.identifyApiVersion();
+    }
+    return this._apigeeLiveApiVersion;
+  }
+  async *generateContentAsync(llmRequest, stream = false) {
+    var _a;
+    const modelToUse = (_a = llmRequest.model) != null ? _a : this.model;
+    llmRequest.model = getModelId(modelToUse);
+    yield* super.generateContentAsync(llmRequest, stream);
+  }
+  async connect(llmRequest) {
+    var _a;
+    const modelToUse = (_a = llmRequest.model) != null ? _a : this.model;
+    llmRequest.model = getModelId(modelToUse);
+    return super.connect(llmRequest);
+  }
+}
+/**
+ * A list of model name patterns that are supported by this LLM.
+ *
+ * @returns A list of supported models.
+ */
+ApigeeLlm.supportedModels = [
+  /apigee\/.*/
+];
+function apigeeToGeminiInitParams({
+  model,
+  vertexai,
+  project,
+  location,
+  apiKey
+}) {
+  var _a;
+  const params = (0, import_google_llm.geminiInitParams)({ model, vertexai, project, location, apiKey });
+  params.vertexai = params.vertexai || ((_a = params.model) == null ? void 0 : _a.startsWith("apigee/vertex_ai/"));
+  if (params.vertexai) {
+    return params;
+  }
+  if (!params.apiKey) {
+    import_logger.logger.warn(
+      `No API key provided when using a Gemini model, using a fake key "-".`
+    );
+    params.apiKey = "-";
+  }
+  return params;
+}
+function getModelId(model) {
+  if (!validateModel(model)) {
+    throw new Error(
+      `Model ${model} is not a valid Apigee model, expected apigee/[<provider>/][<version>/]<model_id>`
+    );
+  }
+  const components = model.split("/");
+  return components[components.length - 1];
+}
+function validateModel(model) {
+  const validProviders = ["vertex_ai", "gemini"];
+  if (!model.startsWith("apigee/")) {
+    return false;
+  }
+  const modelPart = model.substring("apigee/".length);
+  if (modelPart.length === 0) {
+    return false;
+  }
+  const components = modelPart.split("/", -1);
+  if (components[components.length - 1].length === 0) {
+    return false;
+  }
+  if (components.length == 1) {
+    return true;
+  }
+  if (components.length == 2) {
+    if (validProviders.includes(components[0])) {
+      return true;
+    }
+    return components[0].startsWith("v");
+  }
+  if (components.length == 3) {
+    if (!validProviders.includes(components[0])) {
+      return false;
+    }
+    return components[1].startsWith("v");
+  }
+  return false;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ApigeeLlm
+});

package/dist/cjs/models/base_llm.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/models/base_llm_connection.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/models/gemini_llm_connection.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/models/google_llm.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -24,10 +24,12 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var google_llm_exports = {};
 __export(google_llm_exports, {
-  Gemini: () => Gemini
+  Gemini: () => Gemini,
+  geminiInitParams: () => geminiInitParams
 });
 module.exports = __toCommonJS(google_llm_exports);
 var import_genai = require("@google/genai");
+var import_env_aware_utils = require("../utils/env_aware_utils.js");
 var import_logger = require("../utils/logger.js");
 var import_variant_utils = require("../utils/variant_utils.js");
 var import_base_llm = require("./base_llm.js");
@@ -54,45 +56,23 @@ class Gemini extends import_base_llm.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.
@@ -181,25 +161,24 @@ class Gemini extends import_base_llm.BaseLlm {
       yield (0, import_llm_response.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 import_genai.GoogleGenAI({
         vertexai: this.vertexai,
         project: this.project,
         location: this.location,
-        httpOptions: { headers: combinedHeaders }
+        httpOptions: this.getHttpOptions()
       });
     } else {
       this._apiClient = new import_genai.GoogleGenAI({
         apiKey: this.apiKey,
-        httpOptions: { headers: combinedHeaders }
+        httpOptions: this.getHttpOptions()
       });
     }
     return this._apiClient;
@@ -216,14 +195,17 @@ class Gemini extends import_base_llm.BaseLlm {
     }
     return this._liveApiVersion;
   }
+  getLiveHttpOptions() {
+    return {
+      headers: this.trackingHeaders,
+      apiVersion: this.liveApiVersion
+    };
+  }
   get liveApiClient() {
     if (!this._liveApiClient) {
       this._liveApiClient = new import_genai.GoogleGenAI({
         apiKey: this.apiKey,
-        httpOptions: {
-          headers: this.trackingHeaders,
-          apiVersion: this.liveApiVersion
-        }
+        httpOptions: this.getLiveHttpOptions()
       });
     }
     return this._liveApiClient;
@@ -301,7 +283,44 @@ 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 && !(0, import_env_aware_utils.isBrowser)()) {
+    params.vertexai = (0, import_env_aware_utils.getBooleanEnvVar)("GOOGLE_GENAI_USE_VERTEXAI");
+  }
+  if (params.vertexai) {
+    if (!(0, import_env_aware_utils.isBrowser)() && !params.project) {
+      params.project = process.env["GOOGLE_CLOUD_PROJECT"];
+    }
+    if (!(0, import_env_aware_utils.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 && !(0, import_env_aware_utils.isBrowser)()) {
+      params.apiKey = process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"];
+    }
+  }
+  return params;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  Gemini
+  Gemini,
+  geminiInitParams
 });

package/dist/cjs/models/llm_request.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/models/llm_response.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/models/registry.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -28,6 +28,7 @@ __export(registry_exports, {
 });
 module.exports = __toCommonJS(registry_exports);
 var import_logger = require("../utils/logger.js");
+var import_apigee_llm = require("./apigee_llm.js");
 var import_google_llm = require("./google_llm.js");
 /**
  * @license
@@ -115,6 +116,7 @@ _LLMRegistry.llmRegistryDict = /* @__PURE__ */ new Map();
 _LLMRegistry.resolveCache = new LRUCache(32);
 let LLMRegistry = _LLMRegistry;
 LLMRegistry.register(import_google_llm.Gemini);
+LLMRegistry.register(import_apigee_llm.ApigeeLlm);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   LLMRegistry

package/dist/cjs/plugins/base_plugin.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/plugins/logging_plugin.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/plugins/plugin_manager.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/plugins/security_plugin.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/runner/in_memory_runner.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/runner/runner.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -55,6 +55,37 @@ class Runner {
     this.memoryService = input.memoryService;
     this.credentialService = input.credentialService;
   }
+  /**
+   * Runs the agent with a new, ephemeral session.
+   *
+   * @param params.userId The user ID of the session.
+   * @param params.newMessage A new message to append to the session.
+   * @param params.stateDelta An optional state delta to apply to the session.
+   * @param params.runConfig The run config for the agent.
+   * @yields The Events generated by the agent.
+   */
+  async *runEphemeral(params) {
+    const session = await this.sessionService.createSession({
+      appName: this.appName,
+      userId: params.userId
+    });
+    const sessionId = session.id;
+    try {
+      yield* this.runAsync({
+        userId: params.userId,
+        sessionId,
+        newMessage: params.newMessage,
+        stateDelta: params.stateDelta,
+        runConfig: params.runConfig
+      });
+    } finally {
+      await this.sessionService.deleteSession({
+        appName: this.appName,
+        userId: params.userId,
+        sessionId
+      });
+    }
+  }
   /**
    * Runs the agent with the given message, and returns an async generator of
    * events.

package/dist/cjs/sessions/base_session_service.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -24,9 +24,12 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var base_session_service_exports = {};
 __export(base_session_service_exports, {
-  BaseSessionService: () => BaseSessionService
+  BaseSessionService: () => BaseSessionService,
+  mergeStates: () => mergeStates,
+  trimTempDeltaState: () => trimTempDeltaState
 });
 module.exports = __toCommonJS(base_session_service_exports);
+var import_lodash_es = require("lodash-es");
 var import_state = require("./state.js");
 /**
  * @license
@@ -34,6 +37,26 @@ var import_state = require("./state.js");
  * SPDX-License-Identifier: Apache-2.0
  */
 class BaseSessionService {
+  /**
+   * Gets a session or creates one if it doesn't exist.
+   *
+   * @param request The request to get or create a session.
+   * @return A promise that resolves to the session instance.
+   */
+  async getOrCreateSession(request) {
+    if (!request.sessionId) {
+      return this.createSession(request);
+    }
+    const session = await this.getSession({
+      appName: request.appName,
+      userId: request.userId,
+      sessionId: request.sessionId
+    });
+    if (session) {
+      return session;
+    }
+    return this.createSession(request);
+  }
   /**
    * Appends an event to a session.
    *
@@ -44,6 +67,7 @@ class BaseSessionService {
     if (event.partial) {
       return event;
     }
+    event = trimTempDeltaState(event);
     this.updateSessionState({ session, event });
     session.events.push(event);
     return event;
@@ -65,7 +89,33 @@ class BaseSessionService {
     }
   }
 }
+function trimTempDeltaState(event) {
+  if (!event.actions || !event.actions.stateDelta) {
+    return event;
+  }
+  const stateDelta = event.actions.stateDelta;
+  const filteredStateDelta = {};
+  for (const [key, value] of Object.entries(stateDelta)) {
+    if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
+      filteredStateDelta[key] = value;
+    }
+  }
+  event.actions.stateDelta = filteredStateDelta;
+  return event;
+}
+function mergeStates(appState = {}, userState = {}, sessionState = {}) {
+  const merged = (0, import_lodash_es.cloneDeep)(sessionState);
+  for (const [k, v] of Object.entries(appState)) {
+    merged[import_state.State.APP_PREFIX + k] = v;
+  }
+  for (const [k, v] of Object.entries(userState)) {
+    merged[import_state.State.USER_PREFIX + k] = v;
+  }
+  return merged;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  BaseSessionService
+  BaseSessionService,
+  mergeStates,
+  trimTempDeltaState
 });