wave-agent-sdk 1.0.4 → 1.0.5

package/dist/managers/messageManager.js

@@ -1,4 +1,4 @@
-import { addAssistantMessageToMessages, updateToolBlockInMessage, addErrorBlockToMessage, addUserMessageToMessages, updateUserMessageInMessages, addBangMessage, updateBangInMessage, completeBangInMessage, removeLastUserMessage, addToolBlockToMessageInMessages, addNotificationMessageToMessages, generateMessageId, } from "../utils/messageOperations.js";
+import { addAssistantMessageToMessages, updateToolBlockInMessage, addErrorBlockToMessage, addUserMessageToMessages, updateUserMessageInMessages, addBangMessage, updateBangInMessage, completeBangInMessage, removeLastUserMessage, addToolBlockToMessageInMessages, addNotificationMessageToMessages, generateMessageId, sliceFromLastCompact, } from "../utils/messageOperations.js";
 import { getLastApiRounds } from "../utils/groupMessagesByApiRound.js";
 import { join, isAbsolute, relative } from "path";
 import { appendMessages, createSession, generateSessionId, SESSION_DIR, } from "../services/session.js";
@@ -717,9 +717,12 @@ export class MessageManager {
         if (reversionManager && messageIdsToRemove.length > 0) {
             await reversionManager.revertTo(messageIdsToRemove, messages);
         }
-        // Rewrite file with truncated messages
+        // Rewrite file with truncated messages (full history kept on disk)
         await this.rewriteSessionFile(newMessages);
-        this.setMessages(newMessages);
+        // Fold the in-memory messages at the last compact boundary so the agent
+        // only sees messages from the latest compact summary forward — matching
+        // the compact and resume behaviors (which also fold memory).
+        this.setMessages(sliceFromLastCompact(newMessages));
         this.savedMessageCount = newMessages.length;
     }
     /**

package/dist/services/session.js

@@ -22,7 +22,7 @@ import { PathEncoder } from "../utils/pathEncoder.js";
 import { JsonlHandler } from "../services/jsonlHandler.js";
 import { extractLatestTotalTokens } from "../utils/tokenCalculation.js";
 import { logger } from "../utils/globalLogger.js";
-import { getMessageContent } from "../utils/messageOperations.js";
+import { getMessageContent, sliceFromLastCompact, } from "../utils/messageOperations.js";
 /**
  * Generate a new session ID using Node.js native crypto.randomUUID()
  * @returns UUID string for session identification
@@ -149,14 +149,7 @@ export async function loadSessionFromJsonl(sessionId, workdir, sessionType = "ma
         }
         const allMessages = await jsonlHandler.read(filePath);
         // Find the last compact boundary — only return messages from there forward
-        let lastCompactIndex = -1;
-        for (let i = allMessages.length - 1; i >= 0; i--) {
-            if (allMessages[i].blocks?.some((b) => b.type === "compact")) {
-                lastCompactIndex = i;
-                break;
-            }
-        }
-        const messages = lastCompactIndex >= 0 ? allMessages.slice(lastCompactIndex) : allMessages;
+        const messages = sliceFromLastCompact(allMessages);
         // Extract metadata from messages
         const lastMessage = messages.length > 0 ? messages[messages.length - 1] : null;
         const sessionData = {

package/dist/utils/messageOperations.d.ts

@@ -112,6 +112,14 @@ export declare function cloneMessage(message: Message): Message;
  * Helper to format tool and token summary
  */
 export declare function formatToolTokenSummary(toolCount: number, tokens: number): string;
