@paean-ai/adk 0.2.19 → 0.2.21

package/dist/cjs/utils/simple_zod_to_json.js

@@ -0,0 +1,191 @@
+/**
+  * @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 simple_zod_to_json_exports = {};
+__export(simple_zod_to_json_exports, {
+  isZodObject: () => isZodObject,
+  zodObjectToSchema: () => zodObjectToSchema
+});
+module.exports = __toCommonJS(simple_zod_to_json_exports);
+var import_genai = require("@google/genai");
+var import_zod = require("zod");
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function isZodObject(obj) {
+  var _a;
+  return obj !== null && typeof obj === "object" && ((_a = obj._def) == null ? void 0 : _a.typeName) === "ZodObject";
+}
+function parseZodType(zodType) {
+  const def = zodType._def;
+  if (!def) {
+    return {};
+  }
+  const description = def.description;
+  const result = {};
+  if (description) result.description = description;
+  const returnResult = (result2) => {
+    if (result2.description === void 0) {
+      delete result2.description;
+    }
+    return result2;
+  };
+  switch (def.typeName) {
+    case import_zod.z.ZodFirstPartyTypeKind.ZodString:
+      result.type = import_genai.Type.STRING;
+      for (const check of def.checks || []) {
+        if (check.kind === "min")
+          result.minLength = check.value.toString();
+        else if (check.kind === "max")
+          result.maxLength = check.value.toString();
+        else if (check.kind === "email")
+          result.format = "email";
+        else if (check.kind === "uuid")
+          result.format = "uuid";
+        else if (check.kind === "url")
+          result.format = "uri";
+        else if (check.kind === "regex")
+          result.pattern = check.regex.source;
+      }
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodNumber:
+      result.type = import_genai.Type.NUMBER;
+      for (const check of def.checks || []) {
+        if (check.kind === "min")
+          result.minimum = check.value;
+        else if (check.kind === "max")
+          result.maximum = check.value;
+        else if (check.kind === "int")
+          result.type = import_genai.Type.INTEGER;
+      }
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodBoolean:
+      result.type = import_genai.Type.BOOLEAN;
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodArray:
+      result.type = import_genai.Type.ARRAY;
+      result.items = parseZodType(def.type);
+      if (def.minLength) result.minItems = def.minLength.value.toString();
+      if (def.maxLength) result.maxItems = def.maxLength.value.toString();
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodObject: {
+      const nestedSchema = zodObjectToSchema(zodType);
+      return nestedSchema;
+    }
+    case import_zod.z.ZodFirstPartyTypeKind.ZodLiteral:
+      const literalType = typeof def.value;
+      result.enum = [def.value.toString()];
+      if (literalType === "string") {
+        result.type = import_genai.Type.STRING;
+      } else if (literalType === "number") {
+        result.type = import_genai.Type.NUMBER;
+      } else if (literalType === "boolean") {
+        result.type = import_genai.Type.BOOLEAN;
+      } else if (def.value === null) {
+        result.type = import_genai.Type.NULL;
+      } else {
+        throw new Error(`Unsupported ZodLiteral value type: ${literalType}`);
+      }
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodEnum:
+      result.type = import_genai.Type.STRING;
+      result.enum = def.values;
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodNativeEnum:
+      result.type = import_genai.Type.STRING;
+      result.enum = Object.values(def.values);
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodUnion:
+      result.anyOf = def.options.map(parseZodType);
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodOptional:
+      return parseZodType(def.innerType);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodNullable:
+      const nullableInner = parseZodType(def.innerType);
+      return nullableInner ? returnResult({
+        anyOf: [nullableInner, { type: import_genai.Type.NULL }],
+        ...description && { description }
+      }) : returnResult({ type: import_genai.Type.NULL, ...description && { description } });
+    case import_zod.z.ZodFirstPartyTypeKind.ZodDefault:
+      const defaultInner = parseZodType(def.innerType);
+      if (defaultInner) defaultInner.default = def.defaultValue();
+      return defaultInner;
+    case import_zod.z.ZodFirstPartyTypeKind.ZodBranded:
+      return parseZodType(def.type);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodReadonly:
+      return parseZodType(def.innerType);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodNull:
+      result.type = import_genai.Type.NULL;
+      return returnResult(result);
+    case import_zod.z.ZodFirstPartyTypeKind.ZodAny:
+    case import_zod.z.ZodFirstPartyTypeKind.ZodUnknown:
+      return returnResult({ ...description && { description } });
+    default:
+      throw new Error(`Unsupported Zod type: ${def.typeName}`);
+  }
+}
+function zodObjectToSchema(schema) {
+  if (schema._def.typeName !== import_zod.z.ZodFirstPartyTypeKind.ZodObject) {
+    throw new Error("Expected a ZodObject");
+  }
+  const shape = schema.shape;
+  const properties = {};
+  const required = [];
+  for (const key in shape) {
+    const fieldSchema = shape[key];
+    const parsedField = parseZodType(fieldSchema);
+    if (parsedField) {
+      properties[key] = parsedField;
+    }
+    let currentSchema = fieldSchema;
+    let isOptional = false;
+    while (currentSchema._def.typeName === import_zod.z.ZodFirstPartyTypeKind.ZodOptional || currentSchema._def.typeName === import_zod.z.ZodFirstPartyTypeKind.ZodDefault) {
+      isOptional = true;
+      currentSchema = currentSchema._def.innerType;
+    }
+    if (!isOptional) {
+      required.push(key);
+    }
+  }
+  const catchall = schema._def.catchall;
+  let additionalProperties = false;
+  if (catchall && catchall._def.typeName !== import_zod.z.ZodFirstPartyTypeKind.ZodNever) {
+    additionalProperties = parseZodType(catchall) || true;
+  } else {
+    additionalProperties = schema._def.unknownKeys === "passthrough";
+  }
+  return {
+    type: import_genai.Type.OBJECT,
+    properties,
+    required: required.length > 0 ? required : [],
+    ...schema._def.description ? { description: schema._def.description } : {}
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  isZodObject,
+  zodObjectToSchema
+});

package/dist/cjs/utils/variant_utils.js

@@ -0,0 +1,55 @@
+/**
+  * @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 variant_utils_exports = {};
+__export(variant_utils_exports, {
+  GoogleLLMVariant: () => GoogleLLMVariant,
+  getGoogleLlmVariant: () => getGoogleLlmVariant
+});
+module.exports = __toCommonJS(variant_utils_exports);
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+var GoogleLLMVariant = /* @__PURE__ */ ((GoogleLLMVariant2) => {
+  GoogleLLMVariant2["VERTEX_AI"] = "VERTEX_AI";
+  GoogleLLMVariant2["GEMINI_API"] = "GEMINI_API";
+  return GoogleLLMVariant2;
+})(GoogleLLMVariant || {});
+function getGoogleLlmVariant() {
+  return getBooleanEnvVar("GOOGLE_GENAI_USE_VERTEXAI") ? "VERTEX_AI" /* VERTEX_AI */ : "GEMINI_API" /* GEMINI_API */;
+}
+function getBooleanEnvVar(envVar) {
+  if (!process.env) {
+    return false;
+  }
+  const envVarValue = (process.env[envVar] || "").toLowerCase();
+  return ["true", "1"].includes(envVar.toLowerCase());
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GoogleLLMVariant,
+  getGoogleLlmVariant
+});

