wave-code 1.0.8 → 1.0.9

package/bin/wave-code.js

@@ -1,5 +1,26 @@
 #!/usr/bin/env node
 
+import { readFileSync, writeSync } from "node:fs";
+import path from "node:path";
+import { fileURLToPath } from "node:url";
+
+// `wave -v` / `wave --version` must be fast: editors probe the installed CLI
+// version on every launch (e.g. the desktop app's auto-update check). Loading
+// the full app graph (wave-agent-sdk, ink, highlight.js, ...) just to print
+// the version takes 2-3s+ on a warm machine and can exceed callers' probe
+// timeouts on cold starts (AV scan of freshly installed files), which they
+// misread as "CLI missing/corrupt" → spurious re-installs. Print the version
+// straight from package.json and exit before touching the app graph.
+const versionArgs = ["-v", "--version"];
+if (process.argv.slice(2).some((a) => versionArgs.includes(a))) {
+  const __dirname = path.dirname(fileURLToPath(import.meta.url));
+  const packageJson = JSON.parse(
+    readFileSync(path.resolve(__dirname, "../package.json"), "utf-8"),
+  );
+  writeSync(1, `${packageJson.version}\n`);
+  process.exit(0);
+}
+
 // Import and start the CLI
 import("../dist/index.js")
   .then(async ({ main }) => {

package/dist/contexts/useChat.js

@@ -14,6 +14,21 @@ export const useChat = () => {
     }
     return context;
 };
+/**
+ * Snapshot a SDK message for consumer state. The SDK mutates its internal
+ * message blocks in-place BEFORE firing the delta callback (it writes the full
+ * accumulated value to the shared block, then computes the chunk delta by
+ * slicing the new value). A consumer that pushed the SDK message object by
+ * live reference would read the already-updated block and append the delta
+ * again — the first delta is double-counted ("LetLet me think..."), affecting
+ * reasoning and text content alike. See docs/specs/core/stream-content-updates.md.
+ * The clone must be at least one layer deep (message + blocks) so the in-place
+ * block mutation never leaks into consumer state.
+ */
+const snapshotMessage = (message) => ({
+    ...message,
+    blocks: message.blocks.map((block) => ({ ...block })),
+});
 /**
  * Window-concat throttle for pure-delta streaming updates: chunks arriving
  * within the cooldown window are merged so no delta is lost (a dropped delta
@@ -322,7 +337,7 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
     // the incremental callbacks in initializeAgent below.
     const refreshMessages = useCallback(() => {
         if (!isExpandedRef.current && agentRef.current) {
-            const msgs = [...agentRef.current.messages];
+            const msgs = agentRef.current.messages.map(snapshotMessage);
             setMessages(msgs);
             setLatestTotalTokens(extractLatestTotalTokens(msgs));
         }
@@ -357,7 +372,9 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
                 const last = msgs[msgs.length - 1];
                 if (!last || last.role !== "user")
                     return;
-                setMessages((prev) => prev.some((m) => m.id === last.id) ? prev : [...prev, last]);
+                setMessages((prev) => prev.some((m) => m.id === last.id)
+                    ? prev
+                    : [...prev, snapshotMessage(last)]);
             },
             onAssistantMessageAdded: (messageId) => {
                 if (isExpandedRef.current || !agentRef.current)
@@ -365,7 +382,9 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
                 const msg = agentRef.current.messages.find((m) => m.id === messageId);
                 if (!msg)
                     return;
-                setMessages((prev) => prev.some((m) => m.id === messageId) ? prev : [...prev, msg]);
+                setMessages((prev) => prev.some((m) => m.id === messageId)
+                    ? prev
+                    : [...prev, snapshotMessage(msg)]);
             },
             onAssistantContentUpdated: (params) => {
                 if (isExpandedRef.current)
@@ -562,9 +581,10 @@ export const ChatProvider = ({ children, bypassPermissions, permissionMode: init
                 };
                 agent.setWorktreeSession(session);
             }
-            // Get initial state
+            // Get initial state — snapshot the SDK messages (never hold live
+            // references; see snapshotMessage)
             setSessionId(agent.sessionId);
-            setMessages(agent.messages);
+            setMessages(agent.messages.map(snapshotMessage));
             setIsLoading(agent.isLoading);
             setLatestTotalTokens(extractLatestTotalTokens(agent.messages));
             setIsCommandRunning(agent.isCommandRunning);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-code",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "CLI-based code assistant powered by AI, built with React and Ink",
   "repository": {
     "type": "git",
@@ -43,7 +43,7 @@
     "wrap-ansi": "^10.0.0",
     "yargs": "^17.7.2",
     "zod": "^3.23.8",
-    "wave-agent-sdk": "1.0.8"
+    "wave-agent-sdk": "1.0.9"
   },
   "devDependencies": {
     "@types/react": "^19.1.8",

package/src/contexts/useChat.tsx

@@ -168,6 +168,22 @@ interface StreamingUpdateParams {
   stage: "streaming" | "end";
 }
 
+/**
+ * Snapshot a SDK message for consumer state. The SDK mutates its internal
+ * message blocks in-place BEFORE firing the delta callback (it writes the full
+ * accumulated value to the shared block, then computes the chunk delta by
+ * slicing the new value). A consumer that pushed the SDK message object by
+ * live reference would read the already-updated block and append the delta
+ * again — the first delta is double-counted ("LetLet me think..."), affecting
+ * reasoning and text content alike. See docs/specs/core/stream-content-updates.md.
+ * The clone must be at least one layer deep (message + blocks) so the in-place
+ * block mutation never leaks into consumer state.
+ */
+const snapshotMessage = (message: Message): Message => ({
+  ...message,
+  blocks: message.blocks.map((block) => ({ ...block })),
+});
+
 /**
  * Window-concat throttle for pure-delta streaming updates: chunks arriving
  * within the cooldown window are merged so no delta is lost (a dropped delta
@@ -605,7 +621,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   // the incremental callbacks in initializeAgent below.
   const refreshMessages = useCallback(() => {
     if (!isExpandedRef.current && agentRef.current) {
-      const msgs = [...agentRef.current.messages];
+      const msgs = agentRef.current.messages.map(snapshotMessage);
       setMessages(msgs);
       setLatestTotalTokens(extractLatestTotalTokens(msgs));
     }
@@ -656,7 +672,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           const last = msgs[msgs.length - 1];
           if (!last || last.role !== "user") return;
           setMessages((prev) =>
-            prev.some((m) => m.id === last.id) ? prev : [...prev, last],
+            prev.some((m) => m.id === last.id)
+              ? prev
+              : [...prev, snapshotMessage(last)],
           );
         },
         onAssistantMessageAdded: (messageId: string) => {
@@ -664,7 +682,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           const msg = agentRef.current.messages.find((m) => m.id === messageId);
           if (!msg) return;
           setMessages((prev) =>
-            prev.some((m) => m.id === messageId) ? prev : [...prev, msg],
+            prev.some((m) => m.id === messageId)
+              ? prev
+              : [...prev, snapshotMessage(msg)],
           );
         },
         onAssistantContentUpdated: (params) => {
@@ -889,9 +909,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           agent.setWorktreeSession(session);
         }
 
-        // Get initial state
+        // Get initial state — snapshot the SDK messages (never hold live
+        // references; see snapshotMessage)
         setSessionId(agent.sessionId);
-        setMessages(agent.messages);
+        setMessages(agent.messages.map(snapshotMessage));
         setIsLoading(agent.isLoading);
         setLatestTotalTokens(extractLatestTotalTokens(agent.messages));
         setIsCommandRunning(agent.isCommandRunning);