@iqai/adk 0.0.3 → 0.0.5

package/dist/index.mjs

@@ -23,20 +23,39 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var BaseTool;
 var init_base_tool = __esm({
   "src/tools/base/base-tool.ts"() {
-    "use strict";
     BaseTool = class {
+      /**
+       * Name of the tool
+       */
+      name;
+      /**
+       * Description of the tool
+       */
+      description;
+      /**
+       * Whether the tool is a long running operation
+       */
+      isLongRunning;
+      /**
+       * Whether the tool execution should be retried on failure
+       */
+      shouldRetryOnFailure;
+      /**
+       * Maximum retry attempts
+       */
+      maxRetryAttempts;
+      /**
+       * Base delay for retry in ms (will be used with exponential backoff)
+       */
+      baseRetryDelay = 1e3;
+      /**
+       * Maximum delay for retry in ms
+       */
+      maxRetryDelay = 1e4;
       /**
        * Constructor for BaseTool
        */
       constructor(config) {
-        /**
-         * Base delay for retry in ms (will be used with exponential backoff)
-         */
-        this.baseRetryDelay = 1e3;
-        /**
-         * Maximum delay for retry in ms
-         */
-        this.maxRetryDelay = 1e4;
         this.name = config.name;
         this.description = config.description;
         this.isLongRunning = config.isLongRunning || false;
@@ -231,7 +250,6 @@ function extractJSDocParams(funcStr) {
 }
 var init_function_utils = __esm({
   "src/tools/function/function-utils.ts"() {
-    "use strict";
   }
 });
 
@@ -243,10 +261,11 @@ __export(function_tool_exports, {
 var FunctionTool;
 var init_function_tool = __esm({
   "src/tools/function/function-tool.ts"() {
-    "use strict";
     init_base_tool();
     init_function_utils();
     FunctionTool = class extends BaseTool {
+      func;
+      mandatoryArgs = [];
       /**
        * Creates a new FunctionTool wrapping the provided function.
        *
@@ -263,7 +282,6 @@ var init_function_tool = __esm({
           shouldRetryOnFailure: options?.shouldRetryOnFailure || false,
           maxRetryAttempts: options?.maxRetryAttempts || 3
         });
-        this.mandatoryArgs = [];
         this.func = func;
         this.mandatoryArgs = this.getMandatoryArgs(func);
       }
@@ -359,6 +377,25 @@ __export(agents_exports, {
 
 // src/agents/base-agent.ts
 var BaseAgent = class {
+  /**
+   * The agent's name
+   * Agent name must be a unique identifier within the agent tree
+   */
+  name;
+  /**
+   * Description about the agent's capability
+   * The LLM uses this to determine whether to delegate control to the agent
+   */
+  description;
+  /**
+   * The parent agent of this agent
+   * Note that an agent can ONLY be added as sub-agent once
+   */
+  parentAgent;
+  /**
+   * The sub-agents of this agent
+   */
+  subAgents;
   /**
    * Constructs a new BaseAgent
    */
@@ -423,7 +460,11 @@ var BaseAgent = class {
 };
 
 // src/models/llm-registry.ts
-var _LLMRegistry = class _LLMRegistry {
+var LLMRegistry = class _LLMRegistry {
+  /**
+   * Map of model name regex to LLM class
+   */
+  static llmRegistry = /* @__PURE__ */ new Map();
   /**
    * Creates a new LLM instance
    *
@@ -483,14 +524,17 @@ var _LLMRegistry = class _LLMRegistry {
     }
   }
 };
-/**
- * Map of model name regex to LLM class
- */
-_LLMRegistry.llmRegistry = /* @__PURE__ */ new Map();
-var LLMRegistry = _LLMRegistry;
 
 // src/models/llm-request.ts
 var LLMRequest = class {
+  /**
+   * The conversation history
+   */
+  messages;
+  /**
+   * LLM configuration parameters
+   */
+  config;
   constructor(data) {
     this.messages = data.messages;
     this.config = data.config || {};
@@ -499,18 +543,34 @@ var LLMRequest = class {
 
 // src/tools/tool-context.ts
 var ToolContext = class {
+  /**
+   * The parent invocation context
+   */
+  invocationContext;
+  /**
+   * Authentication handler for the tool
+   */
+  auth;
+  /**
+   * Additional parameters for the tool
+   */
+  parameters;
+  /**
+   * Tool name
+   */
+  toolName = "";
+  /**
+   * Tool ID
+   */
+  toolId = "";
+  /**
+   * Variables stored in the context
+   */
+  _variables;
   /**
    * Constructor for ToolContext
    */
   constructor(options) {
-    /**
-     * Tool name
-     */
-    this.toolName = "";
-    /**
-     * Tool ID
-     */
-    this.toolId = "";
     this.invocationContext = options.invocationContext;
     this.auth = options.auth;
     this.parameters = options.parameters || {};
@@ -586,6 +646,36 @@ var StreamingMode = /* @__PURE__ */ ((StreamingMode2) => {
   return StreamingMode2;
 })(StreamingMode || {});
 var RunConfig = class {
+  /**
+   * Speech configuration for the live agent
+   */
+  speechConfig;
+  /**
+   * The output modalities
+   */
+  responseModalities;
+  /**
+   * Whether to save input blobs as artifacts
+   */
+  saveInputBlobsAsArtifacts;
+  /**
+   * Whether to support Compositional Function Calling
+   * Only applicable for StreamingMode.SSE
+   */
+  supportCFC;
+  /**
+   * Streaming mode
+   */
+  streamingMode;
+  /**
+   * Whether to stream the response
+   * This is derived from streamingMode and used by LLM implementations
+   */
+  stream;
+  /**
+   * Output audio transcription configuration
+   */
+  outputAudioTranscription;
   constructor(config) {
     this.speechConfig = config?.speechConfig;
     this.responseModalities = config?.responseModalities;
@@ -599,14 +689,50 @@ var RunConfig = class {
 
 // src/agents/invocation-context.ts
 var InvocationContext = class _InvocationContext {
+  /**
+   * Unique session ID for the current conversation
+   */
+  sessionId;
+  /**
+   * Current conversation history
+   */
+  messages;
+  /**
+   * Run configuration
+   */
+  config;
+  /**
+   * User identifier associated with the session
+   */
+  userId;
+  /**
+   * Application name (for multi-app environments)
+   */
+  appName;
+  /**
+   * Memory service for long-term storage
+   */
+  memoryService;
+  /**
+   * Session service for session management
+   */
+  sessionService;
+  /**
+   * Additional context metadata
+   */
+  metadata;
+  /**
+   * Variables stored in the context
+   */
+  variables;
+  /**
+   * In-memory storage for node execution results
+   */
+  memory = /* @__PURE__ */ new Map();
   /**
    * Constructor for InvocationContext
    */
   constructor(options = {}) {
-    /**
-     * In-memory storage for node execution results
-     */
-    this.memory = /* @__PURE__ */ new Map();
     this.sessionId = options.sessionId || this.generateSessionId();
     this.messages = options.messages || [];
     this.config = options.config || new RunConfig();
@@ -713,6 +839,54 @@ var InvocationContext = class _InvocationContext {
 
 // src/agents/llm-agent.ts
 var Agent = class extends BaseAgent {
+  /**
+   * The LLM model to use
+   */
+  model;
+  /**
+   * The LLM instance
+   */
+  llm;
+  /**
+   * Instructions for the agent
+   */
+  instructions;
+  /**
+   * Tools available to the agent
+   */
+  tools;
+  /**
+   * Maximum number of tool execution steps to prevent infinite loops
+   */
+  maxToolExecutionSteps;
+  /**
+   * Memory service for long-term storage and retrieval
+   */
+  memoryService;
+  /**
+   * Session service for managing conversations
+   */
+  sessionService;
+  /**
+   * User ID for the session
+   */
+  userId;
+  /**
+   * Application name
+   */
+  appName;
+  /**
+   * Whether to automatically augment prompts with relevant memory
+   */
+  useMemoryAugmentation;
+  /**
+   * The maximum number of memory items to include in augmentation
+   */
+  maxMemoryItems;
+  /**
+   * The minimum relevance score for memory augmentation (0-1)
+   */
+  memoryRelevanceThreshold;
   /**
    * Constructor for Agent
    */
@@ -1417,6 +1591,18 @@ ${response.content || "No content"}
 
 // src/agents/loop-agent.ts
 var LoopAgent = class extends BaseAgent {
+  /**
+   * Maximum number of iterations to prevent infinite loops
+   */
+  maxIterations;
+  /**
+   * Agent that decides whether to continue the loop
+   */
+  conditionAgent;
+  /**
+   * Custom condition check function
+   */
+  conditionCheck;
   /**
    * Constructor for LoopAgent
    */
@@ -1673,6 +1859,22 @@ ${lastResponse.content || ""}`,
 
 // src/agents/lang-graph-agent.ts
 var LangGraphAgent = class extends BaseAgent {
+  /**
+   * Graph nodes (agents and their connections)
+   */
+  nodes;
+  /**
+   * Root node to start execution from
+   */
+  rootNode;
+  /**
+   * Maximum number of steps to prevent infinite loops
+   */
+  maxSteps;
+  /**
+   * Results from node executions
+   */
+  results = [];
   /**
    * Constructor for LangGraphAgent
    */
@@ -1681,10 +1883,6 @@ var LangGraphAgent = class extends BaseAgent {
       name: config.name,
       description: config.description
     });
-    /**
-     * Results from node executions
-     */
-    this.results = [];
     this.nodes = /* @__PURE__ */ new Map();
     for (const node of config.nodes) {
       if (this.nodes.has(node.name)) {
@@ -2236,6 +2434,7 @@ init_base_tool();
 import fs from "fs/promises";
 import path from "path";
 var FileOperationsTool = class extends BaseTool {
+  basePath;
   constructor(options) {
     super({
       name: "file_operations",
@@ -2791,6 +2990,8 @@ var McpErrorType = /* @__PURE__ */ ((McpErrorType2) => {
   return McpErrorType2;
 })(McpErrorType || {});
 var McpError = class extends Error {
+  type;
+  originalError;
   constructor(message, type, originalError) {
     super(message);
     this.name = "McpError";
@@ -2829,10 +3030,11 @@ function withRetry(fn, instance, reinitMethod, maxRetries = 1) {
 
 // src/tools/mcp/client.ts
 var McpClientService = class {
+  config;
+  client = null;
+  transport = null;
+  isClosing = false;
   constructor(config) {
-    this.client = null;
-    this.transport = null;
-    this.isClosing = false;
     this.config = config;
   }
   /**
@@ -3235,6 +3437,9 @@ async function createTool(mcpTool, client) {
   }
 }
 var McpToolAdapter = class extends BaseTool {
+  mcpTool;
+  client;
+  clientService = null;
   constructor(mcpTool, client) {
     const metadata = mcpTool.metadata || {};
     super({
@@ -3244,7 +3449,6 @@ var McpToolAdapter = class extends BaseTool {
       shouldRetryOnFailure: metadata.shouldRetryOnFailure ?? false,
       maxRetryAttempts: metadata.maxRetryAttempts ?? 3
     });
-    this.clientService = null;
     this.mcpTool = mcpTool;
     this.client = client;
     if (client.reinitialize && typeof client.reinitialize === "function") {
@@ -3323,11 +3527,12 @@ var McpToolAdapter = class extends BaseTool {
 
 // src/tools/mcp/index.ts
 var McpToolset = class {
+  config;
+  clientService = null;
+  toolFilter = null;
+  tools = [];
+  isClosing = false;
   constructor(config, toolFilter = null) {
-    this.clientService = null;
-    this.toolFilter = null;
-    this.tools = [];
-    this.isClosing = false;
     this.config = config;
     this.toolFilter = toolFilter;
     this.clientService = new McpClientService(config);
@@ -3505,6 +3710,30 @@ __export(models_exports, {
 
 // src/models/llm-response.ts
 var LLMResponse = class {
+  /**
+   * Content of the response
+   */
+  content;
+  /**
+   * Function calls in the response
+   */
+  function_call;
+  /**
+   * Tool calls in the response
+   */
+  tool_calls;
+  /**
+   * Role of the message (usually 'assistant')
+   */
+  role;
+  /**
+   * Whether this is a partial response in a stream
+   */
+  is_partial;
+  /**
+   * Raw provider response
+   */
+  raw_response;
   constructor(data) {
     this.content = data.content;
     this.function_call = data.function_call;
@@ -3517,6 +3746,10 @@ var LLMResponse = class {
 
 // src/models/base-llm.ts
 var BaseLLM = class {
+  /**
+   * The name of the LLM model
+   */
+  model;
   /**
    * Constructor for BaseLLM
    */
@@ -3542,12 +3775,10 @@ var BaseLLM = class {
 
 // src/models/base-llm-connection.ts
 var BaseLLMConnection = class {
-  constructor() {
-    /**
-     * Whether the connection is active
-     */
-    this._isActive = true;
-  }
+  /**
+   * Whether the connection is active
+   */
+  _isActive = true;
   /**
    * Gets whether the connection is active
    */
@@ -3567,6 +3798,32 @@ import axios from "axios";
 
 // src/models/anthropic-llm-connection.ts
 var AnthropicLLMConnection = class extends BaseLLMConnection {
+  /**
+   * Axios instance for API calls
+   */
+  client;
+  /**
+   * Current model to use
+   */
+  model;
+  /**
+   * Current messages in the conversation
+   */
+  messages;
+  /**
+   * System message if present
+   */
+  systemMessage;
+  /**
+   * Default parameters for requests
+   */
+  defaultParams;
+  /**
+   * Callbacks for handling responses, errors, and connection end
+   */
+  responseCallback;
+  errorCallback;
+  endCallback;
   /**
    * Constructor
    */
@@ -3834,6 +4091,18 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
 
 // src/models/anthropic-llm.ts
 var AnthropicLLM = class extends BaseLLM {
+  /**
+   * Anthropic API key
+   */
+  apiKey;
+  /**
+   * Anthropic API base URL
+   */
+  baseURL;
+  /**
+   * Default parameters for requests
+   */
+  defaultParams;
   /**
    * Constructor for AnthropicLLM
    */
@@ -4134,6 +4403,14 @@ import {
   GoogleGenAI
 } from "@google/genai";
 var GoogleLLM = class extends BaseLLM {
+  /**
+   * Generative model instance
+   */
+  ai;
+  /**
+   * Default parameters for requests
+   */
+  defaultParams;
   /**
    * Constructor for GoogleLLM
    */
@@ -4495,15 +4772,43 @@ import OpenAI from "openai";
 
 // src/models/openai-llm-connection.ts
 var OpenAILLMConnection = class extends BaseLLMConnection {
+  /**
+   * OpenAI client
+   */
+  client;
+  /**
+   * The model name
+   */
+  model;
+  /**
+   * The initial request
+   */
+  initialRequest;
+  /**
+   * Default parameters
+   */
+  defaultParams;
+  /**
+   * Response callback
+   */
+  responseCallback;
+  /**
+   * Error callback
+   */
+  errorCallback;
+  /**
+   * End callback
+   */
+  endCallback;
+  /**
+   * Ongoing chat history
+   */
+  messages = [];
   /**
    * Constructor for OpenAILLMConnection
    */
   constructor(client, model, initialRequest, defaultParams) {
     super();
-    /**
-     * Ongoing chat history
-     */
-    this.messages = [];
     this.client = client;
     this.model = model;
     this.initialRequest = initialRequest;
@@ -4714,6 +5019,14 @@ var OpenAILLMConnection = class extends BaseLLMConnection {
 
 // src/models/openai-llm.ts
 var OpenAILLM = class extends BaseLLM {
+  /**
+   * OpenAI client instance
+   */
+  client;
+  /**
+   * Default parameters for requests
+   */
+  defaultParams;
   /**
    * Constructor for OpenAILLM
    */
@@ -5042,6 +5355,10 @@ var AuthCredentialType = /* @__PURE__ */ ((AuthCredentialType2) => {
   return AuthCredentialType2;
 })(AuthCredentialType || {});
 var AuthCredential = class {
+  /**
+   * Type of credential
+   */
+  type;
   /**
    * Constructor for AuthCredential
    */
@@ -5062,6 +5379,10 @@ var AuthCredential = class {
   }
 };
 var ApiKeyCredential = class extends AuthCredential {
+  /**
+   * The API key
+   */
+  apiKey;
   /**
    * Constructor for ApiKeyCredential
    */
@@ -5087,6 +5408,14 @@ var ApiKeyCredential = class extends AuthCredential {
   }
 };
 var BasicAuthCredential = class extends AuthCredential {
+  /**
+   * The username
+   */
+  username;
+  /**
+   * The password
+   */
+  password;
   /**
    * Constructor for BasicAuthCredential
    */
@@ -5111,6 +5440,10 @@ var BasicAuthCredential = class extends AuthCredential {
   }
 };
 var BearerTokenCredential = class extends AuthCredential {
+  /**
+   * The bearer token
+   */
+  token;
   /**
    * Constructor for BearerTokenCredential
    */
@@ -5134,6 +5467,22 @@ var BearerTokenCredential = class extends AuthCredential {
   }
 };
 var OAuth2Credential = class extends AuthCredential {
+  /**
+   * The access token
+   */
+  accessToken;
+  /**
+   * The refresh token
+   */
+  refreshToken;
+  /**
+   * When the token expires
+   */
+  expiresAt;
+  /**
+   * Function to refresh the token
+   */
+  refreshFunction;
   /**
    * Constructor for OAuth2Credential
    */
@@ -5200,6 +5549,14 @@ var OAuth2Credential = class extends AuthCredential {
 
 // src/auth/auth-config.ts
 var AuthConfig = class {
+  /**
+   * The authentication scheme
+   */
+  authScheme;
+  /**
+   * Additional context properties
+   */
+  context;
   /**
    * Constructor for AuthConfig
    */
@@ -5211,6 +5568,14 @@ var AuthConfig = class {
 
 // src/auth/auth-handler.ts
 var AuthHandler = class {
+  /**
+   * The authentication configuration
+   */
+  authConfig;
+  /**
+   * The authentication credential
+   */
+  credential;
   /**
    * Constructor for AuthHandler
    */
@@ -5252,11 +5617,27 @@ var AuthSchemeType = /* @__PURE__ */ ((AuthSchemeType2) => {
   return AuthSchemeType2;
 })(AuthSchemeType || {});
 var AuthScheme = class {
+  /**
+   * The type of authentication scheme
+   */
+  type;
   constructor(type) {
     this.type = type;
   }
 };
 var ApiKeyScheme = class extends AuthScheme {
+  /**
+   * Where the API key is sent
+   */
+  in;
+  /**
+   * Name of the parameter
+   */
+  name;
+  /**
+   * Description of the API key
+   */
+  description;
   /**
    * Constructor for ApiKeyScheme
    */
@@ -5268,6 +5649,18 @@ var ApiKeyScheme = class extends AuthScheme {
   }
 };
 var HttpScheme = class extends AuthScheme {
+  /**
+   * The HTTP authentication scheme
+   */
+  scheme;
+  /**
+   * Bearer format when scheme is 'bearer'
+   */
+  bearerFormat;
+  /**
+   * Description of the scheme
+   */
+  description;
   /**
    * Constructor for HttpScheme
    */
@@ -5279,6 +5672,14 @@ var HttpScheme = class extends AuthScheme {
   }
 };
 var OAuth2Scheme = class extends AuthScheme {
+  /**
+   * OAuth flows
+   */
+  flows;
+  /**
+   * Description of the scheme
+   */
+  description;
   /**
    * Constructor for OAuth2Scheme
    */
@@ -5289,6 +5690,14 @@ var OAuth2Scheme = class extends AuthScheme {
   }
 };
 var OpenIdConnectScheme = class extends AuthScheme {
+  /**
+   * OpenID Connect URL
+   */
+  openIdConnectUrl;
+  /**
+   * Description of the scheme
+   */
+  description;
   /**
    * Constructor for OpenIdConnectScheme
    */
@@ -5301,6 +5710,8 @@ var OpenIdConnectScheme = class extends AuthScheme {
 
 // src/sessions/state.ts
 var SessionState = class _SessionState {
+  state;
+  dirty;
   constructor() {
     this.state = /* @__PURE__ */ new Map();
     this.dirty = /* @__PURE__ */ new Set();
@@ -5388,6 +5799,10 @@ __export(memory_exports, {
 
 // src/memory/in-memory-memory-service.ts
 var InMemoryMemoryService = class {
+  /**
+   * Map of sessions by ID
+   */
+  sessions;
   /**
    * Constructor for InMemoryMemoryService
    */
@@ -5491,6 +5906,18 @@ var InMemoryMemoryService = class {
 import fs2 from "fs";
 import path2 from "path";
 var PersistentMemoryService = class {
+  /**
+   * In-memory service used for search operations
+   */
+  inMemoryService;
+  /**
+   * Directory where memory files will be stored
+   */
+  storageDir;
+  /**
+   * File prefix for memory files
+   */
+  filePrefix;
   /**
    * Constructor for PersistentMemoryService
    */
@@ -5649,11 +6076,16 @@ __export(sessions_exports, {
   SessionState: () => SessionState,
   cloneSession: () => cloneSession,
   generateSessionId: () => generateSessionId,
+  sessionsSchema: () => sessionsSchema2,
   validateSession: () => validateSession
 });
 
 // src/sessions/in-memory-session-service.ts
 var InMemorySessionService = class {
+  /**
+   * Map of sessions by ID
+   */
+  sessions;
   /**
    * Constructor for InMemorySessionService
    */
@@ -5799,6 +6231,8 @@ var sessionsSchema = pgTable("sessions", {
   // Store serialized SessionState as JSONB
 });
 var PostgresSessionService = class {
+  db;
+  sessionsTable;
   constructor(config) {
     this.db = config.db;
     this.sessionsTable = config.sessionsTable || sessionsSchema;
@@ -5921,22 +6355,25 @@ var PostgresSessionService = class {
 // src/sessions/pglite-session-service.ts
 import { eq as eq2 } from "drizzle-orm";
 import { jsonb as jsonb2, pgTable as pgTable2, timestamp as timestamp2, varchar as varchar2 } from "drizzle-orm/pg-core";
+import { drizzle } from "drizzle-orm/pglite";
 var sessionsSchema2 = pgTable2("sessions", {
   id: varchar2("id", { length: 255 }).primaryKey(),
   userId: varchar2("user_id", { length: 255 }).notNull(),
   messages: jsonb2("messages").default("[]").$type(),
-  // Store Message array as JSONB
   metadata: jsonb2("metadata").default("{}").$type(),
   createdAt: timestamp2("created_at", { withTimezone: true }).defaultNow().notNull(),
   updatedAt: timestamp2("updated_at", { withTimezone: true }).defaultNow().notNull(),
   state: jsonb2("state").default("{}").$type()
-  // Store serialized SessionState as JSONB
 });
 var PgLiteSessionService = class {
+  db;
+  sessionsTable;
+  initialized = false;
   constructor(config) {
-    this.initialized = false;
-    this.db = config.db;
-    this.sessionsTable = config.sessionsTable || sessionsSchema2;
+    this.db = drizzle(config.pglite, {
+      schema: { sessions: sessionsSchema2 }
+    });
+    this.sessionsTable = sessionsSchema2;
     if (!config.skipTableCreation) {
       this.initializeDatabase().catch((error) => {
         console.error("Failed to initialize PgLite database:", error);
@@ -5991,9 +6428,7 @@ var PgLiteSessionService = class {
       metadata,
       createdAt: now,
       updatedAt: now,
-      // Drizzle's defaultNow() on schema handles this, but explicit is fine
       state: sessionState.toObject()
-      // Serialize SessionState
     };
     const results = await this.db.insert(this.sessionsTable).values(newSessionData).returning();
     const result = results[0];
@@ -6008,7 +6443,6 @@ var PgLiteSessionService = class {
       messages: Array.isArray(result.messages) ? result.messages : [],
       metadata: result.metadata || {},
       state: SessionState.fromObject(result.state || {}),
-      // Ensure dates are Date objects if Drizzle returns strings for some drivers/configs
       createdAt: new Date(result.createdAt),
       updatedAt: new Date(result.updatedAt)
     };
@@ -6036,7 +6470,6 @@ var PgLiteSessionService = class {
       userId: session.userId,
       messages: session.messages,
       metadata: session.metadata,
-      // createdAt should typically not be updated after creation
       updatedAt: /* @__PURE__ */ new Date(),
       state: session.state.toObject()
     };
@@ -6063,12 +6496,6 @@ var PgLiteSessionService = class {
     await this.ensureInitialized();
     await this.db.delete(this.sessionsTable).where(eq2(this.sessionsTable.id, sessionId));
   }
-  /**
-   * Appends an event to a session object
-   * @param session The session to append the event to
-   * @param event The event to append
-   * @returns The appended event
-   */
   async appendEvent(session, event) {
     await this.ensureInitialized();
     if (event.is_partial) {
@@ -6090,11 +6517,6 @@ var PgLiteSessionService = class {
     await this.updateSession(session);
     return event;
   }
-  // TODO: Consider if table creation/migration logic is needed here or handled externally (e.g., drizzle-kit migrations)
-  // TODO: Implement methods corresponding to Python's append_event, list_events,
-  // get_app_state, update_app_state, get_user_state, update_user_state
-  // if full parity with Python's DatabaseSessionService is desired.
-  // This would require defining corresponding Drizzle schemas for Events, AppState, UserState.
 };
 
 // src/sessions/session-util.ts
@@ -6132,19 +6554,32 @@ import { v4 as uuidv4 } from "uuid";
 
 // src/events/event-actions.ts
 var EventActions = class {
+  /**
+   * If true, it won't call model to summarize function response.
+   * Only used for function_response event.
+   */
+  skipSummarization;
+  /**
+   * Indicates that the event is updating the state with the given delta.
+   */
+  stateDelta = {};
+  /**
+   * Indicates that the event is updating an artifact. key is the filename,
+   * value is the version.
+   */
+  artifactDelta = {};
+  /**
+   * If set, the event transfers to the specified agent.
+   */
+  transferToAgent;
+  /**
+   * The agent is escalating to a higher level agent.
+   */
+  escalate;
   /**
    * Constructor for EventActions
    */
   constructor(options = {}) {
-    /**
-     * Indicates that the event is updating the state with the given delta.
-     */
-    this.stateDelta = {};
-    /**
-     * Indicates that the event is updating an artifact. key is the filename,
-     * value is the version.
-     */
-    this.artifactDelta = {};
     this.skipSummarization = options.skipSummarization;
     this.stateDelta = options.stateDelta || {};
     this.artifactDelta = options.artifactDelta || {};
@@ -6155,6 +6590,40 @@ var EventActions = class {
 
 // src/events/event.ts
 var Event = class _Event extends LLMResponse {
+  /**
+   * The invocation ID of the event.
+   */
+  invocationId = "";
+  /**
+   * 'user' or the name of the agent, indicating who appended the event to the session.
+   */
+  author;
+  /**
+   * The actions taken by the agent.
+   */
+  actions = new EventActions();
+  /**
+   * Set of ids of the long running function calls.
+   * Agent client will know from this field about which function call is long running.
+   * Only valid for function call event.
+   */
+  longRunningToolIds;
+  /**
+   * The branch of the event.
+   * The format is like agent_1.agent_2.agent_3, where agent_1 is the parent of
+   * agent_2, and agent_2 is the parent of agent_3.
+   * Branch is used when multiple sub-agent shouldn't see their peer agents'
+   * conversation history.
+   */
+  branch;
+  /**
+   * The unique identifier of the event.
+   */
+  id = "";
+  /**
+   * The timestamp of the event.
+   */
+  timestamp;
   /**
    * Constructor for Event
    */
@@ -6181,18 +6650,6 @@ var Event = class _Event extends LLMResponse {
       is_partial: partial,
       raw_response
     });
-    /**
-     * The invocation ID of the event.
-     */
-    this.invocationId = "";
-    /**
-     * The actions taken by the agent.
-     */
-    this.actions = new EventActions();
-    /**
-     * The unique identifier of the event.
-     */
-    this.id = "";
     this.invocationId = invocationId;
     this.author = author;
     this.actions = actions;
@@ -6230,6 +6687,22 @@ var Event = class _Event extends LLMResponse {
 
 // src/runners.ts
 var Runner = class {
+  /**
+   * The app name of the runner.
+   */
+  appName;
+  /**
+   * The root agent to run.
+   */
+  agent;
+  /**
+   * The session service for the runner.
+   */
+  sessionService;
+  /**
+   * The memory service for the runner.
+   */
+  memoryService;
   /**
    * Initializes the Runner.
    */
@@ -6429,5 +6902,6 @@ export {
   mcpSchemaToParameters,
   normalizeJsonSchema,
   registerProviders,
+  sessionsSchema2 as sessionsSchema,
   validateSession
 };