wave-code 0.6.5 → 0.7.1

package/dist/components/HistorySearch.js

@@ -1,34 +1,33 @@
 import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
-import React, { useState, useEffect } from "react";
+import { useState, useEffect } from "react";
 import { Box, Text, useInput } from "ink";
 import { PromptHistoryManager } from "wave-agent-sdk";
 export const HistorySearch = ({ searchQuery, onSelect, onCancel, }) => {
     const MAX_VISIBLE_ITEMS = 5;
-    const [selectedIndex, setSelectedIndex] = useState(0);
-    const [entries, setEntries] = useState([]);
-    const entriesRef = React.useRef([]);
-    const selectedIndexRef = React.useRef(0);
-    useEffect(() => {
-        entriesRef.current = entries;
-    }, [entries]);
-    useEffect(() => {
-        selectedIndexRef.current = selectedIndex;
-    }, [selectedIndex]);
+    const [state, setState] = useState({
+        selectedIndex: 0,
+        entries: [],
+    });
     useEffect(() => {
         const fetchHistory = async () => {
             const results = await PromptHistoryManager.searchHistory(searchQuery);
             const limitedResults = results.slice(0, 20);
-            setEntries(limitedResults); // Limit to 20 results
-            setSelectedIndex(0);
+            setState({
+                entries: limitedResults,
+                selectedIndex: 0,
+            });
         };
         fetchHistory();
     }, [searchQuery]);
     useInput((input, key) => {
         if (key.return) {
-            if (entriesRef.current.length > 0 &&
-                selectedIndexRef.current < entriesRef.current.length) {
-                onSelect(entriesRef.current[selectedIndexRef.current].prompt);
-            }
+            setState((prev) => {
+                if (prev.entries.length > 0 &&
+                    prev.selectedIndex < prev.entries.length) {
+                    onSelect(prev.entries[prev.selectedIndex].prompt);
+                }
+                return prev;
+            });
             return;
         }
         if (key.escape) {
@@ -36,14 +35,21 @@ export const HistorySearch = ({ searchQuery, onSelect, onCancel, }) => {
             return;
         }
         if (key.upArrow) {
-            setSelectedIndex((prev) => Math.max(0, prev - 1));
+            setState((prev) => ({
+                ...prev,
+                selectedIndex: Math.max(0, prev.selectedIndex - 1),
+            }));
             return;
         }
         if (key.downArrow) {
-            setSelectedIndex((prev) => Math.min(entriesRef.current.length - 1, prev + 1));
+            setState((prev) => ({
+                ...prev,
+                selectedIndex: Math.min(prev.entries.length - 1, prev.selectedIndex + 1),
+            }));
             return;
         }
     });
+    const { entries, selectedIndex } = state;
     if (entries.length === 0) {
         return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "yellow", borderBottom: false, borderLeft: false, borderRight: false, children: [_jsxs(Text, { color: "yellow", children: ["No history found ", searchQuery && `for "${searchQuery}"`] }), _jsx(Text, { dimColor: true, children: "Press Escape to cancel" })] }));
     }
