@google/adk 1.0.0 → 1.2.0

package/dist/cjs/tools/vertex_ai_search_tool.js

@@ -0,0 +1,127 @@
+/**
+  * @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 vertex_ai_search_tool_exports = {};
+__export(vertex_ai_search_tool_exports, {
+  VertexAiSearchTool: () => VertexAiSearchTool
+});
+module.exports = __toCommonJS(vertex_ai_search_tool_exports);
+var import_logger = require("../utils/logger.js");
+var import_model_name = require("../utils/model_name.js");
+var import_base_tool = require("./base_tool.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const logger = (0, import_logger.getLogger)();
+class VertexAiSearchTool extends import_base_tool.BaseTool {
+  constructor(params) {
+    super({ name: "vertex_ai_search", description: "vertex_ai_search" });
+    const {
+      dataStoreId,
+      dataStoreSpecs,
+      searchEngineId,
+      filter,
+      maxResults,
+      bypassMultiToolsLimit = false
+    } = params;
+    if (dataStoreId === void 0 && searchEngineId === void 0 || dataStoreId !== void 0 && searchEngineId !== void 0) {
+      throw new Error(
+        "Either dataStoreId or searchEngineId must be specified."
+      );
+    }
+    if (dataStoreSpecs !== void 0 && searchEngineId === void 0) {
+      throw new Error(
+        "searchEngineId must be specified if dataStoreSpecs is specified."
+      );
+    }
+    this.dataStoreId = dataStoreId;
+    this.dataStoreSpecs = dataStoreSpecs;
+    this.searchEngineId = searchEngineId;
+    this.filter = filter;
+    this.maxResults = maxResults;
+    this.bypassMultiToolsLimit = bypassMultiToolsLimit;
+  }
+  runAsync() {
+    return Promise.resolve();
+  }
+  /**
+   * Builds the VertexAISearch configuration.
+   *
+   * Override this method in a subclass to dynamically customize the search
+   * configuration based on the context (e.g., set filter based on session
+   * state).
+   */
+  buildVertexAiSearchConfig(_readonlyContext) {
+    return {
+      datastore: this.dataStoreId,
+      dataStoreSpecs: this.dataStoreSpecs,
+      engine: this.searchEngineId,
+      filter: this.filter,
+      maxResults: this.maxResults
+    };
+  }
+  async processLlmRequest({
+    toolContext,
+    llmRequest
+  }) {
+    if (!llmRequest.model) {
+      return;
+    }
+    const modelCheckDisabled = (0, import_model_name.isGeminiModelIdCheckDisabled)();
+    llmRequest.config = llmRequest.config || {};
+    llmRequest.config.tools = llmRequest.config.tools || [];
+    if (!(0, import_model_name.isGeminiModel)(llmRequest.model) && !modelCheckDisabled) {
+      throw new Error(
+        `Vertex AI search tool is not supported for model ${llmRequest.model}`
+      );
+    }
+    if ((0, import_model_name.isGemini1Model)(llmRequest.model) && llmRequest.config.tools.length > 0 && !this.bypassMultiToolsLimit) {
+      throw new Error(
+        "Vertex AI search tool cannot be used with other tools in Gemini 1.x."
+      );
+    }
+    const vertexAiSearchConfig = this.buildVertexAiSearchConfig(toolContext);
+    let specsInfo;
+    if (vertexAiSearchConfig.dataStoreSpecs) {
+      const specIds = vertexAiSearchConfig.dataStoreSpecs.map(
+        (spec) => spec.dataStore ? spec.dataStore.split("/").pop() : "unnamed"
+      );
+      specsInfo = `${vertexAiSearchConfig.dataStoreSpecs.length} spec(s): [${specIds.join(", ")}]`;
+    }
+    logger.debug(
+      `Adding Vertex AI Search tool config to LLM request: datastore=${vertexAiSearchConfig.datastore}, engine=${vertexAiSearchConfig.engine}, filter=${vertexAiSearchConfig.filter}, maxResults=${vertexAiSearchConfig.maxResults}, dataStoreSpecs=${specsInfo}`
+    );
+    llmRequest.config.tools.push({
+      retrieval: {
+        vertexAiSearch: vertexAiSearchConfig
+      }
+    });
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  VertexAiSearchTool
+});

package/dist/cjs/tools/vertex_rag_retrieval_tool.js

@@ -0,0 +1,64 @@
+/**
+  * @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 vertex_rag_retrieval_tool_exports = {};
+__export(vertex_rag_retrieval_tool_exports, {
+  VertexRagRetrievalTool: () => VertexRagRetrievalTool
+});
+module.exports = __toCommonJS(vertex_rag_retrieval_tool_exports);
+var import_base_tool = require("./base_tool.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class VertexRagRetrievalTool extends import_base_tool.BaseTool {
+  constructor(config) {
+    super({
+      name: "vertex_rag_retrieval",
+      description: "Vertex AI RAG Retrieval Tool"
+    });
+    this.vertexRagStore = config;
+  }
+  /**
+   * This tool is executed server-side by the Vertex AI RAG Engine.
+   * Local execution is not required.
+   */
+  runAsync() {
+    return Promise.resolve();
+  }
+  async processLlmRequest({
+    llmRequest
+  }) {
+    llmRequest.config = llmRequest.config || {};
+    llmRequest.config.tools = llmRequest.config.tools || [];
+    llmRequest.config.tools.push({
+      retrieval: { vertexRagStore: this.vertexRagStore }
+    });
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  VertexRagRetrievalTool
+});

