@google/adk 1.2.0 → 1.4.0

package/dist/cjs/tools/openapi_tool/openapi_spec_parser/operation_parser.js

@@ -0,0 +1,224 @@
+/**
+  * @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 __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 operation_parser_exports = {};
+__export(operation_parser_exports, {
+  OperationParser: () => OperationParser
+});
+module.exports = __toCommonJS(operation_parser_exports);
+var import_experimental = require("../../../utils/experimental.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+let OperationParser = class {
+  constructor(operation, options = {}) {
+    this.operation = operation;
+    this.params = [];
+    var _a;
+    this.preservePropertyNames = (_a = options.preservePropertyNames) != null ? _a : false;
+    this.processOperationParameters();
+    this.processRequestBody();
+    this.processReturnValue();
+    this.dedupeParamNames();
+  }
+  getParamName(originalName) {
+    if (this.preservePropertyNames) {
+      return originalName;
+    }
+    return originalName.replace(/[A-Z]/g, (g) => "_" + g.toLowerCase()).replace(/^_/, "");
+  }
+  processOperationParameters() {
+    const parameters = this.operation.parameters || [];
+    for (const param of parameters) {
+      if ("name" in param) {
+        const originalName = param.name;
+        const description = param.description || "";
+        const location = param.in || "";
+        const schema = param.schema || {};
+        this.params.push({
+          originalName,
+          paramLocation: location,
+          paramSchema: schema,
+          description,
+          required: param.required || false,
+          name: this.getParamName(originalName)
+        });
+      }
+    }
+  }
+  processRequestBody() {
+    const requestBody = this.operation.requestBody;
+    if (!requestBody || "$ref" in requestBody) {
+      return;
+    }
+    const content = requestBody.content || {};
+    const firstMimeType = Object.keys(content)[0];
+    if (!firstMimeType) {
+      return;
+    }
+    const mediaTypeObject = content[firstMimeType];
+    const schema = mediaTypeObject.schema;
+    const description = requestBody.description || "";
+    if (schema && !("$ref" in schema)) {
+      if (schema.type === "object") {
+        const properties = schema.properties || {};
+        if (Object.keys(properties).length > 0) {
+          for (const [propName, propDetails] of Object.entries(properties)) {
+            if (!("$ref" in propDetails)) {
+              this.params.push({
+                originalName: propName,
+                paramLocation: "body",
+                paramSchema: propDetails,
+                description: propDetails.description,
+                required: (schema.required || []).includes(propName),
+                name: this.getParamName(propName)
+              });
+            }
+          }
+        } else {
+          this.params.push({
+            originalName: "",
+            paramLocation: "body",
+            paramSchema: schema,
+            description,
+            required: true,
+            name: "body"
+          });
+        }
+      } else if (schema.type === "array") {
+        this.params.push({
+          originalName: "array",
+          paramLocation: "body",
+          paramSchema: schema,
+          description,
+          required: true,
+          name: "body"
+        });
+      } else {
+        this.params.push({
+          originalName: "body",
+          paramLocation: "body",
+          paramSchema: schema,
+          description,
+          required: true,
+          name: "body"
+        });
+      }
+    }
+  }
+  processReturnValue() {
+    const responses = this.operation.responses || {};
+    const validCodes = Object.keys(responses).filter((k) => k.startsWith("2"));
+    const min20x = validCodes.sort()[0];
+    let returnSchema = {};
+    if (min20x) {
+      const response = responses[min20x];
+      if (!("$ref" in response) && response.content) {
+        const firstMimeType = Object.keys(response.content)[0];
+        if (firstMimeType) {
+          const schema = response.content[firstMimeType].schema;
+          if (schema && !("$ref" in schema)) {
+            returnSchema = schema;
+          }
+        }
+      }
+    }
+    this.returnValue = {
+      originalName: "",
+      paramLocation: "",
+      paramSchema: returnSchema,
+      required: true,
+      name: "return"
+    };
+  }
+  dedupeParamNames() {
+    const nameCounts = /* @__PURE__ */ new Map();
+    for (const param of this.params) {
+      const name = param.name;
+      const count = nameCounts.get(name) || 0;
+      if (count > 0) {
+        param.name = `${name}_${count}`;
+      }
+      nameCounts.set(name, count + 1);
+    }
+  }
+  getParameters() {
+    return this.params;
+  }
+  getJsonSchema() {
+    const properties = {};
+    const required = [];
+    for (const param of this.params) {
+      properties[param.name] = param.paramSchema;
+      if (param.required) {
+        required.push(param.name);
+      }
+    }
+    return {
+      type: "object",
+      properties,
+      required: required.length > 0 ? required : void 0,
+      title: `${this.operation.operationId || "unnamed"}_Arguments`
+    };
+  }
+  getFunctionName() {
+    const operationId = this.operation.operationId;
+    if (!operationId) {
+      throw new Error("Operation ID is missing");
+    }
+    return this.getParamName(operationId).substring(0, 60);
+  }
+  getDescription() {
+    return this.operation.description || this.operation.summary || "";
+  }
+};
+__decorateClass([
+  import_experimental.experimental
+], OperationParser.prototype, "getParameters", 1);
+__decorateClass([
+  import_experimental.experimental
+], OperationParser.prototype, "getJsonSchema", 1);
+__decorateClass([
+  import_experimental.experimental
+], OperationParser.prototype, "getFunctionName", 1);
+__decorateClass([
+  import_experimental.experimental
+], OperationParser.prototype, "getDescription", 1);
+OperationParser = __decorateClass([
+  import_experimental.experimental
+], OperationParser);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  OperationParser
+});

