@google/adk 2.0.0 → 2.1.0

package/dist/web/code_executors/code_executor_context.js

@@ -1,174 +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 { cloneDeep } from "lodash-es";
-const CONTEXT_KEY = "_code_execution_context";
-const SESSION_ID_KEY = "execution_session_id";
-const PROCESSED_FILE_NAMES_KEY = "processed_input_files";
-const INPUT_FILE_KEY = "_code_executor_input_files";
-const ERROR_COUNT_KEY = "_code_executor_error_counts";
-const CODE_EXECUTION_RESULTS_KEY = "_code_execution_results";
-const _CodeExecutorContext = class _CodeExecutorContext {
-  constructor(sessionState) {
-    this.sessionState = sessionState;
-    var _a;
-    this.context = (_a = sessionState.get(CONTEXT_KEY)) != null ? _a : {};
-    this.sessionState = sessionState;
-  }
-  /**
-   * Gets the state delta to update in the persistent session state.
-   * @return The state delta to update in the persistent session state.
-   */
-  getStateDelta() {
-    return {
-      [CONTEXT_KEY]: cloneDeep(this.context)
-    };
-  }
-  /**
-   * Gets the execution ID for the code executor.
-   * @return The execution ID for the code executor.
-   */
-  getExecutionId() {
-    if (!(SESSION_ID_KEY in this.context)) {
-      return void 0;
-    }
-    return this.context[SESSION_ID_KEY];
-  }
-  /**
-   * Sets the execution ID for the code executor.
-   * @param executionId The execution ID to set.
-   */
-  setExecutionId(executionId) {
-    this.context[SESSION_ID_KEY] = executionId;
-  }
-  /**
-   * Gets the processed file names from the session state.
-   * @return A list of processed file names in the code executor context.
-   */
-  getProcessedFileNames() {
-    if (!(PROCESSED_FILE_NAMES_KEY in this.context)) {
-      return [];
-    }
-    return this.context[PROCESSED_FILE_NAMES_KEY];
-  }
-  /**
-   * Adds the processed file names to the session state.
-   * @param fileNames The file names to add to the session state.
-   */
-  addProcessedFileNames(fileNames) {
-    if (!(PROCESSED_FILE_NAMES_KEY in this.context)) {
-      this.context[PROCESSED_FILE_NAMES_KEY] = [];
-    }
-    this.context[PROCESSED_FILE_NAMES_KEY].push(...fileNames);
-  }
-  /**
-   * Gets the input files from the session state.
-   * @return A list of input files in the code executor context.
-   */
-  getInputFiles() {
-    if (!this.sessionState.has(INPUT_FILE_KEY)) {
-      return [];
-    }
-    return this.sessionState.get(INPUT_FILE_KEY);
-  }
-  /**
-   * Adds the input files to the session state.
-   * @param inputFiles The input files to add to the session state.
-   */
-  addInputFiles(inputFiles) {
-    if (!this.sessionState.has(INPUT_FILE_KEY)) {
-      this.sessionState.set(INPUT_FILE_KEY, []);
-    }
-    this.sessionState.get(INPUT_FILE_KEY).push(...inputFiles);
-  }
-  clearInputFiles() {
-    if (this.sessionState.has(INPUT_FILE_KEY)) {
-      this.sessionState.set(INPUT_FILE_KEY, []);
-    }
-    if (PROCESSED_FILE_NAMES_KEY in this.context) {
-      this.context[PROCESSED_FILE_NAMES_KEY] = [];
-    }
-  }
-  /**
-   * Gets the error count from the session state.
-   * @param invocationId The invocation ID to get the error count for.
-   * @return The error count for the given invocation ID.
-   */
-  getErrorCount(invocationId) {
-    if (!this.sessionState.has(ERROR_COUNT_KEY)) {
-      return 0;
-    }
-    return this.sessionState.get(ERROR_COUNT_KEY)[invocationId] || 0;
-  }
-  /**
-   * Increments the error count from the session state.
-   * @param invocationId The invocation ID to increment the error count for.
-   */
-  incrementErrorCount(invocationId) {
-    if (!this.sessionState.has(ERROR_COUNT_KEY)) {
-      this.sessionState.set(ERROR_COUNT_KEY, {});
-    }
-    this.sessionState.get(ERROR_COUNT_KEY)[invocationId] = this.getErrorCount(invocationId) + 1;
-  }
-  /**
-   * Resets the error count from the session state.
-   * @param invocationId The invocation ID to reset the error count for.
-   */
-  resetErrorCount(invocationId) {
-    if (!this.sessionState.has(ERROR_COUNT_KEY)) {
-      return;
-    }
-    const errorCounts = this.sessionState.get(ERROR_COUNT_KEY);
-    if (invocationId in errorCounts) {
-      delete errorCounts[invocationId];
-    }
-  }
-  /**
-   * Updates the code execution result.
-   * @param invocationId The invocation ID to update the code execution result
-   *     for.
-   * @param code The code to execute.
-   * @param resultStdout The standard output of the code execution.
-   * @param resultStderr The standard error of the code execution.
-   */
-  updateCodeExecutionResult({
-    invocationId,
-    code,
-    resultStdout,
-    resultStderr
-  }) {
-    if (!this.sessionState.has(CODE_EXECUTION_RESULTS_KEY)) {
-      this.sessionState.set(CODE_EXECUTION_RESULTS_KEY, {});
-    }
-    const codeExecutionResults = this.sessionState.get(
-      CODE_EXECUTION_RESULTS_KEY
-    );
-    if (!(invocationId in codeExecutionResults)) {
-      codeExecutionResults[invocationId] = [];
-    }
-    codeExecutionResults[invocationId].push({
-      code,
-      resultStdout,
-      resultStderr,
-      timestamp: Date.now()
-    });
-  }
-  /**
-   * Gets the code execution context from the session state.
-   * @return The code execution context for the given invocation ID.
-   */
-  getCodeExecutionContext() {
-    return this.sessionState.get(CONTEXT_KEY) || {};
-  }
-};
-__name(_CodeExecutorContext, "CodeExecutorContext");
-let CodeExecutorContext = _CodeExecutorContext;
-export {
-  CodeExecutorContext
-};

