@google/adk 0.1.0

package/dist/esm/agents/loop_agent.js

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseAgent } from "./base_agent.js";
+class LoopAgent extends BaseAgent {
+  constructor(config) {
+    var _a;
+    super(config);
+    this.maxIterations = (_a = config.maxIterations) != null ? _a : Number.MAX_SAFE_INTEGER;
+  }
+  async *runAsyncImpl(context) {
+    let iteration = 0;
+    while (iteration < this.maxIterations) {
+      for (const subAgent of this.subAgents) {
+        let shouldExit = false;
+        for await (const event of subAgent.runAsync(context)) {
+          yield event;
+          if (event.actions.escalate) {
+            shouldExit = true;
+          }
+        }
+        if (shouldExit) {
+          return;
+        }
+      }
+      iteration++;
+    }
+    return;
+  }
+  async *runLiveImpl(context) {
+    throw new Error("This is not supported yet for LoopAgent.");
+  }
+}
+export {
+  LoopAgent
+};

package/dist/esm/agents/parallel_agent.js

@@ -0,0 +1,48 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseAgent } from "./base_agent.js";
+import { InvocationContext } from "./invocation_context.js";
+class ParallelAgent extends BaseAgent {
+  async *runAsyncImpl(context) {
+    const agentRuns = this.subAgents.map(
+      (subAgent) => subAgent.runAsync(
+        createBranchCtxForSubAgent(this, subAgent, context)
+      )
+    );
+    for await (const event of mergeAgentRuns(agentRuns)) {
+      yield event;
+    }
+  }
+  async *runLiveImpl(context) {
+    throw new Error("This is not supported yet for ParallelAgent.");
+  }
+}
+function createBranchCtxForSubAgent(agent, subAgent, originalContext) {
+  const invocationContext = new InvocationContext(originalContext);
+  const branchSuffix = `${agent.name}.${subAgent.name}`;
+  invocationContext.branch = invocationContext.branch ? `${invocationContext.branch}.${branchSuffix}` : branchSuffix;
+  return invocationContext;
+}
+async function* mergeAgentRuns(agentRuns) {
+  const pendingPromises = /* @__PURE__ */ new Map();
+  for (const [index, generator] of agentRuns.entries()) {
+    const promise = generator.next().then((result) => ({ result, index }));
+    pendingPromises.set(index, promise);
+  }
+  while (pendingPromises.size > 0) {
+    const { result, index } = await Promise.race(pendingPromises.values());
+    if (result.done) {
+      pendingPromises.delete(index);
+      continue;
+    }
+    yield result.value;
+    const nextPromise = agentRuns[index].next().then((result2) => ({ result: result2, index }));
+    pendingPromises.set(index, nextPromise);
+  }
+}
+export {
+  ParallelAgent
+};

package/dist/esm/agents/readonly_context.js

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { State } from "../sessions/state.js";
+class ReadonlyContext {
+  constructor(invocationContext) {
+    this.invocationContext = invocationContext;
+  }
+  /**
+   * The user content that started this invocation.
+   */
+  get userContent() {
+    return this.invocationContext.userContent;
+  }
+  /**
+   * The current invocation id.
+   */
+  get invocationId() {
+    return this.invocationContext.invocationId;
+  }
+  /**
+   * The current agent name.
+   */
+  get agentName() {
+    return this.invocationContext.agent.name;
+  }
+  /**
+   * The state of the current session.
+   */
+  get state() {
+    return new State(this.invocationContext.session.state, {});
+  }
+}
+export {
+  ReadonlyContext
+};

package/dist/esm/agents/run_config.js

@@ -0,0 +1,43 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { logger } from "../utils/logger.js";
+var StreamingMode = /* @__PURE__ */ ((StreamingMode2) => {
+  StreamingMode2["NONE"] = "none";
+  StreamingMode2["SSE"] = "sse";
+  StreamingMode2["BIDI"] = "bidi";
+  return StreamingMode2;
+})(StreamingMode || {});
+class RunConfig {
+  constructor(params = {}) {
+    this.speechConfig = params.speechConfig;
+    this.responseModalities = params.responseModalities;
+    this.saveInputBlobsAsArtifacts = params.saveInputBlobsAsArtifacts || false;
+    this.supportCfc = params.supportCfc || false;
+    this.streamingMode = params.streamingMode || "none" /* NONE */;
+    this.outputAudioTranscription = params.outputAudioTranscription;
+    this.inputAudioTranscription = params.inputAudioTranscription;
+    this.enableAffectiveDialog = params.enableAffectiveDialog || false;
+    this.realtimeInputConfig = params.realtimeInputConfig;
+    this.maxLlmCalls = validateMaxLlmCalls(params.maxLlmCalls || 500);
+  }
+}
+function validateMaxLlmCalls(value) {
+  if (value > Number.MAX_SAFE_INTEGER) {
+    throw new Error(
+      `maxLlmCalls should be less than ${Number.MAX_SAFE_INTEGER}.`
+    );
+  }
+  if (value <= 0) {
+    logger.warn(
+      "maxLlmCalls is less than or equal to 0. This will result in no enforcement on total number of llm calls that will be made for a run. This may not be ideal, as this could result in a never ending communication between the model and the agent in certain cases."
+    );
+  }
+  return value;
+}
+export {
+  RunConfig,
+  StreamingMode
+};