package/dist/cjs/version.js

@@ -0,0 +1,39 @@
+/**
+  * @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 version_exports = {};
+__export(version_exports, {
+  version: () => version
+});
+module.exports = __toCommonJS(version_exports);
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const version = "0.2.4";
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  version
+});

package/dist/esm/agents/active_streaming_tool.js

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class ActiveStreamingTool {
+  constructor(params = {}) {
+    this.task = params.task;
+    this.stream = params.stream;
+  }
+}
+export {
+  ActiveStreamingTool
+};

package/dist/esm/agents/base_agent.js

@@ -0,0 +1,214 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { trace } from "@opentelemetry/api";
+import { createEvent } from "../events/event.js";
+import { CallbackContext } from "./callback_context.js";
+import { InvocationContext } from "./invocation_context.js";
+class BaseAgent {
+  constructor(config) {
+    this.name = validateAgentName(config.name);
+    this.description = config.description;
+    this.parentAgent = config.parentAgent;
+    this.subAgents = config.subAgents || [];
+    this.rootAgent = getRootAgent(this);
+    this.beforeAgentCallback = getCannonicalCallback(config.beforeAgentCallback);
+    this.afterAgentCallback = getCannonicalCallback(config.afterAgentCallback);
+    this.setParentAgentForSubAgents();
+  }
+  /**
+   * Entry method to run an agent via text-based conversation.
+   *
+   * @param parentContext The invocation context of the parent agent.
+   * @yields The events generated by the agent.
+   * @returns An AsyncGenerator that yields the events generated by the agent.
+   */
+  async *runAsync(parentContext) {
+    const span = trace.getTracer("gcp.vertex.agent").startSpan(`agent_run [${this.name}]`);
+    try {
+      const context = this.createInvocationContext(parentContext);
+      const beforeAgentCallbackEvent = await this.handleBeforeAgentCallback(context);
+      if (beforeAgentCallbackEvent) {
+        yield beforeAgentCallbackEvent;
+      }
+      if (context.endInvocation) {
+        return;
+      }
+      for await (const event of this.runAsyncImpl(context)) {
+        yield event;
+      }
+      if (context.endInvocation) {
+        return;
+      }
+      const afterAgentCallbackEvent = await this.handleAfterAgentCallback(context);
+      if (afterAgentCallbackEvent) {
+        yield afterAgentCallbackEvent;
+      }
+    } finally {
+      span.end();
+    }
+  }
+  /**
+   * Entry method to run an agent via video/audio-based conversation.
+   *
+   * @param parentContext The invocation context of the parent agent.
+   * @yields The events generated by the agent.
+   * @returns An AsyncGenerator that yields the events generated by the agent.
+   */
+  async *runLive(parentContext) {
+    const span = trace.getTracer("gcp.vertex.agent").startSpan(`agent_run [${this.name}]`);
+    try {
+      throw new Error("Live mode is not implemented yet.");
+    } finally {
+      span.end();
+    }
+  }
+  /**
+   * Finds the agent with the given name in this agent and its descendants.
+   *
+   * @param name The name of the agent to find.
+   * @return The agent with the given name, or undefined if not found.
+   */
+  findAgent(name) {
+    if (this.name === name) {
+      return this;
+    }
+    return this.findSubAgent(name);
+  }
+  /**
+   * Finds the agent with the given name in this agent's descendants.
+   *
+   * @param name The name of the agent to find.
+   * @return The agent with the given name, or undefined if not found.
+   */
+  findSubAgent(name) {
+    for (const subAgent of this.subAgents) {
+      const result = subAgent.findAgent(name);
+      if (result) {
+        return result;
+      }
+    }
+    return void 0;
+  }
+  /**
+   * Creates an invocation context for this agent.
+   *
+   * @param parentContext The invocation context of the parent agent.
+   * @return The invocation context for this agent.
+   */
+  createInvocationContext(parentContext) {
+    return new InvocationContext({
+      ...parentContext,
+      agent: this
+    });
+  }
+  /**
+   * Runs the before agent callback if it exists.
+   *
+   * @param invocationContext The invocation context of the agent.
+   * @return The event to return to the user, or undefined if no event is
+   *     generated.
+   */
+  async handleBeforeAgentCallback(invocationContext) {
+    if (this.beforeAgentCallback.length === 0) {
+      return void 0;
+    }
+    const callbackContext = new CallbackContext({ invocationContext });
+    for (const callback of this.beforeAgentCallback) {
+      const content = await callback(callbackContext);
+      if (content) {
+        invocationContext.endInvocation = true;
+        return createEvent({
+          invocationId: invocationContext.invocationId,
+          author: this.name,
+          branch: invocationContext.branch,
+          content,
+          actions: callbackContext.eventActions
+        });
+      }
+    }
+    if (callbackContext.state.hasDelta()) {
+      return createEvent({
+        invocationId: invocationContext.invocationId,
+        author: this.name,
+        branch: invocationContext.branch,
+        actions: callbackContext.eventActions
+      });
+    }
+    return void 0;
+  }
+  /**
+   * Runs the after agent callback if it exists.
+   *
+   * @param invocationContext The invocation context of the agent.
+   * @return The event to return to the user, or undefined if no event is
+   *     generated.
+   */
+  async handleAfterAgentCallback(invocationContext) {
+    if (this.afterAgentCallback.length === 0) {
+      return void 0;
+    }
+    const callbackContext = new CallbackContext({ invocationContext });
+    for (const callback of this.afterAgentCallback) {
+      const content = await callback(callbackContext);
+      if (content) {
+        return createEvent({
+          invocationId: invocationContext.invocationId,
+          author: this.name,
+          branch: invocationContext.branch,
+          content,
+          actions: callbackContext.eventActions
+        });
+      }
+    }
+    if (callbackContext.state.hasDelta()) {
+      return createEvent({
+        invocationId: invocationContext.invocationId,
+        author: this.name,
+        branch: invocationContext.branch,
+        actions: callbackContext.eventActions
+      });
+    }
+    return void 0;
+  }
+  setParentAgentForSubAgents() {
+    for (const subAgent of this.subAgents) {
+      if (subAgent.parentAgent) {
+        throw new Error(`Agent "${subAgent.name}" already has a parent agent, current parent: "${subAgent.parentAgent.name}", trying to add: "${this.name}"`);
+      }
+      subAgent.parentAgent = this;
+    }
+  }
+}
+function validateAgentName(name) {
+  if (!isIdentifier(name)) {
+    throw new Error(`Found invalid agent name: "${name}". Agent name must be a valid identifier. It should start with a letter (a-z, A-Z) or an underscore (_), and can only contain letters, digits (0-9), and underscores.`);
+  }
+  if (name === "user") {
+    throw new Error(
+      `Agent name cannot be 'user'. 'user' is reserved for end-user's input.`
+    );
+  }
+  return name;
+}
+function isIdentifier(str) {
+  return /^[\p{ID_Start}$_][\p{ID_Continue}$_]*$/u.test(str);
+}
+function getRootAgent(rootAgent) {
+  while (rootAgent.parentAgent) {
+    rootAgent = rootAgent.parentAgent;
+  }
+  return rootAgent;
+}
+function getCannonicalCallback(callbacks) {
+  if (!callbacks) {
+    return [];
+  }
+  return Array.isArray(callbacks) ? callbacks : [callbacks];
+}
+export {
+  BaseAgent,
+  getCannonicalCallback
+};