package/dist/cjs/utils/file_utils.js

@@ -34,6 +34,8 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var file_utils_exports = {};
 __export(file_utils_exports, {
+  EXTENSION_TO_MIME_TYPE: () => EXTENSION_TO_MIME_TYPE,
+  guessMimeType: () => guessMimeType,
   materializeFiles: () => materializeFiles
 });
 module.exports = __toCommonJS(file_utils_exports);
@@ -88,7 +90,33 @@ async function materializeFiles(files, dir = process.cwd()) {
   }
   return createdFiles;
 }
+const EXTENSION_TO_MIME_TYPE = {
+  "pdf": "application/pdf",
+  "jpg": "image/jpeg",
+  "jpeg": "image/jpeg",
+  "png": "image/png",
+  "gif": "image/gif",
+  "csv": "text/csv",
+  "json": "application/json",
+  "xml": "application/xml",
+  "sh": "text/x-shellscript",
+  "bash": "text/x-shellscript",
+  "py": "text/x-python",
+  "js": "text/javascript",
+  "cjs": "text/javascript",
+  "mjs": "text/javascript",
+  "ts": "text/javascript",
+  "cts": "text/javascript",
+  "mts": "text/javascript"
+};
+function guessMimeType(filePath) {
+  var _a;
+  const ext = ((_a = filePath.split(".").pop()) == null ? void 0 : _a.toLowerCase()) || "";
+  return EXTENSION_TO_MIME_TYPE[ext] || "application/octet-stream";
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  EXTENSION_TO_MIME_TYPE,
+  guessMimeType,
   materializeFiles
 });

package/dist/cjs/utils/gemini_schema_util.js

@@ -108,6 +108,21 @@ function toGeminiSchema(mcpSchema) {
         mcp.type = "array";
       } else if (isNullable) {
         mcp.type = "null";
+      } else if (mcp.enum) {
+        const enumTypes = new Set(mcp.enum.map((v) => typeof v));
+        if (enumTypes.size === 1) {
+          const jsType = [...enumTypes][0];
+          if (jsType === "string") mcp.type = "string";
+          else if (jsType === "number") mcp.type = "number";
+          else if (jsType === "boolean") mcp.type = "boolean";
+        }
+      } else if (mcp.const !== void 0) {
+        const jsType = typeof mcp.const;
+        let inferredType;
+        if (jsType === "string") inferredType = "string";
+        else if (jsType === "number") inferredType = "number";
+        else if (jsType === "boolean") inferredType = "boolean";
+        mcp = { ...mcp, type: inferredType, enum: [mcp.const] };
       }
     }
     const geminiType = toGeminiType(mcp.type);
@@ -122,6 +137,9 @@ function toGeminiSchema(mcpSchema) {
     if (mcp.description) {
       geminiSchema.description = mcp.description;
     }
+    if (mcp.enum) {
+      geminiSchema.enum = mcp.enum.map(String);
+    }
     if (isNullable && mcp.type !== "null") {
       geminiSchema.nullable = true;
     }

package/dist/cjs/utils/model_name.js

@@ -27,9 +27,11 @@ __export(model_name_exports, {
   extractModelName: () => extractModelName,
   isGemini1Model: () => isGemini1Model,
   isGemini2OrAbove: () => isGemini2OrAbove,
-  isGeminiModel: () => isGeminiModel
+  isGeminiModel: () => isGeminiModel,
+  isGeminiModelIdCheckDisabled: () => isGeminiModelIdCheckDisabled
 });
 module.exports = __toCommonJS(model_name_exports);
+var import_env_aware_utils = require("./env_aware_utils.js");
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -75,10 +77,14 @@ function isGemini2OrAbove(modelString) {
   const parsedVersion = parseVersion(versionString);
   return parsedVersion.valid && parsedVersion.major >= 2;
 }
+function isGeminiModelIdCheckDisabled() {
+  return (0, import_env_aware_utils.getBooleanEnvVar)("ADK_DISABLE_GEMINI_MODEL_ID_CHECK");
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   extractModelName,
   isGemini1Model,
   isGemini2OrAbove,
-  isGeminiModel
+  isGeminiModel,
+  isGeminiModelIdCheckDisabled
 });

package/dist/cjs/utils/partial_copy.js

@@ -0,0 +1,51 @@
+/**
+  * @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 partial_copy_exports = {};
+__export(partial_copy_exports, {
+  partialCopy: () => partialCopy
+});
+module.exports = __toCommonJS(partial_copy_exports);
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function partialCopy(source, targetKeys) {
+  const result = {};
+  const sourceAsUnknown = source;
+  targetKeys.forEach((key) => {
+    const keyStr = key;
+    if (keyStr in source) {
+      result[key] = sourceAsUnknown[keyStr];
+    } else {
+      result[key] = void 0;
+    }
+  });
+  return result;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  partialCopy
+});