@google/adk 0.1.0

package/dist/web/code_executors/code_execution_utils.js

@@ -0,0 +1,105 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Language, Outcome } from "@google/genai";
+import { deepClone } from "../utils/deep_clone.js";
+import { base64Encode, isBase64Encoded } from "../utils/env_aware_utils.js";
+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 = deepClone(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;
+}
+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:\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
+    }
+  };
+}
+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 {
+  buildCodeExecutionResultPart,
+  buildExecutableCodePart,
+  convertCodeExecutionParts,
+  extractCodeAndTruncateContent,
+  getEncodedFileContent
+};

package/dist/web/code_executors/code_executor_context.js

@@ -0,0 +1,168 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { deepClone } from "../utils/deep_clone.js";
+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]: deepClone(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 executor context from the session state.
+   * @param invocationId The session state to get the code executor context
+   *     from.
+   * @return The code execution context for the given invocation ID.
+   */
+  getCodeExecutionContext(invocationId) {
+    return this.sessionState.get(CONTEXT_KEY) || {};
+  }
+}
+export {
+  CodeExecutorContext
+};

package/dist/web/common.js

@@ -0,0 +1,85 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseAgent } 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";
+import { LiveRequestQueue } from "./agents/live_request_queue.js";
+import { LlmAgent } from "./agents/llm_agent.js";
+import { LoopAgent } from "./agents/loop_agent.js";
+import { ParallelAgent } from "./agents/parallel_agent.js";
+import { RunConfig, StreamingMode } from "./agents/run_config.js";
+import { SequentialAgent } from "./agents/sequential_agent.js";
+import { InMemoryArtifactService } from "./artifacts/in_memory_artifact_service.js";
+import { createEvent, getFunctionCalls, getFunctionResponses, hasTrailingCodeExecutionResult, isFinalResponse } from "./events/event.js";
+import { InMemoryMemoryService } from "./memory/in_memory_memory_service.js";
+import { BaseLlm } from "./models/base_llm.js";
+import { BasePlugin } from "./plugins/base_plugin.js";
+import { LoggingPlugin } from "./plugins/logging_plugin.js";
+import { PluginManager } from "./plugins/plugin_manager.js";
+import { getAskUserConfirmationFunctionCalls, InMemoryPolicyEngine, PolicyOutcome, REQUEST_CONFIRMATION_FUNCTION_CALL_NAME, SecurityPlugin } from "./plugins/security_plugin.js";
+import { InMemoryRunner } from "./runner/in_memory_runner.js";
+import { Runner } from "./runner/runner.js";
+import { InMemorySessionService } from "./sessions/in_memory_session_service.js";
+import { createSession } from "./sessions/session.js";
+import { AgentTool } from "./tools/agent_tool.js";
+import { BaseTool } from "./tools/base_tool.js";
+import { BaseToolset } from "./tools/base_toolset.js";
+import { FunctionTool } from "./tools/function_tool.js";
+import { GOOGLE_SEARCH } from "./tools/google_search_tool.js";
+import { LongRunningFunctionTool } from "./tools/long_running_tool.js";
+import { ToolConfirmation } from "./tools/tool_confirmation.js";
+import { ToolContext } from "./tools/tool_context.js";
+import { LogLevel, setLogLevel } from "./utils/logger.js";
+import { zodObjectToSchema } from "./utils/simple_zod_to_json.js";
+export * from "./artifacts/base_artifact_service.js";
+export * from "./memory/base_memory_service.js";
+export * from "./sessions/base_session_service.js";
+export * from "./tools/base_tool.js";
+export {
+  AgentTool,
+  BaseAgent,
+  BaseLlm,
+  BasePlugin,
+  BaseTool,
+  BaseToolset,
+  CallbackContext,
+  FunctionTool,
+  GOOGLE_SEARCH,
+  InMemoryArtifactService,
+  InMemoryMemoryService,
+  InMemoryPolicyEngine,
+  InMemoryRunner,
+  InMemorySessionService,
+  InvocationContext,
+  LiveRequestQueue,
+  LlmAgent,
+  LogLevel,
+  LoggingPlugin,
+  LongRunningFunctionTool,
+  LoopAgent,
+  ParallelAgent,
+  PluginManager,
+  PolicyOutcome,
+  REQUEST_CONFIRMATION_FUNCTION_CALL_NAME,
+  RunConfig,
+  Runner,
+  SecurityPlugin,
+  SequentialAgent,
+  StreamingMode,
+  ToolConfirmation,
+  ToolContext,
+  createEvent,
+  createSession,
+  functionsExportedForTestingOnly,
+  getAskUserConfirmationFunctionCalls,
+  getFunctionCalls,
+  getFunctionResponses,
+  hasTrailingCodeExecutionResult,
+  isFinalResponse,
+  setLogLevel,
+  zodObjectToSchema
+};

