wave-agent-sdk 0.4.0 → 0.6.1

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();
@@ -126,7 +67,7 @@ export const readMemoryFile = async (workdir) => {
             return "";
         }
         logger.error("Failed to read memory file", { memoryFilePath, error });
-        throw error;
+        return "";
     }
 };
 // Get merged memory content (project memory + user memory)

package/dist/services/session.d.ts

@@ -17,6 +17,8 @@
 import type { Message } from "../types/index.js";
 export interface SessionData {
     id: string;
+    rootSessionId?: string;
+    parentSessionId?: string;
     messages: Message[];
     metadata: {
         workdir: string;
@@ -26,6 +28,8 @@ export interface SessionData {
 }
 export interface SessionMetadata {
     id: string;
+    rootSessionId?: string;
+    parentSessionId?: string;
     sessionType: "main" | "subagent";
     subagentType?: string;
     workdir: string;
@@ -86,7 +90,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, parentSessionId?: string): Promise<void>;
 /**
  * Load session data from JSONL file (new approach)
  *
@@ -171,4 +175,14 @@ export declare function truncateContent(content: string, maxLength?: number): st
  * @returns Promise that resolves to session data or undefined
  */
 export declare function handleSessionRestoration(restoreSessionId?: string, continueLastSession?: boolean, workdir?: string): Promise<SessionData | undefined>;
+/**
+ * Load the full message thread by following parentSessionId links
+ * @param currentSessionId - The ID of the current session
+ * @param workdir - Working directory for the session
+ * @returns Promise that resolves to an array of all messages in the thread
+ */
+export declare function loadFullMessageThread(currentSessionId: string, workdir: string): Promise<{
+    messages: Message[];
+    sessionIds: string[];
+}>;
 //# sourceMappingURL=session.d.ts.map

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,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,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,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,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;AAoChE;;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,EACtB,eAAe,CAAC,EAAE,MAAM,GACvB,OAAO,CAAC,IAAI,CAAC,CAuEf;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,CA+E7B;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;AAED;;;;;GAKG;AACH,wBAAsB,qBAAqB,CACzC,gBAAgB,EAAE,MAAM,EACxB,OAAO,EAAE,MAAM,GACd,OAAO,CAAC;IAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAAC,UAAU,EAAE,MAAM,EAAE,CAAA;CAAE,CAAC,CAiCxD"}

package/dist/services/session.js

@@ -62,6 +62,7 @@ async function updateSessionIndex(projectDirPath, metadata) {
         ...rest,
         lastActiveAt: metadata.lastActiveAt.toISOString(),
         firstMessage: metadata.firstMessage || index.sessions[id]?.firstMessage,
+        parentSessionId: metadata.parentSessionId,
     };
     index.lastUpdated = new Date().toISOString();
     await fs.writeFile(indexPath, JSON.stringify(index, null, 2), "utf8");
@@ -126,7 +127,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, parentSessionId) {
     // Do not save session files in test environment
     if (process.env.NODE_ENV === "test") {
         return;
@@ -174,6 +175,8 @@ export async function appendMessages(sessionId, newMessages, workdir, sessionTyp
     }
     await updateSessionIndex(projectDir.encodedPath, {
         id: sessionId,
+        rootSessionId,
+        parentSessionId,
         sessionType,
         workdir,
         lastActiveAt: new Date(lastMessage.timestamp),
@@ -202,8 +205,25 @@ export async function loadSessionFromJsonl(sessionId, workdir, sessionType = "ma
         }
         // Extract metadata from messages
         const lastMessage = messages[messages.length - 1];
+        // Try to get rootSessionId and parentSessionId from index
+        let rootSessionId;
+        let parentSessionId;
+        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;
+            parentSessionId = index.sessions[sessionId]?.parentSessionId;
+        }
+        catch {
+            // Ignore index errors
+        }
         const sessionData = {
             id: sessionId,
+            rootSessionId: rootSessionId || sessionId,
+            parentSessionId,
             messages: messages.map((msg) => {
                 // Remove timestamp property for backward compatibility
                 const { timestamp: _ignored, ...messageWithoutTimestamp } = msg;
@@ -644,3 +664,39 @@ export async function handleSessionRestoration(restoreSessionId, continueLastSes
         process.exit(1);
     }
 }
+/**
+ * Load the full message thread by following parentSessionId links
+ * @param currentSessionId - The ID of the current session
+ * @param workdir - Working directory for the session
+ * @returns Promise that resolves to an array of all messages in the thread
+ */
+export async function loadFullMessageThread(currentSessionId, workdir) {
+    const sessionIds = [];
+    let currentId = currentSessionId;
+    const allMessages = [];
+    while (currentId) {
+        const sessionData = await loadSessionFromJsonl(currentId, workdir);
+        if (!sessionData)
+            break;
+        sessionIds.unshift(currentId);
+        // Add messages from this session to the beginning of the list
+        // But skip the "compress" block if it's not the first session in our traversal (which is the latest)
+        // Actually, we should probably keep all messages and let the UI/logic handle it.
+        // But wait, if we are concatenating, the "compress" block in session N summarizes session N-1.
+        // So if we have session N-1 and session N, we should probably skip the compress block in session N.
+        const messages = sessionData.messages;
+        if (allMessages.length > 0) {
+            // If we already have messages (from "later" sessions),
+            // we are now adding messages from an "earlier" session.
+            // The later session's first message might be a "compress" block.
+            if (allMessages[0].blocks.some((b) => b.type === "compress")) {
+                // Remove the compress block from the later session's messages
+                // because we are now providing the actual messages it summarized.
+                allMessages.shift();
+            }
+        }
+        allMessages.unshift(...messages);
+        currentId = sessionData.parentSessionId;
+    }
+    return { messages: allMessages, sessionIds };
+}

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"}