wave-code 0.12.0 → 0.12.2

package/dist/contexts/useChat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useChat.d.ts","sourceRoot":"","sources":["../../src/contexts/useChat.tsx"],"names":[],"mappings":"AAAA,OAAO,KAON,MAAM,OAAO,CAAC;AAGf,OAAO,KAAK,EACV,OAAO,EACP,eAAe,EACf,cAAc,EACd,IAAI,EACJ,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACf,MAAM,gBAAgB,CAAC;AAWxB,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,aAAa,EAAE,OAAO,CAAC;IAEvB,UAAU,EAAE,OAAO,CAAC;IACpB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,oBAAoB,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACjD,cAAc,EAAE,KAAK,CAAC;QACpB,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,QAAQ,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACnD,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACtC,CAAC,CAAC;IAEH,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,CACX,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,EAClD,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KACjC,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,MAAM,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9C,YAAY,EAAE,MAAM,IAAI,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAE1B,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,mBAAmB,EAAE,MAAM,MAAM,EAAE,CAAC;IACpC,QAAQ,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAElC,UAAU,EAAE,eAAe,EAAE,CAAC;IAC9B,gBAAgB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3D,mBAAmB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAE9D,eAAe,EAAE,cAAc,EAAE,CAAC;IAElC,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,uBAAuB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK;QAC3C,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,IAAI,CAAC;IACT,kBAAkB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,aAAa,EAAE,YAAY,EAAE,CAAC;IAC9B,eAAe,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5C,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAE7C,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IAClD,kBAAkB,EAAE,OAAO,CAAC;IAE5B,qBAAqB,EAAE,OAAO,CAAC;IAC/B,cAAc,CAAC,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChC,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,oBAAoB,CAAC,EAAE,OAAO,CAAC;QAC/B,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,gBAAgB,EAAE,CAChB,QAAQ,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,eAAe,CAAC,EAAE,MAAM,EACxB,oBAAoB,CAAC,EAAE,OAAO,EAC9B,WAAW,CAAC,EAAE,MAAM,KACjB,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,gBAAgB,EAAE,MAAM,IAAI,CAAC;IAC7B,0BAA0B,EAAE,CAAC,QAAQ,EAAE,kBAAkB,KAAK,IAAI,CAAC;IACnE,wBAAwB,EAAE,MAAM,IAAI,CAAC;IAErC,qBAAqB,EAAE,MAAM,IAAI,CAAC;IAElC,QAAQ,EAAE,MAAM,CAAC;IACjB,kBAAkB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,oBAAoB,EAAE,MAAM,OAAO,CAAC;QAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,CAAC,CAAC;IACH,qBAAqB,EAAE,MAAM,CAAC;IAC9B,wBAAwB,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;IAEvE,gBAAgB,EAAE,MAAM,OAAO,gBAAgB,EAAE,aAAa,CAAC;IAC/D,cAAc,EAAE,MAAM,OAAO,gBAAgB,EAAE,WAAW,CAAC;IAC3D,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,6BAA6B,EAAE,QAAQ,CAAC;IACzD,WAAW,EAAE,CACX,OAAO,EAAE,OAAO,CAAC,OAAO,6BAA6B,EAAE,QAAQ,CAAC,KAC7D,IAAI,CAAC;CACX;AAID,eAAO,MAAM,OAAO,uBAMnB,CAAC;AAEF,MAAM,WAAW,iBAAkB,SAAQ,YAAY;IACrD,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;CAC3B;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAqpBpD,CAAC"}
+{"version":3,"file":"useChat.d.ts","sourceRoot":"","sources":["../../src/contexts/useChat.tsx"],"names":[],"mappings":"AAAA,OAAO,KAQN,MAAM,OAAO,CAAC;AAGf,OAAO,KAAK,EACV,OAAO,EACP,eAAe,EACf,cAAc,EACd,IAAI,EACJ,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACf,MAAM,gBAAgB,CAAC;AAaxB,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,aAAa,EAAE,OAAO,CAAC;IAEvB,UAAU,EAAE,OAAO,CAAC;IACpB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,oBAAoB,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACjD,cAAc,EAAE,KAAK,CAAC;QACpB,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,QAAQ,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACnD,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACtC,CAAC,CAAC;IAEH,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,CACX,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,EAClD,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KACjC,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,MAAM,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9C,YAAY,EAAE,MAAM,IAAI,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAE1B,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,mBAAmB,EAAE,MAAM,MAAM,EAAE,CAAC;IACpC,QAAQ,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAElC,UAAU,EAAE,eAAe,EAAE,CAAC;IAC9B,gBAAgB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3D,mBAAmB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAE9D,eAAe,EAAE,cAAc,EAAE,CAAC;IAElC,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,uBAAuB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK;QAC3C,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,IAAI,CAAC;IACT,kBAAkB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,aAAa,EAAE,YAAY,EAAE,CAAC;IAC9B,eAAe,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IAClD,kBAAkB,EAAE,OAAO,CAAC;IAE5B,qBAAqB,EAAE,OAAO,CAAC;IAC/B,cAAc,CAAC,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChC,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,oBAAoB,CAAC,EAAE,OAAO,CAAC;QAC/B,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,gBAAgB,EAAE,CAChB,QAAQ,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,eAAe,CAAC,EAAE,MAAM,EACxB,oBAAoB,CAAC,EAAE,OAAO,EAC9B,WAAW,CAAC,EAAE,MAAM,KACjB,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,gBAAgB,EAAE,MAAM,IAAI,CAAC;IAC7B,0BAA0B,EAAE,CAAC,QAAQ,EAAE,kBAAkB,KAAK,IAAI,CAAC;IACnE,wBAAwB,EAAE,MAAM,IAAI,CAAC;IAErC,qBAAqB,EAAE,MAAM,IAAI,CAAC;IAElC,QAAQ,EAAE,MAAM,CAAC;IACjB,kBAAkB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,oBAAoB,EAAE,MAAM,OAAO,CAAC;QAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,CAAC,CAAC;IACH,qBAAqB,EAAE,MAAM,CAAC;IAC9B,wBAAwB,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;IAEvE,gBAAgB,EAAE,MAAM,OAAO,gBAAgB,EAAE,aAAa,CAAC;IAC/D,cAAc,EAAE,MAAM,OAAO,gBAAgB,EAAE,WAAW,CAAC;IAC3D,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,6BAA6B,EAAE,QAAQ,CAAC;IACzD,WAAW,EAAE,CACX,OAAO,EAAE,OAAO,CAAC,OAAO,6BAA6B,EAAE,QAAQ,CAAC,KAC7D,IAAI,CAAC;CACX;AAID,eAAO,MAAM,OAAO,uBAMnB,CAAC;AAEF,MAAM,WAAW,iBAAkB,SAAQ,YAAY;IACrD,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;CAC3B;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CA4nBpD,CAAC"}