package/dist/esm/agents/sequential_agent.js

@@ -0,0 +1,54 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionTool } from "../tools/function_tool.js";
+import { BaseAgent } from "./base_agent.js";
+import { LlmAgent } from "./llm_agent.js";
+import { ReadonlyContext } from "./readonly_context.js";
+const TASK_COMPLETED_TOOL_NAME = "task_completed";
+class SequentialAgent extends BaseAgent {
+  async *runAsyncImpl(context) {
+    for (const subAgent of this.subAgents) {
+      for await (const event of subAgent.runAsync(context)) {
+        yield event;
+      }
+    }
+  }
+  /**
+   * Implementation for live SequentialAgent.
+   *
+   * Compared to the non-live case, live agents process a continuous stream of
+   * audio or video, so there is no way to tell if it's finished and should pass
+   * to the next agent or not. So we introduce a task_completed() function so
+   * the model can call this function to signal that it's finished the task and
+   * we can move on to the next agent.
+   *
+   * @param context: The invocation context of the agent.
+   */
+  async *runLiveImpl(context) {
+    for (const subAgent of this.subAgents) {
+      if (subAgent instanceof LlmAgent) {
+        const agentTools = await subAgent.canonicalTools(new ReadonlyContext(context));
+        const taskCompletedToolAlreadyAdded = agentTools.some((tool) => tool.name === TASK_COMPLETED_TOOL_NAME);
+        if (!taskCompletedToolAlreadyAdded) {
+          subAgent.tools.push(new FunctionTool({
+            name: TASK_COMPLETED_TOOL_NAME,
+            description: `Signals that the model has successfully completed the user's question or task.`,
+            execute: () => "Task completion signaled."
+          }));
+          subAgent.instruction += `If you finished the user's request according to its description, call the ${TASK_COMPLETED_TOOL_NAME} function to exit so the next agents can take over. When calling this function, do not generate any text other than the function call.`;
+        }
+      }
+    }
+    for (const subAgent of this.subAgents) {
+      for await (const event of subAgent.runLive(context)) {
+        yield event;
+      }
+    }
+  }
+}
+export {
+  SequentialAgent
+};

package/dist/esm/agents/transcription_entry.js

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

package/dist/esm/artifacts/base_artifact_service.js

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

package/dist/esm/artifacts/in_memory_artifact_service.js

@@ -0,0 +1,89 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class InMemoryArtifactService {
+  constructor() {
+    this.artifacts = {};
+  }
+  saveArtifact({
+    appName,
+    userId,
+    sessionId,
+    filename,
+    artifact
+  }) {
+    const path = artifactPath(appName, userId, sessionId, filename);
+    if (!this.artifacts[path]) {
+      this.artifacts[path] = [];
+    }
+    const version = this.artifacts[path].length;
+    this.artifacts[path].push(artifact);
+    return Promise.resolve(version);
+  }
+  loadArtifact({
+    appName,
+    userId,
+    sessionId,
+    filename,
+    version
+  }) {
+    const path = artifactPath(appName, userId, sessionId, filename);
+    const versions = this.artifacts[path];
+    if (!versions) {
+      return Promise.resolve(void 0);
+    }
+    if (version === void 0) {
+      version = versions.length - 1;
+    }
+    return Promise.resolve(versions[version]);
+  }
+  listArtifactKeys({ appName, userId, sessionId }) {
+    const sessionPrefix = `${appName}/${userId}/${sessionId}/`;
+    const usernamespacePrefix = `${appName}/${userId}/user/`;
+    const filenames = [];
+    for (const path in this.artifacts) {
+      if (path.startsWith(sessionPrefix)) {
+        const filename = path.replace(sessionPrefix, "");
+        filenames.push(filename);
+      } else if (path.startsWith(usernamespacePrefix)) {
+        const filename = path.replace(usernamespacePrefix, "");
+        filenames.push(filename);
+      }
+    }
+    return Promise.resolve(filenames.sort());
+  }
+  deleteArtifact({ appName, userId, sessionId, filename }) {
+    const path = artifactPath(appName, userId, sessionId, filename);
+    if (!this.artifacts[path]) {
+      return Promise.resolve();
+    }
+    delete this.artifacts[path];
+    return Promise.resolve();
+  }
+  listVersions({ appName, userId, sessionId, filename }) {
+    const path = artifactPath(appName, userId, sessionId, filename);
+    const artifacts = this.artifacts[path];
+    if (!artifacts) {
+      return Promise.resolve([]);
+    }
+    let versions = [];
+    for (let i = 0; i < artifacts.length; i++) {
+      versions.push(i);
+    }
+    return Promise.resolve(versions);
+  }
+}
+function artifactPath(appName, userId, sessionId, filename) {
+  if (fileHasUserNamespace(filename)) {
+    return `${appName}/${userId}/user/${filename}`;
+  }
+  return `${appName}/${userId}/${sessionId}/${filename}`;
+}
+function fileHasUserNamespace(filename) {
+  return filename.startsWith("user:");
+}
+export {
+  InMemoryArtifactService
+};

