wave-agent-sdk 0.5.0 → 0.6.0

package/dist/services/aiService.js

@@ -4,33 +4,7 @@ import { transformMessagesForClaudeCache, addCacheControlToLastTool, isClaudeMod
 import * as os from "os";
 import * as fs from "fs";
 import * as path from "path";
-import { DEFAULT_SYSTEM_PROMPT, buildSystemPrompt, } from "../constants/prompts.js";
-/**
- * Use parametersChunk as compact param for better performance
- * Instead of parsing JSON, we use the raw chunk for efficient streaming
- */
-/**
- * Check if a directory is a git repository
- * @param dirPath Directory path to check
- * @returns "Yes" if it's a git repo, "No" otherwise
- */
-function isGitRepository(dirPath) {
-    try {
-        // Check if .git directory exists in current directory or any parent directory
-        let currentPath = path.resolve(dirPath);
-        while (currentPath !== path.dirname(currentPath)) {
-            const gitPath = path.join(currentPath, ".git");
-            if (fs.existsSync(gitPath)) {
-                return "Yes";
-            }
-            currentPath = path.dirname(currentPath);
-        }
-        return "No";
-    }
-    catch {
-        return "No";
-    }
-}
+import { COMPRESS_MESSAGES_SYSTEM_PROMPT } from "../constants/prompts.js";
 /**
  * Get specific configuration parameters based on model name
  * @param modelName Model name
@@ -58,7 +32,7 @@ function getModelConfig(modelName, baseConfig = {}) {
     return config;
 }
 export async function callAgent(options) {
-    const { gatewayConfig, modelConfig, messages, abortSignal, memory, workdir, tools, model, systemPrompt, onContentUpdate, onToolUpdate, onReasoningUpdate, } = options;
+    const { gatewayConfig, modelConfig, messages, abortSignal, workdir, tools, model, systemPrompt, onContentUpdate, onToolUpdate, onReasoningUpdate, } = options;
     // Declare variables outside try block for error handling access
     let openaiMessages;
     let createParams;
@@ -73,23 +47,7 @@ export async function callAgent(options) {
             fetch: gatewayConfig.fetch,
         });
         // Build system prompt content
-        let systemContent = buildSystemPrompt(systemPrompt || DEFAULT_SYSTEM_PROMPT, tools || []);
-        // Always add environment information
-        systemContent += `
-
-Here is useful information about the environment you are running in:
-<env>
-Working directory: ${workdir}
-Is directory a git repo: ${isGitRepository(workdir)}
-Platform: ${os.platform()}
-OS Version: ${os.type()} ${os.release()}
-Today's date: ${new Date().toISOString().split("T")[0]}
-</env>
-`;
-        // If there is memory content, add it to the system prompt
-        if (memory && memory.trim()) {
-            systemContent += `\n## Memory Context\n\nThe following is important context and memory from previous interactions:\n\n${memory}`;
-        }
+        const systemContent = systemPrompt || "";
         // Add system prompt
         const systemMessage = {
             role: "system",
@@ -470,101 +428,7 @@ export async function compressMessages(options) {
             messages: [
                 {
                     role: "system",
-                    content: `Your task is to create a detailed summary of the conversation so far, paying close attention to the user's explicit requests and your previous actions.
-This summary should be thorough in capturing technical details, code patterns, and architectural decisions that would be essential for continuing development work without losing context.
-
-Before providing your final summary, wrap your analysis in <analysis> tags to organize your thoughts and ensure you've covered all necessary points. In your analysis process:
-
-1. Chronologically analyze each message and section of the conversation. For each section thoroughly identify:
-   - The user's explicit requests and intents
-   - Your approach to addressing the user's requests
-   - Key decisions, technical concepts and code patterns
-   - Specific details like:
-     - file names
-     - full code snippets
-     - function signatures
-     - file edits
-  - Errors that you ran into and how you fixed them
-  - Pay special attention to specific user feedback that you received, especially if the user told you to do something differently.
-2. Double-check for technical accuracy and completeness, addressing each required element thoroughly.
-
-Your summary should include the following sections:
-
-1. Primary Request and Intent: Capture all of the user's explicit requests and intents in detail
-2. Key Technical Concepts: List all important technical concepts, technologies, and frameworks discussed.
-3. Files and Code Sections: Enumerate specific files and code sections examined, modified, or created. Pay special attention to the most recent messages and include full code snippets where applicable and include a summary of why this file read or edit is important.
-4. Errors and fixes: List all errors that you ran into, and how you fixed them. Pay special attention to specific user feedback that you received, especially if the user told you to do something differently.
-5. Problem Solving: Document problems solved and any ongoing troubleshooting efforts.
-6. All user messages: List ALL user messages that are not tool results. These are critical for understanding the users' feedback and changing intent.
-6. Pending Tasks: Outline any pending tasks that you have explicitly been asked to work on.
-7. Current Work: Describe in detail precisely what was being worked on immediately before this summary request, paying special attention to the most recent messages from both user and assistant. Include file names and code snippets where applicable.
-8. Optional Next Step: List the next step that you will take that is related to the most recent work you were doing. IMPORTANT: ensure that this step is DIRECTLY in line with the user's most recent explicit requests, and the task you were working on immediately before this summary request. If your last task was concluded, then only list next steps if they are explicitly in line with the users request. Do not start on tangential requests or really old requests that were already completed without confirming with the user first.
-                       If there is a next step, include direct quotes from the most recent conversation showing exactly what task you were working on and where you left off. This should be verbatim to ensure there's no drift in task interpretation.
-
-Here's an example of how your output should be structured:
-
-<example>
-<analysis>
-[Your thought process, ensuring all points are covered thoroughly and accurately]
-</analysis>
-
-<summary>
-1. Primary Request and Intent:
-   [Detailed description]
-
-2. Key Technical Concepts:
-   - [Concept 1]
-   - [Concept 2]
-   - [...]
-
-3. Files and Code Sections:
-   - [File Name 1]
-      - [Summary of why this file is important]
-      - [Summary of the changes made to this file, if any]
-      - [Important Code Snippet]
-   - [File Name 2]
-      - [Important Code Snippet]
-   - [...]
-
-4. Errors and fixes:
-    - [Detailed description of error 1]:
-      - [How you fixed the error]
-      - [User feedback on the error if any]
-    - [...]
-
-5. Problem Solving:
-   [Description of solved problems and ongoing troubleshooting]
-
-6. All user messages: 
-    - [Detailed non tool use user message]
-    - [...]
-
-7. Pending Tasks:
-   - [Task 1]
-   - [Task 2]
-   - [...]
-
-8. Current Work:
-   [Precise description of current work]
-
-9. Optional Next Step:
-   [Optional Next step to take]
-
-</summary>
-</example>
-
-Please provide your summary based on the conversation so far, following this structure and ensuring precision and thoroughness in your response. 
-
-There may be additional summarization instructions provided in the included context. If so, remember to follow these instructions when creating the above summary. Examples of instructions include:
-<example>
-## Compact Instructions
-When summarizing the conversation focus on typescript code changes and also remember the mistakes you made and how you fixed them.
-</example>
-
-<example>
-# Summary instructions
-When you are using compact - please focus on test output and code changes. Include file reads verbatim.
-</example>`,
+                    content: COMPRESS_MESSAGES_SYSTEM_PROMPT,
                 },
                 ...messages,
                 {

package/dist/services/memory.d.ts

@@ -1,7 +1,4 @@
-export declare const isMemoryMessage: (message: string) => boolean;
-export declare const addMemory: (message: string, workdir: string) => Promise<void>;
 export declare const ensureUserMemoryFile: () => Promise<void>;
-export declare const addUserMemory: (message: string) => Promise<void>;
 export declare const getUserMemoryContent: () => Promise<string>;
 export declare const readMemoryFile: (workdir: string) => Promise<string>;
 export declare const getCombinedMemoryContent: (workdir: string) => Promise<string>;

package/dist/services/memory.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../src/services/memory.ts"],"names":[],"mappings":"AAMA,eAAO,MAAM,eAAe,GAAI,SAAS,MAAM,KAAG,OAEjD,CAAC;AAEF,eAAO,MAAM,SAAS,GACpB,SAAS,MAAM,EACf,SAAS,MAAM,KACd,OAAO,CAAC,IAAI,CAuCd,CAAC;AAGF,eAAO,MAAM,oBAAoB,QAAa,OAAO,CAAC,IAAI,CA4BzD,CAAC;AAEF,eAAO,MAAM,aAAa,GAAU,SAAS,MAAM,KAAG,OAAO,CAAC,IAAI,CAsBjE,CAAC;AAEF,eAAO,MAAM,oBAAoB,QAAa,OAAO,CAAC,MAAM,CAa3D,CAAC;AAGF,eAAO,MAAM,cAAc,GAAU,SAAS,MAAM,KAAG,OAAO,CAAC,MAAM,CAqBpE,CAAC;AAGF,eAAO,MAAM,wBAAwB,GACnC,SAAS,MAAM,KACd,OAAO,CAAC,MAAM,CAoBhB,CAAC"}
+{"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../src/services/memory.ts"],"names":[],"mappings":"AAOA,eAAO,MAAM,oBAAoB,QAAa,OAAO,CAAC,IAAI,CA4BzD,CAAC;AAEF,eAAO,MAAM,oBAAoB,QAAa,OAAO,CAAC,MAAM,CAa3D,CAAC;AAGF,eAAO,MAAM,cAAc,GAAU,SAAS,MAAM,KAAG,OAAO,CAAC,MAAM,CAqBpE,CAAC;AAGF,eAAO,MAAM,wBAAwB,GACnC,SAAS,MAAM,KACd,OAAO,CAAC,MAAM,CAoBhB,CAAC"}

package/dist/services/memory.js

@@ -3,46 +3,6 @@ import path from "path";
 import { USER_MEMORY_FILE, DATA_DIRECTORY } from "../utils/constants.js";
 import { logger } from "../utils/globalLogger.js";
 // Project memory related methods
-export const isMemoryMessage = (message) => {
-    return message.trim().startsWith("#");
-};
-export const addMemory = async (message, workdir) => {
-    if (!isMemoryMessage(message)) {
-        return;
-    }
-    try {
-        const memoryFilePath = path.join(workdir, "AGENTS.md");
-        // Format memory entry, starting with -, no timestamp
-        const memoryEntry = `- ${message.substring(1).trim()}\n`;
-        // Check if file exists
-        let existingContent = "";
-        try {
-            existingContent = await fs.readFile(memoryFilePath, "utf-8");
-        }
-        catch (error) {
-            // File does not exist, create new file
-            if (error.code === "ENOENT") {
-                logger.info("Memory file does not exist, will create new one", {
-                    memoryFilePath,
-                });
-                existingContent =
-                    "# Memory\n\nThis is the AI assistant's memory file, recording important information and context.\n\n";
-            }
-            else {
-                throw error;
-            }
-        }
-        // Append new memory entry to the end of the file
-        const updatedContent = existingContent + memoryEntry;
-        // Write file
-        await fs.writeFile(memoryFilePath, updatedContent, "utf-8");
-        logger.debug(`Memory added to ${memoryFilePath}:`, message);
-    }
-    catch (error) {
-        logger.error("Failed to add memory:", error);
-        throw new Error(`Failed to add memory: ${error.message}`);
-    }
-};
 // User memory related methods
 export const ensureUserMemoryFile = async () => {
     try {
@@ -72,25 +32,6 @@ export const ensureUserMemoryFile = async () => {
         throw new Error(`Failed to ensure user memory file: ${error.message}`);
     }
 };
-export const addUserMemory = async (message) => {
-    try {
-        // Ensure user memory file exists
-        await ensureUserMemoryFile();
-        // Format memory entry, starting with -
-        const memoryEntry = `- ${message.substring(1).trim()}\n`;
-        // Read existing content
-        const existingContent = await fs.readFile(USER_MEMORY_FILE, "utf-8");
-        // Append new memory entry to the end of the file
-        const updatedContent = existingContent + memoryEntry;
-        // Write file
-        await fs.writeFile(USER_MEMORY_FILE, updatedContent, "utf-8");
-        logger.debug(`User memory added to ${USER_MEMORY_FILE}:`, message);
-    }
-    catch (error) {
-        logger.error("Failed to add user memory:", error);
-        throw new Error(`Failed to add user memory: ${error.message}`);
-    }
-};
 export const getUserMemoryContent = async () => {
     try {
         await ensureUserMemoryFile();

package/dist/services/session.d.ts

@@ -17,6 +17,7 @@
 import type { Message } from "../types/index.js";
 export interface SessionData {
     id: string;
+    rootSessionId?: string;
     messages: Message[];
     metadata: {
         workdir: string;
@@ -26,6 +27,7 @@ export interface SessionData {
 }
 export interface SessionMetadata {
     id: string;
+    rootSessionId?: string;
     sessionType: "main" | "subagent";
     subagentType?: string;
     workdir: string;
@@ -86,7 +88,7 @@ export declare function createSession(sessionId: string, workdir: string, sessio
  * @param workdir - Working directory for the session
  * @param sessionType - Type of session ("main" or "subagent", defaults to "main")
  */
-export declare function appendMessages(sessionId: string, newMessages: Message[], workdir: string, sessionType?: "main" | "subagent"): Promise<void>;
+export declare function appendMessages(sessionId: string, newMessages: Message[], workdir: string, sessionType?: "main" | "subagent", rootSessionId?: string): Promise<void>;
 /**
  * Load session data from JSONL file (new approach)
  *

package/dist/services/session.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/services/session.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAMH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAOjD,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,QAAQ,EAAE;QACR,OAAO,EAAE,MAAM,CAAC;QAChB,YAAY,EAAE,MAAM,CAAC;QACrB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,GAAG,UAAU,CAAC;IACjC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,IAAI,CAAC;IACnB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,MAAM,CACd,MAAM,EACN,IAAI,CAAC,eAAe,EAAE,IAAI,GAAG,cAAc,CAAC,GAAG;QAAE,YAAY,EAAE,MAAM,CAAA;KAAE,CACxE,CAAC;IACF,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,CAE1C;AAED;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAElE;AAGD,eAAO,MAAM,WAAW,QAAuC,CAAC;AAmChE;;GAEG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC,CAMtD;AAED;;;;;;GAMG;AACH,wBAAsB,uBAAuB,CAC3C,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,MAAM,CAAC,CASjB;AAED;;;;;;GAMG;AACH,wBAAsB,kBAAkB,CACtC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,MAAM,CAAC,CASjB;AAED;;;;;GAKG;AACH,wBAAsB,aAAa,CACjC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,IAAI,CAAC,CAIf;AAED;;;;;;;GAOG;AACH,wBAAsB,cAAc,CAClC,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,OAAO,EAAE,EACtB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,IAAI,CAAC,CAqEf;AAED;;;;;;;GAOG;AACH,wBAAsB,oBAAoB,CACxC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CA2D7B;AAED;;;;;;;GAOG;AACH,wBAAsB,yBAAyB,CAC7C,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAU7B;AAED;;;;;;GAMG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,eAAe,EAAE,CAAC,CAE5B;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,eAAe,EAAE,CAAC,CAsI5B;AAED;;;;;GAKG;AACH,wBAAsB,+BAA+B,CACnD,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,MAAM,CAAC,CAmFjB;AAED;;GAEG;AACH,wBAAsB,8BAA8B,IAAI,OAAO,CAAC,IAAI,CAAC,CA+BpE;AAED;;;;;;;GAOG;AACH,wBAAsB,oBAAoB,CACxC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,CAAC,EAAE,MAAM,GAAG,UAAU,GAChC,OAAO,CAAC,OAAO,CAAC,CAsClB;AAED;;;;;;;GAOG;AACH,wBAAsB,sBAAsB,CAC1C,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAsDxB;AAED;;;;;GAKG;AACH,wBAAgB,eAAe,CAC7B,OAAO,EAAE,MAAM,EACf,SAAS,GAAE,MAAW,GACrB,MAAM,CAKR;AAED;;;;;;GAMG;AACH,wBAAsB,wBAAwB,CAC5C,gBAAgB,CAAC,EAAE,MAAM,EACzB,mBAAmB,CAAC,EAAE,OAAO,EAC7B,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,CA4ClC"}
+{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/services/session.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAMH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAOjD,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,QAAQ,EAAE;QACR,OAAO,EAAE,MAAM,CAAC;QAChB,YAAY,EAAE,MAAM,CAAC;QACrB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,EAAE,MAAM,GAAG,UAAU,CAAC;IACjC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,IAAI,CAAC;IACnB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,MAAM,CACd,MAAM,EACN,IAAI,CAAC,eAAe,EAAE,IAAI,GAAG,cAAc,CAAC,GAAG;QAAE,YAAY,EAAE,MAAM,CAAA;KAAE,CACxE,CAAC;IACF,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,CAE1C;AAED;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAElE;AAGD,eAAO,MAAM,WAAW,QAAuC,CAAC;AAmChE;;GAEG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC,CAMtD;AAED;;;;;;GAMG;AACH,wBAAsB,uBAAuB,CAC3C,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,MAAM,CAAC,CASjB;AAED;;;;;;GAMG;AACH,wBAAsB,kBAAkB,CACtC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,MAAM,CAAC,CASjB;AAED;;;;;GAKG;AACH,wBAAsB,aAAa,CACjC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,IAAI,CAAC,CAIf;AAED;;;;;;;GAOG;AACH,wBAAsB,cAAc,CAClC,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,OAAO,EAAE,EACtB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,EACzC,aAAa,CAAC,EAAE,MAAM,GACrB,OAAO,CAAC,IAAI,CAAC,CAsEf;AAED;;;;;;;GAOG;AACH,wBAAsB,oBAAoB,CACxC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAAM,GAAG,UAAmB,GACxC,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CA4E7B;AAED;;;;;;;GAOG;AACH,wBAAsB,yBAAyB,CAC7C,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAU7B;AAED;;;;;;GAMG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,eAAe,EAAE,CAAC,CAE5B;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,eAAe,EAAE,CAAC,CAsI5B;AAED;;;;;GAKG;AACH,wBAAsB,+BAA+B,CACnD,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,MAAM,CAAC,CAmFjB;AAED;;GAEG;AACH,wBAAsB,8BAA8B,IAAI,OAAO,CAAC,IAAI,CAAC,CA+BpE;AAED;;;;;;;GAOG;AACH,wBAAsB,oBAAoB,CACxC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,WAAW,CAAC,EAAE,MAAM,GAAG,UAAU,GAChC,OAAO,CAAC,OAAO,CAAC,CAsClB;AAED;;;;;;;GAOG;AACH,wBAAsB,sBAAsB,CAC1C,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAsDxB;AAED;;;;;GAKG;AACH,wBAAgB,eAAe,CAC7B,OAAO,EAAE,MAAM,EACf,SAAS,GAAE,MAAW,GACrB,MAAM,CAKR;AAED;;;;;;GAMG;AACH,wBAAsB,wBAAwB,CAC5C,gBAAgB,CAAC,EAAE,MAAM,EACzB,mBAAmB,CAAC,EAAE,OAAO,EAC7B,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,CA4ClC"}

package/dist/services/session.js

@@ -126,7 +126,7 @@ export async function createSession(sessionId, workdir, sessionType = "main") {
  * @param workdir - Working directory for the session
  * @param sessionType - Type of session ("main" or "subagent", defaults to "main")
  */
-export async function appendMessages(sessionId, newMessages, workdir, sessionType = "main") {
+export async function appendMessages(sessionId, newMessages, workdir, sessionType = "main", rootSessionId) {
     // Do not save session files in test environment
     if (process.env.NODE_ENV === "test") {
         return;
@@ -174,6 +174,7 @@ export async function appendMessages(sessionId, newMessages, workdir, sessionTyp
     }
     await updateSessionIndex(projectDir.encodedPath, {
         id: sessionId,
+        rootSessionId,
         sessionType,
         workdir,
         lastActiveAt: new Date(lastMessage.timestamp),
@@ -202,8 +203,22 @@ export async function loadSessionFromJsonl(sessionId, workdir, sessionType = "ma
         }
         // Extract metadata from messages
         const lastMessage = messages[messages.length - 1];
+        // Try to get rootSessionId from index
+        let rootSessionId;
+        try {
+            const encoder = new PathEncoder();
+            const projectDir = await encoder.getProjectDirectory(workdir, SESSION_DIR);
+            const indexPath = join(projectDir.encodedPath, SESSION_INDEX_FILENAME);
+            const indexContent = await fs.readFile(indexPath, "utf8");
+            const index = JSON.parse(indexContent);
+            rootSessionId = index.sessions[sessionId]?.rootSessionId;
+        }
+        catch {
+            // Ignore index errors
+        }
         const sessionData = {
             id: sessionId,
+            rootSessionId: rootSessionId || sessionId,
             messages: messages.map((msg) => {
                 // Remove timestamp property for backward compatibility
                 const { timestamp: _ignored, ...messageWithoutTimestamp } = msg;

package/dist/services/taskManager.d.ts

@@ -0,0 +1,21 @@
+import { EventEmitter } from "events";
+import { Task } from "../types/tasks.js";
+export declare class TaskManager extends EventEmitter {
+    private readonly baseDir;
+    private taskListId;
+    constructor(taskListId: string);
+    getTaskListId(): string;
+    setTaskListId(taskListId: string): void;
+    private getSessionDir;
+    private getTaskPath;
+    private getLockPath;
+    ensureSessionDir(): Promise<void>;
+    private withLock;
+    private validateTask;
+    createTask(task: Omit<Task, "id">): Promise<string>;
+    getTask(taskId: string): Promise<Task | null>;
+    updateTask(task: Task): Promise<void>;
+    listTasks(): Promise<Task[]>;
+    getNextTaskId(): Promise<string>;
+}
+//# sourceMappingURL=taskManager.d.ts.map

package/dist/services/taskManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"taskManager.d.ts","sourceRoot":"","sources":["../../src/services/taskManager.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAGzC,qBAAa,WAAY,SAAQ,YAAY;IAC3C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,UAAU,CAAS;gBAEf,UAAU,EAAE,MAAM;IAMvB,aAAa,IAAI,MAAM;IAIvB,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAI9C,OAAO,CAAC,aAAa;IAIrB,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,WAAW;IAIb,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;YAIzB,QAAQ;IA2CtB,OAAO,CAAC,YAAY;IAed,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAcnD,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAmB7C,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAWrC,SAAS,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;IA+B5B,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC;CAavC"}

package/dist/services/taskManager.js

@@ -0,0 +1,158 @@
+import { promises as fs } from "fs";
+import { join } from "path";
+import { homedir } from "os";
+import { EventEmitter } from "events";
+import { logger } from "../utils/globalLogger.js";
+export class TaskManager extends EventEmitter {
+    constructor(taskListId) {
+        super();
+        this.taskListId = taskListId;
+        this.baseDir = join(homedir(), ".wave", "tasks");
+    }
+    getTaskListId() {
+        return this.taskListId;
+    }
+    setTaskListId(taskListId) {
+        this.taskListId = taskListId;
+    }
+    getSessionDir() {
+        return join(this.baseDir, this.taskListId);
+    }
+    getTaskPath(taskId) {
+        return join(this.getSessionDir(), `${taskId}.json`);
+    }
+    getLockPath() {
+        return join(this.getSessionDir(), `.lock`);
+    }
+    async ensureSessionDir() {
+        await fs.mkdir(this.getSessionDir(), { recursive: true });
+    }
+    async withLock(operation) {
+        const lockPath = this.getLockPath();
+        let lockHandle;
+        const maxRetries = 100;
+        const retryDelay = process.env.NODE_ENV === "test" ? 10 : 100;
+        await this.ensureSessionDir();
+        for (let i = 0; i < maxRetries; i++) {
+            try {
+                lockHandle = await fs.open(lockPath, "wx");
+                break;
+            }
+            catch (error) {
+                if (error.code === "EEXIST") {
+                    if (i === maxRetries - 1) {
+                        throw new Error(`Could not acquire lock for task list ${this.taskListId} after ${maxRetries} retries`);
+                    }
+                    await new Promise((resolve) => setTimeout(resolve, retryDelay));
+                    continue;
+                }
+                throw error;
+            }
+        }
+        try {
+            return await operation();
+        }
+        finally {
+            if (lockHandle) {
+                await lockHandle.close();
+            }
+            try {
+                await fs.unlink(lockPath);
+            }
+            catch (error) {
+                logger.error(`Failed to release lock for task list ${this.taskListId}:`, error);
+            }
+        }
+    }
+    validateTask(task) {
+        if (!task.id || typeof task.id !== "string")
+            throw new Error("Invalid task ID");
+        if (!task.subject || typeof task.subject !== "string")
+            throw new Error("Invalid task subject");
+        if (!task.description || typeof task.description !== "string")
+            throw new Error("Invalid task description");
+        if (!task.status ||
+            !["pending", "in_progress", "completed", "deleted"].includes(task.status)) {
+            throw new Error(`Invalid task status: ${task.status}`);
+        }
+    }
+    async createTask(task) {
+        return await this.withLock(async () => {
+            const taskId = await this.getNextTaskId();
+            const fullTask = { ...task, id: taskId };
+            this.validateTask(fullTask);
+            const taskPath = this.getTaskPath(taskId);
+            const content = JSON.stringify(fullTask, null, 2);
+            await fs.writeFile(taskPath, content, "utf8");
+            this.emit("tasksChange", this.taskListId);
+            logger.info(`Task ${taskId} created in task list ${this.taskListId}`);
+            return taskId;
+        });
+    }
+    async getTask(taskId) {
+        const taskPath = this.getTaskPath(taskId);
+        try {
+            const content = await fs.readFile(taskPath, "utf8");
+            try {
+                return JSON.parse(content.trim());
+            }
+            catch (parseError) {
+                logger.error(`Failed to parse task file ${taskPath}:`, parseError);
+                logger.error(`Corrupted content: ${content}`);
+                throw parseError;
+            }
+        }
+        catch (error) {
+            if (error.code === "ENOENT") {
+                return null;
+            }
+            throw error;
+        }
+    }
+    async updateTask(task) {
+        this.validateTask(task);
+        await this.withLock(async () => {
+            const taskPath = this.getTaskPath(task.id);
+            const content = JSON.stringify(task, null, 2);
+            await fs.writeFile(taskPath, content, "utf8");
+            this.emit("tasksChange", this.taskListId);
+            logger.info(`Task ${task.id} updated in task list ${this.taskListId}`);
+        });
+    }
+    async listTasks() {
+        const sessionDir = this.getSessionDir();
+        try {
+            const files = await fs.readdir(sessionDir);
+            const taskFiles = files.filter((f) => f.endsWith(".json"));
+            const tasks = await Promise.all(taskFiles.map(async (file) => {
+                const taskPath = join(sessionDir, file);
+                try {
+                    const content = await fs.readFile(taskPath, "utf8");
+                    return JSON.parse(content.trim());
+                }
+                catch (error) {
+                    logger.error(`Failed to read or parse task file ${taskPath}:`, error);
+                    return null;
+                }
+            }));
+            return tasks.filter((t) => t !== null);
+        }
+        catch (error) {
+            if (error.code === "ENOENT") {
+                return [];
+            }
+            throw error;
+        }
+    }
+    async getNextTaskId() {
+        const tasks = await this.listTasks();
+        if (tasks.length === 0) {
+            return "1";
+        }
+        const ids = tasks.map((t) => parseInt(t.id, 10)).filter((id) => !isNaN(id));
+        if (ids.length === 0) {
+            return (tasks.length + 1).toString();
+        }
+        return (Math.max(...ids) + 1).toString();
+    }
+}

package/dist/tools/askUserQuestion.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"askUserQuestion.d.ts","sourceRoot":"","sources":["../../src/tools/askUserQuestion.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAOxC,eAAO,MAAM,mBAAmB,EAAE,UAwHjC,CAAC"}
+{"version":3,"file":"askUserQuestion.d.ts","sourceRoot":"","sources":["../../src/tools/askUserQuestion.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAIxC,eAAO,MAAM,mBAAmB,EAAE,UAiJjC,CAAC"}

package/dist/tools/askUserQuestion.js

@@ -1,23 +1,11 @@
-import { ASK_USER_QUESTION_TOOL_NAME, EXIT_PLAN_MODE_TOOL_NAME, } from "../constants/tools.js";
+import { ASK_USER_QUESTION_TOOL_NAME } from "../constants/tools.js";
 export const askUserQuestionTool = {
     name: ASK_USER_QUESTION_TOOL_NAME,
     config: {
         type: "function",
         function: {
             name: ASK_USER_QUESTION_TOOL_NAME,
-            description: `Asks the user multiple choice questions to gather information, clarify ambiguity, understand preferences, make decisions or offer them choices.
-Use this tool when you need to ask the user questions during execution. This allows you to:
-1. Gather user preferences or requirements
-2. Clarify ambiguous instructions
-3. Get decisions on implementation choices as you work
-4. Offer choices to the user about what direction to take.
-
-Usage notes:
-- Users will always be able to select "Other" to provide custom text input
-- Use multiSelect: true to allow multiple answers to be selected for a question
-- If you recommend a specific option, make that the first option in the list and add "(Recommended)" at the end of the label
-
-Plan mode note: In plan mode, use this tool to clarify requirements or choose between approaches BEFORE finalizing your plan. Do NOT use this tool to ask "Is my plan ready?" or "Should I proceed?" - use ${EXIT_PLAN_MODE_TOOL_NAME} for plan approval.`,
+            description: "Asks the user multiple choice questions to gather information, clarify ambiguity, understand preferences, make decisions or offer them choices.",
             parameters: {
                 type: "object",
                 properties: {
@@ -25,36 +13,34 @@ Plan mode note: In plan mode, use this tool to clarify requirements or choose be
                         type: "array",
                         minItems: 1,
                         maxItems: 4,
+                        description: "Questions to ask the user (1-4 questions)",
                         items: {
                             type: "object",
                             properties: {
                                 question: {
                                     type: "string",
-                                    description: "The complete question to ask the user.",
+                                    description: 'The complete question to ask the user. Should be clear, specific, and end with a question mark. Example: "Which library should we use for date formatting?" If multiSelect is true, phrase it accordingly, e.g. "Which features do you want to enable?"',
                                 },
                                 header: {
                                     type: "string",
                                     maxLength: 12,
-                                    description: "Very short label displayed as a chip/tag (max 12 chars).",
+                                    description: `Very short label displayed as a chip/tag (max 12 chars). Examples: "Auth method", "Library", "Approach".`,
                                 },
                                 options: {
                                     type: "array",
                                     minItems: 2,
                                     maxItems: 4,
+                                    description: "The available choices for this question. Must have 2-4 options. Each option should be a distinct, mutually exclusive choice (unless multiSelect is enabled). There should be no 'Other' option, that will be provided automatically.",
                                     items: {
                                         type: "object",
                                         properties: {
                                             label: {
                                                 type: "string",
-                                                description: "The display text for this option.",
+                                                description: "The display text for this option that the user will see and select. Should be concise (1-5 words) and clearly describe the choice.",
                                             },
                                             description: {
                                                 type: "string",
-                                                description: "Explanation of what this option means.",
-                                            },
-                                            isRecommended: {
-                                                type: "boolean",
-                                                description: "Whether this option is recommended.",
+                                                description: "Explanation of what this option means or what will happen if chosen. Useful for providing context about trade-offs or implications.",
                                             },
                                         },
                                         required: ["label"],
@@ -63,23 +49,51 @@ Plan mode note: In plan mode, use this tool to clarify requirements or choose be
                                 multiSelect: {
                                     type: "boolean",
                                     default: false,
-                                    description: "Allow multiple answers to be selected.",
+                                    description: "Set to true to allow the user to select multiple options instead of just one. Use when choices are not mutually exclusive.",
                                 },
                             },
                             required: ["question", "header", "options"],
                         },
                     },
+                    answers: {
+                        type: "object",
+                        additionalProperties: { type: "string" },
+                        description: "User answers collected by the permission component",
+                    },
+                    metadata: {
+                        type: "object",
+                        properties: {
+                            source: {
+                                type: "string",
+                                description: 'Optional identifier for the source of this question (e.g., "remember" for /remember command). Used for analytics tracking.',
+                            },
+                        },
+                        description: "Optional metadata for the question",
+                    },
                 },
                 required: ["questions"],
             },
         },
     },
+    prompt: () => `Use this tool when you need to ask the user questions during execution. This allows you to:
+1. Gather user preferences or requirements
+2. Clarify ambiguous instructions
+3. Get decisions on implementation choices as you work
+4. Offer choices to the user about what direction to take.
+
+Usage notes:
+- Users will always be able to select "Other" to provide custom text input
+- Use multiSelect: true to allow multiple answers to be selected for a question
+- If you recommend a specific option, make that the first option in the list and add "(Recommended)" at the end of the label
+
+Plan mode note: In plan mode, use this tool to clarify requirements or choose between approaches BEFORE finalizing your plan. Do NOT use this tool to ask "Is my plan ready?" or "Should I proceed?" - use ExitPlanMode for plan approval.
+`,
     execute: async (args, context) => {
-        const { questions } = args;
+        const { questions, answers: existingAnswers, metadata, } = args;
         if (!context.permissionManager) {
             throw new Error(`Permission manager is required for ${ASK_USER_QUESTION_TOOL_NAME} tool`);
         }
-        const permissionContext = context.permissionManager.createContext(ASK_USER_QUESTION_TOOL_NAME, context.permissionMode || "default", context.canUseToolCallback, { questions });
+        const permissionContext = context.permissionManager.createContext(ASK_USER_QUESTION_TOOL_NAME, context.permissionMode || "default", context.canUseToolCallback, { questions, answers: existingAnswers, metadata });
         permissionContext.hidePersistentOption = true; // Always hide persistent option for questions
         const decision = await context.permissionManager.checkPermission(permissionContext);
         if (decision.behavior === "deny") {

package/dist/tools/bashTool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"bashTool.d.ts","sourceRoot":"","sources":["../../src/tools/bashTool.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAA2B,MAAM,YAAY,CAAC;AActE;;GAEG;AACH,eAAO,MAAM,QAAQ,EAAE,UAmXtB,CAAC"}
+{"version":3,"file":"bashTool.d.ts","sourceRoot":"","sources":["../../src/tools/bashTool.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAA2B,MAAM,YAAY,CAAC;AActE;;GAEG;AACH,eAAO,MAAM,QAAQ,EAAE,UA6WtB,CAAC"}