wave-code 0.12.0 → 0.12.1

package/README.md

@@ -31,38 +31,26 @@ export WAVE_API_KEY="your_token_here"
 
 # AI Gateway API URL (required)
 export WAVE_BASE_URL="https://your-api-gateway-url.com"
+
+# AI model (required)
+export WAVE_MODEL="MiniMax-M2.7"
+
+# Fast AI model (required)
+export WAVE_FAST_MODEL="MiniMax-M2.7-highspeed"
 ```
 
 ### Optional Environment Variables
 
 ```bash
-# Specify AI model (optional, defaults to gemini-3-flash)
-export WAVE_MODEL="gemini-3-flash"
-
-# Specify fast AI model (optional, defaults to gemini-2.5-flash)
-export WAVE_FAST_MODEL="gemini-2.5-flash"
-
 # Token limit (optional, defaults to 96000)
 export WAVE_MAX_INPUT_TOKENS="96000"
-
 ```
 
 ## Usage
 
-### Full Command
-
 ```bash
 # Start the CLI
-wave-code
-
-# Continue from last session
-wave-code --continue
-
-# Restore specific session
-wave-code --restore session_id
-
-# List available sessions
-wave-code --list-sessions
+wave-code -h
 ```
 
 ### Short Command (wave)
@@ -70,42 +58,12 @@ wave-code --list-sessions
 For convenience, you can also use the shorter `wave` command:
 
 ```bash
