@paean-ai/adk 0.2.14 → 0.2.15

package/dist/cjs/runner/runner.js

@@ -1,277 +0,0 @@
-/**
-  * @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 runner_exports = {};
-__export(runner_exports, {
-  Runner: () => Runner
-});
-module.exports = __toCommonJS(runner_exports);
-var import_genai = require("@google/genai");
-var import_api = require("@opentelemetry/api");
-var import_invocation_context = require("../agents/invocation_context.js");
-var import_llm_agent = require("../agents/llm_agent.js");
-var import_run_config = require("../agents/run_config.js");
-var import_event = require("../events/event.js");
-var import_event_actions = require("../events/event_actions.js");
-var import_plugin_manager = require("../plugins/plugin_manager.js");
-var import_logger = require("../utils/logger.js");
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-class Runner {
-  constructor(input) {
-    var _a;
-    this.appName = input.appName;
-    this.agent = input.agent;
-    this.pluginManager = new import_plugin_manager.PluginManager((_a = input.plugins) != null ? _a : []);
-    this.artifactService = input.artifactService;
-    this.sessionService = input.sessionService;
-    this.memoryService = input.memoryService;
-    this.credentialService = input.credentialService;
-  }
-  /**
-   * Runs the agent with the given message, and returns an async generator of
-   * events.
-   *
-   * @param userId The user ID of the session.
-   * @param sessionId The session ID of the session.
-   * @param newMessage A new message to append to the session.
-   * @param stateDelta An optional state delta to apply to the session.
-   * @param runConfig The run config for the agent.
-   * @yields The events generated by the agent.
-   */
-  // TODO - b/425992518: user, sessionId, and runConfig can be internalized.
-  async *runAsync({
-    userId,
-    sessionId,
-    newMessage,
-    stateDelta,
-    runConfig
-  }) {
-    var _a;
-    runConfig = (0, import_run_config.createRunConfig)(runConfig);
-    const span = import_api.trace.getTracer("gcp.vertex.agent").startSpan("invocation");
-    try {
-      const session = await this.sessionService.getSession({ appName: this.appName, userId, sessionId });
-      if (!session) {
-        throw new Error(`Session not found: ${sessionId}`);
-      }
-      if (runConfig.supportCfc && this.agent instanceof import_llm_agent.LlmAgent) {
-        const modelName = this.agent.canonicalModel.model;
-        if (!modelName.startsWith("gemini-2")) {
-          throw new Error(`CFC is not supported for model: ${modelName} in agent: ${this.agent.name}`);
-        }
-      }
-      const invocationContext = new import_invocation_context.InvocationContext({
-        artifactService: this.artifactService,
-        sessionService: this.sessionService,
-        memoryService: this.memoryService,
-        credentialService: this.credentialService,
-        invocationId: (0, import_invocation_context.newInvocationContextId)(),
-        agent: this.agent,
-        session,
-        userContent: newMessage,
-        runConfig,
-        pluginManager: this.pluginManager
-      });
-      const pluginUserMessage = await this.pluginManager.runOnUserMessageCallback({
-        userMessage: newMessage,
-        invocationContext
-      });
-      if (pluginUserMessage) {
-        newMessage = pluginUserMessage;
-      }
-      if (newMessage) {
-        if (!((_a = newMessage.parts) == null ? void 0 : _a.length)) {
-          throw new Error("No parts in the newMessage.");
-        }
-        if (runConfig.saveInputBlobsAsArtifacts) {
-          await this.saveArtifacts(
-            invocationContext.invocationId,
-            session.userId,
-            session.id,
-            newMessage
-          );
-        }
-        await this.sessionService.appendEvent({
-          session,
-          event: (0, import_event.createEvent)({
-            invocationId: invocationContext.invocationId,
-            author: "user",
-            actions: stateDelta ? (0, import_event_actions.createEventActions)({ stateDelta }) : void 0,
-            content: newMessage
-          })
-        });
-      }
-      invocationContext.agent = this.determineAgentForResumption(session, this.agent);
-      const beforeRunCallbackResponse = await this.pluginManager.runBeforeRunCallback({ invocationContext });
-      if (beforeRunCallbackResponse) {
-        const earlyExitEvent = (0, import_event.createEvent)({
-          invocationId: invocationContext.invocationId,
-          author: "model",
-          content: beforeRunCallbackResponse
-        });
-        await this.sessionService.appendEvent({ session, event: earlyExitEvent });
-        yield earlyExitEvent;
-      } else {
-        for await (const event of invocationContext.agent.runAsync(
-          invocationContext
-        )) {
-          if (!event.partial) {
-            await this.sessionService.appendEvent({ session, event });
-          }
-          const modifiedEvent = await this.pluginManager.runOnEventCallback(
-            { invocationContext, event }
-          );
-          if (modifiedEvent) {
-            yield modifiedEvent;
-          } else {
-            yield event;
-          }
-        }
-      }
-      await this.pluginManager.runAfterRunCallback({ invocationContext });
-    } finally {
-      span.end();
-    }
-  }
-  /**
-   * Saves artifacts from the message parts and replaces the inline data with
-   * a file name placeholder.
-   *
-   * @param invocationId The current invocation ID.
-   * @param userId The user ID of the session.
-   * @param sessionId The session ID of the session.
-   * @param message The message containing parts to process.
-   */
-  async saveArtifacts(invocationId, userId, sessionId, message) {
-    var _a;
-    if (!this.artifactService || !((_a = message.parts) == null ? void 0 : _a.length)) {
-      return;
-    }
-    for (let i = 0; i < message.parts.length; i++) {
-      const part = message.parts[i];
-      if (!part.inlineData) {
-        continue;
-      }
-      const fileName = `artifact_${invocationId}_${i}`;
-      await this.artifactService.saveArtifact({
-        appName: this.appName,
-        userId,
-        sessionId,
-        filename: fileName,
-        artifact: part
-      });
-      message.parts[i] = (0, import_genai.createPartFromText)(
-        `Uploaded file: ${fileName}. It is saved into artifacts`
-      );
-    }
-  }
-  /**
-   * Determines the next agent to run to continue the session. This is primarily
-   * used for session resumption.
-   */
-  // TODO - b/425992518: This is where LRO integration should happen.
-  // Needs clean up before we can generalize it.
-  determineAgentForResumption(session, rootAgent) {
-    const event = findEventByLastFunctionResponseId(session.events);
-    if (event && event.author) {
-      return rootAgent.findAgent(event.author) || rootAgent;
-    }
-    for (let i = session.events.length - 1; i >= 0; i--) {
-      import_logger.logger.info("event: ", JSON.stringify(session.events[i]));
-      const event2 = session.events[i];
-      if (event2.author === "user" || !event2.author) {
-        continue;
-      }
-      if (event2.author === rootAgent.name) {
-        return rootAgent;
-      }
-      const agent = rootAgent.findSubAgent(event2.author);
-      if (!agent) {
-        import_logger.logger.warn(`Event from an unknown agent: ${event2.author}, event id: ${event2.id}`);
-        continue;
-      }
-      if (this.isRoutableLlmAgent(agent)) {
-        return agent;
-      }
-    }
-    return rootAgent;
-  }
-  /**
-   * Whether the agent to run can transfer to any other agent in the agent tree.
-   *
-   * An agent is transferable if:
-   *  - It is an instance of `LlmAgent`.
-   *  - All its ancestors are also transferable (i.e., they have
-   *    `disallowTransferToParent` set to false).
-   *
-   * @param agentToRun The agent to check for transferability.
-   * @returns True if the agent can transfer, False otherwise.
-   */
-  isRoutableLlmAgent(agentToRun) {
-    let agent = agentToRun;
-    while (agent) {
-      if (!(agent instanceof import_llm_agent.LlmAgent)) {
-        return false;
-      }
-      if (agent.disallowTransferToParent) {
-        return false;
-      }
-      agent = agent.parentAgent;
-    }
-    return true;
-  }
-  // TODO - b/425992518: Implement runLive and related methods.
-}
-function findEventByLastFunctionResponseId(events) {
-  var _a, _b, _c, _d;
-  if (!events.length) {
-    return null;
-  }
-  const lastEvent = events[events.length - 1];
-  const functionCallId = (_d = (_c = (_b = (_a = lastEvent.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.find((part) => part.functionResponse)) == null ? void 0 : _c.functionResponse) == null ? void 0 : _d.id;
-  if (!functionCallId) {
-    return null;
-  }
-  for (let i = events.length - 2; i >= 0; i--) {
-    const event = events[i];
-    const functionCalls = (0, import_event.getFunctionCalls)(event);
-    if (!functionCalls) {
-      continue;
-    }
-    for (const functionCall of functionCalls) {
-      if (functionCall.id === functionCallId) {
-        return event;
-      }
-    }
-  }
-  return null;
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  Runner
-});

package/dist/cjs/sessions/base_session_service.js

@@ -1,71 +0,0 @@
-/**
-  * @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 base_session_service_exports = {};
-__export(base_session_service_exports, {
-  BaseSessionService: () => BaseSessionService
-});
-module.exports = __toCommonJS(base_session_service_exports);
-var import_state = require("./state.js");
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-class BaseSessionService {
-  /**
-   * Appends an event to a session.
-   *
-   * @param request The request to append an event.
-   * @return A promise that resolves to the event that was appended.
-   */
-  async appendEvent({ session, event }) {
-    if (event.partial) {
-      return event;
-    }
-    this.updateSessionState({ session, event });
-    session.events.push(event);
-    return event;
-  }
-  /**
-   * Updates the session state based on the event.
-   *
-   * @param request The request to update the session state.
-   */
-  updateSessionState({ session, event }) {
-    if (!event.actions || !event.actions.stateDelta) {
-      return;
-    }
-    for (const [key, value] of Object.entries(event.actions.stateDelta)) {
-      if (key.startsWith(import_state.State.TEMP_PREFIX)) {
-        continue;
-      }
-      session.state[key] = value;
-    }
-  }
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  BaseSessionService
-});