package/dist/esm/agents/base_llm_processor.js

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

package/dist/esm/agents/callback_context.js

@@ -0,0 +1,68 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { createEventActions } from "../events/event_actions.js";
+import { State } from "../sessions/state.js";
+import { ReadonlyContext } from "./readonly_context.js";
+class CallbackContext extends ReadonlyContext {
+  constructor({ invocationContext, eventActions }) {
+    super(invocationContext);
+    this.eventActions = eventActions || createEventActions();
+    this._state = new State(
+      invocationContext.session.state,
+      this.eventActions.stateDelta
+    );
+  }
+  /**
+   * The delta-aware state of the current session.
+   */
+  get state() {
+    return this._state;
+  }
+  /**
+   * Loads an artifact attached to the current session.
+   *
+   * @param filename The filename of the artifact.
+   * @param version The version of the artifact. If not provided, the latest
+   *     version will be used.
+   * @return A promise that resolves to the loaded artifact.
+   */
+  loadArtifact(filename, version) {
+    if (!this.invocationContext.artifactService) {
+      throw new Error("Artifact service is not initialized.");
+    }
+    return this.invocationContext.artifactService.loadArtifact({
+      appName: this.invocationContext.appName,
+      userId: this.invocationContext.userId,
+      sessionId: this.invocationContext.session.id,
+      filename,
+      version
+    });
+  }
+  /**
+   * Saves an artifact attached to the current session.
+   *
+   * @param filename The filename of the artifact.
+   * @param artifact The artifact to save.
+   * @return A promise that resolves to the version of the saved artifact.
+   */
+  async saveArtifact(filename, artifact) {
+    if (!this.invocationContext.artifactService) {
+      throw new Error("Artifact service is not initialized.");
+    }
+    const version = await this.invocationContext.artifactService.saveArtifact({
+      appName: this.invocationContext.appName,
+      userId: this.invocationContext.userId,
+      sessionId: this.invocationContext.session.id,
+      filename,
+      artifact
+    });
+    this.eventActions.artifactDelta[filename] = version;
+    return version;
+  }
+}
+export {
+  CallbackContext
+};