-# Start the CLI (equivalent to wave-code)
-wave
-
-# Continue from last session
-wave --continue
-
-# Restore specific session
-wave --restore session_id
-
-# List available sessions
-wave --list-sessions
-
-# Show help
-wave --help
+# Start the CLI
+wave -h
 ```
 
 The `wave` command is an alias for `wave-code` and supports all the same options and functionality.
 
-## Development
-
-This package depends on `wave-agent-sdk` for core functionality including AI services, tools, and utilities.
-
-```bash
-# Install dependencies
-pnpm install
-
-# Start development
-pnpm dev
-
-# Build
-pnpm build
-
-# Test
-pnpm test
-```
-
 ## License
 
 MIT

package/dist/acp/agent.js

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

package/dist/components/DiffDisplay.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DiffDisplay.d.ts","sourceRoot":"","sources":["../../src/components/DiffDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkB,MAAM,OAAO,CAAC;AAYvC,UAAU,gBAAgB;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CA6WlD,CAAC"}
+{"version":3,"file":"DiffDisplay.d.ts","sourceRoot":"","sources":["../../src/components/DiffDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkB,MAAM,OAAO,CAAC;AAMvC,UAAU,gBAAgB;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CA0VlD,CAAC"}

package/dist/components/DiffDisplay.js

@@ -1,11 +1,9 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import { useMemo } from "react";
 import { Box, Text } from "ink";
-import { WRITE_TOOL_NAME, EDIT_TOOL_NAME, } from "wave-agent-sdk";
+import { WRITE_TOOL_NAME, EDIT_TOOL_NAME } from "wave-agent-sdk";
 import { transformToolBlockToChanges } from "../utils/toolParameterTransforms.js";
 import { diffLines, diffWords } from "diff";
-import path from "path";
-import { Markdown } from "./Markdown.js";
 export const DiffDisplay = ({ toolName, parameters, startLineNumber, }) => {
     const showDiff = toolName && [WRITE_TOOL_NAME, EDIT_TOOL_NAME].includes(toolName);
     // Diff detection and transformation using typed parameters
@@ -55,23 +53,6 @@ export const DiffDisplay = ({ toolName, parameters, startLineNumber, }) => {
             };
         }
     };
-    // Render highlighted code for Write tool
-    const renderWriteContent = () => {
-        if (!parameters)
-            return null;
-        try {
-            const parsed = JSON.parse(parameters);
-            const content = parsed.content || "";
-            const filePath = parsed.file_path || "";
-            const ext = path.extname(filePath).slice(1);
-            const markdown = `\`\`\`${ext}\n${content}\n\`\`\``;
-            return _jsx(Markdown, { children: markdown });
-        }
-        catch (error) {
-            console.warn("Error rendering write content:", error);
-            return null;
-        }
-    };
     // Render expanded diff display
     const renderExpandedDiff = () => {
         try {
@@ -216,7 +197,5 @@ export const DiffDisplay = ({ toolName, parameters, startLineNumber, }) => {
     if (!showDiff) {
         return null;
     }
-    return (_jsx(Box, { flexDirection: "column", children: _jsx(Box, { paddingLeft: 2, borderLeft: true, borderColor: "cyan", flexDirection: "column", children: toolName === WRITE_TOOL_NAME
-                ? renderWriteContent()
-                : renderExpandedDiff() }) }));
+    return (_jsx(Box, { flexDirection: "column", children: _jsx(Box, { paddingLeft: 2, borderLeft: true, borderColor: "cyan", flexDirection: "column", children: renderExpandedDiff() }) }));
 };

package/dist/components/MessageList.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageList.d.ts","sourceRoot":"","sources":["../../src/components/MessageList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkC,MAAM,OAAO,CAAC;AAGvD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAG9C,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,6BAA6B,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;CAC1D;AAED,eAAO,MAAM,WAAW,sHAQnB,gBAAgB,6CAkHpB,CAAC"}
+{"version":3,"file":"MessageList.d.ts","sourceRoot":"","sources":["../../src/components/MessageList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkC,MAAM,OAAO,CAAC;AAGvD,OAAO,KAAK,EAAE,OAAO,EAAgB,MAAM,gBAAgB,CAAC;AAG5D,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,6BAA6B,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;CAC1D;AAED,eAAO,MAAM,WAAW,sHAQnB,gBAAgB,6CA4HpB,CAAC"}

package/dist/components/MessageList.js

@@ -9,6 +9,13 @@ export const MessageList = React.memo(({ messages, isExpanded = false, forceStat
     const maxMessages = 10;
     // Filter out meta messages
     const visibleMessages = messages.filter((m) => !m.isMeta);
+    const isRunning = (b) => (b.type === "tool" && b.stage === "running") ||
+        (b.type === "bang" && b.isRunning) ||
+        (b.type === "slash" && b.stage === "running");
+    const messagesWithRunningBlocks = new Set(visibleMessages
+        .map((m, i) => ({ m, i }))
+        .filter(({ m }) => m.blocks.some(isRunning))
+        .map(({ i }) => i));
     // Flatten messages into blocks with metadata
     const allBlocks = visibleMessages.flatMap((message, messageIndex) => {
         return message.blocks.map((block, blockIndex) => ({
@@ -23,9 +30,7 @@ export const MessageList = React.memo(({ messages, isExpanded = false, forceStat
     const blocksWithStatus = allBlocks.map((item) => {
         const { block } = item;
         const isDynamic = !forceStatic &&
-            ((block.type === "tool" && block.stage === "running") ||
-                (block.type === "bang" && block.isRunning) ||
-                (block.type === "slash" && block.stage === "running"));
+            (messagesWithRunningBlocks.has(item.messageIndex) || isRunning(block));
         return { ...item, isDynamic };
     });
     const staticBlocks = blocksWithStatus.filter((b) => !b.isDynamic);

package/dist/components/StatusCommand.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"StatusCommand.d.ts","sourceRoot":"","sources":["../../src/components/StatusCommand.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAU1B,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAAC,kBAAkB,CA+EtD,CAAC"}
+{"version":3,"file":"StatusCommand.d.ts","sourceRoot":"","sources":["../../src/components/StatusCommand.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAU1B,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAAC,kBAAkB,CAsFtD,CAAC"}

package/dist/components/StatusCommand.js

@@ -24,5 +24,5 @@ export const StatusCommand = ({ onCancel }) => {
     catch {
         // Fallback if package.json cannot be read
     }
-    return (_jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "cyan", borderLeft: false, borderRight: false, paddingX: 1, children: [_jsx(Box, { marginBottom: 1, children: _jsx(Text, { color: "cyan", bold: true, underline: true, children: "Agent Status" }) }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Version:" }) }), _jsx(Text, { color: "white", children: version })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Session ID:" }) }), _jsx(Text, { color: "white", children: sessionId })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "cwd:" }) }), _jsx(Text, { color: "white", wrap: "wrap", children: workingDirectory })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Wave base URL:" }) }), _jsx(Text, { color: "white", children: gatewayConfig.baseURL })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Model:" }) }), _jsx(Text, { color: "white", children: modelConfig.model })] }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { dimColor: true, children: "Esc to cancel" }) })] }));
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "cyan", borderLeft: false, borderRight: false, paddingX: 1, children: [_jsx(Box, { marginBottom: 1, children: _jsx(Text, { color: "cyan", bold: true, underline: true, children: "Agent Status" }) }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Version:" }) }), _jsx(Text, { color: "white", children: version })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Session ID:" }) }), _jsx(Text, { color: "white", children: sessionId })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "cwd:" }) }), _jsx(Text, { color: "white", wrap: "wrap", children: workingDirectory })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Wave base URL:" }) }), _jsx(Text, { color: "white", children: gatewayConfig.baseURL })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Model:" }) }), _jsx(Text, { color: "white", children: modelConfig.model })] }), _jsxs(Box, { children: [_jsx(Box, { width: 20, children: _jsx(Text, { color: "yellow", children: "Fast model:" }) }), _jsx(Text, { color: "white", children: modelConfig.fastModel })] }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { dimColor: true, children: "Esc to cancel" }) })] }));
 };