package/dist/contexts/useChat.js

@@ -1,9 +1,10 @@
 import { jsx as _jsx } from "react/jsx-runtime";
-import { createContext, useContext, useCallback, useRef, useEffect, useState, } from "react";
+import { createContext, useContext, useCallback, useRef, useEffect, useState, useMemo, } from "react";
 import { useInput } from "ink";
 import { useAppConfig } from "./useAppConfig.js";
-import { Agent, OPERATION_CANCELLED_BY_USER, } from "wave-agent-sdk";
+import { Agent, OPERATION_CANCELLED_BY_USER, cloneMessage, } from "wave-agent-sdk";
 import { logger } from "../utils/logger.js";
+import { throttle } from "../utils/throttle.js";
 import { displayUsageSummary } from "../utils/usageSummary.js";
 import { expandLongTextPlaceholders } from "../managers/inputHandlers.js";
 const ChatContext = createContext(null);
@@ -19,13 +20,25 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
     // Message Display State
     const [isExpanded, setIsExpanded] = useState(false);
     const isExpandedRef = useRef(isExpanded);
-    useEffect(() => {
-        isExpandedRef.current = isExpanded;
-    }, [isExpanded]);
     const [isTaskListVisible, setIsTaskListVisible] = useState(true);
     const [queuedMessages, setQueuedMessages] = useState([]);
-    // AI State
     const [messages, setMessages] = useState([]);
+    const throttledSetMessages = useMemo(() => throttle(() => {
+        if (!isExpandedRef.current && agentRef.current) {
+            setMessages([...agentRef.current.messages]);
+        }
+    }, 300), []);
+    useEffect(() => {
+        isExpandedRef.current = isExpanded;
+        if (isExpanded) {
+            throttledSetMessages.cancel();
+        }
+    }, [isExpanded, throttledSetMessages]);
+    useEffect(() => {
+        return () => {
+            throttledSetMessages.cancel();
+        };
+    }, [throttledSetMessages]);
     const [isLoading, setIsLoading] = useState(false);
     const [latestTotalTokens, setlatestTotalTokens] = useState(0);
     const [sessionId, setSessionId] = useState("");
