@google/adk 2.3.0 → 2.5.0

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

@@ -33,18 +33,24 @@ var __decorateClass = (decorators, target, key, kind) => {
 };
 var application_integration_toolset_exports = {};
 __export(application_integration_toolset_exports, {
-  ApplicationIntegrationToolset: () => ApplicationIntegrationToolset
+  ApplicationIntegrationToolset: () => ApplicationIntegrationToolset,
+  IntegrationConnectorTool: () => import_integration_connector_tool.IntegrationConnectorTool
 });
 module.exports = __toCommonJS(application_integration_toolset_exports);
 var import_auth_credential = require("../../auth/auth_credential.js");
+var import_case_utils = require("../../utils/case_utils.js");
 var import_experimental = require("../../utils/experimental.js");
 var import_json_utils = require("../../utils/json_utils.js");
+var import_logger = require("../../utils/logger.js");
 var import_base_toolset = require("../base_toolset.js");
 var import_auth_helpers = require("../openapi_tool/auth/auth_helpers.js");
+var import_openapi_spec_parser = require("../openapi_tool/openapi_spec_parser/openapi_spec_parser.js");
 var import_openapi_toolset = require("../openapi_tool/openapi_toolset.js");
+var import_rest_api_tool = require("../openapi_tool/rest_api_tool.js");
 var import_cloud_request = require("./clients/cloud_request.js");
 var import_connections_client = require("./clients/connections_client.js");
 var import_integration_client = require("./clients/integration_client.js");