@@ -70,6 +76,6 @@ export const HistorySearch = ({ searchQuery, onSelect, onCancel, }) => {
     return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "blue", borderBottom: false, borderLeft: false, borderRight: false, gap: 1, children: [_jsx(Box, { children: _jsxs(Text, { color: "blue", bold: true, children: ["Prompt History ", searchQuery && `(filtering: "${searchQuery}")`] }) }), _jsx(Box, { flexDirection: "column", children: visibleEntries.map((entry, index) => {
                     const actualIndex = startIndex + index;
                     const isSelected = actualIndex === selectedIndex;
-                    return (_jsxs(Box, { flexDirection: "row", justifyContent: "space-between", children: [_jsx(Box, { flexShrink: 1, children: _jsx(Text, { color: isSelected ? "black" : "white", backgroundColor: isSelected ? "blue" : undefined, wrap: "truncate-end", children: entry.prompt.replace(/\n/g, " ") }) }), isSelected && (_jsx(Box, { marginLeft: 2, flexShrink: 0, children: _jsx(Text, { color: "gray", dimColor: true, children: formatTimestamp(entry.timestamp) }) }))] }, actualIndex));
+                    return (_jsxs(Box, { flexDirection: "row", justifyContent: "space-between", children: [_jsx(Box, { flexShrink: 1, children: _jsxs(Text, { color: isSelected ? "black" : "white", backgroundColor: isSelected ? "blue" : undefined, wrap: "truncate-end", children: [isSelected ? "> " : "  ", entry.prompt.replace(/\n/g, " ")] }) }), isSelected && (_jsx(Box, { marginLeft: 2, flexShrink: 0, children: _jsx(Text, { color: "gray", dimColor: true, children: formatTimestamp(entry.timestamp) }) }))] }, actualIndex));
                 }) }), _jsx(Box, { children: _jsx(Text, { dimColor: true, children: "Use \u2191\u2193 to navigate, Enter to select, Escape to cancel" }) })] }));
 };

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

