@google/adk 0.1.2 → 0.2.0

package/dist/esm/agents/functions.js

@@ -3,6 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { createUserContent } from "@google/genai";
 import { createEvent, getFunctionCalls } from "../events/event.js";
 import { mergeEventActions } from "../events/event_actions.js";
 import { ToolContext } from "../tools/tool_context.js";
@@ -126,30 +127,6 @@ async function callToolAsync(tool, args, toolContext) {
   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 createEvent({
-    invocationId: invocationContext.invocationId,
-    author: invocationContext.agent.name,
-    content,
-    actions: toolContext.actions,
-    branch: invocationContext.branch
-  });
-}
 async function handleFunctionCallsAsync({
   invocationContext,
   functionCallEvent,
@@ -200,6 +177,7 @@ async function handleFunctionCallList({
     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,
@@ -236,10 +214,11 @@ async function handleFunctionCallList({
           );
           if (onToolErrorResponse) {
             functionResponse = onToolErrorResponse;
+          } else {
+            functionResponseError = e.message;
           }
         } else {
-          logger.error("Unknown error on tool execution type", e);
-          throw e;
+          functionResponseError = e;
         }
       }
     }
@@ -268,12 +247,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 = createEvent({
+      invocationId: invocationContext.invocationId,
+      author: invocationContext.agent.name,
+      content: createUserContent({
+        functionResponse: {
+          id: toolContext.functionCallId,
+          name: tool.name,
+          response: functionResponse
+        }
+      }),
+      actions: toolContext.actions,
+      branch: invocationContext.branch
+    });
     logger.debug("traceToolCall", {
       tool: tool.name,
       args: functionArgs,

package/dist/esm/agents/llm_agent.js

@@ -140,6 +140,7 @@ class AgentTransferLlmRequestProcessor extends BaseLlmRequestProcessor {
           throw new Error("toolContext is required.");
         }
         toolContext.actions.transferToAgent = args.agentName;
+        return "Transfer queued";
       }
     });
   }

package/dist/esm/artifacts/gcs_artifact_service.js

@@ -0,0 +1,110 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Storage } from "@google-cloud/storage";
+import { createPartFromBase64, createPartFromText } from "@google/genai";
+class GcsArtifactService {
+  constructor(bucket) {
+    this.bucket = new 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 createPartFromText(rawDataBuffer.toString("utf-8"));
+    }
+    return 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}`;
+}
+export {
+  GcsArtifactService
+};

package/dist/esm/common.js

@@ -15,9 +15,11 @@ import { 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, 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 { Gemini } from "./models/google_llm.js";
+import { LLMRegistry } from "./models/registry.js";
 import { BasePlugin } from "./plugins/base_plugin.js";
 import { LoggingPlugin } from "./plugins/logging_plugin.js";
 import { PluginManager } from "./plugins/plugin_manager.js";
@@ -26,6 +28,7 @@ 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 { State } from "./sessions/state.js";
 import { AgentTool } from "./tools/agent_tool.js";
 import { BaseTool } from "./tools/base_tool.js";
 import { BaseToolset } from "./tools/base_toolset.js";
@@ -36,6 +39,7 @@ 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";
+import { version } from "./version.js";
 export * from "./artifacts/base_artifact_service.js";
 export * from "./memory/base_memory_service.js";
 export * from "./sessions/base_session_service.js";
@@ -57,6 +61,7 @@ export {
   InMemoryRunner,
   InMemorySessionService,
   InvocationContext,
+  LLMRegistry,
   LiveRequestQueue,
   LlmAgent,
   LogLevel,
@@ -70,10 +75,12 @@ export {
   Runner,
   SecurityPlugin,
   SequentialAgent,
+  State,
   StreamingMode,
   ToolConfirmation,
   ToolContext,
   createEvent,
+  createEventActions,
   createSession,
   functionsExportedForTestingOnly,
   getAskUserConfirmationFunctionCalls,
@@ -83,5 +90,6 @@ export {
   isFinalResponse,
   setLogLevel,
   stringifyContent,
+  version,
   zodObjectToSchema
 };