package/dist/cjs/tools/openapi_tool/openapi_spec_parser/tool_auth_handler.js

@@ -0,0 +1,117 @@
+/**
+  * @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 __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 tool_auth_handler_exports = {};
+__export(tool_auth_handler_exports, {
+  ToolAuthHandler: () => ToolAuthHandler
+});
+module.exports = __toCommonJS(tool_auth_handler_exports);
+var import_experimental = require("../../../utils/experimental.js");
+var import_auto_auth_credential_exchanger = require("../auth/credential_exchangers/auto_auth_credential_exchanger.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class ToolContextCredentialStore {
+  constructor(context) {
+    this.context = context;
+  }
+  getCredentialKey(authScheme) {
+    const schemeName = (authScheme == null ? void 0 : authScheme.type) || "default";
+    return `${schemeName}_existing_exchanged_credential`;
+  }
+  getCredential(authScheme) {
+    const key = this.getCredentialKey(authScheme);
+    return this.context.state.get(key);
+  }
+  storeCredential(key, credential) {
+    this.context.state.set(key, credential);
+  }
+}
+let ToolAuthHandler = class {
+  constructor(context, authScheme, authCredential, credentialKey) {
+    this.context = context;
+    this.authScheme = authScheme;
+    this.authCredential = authCredential;
+    this.credentialKey = credentialKey;
+  }
+  static fromToolContext(context, authScheme, authCredential, options = {}) {
+    return new ToolAuthHandler(
+      context,
+      authScheme,
+      authCredential,
+      options.credentialKey
+    );
+  }
+  async prepareAuthCredentials() {
+    if (!this.authScheme) {
+      return { state: "done" };
+    }
+    const store = new ToolContextCredentialStore(this.context);
+    const existingCredential = store.getCredential(this.authScheme);
+    if (existingCredential) {
+      return { state: "done", authCredential: existingCredential };
+    }
+    const authConfig = {
+      authScheme: this.authScheme,
+      rawAuthCredential: this.authCredential,
+      credentialKey: this.credentialKey || "default_openapi_key"
+    };
+    const credential = this.context.getAuthResponse(authConfig);
+    if (credential) {
+      const exchanger = new import_auto_auth_credential_exchanger.AutoAuthCredentialExchanger();
+      const result = await exchanger.exchange({
+        authScheme: this.authScheme,
+        authCredential: credential
+      });
+      const key = store.getCredentialKey(this.authScheme);
+      store.storeCredential(key, result.credential);
+      return { state: "done", authCredential: result.credential };
+    }
+    this.context.requestCredential(authConfig);
+    return { state: "pending" };
+  }
+};
+__decorateClass([
+  import_experimental.experimental
+], ToolAuthHandler.prototype, "prepareAuthCredentials", 1);
+__decorateClass([
+  import_experimental.experimental
+], ToolAuthHandler, "fromToolContext", 1);
+ToolAuthHandler = __decorateClass([
+  import_experimental.experimental
+], ToolAuthHandler);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ToolAuthHandler
+});

package/dist/cjs/tools/openapi_tool/openapi_toolset.js

@@ -0,0 +1,133 @@
+/**
+  * @license
+  * Copyright 2026 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+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 __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+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 openapi_toolset_exports = {};
+__export(openapi_toolset_exports, {
+  OpenAPIToolset: () => OpenAPIToolset
+});
+module.exports = __toCommonJS(openapi_toolset_exports);
+var import_js_yaml = __toESM(require("js-yaml"), 1);
+var import_experimental = require("../../utils/experimental.js");
+var import_base_toolset = require("../base_toolset.js");
+var import_openapi_spec_parser = require("./openapi_spec_parser/openapi_spec_parser.js");
+var import_rest_api_tool = require("./rest_api_tool.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+let OpenAPIToolset = class extends import_base_toolset.BaseToolset {
+  constructor(options = {}) {
+    super(options.toolFilter || [], options.prefix);
+    this.tools = [];
+    let spec = options.specDict;
+    if (!spec && options.specStr) {
+      if (options.specType === "yaml" || !options.specType && options.specStr.trim().startsWith("---")) {
+        spec = import_js_yaml.default.load(options.specStr);
+      } else {
+        spec = JSON.parse(options.specStr);
+      }
+    }
+    if (!spec) {
+      throw new Error("Either specDict or specStr must be provided.");
+    }
+    const parser = new import_openapi_spec_parser.OpenApiSpecParser({
+      preservePropertyNames: options.preservePropertyNames
+    });
+    const parsedOperations = parser.parse(spec);
+    for (const op of parsedOperations) {
+      let toolName = op.name;
+      if (this.prefix) {
+        toolName = `${this.prefix}_${toolName}`;
+      }
+      const tool = (0, import_rest_api_tool.createRestApiTool)(
+        {
+          name: toolName,
+          description: op.description,
+          endpoint: op.endpoint,
+          operation: op.operation,
+          authScheme: op.authScheme
+        },
+        {
+          preservePropertyNames: options.preservePropertyNames,
+          headerProvider: options.headerProvider,
+          credentialKey: options.credentialKey
+        }
+      );
+      this.tools.push(tool);
+    }
+    if (options.authScheme || options.authCredential) {
+      for (const tool of this.tools) {
+        if (options.authScheme) tool.configureAuthScheme(options.authScheme);
+        if (options.authCredential)
+          tool.configureAuthCredential(options.authCredential);
+      }
+    }
+  }
+  async getTools(context) {
+    return this.tools.filter((tool) => {
+      if (Array.isArray(this.toolFilter) && this.toolFilter.length > 0) {
+        return this.toolFilter.includes(tool.name);
+      }
+      if (context) {
+        return this.isToolSelected(tool, context);
+      }
+      return true;
+    });
+  }
+  async close() {
+    return Promise.resolve();
+  }
+};
+__decorateClass([
+  import_experimental.experimental
+], OpenAPIToolset.prototype, "getTools", 1);
+__decorateClass([
+  import_experimental.experimental
+], OpenAPIToolset.prototype, "close", 1);
+OpenAPIToolset = __decorateClass([
+  import_experimental.experimental
+], OpenAPIToolset);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  OpenAPIToolset
+});