package/dist/cjs/sessions/in_memory_session_service.js

@@ -1,184 +0,0 @@
-/**
-  * @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 in_memory_session_service_exports = {};
-__export(in_memory_session_service_exports, {
-  InMemorySessionService: () => InMemorySessionService
-});
-module.exports = __toCommonJS(in_memory_session_service_exports);
-var import_deep_clone = require("../utils/deep_clone.js");
-var import_env_aware_utils = require("../utils/env_aware_utils.js");
-var import_logger = require("../utils/logger.js");
-var import_base_session_service = require("./base_session_service.js");
-var import_session = require("./session.js");
-var import_state = require("./state.js");
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-class InMemorySessionService extends import_base_session_service.BaseSessionService {
-  constructor() {
-    super(...arguments);
-    /**
-     * A map from app name to a map from user ID to a map from session ID to
-     * session.
-     */
-    this.sessions = {};
-    /**
-     * A map from app name to a map from user ID to a map from key to the value.
-     */
-    this.userState = {};
-    /**
-     * A map from app name to a map from key to the value.
-     */
-    this.appState = {};
-  }
-  createSession({ appName, userId, state, sessionId }) {
-    const session = (0, import_session.createSession)({
-      id: sessionId || (0, import_env_aware_utils.randomUUID)(),
-      appName,
-      userId,
-      state,
-      events: [],
-      lastUpdateTime: Date.now()
-    });
-    if (!this.sessions[appName]) {
-      this.sessions[appName] = {};
-    }
-    if (!this.sessions[appName][userId]) {
-      this.sessions[appName][userId] = {};
-    }
-    this.sessions[appName][userId][session.id] = session;
-    return Promise.resolve(
-      this.mergeState(appName, userId, (0, import_deep_clone.deepClone)(session))
-    );
-  }
-  getSession({ appName, userId, sessionId, config }) {
-    if (!this.sessions[appName] || !this.sessions[appName][userId] || !this.sessions[appName][userId][sessionId]) {
-      return Promise.resolve(void 0);
-    }
-    const session = this.sessions[appName][userId][sessionId];
-    const copiedSession = (0, import_deep_clone.deepClone)(session);
-    if (config) {
-      if (config.numRecentEvents) {
-        copiedSession.events = copiedSession.events.slice(-config.numRecentEvents);
-      }
-      if (config.afterTimestamp) {
-        let i = copiedSession.events.length - 1;
-        while (i >= 0) {
-          if (copiedSession.events[i].timestamp < config.afterTimestamp) {
-            break;
-          }
-          i--;
-        }
-        if (i >= 0) {
-          copiedSession.events = copiedSession.events.slice(i + 1);
-        }
-      }
-    }
-    return Promise.resolve(this.mergeState(appName, userId, copiedSession));
-  }
-  listSessions({ appName, userId }) {
-    if (!this.sessions[appName] || !this.sessions[appName][userId]) {
-      return Promise.resolve({ sessions: [] });
-    }
-    const sessionsWithoutEvents = [];
-    for (const session of Object.values(this.sessions[appName][userId])) {
-      sessionsWithoutEvents.push((0, import_session.createSession)({
-        id: session.id,
-        appName: session.appName,
-        userId: session.userId,
-        state: {},
-        events: [],
-        lastUpdateTime: session.lastUpdateTime
-      }));
-    }
-    return Promise.resolve({ sessions: sessionsWithoutEvents });
-  }
-  async deleteSession({ appName, userId, sessionId }) {
-    const session = await this.getSession({ appName, userId, sessionId });
-    if (!session) {
-      return;
-    }
-    delete this.sessions[appName][userId][sessionId];
-  }
-  async appendEvent({ session, event }) {
-    await super.appendEvent({ session, event });
-    session.lastUpdateTime = event.timestamp;
-    const appName = session.appName;
-    const userId = session.userId;
-    const sessionId = session.id;
-    const warning = (message) => {
-      import_logger.logger.warn(`Failed to append event to session ${sessionId}: ${message}`);
-    };
-    if (!this.sessions[appName]) {
-      warning(`appName ${appName} not in sessions`);
-      return event;
-    }
-    if (!this.sessions[appName][userId]) {
-      warning(`userId ${userId} not in sessions[appName]`);
-      return event;
-    }
-    if (!this.sessions[appName][userId][sessionId]) {
-      warning(`sessionId ${sessionId} not in sessions[appName][userId]`);
-      return event;
-    }
-    if (event.actions && event.actions.stateDelta) {
-      for (const key of Object.keys(event.actions.stateDelta)) {
-        if (key.startsWith(import_state.State.APP_PREFIX)) {
-          this.appState[appName] = this.appState[appName] || {};
-          this.appState[appName][key.replace(import_state.State.APP_PREFIX, "")] = event.actions.stateDelta[key];
-        }
-        if (key.startsWith(import_state.State.USER_PREFIX)) {
-          this.userState[appName] = this.userState[appName] || {};
-          this.userState[appName][userId] = this.userState[appName][userId] || {};
-          this.userState[appName][userId][key.replace(import_state.State.USER_PREFIX, "")] = event.actions.stateDelta[key];
-        }
-      }
-    }
-    const storageSession = this.sessions[appName][userId][sessionId];
-    await super.appendEvent({ session: storageSession, event });
-    storageSession.lastUpdateTime = event.timestamp;
-    return event;
-  }
-  mergeState(appName, userId, copiedSession) {
-    if (this.appState[appName]) {
-      for (const key of Object.keys(this.appState[appName])) {
-        copiedSession.state[import_state.State.APP_PREFIX + key] = this.appState[appName][key];
-      }
-    }
-    if (!this.userState[appName] || !this.userState[appName][userId]) {
-      return copiedSession;
-    }
-    for (const key of Object.keys(this.userState[appName][userId])) {
-      copiedSession.state[import_state.State.USER_PREFIX + key] = this.userState[appName][userId][key];
-    }
-    return copiedSession;
-  }
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  InMemorySessionService
-});

