@google/adk 2.0.0 → 2.1.0

package/dist/web/code_executors/agent_engine_sandbox_code_executor.js

@@ -1,284 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-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;
-};
-/**
- * @license
- * Copyright 2026 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { Client } from "@google-cloud/vertexai";
-import { Language } from "@google-cloud/vertexai/build/src/genai/types.js";
-import { experimental } from "../utils/experimental.js";
-import { guessMimeType } from "../utils/file_utils.js";
-const SANDBOX_PATTERN = /^projects\/([a-zA-Z0-9-_]+)\/locations\/([a-zA-Z0-9-_]+)\/reasoningEngines\/(\d+)\/sandboxEnvironments\/(\d+)$/;
-const ENGINE_PATTERN = /^projects\/([a-zA-Z0-9-_]+)\/locations\/([a-zA-Z0-9-_]+)\/reasoningEngines\/(\d+)$/;
-import { logger } from "../utils/logger.js";
-import { BaseCodeExecutor } from "./base_code_executor.js";
-import {
-  CodeExecutionLanguage
-} from "./code_execution_utils.js";
-const DEFAULT_MAX_ATTEMPTS = 180;
-const DEFAULT_SANDBOX_TTL = "31536000s";
-const DEFAULT_SANDBOX_DISPLAY_NAME = "default_sandbox";
-const DEFAULT_ENGINE_DISPLAY_NAME = "default_engine";
-function mapLanguage(lang) {
-  switch (lang) {
-    case CodeExecutionLanguage.PYTHON:
-      return Language.LANGUAGE_PYTHON;
-    case CodeExecutionLanguage.JAVASCRIPT:
-      return Language.LANGUAGE_JAVASCRIPT;
-    default:
-      throw new Error("Unsupported language for Agent Engine Sandbox: ".concat(lang));
-  }
-}
-__name(mapLanguage, "mapLanguage");
-let AgentEngineSandboxCodeExecutor = class extends BaseCodeExecutor {
-  constructor(options = {}) {
-    super();
-    this.sandboxResourceName = options.sandboxResourceName;
-    this.agentEngineResourceName = options.agentEngineResourceName;
-    this.projectId = options.projectId || process.env.GOOGLE_CLOUD_PROJECT;
-    this.location = options.location || process.env.GOOGLE_CLOUD_LOCATION || "us-central1";
-    if (this.sandboxResourceName) {
-      const match = this.sandboxResourceName.match(SANDBOX_PATTERN);
-      if (match) {
-        this.projectId = match[1];
-        this.location = match[2];
-      } else {
-        throw new Error(
-          "Invalid sandbox resource name: ".concat(this.sandboxResourceName)
-        );
-      }
-    } else if (this.agentEngineResourceName) {
-      const match = this.agentEngineResourceName.match(ENGINE_PATTERN);
-      if (match) {
-        this.projectId = match[1];
-        this.location = match[2];
-      } else {
-        throw new Error(
-          "Invalid agent engine resource name: ".concat(this.agentEngineResourceName)
-        );
-      }
-    }
-    if (options.client) {
-      this.client = options.client;
-    } else {
-      if (!this.projectId) {
-        throw new Error("Project ID is required.");
-      }
-      this.client = new Client({
-        project: this.projectId,
-        location: this.location
-      });
-    }
-  }
-  async executeCode(params) {
-    var _a;
-    const { invocationContext, codeExecutionInput } = params;
-    const language = mapLanguage(codeExecutionInput.language);
-    const agentEngineName = await this.getOrCreateAgentEngine();
-    const sandboxName = await this.getOrCreateSandbox(
-      invocationContext,
-      agentEngineName,
-      language
-    );
-    const inputs = [
-      {
-        mimeType: "application/json",
-        data: Buffer.from(
-          JSON.stringify({ code: codeExecutionInput.code })
-        ).toString("base64")
-      }
-    ];
-    if (codeExecutionInput.inputFiles) {
-      for (const file of codeExecutionInput.inputFiles) {
-        inputs.push({
-          mimeType: file.mimeType,
-          data: file.content,
-          // Assumed to be already base64 encoded based on CodeExecutionInput definition
-          metadata: {
-            attributes: {
-              file_name: Buffer.from(file.name).toString("base64")
-            }
-          }
-        });
-      }
-    }
-    logger.debug("Executing code in sandbox ".concat(sandboxName, "..."));
-    const response = await this.client.agentEnginesInternal.sandboxes.executeCodeInternal({
-      name: sandboxName,
-      inputs
-    });
-    let stdout = "";
-    let stderr = "";
-    const outputFiles = [];
-    if (response.outputs) {
-      for (const output of response.outputs) {
-        const attributes = ((_a = output.metadata) == null ? void 0 : _a.attributes) || {};
-        const encodedFileName = attributes["file_name"];
-        const fileName = encodedFileName ? Buffer.from(encodedFileName, "base64").toString("utf-8") : void 0;
-        if (output.mimeType === "application/json" && !fileName) {
-          if (output.data) {
-            const jsonStr = Buffer.from(output.data, "base64").toString(
-              "utf-8"
-            );
-            try {
-              const jsonData = JSON.parse(jsonStr);
-              if (jsonData.msg_out !== void 0) {
-                stdout = jsonData.msg_out;
-              } else {
-                stdout = "";
-              }
-              if (jsonData.msg_err !== void 0) {
-                stderr = jsonData.msg_err;
-              } else {
-                stderr = "";
-              }
-            } catch (e) {
-              logger.warn("Failed to parse JSON output from sandbox", e);
-              stdout = jsonStr;
-            }
-          }
-        } else {
-          let mimeType = output.mimeType;
-          const name = fileName || "output_file";
-          if (!mimeType) {
-            mimeType = guessMimeType(name);
-          }
-          outputFiles.push({
-            name,
-            content: output.data || "",
-            mimeType
-          });
-        }
-      }
-    }
-    return {
-      stdout,
-      stderr,
-      outputFiles
-    };
-  }
-  async getOrCreateAgentEngine() {
-    if (this.agentEngineResourceName) {
-      return this.agentEngineResourceName;
-    }
-    if (!this.agentEngineCreationPromise) {
-      this.agentEngineCreationPromise = (async () => {
-        logger.debug(
-          "No Agent Engine resource name provided. Creating a new one..."
-        );
-        const operation = await this.client.agentEnginesInternal.createInternal(
-          {
-            config: {
-              displayName: DEFAULT_ENGINE_DISPLAY_NAME
-            }
-          }
-        );
-        let apiResponse = operation;
-        let attempts = 0;
-        while (!apiResponse.done && attempts < DEFAULT_MAX_ATTEMPTS) {
-          await new Promise((resolve) => setTimeout(resolve, 1e3));
-          apiResponse = await this.client.agentEnginesInternal.getAgentOperationInternal({
-            operationName: operation.name
-          });
-          attempts++;
-        }
-        if (!apiResponse.done) {
-          throw new Error(
-            "Agent Engine creation operation ".concat(operation.name, " did not complete in time.")
-          );
-        }
-        const response = apiResponse.response;
-        this.agentEngineResourceName = response.name;
-        logger.debug("Created Agent Engine: ".concat(this.agentEngineResourceName));
-        return this.agentEngineResourceName;
-      })();
-    }
-    return this.agentEngineCreationPromise;
-  }
-  async getOrCreateSandbox(invocationContext, agentEngineName, language) {
-    var _a, _b;
-    if (this.sandboxResourceName) {
-      return this.sandboxResourceName;
-    }
-    const stateKey = "sandbox_name_".concat(language.toLowerCase());
-    let sandboxName = (_b = (_a = invocationContext.session) == null ? void 0 : _a.state) == null ? void 0 : _b[stateKey];
-    let createNewSandbox = false;
-    if (!sandboxName) {
-      createNewSandbox = true;
-    } else {
-      try {
-        const sandbox = await this.client.agentEnginesInternal.sandboxes.getInternal({
-          name: sandboxName
-        });
-        if (!sandbox || sandbox.state !== "STATE_RUNNING") {
-          createNewSandbox = true;
-        }
-      } catch (error) {
-        logger.debug(
-          "Failed to get sandbox ".concat(sandboxName, ", will create a new one"),
-          error
-        );
-        createNewSandbox = true;
-      }
-    }
-    if (createNewSandbox) {
-      logger.debug("Creating a new ".concat(language, " code execution sandbox..."));
-      const operation = await this.client.agentEnginesInternal.sandboxes.createInternal({
-        name: agentEngineName,
-        spec: {
-          codeExecutionEnvironment: {
-            codeLanguage: language
-          }
-        },
-        config: {
-          displayName: DEFAULT_SANDBOX_DISPLAY_NAME,
-          ttl: DEFAULT_SANDBOX_TTL
-        }
-      });
-      let apiResponse = operation;
-      let attempts = 0;
-      while (!apiResponse.done && attempts < DEFAULT_MAX_ATTEMPTS) {
-        await new Promise((resolve) => setTimeout(resolve, 1e3));
-        apiResponse = await this.client.agentEnginesInternal.sandboxes.getSandboxOperationInternal(
-          {
-            operationName: operation.name
-          }
-        );
-        attempts++;
-      }
-      if (!apiResponse.done) {
-        throw new Error(
-          "Sandbox creation operation ".concat(operation.name, " did not complete in time.")
-        );
-      }
-      const response = apiResponse.response;
-      sandboxName = response.name;
-      if (invocationContext.session) {
-        if (!invocationContext.session.state) {
-          invocationContext.session.state = {};
-        }
-        invocationContext.session.state[stateKey] = sandboxName;
-      }
-    }
-    return sandboxName;
-  }
-};
-__name(AgentEngineSandboxCodeExecutor, "AgentEngineSandboxCodeExecutor");
-AgentEngineSandboxCodeExecutor = __decorateClass([
-  experimental
-], AgentEngineSandboxCodeExecutor);
-export {
-  AgentEngineSandboxCodeExecutor
-};

package/dist/web/code_executors/base_code_executor.js

@@ -1,68 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-var __defProp = Object.defineProperty;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-var _a;
-const BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL = Symbol.for(
-  "google.adk.baseCodeExecutor"
-);
-function isBaseCodeExecutor(obj) {
-  return typeof obj === "object" && obj !== null && BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL in obj && obj[BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL] === true;
-}
-__name(isBaseCodeExecutor, "isBaseCodeExecutor");
-_a = BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL;
-const _BaseCodeExecutor = class _BaseCodeExecutor {
-  constructor() {
-    /** A unique symbol to identify BaseCodeExecutor class. */
-    this[_a] = true;
-    /**
-     * If true, extract and process data files from the model request
-     * and attach them to the code executor.
-     *
-     * Supported data file MimeTypes are [text/csv].
-     * Default to false.
-     */
-    this.optimizeDataFile = false;
-    /**
-     * Whether the code executor is stateful. Default to false.
-     */
-    this.stateful = false;
-    /**
-     * The number of attempts to retry on consecutive code execution errors.
-     * Default to 2.
-     */
-    this.errorRetryAttempts = 2;
-    /**
-     * The list of the enclosing delimiters to identify the code blocks.
-     * For example, the delimiter('```javascript\\n', '\\n```') can be used to
-     * identify code blocks with the following format:
-     *
-     * ```javascript
-     *  console.log("hello")
-     * ```
-     */
-    this.codeBlockDelimiters = [
-      ["```tool_code\n", "\n```"],
-      ["```python\n", "\n```"],
-      ["```javascript\n", "\n```"],
-      ["```typescript\n", "\n```"],
-      ["```bash\n", "\n```"],
-      ["```sh\n", "\n```"]
-    ];
-    /**
-     * The delimiters to format the code execution result.
-     */
-    this.executionResultDelimiters = ["```tool_output\n", "\n```"];
-  }
-};
-__name(_BaseCodeExecutor, "BaseCodeExecutor");
-let BaseCodeExecutor = _BaseCodeExecutor;
-export {
-  BaseCodeExecutor,
-  isBaseCodeExecutor
-};

