@google/adk 0.2.1 → 0.2.2

package/dist/web/agents/llm_agent.js

@@ -36,15 +36,23 @@ var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")])
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { cloneDeep } from "lodash";
 import { z } from "zod";
+import { BaseCodeExecutor } from "../code_executors/base_code_executor.js";
+import { BuiltInCodeExecutor } from "../code_executors/built_in_code_executor.js";
+import { buildCodeExecutionResultPart, buildExecutableCodePart, convertCodeExecutionParts, extractCodeAndTruncateContent } from "../code_executors/code_execution_utils.js";
+import { CodeExecutorContext } from "../code_executors/code_executor_context.js";
 import { createEvent, createNewEventId, getFunctionCalls, getFunctionResponses, isFinalResponse } from "../events/event.js";
-import { BaseLlm } from "../models/base_llm.js";
+import { createEventActions } from "../events/event_actions.js";
+import { isBaseLlm } from "../models/base_llm.js";
 import { appendInstructions, setOutputSchema } from "../models/llm_request.js";
 import { LLMRegistry } from "../models/registry.js";
+import { State } from "../sessions/state.js";
 import { BaseTool } from "../tools/base_tool.js";
 import { FunctionTool } from "../tools/function_tool.js";
 import { ToolConfirmation } from "../tools/tool_confirmation.js";
 import { ToolContext } from "../tools/tool_context.js";
+import { base64Decode } from "../utils/env_aware_utils.js";
 import { logger } from "../utils/logger.js";
 import { BaseAgent } from "./base_agent.js";
 import { BaseLlmRequestProcessor } from "./base_llm_processor.js";
@@ -349,6 +357,306 @@ class RequestConfirmationLlmRequestProcessor extends BaseLlmRequestProcessor {
   }
 }
 const REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR = new RequestConfirmationLlmRequestProcessor();
