wave-code 1.0.7 → 1.0.9

package/dist/stdio/agentBridge.js

@@ -59,6 +59,8 @@ export class AgentBridge {
                 return this.getSessionInfo(sessionId);
             case "listPendingPermissions":
                 return this.listPendingPermissions();
+            case "listDaemonSessions":
+                return this.listDaemonSessions();
             case "updateConfig":
                 return this.updateConfig(p, sessionId);
             case "getConfiguredModels":
@@ -474,9 +476,39 @@ export class AgentBridge {
         catch {
             // Best-effort; don't block message sending on history save failure
         }
-        await entry.agent.sendMessage(params.text, params.images);
+        await entry.agent.sendMessage(params.text, this.persistDataUrlImages(params.images));
         return null;
     }
+    /**
+     * Webview hosts (desktop/vscode/jetbrains) send pasted images as inline
+     * data URLs — there is no local file behind them. Persist each to a temp
+     * file so the model gets a real path it can reference with tools (aligned
+     * with Claude Code's `[Image source: <path>]` metadata). Real paths pass
+     * through untouched; unparseable data URLs pass through as-is and are
+     * skipped by the SDK rather than blocking the message.
+     */
+    persistDataUrlImages(images) {
+        if (!images || images.length === 0)
+            return images;
+        return images.map((img) => {
+            if (!img.path.startsWith("data:"))
+                return img;
+            const match = /^data:([^;,]+);base64,(.*)$/s.exec(img.path);
+            if (!match)
+                return img;
+            try {
+                const mimeType = match[1];
+                const ext = mimeType.split("/")[1]?.replace("jpeg", "jpg") || "png";
+                const filePath = join(tmpdir(), `wave-image-${Date.now()}-${Math.random().toString(36).slice(2, 8)}.${ext}`);
+                writeFileSync(filePath, Buffer.from(match[2], "base64"));
+                return { path: filePath, mimeType };
+            }
+            catch (error) {
+                logger.warn("Failed to persist pasted image to temp file:", error);
+                return img;
+            }
+        });
+    }
     async bang(command, sessionId) {
         const entry = this.requireSession(sessionId);
         await entry.agent.bang(command);
@@ -511,7 +543,10 @@ export class AgentBridge {
     async rewindToMessage(messageId, sessionId) {
         const entry = this.requireSession(sessionId);
         const { messages } = await entry.agent.getFullMessageThread();
-        const index = messages.findIndex((m) => m.id === messageId);
+        // 压缩是 append-only：同 id 消息（压缩前历史 + 压缩后 append 的重复）会
+        // 在磁盘完整线程中出现多次。用户看到的折叠视图对应最后一次出现，
+        // 因此匹配最后一个而非第一个，避免回滚时连压缩摘要一起删掉。
+        const index = messages.map((m) => m.id).lastIndexOf(messageId);
         if (index === -1) {
             throw new RpcError(PROTOCOL_INTERNAL_ERROR, `Message not found: ${messageId}`);
         }
@@ -523,13 +558,19 @@ export class AgentBridge {
     async listRewindCheckpoints(sessionId) {
         const entry = this.requireSession(sessionId);
         const { messages } = await entry.agent.getFullMessageThread();
-        const checkpoints = messages
-            .filter((m) => isUserCheckpointMessage(m) && m.id)
-            .map((m) => ({
-            id: m.id,
-            content: getMessageContent(m).replace(/\s+/g, " ").trim(),
-        }));
-        return { checkpoints };
+        // 压缩 append-only 后同 id 消息在磁盘完整线程中重复出现（压缩前历史 +
+        // 压缩后 append 的重复）。按 id 去重并保留最后一次出现（与折叠后的
+        // UI/内存视图一致），避免弹窗把同一条用户消息显示两遍。
+        const checkpointMap = new Map();
+        for (const m of messages) {
+            if (isUserCheckpointMessage(m) && m.id) {
+                checkpointMap.set(m.id, {
+                    id: m.id,
+                    content: getMessageContent(m).replace(/\s+/g, " ").trim(),
+                });
+            }
+        }
+        return { checkpoints: Array.from(checkpointMap.values()) };
     }
     deleteQueuedMessage(index, sessionId) {
         const entry = this.requireSession(sessionId);
@@ -540,7 +581,7 @@ export class AgentBridge {
         const entry = this.requireSession(sessionId);
         const ok = entry.agent.updateQueuedMessageById(id, {
             content: text,
-            images,
+            images: this.persistDataUrlImages(images),
         });
         return { ok };
     }
@@ -698,6 +739,18 @@ export class AgentBridge {
             })),
         };
     }
+    /** Daemon list: expose the in-memory session registry (live sessions only,
+     * not disk-scanning). Registration order is preserved. */
+    listDaemonSessions() {
+        return {
+            sessions: [...this.sessions.entries()].map(([sessionId, entry]) => ({
+                sessionId,
+                workingDirectory: entry.agent.workingDirectory,
+                isLoading: entry.agent.isLoading,
+                messageCount: entry.agent.messages.length,
+            })),
+        };
+    }
     // ── Auth (global) ────────────────────────────────────────────
     async getAuthStatus() {
         const authService = AuthService.getInstance();

package/dist/stdio/protocol.d.ts

@@ -34,7 +34,7 @@ export declare const INVALID_REQUEST = -32600;
 export declare const METHOD_NOT_FOUND = -32601;
 export declare const INVALID_PARAMS = -32602;
 export declare const INTERNAL_ERROR = -32603;
-export type RequestMethod = "initialize" | "destroy" | "restoreSession" | "listSessions" | "getSessionInfo" | "sendMessage" | "bang" | "askBtw" | "abortMessage" | "clearMessages" | "rewindToMessage" | "listRewindCheckpoints" | "deleteQueuedMessage" | "updateQueuedMessage" | "deleteQueuedMessageById" | "getMessages" | "getFullMessageThread" | "setPermissionMode" | "getPermissionMode" | "getMcpServers" | "connectMcpServer" | "disconnectMcpServer" | "getSlashCommands" | "searchFiles" | "writeArtifactFile" | "getPromptHistory" | "searchPromptHistory" | "updateConfig" | "getConfiguredModels" | "setModel" | "listPendingPermissions" | "getAuthStatus" | "login" | "logout" | "listPlugins" | "installPlugin" | "uninstallPlugin" | "enablePlugin" | "disablePlugin" | "updatePlugin" | "listMarketplaces" | "addMarketplace" | "removeMarketplace" | "updateMarketplace" | "compact" | "getBackgroundTaskOutput" | "stopBackgroundTask" | "getWorkflowRuns" | "stopWorkflowRun" | "listGitBranches" | "createWorktree" | "removeWorktree";
+export type RequestMethod = "initialize" | "destroy" | "restoreSession" | "listSessions" | "getSessionInfo" | "sendMessage" | "bang" | "askBtw" | "abortMessage" | "clearMessages" | "rewindToMessage" | "listRewindCheckpoints" | "deleteQueuedMessage" | "updateQueuedMessage" | "deleteQueuedMessageById" | "getMessages" | "getFullMessageThread" | "setPermissionMode" | "getPermissionMode" | "getMcpServers" | "connectMcpServer" | "disconnectMcpServer" | "getSlashCommands" | "searchFiles" | "writeArtifactFile" | "getPromptHistory" | "searchPromptHistory" | "updateConfig" | "getConfiguredModels" | "setModel" | "listPendingPermissions" | "listDaemonSessions" | "getAuthStatus" | "login" | "logout" | "listPlugins" | "installPlugin" | "uninstallPlugin" | "enablePlugin" | "disablePlugin" | "updatePlugin" | "listMarketplaces" | "addMarketplace" | "removeMarketplace" | "updateMarketplace" | "compact" | "getBackgroundTaskOutput" | "stopBackgroundTask" | "getWorkflowRuns" | "stopWorkflowRun" | "listGitBranches" | "createWorktree" | "removeWorktree";
 export type ClientNotificationMethod = "permissionResponse";
 export type ServerNotificationMethod = "userMessageAdded" | "assistantMessageAdded" | "assistantContentUpdated" | "assistantReasoningUpdated" | "toolBlockUpdated" | "errorBlockAdded" | "loadingChange" | "commandRunningChange" | "queuedMessagesChange" | "tasksChange" | "sessionIdChange" | "permissionModeChange" | "mcpServersChange" | "workdirChange" | "bangMessageAdded" | "bangMessageUpdated" | "bangMessageCompleted" | "notificationMessageAdded" | "permissionRequest" | "authUrl" | "compactBlockAdded" | "compactionStateChange" | "backgroundTasksChange" | "btwContent";
 export declare function isRequest(msg: unknown): msg is JsonRpcRequest;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-code",
-  "version": "1.0.7",
+  "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.7"
+    "wave-agent-sdk": "1.0.9"
   },
   "devDependencies": {
     "@types/react": "^19.1.8",

package/src/components/ChatInterface.tsx

@@ -142,6 +142,7 @@ export const ChatInterface: React.FC = () => {
                   toolName={confirmingTool!.name}
                   toolInput={confirmingTool!.input}
                   planContent={confirmingTool!.planContent}
+                  warning={confirmingTool!.warning}
                   isExpanded={isExpanded}
                 />
               )}
@@ -151,6 +152,7 @@ export const ChatInterface: React.FC = () => {
               toolName={confirmingTool!.name}
               toolInput={confirmingTool!.input}
               planContent={confirmingTool!.planContent}
+              warning={confirmingTool!.warning}
               isExpanded={isExpanded}
             />
           )}

package/src/components/ConfirmationDetails.tsx

@@ -7,6 +7,7 @@ import {
   EXIT_PLAN_MODE_TOOL_NAME,
   ENTER_PLAN_MODE_TOOL_NAME,
   ASK_USER_QUESTION_TOOL_NAME,
+  ARTIFACT_TOOL_NAME,
 } from "wave-agent-sdk";
 import { DiffDisplay } from "./DiffDisplay.js";
 import { PlanDisplay } from "./PlanDisplay.js";
@@ -34,6 +35,8 @@ const getActionDescription = (
       return "Enter plan mode for complex task planning";
     case ASK_USER_QUESTION_TOOL_NAME:
       return "Answer questions to clarify intent";
+    case ARTIFACT_TOOL_NAME:
+      return `Publish file: ${toolInput.file_path || "unknown file"}`;
     default:
       return "Execute operation";
   }
@@ -43,6 +46,7 @@ export interface ConfirmationDetailsProps {
   toolName: string;
   toolInput?: Record<string, unknown>;
   planContent?: string;
+  warning?: string;
   isExpanded?: boolean;
 }
 
@@ -50,6 +54,7 @@ export const ConfirmationDetails: React.FC<ConfirmationDetailsProps> = ({
   toolName,
   toolInput,
   planContent,
+  warning,
   isExpanded = false,
 }) => {
   const startLineNumber =
@@ -69,6 +74,7 @@ export const ConfirmationDetails: React.FC<ConfirmationDetailsProps> = ({
         Tool: {toolName}
       </Text>
       <Text color="yellow">{getActionDescription(toolName, toolInput)}</Text>
+      {warning && <Text color="red">⚠ {warning}</Text>}
 
       <DiffDisplay
         toolName={toolName}

package/src/components/RewindCommand.tsx

@@ -34,10 +34,16 @@ export const RewindCommand: React.FC<RewindCommandProps> = ({
   }, [getFullMessageThread]);
 
   // Filter user messages as checkpoints, excluding meta messages and
-  // system-generated user-role messages (task notifications, hook injections)
-  const checkpoints = messages
-    .map((msg, index) => ({ msg, index }))
-    .filter(({ msg }) => isUserCheckpointMessage(msg));
+  // system-generated user-role messages (task notifications, hook injections).
+  // Compaction is append-only: the same message id appears twice on the full
+  // thread (pre-compact history + post-compact append), so dedupe by id and
+  // keep the last occurrence (matching the folded view the user sees).
+  const checkpointMap = new Map<string, { msg: Message; index: number }>();
+  messages.forEach((msg, index) => {
+    if (!isUserCheckpointMessage(msg)) return;
+    checkpointMap.set(msg.id ?? `index:${index}`, { msg, index });
+  });
+  const checkpoints = Array.from(checkpointMap.values());
 
   const MAX_VISIBLE_ITEMS = 3;
 

package/src/contexts/useChat.tsx

@@ -31,7 +31,6 @@ import {
   extractLatestTotalTokens,
 } 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";
 
@@ -112,6 +111,7 @@ export interface ChatContextType {
     hidePersistentOption?: boolean;
     planContent?: string;
     permissionMode?: PermissionMode;
+    warning?: string;
   };
   showConfirmation: (
     toolName: string,
@@ -120,6 +120,7 @@ export interface ChatContextType {
     hidePersistentOption?: boolean,
     planContent?: string,
     permissionMode?: PermissionMode,
+    warning?: string,
   ) => Promise<PermissionDecision>;
   hideConfirmation: () => void;
   handleConfirmationDecision: (decision: PermissionDecision) => void;
@@ -167,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
@@ -239,6 +256,99 @@ function createStreamingWindowThrottle(
   return throttled;
 }
 
+/**
+ * Per-tool window-concat throttle for pure-delta tool parameter streaming:
+ * `parametersChunk` deltas are accumulated independently per tool block id
+ * within the cooldown window, so interleaved multi-tool streams lose no delta
+ * (a plain throttle's single last-args slot would drop every earlier tool's
+ * deltas, leaving the first tool without streaming parameters). `start` /
+ * `running` apply immediately (one-shot snapshots); `end` flushes pending
+ * streaming deltas first, then applies the authoritative parameters/result.
+ */
+export function createToolStreamingThrottle(
+  fn: (params: ToolBlockUpdateCallbackParams) => void,
+  wait: number,
+): {
+  (params: ToolBlockUpdateCallbackParams): void;
+  cancel: () => void;
+  flush: () => void;
+} {
+  let timer: NodeJS.Timeout | null = null;
+  let pending: { messageId: string; chunks: Map<string, string> } | null = null;
+
+  const fire = () => {
+    if (pending && pending.chunks.size > 0) {
+      const { messageId, chunks } = pending;
+      pending = null;
+      for (const [id, chunk] of chunks) {
+        fn({ messageId, id, parametersChunk: chunk, stage: "streaming" });
+      }
+    }
+  };
+
+  const throttled = (params: ToolBlockUpdateCallbackParams) => {
+    if (params.stage === "end") {
+      // Flush any deltas still pending inside the cooldown window first
+      if (timer) {
+        clearTimeout(timer);
+        timer = null;
+      }
+      fire();
+      fn(params);
+      return;
+    }
+    if (params.stage === "streaming") {
+      if (!pending) {
+        pending = { messageId: params.messageId, chunks: new Map() };
+      }
+      const prev = pending.chunks.get(params.id) || "";
+      pending.chunks.set(params.id, prev + (params.parametersChunk || ""));
+      if (!timer) {
+        timer = setTimeout(() => {
+          timer = null;
+          fire();
+        }, wait);
+      }
+      return;
+    }
+    // start / running — one-shot snapshots applied immediately. Drop this
+    // tool's buffered streaming deltas first: start/running carry the
+    // authoritative parameters, and a pending timer would otherwise fire late
+    // with a stale `streaming` event, regressing this tool block's stage back
+    // to streaming (yellow dot -> gray) mid-execution. Other tools' in-flight
+    // chunks are kept so interleaved multi-tool streaming still accumulates.
+    if (pending) {
+      pending.chunks.delete(params.id);
+      if (pending.chunks.size === 0) {
+        pending = null;
+        if (timer) {
+          clearTimeout(timer);
+          timer = null;
+        }
+      }
+    }
+    fn(params);
+  };
+
+  throttled.cancel = () => {
+    if (timer) {
+      clearTimeout(timer);
+      timer = null;
+    }
+    pending = null;
+  };
+
+  throttled.flush = () => {
+    if (timer) {
+      clearTimeout(timer);
+      timer = null;
+    }
+    fire();
+  };
+
+  return throttled;
+}
+
 export const ChatProvider: React.FC<ChatProviderProps> = ({
   children,
   bypassPermissions,
@@ -345,8 +455,13 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
   const throttledToolBlockUpdate = useMemo(
     () =>
-      throttle((params: ToolBlockUpdateCallbackParams) => {
-        const { messageId, id: toolBlockId, ...updates } = params;
+      createToolStreamingThrottle((params) => {
+        const {
+          messageId,
+          id: toolBlockId,
+          parametersChunk,
+          ...updates
+        } = params;
         setMessages((prev) =>
           prev.map((m) => {
             if (m.id !== messageId) return m;
@@ -363,7 +478,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
                     id: toolBlockId,
                     name: updates.name || "",
                     stage: updates.stage || "start",
-                    parameters: updates.parameters || "",
+                    parameters:
+                      (updates.parameters || "") + (parametersChunk || ""),
                     result: updates.result || "",
                     ...updates,
                   },
@@ -374,7 +490,18 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
               ...m,
               blocks: m.blocks.map((b, idx) =>
                 idx === toolBlockIndex && b.type === "tool"
-                  ? { ...b, ...updates }
+                  ? {
+                      ...b,
+                      ...updates,
+                      // Streaming carries only the delta; append it to the
+                      // accumulated parameters. start/running/end carry the
+                      // authoritative value and replace wholesale.
+                      parameters: parametersChunk
+                        ? (b.parameters || "") + parametersChunk
+                        : updates.parameters !== undefined
+                          ? updates.parameters
+                          : b.parameters,
+                    }
                   : b,
               ),
             };
@@ -443,6 +570,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         hidePersistentOption?: boolean;
         planContent?: string;
         permissionMode?: PermissionMode;
+        warning?: string;
       }
     | undefined
   >();
@@ -454,6 +582,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
       hidePersistentOption?: boolean;
       planContent?: string;
       permissionMode?: PermissionMode;
+      warning?: string;
       resolver: (decision: PermissionDecision) => void;
       reject: () => void;
     }>
@@ -465,6 +594,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     hidePersistentOption?: boolean;
     planContent?: string;
     permissionMode?: PermissionMode;
+    warning?: string;
     resolver: (decision: PermissionDecision) => void;
     reject: () => void;
   } | null>(null);
@@ -491,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));
     }
@@ -506,6 +636,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
       hidePersistentOption?: boolean,
       planContent?: string,
       permissionMode?: PermissionMode,
+      warning?: string,
     ): Promise<PermissionDecision> => {
       return new Promise<PermissionDecision>((resolve, reject) => {
         const queueItem = {
@@ -515,6 +646,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           hidePersistentOption,
           planContent,
           permissionMode,
+          warning,
           resolver: resolve,
           reject,
         };
@@ -540,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) => {
@@ -548,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) => {
@@ -723,6 +859,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
               context.hidePersistentOption,
               context.planContent,
               context.permissionMode,
+              context.warning,
             );
           } catch {
             // If confirmation was cancelled or failed, deny the operation
@@ -772,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);
@@ -1070,6 +1208,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         hidePersistentOption: next.hidePersistentOption,
         planContent: next.planContent,
         permissionMode: next.permissionMode,
+        warning: next.warning,
       });
       setIsConfirmationVisible(true);
       setConfirmationQueue((prev) => prev.slice(1));