wave-agent-sdk 0.0.3 → 0.0.5

package/dist/agent.d.ts

@@ -2,7 +2,7 @@ import { type MessageManagerCallbacks } from "./managers/messageManager.js";
 import { type McpManagerCallbacks } from "./managers/mcpManager.js";
 import { type BackgroundBashManagerCallbacks } from "./managers/backgroundBashManager.js";
 import type { SlashCommand, CustomSlashCommand } from "./types.js";
-import type { Message, Logger, McpServerStatus } from "./types.js";
+import type { Message, Logger, McpServerStatus, Usage } from "./types.js";
 /**
  * Configuration options for Agent instances
  *
@@ -25,6 +25,12 @@ export interface AgentOptions {
     workdir?: string;
     /**Optional custom system prompt - if provided, replaces default system prompt */
     systemPrompt?: string;
+    /**
+     * Optional custom directory for session file storage
+     * @default join(homedir(), ".wave", "sessions")
+     * @example "/path/to/custom/sessions"
+     */
+    sessionDir?: string;
 }
 export interface AgentCallbacks extends MessageManagerCallbacks, BackgroundBashManagerCallbacks, McpManagerCallbacks {
 }
@@ -41,20 +47,34 @@ export declare class Agent {
     private hookManager;
     private workdir;
     private systemPrompt?;
+    private _usages;
     private gatewayConfig;
     private modelConfig;
     private tokenLimit;
     /**
      * Agent constructor - handles configuration resolution and validation
      *
-     * IMPORTANT: Keep this constructor's signature exactly the same as Agent.create()
-     * to maintain API consistency. Both methods should accept the same AgentOptions.
+     * IMPORTANT: This constructor is private. Use Agent.create() instead for proper
+     * async initialization. Keep this constructor's signature exactly the same as
+     * Agent.create() to maintain API consistency.
      *
      * @param options - Configuration options for the Agent instance
+     * @param options.sessionDir - Optional custom directory for session storage
      */
-    constructor(options: AgentOptions);
+    private constructor();
     get sessionId(): string;
     get messages(): Message[];
+    get usages(): Usage[];
+    /**
+     * Rebuild usage array from messages containing usage metadata
+     * Called during session restoration to reconstruct usage tracking
+     */
+    private rebuildUsageFromMessages;
+    /**
+     * Add usage data to the tracking array and trigger callbacks
+     * @param usage Usage data from AI operations
+     */
+    private addUsage;
     get latestTotalTokens(): number;
     get userInputHistory(): string[];
     /** Get working directory */
@@ -82,6 +102,43 @@ export declare class Agent {
      * @param options - Same AgentOptions interface used by constructor
      * @returns Promise<Agent> - Fully initialized Agent instance
      */
+    /**
+     * Create a new Agent instance with async initialization
+     *
+     * This is the recommended way to create Agent instances. The constructor is private
+     * to ensure proper async initialization of all components.
+     *
+     * @param options - Configuration options for the Agent instance
+     * @param options.apiKey - API key for the AI service (or set WAVE_API_KEY env var)
+     * @param options.baseURL - Base URL for the AI service (or set WAVE_BASE_URL env var)
+     * @param options.sessionDir - Optional custom directory for session file storage.
+     *   If not provided, defaults to ~/.wave/sessions/. Can be relative or absolute path.
+     *   Examples: "./app-sessions", "/var/myapp/sessions", "~/Documents/sessions"
+     * @param options.callbacks - Optional callbacks for various Agent events
+     * @param options.restoreSessionId - Optional session ID to restore from
+     * @param options.continueLastSession - Whether to continue the last session automatically
+     * @param options.logger - Optional custom logger implementation
+     * @param options.messages - Optional initial messages for testing convenience
+     * @param options.workdir - Working directory (defaults to process.cwd())
+     * @param options.systemPrompt - Optional custom system prompt
+     * @returns Promise that resolves to initialized Agent instance
+     *
+     * @example
+     * ```typescript
+     * // Basic usage with default session directory
+     * const agent = await Agent.create({
+     *   apiKey: 'your-api-key',
+     *   baseURL: 'https://api.example.com'
+     * });
+     *
+     * // Custom session directory
+     * const agent = await Agent.create({
+     *   apiKey: 'your-api-key',
+     *   baseURL: 'https://api.example.com',
+     *   sessionDir: './app-sessions'
+     * });
+     * ```
+     */
     static create(options: AgentOptions): Promise<Agent>;
     /** Private initialization method, handles async initialization logic */
     private initialize;

package/dist/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,uBAAuB,EAC7B,MAAM,8BAA8B,CAAC;AAKtC,OAAO,EAAc,KAAK,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAEhF,OAAO,EAEL,KAAK,8BAA8B,EACpC,MAAM,qCAAqC,CAAC;AAE7C,OAAO,KAAK,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AACnE,OAAO,KAAK,EACV,OAAO,EACP,MAAM,EACN,eAAe,EAGhB,MAAM,YAAY,CAAC;AAMpB;;;;;GAKG;AACH,MAAM,WAAW,YAAY;IAE3B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,SAAS,CAAC,EAAE,cAAc,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,qEAAqE;IACrE,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;IACrB,6DAA6D;IAC7D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,iFAAiF;IACjF,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,cACf,SAAQ,uBAAuB,EAC7B,8BAA8B,EAC9B,mBAAmB;CAAG;AAE1B,qBAAa,KAAK;IAChB,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,SAAS,CAAY;IAE7B,OAAO,CAAC,WAAW,CAA4B;IAC/C,OAAO,CAAC,qBAAqB,CAAwB;IACrD,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAS;IAG9B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAS;IAE3B;;;;;;;OAOG;gBACS,OAAO,EAAE,YAAY;IAkGjC,IAAW,SAAS,IAAI,MAAM,CAE7B;IAED,IAAW,QAAQ,IAAI,OAAO,EAAE,CAE/B;IAED,IAAW,iBAAiB,IAAI,MAAM,CAErC;IAED,IAAW,gBAAgB,IAAI,MAAM,EAAE,CAEtC;IAED,4BAA4B;IAC5B,IAAW,gBAAgB,IAAI,MAAM,CAEpC;IAED,4BAA4B;IAC5B,IAAW,SAAS,IAAI,OAAO,CAE9B;IAED,qCAAqC;IACrC,IAAW,aAAa,IAAI,OAAO,CAElC;IAED,wCAAwC;IACxC,IAAW,gBAAgB,IAAI,OAAO,CAErC;IAED,uCAAuC;IAChC,wBAAwB,CAC7B,EAAE,EAAE,MAAM,EACV,MAAM,CAAC,EAAE,MAAM,GACd;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAI5D,iCAAiC;IAC1B,mBAAmB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAI/C;;;;;;;;OAQG;WACU,MAAM,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC;IAW1D,wEAAwE;YAC1D,UAAU;IAwDjB,cAAc,IAAI,IAAI;IAI7B,2BAA2B;IACd,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAM/D,uCAAuC;IAChC,aAAa,IAAI,IAAI;IAI5B,kFAAkF;IAC3E,YAAY,IAAI,IAAI;IAO3B,2BAA2B;IAC3B,OAAO,CAAC,iBAAiB;IAIzB,uCAAuC;IAChC,gBAAgB,IAAI,IAAI;IAI/B,wCAAwC;IACjC,iBAAiB,IAAI,IAAI;IAIhC,uCAAuC;IAChC,cAAc,IAAI,IAAI;IAI7B,4CAA4C;IACrC,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;IAIjD,2CAA2C;IAC9B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAcxB,WAAW,CACtB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,GACjD,OAAO,CAAC,IAAI,CAAC;IAgEhB,iDAAiD;IACpC,UAAU,CACrB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,SAAS,GAAG,MAAM,GACvB,OAAO,CAAC,IAAI,CAAC;IAqChB,gCAAgC;IACzB,aAAa,IAAI,eAAe,EAAE;IAIzC,yBAAyB;IACZ,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAInE,4BAA4B;IACf,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAMtE,uCAAuC;IAChC,gBAAgB,IAAI,YAAY,EAAE;IAIzC,oCAAoC;IAC7B,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAIlD,6BAA6B;IACtB,oBAAoB,IAAI,IAAI;IAInC,iCAAiC;IAC1B,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,kBAAkB,GAAG,SAAS;IAI1E,8BAA8B;IACvB,iBAAiB,IAAI,kBAAkB,EAAE;CAGjD"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,uBAAuB,EAC7B,MAAM,8BAA8B,CAAC;AAKtC,OAAO,EAAc,KAAK,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAEhF,OAAO,EAEL,KAAK,8BAA8B,EACpC,MAAM,qCAAqC,CAAC;AAE7C,OAAO,KAAK,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AACnE,OAAO,KAAK,EACV,OAAO,EACP,MAAM,EACN,eAAe,EAGf,KAAK,EACN,MAAM,YAAY,CAAC;AAMpB;;;;;GAKG;AACH,MAAM,WAAW,YAAY;IAE3B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,SAAS,CAAC,EAAE,cAAc,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,qEAAqE;IACrE,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;IACrB,6DAA6D;IAC7D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,iFAAiF;IACjF,YAAY,CAAC,EAAE,MAAM,CAAC;IAGtB;;;;OAIG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,cACf,SAAQ,uBAAuB,EAC7B,8BAA8B,EAC9B,mBAAmB;CAAG;AAE1B,qBAAa,KAAK;IAChB,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,SAAS,CAAY;IAE7B,OAAO,CAAC,WAAW,CAA4B;IAC/C,OAAO,CAAC,qBAAqB,CAAwB;IACrD,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,OAAO,CAAe;IAG9B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAS;IAE3B;;;;;;;;;OASG;IACH,OAAO;IA+GP,IAAW,SAAS,IAAI,MAAM,CAE7B;IAED,IAAW,QAAQ,IAAI,OAAO,EAAE,CAE/B;IAED,IAAW,MAAM,IAAI,KAAK,EAAE,CAE3B;IAED;;;OAGG;IACH,OAAO,CAAC,wBAAwB;IAWhC;;;OAGG;IACH,OAAO,CAAC,QAAQ;IAKhB,IAAW,iBAAiB,IAAI,MAAM,CAErC;IAED,IAAW,gBAAgB,IAAI,MAAM,EAAE,CAEtC;IAED,4BAA4B;IAC5B,IAAW,gBAAgB,IAAI,MAAM,CAEpC;IAED,4BAA4B;IAC5B,IAAW,SAAS,IAAI,OAAO,CAE9B;IAED,qCAAqC;IACrC,IAAW,aAAa,IAAI,OAAO,CAElC;IAED,wCAAwC;IACxC,IAAW,gBAAgB,IAAI,OAAO,CAErC;IAED,uCAAuC;IAChC,wBAAwB,CAC7B,EAAE,EAAE,MAAM,EACV,MAAM,CAAC,EAAE,MAAM,GACd;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAI5D,iCAAiC;IAC1B,mBAAmB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAI/C;;;;;;;;OAQG;IACH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoCG;WACU,MAAM,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC;IAW1D,wEAAwE;YAC1D,UAAU;IA4DjB,cAAc,IAAI,IAAI;IAI7B,2BAA2B;IACd,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAM/D,uCAAuC;IAChC,aAAa,IAAI,IAAI;IAI5B,kFAAkF;IAC3E,YAAY,IAAI,IAAI;IAO3B,2BAA2B;IAC3B,OAAO,CAAC,iBAAiB;IAIzB,uCAAuC;IAChC,gBAAgB,IAAI,IAAI;IAI/B,wCAAwC;IACjC,iBAAiB,IAAI,IAAI;IAIhC,uCAAuC;IAChC,cAAc,IAAI,IAAI;IAI7B,4CAA4C;IACrC,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;IAIjD,2CAA2C;IAC9B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAcxB,WAAW,CACtB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,GACjD,OAAO,CAAC,IAAI,CAAC;IAgEhB,iDAAiD;IACpC,UAAU,CACrB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,SAAS,GAAG,MAAM,GACvB,OAAO,CAAC,IAAI,CAAC;IAqChB,gCAAgC;IACzB,aAAa,IAAI,eAAe,EAAE;IAIzC,yBAAyB;IACZ,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAInE,4BAA4B;IACf,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAMtE,uCAAuC;IAChC,gBAAgB,IAAI,YAAY,EAAE;IAIzC,oCAAoC;IAC7B,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAIlD,6BAA6B;IACtB,oBAAoB,IAAI,IAAI;IAInC,iCAAiC;IAC1B,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,kBAAkB,GAAG,SAAS;IAI1E,8BAA8B;IACvB,iBAAiB,IAAI,kBAAkB,EAAE;CAGjD"}

package/dist/agent.js

@@ -15,14 +15,17 @@ export class Agent {
     /**
      * Agent constructor - handles configuration resolution and validation
      *
-     * IMPORTANT: Keep this constructor's signature exactly the same as Agent.create()
-     * to maintain API consistency. Both methods should accept the same AgentOptions.
+     * IMPORTANT: This constructor is private. Use Agent.create() instead for proper
+     * async initialization. Keep this constructor's signature exactly the same as
+     * Agent.create() to maintain API consistency.
      *
      * @param options - Configuration options for the Agent instance
+     * @param options.sessionDir - Optional custom directory for session storage
      */
     constructor(options) {
         this.bashManager = null;
-        const { callbacks = {}, logger, workdir, systemPrompt } = options;
+        this._usages = []; // Usage tracking array
+        const { callbacks = {}, logger, workdir, systemPrompt, sessionDir, } = options;
         // Resolve configuration from constructor args and environment variables
         const gatewayConfig = configResolver.resolveGatewayConfig(options.apiKey, options.baseURL);
         const modelConfig = configResolver.resolveModelConfig(options.agentModel, options.fastModel);
@@ -53,6 +56,7 @@ export class Agent {
             callbacks,
             workdir: this.workdir,
             logger: this.logger,
+            sessionDir,
         });
         // Initialize subagent manager with all dependencies in constructor
         // IMPORTANT: Must be initialized AFTER MessageManager
@@ -64,6 +68,7 @@ export class Agent {
             gatewayConfig,
             modelConfig,
             tokenLimit,
+            onUsageAdded: (usage) => this.addUsage(usage),
         });
         // Initialize AI manager with resolved configuration
         this.aiManager = new AIManager({
@@ -72,7 +77,12 @@ export class Agent {
             logger: this.logger,
             backgroundBashManager: this.backgroundBashManager,
             hookManager: this.hookManager,
-            callbacks,
+            callbacks: {
+                ...callbacks,
+                onUsageAdded: (usage) => {
+                    this.addUsage(usage);
+                },
+            },
             workdir: this.workdir,
             systemPrompt: this.systemPrompt,
             gatewayConfig: this.gatewayConfig,
@@ -99,6 +109,31 @@ export class Agent {
     get messages() {
         return this.messageManager.getMessages();
     }
+    get usages() {
+        return [...this._usages]; // Return copy to prevent external modification
+    }
+    /**
+     * Rebuild usage array from messages containing usage metadata
+     * Called during session restoration to reconstruct usage tracking
+     */
+    rebuildUsageFromMessages() {
+        this._usages = [];
+        this.messages.forEach((message) => {
+            if (message.role === "assistant" && message.usage) {
+                this._usages.push(message.usage);
+            }
+        });
+        // Trigger callback after rebuilding usage array
+        this.messageManager.triggerUsageChange();
+    }
+    /**
+     * Add usage data to the tracking array and trigger callbacks
+     * @param usage Usage data from AI operations
+     */
+    addUsage(usage) {
+        this._usages.push(usage);
+        this.messageManager.triggerUsageChange();
+    }
     get latestTotalTokens() {
         return this.messageManager.getlatestTotalTokens();
     }
@@ -138,6 +173,43 @@ export class Agent {
      * @param options - Same AgentOptions interface used by constructor
      * @returns Promise<Agent> - Fully initialized Agent instance
      */
+    /**
+     * Create a new Agent instance with async initialization
+     *
+     * This is the recommended way to create Agent instances. The constructor is private
+     * to ensure proper async initialization of all components.
+     *
+     * @param options - Configuration options for the Agent instance
+     * @param options.apiKey - API key for the AI service (or set WAVE_API_KEY env var)
+     * @param options.baseURL - Base URL for the AI service (or set WAVE_BASE_URL env var)
+     * @param options.sessionDir - Optional custom directory for session file storage.
+     *   If not provided, defaults to ~/.wave/sessions/. Can be relative or absolute path.
+     *   Examples: "./app-sessions", "/var/myapp/sessions", "~/Documents/sessions"
+     * @param options.callbacks - Optional callbacks for various Agent events
+     * @param options.restoreSessionId - Optional session ID to restore from
+     * @param options.continueLastSession - Whether to continue the last session automatically
+     * @param options.logger - Optional custom logger implementation
+     * @param options.messages - Optional initial messages for testing convenience
+     * @param options.workdir - Working directory (defaults to process.cwd())
+     * @param options.systemPrompt - Optional custom system prompt
+     * @returns Promise that resolves to initialized Agent instance
+     *
+     * @example
+     * ```typescript
+     * // Basic usage with default session directory
+     * const agent = await Agent.create({
+     *   apiKey: 'your-api-key',
+     *   baseURL: 'https://api.example.com'
+     * });
+     *
+     * // Custom session directory
+     * const agent = await Agent.create({
+     *   apiKey: 'your-api-key',
+     *   baseURL: 'https://api.example.com',
+     *   sessionDir: './app-sessions'
+     * });
+     * ```
+     */
     static async create(options) {
         // Create Agent instance - configuration resolution and validation now happens in constructor
         const instance = new Agent(options);
@@ -178,7 +250,7 @@ export class Agent {
         // Initialize hooks configuration
         try {
             // Load hooks configuration from user and project settings
-            this.logger?.info("Loading hooks configuration...");
+            this.logger?.debug("Loading hooks configuration...");
             this.hookManager.loadConfigurationFromSettings();
             this.logger?.debug("Hooks system initialized successfully");
         }
@@ -190,10 +262,14 @@ export class Agent {
         if (options?.messages) {
             // If messages are provided, use them directly (useful for testing)
             this.messageManager.setMessages(options.messages);
+            // Rebuild usage array from restored messages
+            this.rebuildUsageFromMessages();
         }
         else {
             // Otherwise, handle session restoration
             await this.messageManager.handleSessionRestoration(options?.restoreSessionId, options?.continueLastSession);
+            // Rebuild usage array from restored messages
+            this.rebuildUsageFromMessages();
         }
     }
     abortAIMessage() {
@@ -238,7 +314,7 @@ export class Agent {
     }
     /** Destroy managers, clean up resources */
     async destroy() {
-        this.messageManager.saveSession();
+        await this.messageManager.saveSession();
         this.abortAIMessage();
         this.abortBashCommand();
         this.abortSlashCommand();
@@ -316,13 +392,13 @@ export class Agent {
             // Add successful MemoryBlock to the last assistant message
             const memoryText = message.substring(1).trim();
             const typeLabel = type === "project" ? "Project Memory" : "User Memory";
-            const storagePath = type === "project" ? "WAVE.md" : "user-memory.md";
+            const storagePath = type === "project" ? "AGENTS.md" : "user-memory.md";
             this.messageManager.addMemoryBlock(`${typeLabel}: ${memoryText}`, true, type, storagePath);
         }
         catch (error) {
             // Add failed MemoryBlock to the last assistant message
             const typeLabel = type === "project" ? "Project Memory" : "User Memory";
-            const storagePath = type === "project" ? "WAVE.md" : "user-memory.md";
+            const storagePath = type === "project" ? "AGENTS.md" : "user-memory.md";
             this.messageManager.addMemoryBlock(`${typeLabel} add failed: ${error instanceof Error ? error.message : String(error)}`, false, type, storagePath);
         }
     }

package/dist/hooks/manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/hooks/manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EACL,KAAK,SAAS,EAEd,KAAK,iBAAiB,EACtB,KAAK,wBAAwB,EAC7B,KAAK,oBAAoB,EACzB,KAAK,4BAA4B,EACjC,KAAK,mBAAmB,EACxB,KAAK,gBAAgB,EAItB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,KAAK,YAAY,EAAe,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,KAAK,aAAa,EAAgB,MAAM,eAAe,CAAC;AAEjE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAE1C,MAAM,WAAW,YAAY;IAE3B,iBAAiB,CACf,SAAS,CAAC,EAAE,wBAAwB,EACpC,YAAY,CAAC,EAAE,wBAAwB,GACtC,IAAI,CAAC;IAGR,6BAA6B,IAAI,IAAI,CAAC;IAGtC,YAAY,CACV,KAAK,EAAE,SAAS,EAChB,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,GAC3D,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAGlC,QAAQ,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;IAGvD,qBAAqB,CAAC,MAAM,EAAE,iBAAiB,GAAG,gBAAgB,CAAC;IAGnE,gBAAgB,IAAI,wBAAwB,GAAG,SAAS,CAAC;CAC1D;AAED,qBAAa,WAAY,YAAW,YAAY;IAC9C,OAAO,CAAC,aAAa,CAAuC;IAC5D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAe;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAgB;IACzC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAG/B,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAgC,EACzC,QAAQ,CAAC,EAAE,aAAa,EACxB,MAAM,CAAC,EAAE,MAAM;IAWjB;;;OAGG;IACH,iBAAiB,CACf,SAAS,CAAC,EAAE,wBAAwB,EACpC,YAAY,CAAC,EAAE,wBAAwB,GACtC,IAAI;IAyBP;;;OAGG;IACH,6BAA6B,IAAI,IAAI;IAmCrC;;OAEG;IACG,YAAY,CAChB,KAAK,EAAE,SAAS,EAChB,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,GAC3D,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAsHjC;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO;IAWtD;;OAEG;IACH,qBAAqB,CAAC,MAAM,EAAE,iBAAiB,GAAG,gBAAgB;IA8ClE;;OAEG;IACH,OAAO,CAAC,4BAA4B;IAyCpC;;OAEG;IACH,gBAAgB,IAAI,wBAAwB,GAAG,SAAS;IAOxD;;OAEG;IACH,kBAAkB,IAAI,IAAI;IAI1B;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAmDhC;;OAEG;IACH,OAAO,CAAC,wBAAwB;IA8BhC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAa/B;;OAEG;IACH,OAAO,CAAC,aAAa;IA4BrB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAqC3B;;OAEG;IACH,qBAAqB,IAAI;QACvB,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;KAC3C;CA2CF"}
+{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/hooks/manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EACL,KAAK,SAAS,EAEd,KAAK,iBAAiB,EACtB,KAAK,wBAAwB,EAC7B,KAAK,oBAAoB,EACzB,KAAK,4BAA4B,EACjC,KAAK,mBAAmB,EACxB,KAAK,gBAAgB,EAItB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,KAAK,YAAY,EAAe,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,KAAK,aAAa,EAAgB,MAAM,eAAe,CAAC;AAEjE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAE1C,MAAM,WAAW,YAAY;IAE3B,iBAAiB,CACf,SAAS,CAAC,EAAE,wBAAwB,EACpC,YAAY,CAAC,EAAE,wBAAwB,GACtC,IAAI,CAAC;IAGR,6BAA6B,IAAI,IAAI,CAAC;IAGtC,YAAY,CACV,KAAK,EAAE,SAAS,EAChB,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,GAC3D,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAGlC,QAAQ,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;IAGvD,qBAAqB,CAAC,MAAM,EAAE,iBAAiB,GAAG,gBAAgB,CAAC;IAGnE,gBAAgB,IAAI,wBAAwB,GAAG,SAAS,CAAC;CAC1D;AAED,qBAAa,WAAY,YAAW,YAAY;IAC9C,OAAO,CAAC,aAAa,CAAuC;IAC5D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAe;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAgB;IACzC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAG/B,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAgC,EACzC,QAAQ,CAAC,EAAE,aAAa,EACxB,MAAM,CAAC,EAAE,MAAM;IAWjB;;;OAGG;IACH,iBAAiB,CACf,SAAS,CAAC,EAAE,wBAAwB,EACpC,YAAY,CAAC,EAAE,wBAAwB,GACtC,IAAI;IAyBP;;;OAGG;IACH,6BAA6B,IAAI,IAAI;IAmCrC;;OAEG;IACG,YAAY,CAChB,KAAK,EAAE,SAAS,EAChB,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,GAC3D,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAwHjC;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO;IAWtD;;OAEG;IACH,qBAAqB,CAAC,MAAM,EAAE,iBAAiB,GAAG,gBAAgB;IA8ClE;;OAEG;IACH,OAAO,CAAC,4BAA4B;IAyCpC;;OAEG;IACH,gBAAgB,IAAI,wBAAwB,GAAG,SAAS;IAOxD;;OAEG;IACH,kBAAkB,IAAI,IAAI;IAI1B;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAmDhC;;OAEG;IACH,OAAO,CAAC,wBAAwB;IA8BhC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAa/B;;OAEG;IACH,OAAO,CAAC,aAAa;IA4BrB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAqC3B;;OAEG;IACH,qBAAqB,IAAI;QACvB,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;KAC3C;CA2CF"}

package/dist/hooks/manager.js

@@ -138,7 +138,7 @@ export class HookManager {
         // Generate execution summary
         const totalDuration = Date.now() - startTime;
         const summary = this.generateExecutionSummary(event, results, totalDuration);
-        this.logger?.info(`[HookManager] ${event} execution summary: ${summary}`);
+        this.logger?.debug(`[HookManager] ${event} execution summary: ${summary}`);
         return results;
     }
     /**

package/dist/managers/aiManager.d.ts

@@ -1,10 +1,11 @@
-import type { Logger, GatewayConfig, ModelConfig } from "../types.js";
+import type { Logger, GatewayConfig, ModelConfig, Usage } from "../types.js";
 import type { ToolManager } from "./toolManager.js";
 import type { MessageManager } from "./messageManager.js";
 import type { BackgroundBashManager } from "./backgroundBashManager.js";
 import type { HookManager } from "../hooks/index.js";
 export interface AIManagerCallbacks {
     onCompressionStateChange?: (isCompressing: boolean) => void;
+    onUsageAdded?: (usage: Usage) => void;
 }
 export interface AIManagerOptions {
     messageManager: MessageManager;

package/dist/managers/aiManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"aiManager.d.ts","sourceRoot":"","sources":["../../src/managers/aiManager.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AACtE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAEpD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAExE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGrD,MAAM,WAAW,kBAAkB;IACjC,wBAAwB,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;CAC7D;AAED,MAAM,WAAW,gBAAgB;IAC/B,cAAc,EAAE,cAAc,CAAC;IAC/B,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;IAC9C,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,kBAAkB,CAAC;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,SAAS;IACb,SAAS,EAAE,OAAO,CAAS;IAClC,OAAO,CAAC,eAAe,CAAgC;IACvD,OAAO,CAAC,mBAAmB,CAAgC;IAC3D,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,qBAAqB,CAAC,CAAwB;IACtD,OAAO,CAAC,WAAW,CAAC,CAAc;IAClC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAS;IAG9B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAS;gBAEf,OAAO,EAAE,gBAAgB;IAgBrC,OAAO,CAAC,aAAa,CAAkB;IACvC,OAAO,CAAC,SAAS,CAAqB;IAEtC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAYvB,YAAY,CAAC,SAAS,EAAE,OAAO,GAAG,IAAI;IAItC,cAAc,IAAI,IAAI;IAuB7B,OAAO,CAAC,qBAAqB;YAqBf,8BAA8B;IAoDrC,gBAAgB,IAAI,OAAO;IAI3B,gBAAgB,CAAC,aAAa,EAAE,OAAO,GAAG,IAAI;IAOxC,aAAa,CACxB,OAAO,GAAE;QACP,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;KACpB,GACL,OAAO,CAAC,IAAI,CAAC;IA+PhB;;OAEG;YACW,gBAAgB;IAmC9B;;OAEG;YACW,sBAAsB;IA0CpC;;OAEG;YACW,uBAAuB;CA2CtC"}
+{"version":3,"file":"aiManager.d.ts","sourceRoot":"","sources":["../../src/managers/aiManager.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAC7E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAEpD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAExE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGrD,MAAM,WAAW,kBAAkB;IACjC,wBAAwB,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;IAC5D,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACvC;AAED,MAAM,WAAW,gBAAgB;IAC/B,cAAc,EAAE,cAAc,CAAC;IAC/B,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;IAC9C,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,kBAAkB,CAAC;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,SAAS;IACb,SAAS,EAAE,OAAO,CAAS;IAClC,OAAO,CAAC,eAAe,CAAgC;IACvD,OAAO,CAAC,mBAAmB,CAAgC;IAC3D,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,qBAAqB,CAAC,CAAwB;IACtD,OAAO,CAAC,WAAW,CAAC,CAAc;IAClC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAS;IAG9B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAS;gBAEf,OAAO,EAAE,gBAAgB;IAgBrC,OAAO,CAAC,aAAa,CAAkB;IACvC,OAAO,CAAC,SAAS,CAAqB;IAEtC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAYvB,YAAY,CAAC,SAAS,EAAE,OAAO,GAAG,IAAI;IAItC,cAAc,IAAI,IAAI;IAuB7B,OAAO,CAAC,qBAAqB;YAqBf,8BAA8B;IAoErC,gBAAgB,IAAI,OAAO;IAI3B,gBAAgB,CAAC,aAAa,EAAE,OAAO,GAAG,IAAI;IAOxC,aAAa,CACxB,OAAO,GAAE;QACP,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;KACpB,GACL,OAAO,CAAC,IAAI,CAAC;IAkRhB;;OAEG;YACW,gBAAgB;IAmC9B;;OAEG;YACW,sBAAsB;IA0CpC;;OAEG;YACW,uBAAuB;CA2CtC"}

package/dist/managers/aiManager.js

@@ -83,7 +83,7 @@ export class AIManager {
         this.messageManager.setlatestTotalTokens(usage.total_tokens);
         // Check if token limit exceeded - use injected configuration
         if (usage.total_tokens > this.tokenLimit) {
-            this.logger?.info(`Token usage exceeded ${this.tokenLimit}, compressing messages...`);
+            this.logger?.debug(`Token usage exceeded ${this.tokenLimit}, compressing messages...`);
             // Check if messages need compression
             const { messagesToCompress, insertIndex } = getMessagesToCompress(this.messageManager.getMessages(), 7);
             // If there are messages to compress, perform compression
@@ -91,15 +91,29 @@ export class AIManager {
                 const recentChatMessages = convertMessagesForAPI(messagesToCompress);
                 this.setIsCompressing(true);
                 try {
-                    const compressedContent = await compressMessages({
+                    const compressionResult = await compressMessages({
                         gatewayConfig: this.gatewayConfig,
                         modelConfig: this.modelConfig,
                         messages: recentChatMessages,
                         abortSignal: abortController.signal,
                     });
                     // Execute message reconstruction and sessionId update after compression
-                    this.messageManager.compressMessagesAndUpdateSession(insertIndex, compressedContent);
-                    this.logger?.info(`Successfully compressed ${messagesToCompress.length} messages and updated session`);
+                    this.messageManager.compressMessagesAndUpdateSession(insertIndex, compressionResult.content);
+                    // Handle usage tracking for compression operations
+                    if (compressionResult.usage) {
+                        const usage = {
+                            prompt_tokens: compressionResult.usage.prompt_tokens,
+                            completion_tokens: compressionResult.usage.completion_tokens,
+                            total_tokens: compressionResult.usage.total_tokens,
+                            model: this.modelConfig.fastModel,
+                            operation_type: "compress",
+                        };
+                        // Notify Agent to add to usage tracking
+                        if (this.callbacks?.onUsageAdded) {
+                            this.callbacks.onUsageAdded(usage);
+                        }
+                    }
+                    this.logger?.debug(`Successfully compressed ${messagesToCompress.length} messages and updated session`);
                 }
                 catch (compressError) {
                     this.logger?.error("Failed to compress messages:", compressError);
@@ -163,8 +177,25 @@ export class AIManager {
                     }
                 }
             }
-            // Add assistant message at once (including content and tool calls)
-            this.messageManager.addAssistantMessage(content, toolCalls);
+            // Handle usage tracking for agent operations
+            let usage;
+            if (result.usage) {
+                usage = {
+                    prompt_tokens: result.usage.prompt_tokens,
+                    completion_tokens: result.usage.completion_tokens,
+                    total_tokens: result.usage.total_tokens,
+                    model: model || this.modelConfig.agentModel,
+                    operation_type: "agent",
+                };
+            }
+            // Add assistant message at once (including content, tool calls, and usage)
+            this.messageManager.addAssistantMessage(content, toolCalls, usage);
+            // Notify Agent to add to usage tracking
+            if (usage) {
+                if (this.callbacks?.onUsageAdded) {
+                    this.callbacks.onUsageAdded(usage);
+                }
+            }
             if (toolCalls.length > 0) {
                 for (const functionToolCall of toolCalls) {
                     const toolId = functionToolCall.id || "";

package/dist/managers/mcpManager.js

@@ -20,17 +20,17 @@ export class McpManager {
         this.configPath = join(workdir, ".mcp.json");
         this.workdir = workdir;
         if (autoConnect) {
-            this.logger?.info("Initializing MCP servers...");
+            this.logger?.debug("Initializing MCP servers...");
             // Ensure MCP configuration is loaded
             const config = await this.ensureConfigLoaded();
             if (config && config.mcpServers) {
                 // Connect to all configured servers
                 const connectionPromises = Object.keys(config.mcpServers).map(async (serverName) => {
                     try {
-                        this.logger?.info(`Connecting to MCP server: ${serverName}`);
+                        this.logger?.debug(`Connecting to MCP server: ${serverName}`);
                         const success = await this.connectServer(serverName);
                         if (success) {
-                            this.logger?.info(`Successfully connected to MCP server: ${serverName}`);
+                            this.logger?.debug(`Successfully connected to MCP server: ${serverName}`);
                         }
                         else {
                             this.logger?.warn(`Failed to connect to MCP server: ${serverName}`);
@@ -43,7 +43,7 @@ export class McpManager {
                 // Wait for all connection attempts to complete
                 await Promise.all(connectionPromises);
             }
-            this.logger?.info("MCP servers initialization completed");
+            this.logger?.debug("MCP servers initialization completed");
             // Trigger state change callback after initialization
             this.callbacks.onServersChange?.(this.getAllServers());
         }
@@ -188,7 +188,7 @@ export class McpManager {
                 });
             };
             transport.onclose = () => {
-                this.logger?.info(`MCP Server ${name} transport closed`);
+                this.logger?.debug(`MCP Server ${name} transport closed`);
                 this.connections.delete(name);
                 this.updateServerStatus(name, {
                     status: "disconnected",

package/dist/managers/messageManager.d.ts

@@ -1,11 +1,12 @@
 import { type AgentToolBlockUpdateParams } from "../utils/messageOperations.js";
-import type { Logger, Message } from "../types.js";
+import type { Logger, Message, Usage } from "../types.js";
 import { ChatCompletionMessageFunctionToolCall } from "openai/resources.js";
 export interface MessageManagerCallbacks {
     onMessagesChange?: (messages: Message[]) => void;
     onSessionIdChange?: (sessionId: string) => void;
     onLatestTotalTokensChange?: (latestTotalTokens: number) => void;
     onUserInputHistoryChange?: (history: string[]) => void;
+    onUsagesChange?: (usages: Usage[]) => void;
     onUserMessageAdded?: (content: string, images?: Array<{
         path: string;
         mimeType: string;
@@ -28,6 +29,11 @@ export interface MessageManagerOptions {
     callbacks: MessageManagerCallbacks;
     workdir: string;
     logger?: Logger;
+    /**
+     * Custom session directory path
+     * @default join(homedir(), ".wave", "sessions")
+     */
+    sessionDir?: string;
 }
 export declare class MessageManager {
     private sessionId;
@@ -38,6 +44,7 @@ export declare class MessageManager {
     private workdir;
     private logger?;
     private callbacks;
+    private sessionDir?;
     constructor(options: MessageManagerOptions);
     getSessionId(): string;
     getMessages(): Message[];
@@ -68,7 +75,7 @@ export declare class MessageManager {
         mimeType: string;
     }>): void;
     addCustomCommandMessage(commandName: string, content: string, originalInput?: string): void;
-    addAssistantMessage(content?: string, toolCalls?: ChatCompletionMessageFunctionToolCall[]): void;
+    addAssistantMessage(content?: string, toolCalls?: ChatCompletionMessageFunctionToolCall[], usage?: Usage): void;
     updateToolBlock(params: AgentToolBlockUpdateParams): void;
     addDiffBlock(filePath: string, diffResult: Array<{
         value: string;
@@ -89,5 +96,9 @@ export declare class MessageManager {
         status: "active" | "completed" | "error" | "aborted";
         messages: Message[];
     }>): void;
+    /**
+     * Trigger usage change callback with all usage data from assistant messages
+     */
+    triggerUsageChange(): void;
 }
 //# sourceMappingURL=messageManager.d.ts.map

package/dist/managers/messageManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"messageManager.d.ts","sourceRoot":"","sources":["../../src/managers/messageManager.ts"],"names":[],"mappings":"AACA,OAAO,EAeL,KAAK,0BAA0B,EAChC,MAAM,+BAA+B,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AASnD,OAAO,EAAE,qCAAqC,EAAE,MAAM,qBAAqB,CAAC;AAE5E,MAAM,WAAW,uBAAuB;IACtC,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;IACjD,iBAAiB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IAChD,yBAAyB,CAAC,EAAE,CAAC,iBAAiB,EAAE,MAAM,KAAK,IAAI,CAAC;IAChE,wBAAwB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IAEvD,kBAAkB,CAAC,EAAE,CACnB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,KAC/C,IAAI,CAAC;IACV,uBAAuB,CAAC,EAAE,CACxB,OAAO,CAAC,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,qCAAqC,EAAE,KAChD,IAAI,CAAC;IACV,kBAAkB,CAAC,EAAE,CAAC,MAAM,EAAE,0BAA0B,KAAK,IAAI,CAAC;IAClE,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IAClE,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,oBAAoB,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACtE,wBAAwB,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;IAC5D,kBAAkB,CAAC,EAAE,CACnB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,OAAO,EAChB,IAAI,EAAE,SAAS,GAAG,MAAM,EACxB,WAAW,EAAE,MAAM,KAChB,IAAI,CAAC;IAEV,oBAAoB,CAAC,EAAE,CACrB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,EACf,aAAa,CAAC,EAAE,MAAM,KACnB,IAAI,CAAC;IAEV,yBAAyB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACtD,4BAA4B,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IACzE,wBAAwB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;IAEvE,oBAAoB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IACpD,sBAAsB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CAC5E;AAED,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,uBAAuB,CAAC;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,cAAc;IAEzB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,QAAQ,CAAY;IAC5B,OAAO,CAAC,iBAAiB,CAAS;IAClC,OAAO,CAAC,gBAAgB,CAAW;IACnC,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,SAAS,CAA0B;gBAE/B,OAAO,EAAE,qBAAqB;IAYnC,YAAY,IAAI,MAAM;IAItB,WAAW,IAAI,OAAO,EAAE;IAIxB,oBAAoB,IAAI,MAAM;IAI9B,mBAAmB,IAAI,MAAM,EAAE;IAI/B,iBAAiB,IAAI,MAAM;IAK3B,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAOrC,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,IAAI;IAK7C;;OAEG;IACU,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAczC;;OAEG;IACU,wBAAwB,CACnC,gBAAgB,CAAC,EAAE,MAAM,EACzB,mBAAmB,CAAC,EAAE,OAAO,GAC5B,OAAO,CAAC,IAAI,CAAC;IA+CT,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAOrD,mBAAmB,CAAC,gBAAgB,EAAE,MAAM,EAAE,GAAG,IAAI;IAK5D;;OAEG;IACI,aAAa,IAAI,IAAI;IASrB,qBAAqB,CAC1B,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,iBAAiB,EAAE,MAAM,GACxB,IAAI;IAUA,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAatC,iBAAiB,IAAI,IAAI;IAKzB,cAAc,CACnB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,GACjD,IAAI;IAUA,uBAAuB,CAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,EACf,aAAa,CAAC,EAAE,MAAM,GACrB,IAAI;IAeA,mBAAmB,CACxB,OAAO,CAAC,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,qCAAqC,EAAE,GAClD,IAAI;IAUA,eAAe,CAAC,MAAM,EAAE,0BAA0B,GAAG,IAAI;IAiBzD,YAAY,CACjB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,GACvE,IAAI;IAUA,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IASzC;;OAEG;IACI,gCAAgC,CACrC,WAAW,EAAE,MAAM,EACnB,iBAAiB,EAAE,MAAM,GACxB,IAAI;IAkCA,cAAc,CACnB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,OAAO,EAChB,IAAI,EAAE,SAAS,GAAG,MAAM,EACxB,WAAW,EAAE,MAAM,GAClB,IAAI;IAaA,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAS9C,0BAA0B,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAUjE,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI;IAW/D,gBAAgB,CACrB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,EACpB,MAAM,GAAE,QAAQ,GAAG,WAAW,GAAG,OAAkB,EACnD,gBAAgB,GAAE,OAAO,EAAO,GAC/B,IAAI;IAaA,mBAAmB,CACxB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,OAAO,CAAC;QACf,MAAM,EAAE,QAAQ,GAAG,WAAW,GAAG,OAAO,GAAG,SAAS,CAAC;QACrD,QAAQ,EAAE,OAAO,EAAE,CAAC;KACrB,CAAC,GACD,IAAI;CAeR"}
+{"version":3,"file":"messageManager.d.ts","sourceRoot":"","sources":["../../src/managers/messageManager.ts"],"names":[],"mappings":"AACA,OAAO,EAeL,KAAK,0BAA0B,EAChC,MAAM,+BAA+B,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAS1D,OAAO,EAAE,qCAAqC,EAAE,MAAM,qBAAqB,CAAC;AAE5E,MAAM,WAAW,uBAAuB;IACtC,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;IACjD,iBAAiB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IAChD,yBAAyB,CAAC,EAAE,CAAC,iBAAiB,EAAE,MAAM,KAAK,IAAI,CAAC;IAChE,wBAAwB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IACvD,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,IAAI,CAAC;IAE3C,kBAAkB,CAAC,EAAE,CACnB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,KAC/C,IAAI,CAAC;IACV,uBAAuB,CAAC,EAAE,CACxB,OAAO,CAAC,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,qCAAqC,EAAE,KAChD,IAAI,CAAC;IACV,kBAAkB,CAAC,EAAE,CAAC,MAAM,EAAE,0BAA0B,KAAK,IAAI,CAAC;IAClE,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IAClE,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,oBAAoB,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACtE,wBAAwB,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;IAC5D,kBAAkB,CAAC,EAAE,CACnB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,OAAO,EAChB,IAAI,EAAE,SAAS,GAAG,MAAM,EACxB,WAAW,EAAE,MAAM,KAChB,IAAI,CAAC;IAEV,oBAAoB,CAAC,EAAE,CACrB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,EACf,aAAa,CAAC,EAAE,MAAM,KACnB,IAAI,CAAC;IAEV,yBAAyB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACtD,4BAA4B,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IACzE,wBAAwB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;IAEvE,oBAAoB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IACpD,sBAAsB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CAC5E;AAED,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,uBAAuB,CAAC;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAGhB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,cAAc;IAEzB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,QAAQ,CAAY;IAC5B,OAAO,CAAC,iBAAiB,CAAS;IAClC,OAAO,CAAC,gBAAgB,CAAW;IACnC,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,SAAS,CAA0B;IAC3C,OAAO,CAAC,UAAU,CAAC,CAAS;gBAEhB,OAAO,EAAE,qBAAqB;IAanC,YAAY,IAAI,MAAM;IAItB,WAAW,IAAI,OAAO,EAAE;IAIxB,oBAAoB,IAAI,MAAM;IAI9B,mBAAmB,IAAI,MAAM,EAAE;IAI/B,iBAAiB,IAAI,MAAM;IAK3B,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAOrC,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,IAAI;IAK7C;;OAEG;IACU,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAezC;;OAEG;IACU,wBAAwB,CACnC,gBAAgB,CAAC,EAAE,MAAM,EACzB,mBAAmB,CAAC,EAAE,OAAO,GAC5B,OAAO,CAAC,IAAI,CAAC;IAkDT,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAOrD,mBAAmB,CAAC,gBAAgB,EAAE,MAAM,EAAE,GAAG,IAAI;IAK5D;;OAEG;IACI,aAAa,IAAI,IAAI;IASrB,qBAAqB,CAC1B,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,iBAAiB,EAAE,MAAM,GACxB,IAAI;IAUA,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAatC,iBAAiB,IAAI,IAAI;IAKzB,cAAc,CACnB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,GACjD,IAAI;IAUA,uBAAuB,CAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,EACf,aAAa,CAAC,EAAE,MAAM,GACrB,IAAI;IAeA,mBAAmB,CACxB,OAAO,CAAC,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,qCAAqC,EAAE,EACnD,KAAK,CAAC,EAAE,KAAK,GACZ,IAAI;IAWA,eAAe,CAAC,MAAM,EAAE,0BAA0B,GAAG,IAAI;IAiBzD,YAAY,CACjB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,GACvE,IAAI;IAUA,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IASzC;;OAEG;IACI,gCAAgC,CACrC,WAAW,EAAE,MAAM,EACnB,iBAAiB,EAAE,MAAM,GACxB,IAAI;IAkCA,cAAc,CACnB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,OAAO,EAChB,IAAI,EAAE,SAAS,GAAG,MAAM,EACxB,WAAW,EAAE,MAAM,GAClB,IAAI;IAaA,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAS9C,0BAA0B,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAUjE,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI;IAW/D,gBAAgB,CACrB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,EACpB,MAAM,GAAE,QAAQ,GAAG,WAAW,GAAG,OAAkB,EACnD,gBAAgB,GAAE,OAAO,EAAO,GAC/B,IAAI;IAaA,mBAAmB,CACxB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,OAAO,CAAC;QACf,MAAM,EAAE,QAAQ,GAAG,WAAW,GAAG,OAAO,GAAG,SAAS,CAAC;QACrD,QAAQ,EAAE,OAAO,EAAE,CAAC;KACrB,CAAC,GACD,IAAI;IAgBP;;OAEG;IACI,kBAAkB,IAAI,IAAI;CASlC"}

package/dist/managers/messageManager.js

@@ -11,6 +11,7 @@ export class MessageManager {
         this.workdir = options.workdir;
         this.callbacks = options.callbacks;
         this.logger = options.logger;
+        this.sessionDir = options.sessionDir;
     }
     // Getter methods
     getSessionId() {
@@ -26,7 +27,7 @@ export class MessageManager {
         return [...this.userInputHistory];
     }
     getTranscriptPath() {
-        return getSessionFilePath(this.sessionId);
+        return getSessionFilePath(this.sessionId, this.sessionDir);
     }
     // Setter methods, will trigger callbacks
     setSessionId(sessionId) {
@@ -44,7 +45,7 @@ export class MessageManager {
      */
     async saveSession() {
         try {
-            await saveSession(this.sessionId, this.messages, this.workdir, this.latestTotalTokens, this.sessionStartTime);
+            await saveSession(this.sessionId, this.messages, this.workdir, this.latestTotalTokens, this.sessionStartTime, this.sessionDir);
         }
         catch (error) {
             this.logger?.error("Failed to save session:", error);
@@ -56,7 +57,7 @@ export class MessageManager {
     async handleSessionRestoration(restoreSessionId, continueLastSession) {
         // Clean up expired sessions first
         try {
-            await cleanupExpiredSessions(this.workdir);
+            await cleanupExpiredSessions(this.workdir, this.sessionDir);
         }
         catch (error) {
             console.warn("Failed to cleanup expired sessions:", error);
@@ -67,14 +68,14 @@ export class MessageManager {
         try {
             let sessionToRestore = null;
             if (restoreSessionId) {
-                sessionToRestore = await loadSession(restoreSessionId);
+                sessionToRestore = await loadSession(restoreSessionId, this.sessionDir);
                 if (!sessionToRestore) {
                     console.error(`Session not found: ${restoreSessionId}`);
                     process.exit(1);
                 }
             }
             else if (continueLastSession) {
-                sessionToRestore = await getLatestSession(this.workdir);
+                sessionToRestore = await getLatestSession(this.workdir, this.sessionDir);
                 if (!sessionToRestore) {
                     console.error(`No previous session found for workdir: ${this.workdir}`);
                     process.exit(1);
@@ -157,8 +158,8 @@ export class MessageManager {
         this.setMessages(newMessages);
         this.callbacks.onCustomCommandAdded?.(commandName, content, originalInput);
     }
-    addAssistantMessage(content, toolCalls) {
-        const newMessages = addAssistantMessageToMessages(this.messages, content, toolCalls);
+    addAssistantMessage(content, toolCalls, usage) {
+        const newMessages = addAssistantMessageToMessages(this.messages, content, toolCalls, usage);
         this.setMessages(newMessages);
         this.callbacks.onAssistantMessageAdded?.(content, toolCalls);
     }
@@ -286,4 +287,16 @@ export class MessageManager {
         };
         this.callbacks.onSubAgentBlockUpdated?.(params.subagentId, params.messages);
     }
+    /**
+     * Trigger usage change callback with all usage data from assistant messages
+     */
+    triggerUsageChange() {
+        const usages = [];
+        for (const message of this.messages) {
+            if (message.role === "assistant" && message.usage) {
+                usages.push(message.usage);
+            }
+        }
+        this.callbacks.onUsagesChange?.(usages);
+    }
 }

package/dist/managers/skillManager.js

@@ -19,7 +19,7 @@ export class SkillManager {
      * Initialize the skill manager by discovering available skills
      */
     async initialize() {
-        this.logger?.info("Initializing SkillManager...");
+        this.logger?.debug("Initializing SkillManager...");
         try {
             // Clear existing data before discovery
             this.skillMetadata.clear();
@@ -40,7 +40,7 @@ export class SkillManager {
                 });
             }
             this.initialized = true;
-            this.logger?.info(`SkillManager initialized with ${this.skillMetadata.size} skills`);
+            this.logger?.debug(`SkillManager initialized with ${this.skillMetadata.size} skills`);
         }
         catch (error) {
             this.logger?.error("Failed to initialize SkillManager:", error);
@@ -184,7 +184,7 @@ export class SkillManager {
      */
     async executeSkill(args) {
         const { skill_name } = args;
-        this.logger?.info(`Invoking skill: ${skill_name}`);
+        this.logger?.debug(`Invoking skill: ${skill_name}`);
         try {
             // Load the skill
             const skill = await this.loadSkill(skill_name);

package/dist/managers/slashCommandManager.js

@@ -51,7 +51,7 @@ export class SlashCommandManager {
                     },
                 });
             }
-            this.logger?.info(`Loaded ${customCommands.length} custom commands`);
+            this.logger?.debug(`Loaded ${customCommands.length} custom commands`);
         }
         catch (error) {
             this.logger?.warn("Failed to load custom commands:", error);