@google/adk 0.1.1 → 0.1.3

package/dist/cjs/agents/functions.js

@@ -38,6 +38,7 @@ __export(functions_exports, {
   removeClientFunctionCallId: () => removeClientFunctionCallId
 });
 module.exports = __toCommonJS(functions_exports);
+var import_genai = require("@google/genai");
 var import_event = require("../events/event.js");
 var import_event_actions = require("../events/event_actions.js");
 var import_tool_context = require("../tools/tool_context.js");
@@ -166,30 +167,6 @@ async function callToolAsync(tool, args, toolContext) {
   import_logger.logger.debug(`callToolAsync ${tool.name}`);
   return await tool.runAsync({ args, toolContext });
 }
-function buildResponseEvent(tool, functionResult, toolContext, invocationContext) {
-  let responseResult = functionResult;
-  if (typeof functionResult !== "object" || functionResult == null) {
-    responseResult = { result: functionResult };
-  }
-  const partFunctionResponse = {
-    functionResponse: {
-      name: tool.name,
-      response: responseResult,
-      id: toolContext.functionCallId
-    }
-  };
-  const content = {
-    role: "user",
-    parts: [partFunctionResponse]
-  };
-  return (0, import_event.createEvent)({
-    invocationId: invocationContext.invocationId,
-    author: invocationContext.agent.name,
-    content,
-    actions: toolContext.actions,
-    branch: invocationContext.branch
-  });
-}
 async function handleFunctionCallsAsync({
   invocationContext,
   functionCallEvent,
@@ -240,6 +217,7 @@ async function handleFunctionCallList({
     import_logger.logger.debug(`execute_tool ${tool.name}`);
     const functionArgs = (_a = functionCall.args) != null ? _a : {};
     let functionResponse = null;
+    let functionResponseError;
     functionResponse = await invocationContext.pluginManager.runBeforeToolCallback({
       tool,
       toolArgs: functionArgs,
@@ -276,10 +254,11 @@ async function handleFunctionCallList({
           );
           if (onToolErrorResponse) {
             functionResponse = onToolErrorResponse;
+          } else {
+            functionResponseError = e.message;
           }
         } else {
-          import_logger.logger.error("Unknown error on tool execution type", e);
-          throw e;
+          functionResponseError = e;
         }
       }
     }
@@ -308,12 +287,24 @@ async function handleFunctionCallList({
     if (tool.isLongRunning && !functionResponse) {
       continue;
     }
-    const functionResponseEvent = buildResponseEvent(
-      tool,
-      functionResponse,
-      toolContext,
-      invocationContext
-    );
+    if (functionResponseError) {
+      functionResponse = { error: functionResponseError };
+    } else if (typeof functionResponse !== "object" || functionResponse == null) {
+      functionResponse = { result: functionResponse };
+    }
+    const functionResponseEvent = (0, import_event.createEvent)({
+      invocationId: invocationContext.invocationId,
+      author: invocationContext.agent.name,
+      content: (0, import_genai.createUserContent)({
+        functionResponse: {
+          id: toolContext.functionCallId,
+          name: tool.name,
+          response: functionResponse
+        }
+      }),
+      actions: toolContext.actions,
+      branch: invocationContext.branch
+    });
     import_logger.logger.debug("traceToolCall", {
       tool: tool.name,
       args: functionArgs,

package/dist/cjs/agents/invocation_context.js

@@ -47,9 +47,7 @@ class InvocationCostManager {
   incrementAndEnforceLlmCallsLimit(runConfig) {
     this.numberOfLlmCalls++;
     if (runConfig && runConfig.maxLlmCalls > 0 && this.numberOfLlmCalls > runConfig.maxLlmCalls) {
-      throw new Error(
-        `Max number of llm calls limit of ${runConfig.maxLlmCalls} exceeded`
-      );
+      throw new Error(`Max number of llm calls limit of ${runConfig.maxLlmCalls} exceeded`);
     }
   }
 }

package/dist/cjs/agents/llm_agent.js

@@ -45,6 +45,7 @@ var import_content_processor_utils = require("./content_processor_utils.js");
 var import_functions = require("./functions.js");
 var import_instructions = require("./instructions.js");
 var import_readonly_context = require("./readonly_context.js");
+var import_run_config = require("./run_config.js");
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -743,7 +744,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
     return agentToRun;
   }
   async *callLlmAsync(invocationContext, llmRequest, modelResponseEvent) {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e;
     const beforeModelResponse = await this.handleBeforeModelCallback(
       invocationContext,
       llmRequest,
@@ -763,7 +764,11 @@ class LlmAgent extends import_base_agent.BaseAgent {
       throw new Error("CFC is not yet supported in callLlmAsync");
     } else {
       invocationContext.incrementLlmCallCount();
-      const responsesGenerator = llm.generateContentAsync(llmRequest);
+      const responsesGenerator = llm.generateContentAsync(
+        llmRequest,
+        /* stream= */
+        ((_e = invocationContext.runConfig) == null ? void 0 : _e.streamingMode) === import_run_config.StreamingMode.SSE
+      );
       for await (const llmResponse of this.runAndHandleError(
         responsesGenerator,
         invocationContext,

package/dist/cjs/agents/run_config.js

@@ -24,8 +24,8 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var run_config_exports = {};
 __export(run_config_exports, {
-  RunConfig: () => RunConfig,
-  StreamingMode: () => StreamingMode
+  StreamingMode: () => StreamingMode,
+  createRunConfig: () => createRunConfig
 });
 module.exports = __toCommonJS(run_config_exports);
 var import_logger = require("../utils/logger.js");
@@ -40,19 +40,15 @@ var StreamingMode = /* @__PURE__ */ ((StreamingMode2) => {
   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 createRunConfig(params = {}) {
+  return {
+    saveInputBlobsAsArtifacts: false,
+    supportCfc: false,
+    enableAffectiveDialog: false,
+    streamingMode: "none" /* NONE */,
+    maxLlmCalls: validateMaxLlmCalls(params.maxLlmCalls || 500),
+    ...params
+  };
 }
 function validateMaxLlmCalls(value) {
   if (value > Number.MAX_SAFE_INTEGER) {
@@ -69,6 +65,6 @@ function validateMaxLlmCalls(value) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  RunConfig,
-  StreamingMode
+  StreamingMode,
+  createRunConfig
 });

package/dist/cjs/artifacts/gcs_artifact_service.js

@@ -0,0 +1,140 @@
+/**
+  * @license
+  * Copyright 2025 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var gcs_artifact_service_exports = {};
+__export(gcs_artifact_service_exports, {
+  GcsArtifactService: () => GcsArtifactService
+});
+module.exports = __toCommonJS(gcs_artifact_service_exports);
+var import_storage = require("@google-cloud/storage");
+var import_genai = require("@google/genai");
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class GcsArtifactService {
+  constructor(bucket) {
+    this.bucket = new import_storage.Storage().bucket(bucket);
+  }
+  async saveArtifact(request) {
+    const versions = await this.listVersions(request);
+    const version = versions.length > 0 ? Math.max(...versions) + 1 : 0;
+    const file = this.bucket.file(getFileName({
+      ...request,
+      version
+    }));
+    if (request.artifact.inlineData) {
+      await file.save(JSON.stringify(request.artifact.inlineData.data), {
+        contentType: request.artifact.inlineData.mimeType
+      });
+      return version;
+    }
+    if (request.artifact.text) {
+      await file.save(request.artifact.text, {
+        contentType: "text/plain"
+      });
+      return version;
+    }
+    throw new Error("Artifact must have either inlineData or text.");
+  }
+  async loadArtifact(request) {
+    let version = request.version;
+    if (version === void 0) {
+      const versions = await this.listVersions(request);
+      if (versions.length === 0) {
+        return void 0;
+      }
+      version = Math.max(...versions);
+    }
+    const file = this.bucket.file(getFileName({
+      ...request,
+      version
+    }));
+    const [[metadata], [rawDataBuffer]] = await Promise.all([file.getMetadata(), file.download()]);
+    if (metadata.contentType === "text/plain") {
+      return (0, import_genai.createPartFromText)(rawDataBuffer.toString("utf-8"));
+    }
+    return (0, import_genai.createPartFromBase64)(
+      rawDataBuffer.toString("base64"),
+      metadata.contentType
+    );
+  }
+  async listArtifactKeys(request) {
+    const fileNames = [];
+    const sessionPrefix = `${request.appName}/${request.userId}/${request.sessionId}/`;
+    const usernamePrefix = `${request.appName}/${request.userId}/user/`;
+    const [
+      [sessionFiles],
+      [userSessionFiles]
+    ] = await Promise.all([
+      this.bucket.getFiles({ prefix: sessionPrefix }),
+      this.bucket.getFiles({ prefix: usernamePrefix })
+    ]);
+    for (const file of sessionFiles) {
+      fileNames.push(file.name.split("/").pop());
+    }
+    for (const file of userSessionFiles) {
+      fileNames.push(file.name.split("/").pop());
+    }
+    return fileNames.sort((a, b) => a.localeCompare(b));
+  }
+  async deleteArtifact(request) {
+    const versions = await this.listVersions(request);
+    await Promise.all(versions.map((version) => {
+      const file = this.bucket.file(getFileName({
+        ...request,
+        version
+      }));
+      return file.delete();
+    }));
+    return;
+  }
+  async listVersions(request) {
+    const prefix = getFileName(request);
+    const [files] = await this.bucket.getFiles({ prefix });
+    const versions = [];
+    for (const file of files) {
+      const version = file.name.split("/").pop();
+      versions.push(parseInt(version, 10));
+    }
+    return versions;
+  }
+}
+function getFileName({
+  appName,
+  userId,
+  sessionId,
+  filename,
+  version
+}) {
+  if (filename.startsWith("user:")) {
+    return `${appName}/${userId}/user/${filename}/${version}`;
+  }
+  return `${appName}/${userId}/${sessionId}/${filename}/${version}`;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GcsArtifactService
+});

package/dist/cjs/common.js

@@ -51,7 +51,6 @@ __export(common_exports, {
   PluginManager: () => import_plugin_manager.PluginManager,
   PolicyOutcome: () => import_security_plugin.PolicyOutcome,
   REQUEST_CONFIRMATION_FUNCTION_CALL_NAME: () => import_security_plugin.REQUEST_CONFIRMATION_FUNCTION_CALL_NAME,
-  RunConfig: () => import_run_config.RunConfig,
   Runner: () => import_runner.Runner,
   SecurityPlugin: () => import_security_plugin.SecurityPlugin,
   SequentialAgent: () => import_sequential_agent.SequentialAgent,
@@ -59,6 +58,7 @@ __export(common_exports, {
   ToolConfirmation: () => import_tool_confirmation.ToolConfirmation,
   ToolContext: () => import_tool_context.ToolContext,
   createEvent: () => import_event.createEvent,
+  createEventActions: () => import_event_actions.createEventActions,
   createSession: () => import_session.createSession,
   functionsExportedForTestingOnly: () => import_functions.functionsExportedForTestingOnly,
   getAskUserConfirmationFunctionCalls: () => import_security_plugin.getAskUserConfirmationFunctionCalls,
@@ -83,6 +83,7 @@ var import_run_config = require("./agents/run_config.js");
 var import_sequential_agent = require("./agents/sequential_agent.js");
 var import_in_memory_artifact_service = require("./artifacts/in_memory_artifact_service.js");
 var import_event = require("./events/event.js");
+var import_event_actions = require("./events/event_actions.js");
 var import_in_memory_memory_service = require("./memory/in_memory_memory_service.js");
 var import_base_llm = require("./models/base_llm.js");
 var import_google_llm = require("./models/google_llm.js");
@@ -141,7 +142,6 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   PluginManager,
   PolicyOutcome,
   REQUEST_CONFIRMATION_FUNCTION_CALL_NAME,
-  RunConfig,
   Runner,
   SecurityPlugin,
   SequentialAgent,
@@ -149,6 +149,7 @@ __reExport(common_exports, require("./tools/base_tool.js"), module.exports);
   ToolConfirmation,
   ToolContext,
   createEvent,
+  createEventActions,
   createSession,
   functionsExportedForTestingOnly,
   getAskUserConfirmationFunctionCalls,