snow-ai 0.3.6 → 0.3.8

package/dist/ui/pages/ChatScreen.js

@@ -8,6 +8,7 @@ import PendingMessages from '../components/PendingMessages.js';
 import MCPInfoScreen from '../components/MCPInfoScreen.js';
 import MCPInfoPanel from '../components/MCPInfoPanel.js';
 import SessionListPanel from '../components/SessionListPanel.js';
+import UsagePanel from '../components/UsagePanel.js';
 import MarkdownRenderer from '../components/MarkdownRenderer.js';
 import ToolConfirmation from '../components/ToolConfirmation.js';
 import DiffViewer from '../components/DiffViewer.js';
@@ -38,6 +39,11 @@ import '../../utils/commands/yolo.js';
 import '../../utils/commands/init.js';
 import '../../utils/commands/ide.js';
 import '../../utils/commands/compact.js';
+import '../../utils/commands/home.js';
+import '../../utils/commands/review.js';
+import '../../utils/commands/role.js';
+import '../../utils/commands/usage.js';
+import '../../utils/commands/export.js';
 export default function ChatScreen({ skipWelcome }) {
     const [messages, setMessages] = useState([]);
     const [isSaving] = useState(false);
@@ -62,7 +68,9 @@ export default function ChatScreen({ skipWelcome }) {
     const [compressionError, setCompressionError] = useState(null);
     const [showSessionPanel, setShowSessionPanel] = useState(false);
     const [showMcpPanel, setShowMcpPanel] = useState(false);
+    const [showUsagePanel, setShowUsagePanel] = useState(false);
     const [shouldIncludeSystemInfo, setShouldIncludeSystemInfo] = useState(true); // Include on first message
+    const [restoreInputContent, setRestoreInputContent] = useState(null);
     const { columns: terminalWidth, rows: terminalHeight } = useTerminalSize();
     const { stdout } = useStdout();
     const workingDirectory = process.cwd();
@@ -86,6 +94,17 @@ export default function ChatScreen({ skipWelcome }) {
             // Ignore localStorage errors
         }
     }, [yoloMode]);
