wave-agent-sdk 0.13.6 → 0.14.0

package/dist/utils/groupMessagesByApiRound.d.ts

@@ -0,0 +1,24 @@
+import type { Message } from "../types/index.js";
+export interface ApiRound {
+    messages: Message[];
+    estimatedTokens: number;
+}
+/**
+ * Groups messages into "API rounds" — each round corresponds to one API
+ * call-response cycle. This is critical because in agentic sessions with a
+ * single user prompt, Wave creates a new Message per API round (each recursive
+ * sendAIMessage call creates a new assistant message).
+ *
+ * Boundaries:
+ *  - A new `role: "user"` message starts a new round.
+ *  - A new `role: "assistant"` message with a different `id` starts a new round.
+ *  - A message with a `compress` block is pushed as its own round and starts a
+ *    new round after it.
+ */
+export declare function groupMessagesByApiRound(messages: Message[]): ApiRound[];
+/**
+ * Returns the last `roundCount` complete API rounds as a flat message array.
+ * Never splits a tool_use/tool_result pair. If fewer rounds exist, returns all.
+ */
+export declare function getLastApiRounds(messages: Message[], roundCount: number): Message[];
+//# sourceMappingURL=groupMessagesByApiRound.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"groupMessagesByApiRound.d.ts","sourceRoot":"","sources":["../../src/utils/groupMessagesByApiRound.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEjD,MAAM,WAAW,QAAQ;IACvB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CA8DvE;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,OAAO,EAAE,EACnB,UAAU,EAAE,MAAM,GACjB,OAAO,EAAE,CAIX"}

package/dist/utils/groupMessagesByApiRound.js

@@ -0,0 +1,97 @@
+/**
+ * Groups messages into "API rounds" — each round corresponds to one API
+ * call-response cycle. This is critical because in agentic sessions with a
+ * single user prompt, Wave creates a new Message per API round (each recursive
+ * sendAIMessage call creates a new assistant message).
+ *
+ * Boundaries:
+ *  - A new `role: "user"` message starts a new round.
+ *  - A new `role: "assistant"` message with a different `id` starts a new round.
+ *  - A message with a `compress` block is pushed as its own round and starts a
+ *    new round after it.
+ */
+export function groupMessagesByApiRound(messages) {
+    const rounds = [];
+    let currentRound = [];
+    let lastAssistantId;
+    for (const msg of messages) {
+        let startNewRound = false;
+        if (msg.role === "user") {
+            startNewRound = true;
+        }
+        else if (msg.role === "assistant") {
+            // Compress block is always its own round
+            const hasCompress = msg.blocks.some((b) => b.type === "compress");
+            if (hasCompress) {
+                startNewRound = true;
+            }
+            else if (msg.id !== lastAssistantId) {
+                // New assistant id starts a new round.
+                // Exception: if the current round is [user] (first assistant after a
+                // user prompt in a normal conversation), keep them together as one
+                // round. But if we already have assistant(s) in this round (agentic
+                // tool loop), the new id starts a new round.
+                const roundHasOtherAssistant = currentRound.some((m) => m.role === "assistant" && m.id !== msg.id);
+                if (roundHasOtherAssistant) {
+                    startNewRound = true;
+                }
+            }
+            lastAssistantId = msg.id;
+        }
+        if (startNewRound && currentRound.length > 0) {
+            rounds.push({
+                messages: currentRound,
+                estimatedTokens: estimateTokens(currentRound),
+            });
+            currentRound = [];
+        }
+        currentRound.push(msg);
+        // After pushing a compress message as its own round, flush immediately
+        if (msg.role === "assistant" &&
+            msg.blocks.some((b) => b.type === "compress")) {
+            rounds.push({
+                messages: currentRound,
+                estimatedTokens: estimateTokens(currentRound),
+            });
+            currentRound = [];
+        }
+    }
+    if (currentRound.length > 0) {
+        rounds.push({
+            messages: currentRound,
+            estimatedTokens: estimateTokens(currentRound),
+        });
+    }
+    return rounds;
+}
+/**
+ * Returns the last `roundCount` complete API rounds as a flat message array.
+ * Never splits a tool_use/tool_result pair. If fewer rounds exist, returns all.
+ */
+export function getLastApiRounds(messages, roundCount) {
+    const rounds = groupMessagesByApiRound(messages);
+    const lastRounds = rounds.slice(-roundCount);
+    return lastRounds.flatMap((r) => r.messages);
+}
+/**
+ * Roughly estimate token count from character count (~4 chars per token).
+ */
+function estimateTokens(messages) {
+    let chars = 0;
+    for (const msg of messages) {
+        for (const block of msg.blocks) {
+            if ("content" in block && typeof block.content === "string") {
+                chars += block.content.length;
+            }
+            if (block.type === "tool" &&
+                block.parameters &&
+                typeof block.parameters === "string") {
+                chars += block.parameters.length;
+            }
+            if (block.type === "tool" && block.result) {
+                chars += block.result.length;
+            }
+        }
+    }
+    return Math.ceil(chars / 4);
+}