package/dist/esm/auth/auth_credential.js

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+var AuthCredentialTypes = /* @__PURE__ */ ((AuthCredentialTypes2) => {
+  AuthCredentialTypes2["API_KEY"] = "apiKey";
+  AuthCredentialTypes2["HTTP"] = "http";
+  AuthCredentialTypes2["OAUTH2"] = "oauth2";
+  AuthCredentialTypes2["OPEN_ID_CONNECT"] = "openIdConnect";
+  AuthCredentialTypes2["SERVICE_ACCOUNT"] = "serviceAccount";
+  return AuthCredentialTypes2;
+})(AuthCredentialTypes || {});
+export {
+  AuthCredentialTypes
+};

package/dist/esm/auth/auth_handler.js

@@ -0,0 +1,62 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class AuthHandler {
+  constructor(authConfig) {
+    this.authConfig = authConfig;
+  }
+  getAuthResponse(state) {
+    const credentialKey = "temp:" + this.authConfig.credentialKey;
+    return state.get(credentialKey);
+  }
+  generateAuthRequest() {
+    var _a, _b;
+    const authSchemeType = this.authConfig.authScheme.type;
+    if (!["oauth2", "openIdConnect"].includes(authSchemeType)) {
+      return this.authConfig;
+    }
+    if ((_b = (_a = this.authConfig.exchangedAuthCredential) == null ? void 0 : _a.oauth2) == null ? void 0 : _b.authUri) {
+      return this.authConfig;
+    }
+    if (!this.authConfig.rawAuthCredential) {
+      throw new Error(`Auth Scheme ${authSchemeType} requires authCredential.`);
+    }
+    if (!this.authConfig.rawAuthCredential.oauth2) {
+      throw new Error(
+        `Auth Scheme ${authSchemeType} requires oauth2 in authCredential.`
+      );
+    }
+    if (this.authConfig.rawAuthCredential.oauth2.authUri) {
+      return {
+        credentialKey: this.authConfig.credentialKey,
+        authScheme: this.authConfig.authScheme,
+        rawAuthCredential: this.authConfig.rawAuthCredential,
+        exchangedAuthCredential: this.authConfig.rawAuthCredential
+      };
+    }
+    if (!this.authConfig.rawAuthCredential.oauth2.clientId || !this.authConfig.rawAuthCredential.oauth2.clientSecret) {
+      throw new Error(`Auth Scheme ${authSchemeType} requires both clientId and clientSecret in authCredential.oauth2.`);
+    }
+    return {
+      credentialKey: this.authConfig.credentialKey,
+      authScheme: this.authConfig.authScheme,
+      rawAuthCredential: this.authConfig.rawAuthCredential,
+      exchangedAuthCredential: this.generateAuthUri()
+    };
+  }
+  /**
+   * Generates an response containing the auth uri for user to sign in.
+   *
+   * @return An AuthCredential object containing the auth URI and state.
+   * @throws Error: If the authorization endpoint is not configured in the
+   *     auth scheme.
+   */
+  generateAuthUri() {
+    return this.authConfig.rawAuthCredential;
+  }
+}
+export {
+  AuthHandler
+};

package/dist/esm/auth/auth_schemes.js