package/dist/web/events/event.js

@@ -0,0 +1,90 @@
+var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { createEventActions } from "./event_actions.js";
+function createEvent(params = {}) {
+  return __spreadProps(__spreadValues({}, params), {
+    id: params.id || createNewEventId(),
+    invocationId: params.invocationId || "",
+    author: params.author,
+    actions: params.actions || createEventActions(),
+    longRunningToolIds: params.longRunningToolIds || [],
+    branch: params.branch,
+    timestamp: params.timestamp || Date.now()
+  });
+}
+function isFinalResponse(event) {
+  if (event.actions.skipSummarization || event.longRunningToolIds && event.longRunningToolIds.length > 0) {
+    return true;
+  }
+  return getFunctionCalls(event).length === 0 && getFunctionResponses(event).length === 0 && !event.partial && !hasTrailingCodeExecutionResult(event);
+}
+function getFunctionCalls(event) {
+  const funcCalls = [];
+  if (event.content && event.content.parts) {
+    for (const part of event.content.parts) {
+      if (part.functionCall) {
+        funcCalls.push(part.functionCall);
+      }
+    }
+  }
+  return funcCalls;
+}
+function getFunctionResponses(event) {
+  const funcResponses = [];
+  if (event.content && event.content.parts) {
+    for (const part of event.content.parts) {
+      if (part.functionResponse) {
+        funcResponses.push(part.functionResponse);
+      }
+    }
+  }
+  return funcResponses;
+}
+function hasTrailingCodeExecutionResult(event) {
+  var _a;
+  if (event.content && ((_a = event.content.parts) == null ? void 0 : _a.length)) {
+    const lastPart = event.content.parts[event.content.parts.length - 1];
+    return lastPart.codeExecutionResult !== void 0;
+  }
+  return false;
+}
+const ASCII_LETTERS_AND_NUMBERS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
+function createNewEventId() {
+  let id = "";
+  for (let i = 0; i < 8; i++) {
+    id += ASCII_LETTERS_AND_NUMBERS[Math.floor(
+      Math.random() * ASCII_LETTERS_AND_NUMBERS.length
+    )];
+  }
+  return id;
+}
+export {
+  createEvent,
+  createNewEventId,
+  getFunctionCalls,
+  getFunctionResponses,
+  hasTrailingCodeExecutionResult,
+  isFinalResponse
+};

package/dist/web/events/event_actions.js

@@ -0,0 +1,67 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function createEventActions(state = {}) {
+  return __spreadValues({
+    stateDelta: {},
+    artifactDelta: {},
+    requestedAuthConfigs: {},
+    requestedToolConfirmations: {}
+  }, state);
+}
+function mergeEventActions(sources, target) {
+  const result = createEventActions();
+  if (target) {
+    Object.assign(result, target);
+  }
+  for (const source of sources) {
+    if (!source) continue;
+    if (source.stateDelta) {
+      Object.assign(result.stateDelta, source.stateDelta);
+    }
+    if (source.artifactDelta) {
+      Object.assign(result.artifactDelta, source.artifactDelta);
+    }
+    if (source.requestedAuthConfigs) {
+      Object.assign(result.requestedAuthConfigs, source.requestedAuthConfigs);
+    }
+    if (source.requestedToolConfirmations) {
+      Object.assign(
+        result.requestedToolConfirmations,
+        source.requestedToolConfirmations
+      );
+    }
+    if (source.skipSummarization !== void 0) {
+      result.skipSummarization = source.skipSummarization;
+    }
+    if (source.transferToAgent !== void 0) {
+      result.transferToAgent = source.transferToAgent;
+    }
+    if (source.escalate !== void 0) {
+      result.escalate = source.escalate;
+    }
+  }
+  return result;
+}
+export {
+  createEventActions,
+  mergeEventActions
+};