+class CodeExecutionRequestProcessor extends BaseLlmRequestProcessor {
+  runAsync(invocationContext, llmRequest) {
+    return __asyncGenerator(this, null, function* () {
+      if (!(invocationContext.agent instanceof LlmAgent)) {
+        return;
+      }
+      if (!invocationContext.agent.codeExecutor) {
+        return;
+      }
+      try {
+        for (var iter = __forAwait(runPreProcessor(invocationContext, llmRequest)), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
+          const event = temp.value;
+          yield event;
+        }
+      } catch (temp) {
+        error = [temp];
+      } finally {
+        try {
+          more && (temp = iter.return) && (yield new __await(temp.call(iter)));
+        } finally {
+          if (error)
+            throw error[0];
+        }
+      }
+      if (!(invocationContext.agent.codeExecutor instanceof BaseCodeExecutor)) {
+        return;
+      }
+      for (const content of llmRequest.contents) {
+        const delimeters = invocationContext.agent.codeExecutor.codeBlockDelimiters.length ? invocationContext.agent.codeExecutor.codeBlockDelimiters[0] : ["", ""];
+        const codeExecutionParts = convertCodeExecutionParts(
+          content,
+          delimeters,
+          invocationContext.agent.codeExecutor.executionResultDelimiters
+        );
+      }
+    });
+  }
+}
+const DATA_FILE_UTIL_MAP = {
+  "text/csv": {
+    extension: ".csv",
+    loaderCodeTemplate: "pd.read_csv('{filename}')"
+  }
+};
+const DATA_FILE_HELPER_LIB = "\nimport pandas as pd\n\ndef explore_df(df: pd.DataFrame) -> None:\n  \"\"\"Prints some information about a pandas DataFrame.\"\"\"\n\n  with pd.option_context(\n      'display.max_columns', None, 'display.expand_frame_repr', False\n  ):\n    # Print the column names to never encounter KeyError when selecting one.\n    df_dtypes = df.dtypes\n\n    # Obtain information about data types and missing values.\n    df_nulls = (len(df) - df.isnull().sum()).apply(\n        lambda x: f'{x} / {df.shape[0]} non-null'\n    )\n\n    # Explore unique total values in columns using `.unique()`.\n    df_unique_count = df.apply(lambda x: len(x.unique()))\n\n    # Explore unique values in columns using `.unique()`.\n    df_unique = df.apply(lambda x: crop(str(list(x.unique()))))\n\n    df_info = pd.concat(\n        (\n            df_dtypes.rename('Dtype'),\n            df_nulls.rename('Non-Null Count'),\n            df_unique_count.rename('Unique Values Count'),\n            df_unique.rename('Unique Values'),\n        ),\n        axis=1,\n    )\n    df_info.index.name = 'Columns'\n    print(f\"\"\"Total rows: {df.shape[0]}\nTotal columns: {df.shape[1]}\n\n{df_info}\"\"\")\n";
+class CodeExecutionResponseProcessor {
+  /**
+   * Processes the LLM response asynchronously.
+   *
+   * @param invocationContext The invocation context
+   * @param llmResponse The LLM response to process
+   * @returns An async generator yielding events
+   */
+  runAsync(invocationContext, llmResponse) {
+    return __asyncGenerator(this, null, function* () {
+      if (llmResponse.partial) {
+        return;
+      }
+      try {
+        for (var iter = __forAwait(runPostProcessor(invocationContext, llmResponse)), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
+          const event = temp.value;
+          yield event;
+        }
+      } catch (temp) {
+        error = [temp];
+      } finally {
+        try {
+          more && (temp = iter.return) && (yield new __await(temp.call(iter)));
+        } finally {
+          if (error)
+            throw error[0];
+        }
+      }
+    });
+  }
+}
+const responseProcessor = new CodeExecutionResponseProcessor();
+function runPreProcessor(invocationContext, llmRequest) {
+  return __asyncGenerator(this, null, function* () {
+    const agent = invocationContext.agent;
+    if (!(agent instanceof LlmAgent)) {
+      return;
+    }
+    const codeExecutor = agent.codeExecutor;
+    if (!codeExecutor || !(codeExecutor instanceof BaseCodeExecutor)) {
+      return;
+    }
+    if (codeExecutor instanceof BuiltInCodeExecutor) {
+      codeExecutor.processLlmRequest(llmRequest);
+      return;
+    }
+    if (!codeExecutor.optimizeDataFile) {
+      return;
+    }
+    const codeExecutorContext = new CodeExecutorContext(new State(invocationContext.session.state));
+    if (codeExecutorContext.getErrorCount(invocationContext.invocationId) >= codeExecutor.errorRetryAttempts) {
+      return;
+    }
+    const allInputFiles = extractAndReplaceInlineFiles(codeExecutorContext, llmRequest);
+    const processedFileNames = new Set(codeExecutorContext.getProcessedFileNames());
+    const filesToProcess = allInputFiles.filter((f) => !processedFileNames.has(f.name));
+    for (const file of filesToProcess) {
+      const codeStr = getDataFilePreprocessingCode(file);
+      if (!codeStr) {
+        return;
+      }
+      const codeContent = {
+        role: "model",
+        parts: [
+          { text: "Processing input file: `".concat(file.name, "`") },
+          buildExecutableCodePart(codeStr)
+        ]
+      };
+      llmRequest.contents.push(cloneDeep(codeContent));
+      yield createEvent({
+        invocationId: invocationContext.invocationId,
+        author: agent.name,
+        branch: invocationContext.branch,
+        content: codeContent
+      });
+      const executionId = getOrSetExecutionId(invocationContext, codeExecutorContext);
+      const codeExecutionResult = yield new __await(codeExecutor.executeCode({
+        invocationContext,
+        codeExecutionInput: {
+          code: codeStr,
+          inputFiles: [file],
+          executionId
+        }
+      }));
+      codeExecutorContext.updateCodeExecutionResult({
+        invocationId: invocationContext.invocationId,
+        code: codeStr,
+        resultStdout: codeExecutionResult.stdout,
+        resultStderr: codeExecutionResult.stderr
+      });
+      codeExecutorContext.addProcessedFileNames([file.name]);
+      const executionResultEvent = yield new __await(postProcessCodeExecutionResult(
+        invocationContext,
+        codeExecutorContext,
+        codeExecutionResult
+      ));
+      yield executionResultEvent;
+      llmRequest.contents.push(cloneDeep(executionResultEvent.content));
+    }
+  });
+}
+function runPostProcessor(invocationContext, llmResponse) {
+  return __asyncGenerator(this, null, function* () {
+    const agent = invocationContext.agent;
+    if (!(agent instanceof LlmAgent)) {
+      return;
+    }
+    const codeExecutor = agent.codeExecutor;
+    if (!codeExecutor || !(codeExecutor instanceof BaseCodeExecutor)) {
+      return;
+    }
+    if (!llmResponse || !llmResponse.content) {
+      return;
+    }
+    if (codeExecutor instanceof BuiltInCodeExecutor) {
+      return;
+    }
+    const codeExecutorContext = new CodeExecutorContext(new State(invocationContext.session.state));
+    if (codeExecutorContext.getErrorCount(invocationContext.invocationId) >= codeExecutor.errorRetryAttempts) {
+      return;
+    }
+    const responseContent = llmResponse.content;
+    const codeStr = extractCodeAndTruncateContent(
+      responseContent,
+      codeExecutor.codeBlockDelimiters
+    );
+    if (!codeStr) {
+      return;
+    }
+    yield createEvent({
+      invocationId: invocationContext.invocationId,
+      author: agent.name,
+      branch: invocationContext.branch,
+      content: responseContent
+    });
+    const executionId = getOrSetExecutionId(invocationContext, codeExecutorContext);
+    const codeExecutionResult = yield new __await(codeExecutor.executeCode({
+      invocationContext,
+      codeExecutionInput: {
+        code: codeStr,
+        inputFiles: codeExecutorContext.getInputFiles(),
+        executionId
+      }
+    }));
+    codeExecutorContext.updateCodeExecutionResult({
+      invocationId: invocationContext.invocationId,
+      code: codeStr,
+      resultStdout: codeExecutionResult.stdout,
+      resultStderr: codeExecutionResult.stderr
+    });
+    yield yield new __await(postProcessCodeExecutionResult(
+      invocationContext,
+      codeExecutorContext,
+      codeExecutionResult
+    ));
+    llmResponse.content = null;
+  });
+}
+function extractAndReplaceInlineFiles(codeExecutorContext, llmRequest) {
+  var _a;
+  const allInputFiles = codeExecutorContext.getInputFiles();
+  const savedFileNames = new Set(allInputFiles.map((f) => f.name));
+  for (let i = 0; i < llmRequest.contents.length; i++) {
+    const content = llmRequest.contents[i];
+    if (content.role !== "user" || !content.parts) {
+      continue;
+    }
+    for (let j = 0; j < content.parts.length; j++) {
+      const part = content.parts[j];
+      const mimeType = (_a = part.inlineData) == null ? void 0 : _a.mimeType;
+      if (!mimeType || !part.inlineData || !DATA_FILE_UTIL_MAP[mimeType]) {
+        continue;
+      }
+      const fileName = "data_".concat(i + 1, "_").concat(j + 1).concat(DATA_FILE_UTIL_MAP[mimeType].extension);
+      part.text = "\nAvailable file: `".concat(fileName, "`\n");
+      const file = {
+        name: fileName,
+        content: base64Decode(part.inlineData.data),
+        mimeType
+      };
+      if (!savedFileNames.has(fileName)) {
+        codeExecutorContext.addInputFiles([file]);
+        allInputFiles.push(file);
+      }
+    }
+  }
+  return allInputFiles;
+}
+function getOrSetExecutionId(invocationContext, codeExecutorContext) {
+  var _a;
+  const agent = invocationContext.agent;
+  if (!(agent instanceof LlmAgent) || !((_a = agent.codeExecutor) == null ? void 0 : _a.stateful)) {
+    return void 0;
+  }
+  let executionId = codeExecutorContext.getExecutionId();
+  if (!executionId) {
+    executionId = invocationContext.session.id;
+    codeExecutorContext.setExecutionId(executionId);
+  }
+  return executionId;
+}
+async function postProcessCodeExecutionResult(invocationContext, codeExecutorContext, codeExecutionResult) {
+  if (!invocationContext.artifactService) {
+    throw new Error("Artifact service is not initialized.");
+  }
+  const resultContent = {
+    role: "model",
+    parts: [buildCodeExecutionResultPart(codeExecutionResult)]
+  };
+  const eventActions = createEventActions({ stateDelta: codeExecutorContext.getStateDelta() });
+  if (codeExecutionResult.stderr) {
+    codeExecutorContext.incrementErrorCount(invocationContext.invocationId);
+  } else {
+    codeExecutorContext.resetErrorCount(invocationContext.invocationId);
+  }
+  for (const outputFile of codeExecutionResult.outputFiles) {
+    const version = await invocationContext.artifactService.saveArtifact({
+      appName: invocationContext.appName || "",
+      userId: invocationContext.userId || "",
+      sessionId: invocationContext.session.id,
+      filename: outputFile.name,
+      artifact: {
+        inlineData: { data: outputFile.content, mimeType: outputFile.mimeType }
+      }
+    });
+    eventActions.artifactDelta[outputFile.name] = version;
+  }
+  return createEvent({
+    invocationId: invocationContext.invocationId,
+    author: invocationContext.agent.name,
+    branch: invocationContext.branch,
+    content: resultContent,
+    actions: eventActions
+  });
+}
+function getDataFilePreprocessingCode(file) {
+  function getNormalizedFileName(fileName) {
+    const [varName2] = fileName.split(".");
+    let normalizedName = varName2.replace(/[^a-zA-Z0-9_]/g, "_");
+    if (/^\d/.test(normalizedName)) {
+      normalizedName = "_" + normalizedName;
+    }
+    return normalizedName;
+  }
+  if (!DATA_FILE_UTIL_MAP[file.mimeType]) {
+    return void 0;
+  }
+  const varName = getNormalizedFileName(file.name);
+  const loaderCode = DATA_FILE_UTIL_MAP[file.mimeType].loaderCodeTemplate.replace(
+    "{filename}",
+    file.name
+  );
+  return "\n".concat(DATA_FILE_HELPER_LIB, "\n\n# Load the dataframe.\n").concat(varName, " = ").concat(loaderCode, "\n\n# Use `explore_df` to guide my analysis.\nexplore_df(").concat(varName, ")\n");
+}
+const CODE_EXECUTION_REQUEST_PROCESSOR = new CodeExecutionRequestProcessor();
 class LlmAgent extends BaseAgent {
   constructor(config) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i;
@@ -368,12 +676,14 @@ class LlmAgent extends BaseAgent {
     this.afterModelCallback = config.afterModelCallback;
     this.beforeToolCallback = config.beforeToolCallback;
     this.afterToolCallback = config.afterToolCallback;
+    this.codeExecutor = config.codeExecutor;
     this.requestProcessors = (_g = config.requestProcessors) != null ? _g : [
       BASIC_LLM_REQUEST_PROCESSOR,
       IDENTITY_LLM_REQUEST_PROCESSOR,
       INSTRUCTIONS_LLM_REQUEST_PROCESSOR,
       REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR,
-      CONTENT_REQUEST_PROCESSOR
+      CONTENT_REQUEST_PROCESSOR,
+      CODE_EXECUTION_REQUEST_PROCESSOR
     ];
     this.responseProcessors = (_h = config.responseProcessors) != null ? _h : [];
     const agentTransferDisabled = this.disallowTransferToParent && this.disallowTransferToPeers && !((_i = this.subAgents) == null ? void 0 : _i.length);
@@ -423,7 +733,7 @@ class LlmAgent extends BaseAgent {
    * When not set, the agent will inherit the model from its ancestor.
    */
   get canonicalModel() {
-    if (this.model instanceof BaseLlm) {
+    if (isBaseLlm(this.model)) {
       return this.model;
     }
     if (typeof this.model === "string" && this.model) {
@@ -975,5 +1285,6 @@ class LlmAgent extends BaseAgent {
 }
 export {
   LlmAgent,
-  REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR
+  REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR,
+  responseProcessor
 };

package/dist/web/auth/exchanger/base_credential_exchanger.js

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class CredentialExchangeError extends Error {
+}
+export {
+  CredentialExchangeError
+};

package/dist/web/auth/exchanger/credential_exchanger_registry.js

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class CredentialExchangerRegistry {
+  constructor() {
+    this.exchangers = {};
+  }
+  /**
+   * Register an exchanger instance for a credential type.
+   * @param credentialType - The credential type to register for.
+   * @param exchangerInstance - The exchanger instance to register.
+   */
+  register(credentialType, exchangerInstance) {
+    this.exchangers[credentialType] = exchangerInstance;
+  }
+  /**
+   * Get the exchanger instance for a credential type.
+   * @param credentialType - The credential type to get exchanger for.
+   * @returns The exchanger instance if registered, undefined otherwise.
+   */
+  getExchanger(credentialType) {
+    return this.exchangers[credentialType];
+  }
+}
+export {
+  CredentialExchangerRegistry
+};

package/dist/web/code_executors/code_execution_utils.js

@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { Language, Outcome } from "@google/genai";
-import { deepClone } from "../utils/deep_clone.js";
+import { cloneDeep } from "lodash";
 import { base64Encode, isBase64Encoded } from "../utils/env_aware_utils.js";
 function getEncodedFileContent(data) {
   return isBase64Encoded(data) ? data : base64Encode(data);
@@ -25,7 +25,7 @@ function extractCodeAndTruncateContent(content, codeBlockDelimiters) {
   if (!textParts.length) {
     return "";
   }
-  const firstTextPart = deepClone(textParts[0]);
+  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("|");

package/dist/web/code_executors/code_executor_context.js

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { deepClone } from "../utils/deep_clone.js";
+import { cloneDeep } from "lodash";
 const CONTEXT_KEY = "_code_execution_context";
 const SESSION_ID_KEY = "execution_session_id";
 const PROCESSED_FILE_NAMES_KEY = "processed_input_files";
@@ -23,7 +23,7 @@ class CodeExecutorContext {
    */
   getStateDelta() {
     return {
-      [CONTEXT_KEY]: deepClone(this.context)
+      [CONTEXT_KEY]: cloneDeep(this.context)
     };
   }
   /**

package/dist/web/common.js

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { BaseAgent } from "./agents/base_agent.js";
+import { BaseAgent, isBaseAgent } from "./agents/base_agent.js";
 import { CallbackContext } from "./agents/callback_context.js";
 import { functionsExportedForTestingOnly } from "./agents/functions.js";
 import { InvocationContext } from "./agents/invocation_context.js";
@@ -14,10 +14,11 @@ import { ParallelAgent } from "./agents/parallel_agent.js";
 import { StreamingMode } from "./agents/run_config.js";
 import { SequentialAgent } from "./agents/sequential_agent.js";
 import { InMemoryArtifactService } from "./artifacts/in_memory_artifact_service.js";
+import { BuiltInCodeExecutor } from "./code_executors/built_in_code_executor.js";
 import { createEvent, getFunctionCalls, getFunctionResponses, hasTrailingCodeExecutionResult, isFinalResponse, stringifyContent } from "./events/event.js";
 import { createEventActions } from "./events/event_actions.js";
 import { InMemoryMemoryService } from "./memory/in_memory_memory_service.js";
-import { BaseLlm } from "./models/base_llm.js";
+import { BaseLlm, isBaseLlm } from "./models/base_llm.js";
 import { Gemini } from "./models/google_llm.js";
 import { LLMRegistry } from "./models/registry.js";
 import { BasePlugin } from "./plugins/base_plugin.js";
@@ -51,6 +52,7 @@ export {
   BasePlugin,
   BaseTool,
   BaseToolset,
+  BuiltInCodeExecutor,
   CallbackContext,
   FunctionTool,
   GOOGLE_SEARCH,
@@ -87,6 +89,8 @@ export {
   getFunctionCalls,
   getFunctionResponses,
   hasTrailingCodeExecutionResult,
+  isBaseAgent,
+  isBaseLlm,
   isFinalResponse,
   setLogLevel,
   stringifyContent,