wave-agent-sdk 0.7.1 → 0.8.0

package/src/index.ts

@@ -22,6 +22,6 @@ export * from "./utils/stringUtils.js";
 export * from "./utils/customCommands.js";
 export * from "./utils/hookMatcher.js";
 export * from "./utils/tokenCalculation.js";
-
-// Export types
+export * from "./utils/gitUtils.js";
+export * from "./utils/nameGenerator.js";
 export * from "./types/index.js";

package/src/managers/aiManager.ts

@@ -132,7 +132,17 @@ export class AIManager {
    * Get filtered tool configuration based on tools list
    */
   private getFilteredToolsConfig(tools?: string[]) {
-    const allTools = this.toolManager.getToolsConfig();
+    // Get available subagents and skills for dynamic prompts
+    const availableSubagents = this.subagentManager?.getConfigurations();
+    const availableSkills = this.skillManager
+      ?.getAvailableSkills()
+      .filter((skill) => !skill.disableModelInvocation);
+
+    const allTools = this.toolManager.getToolsConfig({
+      availableSubagents,
+      availableSkills,
+      workdir: this.workdir,
+    });
 
     // If no tools specified, return all tools
     if (!tools || tools.length === 0) {
@@ -241,7 +251,7 @@ export class AIManager {
               prompt_tokens: compressionResult.usage.prompt_tokens,
               completion_tokens: compressionResult.usage.completion_tokens,
               total_tokens: compressionResult.usage.total_tokens,
-              model: model || this.getModelConfig().agentModel,
+              model: model || this.getModelConfig().model,
               operation_type: "compress",
             };
           }
@@ -383,10 +393,6 @@ export class AIManager {
         }
       }
 
-      // Get available subagents and skills for dynamic prompts
-      const availableSubagents = this.subagentManager?.getConfigurations();
-      const availableSkills = this.skillManager?.getAvailableSkills();
-
       // Call AI service with streaming callbacks if enabled
       const callAgentOptions: CallAgentOptions = {
         gatewayConfig: this.getGatewayConfig(),
@@ -406,8 +412,6 @@ export class AIManager {
             language: this.getLanguage(),
             isSubagent: !!this.subagentType,
             planMode: planModeOptions,
-            availableSubagents,
-            availableSkills,
           },
         ), // Pass custom system prompt
         maxTokens: maxTokens, // Pass max tokens override
@@ -512,7 +516,7 @@ export class AIManager {
           prompt_tokens: result.usage.prompt_tokens,
           completion_tokens: result.usage.completion_tokens,
           total_tokens: result.usage.total_tokens,
-          model: model || this.getModelConfig().agentModel,
+          model: model || this.getModelConfig().model,
           operation_type: "agent",
           // Preserve cache fields if present
           ...(result.usage.cache_read_input_tokens !== undefined && {
@@ -553,12 +557,6 @@ export class AIManager {
         }
       }
 
-      if (result.finish_reason === "length" && toolCalls.length === 0) {
-        this.messageManager.addErrorBlock(
-          "AI response was truncated due to length limit. Please try to reduce the complexity of your request or split it into smaller parts.",
-        );
-      }
-
       if (toolCalls.length > 0) {
         // Execute all tools in parallel using Promise.all
         const toolExecutionPromises = toolCalls.map(
@@ -716,8 +714,8 @@ export class AIManager {
         model,
       );
 
-      // Check if there are tool operations, if so automatically initiate next AI service call
-      if (toolCalls.length > 0) {
+      // Check if there are tool operations or response was truncated, if so automatically initiate next AI service call
+      if (toolCalls.length > 0 || result.finish_reason === "length") {
         // Record committed snapshots to message history
         if (this.reversionManager) {
           const snapshots =
@@ -750,6 +748,14 @@ export class AIManager {
             "Some tools were manually backgrounded, stopping recursion.",
           );
         } else if (!isCurrentlyAborted) {
+          // If response was truncated and no tools were called, add a continuation message
+          if (result.finish_reason === "length" && toolCalls.length === 0) {
+            this.messageManager.addUserMessage({
+              content:
+                "Your response was cut off because it exceeded the output token limit. Please break your work into smaller pieces. Continue from where you left off.",
+            });
+          }
+
           // Recursively call AI service, increment recursion depth, and pass same configuration
           await this.sendAIMessage({
             recursionDepth: recursionDepth + 1,

package/src/managers/backgroundTaskManager.ts

@@ -5,7 +5,7 @@ import { logger } from "../utils/globalLogger.js";
 import { Container } from "../utils/container.js";
 
 export interface BackgroundTaskManagerCallbacks {
-  onTasksChange?: (tasks: BackgroundTask[]) => void;
+  onBackgroundTasksChange?: (tasks: BackgroundTask[]) => void;
 }
 
 export interface BackgroundTaskManagerOptions {
@@ -28,7 +28,7 @@ export class BackgroundTaskManager {
   }
 
   private notifyTasksChange(): void {
-    this.callbacks.onTasksChange?.(Array.from(this.tasks.values()));
+    this.callbacks.onBackgroundTasksChange?.(Array.from(this.tasks.values()));
   }
 
   public generateId(): string {

package/src/managers/{bashManager.ts → bangManager.ts}

@@ -2,7 +2,7 @@ import { spawn, type ChildProcess } from "child_process";
 import type { MessageManager } from "./messageManager.js";
 import { Container } from "../utils/container.js";
 
-export interface BashManagerOptions {
+export interface BangManagerOptions {
   workdir: string;
 }
 
@@ -11,14 +11,14 @@ export interface CommandExecutionResult {
   output: string;
 }
 
-export class BashManager {
+export class BangManager {
   private workdir: string;
   public isCommandRunning = false;
   private currentProcess: ChildProcess | null = null;
 
   constructor(
     private container: Container,
-    options: BashManagerOptions,
+    options: BangManagerOptions,
   ) {
     this.workdir = options.workdir;
   }
@@ -38,8 +38,8 @@ export class BashManager {
 
     this.setCommandRunning(true);
 
-    // Add command output placeholder
-    this.messageManager.addCommandOutputMessage(command);
+    // Add bang placeholder
+    this.messageManager.addBangMessage(command);
 
     return new Promise<number>((resolve) => {
       const child = spawn(command, {
@@ -56,7 +56,6 @@ export class BashManager {
 
       const updateOutput = (newData: string) => {
         outputBuffer += newData;
-        this.messageManager.updateCommandOutputMessage(command, outputBuffer);
       };
 
       child.stdout?.on("data", (data) => {
@@ -70,7 +69,11 @@ export class BashManager {
       child.on("exit", (code, signal) => {
         const exitCode = code === null && signal ? 130 : (code ?? 0);
 
-        this.messageManager.completeCommandMessage(command, exitCode);
+        this.messageManager.completeBangMessage(
+          command,
+          exitCode,
+          outputBuffer,
+        );
 
         this.setCommandRunning(false);
         this.currentProcess = null;
@@ -79,7 +82,7 @@ export class BashManager {
 
       child.on("error", (error) => {
         updateOutput(`\nError: ${error.message}\n`);
-        this.messageManager.completeCommandMessage(command, 1);
+        this.messageManager.completeBangMessage(command, 1, outputBuffer);
 
         this.setCommandRunning(false);
         this.currentProcess = null;

package/src/managers/hookManager.ts

@@ -378,6 +378,11 @@ export class HookManager {
         });
         return { shouldBlock: true, errorMessage };
 
+      case "WorktreeCreate":
+        // Non-blocking for now, just show error in error block
+        messageManager.addErrorBlock(errorMessage);
+        return { shouldBlock: false };
+
       default:
         return { shouldBlock: false };
     }
@@ -577,7 +582,8 @@ export class HookManager {
       (event === "UserPromptSubmit" ||
         event === "Stop" ||
         event === "Notification" ||
-        event === "SubagentStop") &&
+        event === "SubagentStop" ||
+        event === "WorktreeCreate") &&
       context.toolName !== undefined
     ) {
       logger?.warn(
@@ -653,7 +659,8 @@ export class HookManager {
       event === "UserPromptSubmit" ||
       event === "Stop" ||
       event === "Notification" ||
-      event === "SubagentStop"
+      event === "SubagentStop" ||
+      event === "WorktreeCreate"
     ) {
       return true;
     }
@@ -704,7 +711,8 @@ export class HookManager {
       (event === "UserPromptSubmit" ||
         event === "Stop" ||
         event === "Notification" ||
-        event === "SubagentStop") &&
+        event === "SubagentStop" ||
+        event === "WorktreeCreate") &&
       config.matcher
     ) {
       errors.push(`${prefix}: Event ${event} should not have a matcher`);
@@ -743,6 +751,7 @@ export class HookManager {
           Stop: 0,
           SubagentStop: 0,
           Notification: 0,
+          WorktreeCreate: 0,
         },
       };
     }
@@ -754,6 +763,7 @@ export class HookManager {
       Stop: 0,
       SubagentStop: 0,
       Notification: 0,
+      WorktreeCreate: 0,
     };
 
     let totalConfigs = 0;

package/src/managers/messageManager.ts

@@ -3,9 +3,9 @@ import {
   updateToolBlockInMessage,
   addErrorBlockToMessage,
   addUserMessageToMessages,
-  addCommandOutputMessage,
-  updateCommandOutputInMessage,
-  completeCommandInMessage,
+  addBangMessage,
+  updateBangInMessage,
+  completeBangInMessage,
   removeLastUserMessage,
   UserMessageParams,
   type AgentToolBlockUpdateParams,
@@ -48,10 +48,10 @@ export interface MessageManagerCallbacks {
     type: "project" | "user",
     storagePath: string,
   ) => void;
-  // Bash command callback
-  onAddCommandOutputMessage?: (command: string) => void;
-  onUpdateCommandOutputMessage?: (command: string, output: string) => void;
-  onCompleteCommandMessage?: (command: string, exitCode: number) => void;
+  // Bang callback
+  onAddBangMessage?: (command: string) => void;
+  onUpdateBangMessage?: (command: string, output: string) => void;
+  onCompleteBangMessage?: (command: string, exitCode: number) => void;
   onSlashCommandsChange?: (commands: SlashCommand[]) => void;
   onInfoBlockAdded?: (content: string) => void;
   // Rewind callbacks
@@ -85,6 +85,7 @@ export class MessageManager {
   private filesInContext: Set<string> = new Set(); // Track files mentioned in the conversation
   private sessionType: "main" | "subagent";
   private subagentType?: string;
+  private _usages: Usage[] = [];
 
   constructor(
     private container: Container,
@@ -128,6 +129,10 @@ export class MessageManager {
     return [...this.messages];
   }
 
+  public getUsages(): Usage[] {
+    return [...this._usages];
+  }
+
   public getlatestTotalTokens(): number {
     return this.latestTotalTokens;
   }
@@ -266,8 +271,9 @@ export class MessageManager {
    */
   public clearMessages(): void {
     this.setMessages([]);
-    this.setSessionId(generateSessionId());
-    this.rootSessionId = this.sessionId;
+    const newSessionId = generateSessionId();
+    this.rootSessionId = newSessionId;
+    this.setSessionId(newSessionId);
     this.setlatestTotalTokens(0);
     this.savedMessageCount = 0; // Reset saved message count
   }
@@ -463,47 +469,70 @@ export class MessageManager {
     }
   }
 
-  // Bash command related message operations
-  public addCommandOutputMessage(command: string): void {
-    const updatedMessages = addCommandOutputMessage({
+  // Bang related message operations
+  public addBangMessage(command: string): void {
+    const updatedMessages = addBangMessage({
       messages: this.messages,
       command,
     });
     this.setMessages(updatedMessages);
-    this.callbacks.onAddCommandOutputMessage?.(command);
+    this.callbacks.onAddBangMessage?.(command);
   }
 
-  public updateCommandOutputMessage(command: string, output: string): void {
-    const updatedMessages = updateCommandOutputInMessage({
+  public updateBangMessage(command: string, output: string): void {
+    const updatedMessages = updateBangInMessage({
       messages: this.messages,
       command,
       output,
     });
     this.setMessages(updatedMessages);
-    this.callbacks.onUpdateCommandOutputMessage?.(command, output);
+    this.callbacks.onUpdateBangMessage?.(command, output);
   }
 
-  public completeCommandMessage(command: string, exitCode: number): void {
-    const updatedMessages = completeCommandInMessage({
+  public completeBangMessage(
+    command: string,
+    exitCode: number,
+    output?: string,
+  ): void {
+    const updatedMessages = completeBangInMessage({
       messages: this.messages,
       command,
       exitCode,
+      output,
     });
     this.setMessages(updatedMessages);
-    this.callbacks.onCompleteCommandMessage?.(command, exitCode);
+    this.callbacks.onCompleteBangMessage?.(command, exitCode);
   }
 
   /**
-   * Trigger usage change callback with all usage data from assistant messages
+   * Rebuild usage array from messages containing usage metadata
+   * Called during session restoration to reconstruct usage tracking
    */
-  public triggerUsageChange(): void {
-    const usages: Usage[] = [];
-    for (const message of this.messages) {
+  public rebuildUsageFromMessages(messages: Message[]): void {
+    this._usages = [];
+    messages.forEach((message) => {
       if (message.role === "assistant" && message.usage) {
-        usages.push(message.usage);
+        this._usages.push(message.usage);
       }
-    }
-    this.callbacks.onUsagesChange?.(usages);
+    });
+    // Trigger callback after rebuilding usage array
+    this.triggerUsageChange();
+  }
+
+  /**
+   * Add usage data to the tracking array and trigger callbacks
+   * @param usage Usage data from AI operations
+   */
+  public addUsage(usage: Usage): void {
+    this._usages.push(usage);
+    this.triggerUsageChange();
+  }
+
+  /**
+   * Trigger usage change callback with all usage data from assistant messages
+   */
+  public triggerUsageChange(): void {
+    this.callbacks.onUsagesChange?.([...this._usages]);
   }
 
   /**