package/dist/components/ToolDisplay.js

@@ -40,5 +40,5 @@ export const ToolDisplay = ({ block, isExpanded = false, }) => {
         return null;
     };
     const shortResult = getShortResult();
-    return (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsxs(Box, { flexShrink: 0, children: [_jsx(Text, { color: getStatusColor(), children: "\u25CF " }), _jsx(Text, { color: "white", children: toolName })] }), !isExpanded && compactParams && (_jsxs(Text, { color: "gray", children: [" ", compactParams] })), hasImages() && _jsxs(Text, { color: "blue", children: [" ", getImageIndicator()] })] }), !isExpanded && shortResult && !error && (_jsx(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: shortResult.split("\n").map((line, index) => (_jsx(Text, { color: "gray", children: line }, index))) })), isExpanded && parameters && (_jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Parameters:" }), _jsx(Text, { color: "gray", children: parameters })] })), isExpanded && result && (_jsx(Box, { flexDirection: "column", children: _jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "green", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Result:" }), _jsx(Text, { color: "white", children: result })] }) })), error && (_jsx(Box, { children: _jsxs(Text, { color: "red", children: ["Error: ", typeof error === "string" ? error : String(error)] }) })), !isExpanded && stage === "end" && success && (_jsx(DiffDisplay, { toolName: name, parameters: parameters, startLineNumber: block.startLineNumber }))] }));
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsxs(Box, { flexShrink: 0, children: [_jsx(Text, { color: getStatusColor(), children: "\u25CF " }), _jsx(Text, { color: "white", children: toolName })] }), !isExpanded && compactParams && (_jsxs(Text, { color: "gray", children: [" ", compactParams] })), hasImages() && _jsxs(Text, { color: "blue", children: [" ", getImageIndicator()] })] }), !isExpanded && shortResult && !error && (_jsx(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: shortResult.split("\n").map((line, index) => (_jsx(Text, { color: "gray", wrap: "truncate-end", children: line }, index))) })), isExpanded && parameters && (_jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Parameters:" }), _jsx(Text, { color: "gray", children: parameters })] })), isExpanded && result && (_jsx(Box, { flexDirection: "column", children: _jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "green", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Result:" }), _jsx(Text, { color: "white", children: result })] }) })), error && (_jsx(Box, { children: _jsxs(Text, { color: "red", children: ["Error: ", typeof error === "string" ? error : String(error)] }) })), !isExpanded && stage === "end" && success && (_jsx(DiffDisplay, { toolName: name, parameters: parameters, startLineNumber: block.startLineNumber }))] }));
 };