+    // Clear restore input content after it's been used
+    useEffect(() => {
+        if (restoreInputContent !== null) {
+            // Clear after a short delay to ensure ChatInput has processed it
+            const timer = setTimeout(() => {
+                setRestoreInputContent(null);
+            }, 100);
+            return () => clearTimeout(timer);
+        }
+        return undefined;
+    }, [restoreInputContent]);
     // Auto-resume last session when skipWelcome is true
     useEffect(() => {
         if (!skipWelcome)
@@ -161,6 +180,7 @@ export default function ChatScreen({ skipWelcome }) {
         setShowSessionPanel,
         setShowMcpInfo,
         setShowMcpPanel,
+        setShowUsagePanel,
         setMcpPanelKey,
         setYoloMode,
         setContextUsage: streamingState.setContextUsage,
@@ -224,6 +244,12 @@ export default function ChatScreen({ skipWelcome }) {
             }
             return;
         }
+        if (showUsagePanel) {
+            if (key.escape) {
+                setShowUsagePanel(false);
+            }
+            return;
+        }
         if (showMcpInfo) {
             if (key.escape) {
                 setShowMcpInfo(false);
@@ -235,6 +261,8 @@ export default function ChatScreen({ skipWelcome }) {
             streamingState.abortController) {
             // Abort the controller
             streamingState.abortController.abort();
+            // Clear retry status immediately when user cancels
+            streamingState.setRetryStatus(null);
             // Remove all pending tool call messages (those with toolPending: true)
             setMessages(prev => prev.filter(msg => !msg.toolPending));
             // Add discontinued message
@@ -253,7 +281,7 @@ export default function ChatScreen({ skipWelcome }) {
             streamingState.setStreamTokenCount(0);
         }
     });
-    const handleHistorySelect = async (selectedIndex, _message) => {
+    const handleHistorySelect = async (selectedIndex, message) => {
         // Count total files that will be rolled back (from selectedIndex onwards)
         let totalFileCount = 0;
         for (const [index, count] of snapshotState.snapshotFileCount.entries()) {
@@ -272,39 +300,95 @@ export default function ChatScreen({ skipWelcome }) {
                 messageIndex: selectedIndex,
                 fileCount: filePaths.length, // Use actual unique file count
                 filePaths,
+                message, // Save message for restore after rollback
             });
         }
         else {
             // No files to rollback, just rollback conversation
-            performRollback(selectedIndex, false);
+            // Note: message is already in input buffer via useHistoryNavigation
+            await performRollback(selectedIndex, false);
         }
     };
     const performRollback = async (selectedIndex, rollbackFiles) => {
+        const currentSession = sessionManager.getCurrentSession();
         // Rollback workspace to checkpoint if requested
-        if (rollbackFiles) {
-            const currentSession = sessionManager.getCurrentSession();
-            if (currentSession) {
-                // Use rollbackToMessageIndex to rollback all snapshots >= selectedIndex
-                await incrementalSnapshotManager.rollbackToMessageIndex(currentSession.id, selectedIndex);
+        if (rollbackFiles && currentSession) {
+            // Use rollbackToMessageIndex to rollback all snapshots >= selectedIndex
+            await incrementalSnapshotManager.rollbackToMessageIndex(currentSession.id, selectedIndex);
+        }
+        // For session file: find the correct truncation point based on session messages
+        // We need to truncate to the same user message in the session file
+        if (currentSession) {
+            // Count how many user messages we're deleting (from selectedIndex onwards in UI)
+            const uiUserMessagesToDelete = messages
+                .slice(selectedIndex)
+                .filter(msg => msg.role === 'user').length;
+            // Find the corresponding user message in session to delete
+            // We start from the end and count backwards
+            let sessionUserMessageCount = 0;
+            let sessionTruncateIndex = currentSession.messages.length;
+            for (let i = currentSession.messages.length - 1; i >= 0; i--) {
+                const msg = currentSession.messages[i];
+                if (msg && msg.role === 'user') {
+                    sessionUserMessageCount++;
+                    if (sessionUserMessageCount === uiUserMessagesToDelete) {
+                        // We want to delete from this user message onwards
+                        sessionTruncateIndex = i;
+                        break;
+                    }
+                }
+            }
+            // Special case: rolling back to index 0 means deleting the entire session
+            if (sessionTruncateIndex === 0 && currentSession) {
+                // Delete all snapshots for this session
+                await incrementalSnapshotManager.clearAllSnapshots(currentSession.id);
+                // Delete the session file
+                await sessionManager.deleteSession(currentSession.id);
+                // Clear current session
+                sessionManager.clearCurrentSession();
+                // Clear all messages
+                setMessages([]);
+                // Clear saved messages
+                clearSavedMessages();
+                // Clear snapshot state
+                snapshotState.setSnapshotFileCount(new Map());
+                // Clear pending rollback dialog
+                snapshotState.setPendingRollback(null);
+                // Trigger remount
+                setRemountKey(prev => prev + 1);
+                return;
+            }
+            // Delete snapshot files >= selectedIndex (regardless of whether files were rolled back)
+            await incrementalSnapshotManager.deleteSnapshotsFromIndex(currentSession.id, selectedIndex);
+            // Reload snapshot file counts from disk after deletion
+            const snapshots = await incrementalSnapshotManager.listSnapshots(currentSession.id);
+            const counts = new Map();
+            for (const snapshot of snapshots) {
+                counts.set(snapshot.messageIndex, snapshot.fileCount);
             }
+            snapshotState.setSnapshotFileCount(counts);
+            // Truncate session messages
+            await sessionManager.truncateMessages(sessionTruncateIndex);
         }
-        // Truncate messages array to remove the selected user message and everything after it
+        // Truncate UI messages array to remove the selected user message and everything after it
         setMessages(prev => prev.slice(0, selectedIndex));
-        // Truncate session messages to match the UI state
-        await sessionManager.truncateMessages(selectedIndex);
         clearSavedMessages();
         setRemountKey(prev => prev + 1);
         // Clear pending rollback dialog
         snapshotState.setPendingRollback(null);
     };
-    const handleRollbackConfirm = (rollbackFiles) => {
+    const handleRollbackConfirm = async (rollbackFiles) => {
         if (rollbackFiles === null) {
             // User cancelled - just close the dialog without doing anything
             snapshotState.setPendingRollback(null);
             return;
         }
         if (snapshotState.pendingRollback) {
-            performRollback(snapshotState.pendingRollback.messageIndex, rollbackFiles);
+            // Restore message to input before rollback
+            if (snapshotState.pendingRollback.message) {
+                setRestoreInputContent(snapshotState.pendingRollback.message);
+            }
+            await performRollback(snapshotState.pendingRollback.messageIndex, rollbackFiles);
         }
     };
     const handleSessionPanelSelect = async (sessionId) => {
@@ -351,6 +435,8 @@ export default function ChatScreen({ skipWelcome }) {
         await processMessage(message, images);
     };
     const processMessage = async (message, images, useBasicModel, hideUserMessage) => {
+        // Clear any previous retry status when starting a new request
+        streamingState.setRetryStatus(null);
         // Parse and validate file references
         const { cleanContent, validFiles } = await parseAndValidateFileReferences(message);
         // Separate image files from regular files
@@ -418,15 +504,18 @@ export default function ChatScreen({ skipWelcome }) {
         }
         catch (error) {
             if (controller.signal.aborted) {
-                return;
+                // Don't return here - let finally block execute
+                // Just skip error display for aborted requests
+            }
+            else {
+                const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
+                const finalMessage = {
+                    role: 'assistant',
+                    content: `Error: ${errorMessage}`,
+                    streaming: false,
+                };
+                setMessages(prev => [...prev, finalMessage]);
             }
-            const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
-            const finalMessage = {
-                role: 'assistant',
-                content: `Error: ${errorMessage}`,
-                streaming: false,
-            };
-            setMessages(prev => [...prev, finalMessage]);
         }
         finally {
             // End streaming
@@ -440,6 +529,8 @@ export default function ChatScreen({ skipWelcome }) {
     const processPendingMessages = async () => {
         if (pendingMessages.length === 0)
             return;
+        // Clear any previous retry status when starting a new request
+        streamingState.setRetryStatus(null);
         // Get current pending messages and clear them immediately
         const messagesToProcess = [...pendingMessages];
         setPendingMessages([]);
@@ -500,15 +591,18 @@ export default function ChatScreen({ skipWelcome }) {
         }
         catch (error) {
             if (controller.signal.aborted) {
-                return;
+                // Don't return here - let finally block execute
+                // Just skip error display for aborted requests
+            }
+            else {
+                const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
+                const finalMessage = {
+                    role: 'assistant',
+                    content: `Error: ${errorMessage}`,
+                    streaming: false,
+                };
+                setMessages(prev => [...prev, finalMessage]);
             }
-            const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
-            const finalMessage = {
-                role: 'assistant',
-                content: `Error: ${errorMessage}`,
-                streaming: false,
-            };
-            setMessages(prev => [...prev, finalMessage]);
         }
         finally {
             // End streaming
@@ -553,10 +647,11 @@ export default function ChatScreen({ skipWelcome }) {
                                 workingDirectory)))),
                 ...messages
                     .filter(m => !m.streaming)
-                    .map((message, index) => {
+                    .map((message, index, filteredMessages) => {
                     // Determine tool message type and color
                     let toolStatusColor = 'cyan';
                     let isToolMessage = false;
+                    const isLastMessage = index === filteredMessages.length - 1;
                     if (message.role === 'assistant') {
                         if (message.content.startsWith('⚡')) {
                             isToolMessage = true;
@@ -574,7 +669,7 @@ export default function ChatScreen({ skipWelcome }) {
                             toolStatusColor = 'blue';
                         }
                     }
-                    return (React.createElement(Box, { key: `msg-${index}`, marginBottom: isToolMessage ? 0 : 1, paddingX: 1, flexDirection: "column", width: terminalWidth },
+                    return (React.createElement(Box, { key: `msg-${index}`, marginTop: index > 0 ? 1 : 0, marginBottom: isLastMessage ? 1 : 0, paddingX: 1, flexDirection: "column", width: terminalWidth },
                         React.createElement(Box, null,
                             React.createElement(Text, { color: message.role === 'user'
                                     ? 'green'
@@ -585,18 +680,18 @@ export default function ChatScreen({ skipWelcome }) {
                                 : message.role === 'command'
                                     ? '⌘'
                                     : '❆'),
-                            React.createElement(Box, { marginLeft: 1, marginBottom: 1, flexDirection: "column" }, message.role === 'command' ? (React.createElement(Text, { color: "gray", dimColor: true },
-                                "\u2514\u2500 ",
-                                message.commandName)) : message.showTodoTree ? (React.createElement(TodoTree, { todos: currentTodos })) : (React.createElement(React.Fragment, null,
-                                React.createElement(MarkdownRenderer, { content: message.content || ' ', color: message.role === 'user'
+                            React.createElement(Box, { marginLeft: 1, flexDirection: "column" }, message.role === 'command' ? (React.createElement(React.Fragment, null,
+                                React.createElement(Text, { color: "gray", dimColor: true },
+                                    "\u2514\u2500 ",
+                                    message.commandName),
+                                message.content && (React.createElement(Text, { color: "white" }, message.content)))) : message.showTodoTree ? (React.createElement(TodoTree, { todos: currentTodos })) : (React.createElement(React.Fragment, null,
+                                message.role === 'user' || isToolMessage ? (React.createElement(Text, { color: message.role === 'user'
                                         ? 'gray'
-                                        : isToolMessage
-                                            ? message.content.startsWith('⚡')
-                                                ? 'yellow'
-                                                : message.content.startsWith('✓')
-                                                    ? 'green'
-                                                    : 'red'
-                                            : undefined }),
+                                        : message.content.startsWith('⚡')
+                                            ? 'yellow'
+                                            : message.content.startsWith('✓')
+                                                ? 'green'
+                                                : 'red' }, message.content || ' ')) : (React.createElement(MarkdownRenderer, { content: message.content || ' ' })),
                                 message.toolDisplay &&
                                     message.toolDisplay.args.length > 0 && (React.createElement(Box, { flexDirection: "column" }, message.toolDisplay.args.map((arg, argIndex) => (React.createElement(Text, { key: argIndex, color: "gray", dimColor: true },
                                     arg.isLast ? '└─' : '├─',
@@ -606,8 +701,7 @@ export default function ChatScreen({ skipWelcome }) {
                                     ' ',
                                     arg.value))))),
                                 message.toolCall &&
-                                    (message.toolCall.name === 'filesystem-create' ||
-                                        message.toolCall.name === 'filesystem-write') &&
+                                    message.toolCall.name === 'filesystem-create' &&
                                     message.toolCall.arguments.content && (React.createElement(Box, { marginTop: 1 },
                                     React.createElement(DiffViewer, { newContent: message.toolCall.arguments.content, filename: message.toolCall.arguments.path }))),
                                 message.toolCall &&
@@ -626,47 +720,27 @@ export default function ChatScreen({ skipWelcome }) {
                                             .completeOldContent, completeNewContent: message.toolCall.arguments
                                             .completeNewContent, startLineNumber: message.toolCall.arguments.contextStartLine }))),
                                 message.toolCall &&
-                                    message.toolCall.name === 'terminal-execute' &&
-                                    message.toolCall.arguments.command && (React.createElement(Box, { marginTop: 1, flexDirection: "column" },
-                                    React.createElement(Text, { color: "gray", dimColor: true },
-                                        "\u2514\u2500 Command:",
-                                        ' ',
-                                        React.createElement(Text, { color: "white" }, message.toolCall.arguments.command)),
-                                    React.createElement(Text, { color: "gray", dimColor: true },
-                                        "\u2514\u2500 Exit Code:",
-                                        ' ',
-                                        React.createElement(Text, { color: message.toolCall.arguments.exitCode === 0
-                                                ? 'green'
-                                                : 'red' }, message.toolCall.arguments.exitCode)),
-                                    message.toolCall.arguments.stdout &&
-                                        message.toolCall.arguments.stdout.trim()
-                                            .length > 0 && (React.createElement(Box, { flexDirection: "column", marginTop: 1 },
-                                        React.createElement(Text, { color: "green", dimColor: true }, "\u2514\u2500 stdout:"),
-                                        React.createElement(Box, { paddingLeft: 2 },
-                                            React.createElement(Text, { color: "white" }, message.toolCall.arguments.stdout
-                                                .trim()
-                                                .split('\n')
-                                                .slice(0, 20)
-                                                .join('\n')),
-                                            message.toolCall.arguments.stdout
-                                                .trim()
-                                                .split('\n').length > 20 && (React.createElement(Text, { color: "gray", dimColor: true }, "... (output truncated)"))))),
-                                    message.toolCall.arguments.stderr &&
-                                        message.toolCall.arguments.stderr.trim()
-                                            .length > 0 && (React.createElement(Box, { flexDirection: "column", marginTop: 1 },
-                                        React.createElement(Text, { color: "red", dimColor: true }, "\u2514\u2500 stderr:"),
-                                        React.createElement(Box, { paddingLeft: 2 },
-                                            React.createElement(Text, { color: "red" }, message.toolCall.arguments.stderr
-                                                .trim()
-                                                .split('\n')
-                                                .slice(0, 10)
-                                                .join('\n')),
-                                            message.toolCall.arguments.stderr
-                                                .trim()
-                                                .split('\n').length > 10 && (React.createElement(Text, { color: "gray", dimColor: true }, "... (output truncated)"))))))),
+                                    (message.toolCall.name === 'filesystem-edit' ||
+                                        message.toolCall.name ===
+                                            'filesystem-edit_search') &&
+                                    message.toolCall.arguments.isBatch &&
+                                    message.toolCall.arguments.batchResults &&
+                                    Array.isArray(message.toolCall.arguments.batchResults) && (React.createElement(Box, { marginTop: 1, flexDirection: "column" }, message.toolCall.arguments.batchResults.map((fileResult, index) => {
+                                    if (fileResult.success &&
+                                        fileResult.oldContent &&
+                                        fileResult.newContent) {
+                                        return (React.createElement(Box, { key: index, flexDirection: "column", marginBottom: 1 },
+                                            React.createElement(Text, { bold: true, color: "cyan" }, `File ${index + 1}: ${fileResult.path}`),
+                                            React.createElement(DiffViewer, { oldContent: fileResult.oldContent, newContent: fileResult.newContent, filename: fileResult.path, completeOldContent: fileResult.completeOldContent, completeNewContent: fileResult.completeNewContent, startLineNumber: fileResult.contextStartLine })));
+                                    }
+                                    return null;
+                                }))),
                                 message.content.startsWith('✓') &&
                                     message.toolResult &&
-                                    !message.toolCall && (React.createElement(ToolResultPreview, { toolName: message.content
+                                    // 只在没有 diff 数据时显示预览（有 diff 的工具会用 DiffViewer 显示）
+                                    !(message.toolCall &&
+                                        (message.toolCall.arguments?.oldContent ||
+                                            message.toolCall.arguments?.batchResults)) && (React.createElement(ToolResultPreview, { toolName: message.content
                                         .replace('✓ ', '')
                                         .split('\n')[0] || '', result: message.toolResult, maxLines: 5 })),
                                 message.role === 'user' && message.systemInfo && (React.createElement(Box, { marginTop: 1, flexDirection: "column" },
@@ -747,13 +821,18 @@ export default function ChatScreen({ skipWelcome }) {
             React.createElement(MCPInfoPanel, null),
             React.createElement(Box, { marginTop: 1 },
                 React.createElement(Text, { color: "gray", dimColor: true }, "Press ESC to close")))),
+        showUsagePanel && (React.createElement(Box, { paddingX: 1, flexDirection: "column", width: terminalWidth },
+            React.createElement(UsagePanel, null),
+            React.createElement(Box, { marginTop: 1 },
+                React.createElement(Text, { color: "gray", dimColor: true }, "Press ESC to close")))),
         snapshotState.pendingRollback && (React.createElement(FileRollbackConfirmation, { fileCount: snapshotState.pendingRollback.fileCount, filePaths: snapshotState.pendingRollback.filePaths || [], onConfirm: handleRollbackConfirm })),
         !pendingToolConfirmation &&
             !isCompressing &&
             !showSessionPanel &&
             !showMcpPanel &&
+            !showUsagePanel &&
             !snapshotState.pendingRollback && (React.createElement(React.Fragment, null,
-            React.createElement(ChatInput, { onSubmit: handleMessageSubmit, onCommand: handleCommandExecution, placeholder: "Ask me anything about coding...", disabled: !!pendingToolConfirmation, chatHistory: messages, onHistorySelect: handleHistorySelect, yoloMode: yoloMode, contextUsage: streamingState.contextUsage
+            React.createElement(ChatInput, { onSubmit: handleMessageSubmit, onCommand: handleCommandExecution, placeholder: "Ask me anything about coding...", disabled: !!pendingToolConfirmation, isProcessing: streamingState.isStreaming || isSaving, chatHistory: messages, onHistorySelect: handleHistorySelect, yoloMode: yoloMode, contextUsage: streamingState.contextUsage
                     ? {
                         inputTokens: streamingState.contextUsage.prompt_tokens,
                         maxContextTokens: getOpenAiConfig().maxContextTokens || 4000,
@@ -761,8 +840,8 @@ export default function ChatScreen({ skipWelcome }) {
                         cacheReadTokens: streamingState.contextUsage.cache_read_input_tokens,
                         cachedTokens: streamingState.contextUsage.cached_tokens,
                     }
-                    : undefined }),
-            vscodeState.vscodeConnectionStatus !== 'disconnected' && (React.createElement(Box, { marginTop: 1 },
+                    : undefined, initialContent: restoreInputContent }),
+            vscodeState.vscodeConnectionStatus !== 'disconnected' && (React.createElement(Box, { marginTop: 1, paddingX: 1 },
                 React.createElement(Text, { color: vscodeState.vscodeConnectionStatus === 'connecting'
                         ? 'yellow'
                         : vscodeState.vscodeConnectionStatus === 'connected'

package/dist/ui/pages/ConfigScreen.js

@@ -294,25 +294,6 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
         if (isFocusEventInput(rawInput)) {
             return;
         }
-        // Handle profile shortcuts (highest priority - works even in Select mode)
-        if (currentField === 'profile' && (input === 'n' || input === 'N')) {
-            // Handle profile creation (works in both normal and editing mode)
-            setProfileMode('creating');
-            setNewProfileName('');
-            setIsEditing(false); // Exit Select editing mode
-            return;
-        }
-        if (currentField === 'profile' && (input === 'd' || input === 'D')) {
-            // Handle profile deletion (works in both normal and editing mode)
-            if (activeProfile === 'default') {
-                setErrors(['Cannot delete the default profile']);
-                setIsEditing(false);
-                return;
-            }
-            setProfileMode('deleting');
-            setIsEditing(false); // Exit Select editing mode
-            return;
-        }
         // Handle profile creation mode
         if (profileMode === 'creating') {
             if (key.return) {
@@ -336,6 +317,29 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
             }
             return;
         }
+        // Handle profile shortcuts (only when in normal profile mode)
+        if (profileMode === 'normal' &&
+            currentField === 'profile' &&
+            (input === 'n' || input === 'N')) {
+            // Handle profile creation (works in both normal and editing mode)
+            setProfileMode('creating');
+            setNewProfileName('');
+            setIsEditing(false); // Exit Select editing mode
+            return;
+        }
+        if (profileMode === 'normal' &&
+            currentField === 'profile' &&
+            (input === 'd' || input === 'D')) {
+            // Handle profile deletion (works in both normal and editing mode)
+            if (activeProfile === 'default') {
+                setErrors(['Cannot delete the default profile']);
+                setIsEditing(false);
+                return;
+            }
+            setProfileMode('deleting');
+            setIsEditing(false); // Exit Select editing mode
+            return;
+        }
         // Handle loading state
         if (loading) {
             if (key.escape) {

package/dist/ui/pages/HeadlessModeScreen.js

@@ -6,6 +6,7 @@ import { handleConversationWithTools } from '../../hooks/useConversation.js';
 import { useStreamingState } from '../../hooks/useStreamingState.js';
 import { useToolConfirmation } from '../../hooks/useToolConfirmation.js';
 import { useVSCodeState } from '../../hooks/useVSCodeState.js';
+import { useSessionSave } from '../../hooks/useSessionSave.js';
 import { parseAndValidateFileReferences, createMessageWithFileInstructions, getSystemInfo, } from '../../utils/fileUtils.js';
 // Console-based markdown renderer functions
 function renderConsoleMarkdown(content) {
@@ -202,6 +203,7 @@ export default function HeadlessModeScreen({ prompt, onComplete }) {
     // Use custom hooks
     const streamingState = useStreamingState();
     const vscodeState = useVSCodeState();
+    const { saveMessage } = useSessionSave();
     // Use tool confirmation hook
     const { requestToolConfirmation, isToolAutoApproved, addMultipleToAlwaysApproved, } = useToolConfirmation();
     // Listen for message changes to display AI responses and tool calls
@@ -335,10 +337,6 @@ export default function HeadlessModeScreen({ prompt, onComplete }) {
             console.log(`\x1b[36m└─ Assistant Response\x1b[0m`);
             // Create message for AI
             const messageForAI = createMessageWithFileInstructions(cleanContent, regularFiles, systemInfo, vscodeState.vscodeConnected ? vscodeState.editorContext : undefined);
-            // Custom save message function for headless mode
-            const saveMessage = async () => {
-                // In headless mode, we don't need to save messages
-            };
             // Start conversation with tool support
             await handleConversationWithTools({
                 userContent: messageForAI,

package/dist/ui/pages/SubAgentConfigScreen.js

@@ -8,30 +8,30 @@ const toolCategories = [
     {
         name: 'Filesystem Tools',
         tools: [
-            'filesystem_read',
-            'filesystem_create',
-            'filesystem_edit',
-            'filesystem_edit_search',
-            'filesystem_delete',
-            'filesystem_list',
+            'filesystem-read',
+            'filesystem-create',
+            'filesystem-edit',
+            'filesystem-edit_search',
+            'filesystem-delete',
+            'filesystem-list',
         ],
     },
     {
         name: 'ACE Code Search Tools',
         tools: [
-            'ace_search_symbols',
-            'ace_find_definition',
-            'ace_find_references',
-            'ace_semantic_search',
-            'ace_text_search',
-            'ace_file_outline',
-            'ace_index_stats',
-            'ace_clear_cache',
+            'ace-search_symbols',
+            'ace-find_definition',
+            'ace-find_references',
+            'ace-semantic_search',
+            'ace-text_search',
+            'ace-file_outline',
+            'ace-index_stats',
+            'ace-clear_cache',
         ],
     },
     {
         name: 'Terminal Tools',
-        tools: ['terminal_execute'],
+        tools: ['terminal-execute'],
     },
     {
         name: 'TODO Management Tools',
@@ -45,11 +45,11 @@ const toolCategories = [
     },
     {
         name: 'Web Search Tools',
-        tools: ['websearch_search', 'websearch_fetch'],
+        tools: ['websearch-search', 'websearch-fetch'],
     },
     {
         name: 'IDE Diagnostics Tools',
-        tools: ['ide_get_diagnostics'],
+        tools: ['ide-get_diagnostics'],
     },
 ];
 export default function SubAgentConfigScreen({ onBack, onSave, inlineMode = false, agentId, }) {

package/dist/ui/pages/SystemPromptConfigScreen.js

@@ -4,7 +4,7 @@ import { spawn } from 'child_process';
 import { writeFileSync, readFileSync, existsSync, mkdirSync } from 'fs';
 import { join } from 'path';
 import { homedir, platform } from 'os';
-import { SYSTEM_PROMPT } from '../../api/systemPrompt.js';
+import { getSystemPrompt } from '../../api/systemPrompt.js';
 const CONFIG_DIR = join(homedir(), '.snow');
 const SYSTEM_PROMPT_FILE = join(CONFIG_DIR, 'system-prompt.txt');
 function getSystemEditor() {
@@ -24,14 +24,14 @@ export default function SystemPromptConfigScreen({ onBack }) {
         const openEditor = async () => {
             ensureConfigDirectory();
             // 读取系统提示词文件，如果不存在则使用默认系统提示词
-            let currentPrompt = SYSTEM_PROMPT;
+            let currentPrompt = getSystemPrompt();
             if (existsSync(SYSTEM_PROMPT_FILE)) {
                 try {
                     currentPrompt = readFileSync(SYSTEM_PROMPT_FILE, 'utf8');
                 }
                 catch {
                     // 读取失败，使用默认
-                    currentPrompt = SYSTEM_PROMPT;
+                    currentPrompt = getSystemPrompt();
                 }
             }
             // 写入临时文件供编辑
@@ -39,7 +39,7 @@ export default function SystemPromptConfigScreen({ onBack }) {
             const editor = getSystemEditor();
             exit();
             const child = spawn(editor, [SYSTEM_PROMPT_FILE], {
-                stdio: 'inherit'
+                stdio: 'inherit',
             });
             child.on('close', () => {
                 // 读取编辑后的内容
@@ -47,7 +47,8 @@ export default function SystemPromptConfigScreen({ onBack }) {
                     try {
                         const editedContent = readFileSync(SYSTEM_PROMPT_FILE, 'utf8');
                         const trimmedContent = editedContent.trim();
-                        if (trimmedContent === '' || trimmedContent === SYSTEM_PROMPT.trim()) {
+                        if (trimmedContent === '' ||
+                            trimmedContent === getSystemPrompt().trim()) {
                             // 内容为空或与默认相同，删除文件，使用默认提示词
                             try {
                                 const fs = require('fs');
@@ -72,7 +73,7 @@ export default function SystemPromptConfigScreen({ onBack }) {
                 }
                 process.exit(0);
             });
-            child.on('error', (error) => {
+            child.on('error', error => {
                 console.error('Failed to open editor:', error.message);
                 process.exit(1);
             });

package/dist/utils/chatExporter.d.ts

@@ -0,0 +1,9 @@
+import type { Message } from '../ui/components/MessageList.js';
+/**
+ * Format messages to plain text for export
+ */
+export declare function formatMessagesAsText(messages: Message[]): string;
+/**
+ * Export messages to a file
+ */
+export declare function exportMessagesToFile(messages: Message[], filePath: string): Promise<void>;