@softactivate/adk 1.1.0

package/dist/esm/code_executors/code_execution_utils.js

@@ -0,0 +1,127 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+/**
+ * @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);
+}
+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>.*?)(${leadingDelimiterPattern})(?<codeStr>.*?)(${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;
+}
+function buildExecutableCodePart(code) {
+  return {
+    text: code,
+    executableCode: {
+      code,
+      language: Language.PYTHON
+    }
+  };
+}
+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:
+${codeExecutionResult.stdout}
+`);
+  }
+  if (codeExecutionResult.outputFiles) {
+    finalResult.push(
+      `Saved artifacts:
+` + codeExecutionResult.outputFiles.map((f) => f.name).join(", ")
+    );
+  }
+  return {
+    text: finalResult.join("\n\n"),
+    codeExecutionResult: {
+      outcome: Outcome.OUTCOME_OK
+    }
+  };
+}
+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";
+  }
+}
+export {
+  CodeExecutionLanguage,
+  FileContentEncoding,
+  buildCodeExecutionResultPart,
+  buildExecutableCodePart,
+  convertCodeExecutionParts,
+  extractCodeAndTruncateContent,
+  getEncodedFileContent
+};

package/dist/esm/code_executors/code_executor_context.js

@@ -0,0 +1,170 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+/**
+ * @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";
+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 (!(INPUT_FILE_KEY in this.sessionState)) {
+      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 (!(INPUT_FILE_KEY in this.sessionState)) {
+      this.sessionState.set(INPUT_FILE_KEY, []);
+    }
+    this.sessionState.get(INPUT_FILE_KEY).push(...inputFiles);
+  }
+  clearInputFiles() {
+    if (INPUT_FILE_KEY in this.sessionState) {
+      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 (!(ERROR_COUNT_KEY in this.sessionState)) {
+      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 (!(ERROR_COUNT_KEY in this.sessionState)) {
+      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 (!(ERROR_COUNT_KEY in this.sessionState)) {
+      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 (!(CODE_EXECUTION_RESULTS_KEY in this.sessionState)) {
+      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) || {};
+  }
+}
+export {
+  CodeExecutorContext
+};

package/dist/esm/code_executors/unsafe_local_code_executor.js

@@ -0,0 +1,205 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { spawn } from "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";
+async function createTempScriptFile(code, language, shellCommandPath) {
+  const tempDir = path.join(
+    os.tmpdir(),
+    "adk_js_unsafe_code_executor",
+    Date.now().toString() + "_" + Math.random().toString(36).slice(2)
+  );
+  await fs.mkdir(tempDir, { recursive: true });
+  const ext = getExtensionForLanguage(language, shellCommandPath) || ".js";
+  const filePath = path.join(tempDir, `script${ext}`);
+  await fs.writeFile(filePath, code);
+  return { filePath, tempDir };
+}
+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 (IS_WINDOWS) {
+      if (shellCommandPath && shellCommandPath.toLowerCase().includes("cmd")) {
+        return ".bat";
+      }
+      return ".ps1";
+    }
+    return ".sh";
+  }
+  return void 0;
+}
+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: ${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 (this.shellCommandPath.toLowerCase().includes("powershell")) {
+          args = ["-NoLogo", "-ExecutionPolicy", "Bypass", "-File", filePath];
+        } else if (this.shellCommandPath.toLowerCase().includes("cmd")) {
+          args = ["/c", filePath];
+        }
+      } else if (language === CodeExecutionLanguage.POWERSHELL) {
+        command = IS_WINDOWS ? "powershell" : "pwsh";
+        args = ["-NoLogo", "-ExecutionPolicy", "Bypass", "-File", filePath];
+      } else if (language === CodeExecutionLanguage.WINDOWS_CMD) {
+        command = "cmd.exe";
+        args = ["/c", 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(`--${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: ${err.message}
+`;
+        });
+        child.on("close", (exitCode, signal) => {
+          if (signal === "SIGKILL" || signal === "SIGTERM") {
+            stderr += `
+Code execution timed out after ${this.timeoutSeconds} seconds.`;
+          } else if (exitCode !== 0 && exitCode !== null) {
+            if (!stderr) {
+              stderr = `Exit code ${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: ${e}`);
+      }
+      return {
+        stdout: executionResult.stdout,
+        stderr: executionResult.stderr,
+        outputFiles
+      };
+    } finally {
+      if (tempDir) {
+        await fs.rm(tempDir, { recursive: true, force: true });
+      }
+    }
+  }
+}
+export {
+  UnsafeLocalCodeExecutor
+};