+var import_integration_connector_tool = require("./integration_connector_tool.js");
 /**
  * @license
  * Copyright 2026 Google LLC
@@ -57,19 +63,20 @@ const AUTH_SCHEME = {
 };
 let ApplicationIntegrationToolset = class extends import_base_toolset.BaseToolset {
   /**
-   * @throws Error Unless the options name either an integration and a trigger,
-   *   or a connection with at least one entity operation or action.
+   * @throws Error Unless the options name either an integration, or a
+   *   connection with at least one entity operation or action.
    */
   constructor(options) {
-    var _a, _b;
-    super([]);
-    const hasIntegration = Boolean(options.integration && options.trigger);
+    var _a, _b, _c;
+    super((_a = options.toolFilter) != null ? _a : []);
+    this.openApiToolsets = [];
+    const hasIntegration = Boolean(options.integration);
     const hasConnection = Boolean(
-      options.connection && (Object.keys((_a = options.entityOperations) != null ? _a : {}).length > 0 || ((_b = options.actions) != null ? _b : []).length > 0)
+      options.connection && (Object.keys((_b = options.entityOperations) != null ? _b : {}).length > 0 || ((_c = options.actions) != null ? _c : []).length > 0)
     );
     if (!hasIntegration && !hasConnection) {
       throw new Error(
-        "Either (integration and trigger) or (connection and (entityOperations or actions)) should be provided."
+        "Either integration or (connection and (entityOperations or actions)) should be provided."
       );
     }
     this.options = options;
@@ -80,72 +87,122 @@ let ApplicationIntegrationToolset = class extends import_base_toolset.BaseToolse
    * The first call fetches the spec and builds the tools. Concurrent calls
    * share that load. When it fails, the next call tries again.
    */
-  async getTools() {
+  async getTools(context) {
     if (!this.toolsPromise) {
       this.toolsPromise = this.loadTools();
     }
     try {
-      return await this.toolsPromise;
+      const tools = await this.toolsPromise;
+      return tools.filter((tool) => this.isToolSelected(tool, context));
     } catch (e) {
       this.toolsPromise = void 0;
       throw e;
     }
   }
   async close() {
+    for (const openApiToolset of this.openApiToolsets) {
+      await openApiToolset.close();
+    }
   }
   async loadTools() {
-    var _a;
+    var _a, _b;
     const {
       project,
       location,
       integration,
-      trigger,
+      triggers,
       connection,
       entityOperations,
       actions,
-      toolName = "",
+      toolNamePrefix,
       toolInstructions = "",
-      serviceAccountJson
+      serviceAccountJson,
+      authScheme,
+      authCredential: dynamicAuthCredential
     } = this.options;
+    const resolvedToolName = toolNamePrefix != null ? toolNamePrefix : "";
+    this.openApiToolsets = [];
     const integrationClient = new import_integration_client.IntegrationClient({
       project,
       location,
       integration,
-      trigger,
+      triggers,
       connection,
       entityOperations,
       actions,
       serviceAccountJson
     });
-    let specDict;
-    if (integration && trigger) {
-      specDict = await integrationClient.getOpenApiSpecForIntegration();
-    } else {
-      const connectionName = connection != null ? connection : "";
-      const { serviceName, host } = await new import_connections_client.ConnectionsClient({
-        project,
-        location,
-        connection: connectionName,
-        serviceAccountJson
-      }).getConnectionDetails();
-      const instructions = `${toolInstructions}ALWAYS use serviceName = ${serviceName}, host = ${host} and the connection name = projects/${project}/locations/${location}/connections/${connectionName} when using this tool. DO NOT ask the user for these values as you already have those.`;
-      specDict = await integrationClient.getOpenApiSpecForConnection(
-        toolName,
-        instructions
-      );
-    }
-    const spec = specDict;
     const authCredential = buildAuthCredential(serviceAccountJson);
     const toolsByName = /* @__PURE__ */ new Map();
-    for (const [path, pathItem] of Object.entries((_a = spec.paths) != null ? _a : {})) {
-      const tools = await new import_openapi_toolset.OpenAPIToolset({
-        specDict: { ...spec, paths: { [removeFragment(path)]: pathItem } },
-        authCredential,
-        authScheme: AUTH_SCHEME
-      }).getTools();
-      for (const tool of tools) {
-        toolsByName.set(tool.name, tool);
+    if (integration) {
+      const specRecord2 = await integrationClient.getOpenApiSpecForIntegration();
+      const spec2 = toOpenApiDocument(specRecord2);
+      for (const [path, pathItem] of Object.entries((_a = spec2.paths) != null ? _a : {})) {
+        const openApiToolset = new import_openapi_toolset.OpenAPIToolset({
+          specDict: { ...spec2, paths: { [removeFragment(path)]: pathItem } },
+          authCredential,
+          authScheme: AUTH_SCHEME
+        });
+        this.openApiToolsets.push(openApiToolset);
+        const tools = await openApiToolset.getTools();
+        for (const tool of tools) {
+          toolsByName.set(tool.name, tool);
+        }
       }
+      return [...toolsByName.values()];
+    }
+    const connectionName = connection != null ? connection : "";
+    const connectionDetails = await new import_connections_client.ConnectionsClient({
+      project,
+      location,
+      connection: connectionName,
+      serviceAccountJson
+    }).getConnectionDetails();
+    if (authScheme && dynamicAuthCredential && !connectionDetails.authOverrideEnabled) {
+      import_logger.logger.warn(
+        "Authentication schema and credentials are not used because authOverrideEnabled is not enabled in the connection."
+      );
+    }
+    const specRecord = await integrationClient.getOpenApiSpecForConnection(
+      resolvedToolName,
+      toolInstructions
+    );
+    const spec = toOpenApiDocument(specRecord);
+    const operations = new import_openapi_spec_parser.OpenApiSpecParser({
+      preservePropertyNames: true
+    }).parse(spec);
+    for (const openApiOperation of operations) {
+      const operationObject = (0, import_json_utils.toJsonObject)(openApiOperation.operation);
+      const operation = toStringValue(operationObject["x-operation"]);
+      const entity = toStringValue(operationObject["x-entity"]);
+      const action = toStringValue(operationObject["x-action"]);
+      const restApiTool = (0, import_rest_api_tool.createRestApiTool)(
+        {
+          ...openApiOperation,
+          name: (0, import_case_utils.toSnakeCaseName)(openApiOperation.name),
+          endpoint: {
+            ...openApiOperation.endpoint,
+            path: removeFragment(openApiOperation.endpoint.path)
+          }
+        },
+        { preservePropertyNames: true }
+      );
+      restApiTool.configureAuthScheme(AUTH_SCHEME);
+      restApiTool.configureAuthCredential(authCredential);
+      const tool = new import_integration_connector_tool.IntegrationConnectorTool({
+        name: restApiTool.name,
+        description: restApiTool.description,
+        connectionName: (_b = connectionDetails.name) != null ? _b : "",
+        connectionHost: connectionDetails.host,
+        connectionServiceName: connectionDetails.serviceName,
+        entity,
+        action,
+        operation,
+        restApiTool,
+        authScheme: connectionDetails.authOverrideEnabled ? authScheme : void 0,
+        authCredential: connectionDetails.authOverrideEnabled ? dynamicAuthCredential : void 0
+      });
+      toolsByName.set(tool.name, tool);
     }
     return [...toolsByName.values()];
   }
@@ -154,6 +211,14 @@ __name(ApplicationIntegrationToolset, "ApplicationIntegrationToolset");
 ApplicationIntegrationToolset = __decorateClass([
   import_experimental.experimental
 ], ApplicationIntegrationToolset);
+function toOpenApiDocument(specRecord) {
+  return specRecord;
+}
+__name(toOpenApiDocument, "toOpenApiDocument");
+function toStringValue(value) {
+  return typeof value === "string" ? value : "";
+}
+__name(toStringValue, "toStringValue");
 function buildAuthCredential(serviceAccountJson) {
   if (!serviceAccountJson) {
     return {
@@ -168,7 +233,7 @@ function buildAuthCredential(serviceAccountJson) {
   if (!(0, import_json_utils.isJsonObject)(key)) {
     throw new Error("serviceAccountJson must hold a JSON object.");
   }
-  const [, authCredential] = (0, import_auth_helpers.serviceAccountDictToSchemeCredential)(key, [
+  const [, authCredential] = (0, import_auth_helpers.serviceAccountRecordToSchemeCredential)(key, [
     import_cloud_request.CLOUD_PLATFORM_SCOPE
   ]);
   return authCredential;
@@ -181,5 +246,6 @@ function removeFragment(path) {
 __name(removeFragment, "removeFragment");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  ApplicationIntegrationToolset
+  ApplicationIntegrationToolset,
+  IntegrationConnectorTool
 });