package/dist/contexts/useChat.d.ts

@@ -43,8 +43,6 @@ export interface ChatContextType {
     stopBackgroundTask: (taskId: string) => boolean;
     slashCommands: SlashCommand[];
     hasSlashCommand: (commandId: string) => boolean;
-    subagentMessages: Record<string, Message[]>;
-    subagentLatestTokens: Record<string, number>;
     permissionMode: PermissionMode;
     setPermissionMode: (mode: PermissionMode) => void;
     allowBypassInCycle: boolean;

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,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,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,CAwmBpD,CAAC"}

package/dist/contexts/useChat.js

@@ -41,9 +41,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 +72,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) => {
@@ -126,20 +113,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 +195,11 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
         workdir,
         worktreeSession,
         model,
-        debouncedSetTasks,
         initialPermissionMode,
     ]);
     // Cleanup on unmount
     useEffect(() => {
         return () => {
-            if (taskUpdateTimerRef.current) {
-                clearTimeout(taskUpdateTimerRef.current);
-            }
             if (agentRef.current) {
                 try {
                     // Display usage summary before cleanup
@@ -499,8 +469,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;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;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;AAqDD;;;;;GAKG;AACH,eAAO,MAAM,WAAW,GACtB,eAAe,OAAO,CAAC,gBAAgB,CAAC,KACvC,OAAO,CAAC,IAAI,CAiBd,CAAC"}

package/dist/utils/logger.js

@@ -106,20 +106,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

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.1",
   "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.1"
   },
   "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/DiffDisplay.tsx

@@ -1,14 +1,8 @@
 import React, { useMemo } from "react";
 import { Box, Text } from "ink";
-import {
-  WRITE_TOOL_NAME,
-  EDIT_TOOL_NAME,
-  type WriteToolParameters,
-} from "wave-agent-sdk";
+import { WRITE_TOOL_NAME, EDIT_TOOL_NAME } from "wave-agent-sdk";
 import { transformToolBlockToChanges } from "../utils/toolParameterTransforms.js";
 import { diffLines, diffWords } from "diff";
-import path from "path";
-import { Markdown } from "./Markdown.js";
 
 interface DiffDisplayProps {
   toolName?: string;
@@ -103,23 +97,6 @@ export const DiffDisplay: React.FC<DiffDisplayProps> = ({
     }
   };
 
-  // Render highlighted code for Write tool
-  const renderWriteContent = () => {
-    if (!parameters) return null;
-    try {
-      const parsed = JSON.parse(parameters) as WriteToolParameters;
-      const content = parsed.content || "";
-      const filePath = parsed.file_path || "";
-      const ext = path.extname(filePath).slice(1);
-
-      const markdown = `\`\`\`${ext}\n${content}\n\`\`\``;
-      return <Markdown>{markdown}</Markdown>;
-    } catch (error) {
-      console.warn("Error rendering write content:", error);
-      return null;
-    }
-  };
-
   // Render expanded diff display
   const renderExpandedDiff = () => {
     try {
@@ -375,9 +352,7 @@ export const DiffDisplay: React.FC<DiffDisplayProps> = ({
   return (
     <Box flexDirection="column">
       <Box paddingLeft={2} borderLeft borderColor="cyan" flexDirection="column">
-        {toolName === WRITE_TOOL_NAME
-          ? renderWriteContent()
-          : renderExpandedDiff()}
+        {renderExpandedDiff()}
       </Box>
     </Box>
   );

package/src/components/MessageList.tsx

@@ -1,7 +1,7 @@
 import React, { useLayoutEffect, useRef } from "react";
 import os from "os";
 import { Box, Text, Static, measureElement } from "ink";
-import type { Message } from "wave-agent-sdk";
+import type { Message, MessageBlock } from "wave-agent-sdk";
 import { MessageBlockItem } from "./MessageBlockItem.js";
 
 export interface MessageListProps {
@@ -39,6 +39,18 @@ export const MessageList = React.memo(
     // Filter out meta messages
     const visibleMessages = messages.filter((m) => !m.isMeta);
 
+    const isRunning = (b: MessageBlock) =>
+      (b.type === "tool" && b.stage === "running") ||
+      (b.type === "bang" && b.isRunning) ||
+      (b.type === "slash" && b.stage === "running");
+
+    const messagesWithRunningBlocks = new Set(
+      visibleMessages
+        .map((m, i) => ({ m, i }))
+        .filter(({ m }) => m.blocks.some(isRunning))
+        .map(({ i }) => i),
+    );
+
     // Flatten messages into blocks with metadata
     const allBlocks = visibleMessages.flatMap((message, messageIndex) => {
       return message.blocks.map((block, blockIndex) => ({
@@ -55,9 +67,7 @@ export const MessageList = React.memo(
       const { block } = item;
       const isDynamic =
         !forceStatic &&
-        ((block.type === "tool" && block.stage === "running") ||
-          (block.type === "bang" && block.isRunning) ||
-          (block.type === "slash" && block.stage === "running"));
+        (messagesWithRunningBlocks.has(item.messageIndex) || isRunning(block));
       return { ...item, isDynamic };
     });
 

package/src/components/StatusCommand.tsx

@@ -86,6 +86,13 @@ export const StatusCommand: React.FC<StatusCommandProps> = ({ onCancel }) => {
         <Text color="white">{modelConfig.model}</Text>
       </Box>
 
+      <Box>
+        <Box width={20}>
+          <Text color="yellow">Fast model:</Text>
+        </Box>
+        <Text color="white">{modelConfig.fastModel}</Text>
+      </Box>
+
       <Box marginTop={1}>
         <Text dimColor>Esc to cancel</Text>
       </Box>

package/src/components/ToolDisplay.tsx

@@ -81,7 +81,7 @@ export const ToolDisplay: React.FC<ToolDisplayProps> = ({
           flexDirection="column"
         >
           {shortResult.split("\n").map((line, index) => (
-            <Text key={index} color="gray">
+            <Text key={index} color="gray" wrap="truncate-end">
               {line}
             </Text>
           ))}

package/src/contexts/useChat.tsx

@@ -77,9 +77,6 @@ export interface ChatContextType {
   // Slash Command functionality
   slashCommands: SlashCommand[];
   hasSlashCommand: (commandId: string) => boolean;
-  // Subagent messages
-  subagentMessages: Record<string, Message[]>;
-  subagentLatestTokens: Record<string, number>;
   // Permission functionality
   permissionMode: PermissionMode;
   setPermissionMode: (mode: PermissionMode) => void;
@@ -195,14 +192,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   // Command state
   const [slashCommands, setSlashCommands] = useState<SlashCommand[]>([]);
 
-  // Subagent messages state
-  const [subagentMessages, setSubagentMessages] = useState<
-    Record<string, Message[]>
-  >({});
-  const [subagentLatestTokens, setSubagentLatestTokens] = useState<
-    Record<string, number>
-  >({});
-
   // Permission state
   const [permissionMode, setPermissionModeState] = useState<PermissionMode>(
     initialPermissionMode ||
@@ -276,20 +265,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     !!bypassPermissions || initialPermissionMode === "bypassPermissions";
 
   const agentRef = useRef<Agent | null>(null);
-  const taskUpdateTimerRef = useRef<NodeJS.Timeout | null>(null);
-
-  const debouncedSetTasks = useCallback(
-    (newTasks: Task[]) => {
-      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(
@@ -343,23 +318,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           setBackgroundTasks([...tasks]);
         },
         onTasksChange: (tasks) => {
-          debouncedSetTasks(tasks);
-        },
-        onSubagentMessagesChange: (subagentId: string, messages: Message[]) => {
-          logger.debug("onSubagentMessagesChange", subagentId, messages.length);
-          setSubagentMessages((prev) => ({
-            ...prev,
-            [subagentId]: [...messages],
-          }));
-        },
-        onSubagentLatestTotalTokensChange: (
-          subagentId: string,
-          tokens: number,
-        ) => {
-          setSubagentLatestTokens((prev) => ({
-            ...prev,
-            [subagentId]: tokens,
-          }));
+          setTasks([...tasks]);
         },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
@@ -455,17 +414,12 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     workdir,
     worktreeSession,
     model,
-    debouncedSetTasks,
     initialPermissionMode,
   ]);
 
   // Cleanup on unmount
   useEffect(() => {
     return () => {
-      if (taskUpdateTimerRef.current) {
-        clearTimeout(taskUpdateTimerRef.current);
-      }
-
       if (agentRef.current) {
         try {
           // Display usage summary before cleanup
@@ -772,8 +726,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     stopBackgroundTask,
     slashCommands,
     hasSlashCommand,
-    subagentMessages,
-    subagentLatestTokens,
     permissionMode,
     setPermissionMode,
     allowBypassInCycle,

package/src/index.ts

@@ -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 { Scope, generateRandomName, type PermissionMode } from "wave-agent-sdk";
 import { createWorktree, type WorktreeSession } from "./utils/worktree.js";
 import path from "path";
@@ -14,6 +15,18 @@ 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/src/utils/highlightUtils.ts

@@ -49,14 +49,6 @@ function nodeToAnsi(node: Node): string {
 
     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/src/utils/logger.ts

@@ -124,21 +124,23 @@ const formatArg = (arg: unknown): string => {
   if (arg === null) return "null";
   if (arg === undefined) return "undefined";
 
+  let result: string;
   if (arg instanceof Error) {
     // Special handling for Error objects, display stack or message
-    return arg.stack || arg.message || String(arg);
-  }
-
-  if (typeof arg === "object") {
+    result = arg.stack || arg.message || String(arg);
+  } 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);
     }
+  } else {
+    result = String(arg);
   }
 
-  return String(arg);
+  // Ensure no newlines in the result to keep log entries single-line
+  return result.replace(/[\n\r]+/g, " ");
 };
 
 /**

package/src/utils/toolParameterTransforms.ts

@@ -3,7 +3,11 @@
  * 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";
 import { logger } from "./logger.js";
 
 /**
@@ -22,6 +26,20 @@ function parseToolParameters(parameters: string): unknown {
   }
 }
 
+/**
+ * Transform Write tool parameters to changes
+ */
+export function transformWriteParameters(
+  parameters: WriteToolParameters,
+): Change[] {
+  return [
+    {
+      oldContent: "", // No previous content for write operations
+      newContent: parameters.content,
+    },
+  ];
+}
+
 /**
  * Transform Edit tool parameters to changes
  */
@@ -54,6 +72,10 @@ export function transformToolBlockToChanges(
 
     let changes: Change[] = [];
     switch (toolName) {
+      case "Write":
+        changes = transformWriteParameters(parsedParams as WriteToolParameters);
+        break;
+
       case "Edit":
         changes = transformEditParameters(parsedParams as EditToolParameters);
         break;

package/src/utils/worktree.ts

@@ -13,8 +13,6 @@ export interface WorktreeSession {
   isNew: boolean;
 }
 
-export const WORKTREE_DIR = ".wave/worktrees";
-
 /**
  * Create a new git worktree
  * @param name Worktree name
@@ -23,7 +21,13 @@ export const WORKTREE_DIR = ".wave/worktrees";
  */
 export function createWorktree(name: string, cwd: string): WorktreeSession {
   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);