package/dist/web/code_executors/unsafe_local_code_executor.js

@@ -1,221 +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 2026 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { spawn } from "node:child_process";
-import * as fs from "node:fs/promises";
-import * as os from "node:os";
-import * as path from "node:path";
-import { getMimeTypeAndEncoding } from "../utils/file_extension_utils.js";
-import { materializeFiles } from "../utils/file_utils.js";
-import { logger } from "../utils/logger.js";
-import { BaseCodeExecutor } from "./base_code_executor.js";
-import {
-  CodeExecutionLanguage
-} from "./code_execution_utils.js";
-const IS_WINDOWS = os.platform() === "win32";
-const POWERSHELL_BASE_ARGS = [
-  "-NoLogo",
-  "-NoProfile",
-  "-ExecutionPolicy",
-  "Bypass",
-  "-File"
-];
-const CMD_BASE_ARGS = ["/D", "/c"];
-function isPowerShellCommand(commandPath) {
-  return /^(powershell|pwsh)(\.exe)?$/i.test(path.win32.basename(commandPath));
-}
-__name(isPowerShellCommand, "isPowerShellCommand");
-async function createTempScriptFile(code, language, shellCommandPath) {
-  const tempDir = await fs.mkdtemp(
-    path.join(os.tmpdir(), "adk_js_unsafe_code_executor_")
-  );
-  const ext = getExtensionForLanguage(language, shellCommandPath) || ".js";
-  const filePath = path.join(tempDir, "script".concat(ext));
-  await fs.writeFile(filePath, code);
-  return { filePath, tempDir };
-}
-__name(createTempScriptFile, "createTempScriptFile");
-function getExtensionForLanguage(language, shellCommandPath) {
-  if (language === CodeExecutionLanguage.JAVASCRIPT) {
-    return ".js";
-  }
-  if (language === CodeExecutionLanguage.PYTHON) {
-    return ".py";
-  }
-  if (language === CodeExecutionLanguage.POWERSHELL) {
-    return ".ps1";
-  }
-  if (language === CodeExecutionLanguage.WINDOWS_CMD) {
-    return ".bat";
-  }
-  if (language === CodeExecutionLanguage.SHELL) {
-    if (shellCommandPath && isPowerShellCommand(shellCommandPath)) {
-      return ".ps1";
-    }
-    if (IS_WINDOWS) {
-      if (shellCommandPath && shellCommandPath.toLowerCase().includes("cmd")) {
-        return ".bat";
-      }
-      return ".ps1";
-    }
-    return ".sh";
-  }
-  return void 0;
-}
-__name(getExtensionForLanguage, "getExtensionForLanguage");
-const _UnsafeLocalCodeExecutor = class _UnsafeLocalCodeExecutor extends BaseCodeExecutor {
-  constructor(options = {}) {
-    var _a, _b, _c, _d;
-    super();
-    this.timeoutSeconds = (_a = options.timeoutSeconds) != null ? _a : 30;
-    this.nodeCommandPath = (_b = options.commandPath) != null ? _b : process.execPath;
-    this.pythonCommandPath = (_c = options.pythonCommandPath) != null ? _c : IS_WINDOWS ? "python" : "python3";
-    this.shellCommandPath = (_d = options.shellCommandPath) != null ? _d : IS_WINDOWS ? "powershell" : "bash";
-    this.stateful = false;
-    this.optimizeDataFile = false;
-  }
-  async executeCode(params) {
-    var _a;
-    const { code, language } = params.codeExecutionInput;
-    if (![
-      CodeExecutionLanguage.JAVASCRIPT,
-      CodeExecutionLanguage.PYTHON,
-      CodeExecutionLanguage.SHELL,
-      CodeExecutionLanguage.WINDOWS_CMD,
-      CodeExecutionLanguage.POWERSHELL
-    ].includes(language)) {
-      return {
-        stdout: "",
-        stderr: "Unsupported language: ".concat(language),
-        outputFiles: []
-      };
-    }
-    logger.warn(
-      "\n====================================================================================\n\u26A0\uFE0F DANGER: UnsafeLocalCodeExecutor is executing code locally on this host machine!\nThis component provides NO sandboxing or container isolation. Arbitrary shell/script\ncommands generated by AI or untrusted sources could lead to serious security risks.\n====================================================================================\n"
-    );
-    let tempDir;
-    try {
-      const res = await createTempScriptFile(
-        code,
-        language,
-        this.shellCommandPath
-      );
-      const filePath = res.filePath;
-      tempDir = res.tempDir;
-      if (params.codeExecutionInput.inputFiles) {
-        await materializeFiles(params.codeExecutionInput.inputFiles, tempDir);
-      }
-      let command = this.nodeCommandPath;
-      let args = [filePath];
-      if (language === CodeExecutionLanguage.PYTHON) {
-        command = this.pythonCommandPath;
-      } else if (language === CodeExecutionLanguage.SHELL) {
-        command = this.shellCommandPath;
-        if (isPowerShellCommand(this.shellCommandPath)) {
-          args = [...POWERSHELL_BASE_ARGS, filePath];
-        } else if (this.shellCommandPath.toLowerCase().includes("cmd")) {
-          args = [...CMD_BASE_ARGS, filePath];
-        }
-      } else if (language === CodeExecutionLanguage.POWERSHELL) {
-        command = IS_WINDOWS ? "powershell" : "pwsh";
-        args = [...POWERSHELL_BASE_ARGS, filePath];
-      } else if (language === CodeExecutionLanguage.WINDOWS_CMD) {
-        command = "cmd.exe";
-        args = [...CMD_BASE_ARGS, filePath];
-      }
-      if (params.codeExecutionInput.args) {
-        if (Array.isArray(params.codeExecutionInput.args)) {
-          args.push(...params.codeExecutionInput.args);
-        } else {
-          for (const [k, v] of Object.entries(params.codeExecutionInput.args)) {
-            args.push("--".concat(k), String(v));
-          }
-        }
-      }
-      const executionResult = await new Promise((resolve) => {
-        const child = spawn(command, args, {
-          timeout: this.timeoutSeconds * 1e3,
-          killSignal: "SIGKILL",
-          cwd: tempDir
-        });
-        let stdout = "";
-        let stderr = "";
-        if (child.stdout) {
-          child.stdout.on("data", (data) => {
-            stdout += data.toString();
-          });
-        }
-        if (child.stderr) {
-          child.stderr.on("data", (data) => {
-            stderr += data.toString();
-          });
-        }
-        child.on("error", (err) => {
-          stderr += "Process error: ".concat(err.message, "\n");
-        });
-        child.on("close", (exitCode, signal) => {
-          if (signal === "SIGKILL" || signal === "SIGTERM") {
-            stderr += "\nCode execution timed out after ".concat(this.timeoutSeconds, " seconds.");
-          } else if (exitCode !== 0 && exitCode !== null) {
-            if (!stderr) {
-              stderr = "Exit code ".concat(exitCode);
-            }
-          }
-          resolve({ stdout, stderr, exitCode });
-        });
-      });
-      const outputFiles = [];
-      try {
-        const allFiles = await fs.readdir(tempDir, { recursive: true });
-        for (const relativeFilePath of allFiles) {
-          const fullPath = path.join(tempDir, relativeFilePath);
-          const stat = await fs.lstat(fullPath);
-          if (!stat.isFile()) {
-            continue;
-          }
-          if (relativeFilePath === path.basename(filePath)) {
-            continue;
-          }
-          const isInputFile = (_a = params.codeExecutionInput.inputFiles) == null ? void 0 : _a.some(
-            (f) => f.name === relativeFilePath
-          );
-          if (isInputFile) {
-            continue;
-          }
-          const fileContent = await fs.readFile(fullPath);
-          const { mimeType, encoding } = getMimeTypeAndEncoding(
-            path.extname(relativeFilePath)
-          );
-          outputFiles.push({
-            name: relativeFilePath,
-            content: fileContent.toString(encoding),
-            contentEncoding: encoding,
-            mimeType
-          });
-        }
-      } catch (e) {
-        logger.error("Error scanning output files: ".concat(e));
-      }
-      return {
-        stdout: executionResult.stdout,
-        stderr: executionResult.stderr,
-        outputFiles
-      };
-    } finally {
-      if (tempDir) {
-        await fs.rm(tempDir, { recursive: true, force: true });
-      }
-    }
-  }
-};
-__name(_UnsafeLocalCodeExecutor, "UnsafeLocalCodeExecutor");
-let UnsafeLocalCodeExecutor = _UnsafeLocalCodeExecutor;
-export {
-  UnsafeLocalCodeExecutor
-};