package/dist/web/code_executors/built_in_code_executor.js

@@ -1,50 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-var __defProp = Object.defineProperty;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-var _a, _b;
-import { isGemini2OrAbove } from "../utils/model_name.js";
-import { BaseCodeExecutor } from "./base_code_executor.js";
-const BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL = Symbol.for(
-  "google.adk.builtInCodeExecutor"
-);
-function isBuiltInCodeExecutor(obj) {
-  return typeof obj === "object" && obj !== null && BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL in obj && obj[BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL] === true;
-}
-__name(isBuiltInCodeExecutor, "isBuiltInCodeExecutor");
-const _BuiltInCodeExecutor = class _BuiltInCodeExecutor extends (_b = BaseCodeExecutor, _a = BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL, _b) {
-  constructor() {
-    super(...arguments);
-    /** A unique symbol to identify BuiltInCodeExecutor class. */
-    this[_a] = true;
-  }
-  executeCode(_params) {
-    return Promise.resolve({
-      stdout: "",
-      stderr: "",
-      outputFiles: []
-    });
-  }
-  processLlmRequest(llmRequest) {
-    if (llmRequest.model && isGemini2OrAbove(llmRequest.model)) {
-      llmRequest.config = llmRequest.config || {};
-      llmRequest.config.tools = llmRequest.config.tools || [];
-      llmRequest.config.tools.push({ codeExecution: {} });
-      return;
-    }
-    throw new Error(
-      "Gemini code execution tool is not supported for model ".concat(llmRequest.model)
-    );
-  }
-};
-__name(_BuiltInCodeExecutor, "BuiltInCodeExecutor");
-let BuiltInCodeExecutor = _BuiltInCodeExecutor;
-export {
-  BuiltInCodeExecutor,
-  isBuiltInCodeExecutor
-};

