wave-agent-sdk 0.5.0 → 0.6.1

package/dist/utils/messageOperations.js

@@ -1,25 +1,6 @@
-import { MessageSource } from "../types/index.js";
-import { DEFAULT_KEEP_LAST_MESSAGES_COUNT } from "./constants.js";
 import { readFileSync } from "fs";
 import { extname } from "path";
 import { logger } from "./globalLogger.js";
-/**
- * Extract text content from user messages in the messages array
- * Excludes messages with source HOOK to prevent hook-generated content from entering user history
- */
-export const extractUserInputHistory = (messages) => {
-    return messages
-        .filter((message) => message.role === "user")
-        .map((message) => {
-        // Extract text block content, excluding HOOK-sourced blocks
-        const textBlocks = message.blocks.filter((block) => block.type === "text" && block.source !== MessageSource.HOOK);
-        return textBlocks
-            .map((block) => block.content)
-            .join(" ")
-            .trim();
-    })
-        .filter((text) => text.length > 0); // Filter out empty text
-};
 /**
  * Convert image file path to base64 format
  * @param imagePath Image file path
@@ -116,7 +97,7 @@ export const addAssistantMessageToMessages = (messages, content, toolCalls, usag
     return [...messages, initialAssistantMessage];
 };
 // Update Tool Block of the last assistant message
-export const updateToolBlockInMessage = ({ messages, id, parameters, result, success, error, stage, name, shortResult, images, compactParams, parametersChunk, }) => {
+export const updateToolBlockInMessage = ({ messages, id, parameters, result, success, error, stage, name, shortResult, images, compactParams, parametersChunk, isManuallyBackgrounded, }) => {
     const newMessages = [...messages];
     // Find the last assistant message
     for (let i = newMessages.length - 1; i >= 0; i--) {
@@ -141,6 +122,8 @@ export const updateToolBlockInMessage = ({ messages, id, parameters, result, suc
                         toolBlock.compactParams = compactParams;
                     if (parametersChunk !== undefined)
                         toolBlock.parametersChunk = parametersChunk;
+                    if (isManuallyBackgrounded !== undefined)
+                        toolBlock.isManuallyBackgrounded = isManuallyBackgrounded;
                 }
             }
             else {
@@ -159,6 +142,7 @@ export const updateToolBlockInMessage = ({ messages, id, parameters, result, suc
                     stage: stage ?? "start",
                     compactParams: compactParams,
                     parametersChunk: parametersChunk,
+                    isManuallyBackgrounded: isManuallyBackgrounded,
                 });
             }
             break;
@@ -198,83 +182,6 @@ export const addErrorBlockToMessage = ({ messages, error, }) => {
     }
     return newMessages;
 };
-// Add Memory Block as new assistant message
-export const addMemoryBlockToMessage = ({ messages, content, isSuccess, memoryType, storagePath, }) => {
-    const newMessages = [...messages];
-    // Create new assistant message containing MemoryBlock
-    const memoryMessage = {
-        role: "assistant",
-        blocks: [
-            {
-                type: "memory",
-                content,
-                isSuccess,
-                memoryType,
-                storagePath,
-            },
-        ],
-    };
-    // Add to end of message list
-    newMessages.push(memoryMessage);
-    return newMessages;
-};
-/**
- * Count valid blocks from the end
- * Only text, image, and tool type blocks are counted
- * @param messages Message array
- * @param targetCount Number of valid blocks to count
- * @returns { messageIndex: number, blockCount: number } Message index and actual counted block count
- */
-export const countValidBlocksFromEnd = (messages, targetCount) => {
-    let validBlockCount = 0;
-    // Iterate messages from end to beginning
-    for (let i = messages.length - 1; i >= 0; i--) {
-        const message = messages[i];
-        // Iterate through all blocks of current message
-        for (const block of message.blocks) {
-            // Only count valid block types
-            if (block.type === "text" ||
-                block.type === "image" ||
-                block.type === "tool") {
-                validBlockCount++;
-                // If target count reached, return current message index
-                if (validBlockCount >= targetCount) {
-                    return { messageIndex: i, blockCount: validBlockCount };
-                }
-            }
-        }
-    }
-    // If target count not reached, return index 0
-    return { messageIndex: 0, blockCount: validBlockCount };
-};
-/**
- * Get messages to be compressed and insertion position
- * @param messages Message array
- * @param keepLastCount Keep the last few valid blocks uncompressed
- * @returns { messagesToCompress: Message[], insertIndex: number }
- */
-export const getMessagesToCompress = (messages, keepLastCount = DEFAULT_KEEP_LAST_MESSAGES_COUNT) => {
-    // Calculate message position to keep from end to beginning
-    const { messageIndex } = countValidBlocksFromEnd(messages, keepLastCount);
-    // Find the last message containing compression block
-    let lastCompressIndex = -1;
-    for (let i = messages.length - 1; i >= 0; i--) {
-        const hasCompressBlock = messages[i].blocks.some((block) => block.type === "compress");
-        if (hasCompressBlock) {
-            lastCompressIndex = i;
-            break;
-        }
-    }
-    // Determine compression start position
-    // If compression block exists, start from compression block position (include compression block)
-    // If no compression block, start from beginning
-    const startIndex = lastCompressIndex >= 0 ? lastCompressIndex : 0;
-    // Messages to compress are all messages from start position to before calculated position
-    const messagesToCompress = messages.slice(startIndex, messageIndex);
-    // Change insertion position to negative number, indicating position from end
-    const insertIndex = messageIndex - messages.length;
-    return { messagesToCompress, insertIndex };
-};
 // Add command output block to message list
 export const addCommandOutputMessage = ({ messages, command, }) => {
     const outputMessage = {

package/dist/utils/nameGenerator.d.ts

@@ -4,5 +4,5 @@
 /**
  * Generates a random English name (adjective-noun)
  */
-export declare function generateRandomName(): string;
+export declare function generateRandomName(seed?: string): string;
 //# sourceMappingURL=nameGenerator.d.ts.map

package/dist/utils/nameGenerator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"nameGenerator.d.ts","sourceRoot":"","sources":["../../src/utils/nameGenerator.ts"],"names":[],"mappings":"AAAA;;GAEG;AAoEH;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,MAAM,CAI3C"}
+{"version":3,"file":"nameGenerator.d.ts","sourceRoot":"","sources":["../../src/utils/nameGenerator.ts"],"names":[],"mappings":"AAAA;;GAEG;AAoEH;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAqBxD"}

package/dist/utils/nameGenerator.js

@@ -68,8 +68,24 @@ const nouns = [
 /**
  * Generates a random English name (adjective-noun)
  */
-export function generateRandomName() {
-    const adj = adjectives[Math.floor(Math.random() * adjectives.length)];
-    const noun = nouns[Math.floor(Math.random() * nouns.length)];
+export function generateRandomName(seed) {
+    let adjIndex;
+    let nounIndex;
+    if (seed) {
+        // Simple hash function to derive indices from seed
+        let hash = 0;
+        for (let i = 0; i < seed.length; i++) {
+            hash = (hash << 5) - hash + seed.charCodeAt(i);
+            hash |= 0; // Convert to 32bit integer
+        }
+        adjIndex = Math.abs(hash) % adjectives.length;
+        nounIndex = Math.abs(hash >> 8) % nouns.length;
+    }
+    else {
+        adjIndex = Math.floor(Math.random() * adjectives.length);
+        nounIndex = Math.floor(Math.random() * nouns.length);
+    }
+    const adj = adjectives[adjIndex];
+    const noun = nouns[nounIndex];
     return `${adj}-${noun}`;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-agent-sdk",
-  "version": "0.5.0",
+  "version": "0.6.1",
   "description": "SDK for building AI-powered development tools and agents",
   "keywords": [
     "ai",

package/src/agent.ts

@@ -9,7 +9,6 @@ import {
   SubagentManager,
   type SubagentManagerCallbacks,
 } from "./managers/subagentManager.js";
-import * as memory from "./services/memory.js";
 import { McpManager, type McpManagerCallbacks } from "./managers/mcpManager.js";
 import { LspManager } from "./managers/lspManager.js";
 import { BashManager } from "./managers/bashManager.js";
@@ -46,6 +45,7 @@ import { MemoryRuleManager } from "./managers/MemoryRuleManager.js";
 import { LiveConfigManager } from "./managers/liveConfigManager.js";
 import { configValidator } from "./utils/configValidator.js";
 import { SkillManager } from "./managers/skillManager.js";
+import { TaskManager } from "./services/taskManager.js";
 import {
   loadSessionFromJsonl,
   handleSessionRestoration,
@@ -107,7 +107,12 @@ export interface AgentCallbacks
     McpManagerCallbacks,
     SubagentManagerCallbacks {
   onTasksChange?: (tasks: BackgroundTask[]) => void;
+  onSessionTasksChange?: (tasks: import("./types/tasks.js").Task[]) => void;
   onPermissionModeChange?: (mode: PermissionMode) => void;
+  onSubagentLatestTotalTokensChange?: (
+    subagentId: string,
+    tokens: number,
+  ) => void;
   onBackgroundCurrentTask?: () => void;
 }
 
@@ -131,6 +136,7 @@ export class Agent {
   private reversionManager: ReversionManager;
   private memoryRuleManager: MemoryRuleManager; // Add memory rule manager instance
   private liveConfigManager: LiveConfigManager; // Add live configuration manager
+  private taskManager: TaskManager;
   private foregroundTaskManager: ForegroundTaskManager;
   private configurationService: ConfigurationService; // Add configuration service
   private workdir: string; // Working directory
@@ -208,6 +214,46 @@ export class Agent {
 
     this.foregroundTaskManager = new ForegroundTaskManager();
 
+    // Initialize memory rule manager
+    this.memoryRuleManager = new MemoryRuleManager({
+      workdir: this.workdir,
+    });
+
+    // Initialize MessageManager
+    this.messageManager = new MessageManager({
+      callbacks: {
+        ...callbacks,
+        onSessionIdChange: (sessionId) => {
+          // When session ID changes (e.g. due to compression),
+          // we update the task manager to use the root session ID
+          // to ensure the task list remains consistent.
+          this.taskManager.setTaskListId(
+            this.messageManager.getRootSessionId(),
+          );
+          callbacks.onSessionIdChange?.(sessionId);
+        },
+        onSubagentTaskStopRequested: (subagentId) => {
+          this.backgroundTaskManager.stopTask(subagentId);
+        },
+      },
+      workdir: this.workdir,
+      logger: this.logger,
+      memoryRuleManager: this.memoryRuleManager,
+    });
+
+    // Resolve taskListId once during construction to ensure stability
+    const resolvedTaskListId =
+      this.configurationService.getEnvironmentVars().WAVE_TASK_LIST_ID ||
+      process.env.WAVE_TASK_LIST_ID ||
+      this.messageManager.getRootSessionId();
+
+    this.taskManager = new TaskManager(resolvedTaskListId);
+    this.taskManager.on("tasksChange", async () => {
+      const tasks = await this.taskManager.listTasks();
+      this.options.callbacks?.onSessionTasksChange?.(tasks);
+    });
+
+    // Initialize BackgroundTaskManager
     this.backgroundTaskManager = new BackgroundTaskManager({
       callbacks: {
         ...callbacks,
@@ -243,20 +289,7 @@ export class Agent {
       workdir: this.workdir,
     });
 
-    // Initialize memory rule manager
-    this.memoryRuleManager = new MemoryRuleManager({
-      workdir: this.workdir,
-    });
-
-    // Initialize MessageManager
-    this.messageManager = new MessageManager({
-      callbacks,
-      workdir: this.workdir,
-      logger: this.logger,
-      memoryRuleManager: this.memoryRuleManager,
-    });
-
-    // Initialize ReversionManager
+    // ReversionManager depends on MessageManager
     this.reversionManager = new ReversionManager(
       new ReversionService(this.messageManager.getTranscriptPath()),
     );
@@ -312,6 +345,7 @@ export class Agent {
       reversionManager: this.reversionManager,
       permissionMode: options.permissionMode, // Let PermissionManager handle defaultMode resolution
       canUseToolCallback: canUseToolWithNotification,
+      taskManager: this.taskManager,
       backgroundTaskManager: this.backgroundTaskManager,
       foregroundTaskManager: this.foregroundTaskManager,
     }); // Initialize tool registry with permission support
@@ -339,6 +373,8 @@ export class Agent {
           callbacks.onSubagentAssistantReasoningUpdated,
         onSubagentToolBlockUpdated: callbacks.onSubagentToolBlockUpdated,
         onSubagentMessagesChange: callbacks.onSubagentMessagesChange,
+        onSubagentLatestTotalTokensChange:
+          callbacks.onSubagentLatestTotalTokensChange,
       }, // Pass subagent callbacks for forwarding
       logger: this.logger,
       getGatewayConfig: () => this.getGatewayConfig(),
@@ -348,6 +384,7 @@ export class Agent {
       hookManager: this.hookManager,
       onUsageAdded: (usage) => this.addUsage(usage),
       backgroundTaskManager: this.backgroundTaskManager,
+      taskManager: this.taskManager,
       memoryRuleManager: this.memoryRuleManager,
     });
 
@@ -355,6 +392,7 @@ export class Agent {
     this.aiManager = new AIManager({
       messageManager: this.messageManager,
       toolManager: this.toolManager,
+      taskManager: this.taskManager,
       logger: this.logger,
       backgroundTaskManager: this.backgroundTaskManager,
       hookManager: this.hookManager,
@@ -381,6 +419,7 @@ export class Agent {
       messageManager: this.messageManager,
       aiManager: this.aiManager,
       backgroundTaskManager: this.backgroundTaskManager,
+      taskManager: this.taskManager,
       workdir: this.workdir,
       logger: this.logger,
     });
@@ -455,10 +494,6 @@ export class Agent {
     return this.messageManager.getlatestTotalTokens();
   }
 
-  public get userInputHistory(): string[] {
-    return this.messageManager.getUserInputHistory();
-  }
-
   /** Get working directory */
   public get workingDirectory(): string {
     return this.workdir;
@@ -780,8 +815,18 @@ export class Agent {
       // After main session is restored, restore any associated subagent sessions
       await this.restoreSubagentSessions(sessionToRestore?.messages || []);
 
-      if (sessionToRestore)
+      if (sessionToRestore) {
         this.messageManager.initializeFromSession(sessionToRestore);
+
+        // Update task manager with the root session ID to ensure continuity across compressions
+        this.taskManager.setTaskListId(
+          sessionToRestore.rootSessionId || sessionToRestore.id,
+        );
+
+        // After session is initialized, load tasks for the session
+        const tasks = await this.taskManager.listTasks();
+        this.options.callbacks?.onSessionTasksChange?.(tasks);
+      }
     }
   }
 
@@ -904,6 +949,13 @@ export class Agent {
 
     // 6. Initialize session state last
     this.messageManager.initializeFromSession(sessionData);
+
+    // Update task manager with the root session ID to ensure continuity across compressions
+    this.taskManager.setTaskListId(sessionData.rootSessionId || sessionData.id);
+
+    // 7. Load tasks for the restored session
+    const tasks = await this.taskManager.listTasks();
+    this.options.callbacks?.onSessionTasksChange?.(tasks);
   }
 
   public abortAIMessage(): void {
@@ -912,12 +964,9 @@ export class Agent {
 
   /** Execute bash command */
   public async executeBashCommand(command: string): Promise<void> {
-    // Add user message to history (but not displayed in UI)
-    this.addToInputHistory(`!${command}`);
     await this.bashManager?.executeCommand(command);
   }
 
-  /** Clear messages and input history */
   public clearMessages(): void {
     this.messageManager.clearMessages();
   }
@@ -929,11 +978,6 @@ export class Agent {
     this.abortSlashCommand();
   }
 
-  /** Add to input history */
-  private addToInputHistory(input: string): void {
-    this.messageManager.addToInputHistory(input);
-  }
-
   /** Interrupt bash command execution */
   public abortBashCommand(): void {
     this.bashManager?.abortCommand();
@@ -964,12 +1008,6 @@ export class Agent {
   public async backgroundCurrentTask(): Promise<void> {
     await this.foregroundTaskManager.backgroundCurrentTask();
     this.options.callbacks?.onBackgroundCurrentTask?.();
-
-    // If there was no foreground task (e.g. a tool that doesn't register one),
-    // or even if there was, we should stop the AI recursion loop.
-    if (!this.foregroundTaskManager.hasActiveTasks()) {
-      this.aiManager.abortRecursion();
-    }
   }
 
   /** Destroy managers, clean up resources */
@@ -1061,8 +1099,6 @@ export class Agent {
           // Execute valid slash command
           await this.slashCommandManager.executeCommand(commandId, args);
 
-          // Add slash command to history
-          this.addToInputHistory(command);
           return;
         }
 
@@ -1071,9 +1107,6 @@ export class Agent {
       }
 
       // Handle normal AI message
-      // Add user message to history
-      this.addToInputHistory(content);
-
       // Add user message first, will automatically sync to UI
       this.messageManager.addUserMessage({
         content,
@@ -1130,67 +1163,6 @@ export class Agent {
     }
   }
 
-  /** Save memory to project or user memory file */
-  public async saveMemory(
-    message: string,
-    type: "project" | "user",
-  ): Promise<void> {
-    try {
-      // Ensure the message starts with # for memory functions
-      const formattedMessage = message.startsWith("#")
-        ? message
-        : `#${message}`;
-
-      if (type === "project") {
-        await memory.addMemory(formattedMessage, this.workdir);
-        // Update internal state after successful save
-        this._projectMemoryContent = await memory.readMemoryFile(this.workdir);
-      } else {
-        await memory.addUserMemory(formattedMessage);
-        // Update internal state after successful save
-        this._userMemoryContent = await memory.getUserMemoryContent();
-      }
-
-      // Add successful MemoryBlock to the last assistant message
-      const memoryText = message.substring(1).trim();
-      const typeLabel = type === "project" ? "Project Memory" : "User Memory";
-      const storagePath = "AGENTS.md";
-
-      const messages = this.messageManager.getMessages();
-      const lastMessage = messages[messages.length - 1];
-      if (lastMessage && lastMessage.role === "assistant") {
-        lastMessage.blocks.push({
-          type: "memory",
-          content: `${typeLabel}: ${memoryText}`,
-          isSuccess: true,
-          memoryType: type,
-          storagePath,
-        });
-        this.messageManager.setMessages(messages);
-      }
-    } catch (error) {
-      // Add failed MemoryBlock to the last assistant message
-      const memoryText = message.substring(1).trim();
-      const typeLabel = type === "project" ? "Project Memory" : "User Memory";
-      const storagePath = "AGENTS.md";
-      const errorMessage =
-        error instanceof Error ? error.message : String(error);
-
-      const messages = this.messageManager.getMessages();
-      const lastMessage = messages[messages.length - 1];
-      if (lastMessage && lastMessage.role === "assistant") {
-        lastMessage.blocks.push({
-          type: "memory",
-          content: `${typeLabel}: ${memoryText} - Error: ${errorMessage}`,
-          isSuccess: false,
-          memoryType: type,
-          storagePath,
-        });
-        this.messageManager.setMessages(messages);
-      }
-    }
-  }
-
   // ========== MCP Management Methods ==========
 
   /** Get all MCP server status */
@@ -1270,6 +1242,16 @@ export class Agent {
     await this.messageManager.truncateHistory(index, this.reversionManager);
   }
 
+  /**
+   * Get the full message thread including parent sessions
+   */
+  public async getFullMessageThread(): Promise<{
+    messages: Message[];
+    sessionIds: string[];
+  }> {
+    return this.messageManager.getFullMessageThread();
+  }
+
   /**
    * Get the current plan file path (for testing and UI)
    */
@@ -1338,7 +1320,7 @@ export class Agent {
   private handlePlanModeTransition(mode: PermissionMode): void {
     if (mode === "plan") {
       this.planManager
-        .getOrGeneratePlanFilePath()
+        .getOrGeneratePlanFilePath(this.messageManager.getRootSessionId())
         .then(({ path }) => {
           this.logger?.debug("Plan file path generated", { path });
           this.permissionManager.setPlanFilePath(path);
@@ -1350,4 +1332,11 @@ export class Agent {
       this.permissionManager.setPlanFilePath(undefined);
     }
   }
+
+  /**
+   * Get the current task list ID
+   */
+  public get taskListId(): string {
+    return this.taskManager.getTaskListId();
+  }
 }