@google/adk 0.2.0 → 0.2.2

package/dist/cjs/models/base_llm.js

@@ -24,7 +24,8 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var base_llm_exports = {};
 __export(base_llm_exports, {
-  BaseLlm: () => BaseLlm
+  BaseLlm: () => BaseLlm,
+  isBaseLlm: () => isBaseLlm
 });
 module.exports = __toCommonJS(base_llm_exports);
 var import_client_labels = require("../utils/client_labels.js");
@@ -33,6 +34,12 @@ var import_client_labels = require("../utils/client_labels.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+var _a;
+const BASE_MODEL_SYMBOL = Symbol.for("google.adk.baseModel");
+function isBaseLlm(obj) {
+  return typeof obj === "object" && obj !== null && BASE_MODEL_SYMBOL in obj && obj[BASE_MODEL_SYMBOL] === true;
+}
+_a = BASE_MODEL_SYMBOL;
 class BaseLlm {
   /**
    * Creates an instance of BaseLLM.
@@ -41,6 +48,10 @@ class BaseLlm {
    *     gemini-1.5-flash-001.
    */
   constructor({ model }) {
+    /**
+     * A unique symbol to identify BaseLlm classes.
+     */
+    this[_a] = true;
     this.model = model;
   }
   get trackingHeaders() {
@@ -57,7 +68,7 @@ class BaseLlm {
    * @param llmRequest LlmRequest, the request to send to the LLM.
    */
   maybeAppendUserContent(llmRequest) {
-    var _a;
+    var _a2;
     if (llmRequest.contents.length === 0) {
       llmRequest.contents.push({
         role: "user",
@@ -66,7 +77,7 @@ class BaseLlm {
         ]
       });
     }
-    if (((_a = llmRequest.contents[llmRequest.contents.length - 1]) == null ? void 0 : _a.role) !== "user") {
+    if (((_a2 = llmRequest.contents[llmRequest.contents.length - 1]) == null ? void 0 : _a2.role) !== "user") {
       llmRequest.contents.push({
         role: "user",
         parts: [{
@@ -82,5 +93,6 @@ class BaseLlm {
 BaseLlm.supportedModels = [];
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  BaseLlm
+  BaseLlm,
+  isBaseLlm
 });

package/dist/cjs/runner/runner.js

@@ -32,6 +32,7 @@ var import_api = require("@opentelemetry/api");
 var import_invocation_context = require("../agents/invocation_context.js");
 var import_llm_agent = require("../agents/llm_agent.js");
 var import_run_config = require("../agents/run_config.js");
+var import_built_in_code_executor = require("../code_executors/built_in_code_executor.js");
 var import_event = require("../events/event.js");
 var import_event_actions = require("../events/event_actions.js");
 var import_plugin_manager = require("../plugins/plugin_manager.js");
@@ -77,6 +78,11 @@ class Runner {
     try {
       const session = await this.sessionService.getSession({ appName: this.appName, userId, sessionId });
       if (!session) {
+        if (!this.appName) {
+          throw new Error(
+            `Session lookup failed: appName must be provided in runner constructor`
+          );
+        }
         throw new Error(`Session not found: ${sessionId}`);
       }
       if (runConfig.supportCfc && this.agent instanceof import_llm_agent.LlmAgent) {
@@ -85,6 +91,9 @@ class Runner {
           throw new Error(`CFC is not supported for model: ${modelName} in agent: ${this.agent.name}`);
         }
       }
+      if (this.agent instanceof import_llm_agent.LlmAgent && !(this.agent.codeExecutor instanceof import_built_in_code_executor.BuiltInCodeExecutor)) {
+        this.agent.codeExecutor = new import_built_in_code_executor.BuiltInCodeExecutor();
+      }
       const invocationContext = new import_invocation_context.InvocationContext({
         artifactService: this.artifactService,
         sessionService: this.sessionService,
@@ -106,7 +115,7 @@ class Runner {
       }
       if (newMessage) {
         if (!((_a = newMessage.parts) == null ? void 0 : _a.length)) {
-          throw new Error("No parts in the new_message.");
+          throw new Error("No parts in the newMessage.");
         }
         if (runConfig.saveInputBlobsAsArtifacts) {
           await this.saveArtifacts(

package/dist/cjs/sessions/in_memory_session_service.js

@@ -27,7 +27,7 @@ __export(in_memory_session_service_exports, {
   InMemorySessionService: () => InMemorySessionService
 });
 module.exports = __toCommonJS(in_memory_session_service_exports);
-var import_deep_clone = require("../utils/deep_clone.js");
+var import_lodash = require("lodash");
 var import_env_aware_utils = require("../utils/env_aware_utils.js");
 var import_logger = require("../utils/logger.js");
 var import_base_session_service = require("./base_session_service.js");
@@ -72,7 +72,7 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     }
     this.sessions[appName][userId][session.id] = session;
     return Promise.resolve(
-      this.mergeState(appName, userId, (0, import_deep_clone.deepClone)(session))
+      this.mergeState(appName, userId, (0, import_lodash.cloneDeep)(session))
     );
   }
   getSession({ appName, userId, sessionId, config }) {
@@ -80,7 +80,7 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
       return Promise.resolve(void 0);
     }
     const session = this.sessions[appName][userId][sessionId];
-    const copiedSession = (0, import_deep_clone.deepClone)(session);
+    const copiedSession = (0, import_lodash.cloneDeep)(session);
     if (config) {
       if (config.numRecentEvents) {
         copiedSession.events = copiedSession.events.slice(-config.numRecentEvents);

package/dist/cjs/sessions/state.js

@@ -33,7 +33,7 @@ module.exports = __toCommonJS(state_exports);
  * SPDX-License-Identifier: Apache-2.0
  */
 class State {
-  constructor(value, delta) {
+  constructor(value = {}, delta = {}) {
     this.value = value;
     this.delta = delta;
   }

package/dist/cjs/tools/agent_tool.js

@@ -124,8 +124,8 @@ class AgentTool extends import_base_tool.BaseTool {
       return "";
     }
     const hasOutputSchema = this.agent instanceof import_llm_agent.LlmAgent && this.agent.outputSchema;
-    const mergetText = lastEvent.content.parts.map((part) => part.text).filter((text) => text).join("\n");
-    return hasOutputSchema ? JSON.parse(mergetText) : mergetText;
+    const mergedText = lastEvent.content.parts.map((part) => part.text).filter((text) => text).join("\n");
+    return hasOutputSchema ? JSON.parse(mergedText) : mergedText;
   }
 }
 // Annotate the CommonJS export names for ESM import in node:

package/dist/cjs/tools/mcp/mcp_session_manager.js

@@ -48,8 +48,14 @@ class MCPSessionManager {
         );
         break;
       case "StreamableHTTPConnectionParams":
+        const transportOptions = this.connectionParams.header ? {
+          requestInit: {
+            headers: this.connectionParams.header
+          }
+        } : void 0;
         await client.connect(new import_streamableHttp.StreamableHTTPClientTransport(
-          new URL(this.connectionParams.url)
+          new URL(this.connectionParams.url),
+          transportOptions
         ));
         break;
       default:

package/dist/cjs/utils/gemini_schema_util.js

@@ -63,6 +63,22 @@ function toGeminiSchema(mcpSchema) {
     return void 0;
   }
   function recursiveConvert(mcp) {
+    if (!mcp.type && mcp.anyOf && Array.isArray(mcp.anyOf)) {
+      const nonNullOption = mcp.anyOf.find((opt) => {
+        const t = opt.type;
+        return t !== "null" && t !== "NULL";
+      });
+      if (nonNullOption) {
+        mcp = nonNullOption;
+      }
+    }
+    if (!mcp.type) {
+      if (mcp.properties || mcp.$ref) {
+        mcp.type = "object";
+      } else if (mcp.items) {
+        mcp.type = "array";
+      }
+    }
     const geminiType = toGeminiType(mcp.type);
     const geminiSchema = { type: geminiType, description: mcp.description };
     if (geminiType === import_genai.Type.OBJECT) {

package/dist/cjs/version.js

@@ -29,10 +29,10 @@ __export(version_exports, {
 module.exports = __toCommonJS(version_exports);
 /**
  * @license
- * Copyright 2025 Google LLC
+ * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const version = "0.2.0";
+const version = "0.2.3";
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   version

package/dist/esm/agents/base_agent.js

@@ -3,12 +3,22 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+var _a;
 import { trace } from "@opentelemetry/api";
 import { createEvent } from "../events/event.js";
 import { CallbackContext } from "./callback_context.js";
 import { InvocationContext } from "./invocation_context.js";
+const BASE_AGENT_SIGNATURE_SYMBOL = Symbol.for("google.adk.baseAgent");
+function isBaseAgent(obj) {
+  return typeof obj === "object" && obj !== null && BASE_AGENT_SIGNATURE_SYMBOL in obj && obj[BASE_AGENT_SIGNATURE_SYMBOL] === true;
+}
+_a = BASE_AGENT_SIGNATURE_SYMBOL;
 class BaseAgent {
   constructor(config) {
+    /**
+     * A unique symbol to identify ADK agent classes.
+     */
+    this[_a] = true;
     this.name = validateAgentName(config.name);
     this.description = config.description;
     this.parentAgent = config.parentAgent;
@@ -210,5 +220,6 @@ function getCannonicalCallback(callbacks) {
 }
 export {
   BaseAgent,
-  getCannonicalCallback
+  getCannonicalCallback,
+  isBaseAgent
 };

package/dist/esm/agents/content_processor_utils.js

@@ -3,8 +3,8 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { cloneDeep } from "lodash";
 import { createEvent, getFunctionCalls, getFunctionResponses } from "../events/event.js";
-import { deepClone } from "../utils/deep_clone.js";
 import { removeClientFunctionCallId, REQUEST_CONFIRMATION_FUNCTION_CALL_NAME, REQUEST_EUC_FUNCTION_CALL_NAME } from "./functions.js";
 function getContents(events, agentName, currentBranch) {
   var _a, _b, _c;
@@ -30,7 +30,7 @@ function getContents(events, agentName, currentBranch) {
   resultEvents = rearrangeEventsForAsyncFunctionResponsesInHistory(resultEvents);
   const contents = [];
   for (const event of resultEvents) {
-    const content = deepClone(event.content);
+    const content = cloneDeep(event.content);
     removeClientFunctionCallId(content);
     contents.push(content);
   }

package/dist/esm/agents/functions.js

@@ -4,6 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { createUserContent } from "@google/genai";
+import { isEmpty } from "lodash";
 import { createEvent, getFunctionCalls } from "../events/event.js";
 import { mergeEventActions } from "../events/event_actions.js";
 import { ToolContext } from "../tools/tool_context.js";
@@ -13,7 +14,9 @@ const AF_FUNCTION_CALL_ID_PREFIX = "adk-";
 const REQUEST_EUC_FUNCTION_CALL_NAME = "adk_request_credential";
 const REQUEST_CONFIRMATION_FUNCTION_CALL_NAME = "adk_request_confirmation";
 const functionsExportedForTestingOnly = {
-  handleFunctionCallList
+  handleFunctionCallList,
+  generateAuthEvent,
+  generateRequestConfirmationEvent
 };
 function generateClientFunctionCallId() {
   return `${AF_FUNCTION_CALL_ID_PREFIX}${randomUUID()}`;
@@ -52,7 +55,7 @@ function getLongRunningFunctionCalls(functionCalls, toolsDict) {
 }
 function generateAuthEvent(invocationContext, functionResponseEvent) {
   var _a;
-  if (!((_a = functionResponseEvent.actions) == null ? void 0 : _a.requestedAuthConfigs)) {
+  if (!((_a = functionResponseEvent.actions) == null ? void 0 : _a.requestedAuthConfigs) || isEmpty(functionResponseEvent.actions.requestedAuthConfigs)) {
     return void 0;
   }
   const parts = [];
@@ -88,7 +91,7 @@ function generateRequestConfirmationEvent({
   functionResponseEvent
 }) {
   var _a, _b;
-  if (!((_a = functionResponseEvent.actions) == null ? void 0 : _a.requestedToolConfirmations)) {
+  if (!((_a = functionResponseEvent.actions) == null ? void 0 : _a.requestedToolConfirmations) || isEmpty(functionResponseEvent.actions.requestedToolConfirmations)) {
     return;
   }
   const parts = [];

package/dist/esm/agents/llm_agent.js

@@ -3,15 +3,23 @@
  * 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";
@@ -323,6 +331,321 @@ class RequestConfirmationLlmRequestProcessor extends BaseLlmRequestProcessor {
   }
 }
 const REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR = new RequestConfirmationLlmRequestProcessor();
+class CodeExecutionRequestProcessor extends BaseLlmRequestProcessor {
+  async *runAsync(invocationContext, llmRequest) {
+    if (!(invocationContext.agent instanceof LlmAgent)) {
+      return;
+    }
+    if (!invocationContext.agent.codeExecutor) {
+      return;
+    }
+    for await (const event of runPreProcessor(invocationContext, llmRequest)) {
+      yield event;
+    }
+    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 = `
+import pandas as pd
+
+def explore_df(df: pd.DataFrame) -> None:
+  """Prints some information about a pandas DataFrame."""
+
+  with pd.option_context(
+      'display.max_columns', None, 'display.expand_frame_repr', False
+  ):
+    # Print the column names to never encounter KeyError when selecting one.
+    df_dtypes = df.dtypes
+
+    # Obtain information about data types and missing values.
+    df_nulls = (len(df) - df.isnull().sum()).apply(
+        lambda x: f'{x} / {df.shape[0]} non-null'
+    )
+
+    # Explore unique total values in columns using \`.unique()\`.
+    df_unique_count = df.apply(lambda x: len(x.unique()))
+
+    # Explore unique values in columns using \`.unique()\`.
+    df_unique = df.apply(lambda x: crop(str(list(x.unique()))))
+
+    df_info = pd.concat(
+        (
+            df_dtypes.rename('Dtype'),
+            df_nulls.rename('Non-Null Count'),
+            df_unique_count.rename('Unique Values Count'),
+            df_unique.rename('Unique Values'),
+        ),
+        axis=1,
+    )
+    df_info.index.name = 'Columns'
+    print(f"""Total rows: {df.shape[0]}
+Total columns: {df.shape[1]}
+
+{df_info}""")
+`;
+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
+   */
+  async *runAsync(invocationContext, llmResponse) {
+    if (llmResponse.partial) {
+      return;
+    }
+    for await (const event of runPostProcessor(invocationContext, llmResponse)) {
+      yield event;
+    }
+  }
+}
+const responseProcessor = new CodeExecutionResponseProcessor();
+async function* runPreProcessor(invocationContext, llmRequest) {
+  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: \`${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 = 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 = await postProcessCodeExecutionResult(
+      invocationContext,
+      codeExecutorContext,
+      codeExecutionResult
+    );
+    yield executionResultEvent;
+    llmRequest.contents.push(cloneDeep(executionResultEvent.content));
+  }
+}
+async function* runPostProcessor(invocationContext, llmResponse) {
+  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 = 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 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_${i + 1}_${j + 1}${DATA_FILE_UTIL_MAP[mimeType].extension}`;
+      part.text = `
+Available file: \`${fileName}\`
+`;
+      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 `
+${DATA_FILE_HELPER_LIB}
+
+# Load the dataframe.
+${varName} = ${loaderCode}
+
+# Use \`explore_df\` to guide my analysis.
+explore_df(${varName})
+`;
+}
+const CODE_EXECUTION_REQUEST_PROCESSOR = new CodeExecutionRequestProcessor();
 class LlmAgent extends BaseAgent {
   constructor(config) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i;
@@ -342,12 +665,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);
@@ -397,7 +722,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) {
@@ -830,5 +1155,6 @@ class LlmAgent extends BaseAgent {
 }
 export {
   LlmAgent,
-  REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR
+  REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR,
+  responseProcessor
 };

package/dist/esm/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
+};