@@ -1 +1 @@
-{"version":3,"file":"Markdown.d.ts","sourceRoot":"","sources":["../../src/components/Markdown.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkB,MAAM,OAAO,CAAC;AAKvC,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAwID,eAAO,MAAM,QAAQ,2CAA6B,aAAa,6CAc7D,CAAC"}
+{"version":3,"file":"Markdown.d.ts","sourceRoot":"","sources":["../../src/components/Markdown.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkB,MAAM,OAAO,CAAC;AAMvC,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAyID,eAAO,MAAM,QAAQ,2CAA6B,aAAa,6CAc7D,CAAC"}

package/dist/components/Markdown.js

@@ -3,6 +3,7 @@ import React, { useMemo } from "react";
 import { Box, Text } from "ink";
 import { Renderer, marked } from "marked";
 import chalk from "chalk";
+import { highlightToAnsi } from "../utils/highlightUtils.js";
 const unescapeHtml = (html) => {
     return html
         .replace(/&/g, "&")
@@ -16,7 +17,8 @@ class AnsiRenderer extends Renderer {
     code({ text, lang }) {
         const prefix = lang ? `\`\`\`${lang}` : "```";
         const suffix = "```";
-        return `\n${chalk.gray(prefix)}\n${text}\n${chalk.gray(suffix)}\n`;
+        const highlighted = highlightToAnsi(text, lang);
+        return `\n${chalk.gray(prefix)}\n${highlighted}\n${chalk.gray(suffix)}\n`;
     }
     blockquote({ tokens }) {
         const body = this.parser.parse(tokens);

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

@@ -1 +1 @@
-{"version":3,"file":"McpManager.d.ts","sourceRoot":"","sources":["../../src/components/McpManager.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,IAAI,CAAC;IACrB,OAAO,EAAE,eAAe,EAAE,CAAC;IAC3B,eAAe,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAC1D,kBAAkB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;CAC9D;AAED,eAAO,MAAM,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,eAAe,CAkUhD,CAAC"}
+{"version":3,"file":"McpManager.d.ts","sourceRoot":"","sources":["../../src/components/McpManager.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,IAAI,CAAC;IACrB,OAAO,EAAE,eAAe,EAAE,CAAC;IAC3B,eAAe,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAC1D,kBAAkB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;CAC9D;AAED,eAAO,MAAM,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,eAAe,CA0ThD,CAAC"}

package/dist/components/McpManager.js

@@ -44,64 +44,61 @@ export const McpManager = ({ onCancel, servers, onConnectServer, onDisconnectSer
         await onDisconnectServer(serverName);
     };
     useInput((input, key) => {
-        if (viewMode === "list") {
-            // List mode navigation
-            if (key.return) {
-                if (servers.length > 0 && selectedIndex < servers.length) {
-                    setViewMode("detail");
+        if (key.return) {
+            setViewMode((prevMode) => {
+                if (prevMode === "list") {
+                    setSelectedIndex((prevIndex) => {
+                        if (servers.length > 0 && prevIndex < servers.length) {
+                            // We can't call setViewMode here because we're already in a setViewMode call
+                            // But we can return the new mode from the outer setViewMode
+                        }
+                        return prevIndex;
+                    });
+                    return "detail";
+                }
+                return prevMode;
+            });
+            return;
+        }
+        if (key.escape) {
+            setViewMode((prev) => {
+                if (prev === "detail") {
+                    return "list";
                 }
-                return;
-            }
-            if (key.escape) {
                 onCancel();
-                return;
-            }
-            if (key.upArrow) {
-                setSelectedIndex(Math.max(0, selectedIndex - 1));
-                return;
-            }
-            if (key.downArrow) {
-                setSelectedIndex(Math.min(servers.length - 1, selectedIndex + 1));
-                return;
-            }
-            // Hotkeys for server actions
-            if (input === "c" &&
-                servers.length > 0 &&
-                selectedIndex < servers.length) {
-                const server = servers[selectedIndex];
-                if (server.status === "disconnected" || server.status === "error") {
+                return prev;
+            });
+            return;
+        }
+        if (key.upArrow) {
+            setSelectedIndex((prev) => Math.max(0, prev - 1));
+            return;
+        }
+        if (key.downArrow) {
+            setSelectedIndex((prev) => Math.min(servers.length - 1, prev + 1));
+            return;
+        }
+        // Hotkeys for server actions
+        if (input === "c") {
+            setSelectedIndex((prev) => {
+                const server = servers[prev];
+                if (server &&
+                    (server.status === "disconnected" || server.status === "error")) {
                     handleConnect(server.name);
                 }
-                return;
-            }
-            if (input === "d" &&
-                servers.length > 0 &&
-                selectedIndex < servers.length) {
-                const server = servers[selectedIndex];
-                if (server.status === "connected") {
-                    handleDisconnect(server.name);
-                }
-                return;
-            }
+                return prev;
+            });
+            return;
         }
-        else if (viewMode === "detail") {
-            // Detail mode navigation
-            if (key.escape) {
-                setViewMode("list");
-                return;
-            }
-            if (selectedServer) {
-                if (input === "c" &&
-                    (selectedServer.status === "disconnected" ||
-                        selectedServer.status === "error")) {
-                    handleConnect(selectedServer.name);
-                    return;
-                }
-                if (input === "d" && selectedServer.status === "connected") {
-                    handleDisconnect(selectedServer.name);
-                    return;
+        if (input === "d") {
+            setSelectedIndex((prev) => {
+                const server = servers[prev];
+                if (server && server.status === "connected") {
+                    handleDisconnect(server.name);
                 }
-            }
+                return prev;
+            });
+            return;
         }
     });
     if (viewMode === "detail" && selectedServer) {

package/dist/components/MessageBlockItem.d.ts

@@ -0,0 +1,9 @@
+import type { Message, MessageBlock } from "wave-agent-sdk";
+export interface MessageBlockItemProps {
+    block: MessageBlock;
+    message: Message;
+    isExpanded: boolean;
+    paddingTop?: number;
+}
+export declare const MessageBlockItem: ({ block, message, isExpanded, paddingTop, }: MessageBlockItemProps) => import("react/jsx-runtime").JSX.Element;
+//# sourceMappingURL=MessageBlockItem.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"MessageBlockItem.d.ts","sourceRoot":"","sources":["../../src/components/MessageBlockItem.tsx"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAQ5D,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,YAAY,CAAC;IACpB,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,eAAO,MAAM,gBAAgB,GAAI,6CAK9B,qBAAqB,4CA4DvB,CAAC"}

package/dist/components/MessageBlockItem.js

@@ -0,0 +1,11 @@
+import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+import { Box, Text } from "ink";
+import { MessageSource } from "wave-agent-sdk";
+import { CommandOutputDisplay } from "./CommandOutputDisplay.js";
+import { ToolDisplay } from "./ToolDisplay.js";
+import { CompressDisplay } from "./CompressDisplay.js";
+import { ReasoningDisplay } from "./ReasoningDisplay.js";
+import { Markdown } from "./Markdown.js";
+export const MessageBlockItem = ({ block, message, isExpanded, paddingTop = 0, }) => {
+    return (_jsxs(Box, { flexDirection: "column", paddingTop: paddingTop, children: [block.type === "text" && block.content.trim() && (_jsxs(Box, { children: [block.customCommandContent && (_jsxs(Text, { color: "cyan", bold: true, children: ["$", " "] })), block.source === MessageSource.HOOK && (_jsxs(Text, { color: "magenta", bold: true, children: ["~", " "] })), message.role === "user" ? (_jsx(Text, { backgroundColor: "gray", color: "white", children: block.content })) : (_jsx(Markdown, { children: block.content }))] })), block.type === "error" && (_jsx(Box, { children: _jsxs(Text, { color: "red", children: ["Error: ", block.content] }) })), block.type === "command_output" && (_jsx(CommandOutputDisplay, { block: block, isExpanded: isExpanded })), block.type === "tool" && (_jsx(ToolDisplay, { block: block, isExpanded: isExpanded })), block.type === "image" && (_jsxs(Box, { children: [_jsx(Text, { color: "magenta", bold: true, children: "# Image" }), block.imageUrls && block.imageUrls.length > 0 && (_jsxs(Text, { color: "gray", dimColor: true, children: [" ", "(", block.imageUrls.length, ")"] }))] })), block.type === "compress" && (_jsx(CompressDisplay, { block: block, isExpanded: isExpanded })), block.type === "reasoning" && _jsx(ReasoningDisplay, { block: block })] }));
+};

package/dist/components/MessageList.d.ts

@@ -2,10 +2,8 @@ import React from "react";
 import type { Message } from "wave-agent-sdk";
 export interface MessageListProps {
     messages: Message[];
-    isLoading?: boolean;
-    isCommandRunning?: boolean;
     isExpanded?: boolean;
-    forceStaticLastMessage?: boolean;
+    hideDynamicBlocks?: boolean;
 }
-export declare const MessageList: React.MemoExoticComponent<({ messages, isLoading, isCommandRunning, isExpanded, forceStaticLastMessage, }: MessageListProps) => import("react/jsx-runtime").JSX.Element>;
+export declare const MessageList: React.MemoExoticComponent<({ messages, isExpanded, hideDynamicBlocks, }: MessageListProps) => import("react/jsx-runtime").JSX.Element>;
 //# sourceMappingURL=MessageList.d.ts.map

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,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAG9C,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,sBAAsB,CAAC,EAAE,OAAO,CAAC;CAClC;AAED,eAAO,MAAM,WAAW,6GAOnB,gBAAgB,6CA4EpB,CAAC"}
+{"version":3,"file":"MessageList.d.ts","sourceRoot":"","sources":["../../src/components/MessageList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,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,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,eAAO,MAAM,WAAW,2EAKnB,gBAAgB,6CAmFpB,CAAC"}

package/dist/components/MessageList.js

@@ -1,8 +1,8 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import React from "react";
 import { Box, Text, Static } from "ink";
-import { MessageItem } from "./MessageItem.js";
-export const MessageList = React.memo(({ messages, isLoading = false, isCommandRunning = false, isExpanded = false, forceStaticLastMessage = false, }) => {
+import { MessageBlockItem } from "./MessageBlockItem.js";
+export const MessageList = React.memo(({ messages, isExpanded = false, hideDynamicBlocks = false, }) => {
     // Empty message state
     if (messages.length === 0) {
         return (_jsx(Box, { flexDirection: "column", gap: 1, children: _jsx(Box, { flexDirection: "column", paddingY: 1, children: _jsx(Text, { color: "gray", children: "Welcome to WAVE Code Assistant!" }) }) }));
@@ -13,27 +13,32 @@ export const MessageList = React.memo(({ messages, isLoading = false, isCommandR
     const displayMessages = shouldLimitMessages
         ? messages.slice(-maxExpandedMessages)
         : messages;
-    // Compute which messages to render statically vs dynamically
-    const lastMessage = displayMessages[displayMessages.length - 1];
-    const hasNonEndTool = lastMessage?.blocks.some((block) => block.type === "tool" && block.stage !== "end");
-    const hasRunningCommand = lastMessage?.blocks.some((block) => block.type === "command_output" && block.isRunning);
-    const shouldRenderLastDynamic = !forceStaticLastMessage &&
-        (isLoading || isCommandRunning || hasNonEndTool || hasRunningCommand);
-    const staticMessages = shouldRenderLastDynamic
-        ? displayMessages.slice(0, -1)
-        : displayMessages;
-    const dynamicMessages = shouldRenderLastDynamic && displayMessages.length > 0
-        ? [displayMessages[displayMessages.length - 1]]
-        : [];
-    return (_jsxs(Box, { flexDirection: "column", paddingBottom: 1, children: [_jsx(Static, { items: staticMessages, children: (message, key) => {
-                    // Get previous message
-                    const previousMessage = key > 0 ? staticMessages[key - 1] : undefined;
-                    return (_jsx(MessageItem, { message: message, shouldShowHeader: previousMessage?.role !== message.role, isExpanded: isExpanded }, key));
-                } }), dynamicMessages.map((message, index) => {
-                const messageIndex = staticMessages.length + index;
-                const previousMessage = messageIndex > 0 ? displayMessages[messageIndex - 1] : undefined;
-                return (_jsx(Box, { children: _jsx(MessageItem, { message: message, shouldShowHeader: previousMessage?.role !== message.role, isExpanded: isExpanded }) }, `dynamic-${index}`));
-            })] }));
+    // Flatten messages into blocks with metadata
+    const allBlocks = displayMessages.flatMap((message, index) => {
+        const messageIndex = shouldLimitMessages
+            ? messages.length - maxExpandedMessages + index
+            : index;
+        return message.blocks.map((block, blockIndex) => ({
+            block,
+            message,
+            isLastMessage: messageIndex === messages.length - 1,
+            // Unique key for each block to help Static component
+            key: `${message.id || messageIndex}-${blockIndex}`,
+        }));
+    });
+    // Determine which blocks are static vs dynamic
+    const blocksWithStatus = allBlocks.map((item) => {
+        const { block, isLastMessage } = item;
+        const isDynamic = isLastMessage &&
+            ((block.type === "tool" && block.stage !== "end") ||
+                (block.type === "command_output" && block.isRunning));
+        return { ...item, isDynamic };
+    });
+    const staticBlocks = blocksWithStatus.filter((b) => !b.isDynamic);
+    const dynamicBlocks = hideDynamicBlocks
+        ? []
+        : blocksWithStatus.filter((b) => b.isDynamic);
+    return (_jsxs(Box, { flexDirection: "column", paddingBottom: 1, children: [staticBlocks.length > 0 && (_jsx(Static, { items: staticBlocks, children: (item) => (_jsx(MessageBlockItem, { block: item.block, message: item.message, isExpanded: isExpanded, paddingTop: 1 }, item.key)) })), dynamicBlocks.length > 0 && (_jsx(Box, { flexDirection: "column", children: dynamicBlocks.map((item) => (_jsx(MessageBlockItem, { block: item.block, message: item.message, isExpanded: isExpanded, paddingTop: 1 }, item.key))) }))] }));
 });
 // Add display name for debugging
 MessageList.displayName = "MessageList";

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

@@ -1 +1 @@
-{"version":3,"file":"PluginDetail.d.ts","sourceRoot":"","sources":["../../src/components/PluginDetail.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAUxC,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAwIhC,CAAC"}
+{"version":3,"file":"PluginDetail.d.ts","sourceRoot":"","sources":["../../src/components/PluginDetail.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAUxC,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAuIhC,CAAC"}

package/dist/components/PluginDetail.js

@@ -24,36 +24,33 @@ export const PluginDetail = () => {
             actions.setView(isFromDiscover ? "DISCOVER" : "INSTALLED");
         }
         else if (key.upArrow) {
-            if (isInstalledAndEnabled) {
-                setSelectedActionIndex((prev) => prev > 0 ? prev - 1 : INSTALLED_ACTIONS.length - 1);
-            }
-            else {
-                setSelectedScopeIndex((prev) => prev > 0 ? prev - 1 : SCOPES.length - 1);
-            }
+            setSelectedActionIndex((prev) => prev > 0 ? prev - 1 : INSTALLED_ACTIONS.length - 1);
+            setSelectedScopeIndex((prev) => prev > 0 ? prev - 1 : SCOPES.length - 1);
         }
         else if (key.downArrow) {
-            if (isInstalledAndEnabled) {
-                setSelectedActionIndex((prev) => prev < INSTALLED_ACTIONS.length - 1 ? prev + 1 : 0);
-            }
-            else {
-                setSelectedScopeIndex((prev) => prev < SCOPES.length - 1 ? prev + 1 : 0);
-            }
+            setSelectedActionIndex((prev) => prev < INSTALLED_ACTIONS.length - 1 ? prev + 1 : 0);
+            setSelectedScopeIndex((prev) => prev < SCOPES.length - 1 ? prev + 1 : 0);
         }
         else if (key.return && plugin) {
             if (isInstalledAndEnabled) {
-                const action = INSTALLED_ACTIONS[selectedActionIndex].id;
-                if (action === "uninstall") {
-                    actions.uninstallPlugin(plugin.name, plugin.marketplace);
-                }
-                else {
-                    actions.updatePlugin(plugin.name, plugin.marketplace);
-                }
-                actions.setView("INSTALLED");
+                setSelectedActionIndex((prev) => {
+                    const action = INSTALLED_ACTIONS[prev].id;
+                    if (action === "uninstall") {
+                        actions.uninstallPlugin(plugin.name, plugin.marketplace);
+                    }
+                    else {
+                        actions.updatePlugin(plugin.name, plugin.marketplace);
+                    }
+                    return prev;
+                });
             }
             else {
-                actions.installPlugin(plugin.name, plugin.marketplace, SCOPES[selectedScopeIndex].id);
-                actions.setView("INSTALLED");
+                setSelectedScopeIndex((prev) => {
+                    actions.installPlugin(plugin.name, plugin.marketplace, SCOPES[prev].id);
+                    return prev;
+                });
             }
+            actions.setView("INSTALLED");
         }
     });
     if (!plugin) {

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

@@ -1 +1 @@
-{"version":3,"file":"SessionSelector.d.ts","sourceRoot":"","sources":["../../src/components/SessionSelector.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEtD,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,CAAC,eAAe,GAAG;QAAE,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,CAAC;IAC1D,QAAQ,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAoH1D,CAAC"}
+{"version":3,"file":"SessionSelector.d.ts","sourceRoot":"","sources":["../../src/components/SessionSelector.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEtD,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,CAAC,eAAe,GAAG;QAAE,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,CAAC;IAC1D,QAAQ,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAuH1D,CAAC"}

package/dist/components/SessionSelector.js

@@ -5,9 +5,12 @@ export const SessionSelector = ({ sessions, onSelect, onCancel, }) => {
     const [selectedIndex, setSelectedIndex] = useState(0);
     useInput((input, key) => {
         if (key.return) {
-            if (sessions.length > 0 && selectedIndex < sessions.length) {
-                onSelect(sessions[selectedIndex].id);
-            }
+            setSelectedIndex((prev) => {
+                if (sessions.length > 0 && prev < sessions.length) {
+                    onSelect(sessions[prev].id);
+                }
+                return prev;
+            });
             return;
         }
         if (key.escape) {
@@ -15,11 +18,11 @@ export const SessionSelector = ({ sessions, onSelect, onCancel, }) => {
             return;
         }
         if (key.upArrow) {
-            setSelectedIndex(Math.max(0, selectedIndex - 1));
+            setSelectedIndex((prev) => Math.max(0, prev - 1));
             return;
         }
         if (key.downArrow) {
-            setSelectedIndex(Math.min(sessions.length - 1, selectedIndex + 1));
+            setSelectedIndex((prev) => Math.min(sessions.length - 1, prev + 1));
             return;
         }
     });

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

@@ -1 +1 @@
-{"version":3,"file":"TaskList.d.ts","sourceRoot":"","sources":["../../src/components/TaskList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAK1B,eAAO,MAAM,QAAQ,EAAE,KAAK,CAAC,EAoD5B,CAAC"}
+{"version":3,"file":"TaskList.d.ts","sourceRoot":"","sources":["../../src/components/TaskList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAK1B,eAAO,MAAM,QAAQ,EAAE,KAAK,CAAC,EAiD5B,CAAC"}

package/dist/components/TaskList.js

@@ -8,10 +8,7 @@ export const TaskList = () => {
     if (tasks.length === 0 || !isTaskListVisible) {
         return null;
     }
-    const getStatusIcon = (status, isBlocked) => {
-        if (isBlocked) {
-            return _jsx(Text, { color: "red", children: "\uD83D\uDD12" });
-        }
+    const getStatusIcon = (status) => {
         switch (status) {
             case "pending":
                 return _jsx(Text, { color: "gray", children: "\u25A1" });
@@ -35,6 +32,6 @@ export const TaskList = () => {
                 ? ` (Blocked by: ${blockingTaskIds.join(", ")})`
                 : "";
             const fullText = `${task.subject}${blockedByText}`;
-            return (_jsxs(Box, { gap: 1, children: [getStatusIcon(task.status, isBlocked), _jsx(Text, { dimColor: isDimmed, children: fullText })] }, task.id));
+            return (_jsxs(Box, { gap: 1, children: [getStatusIcon(task.status), _jsx(Text, { dimColor: isDimmed, children: fullText })] }, task.id));
         }) }));
 };

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

@@ -1 +1 @@
-{"version":3,"file":"ToolDisplay.d.ts","sourceRoot":"","sources":["../../src/components/ToolDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAGhD,UAAU,gBAAgB;IACxB,KAAK,EAAE,SAAS,CAAC;IACjB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CA+HlD,CAAC"}
+{"version":3,"file":"ToolDisplay.d.ts","sourceRoot":"","sources":["../../src/components/ToolDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAGhD,UAAU,gBAAgB;IACxB,KAAK,EAAE,SAAS,CAAC;IACjB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CAmIlD,CAAC"}

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)] }) })), stage === "end" && success && (_jsx(DiffDisplay, { toolName: name, parameters: parameters }))] }));
+    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)] }) })), stage === "end" && success && (_jsx(DiffDisplay, { toolName: name, parameters: parameters, startLineNumber: block.startLineNumber }))] }));
 };

package/dist/contexts/useChat.d.ts

@@ -58,6 +58,7 @@ export interface ChatProviderProps {
     children: React.ReactNode;
     bypassPermissions?: boolean;
     pluginDirs?: string[];
+    tools?: string[];
 }
 export declare const ChatProvider: React.FC<ChatProviderProps>;
 //# sourceMappingURL=useChat.d.ts.map

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;AAUxB,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;IAEjD,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,KAC/C,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,EAAE,MAAM,IAAI,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAE1B,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,YAAY,EAAE,IAAI,EAAE,CAAC;IACrB,uBAAuB,EAAE,CACvB,MAAM,EAAE,MAAM,KACX;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAC/D,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;IAElD,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;KAChC,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,KAC3B,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;CACxE;AAID,eAAO,MAAM,OAAO,uBAMnB,CAAC;AAEF,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAsepD,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;AAUxB,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;IAEjD,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,KAC/C,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,EAAE,MAAM,IAAI,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAE1B,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,YAAY,EAAE,IAAI,EAAE,CAAC;IACrB,uBAAuB,EAAE,CACvB,MAAM,EAAE,MAAM,KACX;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAC/D,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;IAElD,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;KAChC,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,KAC3B,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;CACxE;AAID,eAAO,MAAM,OAAO,uBAMnB,CAAC;AAEF,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAyfpD,CAAC"}

package/dist/contexts/useChat.js

@@ -13,7 +13,7 @@ export const useChat = () => {
     }
     return context;
 };
-export const ChatProvider = ({ children, bypassPermissions, pluginDirs, }) => {
+export const ChatProvider = ({ children, bypassPermissions, pluginDirs, tools, }) => {
     const { restoreSessionId, continueLastSession } = useAppConfig();
     // Message Display State
     const [isExpanded, setIsExpanded] = useState(false);
@@ -24,6 +24,7 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, }) => {
     const [isTaskListVisible, setIsTaskListVisible] = useState(true);
     // AI State
     const [messages, setMessages] = useState([]);
+    const messagesUpdateTimerRef = useRef(null);
     const [isLoading, setIsLoading] = useState(false);
     const [latestTotalTokens, setlatestTotalTokens] = useState(0);
     const [sessionId, setSessionId] = useState("");
@@ -71,9 +72,16 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, }) => {
     useEffect(() => {
         const initializeAgent = async () => {
             const callbacks = {
-                onMessagesChange: (newMessages) => {
+                onMessagesChange: () => {
                     if (!isExpandedRef.current) {
-                        setMessages([...newMessages]);
+                        if (!messagesUpdateTimerRef.current) {
+                            messagesUpdateTimerRef.current = setTimeout(() => {
+                                if (agentRef.current) {
+                                    setMessages([...agentRef.current.messages]);
+                                }
+                                messagesUpdateTimerRef.current = null;
+                            }, 50);
+                        }
                     }
                 },
                 onServersChange: (servers) => {
@@ -139,6 +147,7 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, }) => {
                     canUseTool: permissionCallback,
                     stream: false, // 关闭流式模式
                     plugins: pluginDirs?.map((path) => ({ type: "local", path })),
+                    tools,
                 });
                 agentRef.current = agent;
                 // Get initial state
@@ -167,10 +176,14 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, }) => {
         bypassPermissions,
         showConfirmation,
         pluginDirs,
+        tools,
     ]);
     // Cleanup on unmount
     useEffect(() => {
         return () => {
+            if (messagesUpdateTimerRef.current) {
+                clearTimeout(messagesUpdateTimerRef.current);
+            }
             if (agentRef.current) {
                 try {
                     // Display usage summary before cleanup
@@ -330,6 +343,10 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, }) => {
             setIsExpanded((prev) => {
                 const newExpanded = !prev;
                 if (newExpanded) {
+                    if (messagesUpdateTimerRef.current) {
+                        clearTimeout(messagesUpdateTimerRef.current);
+                        messagesUpdateTimerRef.current = null;
+                    }
                     // Transitioning to EXPANDED: Freeze the current view
                     // Deep copy the last message to ensure it doesn't update if the agent is still writing to it
                     setMessages((currentMessages) => {

package/dist/hooks/usePluginManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"usePluginManager.d.ts","sourceRoot":"","sources":["../../src/hooks/usePluginManager.ts"],"names":[],"mappings":"AAUA,OAAO,EAGL,wBAAwB,EACzB,MAAM,qCAAqC,CAAC;AAE7C,wBAAgB,gBAAgB,IAAI,wBAAwB,CA6R3D"}
+{"version":3,"file":"usePluginManager.d.ts","sourceRoot":"","sources":["../../src/hooks/usePluginManager.ts"],"names":[],"mappings":"AAOA,OAAO,EAGL,wBAAwB,EACzB,MAAM,qCAAqC,CAAC;AAE7C,wBAAgB,gBAAgB,IAAI,wBAAwB,CA4P3D"}