@@ -0,0 +1,31 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+var OAuthGrantType = /* @__PURE__ */ ((OAuthGrantType2) => {
+  OAuthGrantType2["CLIENT_CREDENTIALS"] = "client_credentials";
+  OAuthGrantType2["AUTHORIZATION_CODE"] = "authorization_code";
+  OAuthGrantType2["IMPLICIT"] = "implicit";
+  OAuthGrantType2["PASSWORD"] = "password";
+  return OAuthGrantType2;
+})(OAuthGrantType || {});
+function getOAuthGrantTypeFromFlow(flow) {
+  if (flow.clientCredentials) {
+    return "client_credentials" /* CLIENT_CREDENTIALS */;
+  }
+  if (flow.authorizationCode) {
+    return "authorization_code" /* AUTHORIZATION_CODE */;
+  }
+  if (flow.implicit) {
+    return "implicit" /* IMPLICIT */;
+  }
+  if (flow.password) {
+    return "password" /* PASSWORD */;
+  }
+  return void 0;
+}
+export {
+  OAuthGrantType,
+  getOAuthGrantTypeFromFlow
+};

package/dist/esm/auth/auth_tool.js

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

package/dist/esm/auth/credential_service/base_credential_service.js

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

package/dist/esm/auth/credential_service/in_memory_credential_service.js

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class InMemoryCredentialService {
+  constructor() {
+    this.credentials = {};
+  }
+  loadCredential(authConfig, toolContext) {
+    const credentialBucket = this.getBucketForCurrentContext(toolContext);
+    return Promise.resolve(credentialBucket[authConfig.credentialKey]);
+  }
+  async saveCredential(authConfig, toolContext) {
+    const credentialBucket = this.getBucketForCurrentContext(toolContext);
+    if (authConfig.exchangedAuthCredential) {
+      credentialBucket[authConfig.credentialKey] = authConfig.exchangedAuthCredential;
+    }
+  }
+  getBucketForCurrentContext(toolContext) {
+    const { appName, userId } = toolContext.invocationContext.session;
+    if (!this.credentials[appName]) {
+      this.credentials[appName] = {};
+    }
+    if (!this.credentials[appName][userId]) {
+      this.credentials[appName][userId] = {};
+    }
+    return this.credentials[appName][userId];
+  }
+}
+export {
+  InMemoryCredentialService
+};

package/dist/esm/code_executors/base_code_executor.js

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class BaseCodeExecutor {
+  constructor() {
+    /**
+     * If true, extract and process data files from the model request
+     * and attach them to the code executor.
+     *
+     * Supported data file MimeTypes are [text/csv].
+     * Default to false.
+     */
+    this.optimizeDataFile = false;
+    /**
+     * Whether the code executor is stateful. Default to false.
+     */
+    this.stateful = false;
+    /**
+     * The number of attempts to retry on consecutive code execution errors.
+     * Default to 2.
+     */
+    this.errorRetryAttempts = 2;
+    /**
+     * The list of the enclosing delimiters to identify the code blocks.
+     * For example, the delimiter('```python\\n', '\\n```') can be  used to
+     * identify code blocks with the following format::
+     *
+     * ```python
+     *  print("hello")
+     * ```
+     */
+    this.codeBlockDelimiters = [
+      ["```tool_code\n", "\n```"],
+      ["```python\n", "\n```"]
+    ];
+    /**
+     * The delimiters to format the code execution result.
+     */
+    this.executionResultDelimiters = ["```tool_output\n", "\n```"];
+  }
+}
+export {
+  BaseCodeExecutor
+};

package/dist/esm/code_executors/built_in_code_executor.js

@@ -0,0 +1,28 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isGemini2Model } from "../utils/model_name.js";
+import { BaseCodeExecutor } from "./base_code_executor.js";
+class BuiltInCodeExecutor extends BaseCodeExecutor {
+  executeCode(params) {
+    return Promise.resolve({
+      stdout: "",
+      stderr: "",
+      outputFiles: []
+    });
+  }
+  processLlmRequest(llmRequest) {
+    if (llmRequest.model && isGemini2Model(llmRequest.model)) {
+      llmRequest.config = llmRequest.config || {};
+      llmRequest.config.tools = llmRequest.config.tools || [];
+      llmRequest.config.tools.push({ codeExecution: {} });
+      return;
+    }
+    throw new Error(`Gemini code execution tool is not supported for model ${llmRequest.model}`);
+  }
+}
+export {
+  BuiltInCodeExecutor
+};

package/dist/esm/code_executors/code_execution_utils.js

@@ -0,0 +1,108 @@
+/**
+ * @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>.*?)(${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 {
+  buildCodeExecutionResultPart,
+  buildExecutableCodePart,
+  convertCodeExecutionParts,
+  extractCodeAndTruncateContent,
+  getEncodedFileContent
+};