@@ -41,9 +54,6 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
     const [tasks, setTasks] = useState([]);
     // Command state
     const [slashCommands, setSlashCommands] = useState([]);
-    // Subagent messages state
-    const [subagentMessages, setSubagentMessages] = useState({});
-    const [subagentLatestTokens, setSubagentLatestTokens] = useState({});
     // Permission state
     const [permissionMode, setPermissionModeState] = useState(initialPermissionMode ||
         (bypassPermissions ? "bypassPermissions" : "default"));
@@ -75,16 +85,6 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
     const [wasLastDetailsTooTall, setWasLastDetailsTooTall] = useState(0);
     const allowBypassInCycle = !!bypassPermissions || initialPermissionMode === "bypassPermissions";
     const agentRef = useRef(null);
-    const taskUpdateTimerRef = useRef(null);
-    const debouncedSetTasks = useCallback((newTasks) => {
-        if (taskUpdateTimerRef.current) {
-            clearTimeout(taskUpdateTimerRef.current);
-        }
-        taskUpdateTimerRef.current = setTimeout(() => {
-            setTasks([...newTasks]);
-            taskUpdateTimerRef.current = null;
-        }, 100);
-    }, [setTasks]);
     // Permission confirmation methods with queue support
     const showConfirmation = useCallback(async (toolName, toolInput, suggestedPrefix, hidePersistentOption, planContent) => {
         return new Promise((resolve, reject) => {
@@ -106,9 +106,7 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
         const initializeAgent = async () => {
             const callbacks = {
                 onMessagesChange: () => {
-                    if (!isExpandedRef.current && agentRef.current) {
-                        setMessages([...agentRef.current.messages]);
-                    }
+                    throttledSetMessages();
                 },
                 onServersChange: (servers) => {
                     setMcpServers([...servers]);
@@ -126,20 +124,7 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
                     setBackgroundTasks([...tasks]);
                 },
                 onTasksChange: (tasks) => {
-                    debouncedSetTasks(tasks);
-                },
-                onSubagentMessagesChange: (subagentId, messages) => {
-                    logger.debug("onSubagentMessagesChange", subagentId, messages.length);
-                    setSubagentMessages((prev) => ({
-                        ...prev,
-                        [subagentId]: [...messages],
-                    }));
-                },
-                onSubagentLatestTotalTokensChange: (subagentId, tokens) => {
-                    setSubagentLatestTokens((prev) => ({
-                        ...prev,
-                        [subagentId]: tokens,
-                    }));
+                    setTasks([...tasks]);
                 },
                 onPermissionModeChange: (mode) => {
                     setPermissionModeState(mode);
@@ -221,15 +206,12 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
         workdir,
         worktreeSession,
         model,
-        debouncedSetTasks,
         initialPermissionMode,
+        throttledSetMessages,
     ]);
     // Cleanup on unmount
     useEffect(() => {
         return () => {
-            if (taskUpdateTimerRef.current) {
-                clearTimeout(taskUpdateTimerRef.current);
-            }
             if (agentRef.current) {
                 try {
                     // Display usage summary before cleanup
@@ -270,7 +252,7 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
                 if (command) {
                     setIsCommandRunning(true);
                     try {
-                        await agentRef.current?.executeBashCommand(command);
+                        await agentRef.current?.bang(command);
                     }
                     finally {
                         setIsCommandRunning(false);
@@ -434,14 +416,17 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
             // Clear terminal screen when expanded state changes
             setIsExpanded((prev) => {
                 const newExpanded = !prev;
+                isExpandedRef.current = newExpanded;
                 if (newExpanded) {
                     // Transitioning to EXPANDED: Freeze the current view
+                    // Cancel any pending throttled updates to avoid overwriting the frozen state
+                    throttledSetMessages.cancel();
                     // Deep copy the last message to ensure it doesn't update if the agent is still writing to it
                     setMessages((currentMessages) => {
                         if (currentMessages.length === 0)
                             return currentMessages;
                         const lastMessage = currentMessages[currentMessages.length - 1];
-                        const frozenLastMessage = JSON.parse(JSON.stringify(lastMessage));
+                        const frozenLastMessage = cloneMessage(lastMessage);
                         return [...currentMessages.slice(0, -1), frozenLastMessage];
                     });
                 }
@@ -499,8 +484,6 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
         stopBackgroundTask,
         slashCommands,
         hasSlashCommand,
-        subagentMessages,
-        subagentLatestTokens,
         permissionMode,
         setPermissionMode,
         allowBypassInCycle,

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAeA,wBAAsB,IAAI,kBAoXzB;AAGD,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAGpC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAG3C,OAAO,EACL,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,KAAK,oBAAoB,GAC1B,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAgBA,wBAAsB,IAAI,kBAgYzB;AAGD,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAGpC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAG3C,OAAO,EACL,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,KAAK,oBAAoB,GAC1B,MAAM,sBAAsB,CAAC"}

package/dist/index.js

@@ -1,6 +1,7 @@
 import yargs from "yargs";
 import { hideBin } from "yargs/helpers";
 import { startCli } from "./cli.js";
+import { logger } from "./utils/logger.js";
 import { generateRandomName } from "wave-agent-sdk";
 import { createWorktree } from "./utils/worktree.js";
 import path from "path";
@@ -12,6 +13,15 @@ const packageJson = JSON.parse(readFileSync(packageJsonPath, "utf-8"));
 const version = packageJson.version;
 // Export main function for external use
 export async function main() {
+    // Start memory monitoring in debug mode
+    if (process.env.LOG_LEVEL === "DEBUG") {
+        setInterval(() => {
+            const usage = process.memoryUsage();
+            logger.debug(`[Memory] RSS: ${Math.round(usage.rss / 1024 / 1024)}MB, ` +
+                `Heap: ${Math.round(usage.heapUsed / 1024 / 1024)}/${Math.round(usage.heapTotal / 1024 / 1024)}MB, ` +
+                `External: ${Math.round(usage.external / 1024 / 1024)}MB`);
+        }, 10000).unref();
+    }
     try {
         const originalCwd = process.cwd();
         const argv = await yargs(hideBin(process.argv))

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

@@ -1 +1 @@
-{"version":3,"file":"highlightUtils.d.ts","sourceRoot":"","sources":["../../src/utils/highlightUtils.ts"],"names":[],"mappings":"AAqEA,wBAAgB,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,CAsCvE"}
+{"version":3,"file":"highlightUtils.d.ts","sourceRoot":"","sources":["../../src/utils/highlightUtils.ts"],"names":[],"mappings":"AA6DA,wBAAgB,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,CAsCvE"}

package/dist/utils/highlightUtils.js

@@ -45,14 +45,6 @@ function nodeToAnsi(node) {
         const classes = node.getAttribute("class")?.split(/\s+/) || [];
         for (const className of classes) {
             if (theme[className]) {
-                // If content has newlines, split it and apply style to each line
-                // to ensure ANSI codes are correctly applied when splitting the final string by lines.
-                if (content.includes("\n")) {
-                    return content
-                        .split("\n")
-                        .map((line) => theme[className](line))
-                        .join("\n");
-                }
                 return theme[className](content);
             }
         }

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

@@ -1 +1 @@
-{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../src/utils/logger.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAUH;;GAEG;AACH,oBAAY,QAAQ;IAClB,KAAK,IAAI;IACT,IAAI,IAAI;IACR,IAAI,IAAI;IACR,KAAK,IAAI;CACV;AAsBD;;GAEG;AACH,UAAU,SAAS;IACjB,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC;IACzB,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;CAC7B;AA+HD;;GAEG;AACH,eAAO,MAAM,MAAM;IACjB;;OAEG;qBACc,OAAO,EAAE,KAAG,IAAI;IAIjC;;OAEG;oBACa,OAAO,EAAE,KAAG,IAAI;IAIhC;;OAEG;oBACa,OAAO,EAAE,KAAG,IAAI;IAIhC;;OAEG;qBACc,OAAO,EAAE,KAAG,IAAI;CAGlC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,YAAY,QAAO,SAE/B,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,UAAU,QAAO,MAE7B,CAAC;AAEF;;GAEG;AACH,UAAU,gBAAgB;IACxB,6DAA6D;IAC7D,WAAW,EAAE,MAAM,CAAC;IACpB,kEAAkE;IAClE,SAAS,EAAE,MAAM,CAAC;CACnB;AAqDD;;;;;GAKG;AACH,eAAO,MAAM,WAAW,GACtB,eAAe,OAAO,CAAC,gBAAgB,CAAC,KACvC,OAAO,CAAC,IAAI,CAiBd,CAAC"}
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../src/utils/logger.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAWH;;GAEG;AACH,oBAAY,QAAQ;IAClB,KAAK,IAAI;IACT,IAAI,IAAI;IACR,IAAI,IAAI;IACR,KAAK,IAAI;CACV;AAsBD;;GAEG;AACH,UAAU,SAAS;IACjB,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC;IACzB,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;CAC7B;AAiID;;GAEG;AACH,eAAO,MAAM,MAAM;IACjB;;OAEG;qBACc,OAAO,EAAE,KAAG,IAAI;IAIjC;;OAEG;oBACa,OAAO,EAAE,KAAG,IAAI;IAIhC;;OAEG;oBACa,OAAO,EAAE,KAAG,IAAI;IAIhC;;OAEG;qBACc,OAAO,EAAE,KAAG,IAAI;CAGlC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,YAAY,QAAO,SAE/B,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,UAAU,QAAO,MAE7B,CAAC;AAEF;;GAEG;AACH,UAAU,gBAAgB;IACxB,6DAA6D;IAC7D,WAAW,EAAE,MAAM,CAAC;IACpB,kEAAkE;IAClE,SAAS,EAAE,MAAM,CAAC;CACnB;AAgDD;;;;;GAKG;AACH,eAAO,MAAM,WAAW,GACtB,eAAe,OAAO,CAAC,gBAAgB,CAAC,KACvC,OAAO,CAAC,IAAI,CAiBd,CAAC"}

package/dist/utils/logger.js

@@ -9,6 +9,7 @@
  */
 import * as fs from "fs";
 import { Chalk } from "chalk";
+import { getLastLines } from "wave-agent-sdk";
 import { LOG_FILE, DATA_DIRECTORY } from "./constants.js";
 const chalk = new Chalk({ level: 3 });
 const logFile = process.env.LOG_FILE || LOG_FILE;
@@ -106,20 +107,25 @@ const formatArg = (arg) => {
         return "null";
     if (arg === undefined)
         return "undefined";
+    let result;
     if (arg instanceof Error) {
         // Special handling for Error objects, display stack or message
-        return arg.stack || arg.message || String(arg);
+        result = arg.stack || arg.message || String(arg);
     }
-    if (typeof arg === "object") {
+    else if (typeof arg === "object") {
         try {
-            return JSON.stringify(arg, null, 2);
+            result = JSON.stringify(arg);
         }
         catch {
             // If JSON.stringify fails, fallback to String()
-            return String(arg);
+            result = String(arg);
         }
     }
-    return String(arg);
+    else {
+        result = String(arg);
+    }
+    // Ensure no newlines in the result to keep log entries single-line
+    return result.replace(/[\n\r]+/g, " ");
 };
 /**
  * Generic log output function
@@ -220,15 +226,11 @@ const truncateLogFileIfNeeded = (config) => {
         // If file size exceeds limit, truncate file
         if (stats.size > config.maxFileSize) {
             const content = fs.readFileSync(logFile, "utf8");
-            const lines = content.split("\n");
-            // Keep the last specified number of lines
-            const keepLines = Math.min(config.keepLines, lines.length);
-            const truncatedContent = lines.slice(-keepLines).join("\n");
+            const truncatedContent = getLastLines(content, config.keepLines);
             // Write truncated content
             fs.writeFileSync(logFile, truncatedContent);
             // Record truncation operation
-            const removedLines = lines.length - keepLines;
-            logger.debug(`Log file truncated: removed ${removedLines} lines, kept last ${keepLines} lines`);
+            logger.debug(`Log file truncated: file size ${stats.size} exceeded limit ${config.maxFileSize}, kept last ${config.keepLines} lines`);
         }
     }
     catch (error) {

package/dist/utils/throttle.d.ts

@@ -0,0 +1,15 @@
+export interface ThrottleOptions {
+    leading?: boolean;
+    trailing?: boolean;
+}
+export interface ThrottledFunction<T extends (...args: unknown[]) => void> {
+    (...args: Parameters<T>): void;
+    cancel: () => void;
+    flush: () => void;
+}
+/**
+ * Creates a throttled function that only invokes `func` at most once per
+ * every `wait` milliseconds.
+ */
+export declare function throttle<T extends (...args: unknown[]) => void>(func: T, wait: number, options?: ThrottleOptions): ThrottledFunction<T>;
+//# sourceMappingURL=throttle.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"throttle.d.ts","sourceRoot":"","sources":["../../src/utils/throttle.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI;IACvE,CAAC,GAAG,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAC/B,MAAM,EAAE,MAAM,IAAI,CAAC;IACnB,KAAK,EAAE,MAAM,IAAI,CAAC;CACnB;AAED;;;GAGG;AACH,wBAAgB,QAAQ,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,EAC7D,IAAI,EAAE,CAAC,EACP,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,eAAmD,GAC3D,iBAAiB,CAAC,CAAC,CAAC,CAuDtB"}

package/dist/utils/throttle.js

@@ -0,0 +1,55 @@
+/**
+ * Creates a throttled function that only invokes `func` at most once per
+ * every `wait` milliseconds.
+ */
+export function throttle(func, wait, options = { leading: true, trailing: true }) {
+    let timeoutId = null;
+    let lastArgs = null;
+    let lastCallTime = 0;
+    const invokeFunc = () => {
+        if (lastArgs) {
+            func(...lastArgs);
+            lastCallTime = Date.now();
+            lastArgs = null;
+        }
+    };
+    const throttled = (...args) => {
+        const now = Date.now();
+        const remaining = wait - (now - lastCallTime);
+        lastArgs = args;
+        if (remaining <= 0 || remaining > wait) {
+            if (timeoutId) {
+                clearTimeout(timeoutId);
+                timeoutId = null;
+            }
+            if (options.leading !== false || lastCallTime !== 0) {
+                invokeFunc();
+            }
+            else {
+                lastCallTime = now;
+            }
+        }
+        else if (!timeoutId && options.trailing !== false) {
+            timeoutId = setTimeout(() => {
+                timeoutId = null;
+                invokeFunc();
+            }, remaining);
+        }
+    };
+    throttled.cancel = () => {
+        if (timeoutId) {
+            clearTimeout(timeoutId);
+            timeoutId = null;
+        }
+        lastArgs = null;
+        lastCallTime = 0;
+    };
+    throttled.flush = () => {
+        if (timeoutId) {
+            clearTimeout(timeoutId);
+            timeoutId = null;
+        }
+        invokeFunc();
+    };
+    return throttled;
+}

package/dist/utils/toolParameterTransforms.d.ts

@@ -2,7 +2,11 @@
  * Tool parameter transformation utilities for UI rendering
  * Forces type judgment based on tool name using type assertions
  */
-import { type Change, type EditToolParameters } from "wave-agent-sdk";
+import { type Change, type WriteToolParameters, type EditToolParameters } from "wave-agent-sdk";
+/**
+ * Transform Write tool parameters to changes
+ */
+export declare function transformWriteParameters(parameters: WriteToolParameters): Change[];
 /**
  * Transform Edit tool parameters to changes
  */

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

@@ -1 +1 @@
-{"version":3,"file":"toolParameterTransforms.d.ts","sourceRoot":"","sources":["../../src/utils/toolParameterTransforms.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,KAAK,MAAM,EAAE,KAAK,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAmBtE;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,UAAU,EAAE,kBAAkB,GAC7B,MAAM,EAAE,CAOV;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,eAAe,CAAC,EAAE,MAAM,GACvB,MAAM,EAAE,CA0BV"}
+{"version":3,"file":"toolParameterTransforms.d.ts","sourceRoot":"","sources":["../../src/utils/toolParameterTransforms.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,KAAK,MAAM,EACX,KAAK,mBAAmB,EACxB,KAAK,kBAAkB,EACxB,MAAM,gBAAgB,CAAC;AAmBxB;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,UAAU,EAAE,mBAAmB,GAC9B,MAAM,EAAE,CAOV;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,UAAU,EAAE,kBAAkB,GAC7B,MAAM,EAAE,CAOV;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,eAAe,CAAC,EAAE,MAAM,GACvB,MAAM,EAAE,CA8BV"}

package/dist/utils/toolParameterTransforms.js

@@ -18,6 +18,17 @@ function parseToolParameters(parameters) {
         return {};
     }
 }
+/**
+ * Transform Write tool parameters to changes
+ */
+export function transformWriteParameters(parameters) {
+    return [
+        {
+            oldContent: "", // No previous content for write operations
+            newContent: parameters.content,
+        },
+    ];
+}
 /**
  * Transform Edit tool parameters to changes
  */
@@ -41,6 +52,9 @@ export function transformToolBlockToChanges(toolName, parameters, startLineNumbe
         const parsedParams = parseToolParameters(parameters);
         let changes = [];
         switch (toolName) {
+            case "Write":
+                changes = transformWriteParameters(parsedParams);
+                break;
             case "Edit":
                 changes = transformEditParameters(parsedParams);
                 break;

package/dist/utils/worktree.d.ts

@@ -7,7 +7,6 @@ export interface WorktreeSession {
     hasNewCommits: boolean;
     isNew: boolean;
 }
-export declare const WORKTREE_DIR = ".wave/worktrees";
 /**
  * Create a new git worktree
  * @param name Worktree name

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

@@ -1 +1 @@
-{"version":3,"file":"worktree.d.ts","sourceRoot":"","sources":["../../src/utils/worktree.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,qBAAqB,EAAE,OAAO,CAAC;IAC/B,aAAa,EAAE,OAAO,CAAC;IACvB,KAAK,EAAE,OAAO,CAAC;CAChB;AAED,eAAO,MAAM,YAAY,oBAAoB,CAAC;AAE9C;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,eAAe,CAyEzE;AAED;;;GAGG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,eAAe,GAAG,IAAI,CA6D7D"}
+{"version":3,"file":"worktree.d.ts","sourceRoot":"","sources":["../../src/utils/worktree.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,qBAAqB,EAAE,OAAO,CAAC;IAC/B,aAAa,EAAE,OAAO,CAAC;IACvB,KAAK,EAAE,OAAO,CAAC;CAChB;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,eAAe,CA+EzE;AAED;;;GAGG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,eAAe,GAAG,IAAI,CA6D7D"}

package/dist/utils/worktree.js

@@ -2,7 +2,6 @@ import { execSync } from "node:child_process";
 import * as path from "node:path";
 import * as fs from "node:fs";
 import { getDefaultRemoteBranch, getGitMainRepoRoot } from "wave-agent-sdk";
-export const WORKTREE_DIR = ".wave/worktrees";
 /**
  * Create a new git worktree
  * @param name Worktree name
@@ -11,7 +10,8 @@ export const WORKTREE_DIR = ".wave/worktrees";
  */
 export function createWorktree(name, cwd) {
     const repoRoot = getGitMainRepoRoot(cwd);
-    const worktreePath = path.join(repoRoot, WORKTREE_DIR, name);
+    const projectName = path.basename(repoRoot);
+    const worktreePath = path.join(repoRoot, "..", `${projectName}.worktrees`, name);
     const branchName = `worktree-${name}`;
     const baseBranch = getDefaultRemoteBranch(cwd);
     // Ensure parent directory exists

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-code",
-  "version": "0.12.0",
+  "version": "0.12.2",
   "description": "CLI-based code assistant powered by AI, built with React and Ink",
   "repository": {
     "type": "git",
@@ -42,7 +42,7 @@
     "semver": "^7.7.4",
     "yargs": "^17.7.2",
     "zod": "^3.23.8",
-    "wave-agent-sdk": "0.12.0"
+    "wave-agent-sdk": "0.12.2"
   },
   "devDependencies": {
     "@types/react": "^19.1.8",
@@ -65,7 +65,7 @@
   },
   "license": "MIT",
   "scripts": {
-    "wave": "tsx --tsconfig tsconfig.dev.json src/index.ts",
+    "wave": "NODE_OPTIONS='--heapsnapshot-near-heap-limit=1' tsx --tsconfig tsconfig.dev.json src/index.ts",
     "build": "rimraf dist && tsc -p tsconfig.build.json && tsc-alias -p tsconfig.build.json",
     "type-check": "tsc --noEmit --incremental",
     "watch": "tsc -p tsconfig.build.json --watch & tsc-alias -p tsconfig.build.json --watch",

package/src/acp/agent.ts

@@ -670,7 +670,7 @@ export class WaveAcpAgent implements AcpAgent {
         });
       }
 
-      if (contents.length === 0) {
+      if (name.startsWith("mcp__")) {
         contents.push({
           type: "content",
           content: {

package/src/components/App.tsx

@@ -112,7 +112,7 @@ const AppWithProviders: React.FC<AppWithProvidersProps> = ({
   );
 };
 
-const ChatInterfaceWithRemount: React.FC = () => {
+export const ChatInterfaceWithRemount: React.FC = () => {
   const { stdout } = useStdout();
   const { isExpanded, rewindId, wasLastDetailsTooTall, sessionId } = useChat();
 
@@ -121,6 +121,7 @@ const ChatInterfaceWithRemount: React.FC = () => {
   );
 
   const prevSessionId = useRef(sessionId);
+  const isRemountScheduled = useRef(false);
 
   useEffect(() => {
     const newKey =
@@ -131,13 +132,16 @@ const ChatInterfaceWithRemount: React.FC = () => {
         ? sessionId
         : "");
 
-    if (newKey !== remountKey) {
+    if (newKey !== remountKey && !isRemountScheduled.current) {
+      isRemountScheduled.current = true;
+
       const timeout = setTimeout(() => {
         stdout?.write("\u001b[2J\u001b[3J\u001b[0;0H", (err?: Error | null) => {
           if (err) {
             console.error("Failed to clear terminal:", err);
           }
           setRemountKey(newKey);
+          isRemountScheduled.current = false;
         });
       }, 100);
 

package/src/components/BackgroundTaskManager.tsx

@@ -1,6 +1,7 @@
 import React, { useState, useEffect } from "react";
 import { Box, Text, useInput } from "ink";
 import { useChat } from "../contexts/useChat.js";
+import { getLastLines } from "wave-agent-sdk";
 
 interface Task {
   id: string;
@@ -207,9 +208,7 @@ export const BackgroundTaskManager: React.FC<BackgroundTaskManagerProps> = ({
               OUTPUT (last 10 lines):
             </Text>
             <Box borderStyle="single" borderColor="green" padding={1}>
-              <Text>
-                {detailOutput.stdout.split("\n").slice(-10).join("\n")}
-              </Text>
+              <Text>{getLastLines(detailOutput.stdout, 10)}</Text>
             </Box>
           </Box>
         )}
@@ -220,9 +219,7 @@ export const BackgroundTaskManager: React.FC<BackgroundTaskManagerProps> = ({
               ERRORS:
             </Text>
             <Box borderStyle="single" borderColor="red" padding={1}>
-              <Text color="red">
-                {detailOutput.stderr.split("\n").slice(-10).join("\n")}
-              </Text>
+              <Text color="red">{getLastLines(detailOutput.stderr, 10)}</Text>
             </Box>
           </Box>
         )}

package/src/components/BangDisplay.tsx

@@ -1,6 +1,7 @@
-import React, { useState, useEffect } from "react";
+import React from "react";
 import { Box, Text } from "ink";
 import type { BangBlock } from "wave-agent-sdk";
+import { getLastLines } from "wave-agent-sdk";
 
 interface BangDisplayProps {
   block: BangBlock;
@@ -12,17 +13,8 @@ export const BangDisplay: React.FC<BangDisplayProps> = ({
   isExpanded = false,
 }) => {
   const { command, output, isRunning, exitCode } = block;
-  const [isOverflowing, setIsOverflowing] = useState(false);
   const MAX_LINES = 3; // Set maximum display lines
 
-  // Detect if content is overflowing
-  useEffect(() => {
-    if (output) {
-      const lines = output.split("\n");
-      setIsOverflowing(!isExpanded && lines.length > MAX_LINES);
-    }
-  }, [output, isExpanded]);
-
   const getStatusColor = () => {
     if (isRunning) return "yellow";
     if (exitCode === 0) return "green";
@@ -38,19 +30,9 @@ export const BangDisplay: React.FC<BangDisplayProps> = ({
       </Box>
 
       {output && (
-        <Box
-          paddingLeft={2}
-          height={
-            isExpanded
-              ? undefined
-              : Math.min(output.split("\n").length, MAX_LINES)
-          }
-          overflow="hidden"
-        >
+        <Box paddingLeft={2} overflow="hidden">
           <Text color="gray">
-            {isOverflowing
-              ? output.split("\n").slice(-MAX_LINES).join("\n")
-              : output}
+            {isExpanded ? output : getLastLines(output, MAX_LINES)}
           </Text>
         </Box>
       )}