@google/adk 0.1.2 → 0.1.3

package/dist/esm/sessions/database_session_service.js

@@ -0,0 +1,22 @@
+import { BaseSessionService } from "./base_session_service.js";
+class DatabaseSessionService extends BaseSessionService {
+  constructor(dbUrl) {
+    super();
+    this.dbUrl = dbUrl;
+  }
+  createSession(request) {
+    throw new Error("Method not implemented.");
+  }
+  getSession(request) {
+    throw new Error("Method not implemented.");
+  }
+  listSessions(request) {
+    throw new Error("Method not implemented.");
+  }
+  deleteSession(request) {
+    throw new Error("Method not implemented.");
+  }
+}
+export {
+  DatabaseSessionService
+};

package/dist/types/artifacts/gcs_artifact_service.d.ts

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Part } from '@google/genai';
+import { BaseArtifactService, DeleteArtifactRequest, ListArtifactKeysRequest, ListVersionsRequest, LoadArtifactRequest, SaveArtifactRequest } from './base_artifact_service.js';
+export declare class GcsArtifactService implements BaseArtifactService {
+    private readonly bucket;
+    constructor(bucket: string);
+    saveArtifact(request: SaveArtifactRequest): Promise<number>;
+    loadArtifact(request: LoadArtifactRequest): Promise<Part | undefined>;
+    listArtifactKeys(request: ListArtifactKeysRequest): Promise<string[]>;
+    deleteArtifact(request: DeleteArtifactRequest): Promise<void>;
+    listVersions(request: ListVersionsRequest): Promise<number[]>;
+}

package/dist/types/common.d.ts

@@ -21,6 +21,7 @@ export type { BaseCredentialService } from './auth/credential_service/base_crede
 export { createEvent, getFunctionCalls, getFunctionResponses, hasTrailingCodeExecutionResult, isFinalResponse, stringifyContent } from './events/event.js';
 export type { Event } from './events/event.js';
 export type { EventActions } from './events/event_actions.js';
+export { createEventActions } from './events/event_actions.js';
 export { InMemoryMemoryService } from './memory/in_memory_memory_service.js';
 export { BaseLlm } from './models/base_llm.js';
 export type { BaseLlmConnection } from './models/base_llm_connection.js';
@@ -28,6 +29,7 @@ export { Gemini } from './models/google_llm.js';
 export type { GeminiParams } from './models/google_llm.js';
 export type { LlmRequest } from './models/llm_request.js';
 export type { LlmResponse } from './models/llm_response.js';
+export { LLMRegistry } from './models/registry.js';
 export { BasePlugin } from './plugins/base_plugin.js';
 export { LoggingPlugin } from './plugins/logging_plugin.js';
 export { PluginManager } from './plugins/plugin_manager.js';

package/dist/types/index.d.ts

@@ -7,3 +7,4 @@ export * from './common.js';
 export * from './tools/mcp/mcp_session_manager.js';
 export * from './tools/mcp/mcp_tool.js';
 export * from './tools/mcp/mcp_toolset.js';
+export * from './artifacts/gcs_artifact_service.js';

package/dist/types/models/base_llm.d.ts

@@ -13,11 +13,13 @@ export declare abstract class BaseLlm {
     readonly model: string;
     /**
      * Creates an instance of BaseLLM.
-     *
-     * @param model The name of the LLM, e.g. gemini-1.5-flash or
+     * @param params The parameters for creating a BaseLlm instance.
+     * @param params.model The name of the LLM, e.g. gemini-1.5-flash or
      *     gemini-1.5-flash-001.
      */
-    constructor(model: string);
+    constructor({ model }: {
+        model: string;
+    });
     /**
      * List of supported models in regex for LlmRegistry.
      */

package/dist/types/models/google_llm.d.ts

@@ -52,7 +52,7 @@ export declare class Gemini extends BaseLlm {
     /**
      * @param params The parameters for creating a Gemini instance.
      */
-    constructor({ model, apiKey, vertexai, project, location, headers }?: GeminiParams);
+    constructor({ model, apiKey, vertexai, project, location, headers, }: GeminiParams);
     /**
      * A list of model name patterns that are supported by this LLM.
      *

package/dist/types/models/registry.d.ts

@@ -8,7 +8,9 @@ import { BaseLlm } from './base_llm.js';
  * type[BaseLlm] equivalent in TypeScript, represents a class that can be new-ed
  * to create a BaseLlm instance.
  */
-export type BaseLlmType = (new (model: string) => BaseLlm) & {
+export type BaseLlmType = (new (params: {
+    model: string;
+}) => BaseLlm) & {
     readonly supportedModels: Array<string | RegExp>;
 };
 /**
@@ -32,7 +34,9 @@ export declare class LLMRegistry {
      * Registers a new LLM class.
      * @param llmCls The class that implements the model.
      */
-    static register<T extends BaseLlm>(llmCls: (new (model: string) => T) & {
+    static register<T extends BaseLlm>(llmCls: (new (params: {
+        model: string;
+    }) => T) & {
         readonly supportedModels: Array<string | RegExp>;
     }): void;
     /**

package/dist/types/sessions/database_session_service.d.ts

@@ -0,0 +1,10 @@
+import { BaseSessionService, CreateSessionRequest, DeleteSessionRequest, GetSessionRequest, ListSessionsRequest, ListSessionsResponse } from './base_session_service.js';
+import { Session } from './session.js';
+export declare class DatabaseSessionService extends BaseSessionService {
+    private readonly dbUrl;
+    constructor(dbUrl: string);
+    createSession(request: CreateSessionRequest): Promise<Session>;
+    getSession(request: GetSessionRequest): Promise<Session | undefined>;
+    listSessions(request: ListSessionsRequest): Promise<ListSessionsResponse>;
+    deleteSession(request: DeleteSessionRequest): Promise<void>;
+}

package/dist/web/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 ".concat(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 ".concat(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/web/artifacts/gcs_artifact_service.js

@@ -0,0 +1,126 @@
+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 { 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(__spreadProps(__spreadValues({}, 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(__spreadProps(__spreadValues({}, 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 = "".concat(request.appName, "/").concat(request.userId, "/").concat(request.sessionId, "/");
+    const usernamePrefix = "".concat(request.appName, "/").concat(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(__spreadProps(__spreadValues({}, 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 "".concat(appName, "/").concat(userId, "/user/").concat(filename, "/").concat(version);
+  }
+  return "".concat(appName, "/").concat(userId, "/").concat(sessionId, "/").concat(filename, "/").concat(version);
+}
+export {
+  GcsArtifactService
+};

package/dist/web/common.js

@@ -15,6 +15,7 @@ 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";
@@ -74,6 +75,7 @@ export {
   ToolConfirmation,
   ToolContext,
   createEvent,
+  createEventActions,
   createSession,
   functionsExportedForTestingOnly,
   getAskUserConfirmationFunctionCalls,