package/dist/cjs/sessions/session.js

@@ -1,48 +0,0 @@
-/**
-  * @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 session_exports = {};
-__export(session_exports, {
-  createSession: () => createSession
-});
-module.exports = __toCommonJS(session_exports);
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-function createSession(params) {
-  return {
-    id: params.id,
-    appName: params.appName,
-    userId: params.userId || "",
-    state: params.state || {},
-    events: params.events || [],
-    lastUpdateTime: params.lastUpdateTime || 0
-  };
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  createSession
-});

package/dist/cjs/sessions/state.js

@@ -1,101 +0,0 @@
-/**
-  * @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 state_exports = {};
-__export(state_exports, {
-  State: () => State
-});
-module.exports = __toCommonJS(state_exports);
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-class State {
-  constructor(value = {}, delta = {}) {
-    this.value = value;
-    this.delta = delta;
-  }
-  /**
-   * Returns the value of the state dict for the given key.
-   *
-   * @param key The key to get the value for.
-   * @param defaultValue The default value to return if the key is not found.
-   * @return The value of the state for the given key, or the default value if
-   *     not found.
-   */
-  get(key, defaultValue) {
-    if (key in this.delta) {
-      return this.delta[key];
-    }
-    if (key in this.value) {
-      return this.value[key];
-    }
-    return defaultValue;
-  }
-  /**
-   * Sets the value of the state dict for the given key.
-   *
-   * @param key The key to set the value for.
-   * @param value The value to set.
-   */
-  set(key, value) {
-    this.value[key] = value;
-    this.delta[key] = value;
-  }
-  /**
-   * Whether the state has pending delta.
-   */
-  has(key) {
-    return key in this.value || key in this.delta;
-  }
-  /**
-   * Whether the state has pending delta.
-   */
-  hasDelta() {
-    return Object.keys(this.delta).length > 0;
-  }
-  /**
-   * Updates the state dict with the given delta.
-   *
-   * @param delta The delta to update the state with.
-   */
-  update(delta) {
-    this.delta = { ...this.delta, ...delta };
-    this.value = { ...this.value, ...delta };
-  }
-  /**
-   * Returns the state as a plain JSON object.
-   */
-  toRecord() {
-    return { ...this.value, ...this.delta };
-  }
-}
-State.APP_PREFIX = "app:";
-State.USER_PREFIX = "user:";
-State.TEMP_PREFIX = "temp:";
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  State
-});