@google/adk 2.4.0 → 2.5.0

package/dist/cjs/tools/application_integration_tool/clients/connections_client.js

@@ -62,6 +62,7 @@ var import_cloud_request = require("./cloud_request.js");
  */
 const CONNECTOR_URL = "https://connectors.googleapis.com";
 const OPERATION_POLL_INTERVAL_MS = 1e3;
+const MAX_OPERATION_POLL_ATTEMPTS = 60;
 function toStringValue(value) {
   return typeof value === "string" ? value : "";
 }
@@ -70,6 +71,16 @@ function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
 __name(sleep, "sleep");
+function formatOperationError(error) {
+  if (typeof error === "string" && error.length > 0) {
+    return error;
+  }
+  if ((0, import_json_utils.isJsonObject)(error) && typeof error["message"] === "string" && error["message"].length > 0) {
+    return error["message"];
+  }
+  return JSON.stringify(error);
+}
+__name(formatOperationError, "formatOperationError");
 function schemaRef(name) {
   return { $ref: `#/components/schemas/${name}` };
 }
@@ -151,6 +162,12 @@ const CONNECTOR_BASE_SPEC = {
         default: 120,
         description: "Timeout in seconds for execution of custom query"
       },
+      sortByColumns: {
+        type: "array",
+        items: { type: "string" },
+        default: [],
+        description: "Column to sort the results by"
+      },
       connectorOutputPayload: { type: "object" },
       nextPageToken: { type: "string" },
       "execute-connector_Response": {
@@ -477,7 +494,8 @@ function listOperationRequest(options) {
     connectionName: schemaRef("connectionName"),
     serviceName: schemaRef("serviceName"),
     host: schemaRef("host"),
-    entity: schemaRef("entity")
+    entity: schemaRef("entity"),
+    sortByColumns: schemaRef("sortByColumns")
   };
   if (options == null ? void 0 : options.includeDynamicAuthConfig) {
     properties["dynamicAuthConfig"] = schemaRef("dynamicAuthConfig");
@@ -524,7 +542,20 @@ function actionResponse(action) {
   };
 }
 __name(actionResponse, "actionResponse");
-function executeCustomQueryRequest() {
+function executeCustomQueryRequest(options) {
+  const properties = {
+    operation: schemaRef("operation"),
+    connectionName: schemaRef("connectionName"),
+    serviceName: schemaRef("serviceName"),
+    host: schemaRef("host"),
+    action: schemaRef("action"),
+    query: schemaRef("query"),
+    timeout: schemaRef("timeout"),
+    pageSize: schemaRef("pageSize")
+  };
+  if (options == null ? void 0 : options.includeDynamicAuthConfig) {
+    properties["dynamicAuthConfig"] = schemaRef("dynamicAuthConfig");
+  }
   return {
     type: "object",
     required: [
@@ -537,16 +568,7 @@ function executeCustomQueryRequest() {
       "timeout",
       "pageSize"
     ],
-    properties: {
-      operation: schemaRef("operation"),
-      connectionName: schemaRef("connectionName"),
-      serviceName: schemaRef("serviceName"),
-      host: schemaRef("host"),
-      action: schemaRef("action"),
-      query: schemaRef("query"),
-      timeout: schemaRef("timeout"),
-      pageSize: schemaRef("pageSize")
-    }
+    properties
   };
 }
 __name(executeCustomQueryRequest, "executeCustomQueryRequest");
@@ -574,7 +596,8 @@ let ConnectionsClient = class {
    * Reads the JSON schema of an entity and the operations it supports.
    *
    * @param entity The entity name.
-   * @throws Error When the API starts no operation for the entity.
+   * @throws Error When the API starts no operation for the entity, or the
+   *   operation fails or times out.
    */
   async getEntitySchemaAndOperations(entity) {
     const url = `${CONNECTOR_URL}/v1/projects/${this.project}/locations/${this.location}/connections/${this.connection}/connectionSchemaMetadata:getEntityType?entityId=${entity}`;
@@ -599,7 +622,8 @@ let ConnectionsClient = class {
    * Reads the input and output JSON schemas of an action.
    *
    * @param action The action name.
-   * @throws Error When the API starts no operation for the action.
+   * @throws Error When the API starts no operation for the action, or the
+   *   operation fails or times out.
    */
   async getActionSchema(action) {
     const url = `${CONNECTOR_URL}/v1/projects/${this.project}/locations/${this.location}/connections/${this.connection}/connectionSchemaMetadata:getAction?actionId=${action}`;
@@ -634,15 +658,31 @@ let ConnectionsClient = class {
       invalidRequestMessage: `Invalid request. Please check the provided values of project(${this.project}), location(${this.location}), connection(${this.connection}).`
     });
   }
-  /** Polls a long-running operation until it reports `done`. */
+  /**
+   * Polls a long-running operation until it reports `done`.
+   *
+   * @throws Error When the operation reports an `error` or does not finish
+   *   within `MAX_OPERATION_POLL_ATTEMPTS` polls.
+   */
   async pollOperation(operationId) {
     const url = `${CONNECTOR_URL}/v1/${operationId}`;
-    let operation = (0, import_json_utils.toJsonObject)(await this.executeApiCall(url));
-    while (operation["done"] !== true) {
-      await sleep(OPERATION_POLL_INTERVAL_MS);
-      operation = (0, import_json_utils.toJsonObject)(await this.executeApiCall(url));
+    for (let attempt = 0; attempt < MAX_OPERATION_POLL_ATTEMPTS; attempt++) {
+      if (attempt > 0) {
+        await sleep(OPERATION_POLL_INTERVAL_MS);
+      }
+      const operation = (0, import_json_utils.toJsonObject)(await this.executeApiCall(url));
+      if (operation["error"]) {
+        throw new Error(
+          `Operation ${operationId} failed: ${formatOperationError(operation["error"])}`
+        );
+      }
+      if (operation["done"] === true) {
+        return operation;
+      }
     }
-    return operation;
+    throw new Error(
+      `Operation ${operationId} timed out after ${MAX_OPERATION_POLL_ATTEMPTS} poll attempts.`
+    );
   }
 };
 __name(ConnectionsClient, "ConnectionsClient");

package/dist/cjs/tools/application_integration_tool/clients/integration_client.js

@@ -47,6 +47,7 @@ var import_connections_client = require("./connections_client.js");
  */
 const EXECUTE_CONNECTION_INTEGRATION = "ExecuteConnection";
 const EXECUTE_CUSTOM_QUERY_ACTION = "ExecuteCustomQuery";
+const REQUEST_SCHEMA_OPTIONS = Object.freeze({ includeDynamicAuthConfig: true });
 let IntegrationClient = class {
   constructor(options) {
     var _a, _b;
@@ -68,9 +69,7 @@ let IntegrationClient = class {
    *   that carries no parseable spec.
    */
   async getOpenApiSpecForIntegration() {
-    var _a;
     const url = `https://${this.location}-integrations.googleapis.com/v1/projects/${this.project}/locations/${this.location}:generateOpenApiSpec`;
-    const triggerId = (_a = this.triggers) != null ? _a : null;
     const invalidRequestMessage = `Invalid request. Please check the provided values of project(${this.project}), location(${this.location}), integration(${this.integration}).`;
     const body = await (0, import_cloud_request.sendCloudRequest)({
       url,
@@ -79,7 +78,7 @@ let IntegrationClient = class {
         apiTriggerResources: [
           {
             integrationResource: this.integration,
-            triggerId
+            triggerId: this.triggers
           }
         ],
         fileFormat: "JSON"
@@ -146,7 +145,10 @@ let IntegrationClient = class {
               toolName,
               toolInstructions
             });
-            schemas[`create_${entity}_Request`] = (0, import_connections_client.createOperationRequest)(entity);
+            schemas[`create_${entity}_Request`] = (0, import_connections_client.createOperationRequest)(
+              entity,
+              REQUEST_SCHEMA_OPTIONS
+            );
             break;
           case "update":
             connectorSpec.paths[path] = (0, import_connections_client.updateOperation)({
@@ -154,7 +156,10 @@ let IntegrationClient = class {
               toolName,
               toolInstructions
             });
-            schemas[`update_${entity}_Request`] = (0, import_connections_client.updateOperationRequest)(entity);
+            schemas[`update_${entity}_Request`] = (0, import_connections_client.updateOperationRequest)(
+              entity,
+              REQUEST_SCHEMA_OPTIONS
+            );
             break;
           case "delete":
             connectorSpec.paths[path] = (0, import_connections_client.deleteOperation)({
@@ -162,7 +167,9 @@ let IntegrationClient = class {
               toolName,
               toolInstructions
             });
-            schemas[`delete_${entity}_Request`] = (0, import_connections_client.deleteOperationRequest)();
+            schemas[`delete_${entity}_Request`] = (0, import_connections_client.deleteOperationRequest)(
+              REQUEST_SCHEMA_OPTIONS
+            );
             break;
           case "list":
             connectorSpec.paths[path] = (0, import_connections_client.listOperation)({
@@ -171,7 +178,9 @@ let IntegrationClient = class {
               toolName,
               toolInstructions
             });
-            schemas[`list_${entity}_Request`] = (0, import_connections_client.listOperationRequest)();
+            schemas[`list_${entity}_Request`] = (0, import_connections_client.listOperationRequest)(
+              REQUEST_SCHEMA_OPTIONS
+            );
             break;
           case "get":
             connectorSpec.paths[path] = (0, import_connections_client.getOperation)({
@@ -180,7 +189,9 @@ let IntegrationClient = class {
               toolName,
               toolInstructions
             });
-            schemas[`get_${entity}_Request`] = (0, import_connections_client.getOperationRequest)();
+            schemas[`get_${entity}_Request`] = (0, import_connections_client.getOperationRequest)(
+              REQUEST_SCHEMA_OPTIONS
+            );
             break;
           default:
             throw new Error(
@@ -194,10 +205,15 @@ let IntegrationClient = class {
       const actionDisplayName = displayName.replace(/ /g, "");
       let operation = "EXECUTE_ACTION";
       if (action === EXECUTE_CUSTOM_QUERY_ACTION) {
-        schemas[`${actionDisplayName}_Request`] = (0, import_connections_client.executeCustomQueryRequest)();
+        schemas[`${actionDisplayName}_Request`] = (0, import_connections_client.executeCustomQueryRequest)(
+          REQUEST_SCHEMA_OPTIONS
+        );
         operation = "EXECUTE_QUERY";
       } else {
-        schemas[`${actionDisplayName}_Request`] = (0, import_connections_client.actionRequest)(actionDisplayName);
+        schemas[`${actionDisplayName}_Request`] = (0, import_connections_client.actionRequest)(
+          actionDisplayName,
+          REQUEST_SCHEMA_OPTIONS
+        );
         schemas[`connectorInputPayload_${actionDisplayName}`] = (0, import_connections_client.connectorPayload)(inputSchema);
       }
       schemas[`connectorOutputPayload_${actionDisplayName}`] = (0, import_connections_client.connectorPayload)(outputSchema);

package/dist/cjs/tools/application_integration_tool/integration_connector_tool.js

@@ -45,10 +45,7 @@ var import_tool_auth_handler = require("../openapi_tool/openapi_spec_parser/tool
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const DYNAMIC_AUTH_EXCLUDE_FIELDS = Object.freeze([
-  "dynamic_auth_config",
-  "dynamicAuthConfig"
-]);
+const DYNAMIC_AUTH_CONFIG_FIELD = "dynamicAuthConfig";
 let IntegrationConnectorTool = class extends import_base_tool.BaseTool {
   constructor(options) {
     var _a, _b, _c;
@@ -71,17 +68,15 @@ let IntegrationConnectorTool = class extends import_base_tool.BaseTool {
       parameters = { ...rawParameters };
       if (rawParameters.properties) {
         const properties = { ...rawParameters.properties };
-        for (const field of [
-          ...IntegrationConnectorTool.EXCLUDE_FIELDS,
-          ...DYNAMIC_AUTH_EXCLUDE_FIELDS
-        ]) {
+        for (const field of IntegrationConnectorTool.EXCLUDE_FIELDS) {
           delete properties[field];
         }
+        delete properties[DYNAMIC_AUTH_CONFIG_FIELD];
         parameters.properties = properties;
       }
       if (rawParameters.required) {
         parameters.required = rawParameters.required.filter(
-          (field) => !IntegrationConnectorTool.EXCLUDE_FIELDS.includes(field) && !DYNAMIC_AUTH_EXCLUDE_FIELDS.includes(field) && !IntegrationConnectorTool.OPTIONAL_FIELDS.includes(field)
+          (field) => !IntegrationConnectorTool.EXCLUDE_FIELDS.includes(field) && field !== DYNAMIC_AUTH_CONFIG_FIELD && !IntegrationConnectorTool.OPTIONAL_FIELDS.includes(field)
         );
       }
     }
@@ -106,9 +101,7 @@ let IntegrationConnectorTool = class extends import_base_tool.BaseTool {
     }
     const updatedArgs = {
       ...request.args,
-      connection_name: this.connectionName,
       connectionName: this.connectionName,
-      service_name: this.connectionServiceName,
       serviceName: this.connectionServiceName,
       host: this.connectionHost
     };
@@ -123,11 +116,14 @@ let IntegrationConnectorTool = class extends import_base_tool.BaseTool {
     }
     if (authResult.authCredential) {
       const token = prepareDynamicEuc(authResult.authCredential);
-      const dynamicAuthPayload = {
-        "oauth2_auth_code_flow.access_token": token != null ? token : {}
+      if (!token) {
+        return {
+          error: "The resolved end-user credential has no access token."
+        };
+      }
+      updatedArgs[DYNAMIC_AUTH_CONFIG_FIELD] = {
+        "oauth2_auth_code_flow.access_token": token
       };
-      updatedArgs["dynamic_auth_config"] = dynamicAuthPayload;
-      updatedArgs["dynamicAuthConfig"] = dynamicAuthPayload;
     }
     import_logger.logger.debug(`Running tool: ${this.name}`);
     return this.restApiTool.runAsync({
@@ -142,9 +138,7 @@ __name(IntegrationConnectorTool, "IntegrationConnectorTool");
  * excludes from the function declaration shown to the model.
  */
 IntegrationConnectorTool.EXCLUDE_FIELDS = Object.freeze([
-  "connection_name",
   "connectionName",
-  "service_name",
   "serviceName",
   "host",
   "entity",
@@ -156,13 +150,10 @@ IntegrationConnectorTool.EXCLUDE_FIELDS = Object.freeze([
  * `required` list in the function declaration shown to the model.
  */
 IntegrationConnectorTool.OPTIONAL_FIELDS = Object.freeze([
-  "page_size",
   "pageSize",
-  "page_token",
   "pageToken",
-  "filter",
-  "filter_clause",
-  "filterClause"
+  "filterClause",
+  "sortByColumns"
 ]);
 __decorateClass([
   import_experimental.experimental
@@ -175,7 +166,7 @@ IntegrationConnectorTool = __decorateClass([
 ], IntegrationConnectorTool);
 function prepareDynamicEuc(authCredential) {
   var _a, _b;
-  return ((_b = (_a = authCredential.http) == null ? void 0 : _a.credentials) == null ? void 0 : _b.token) || void 0;
+  return (_b = (_a = authCredential.http) == null ? void 0 : _a.credentials) == null ? void 0 : _b.token;
 }
 __name(prepareDynamicEuc, "prepareDynamicEuc");
 // Annotate the CommonJS export names for ESM import in node:

package/dist/cjs/tools/base_toolset.js

@@ -42,7 +42,7 @@ function isBaseToolset(obj) {
 __name(isBaseToolset, "isBaseToolset");
 _a = BASE_TOOLSET_SIGNATURE_SYMBOL;
 const _BaseToolset = class _BaseToolset {
-  constructor(toolFilter, prefix) {
+  constructor(toolFilter = [], prefix) {
     this.toolFilter = toolFilter;
     this.prefix = prefix;
     this[_a] = true;

package/dist/cjs/tools/computer_use/base_computer.js

@@ -0,0 +1,125 @@
+/**
+  * @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 __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+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 __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 base_computer_exports = {};
+__export(base_computer_exports, {
+  BaseComputer: () => BaseComputer,
+  ComputerEnvironment: () => ComputerEnvironment,
+  checkComputerUseEnabled: () => checkComputerUseEnabled,
+  isBaseComputer: () => isBaseComputer
+});
+module.exports = __toCommonJS(base_computer_exports);
+var import_feature_registry = require("../../features/feature_registry.js");
+var import_experimental = require("../../utils/experimental.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+var _a;
+var ComputerEnvironment = /* @__PURE__ */ ((ComputerEnvironment2) => {
+  ComputerEnvironment2["ENVIRONMENT_UNSPECIFIED"] = "ENVIRONMENT_UNSPECIFIED";
+  ComputerEnvironment2["ENVIRONMENT_BROWSER"] = "ENVIRONMENT_BROWSER";
+  return ComputerEnvironment2;
+})(ComputerEnvironment || {});
+function checkComputerUseEnabled() {
+  if (!(0, import_feature_registry.isFeatureEnabled)(import_feature_registry.FeatureName.COMPUTER_USE)) {
+    throw new Error(`Feature ${import_feature_registry.FeatureName.COMPUTER_USE} is not enabled.`);
+  }
+}
+__name(checkComputerUseEnabled, "checkComputerUseEnabled");
+const BASE_COMPUTER_SIGNATURE_SYMBOL = /* @__PURE__ */ Symbol.for("google.adk.baseComputer");
+function isBaseComputer(obj) {
+  return typeof obj === "object" && obj !== null && BASE_COMPUTER_SIGNATURE_SYMBOL in obj && obj[BASE_COMPUTER_SIGNATURE_SYMBOL] === true;
+}
+__name(isBaseComputer, "isBaseComputer");
+_a = BASE_COMPUTER_SIGNATURE_SYMBOL;
+let BaseComputer = class {
+  constructor() {
+    /** A unique symbol to identify ADK base computer class. */
+    this[_a] = true;
+    checkComputerUseEnabled();
+  }
+  /** Initializes the computer. */
+  async initialize() {
+  }
+  /** Cleans up the resources of the computer. */
+  async close() {
+  }
+  /**
+   * Called before each tool invocation to prepare resources.
+   *
+   * Override this to set up session-level resources (a sandbox, tokens, and so
+   * on), using the state of `toolContext` for persistence across invocations.
+   *
+   * @param _toolContext The tool context with session state access.
+   */
+  async prepare(_toolContext) {
+  }
+  /**
+   * Types text into the currently focused element.
+   *
+   * By default, this calls {@link keyCombination} once per character in `text`
+   * (and once for `"Enter"` when `pressEnter` is true). Because
+   * `keyCombination` is designed for key names and typically captures a new
+   * screenshot on every call, this fallback costs one round trip and screenshot
+   * per character, and relies on the backend accepting single characters in
+   * `keyCombination`. Subclasses should override this method to use native
+   * bulk text input.
+   *
+   * @param text The text to type.
+   * @param pressEnter Whether to press ENTER after typing.
+   * @returns The current state after typing.
+   */
+  async typeText(text, pressEnter = false) {
+    let state;
+    for (const char of text) {
+      state = await this.keyCombination([char]);
+    }
+    if (pressEnter) {
+      state = await this.keyCombination(["Enter"]);
+    }
+    return state != null ? state : this.currentState();
+  }
+};
+__name(BaseComputer, "BaseComputer");
+BaseComputer = __decorateClass([
+  import_experimental.experimental
+], BaseComputer);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  BaseComputer,
+  ComputerEnvironment,
+  checkComputerUseEnabled,
+  isBaseComputer
+});