package/dist/web/code_executors/code_execution_utils.js

@@ -1,131 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-var __defProp = Object.defineProperty;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { Language, Outcome } from "@google/genai";
-import { cloneDeep } from "lodash-es";
-import { base64Encode, isBase64Encoded } from "../utils/env_aware_utils.js";
-var FileContentEncoding = /* @__PURE__ */ ((FileContentEncoding2) => {
-  FileContentEncoding2["UTF8"] = "utf-8";
-  FileContentEncoding2["BASE64"] = "base64";
-  return FileContentEncoding2;
-})(FileContentEncoding || {});
-var CodeExecutionLanguage = /* @__PURE__ */ ((CodeExecutionLanguage2) => {
-  CodeExecutionLanguage2["UNSPECIFIED"] = "unspecified";
-  CodeExecutionLanguage2["PYTHON"] = "python";
-  CodeExecutionLanguage2["JAVASCRIPT"] = "javascript";
-  CodeExecutionLanguage2["TYPESCRIPT"] = "typescript";
-  CodeExecutionLanguage2["SHELL"] = "shell";
-  CodeExecutionLanguage2["POWERSHELL"] = "powershell";
-  CodeExecutionLanguage2["WINDOWS_CMD"] = "cmd";
-  return CodeExecutionLanguage2;
-})(CodeExecutionLanguage || {});
-function getEncodedFileContent(data) {
-  return isBase64Encoded(data) ? data : base64Encode(data);
-}
-__name(getEncodedFileContent, "getEncodedFileContent");
-function extractCodeAndTruncateContent(content, codeBlockDelimiters) {
-  var _a;
-  if (!((_a = content.parts) == null ? void 0 : _a.length)) {
-    return "";
-  }
-  for (let i = 0; i < content.parts.length; i++) {
-    const part = content.parts[i];
-    if (part.executableCode && (i === content.parts.length - 1 || !content.parts[i + 1].codeExecutionResult)) {
-      content.parts = content.parts.slice(0, i + 1);
-      return part.executableCode.code;
-    }
-  }
-  const textParts = content.parts.filter((part) => part.text);
-  if (!textParts.length) {
-    return "";
-  }
-  const firstTextPart = cloneDeep(textParts[0]);
-  const responseText = textParts.map((part) => part.text).join("\n");
-  const leadingDelimiterPattern = codeBlockDelimiters.map((d) => d[0]).join("|");
-  const trailingDelimiterPattern = codeBlockDelimiters.map((d) => d[1]).join("|");
-  const match = new RegExp(
-    "(?<prefix>.*?)(".concat(leadingDelimiterPattern, ")(?<codeStr>.*?)(").concat(trailingDelimiterPattern, ")(?<suffix>.*?)$"),
-    "s"
-  ).exec(responseText);
-  const { prefix, codeStr } = (match == null ? void 0 : match.groups) || {};
-  if (!codeStr) {
-    return "";
-  }
-  content.parts = [];
-  if (prefix) {
-    firstTextPart.text = prefix;
-    content.parts.push(firstTextPart);
-  }
-  content.parts.push(buildExecutableCodePart(codeStr));
-  return codeStr;
-}
-__name(extractCodeAndTruncateContent, "extractCodeAndTruncateContent");
-function buildExecutableCodePart(code) {
-  return {
-    text: code,
-    executableCode: {
-      code,
-      language: Language.PYTHON
-    }
-  };
-}
-__name(buildExecutableCodePart, "buildExecutableCodePart");
-function buildCodeExecutionResultPart(codeExecutionResult) {
-  if (codeExecutionResult.stderr) {
-    return {
-      text: codeExecutionResult.stderr,
-      codeExecutionResult: {
-        outcome: Outcome.OUTCOME_FAILED
-      }
-    };
-  }
-  const finalResult = [];
-  if (codeExecutionResult.stdout || !codeExecutionResult.outputFiles) {
-    finalResult.push("Code execution result:\n".concat(codeExecutionResult.stdout, "\n"));
-  }
-  if (codeExecutionResult.outputFiles) {
-    finalResult.push(
-      "Saved artifacts:\n" + codeExecutionResult.outputFiles.map((f) => f.name).join(", ")
-    );
-  }
-  return {
-    text: finalResult.join("\n\n"),
-    codeExecutionResult: {
-      outcome: Outcome.OUTCOME_OK
-    }
-  };
-}
-__name(buildCodeExecutionResultPart, "buildCodeExecutionResultPart");
-function convertCodeExecutionParts(content, codeBlockDelimiter, executionResultDelimiters) {
-  var _a;
-  if (!((_a = content.parts) == null ? void 0 : _a.length)) {
-    return;
-  }
-  const lastPart = content.parts[content.parts.length - 1];
-  if (lastPart.executableCode) {
-    content.parts[content.parts.length - 1] = {
-      text: codeBlockDelimiter[0] + lastPart.executableCode.code + codeBlockDelimiter[1]
-    };
-  } else if (content.parts.length == 1 && lastPart.codeExecutionResult) {
-    content.parts[content.parts.length - 1] = {
-      text: executionResultDelimiters[0] + lastPart.codeExecutionResult.output + executionResultDelimiters[1]
-    };
-    content.role = "user";
-  }
-}
-__name(convertCodeExecutionParts, "convertCodeExecutionParts");
-export {
-  CodeExecutionLanguage,
-  FileContentEncoding,
-  buildCodeExecutionResultPart,
-  buildExecutableCodePart,
-  convertCodeExecutionParts,
-  extractCodeAndTruncateContent,
-  getEncodedFileContent
-};