+/**
+ * Fold a message list at the last compact boundary — returns only messages from
+ * the most recent compact summary forward. Mirrors what is sent to the LLM
+ * (convertMessagesForAPI stops at the last compact block), so the display layer
+ * matches the AI context. The session file keeps the full history; folding is
+ * display-only.
+ */
+export declare function sliceFromLastCompact(messages: Message[]): Message[];
 /**
  * Extracts displayable text from a Message object, handling various block types.
  * Returns the first available content block.

package/dist/utils/messageOperations.js

@@ -415,6 +415,23 @@ export function formatToolTokenSummary(toolCount, tokens) {
     summary += ")";
     return summary;
 }
+/**
+ * Fold a message list at the last compact boundary — returns only messages from
+ * the most recent compact summary forward. Mirrors what is sent to the LLM
+ * (convertMessagesForAPI stops at the last compact block), so the display layer
+ * matches the AI context. The session file keeps the full history; folding is
+ * display-only.
+ */
+export function sliceFromLastCompact(messages) {
+    let lastCompactIndex = -1;
+    for (let i = messages.length - 1; i >= 0; i--) {
+        if (messages[i].blocks?.some((b) => b.type === "compact")) {
+            lastCompactIndex = i;
+            break;
+        }
+    }
+    return lastCompactIndex >= 0 ? messages.slice(lastCompactIndex) : messages;
+}
 /**
  * Extracts displayable text from a Message object, handling various block types.
  * Returns the first available content block.

package/package.json

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

package/src/managers/messageManager.ts

@@ -15,6 +15,7 @@ import {
   type ToolBlockUpdateCallbackParams,
   type AddNotificationMessageParams,
   generateMessageId,
+  sliceFromLastCompact,
 } from "../utils/messageOperations.js";
 import type { Message, Usage, ToolBlock } from "../types/index.js";
 import { getLastApiRounds } from "../utils/groupMessagesByApiRound.js";
@@ -1014,9 +1015,12 @@ export class MessageManager {
       await reversionManager.revertTo(messageIdsToRemove, messages);
     }
 
-    // Rewrite file with truncated messages
+    // Rewrite file with truncated messages (full history kept on disk)
     await this.rewriteSessionFile(newMessages);
-    this.setMessages(newMessages);
+    // Fold the in-memory messages at the last compact boundary so the agent
+    // only sees messages from the latest compact summary forward — matching
+    // the compact and resume behaviors (which also fold memory).
+    this.setMessages(sliceFromLastCompact(newMessages));
     this.savedMessageCount = newMessages.length;
   }
 

package/src/services/session.ts

@@ -24,7 +24,10 @@ import { PathEncoder } from "../utils/pathEncoder.js";
 import { JsonlHandler } from "../services/jsonlHandler.js";
 import { extractLatestTotalTokens } from "../utils/tokenCalculation.js";
 import { logger } from "../utils/globalLogger.js";
-import { getMessageContent } from "../utils/messageOperations.js";
+import {
+  getMessageContent,
+  sliceFromLastCompact,
+} from "../utils/messageOperations.js";
 
 export interface SessionData {
   id: string;
@@ -222,15 +225,7 @@ export async function loadSessionFromJsonl(
     const allMessages = await jsonlHandler.read(filePath);
 
     // Find the last compact boundary — only return messages from there forward
-    let lastCompactIndex = -1;
-    for (let i = allMessages.length - 1; i >= 0; i--) {
-      if (allMessages[i].blocks?.some((b) => b.type === "compact")) {
-        lastCompactIndex = i;
-        break;
-      }
-    }
-    const messages =
-      lastCompactIndex >= 0 ? allMessages.slice(lastCompactIndex) : allMessages;
+    const messages = sliceFromLastCompact(allMessages);
 
     // Extract metadata from messages
     const lastMessage =

package/src/utils/messageOperations.ts

@@ -580,6 +580,24 @@ export function formatToolTokenSummary(
   return summary;
 }
 
+/**
+ * Fold a message list at the last compact boundary — returns only messages from
+ * the most recent compact summary forward. Mirrors what is sent to the LLM
+ * (convertMessagesForAPI stops at the last compact block), so the display layer
+ * matches the AI context. The session file keeps the full history; folding is
+ * display-only.
+ */
+export function sliceFromLastCompact(messages: Message[]): Message[] {
+  let lastCompactIndex = -1;
+  for (let i = messages.length - 1; i >= 0; i--) {
+    if (messages[i].blocks?.some((b) => b.type === "compact")) {
+      lastCompactIndex = i;
+      break;
+    }
+  }
+  return lastCompactIndex >= 0 ? messages.slice(lastCompactIndex) : messages;
+}
+
 /**
  * Extracts displayable text from a Message object, handling various block types.
  * Returns the first available content block.