package/dist/utils/messageOperations.d.ts

@@ -41,6 +41,7 @@ export interface UpdateToolBlockParams {
     compactParams?: string;
     parametersChunk?: string;
     isManuallyBackgrounded?: boolean;
+    timestamp?: number;
 }
 export type AgentToolBlockUpdateParams = Omit<UpdateToolBlockParams, "messages">;
 export interface AddErrorBlockParams {

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

@@ -1 +1 @@
-{"version":3,"file":"messageOperations.d.ts","sourceRoot":"","sources":["../../src/utils/messageOperations.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,OAAO,EACP,KAAK,EAGN,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGlD,OAAO,EAAE,qCAAqC,EAAE,MAAM,qBAAqB,CAAC;AAI5E,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnD,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAGD,MAAM,WAAW,oBAAqB,SAAQ,iBAAiB;IAC7D,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,GAAG,KAAK,CAAC;IAClD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACrD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,sBAAsB,CAAC,EAAE,OAAO,CAAC;CAClC;AAGD,MAAM,MAAM,0BAA0B,GAAG,IAAI,CAC3C,qBAAqB,EACrB,UAAU,CACX,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;;GAIG;AACH,eAAO,MAAM,oBAAoB,GAAI,WAAW,MAAM,KAAG,MAmCxD,CAAC;AAEF,eAAO,MAAM,iBAAiB,QAAO,MAA+B,CAAC;AAGrE,eAAO,MAAM,wBAAwB,GAAI,0EAQtC,oBAAoB,KAAG,OAAO,EA6BhC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,2BAA2B,GACtC,UAAU,OAAO,EAAE,EACnB,IAAI,MAAM,EACV,QAAQ,OAAO,CAAC,iBAAiB,CAAC,KACjC,OAAO,EAwBT,CAAC;AAGF,eAAO,MAAM,6BAA6B,GACxC,UAAU,OAAO,EAAE,EACnB,UAAU,MAAM,EAChB,YAAY,qCAAqC,EAAE,EACnD,QAAQ,KAAK,EACb,mBAAmB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KACzC,OAAO,EA+BT,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,+BAA+B,GAC1C,UAAU,OAAO,EAAE,EACnB,WAAW,MAAM,EACjB,QAAQ,IAAI,CAAC,0BAA0B,EAAE,IAAI,CAAC,KAC7C;IAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAuB5C,CAAC;AAGF,eAAO,MAAM,wBAAwB,GAAI,6KAgBtC,qBAAqB,KAAG,OAAO,EAsFjC,CAAC;AAGF,eAAO,MAAM,sBAAsB,GAAI,sBAGpC,mBAAmB,KAAG,OAAO,EAgC/B,CAAC;AAGF,eAAO,MAAM,cAAc,GAAI,wBAG5B,aAAa,KAAG,OAAO,EAgBzB,CAAC;AAGF,eAAO,MAAM,mBAAmB,GAAI,gCAIjC,gBAAgB,KAAG,OAAO,EAmB5B,CAAC;AAGF,eAAO,MAAM,qBAAqB,GAAI,0CAKnC,kBAAkB,KAAG,OAAO,EAuB9B,CAAC;AAEF;;GAEG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAU3D;AAED;;;GAGG;AACH,eAAO,MAAM,qBAAqB,GAAI,UAAU,OAAO,EAAE,KAAG,OAAO,EASlE,CAAC;AAEF;;;GAGG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAoBtD;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,MAAM,CAUR;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAoB1D;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC;IAC5B,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,QAAQ,CAAC;IAC1C,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,eAAO,MAAM,gCAAgC,GAAI,8DAO9C,4BAA4B,KAAG,OAAO,EAiBxC,CAAC"}
+{"version":3,"file":"messageOperations.d.ts","sourceRoot":"","sources":["../../src/utils/messageOperations.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,OAAO,EACP,KAAK,EAGN,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGlD,OAAO,EAAE,qCAAqC,EAAE,MAAM,qBAAqB,CAAC;AAI5E,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnD,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAGD,MAAM,WAAW,oBAAqB,SAAQ,iBAAiB;IAC7D,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,GAAG,KAAK,CAAC;IAClD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACrD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,sBAAsB,CAAC,EAAE,OAAO,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAGD,MAAM,MAAM,0BAA0B,GAAG,IAAI,CAC3C,qBAAqB,EACrB,UAAU,CACX,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;;GAIG;AACH,eAAO,MAAM,oBAAoB,GAAI,WAAW,MAAM,KAAG,MAmCxD,CAAC;AAEF,eAAO,MAAM,iBAAiB,QAAO,MAA+B,CAAC;AAGrE,eAAO,MAAM,wBAAwB,GAAI,0EAQtC,oBAAoB,KAAG,OAAO,EA6BhC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,2BAA2B,GACtC,UAAU,OAAO,EAAE,EACnB,IAAI,MAAM,EACV,QAAQ,OAAO,CAAC,iBAAiB,CAAC,KACjC,OAAO,EAwBT,CAAC;AAGF,eAAO,MAAM,6BAA6B,GACxC,UAAU,OAAO,EAAE,EACnB,UAAU,MAAM,EAChB,YAAY,qCAAqC,EAAE,EACnD,QAAQ,KAAK,EACb,mBAAmB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KACzC,OAAO,EA+BT,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,+BAA+B,GAC1C,UAAU,OAAO,EAAE,EACnB,WAAW,MAAM,EACjB,QAAQ,IAAI,CAAC,0BAA0B,EAAE,IAAI,CAAC,KAC7C;IAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAuB5C,CAAC;AAGF,eAAO,MAAM,wBAAwB,GAAI,6KAgBtC,qBAAqB,KAAG,OAAO,EAsFjC,CAAC;AAGF,eAAO,MAAM,sBAAsB,GAAI,sBAGpC,mBAAmB,KAAG,OAAO,EAgC/B,CAAC;AAGF,eAAO,MAAM,cAAc,GAAI,wBAG5B,aAAa,KAAG,OAAO,EAgBzB,CAAC;AAGF,eAAO,MAAM,mBAAmB,GAAI,gCAIjC,gBAAgB,KAAG,OAAO,EAmB5B,CAAC;AAGF,eAAO,MAAM,qBAAqB,GAAI,0CAKnC,kBAAkB,KAAG,OAAO,EAuB9B,CAAC;AAEF;;GAEG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAU3D;AAED;;;GAGG;AACH,eAAO,MAAM,qBAAqB,GAAI,UAAU,OAAO,EAAE,KAAG,OAAO,EASlE,CAAC;AAEF;;;GAGG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAoBtD;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,MAAM,CAUR;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAoB1D;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC;IAC5B,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,QAAQ,CAAC;IAC1C,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,eAAO,MAAM,gCAAgC,GAAI,8DAO9C,4BAA4B,KAAG,OAAO,EAiBxC,CAAC"}

package/dist/utils/microcompact.d.ts

@@ -0,0 +1,7 @@
+import type { Message } from "../types/messaging.js";
+export interface MicrocompactOptions {
+    timeThresholdMS: number;
+    recentResultsToKeep: number;
+}
+export declare function microcompactMessages(messages: Message[], options: MicrocompactOptions): Message[];
+//# sourceMappingURL=microcompact.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"microcompact.d.ts","sourceRoot":"","sources":["../../src/utils/microcompact.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAa,MAAM,uBAAuB,CAAC;AAEhE,MAAM,WAAW,mBAAmB;IAClC,eAAe,EAAE,MAAM,CAAC;IACxB,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAID,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE,mBAAmB,GAC3B,OAAO,EAAE,CAwFX"}

package/dist/utils/microcompact.js

@@ -0,0 +1,78 @@
+const CLEARED_RESULT = "[Old tool result content cleared]";
+export function microcompactMessages(messages, options) {
+    const { timeThresholdMS, recentResultsToKeep } = options;
+    // 1. Find the latest tool block timestamp across all assistant messages
+    let lastAssistantTime = 0;
+    for (const msg of messages) {
+        if (msg.role === "assistant") {
+            for (const block of msg.blocks) {
+                if (block.type === "tool" && block.stage === "end" && block.timestamp) {
+                    if (block.timestamp > lastAssistantTime) {
+                        lastAssistantTime = block.timestamp;
+                    }
+                }
+            }
+        }
+    }
+    // 2. If no prior assistant messages with completed tools, return unchanged
+    if (lastAssistantTime === 0) {
+        return messages;
+    }
+    // 3. If within threshold, return unchanged
+    if (Date.now() - lastAssistantTime < timeThresholdMS) {
+        return messages;
+    }
+    const toolRefs = [];
+    for (let mi = 0; mi < messages.length; mi++) {
+        const msg = messages[mi];
+        if (msg.role === "assistant") {
+            for (let bi = 0; bi < msg.blocks.length; bi++) {
+                const block = msg.blocks[bi];
+                if (block.type === "tool" && block.stage === "end" && block.timestamp) {
+                    toolRefs.push({
+                        msgIndex: mi,
+                        blockIndex: bi,
+                        timestamp: block.timestamp,
+                    });
+                }
+            }
+        }
+    }
+    toolRefs.sort((a, b) => b.timestamp - a.timestamp);
+    // 5. Mark the top N as "keep"
+    const keepSet = new Set();
+    for (let i = 0; i < Math.min(recentResultsToKeep, toolRefs.length); i++) {
+        const ref = toolRefs[i];
+        keepSet.add(`${ref.msgIndex}:${ref.blockIndex}`);
+    }
+    // 6. Deep-copy messages and clear result + shortResult on non-kept blocks
+    const result = messages.map((msg) => ({
+        ...msg,
+        blocks: msg.blocks.map((block) => {
+            if (block.type === "tool" && block.stage === "end" && block.timestamp) {
+                return { ...block };
+            }
+            return block;
+        }),
+    }));
+    // Clear non-kept tool blocks
+    for (const ref of toolRefs) {
+        const key = `${ref.msgIndex}:${ref.blockIndex}`;
+        if (!keepSet.has(key)) {
+            result[ref.msgIndex] = {
+                ...result[ref.msgIndex],
+                blocks: result[ref.msgIndex].blocks.map((b, idx) => {
+                    if (idx === ref.blockIndex && b.type === "tool") {
+                        return {
+                            ...b,
+                            result: CLEARED_RESULT,
+                            shortResult: undefined,
+                        };
+                    }
+                    return b;
+                }),
+            };
+        }
+    }
+    return result;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-agent-sdk",
-  "version": "0.13.6",
+  "version": "0.14.0",
   "description": "SDK for building AI-powered development tools and agents",
   "keywords": [
     "ai",
@@ -33,6 +33,7 @@
     "cron-parser": "^5.5.0",
     "fuzzysort": "^3.1.0",
     "glob": "^13.0.0",
+    "lru-cache": "^11.3.5",
     "minimatch": "^10.0.3",
     "openai": "^5.12.2",
     "turndown": "^7.2.2"

package/src/agent.ts

@@ -579,11 +579,13 @@ export class Agent {
 
   /** Unified interrupt method, interrupts both AI messages and command execution */
   public abortMessage(): void {
+    // Clear queue first to prevent processQueuedMessage from dequeuing
+    // when abortAIMessage triggers onLoadingChange(false)
+    this.messageQueue.clear();
+    this.options.callbacks?.onQueuedMessagesChange?.(this.queuedMessages);
     this.abortAIMessage(); // This will abort tools including Agent tool (subagents)
     this.abortBashCommand();
     this.abortSlashCommand();
-    this.messageQueue.clear();
-    this.options.callbacks?.onQueuedMessagesChange?.(this.queuedMessages);
   }
 
   /** Interrupt bash command execution */

package/src/managers/aiManager.ts

@@ -1,6 +1,7 @@
 import { type CallAgentOptions } from "../services/aiService.js";
 import * as aiService from "../services/aiService.js";
 import { convertMessagesForAPI } from "../utils/convertMessagesForAPI.js";
+import { microcompactMessages } from "../utils/microcompact.js";
 import { parseTaskNotificationXml } from "../utils/notificationXml.js";
 import { calculateComprehensiveTotalTokens } from "../utils/tokenCalculation.js";
 import * as fs from "node:fs/promises";
@@ -15,7 +16,6 @@ import type { ToolManager } from "./toolManager.js";
 import type { ToolContext, ToolResult } from "../tools/types.js";
 import type { MessageManager } from "./messageManager.js";
 import type { BackgroundTaskManager } from "./backgroundTaskManager.js";
-import type { NotificationQueue } from "./notificationQueue.js";
 import { ChatCompletionMessageFunctionToolCall } from "openai/resources.js";
 import type { HookManager } from "./hookManager.js";
 import type { ExtendedHookExecutionContext } from "../types/hooks.js";
@@ -25,6 +25,7 @@ import type { SkillManager } from "./skillManager.js";
 import { buildSystemPrompt } from "../prompts/index.js";
 import { Container } from "../utils/container.js";
 import { ConfigurationService } from "../services/configurationService.js";
+import type { NotificationQueue } from "./notificationQueue.js";
 
 import { logger } from "../utils/globalLogger.js";
 
@@ -51,11 +52,13 @@ export class AIManager {
   onLoadingChange?: (loading: boolean) => void;
   private toolAbortController: AbortController | null = null;
   private workdir: string;
+  private originalWorkdir: string;
   private systemPrompt?: string;
   private subagentType?: string; // Store subagent type for hook context
   private stream: boolean; // Streaming mode flag
   private modelOverride?: string;
   private _onCwdChange?: (newCwd: string) => void; // Store callback for CWD changes
+  private consecutiveCompressionFailures: number = 0;
 
   // Service overrides
   constructor(
@@ -63,6 +66,7 @@ export class AIManager {
     options: AIManagerOptions,
   ) {
     this.workdir = options.workdir;
+    this.originalWorkdir = options.workdir;
     this.systemPrompt = options.systemPrompt;
     this.subagentType = options.subagentType; // Store subagent type
     this.stream = options.stream ?? true; // Default to true if not specified
@@ -165,6 +169,10 @@ export class AIManager {
     return this.workdir;
   }
 
+  public getOriginalWorkdir(): string {
+    return this.originalWorkdir;
+  }
+
   public setOnCwdChange(callback: (newCwd: string) => void): void {
     this._onCwdChange = callback;
   }
@@ -234,6 +242,7 @@ export class AIManager {
       if (toolPlugin?.formatCompactParams) {
         const context: ToolContext = {
           workdir: this.workdir,
+          originalWorkdir: this.originalWorkdir,
           taskManager: this.taskManager,
         };
         return toolPlugin.formatCompactParams(toolArgs, context);
@@ -248,7 +257,6 @@ export class AIManager {
   private async handleTokenUsageAndCompression(
     usage: Usage | undefined,
     abortController: AbortController,
-    model?: string,
   ): Promise<void> {
     if (!usage) return;
 
@@ -272,6 +280,14 @@ export class AIManager {
 
       // If there are messages to compress, perform compression
       if (messagesToCompress.length > 0) {
+        // Circuit breaker: skip compression after 3 consecutive failures
+        if (this.consecutiveCompressionFailures >= 3) {
+          logger?.warn(
+            `Skipping compression: ${this.consecutiveCompressionFailures} consecutive failures`,
+          );
+          return;
+        }
+
         const recentChatMessages = convertMessagesForAPI(messagesToCompress);
 
         // Save session before compression to preserve original messages
@@ -284,7 +300,7 @@ export class AIManager {
             modelConfig: this.getModelConfig(),
             messages: recentChatMessages,
             abortSignal: abortController.signal,
-            model: model,
+            model: this.getModelConfig().fastModel,
           });
 
           // Handle usage tracking for compression operations
@@ -294,14 +310,91 @@ export class AIManager {
               prompt_tokens: compressionResult.usage.prompt_tokens,
               completion_tokens: compressionResult.usage.completion_tokens,
               total_tokens: compressionResult.usage.total_tokens,
-              model: model || this.getModelConfig().model,
+              model: this.getModelConfig().fastModel,
               operation_type: "compress",
             };
           }
 
+          // Build post-compact context restoration
+          const POST_COMPACT_TOKEN_BUDGET = 50_000;
+          const POST_COMPACT_MAX_TOKENS_PER_FILE = 5_000;
+          const POST_COMPACT_MAX_FILES_TO_RESTORE = 5;
+          const contextParts: string[] = [];
+
+          // 1. File context restoration
+          const recentFiles = this.messageManager.getRecentFileReads(
+            POST_COMPACT_MAX_FILES_TO_RESTORE,
+            POST_COMPACT_MAX_TOKENS_PER_FILE,
+          );
+          let usedTokens = 0;
+          for (const file of recentFiles) {
+            const fileTokens = Math.ceil(file.content.length / 4);
+            if (usedTokens + fileTokens > POST_COMPACT_MAX_TOKENS_PER_FILE)
+              continue;
+            if (fileTokens > 0) usedTokens += fileTokens;
+            contextParts.push(
+              `\n\n## ${file.path}\n\`\`\`\n${file.content}\n\`\`\``,
+            );
+            if (contextParts.length >= POST_COMPACT_MAX_FILES_TO_RESTORE) break;
+            if (usedTokens >= POST_COMPACT_TOKEN_BUDGET) break;
+          }
+
+          // 2. Working directory
+          contextParts.push(
+            `\n\n[Working Directory]\nCurrent working directory: ${this.workdir}`,
+          );
+
+          // 3. Plan mode context
+          const currentMode = this.permissionManager?.getCurrentEffectiveMode(
+            this.getModelConfig().permissionMode,
+          );
+          if (currentMode === "plan") {
+            const planFilePath = this.permissionManager?.getPlanFilePath();
+            if (planFilePath) {
+              let planExists = false;
+              try {
+                await fs.access(planFilePath);
+                planExists = true;
+              } catch {
+                // Plan file doesn't exist yet
+              }
+              contextParts.push(
+                `\n\n[Plan Mode]\nYou are in plan mode. Plan file: ${planFilePath} (exists: ${planExists})`,
+              );
+            }
+          }
+
+          // 4. Skills context
+          const skills =
+            this.skillManager
+              ?.getAvailableSkills()
+              .filter((s) => !s.disableModelInvocation) || [];
+          if (skills.length > 0) {
+            const skillList = skills
+              .map((s) => `- ${s.name}: ${s.description || ""}`)
+              .join("\n");
+            contextParts.push(`\n\n[Available Skills]\n${skillList}`);
+          }
+
+          // 5. Background agents status
+          const agents = this.backgroundTaskManager?.getAllTasks() || [];
+          if (agents.length > 0) {
+            const agentList = agents
+              .map((a) => `- Agent "${a.description}": ${a.status}`)
+              .join("\n");
+            contextParts.push(`\n\n[Background Tasks]\n${agentList}`);
+          }
+
+          // Merge context restoration into summary
+          const enhancedSummary =
+            compressionResult.content +
+            (contextParts.length > 0
+              ? `\n\n[Context Restoration]` + contextParts.join("")
+              : "");
+
           // Execute message reconstruction and sessionId update after compression
           this.messageManager.compressMessagesAndUpdateSession(
-            compressionResult.content,
+            enhancedSummary,
             compressionUsage,
           );
 
@@ -313,8 +406,13 @@ export class AIManager {
           logger?.debug(
             `Successfully compressed ${messagesToCompress.length} messages and updated session`,
           );
+          this.consecutiveCompressionFailures = 0;
         } catch (compressError) {
-          logger?.error("Failed to compress messages:", compressError);
+          this.consecutiveCompressionFailures++;
+          logger?.error(
+            `Failed to compress messages (${this.consecutiveCompressionFailures} consecutive):`,
+            compressError,
+          );
           this.messageManager.addErrorBlock(
             `Failed to compress conversation history: ${compressError instanceof Error ? compressError.message : String(compressError)}. You may encounter context limit issues.`,
           );
@@ -403,10 +501,13 @@ export class AIManager {
       toolAbortController = this.toolAbortController!;
     }
 
-    // Get recent message history
-    const recentMessages = convertMessagesForAPI(
-      this.messageManager.getMessages(),
-    );
+    // Get recent message history with microcompact applied
+    const rawMessages = this.messageManager.getMessages();
+    const microcompactedMessages = microcompactMessages(rawMessages, {
+      timeThresholdMS: 30 * 60 * 1000, // 30 minutes
+      recentResultsToKeep: 3,
+    });
+    const recentMessages = convertMessagesForAPI(microcompactedMessages);
 
     try {
       // Get combined memory content
@@ -472,6 +573,7 @@ export class AIManager {
           filteredToolPlugins,
           {
             workdir: this.workdir,
+            originalWorkdir: this.originalWorkdir,
             memory: combinedMemory,
             language: this.getLanguage(),
             isSubagent: !!this.subagentType,
@@ -658,6 +760,7 @@ export class AIManager {
                   stage: "end",
                   name: toolName,
                   compactParams: "",
+                  timestamp: Date.now(),
                 });
                 return;
               }
@@ -710,6 +813,7 @@ export class AIManager {
                 abortSignal: toolAbortController.signal,
                 backgroundTaskManager: this.backgroundTaskManager,
                 workdir: this.workdir,
+                originalWorkdir: this.originalWorkdir,
                 messageId: this.messageManager.getMessages().slice(-1)[0]?.id,
                 sessionId: this.messageManager.getSessionId(),
                 toolCallId: toolId,
@@ -774,6 +878,7 @@ export class AIManager {
                 shortResult: toolResult.shortResult,
                 isManuallyBackgrounded: toolResult.isManuallyBackgrounded,
                 startLineNumber: toolResult.startLineNumber,
+                timestamp: Date.now(),
               });
 
               // Execute PostToolUse hooks after successful tool completion
@@ -799,6 +904,7 @@ export class AIManager {
                 name: toolName,
                 compactParams,
                 isManuallyBackgrounded: false,
+                timestamp: Date.now(),
               });
             }
           },
@@ -809,11 +915,7 @@ export class AIManager {
       }
 
       // Handle token statistics and message compression
-      await this.handleTokenUsageAndCompression(
-        result.usage,
-        abortController,
-        model,
-      );
+      await this.handleTokenUsageAndCompression(result.usage, abortController);
 
       // Finalize text/reasoning blocks for the final response (no tools)
       this.messageManager.finalizeStreamingBlocks();

package/src/managers/messageManager.ts

@@ -16,6 +16,7 @@ import {
   generateMessageId,
 } from "../utils/messageOperations.js";
 import type { Message, Usage } from "../types/index.js";
+import { getLastApiRounds } from "../utils/groupMessagesByApiRound.js";
 import { join, isAbsolute, relative } from "path";
 import {
   appendMessages,
@@ -89,6 +90,8 @@ export class MessageManager {
   private transcriptPath: string; // Cached transcript path
   private savedMessageCount: number; // Track how many messages have been saved to prevent duplication
   private filesInContext: Set<string> = new Set(); // Track files mentioned in the conversation
+  private recentFileReads: Map<string, { content: string; timestamp: number }> =
+    new Map(); // Track file read contents
   private sessionType: "main" | "subagent";
   private subagentType?: string;
   private _usages: Usage[] = [];
@@ -266,11 +269,13 @@ export class MessageManager {
     const newMessages = messages.slice(oldLength);
     for (const message of newMessages) {
       this.addPathsFromMessage(message);
+      this.extractFileReadsFromMessage(message);
     }
 
     // Also check if the last message was updated (common for tool blocks)
     if (messages.length > 0 && messages.length === oldLength) {
       this.addPathsFromMessage(messages[messages.length - 1]);
+      this.extractFileReadsFromMessage(messages[messages.length - 1]);
     }
 
     this.callbacks.onMessagesChange?.([...messages]);
@@ -495,8 +500,8 @@ export class MessageManager {
     compressedContent: string,
     usage?: Usage,
   ): void {
-    // Get last 3 messages to preserve
-    const lastThreeMessages = this.messages.slice(-3);
+    // Get last 2 API rounds to preserve (structurally safe boundary)
+    const lastThreeMessages = getLastApiRounds(this.messages, 2);
 
     // Create compressed message
     const compressMessage: Message = {
@@ -994,4 +999,61 @@ export class MessageManager {
 
     return paths;
   }
+
+  /**
+   * Extract file read contents from tool result blocks in a message.
+   */
+  private extractFileReadsFromMessage(message: Message): void {
+    for (const block of message.blocks) {
+      if (
+        block.type === "tool" &&
+        block.name === "read" &&
+        block.stage === "end" &&
+        block.result &&
+        block.parameters
+      ) {
+        let filePath: string | undefined;
+        try {
+          const params = JSON.parse(block.parameters) as Record<
+            string,
+            unknown
+          >;
+          filePath = params.file_path as string | undefined;
+        } catch {
+          // Ignore parse errors
+        }
+        if (filePath) {
+          this.recentFileReads.set(filePath, {
+            content: block.result,
+            timestamp: Date.now(),
+          });
+        }
+      }
+    }
+  }
+
+  /**
+   * Get recent file read contents, sorted by timestamp (newest first).
+   * @param maxFiles - Maximum number of files to return
+   * @param maxTokensPerFile - Maximum tokens per file (~4 chars/token)
+   * @returns Array of { path, content } sorted by recency
+   */
+  public getRecentFileReads(
+    maxFiles = 5,
+    maxTokensPerFile = 5000,
+  ): Array<{ path: string; content: string }> {
+    const sorted = Array.from(this.recentFileReads.entries())
+      .sort(([, a], [, b]) => b.timestamp - a.timestamp)
+      .slice(0, maxFiles);
+
+    const result: Array<{ path: string; content: string }> = [];
+    for (const [path, { content }] of sorted) {
+      const truncated =
+        content.length > maxTokensPerFile * 4
+          ? content.slice(0, maxTokensPerFile * 4)
+          : content;
+      result.push({ path, content: truncated });
+    }
+    return result;
+  }
 }

package/src/managers/permissionManager.ts

@@ -315,6 +315,13 @@ export class PermissionManager {
     return this.planFilePath;
   }
 
+  /**
+   * Public wrapper for isInsideSafeZone to check if a path is in the safe zone
+   */
+  public isPathInSafeZone(targetPath: string): boolean {
+    return this.isInsideSafeZone(targetPath).isInside;
+  }
+
   /**
    * Check if a path is inside the Safe Zone (workdir + additionalDirectories)
    */