package/dist/web/examples/base_example_provider.js

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

package/dist/web/examples/example.js

@@ -0,0 +1,5 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */

package/dist/web/examples/example_util.js

@@ -0,0 +1,75 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseExampleProvider } from "./base_example_provider.js";
+const EXAMPLES_INTRO = "<EXAMPLES>\nBegin few-shot\nThe following are examples of user queries and model responses using the available tools.\n\n";
+const EXAMPLES_END = "End few-shot\n<EXAMPLES>";
+const EXAMPLE_START = "EXAMPLE {}:\nBegin example\n";
+const EXAMPLE_END = "End example\n\n";
+const USER_PREFIX = "[user]\n";
+const MODEL_PREFIX = "[model]\n";
+const FUNCTION_PREFIX = "```\n";
+const FUNCTION_CALL_PREFIX = "```tool_code\n";
+const FUNCTION_CALL_SUFFIX = "\n```\n";
+const FUNCTION_RESPONSE_PREFIX = "```tool_outputs\n";
+const FUNCTION_RESPONSE_SUFFIX = "\n```\n";
+function convertExamplesToText(examples, model) {
+  var _a;
+  let examplesStr = "";
+  for (const [exampleNum, example] of examples.entries()) {
+    let output = "".concat(EXAMPLE_START.replace("{}", String(exampleNum + 1))).concat(USER_PREFIX);
+    if ((_a = example.input) == null ? void 0 : _a.parts) {
+      output += example.input.parts.filter((part) => part.text).map((part) => part.text).join("\n") + "\n";
+    }
+    const gemini2 = !model || model.includes("gemini-2");
+    let previousRole;
+    for (const content of example.output) {
+      const role = content.role === "model" ? MODEL_PREFIX : USER_PREFIX;
+      if (role !== previousRole) {
+        output += role;
+      }
+      previousRole = role;
+      for (const part of content.parts || []) {
+        if (part.functionCall) {
+          const prefix = gemini2 ? FUNCTION_PREFIX : FUNCTION_CALL_PREFIX;
+          const functionCall = part.functionCall;
+          const args = [];
+          if (functionCall.args) {
+            for (const [k, v] of Object.entries(functionCall.args)) {
+              if (typeof v === "string") {
+                args.push("".concat(k, "='").concat(v, "'"));
+              } else {
+                args.push("".concat(k, "=").concat(v));
+              }
+            }
+          }
+          const functionCallString = "".concat(functionCall.name, "(").concat(args.join(", "), ")");
+          output += "".concat(prefix).concat(functionCallString).concat(FUNCTION_CALL_SUFFIX);
+        } else if (part.functionResponse) {
+          const prefix = gemini2 ? FUNCTION_PREFIX : FUNCTION_RESPONSE_PREFIX;
+          output += "".concat(prefix).concat(JSON.stringify(part.functionResponse)).concat(FUNCTION_RESPONSE_SUFFIX);
+        } else if (part.text) {
+          output += "".concat(part.text, "\n");
+        }
+      }
+    }
+    output += EXAMPLE_END;
+    examplesStr += output;
+  }
+  return "".concat(EXAMPLES_INTRO).concat(examplesStr).concat(EXAMPLES_END);
+}
+function buildExampleSi(examples, query, model) {
+  if (Array.isArray(examples)) {
+    return convertExamplesToText(examples, model);
+  }
+  if (examples instanceof BaseExampleProvider) {
+    return convertExamplesToText(examples.getExamples(query), model);
+  }
+  throw new Error("Invalid example configuration");
+}
+export {
+  buildExampleSi,
+  convertExamplesToText
+};