snow-ai 0.3.4 → 0.3.5

package/dist/ui/components/FileRollbackConfirmation.d.ts

@@ -1,7 +1,8 @@
 import React from 'react';
 type Props = {
     fileCount: number;
-    onConfirm: (rollbackFiles: boolean) => void;
+    filePaths: string[];
+    onConfirm: (rollbackFiles: boolean | null) => void;
 };
-export default function FileRollbackConfirmation({ fileCount, onConfirm }: Props): React.JSX.Element;
+export default function FileRollbackConfirmation({ fileCount, filePaths, onConfirm }: Props): React.JSX.Element;
 export {};

package/dist/ui/components/FileRollbackConfirmation.js

@@ -1,34 +1,71 @@
 import React, { useState } from 'react';
 import { Box, Text, useInput } from 'ink';
-export default function FileRollbackConfirmation({ fileCount, onConfirm }) {
+export default function FileRollbackConfirmation({ fileCount, filePaths, onConfirm }) {
     const [selectedIndex, setSelectedIndex] = useState(0);
+    const [showFullList, setShowFullList] = useState(false);
+    const [fileScrollIndex, setFileScrollIndex] = useState(0);
     const options = [
         { label: 'Yes, rollback files and conversation', value: true },
         { label: 'No, rollback conversation only', value: false }
     ];
     useInput((_, key) => {
-        // Up arrow
-        if (key.upArrow) {
-            setSelectedIndex(prev => Math.max(0, prev - 1));
+        // Tab - toggle full file list view
+        if (key.tab) {
+            setShowFullList(prev => !prev);
+            setFileScrollIndex(0); // Reset scroll when toggling
             return;
         }
-        // Down arrow
-        if (key.downArrow) {
-            setSelectedIndex(prev => Math.min(options.length - 1, prev + 1));
-            return;
+        // In full list mode, use up/down to scroll files
+        if (showFullList) {
+            const maxVisibleFiles = 10;
+            const maxScroll = Math.max(0, filePaths.length - maxVisibleFiles);
+            if (key.upArrow) {
+                setFileScrollIndex(prev => Math.max(0, prev - 1));
+                return;
+            }
+            if (key.downArrow) {
+                setFileScrollIndex(prev => Math.min(maxScroll, prev + 1));
+                return;
+            }
+        }
+        else {
+            // In compact mode, up/down navigate options
+            if (key.upArrow) {
+                setSelectedIndex(prev => Math.max(0, prev - 1));
+                return;
+            }
+            if (key.downArrow) {
+                setSelectedIndex(prev => Math.min(options.length - 1, prev + 1));
+                return;
+            }
         }
-        // Enter - confirm selection
-        if (key.return) {
+        // Enter - confirm selection (only when not in full list mode)
+        if (key.return && !showFullList) {
             onConfirm(options[selectedIndex]?.value ?? false);
             return;
         }
-        // ESC - cancel rollback (select "No")
+        // ESC - exit full list mode or cancel rollback
         if (key.escape) {
-            onConfirm(false);
+            if (showFullList) {
+                setShowFullList(false);
+                setFileScrollIndex(0);
+            }
+            else {
+                onConfirm(null); // null means cancel everything
+            }
             return;
         }
     });
-    return (React.createElement(Box, { flexDirection: "column", marginX: 1, marginBottom: 1, borderStyle: "round", borderColor: "yellow", padding: 1 },
+    // Display logic for file list
+    const maxFilesToShowCompact = 5;
+    const maxFilesToShowFull = 10;
+    const displayFiles = showFullList
+        ? filePaths.slice(fileScrollIndex, fileScrollIndex + maxFilesToShowFull)
+        : filePaths.slice(0, maxFilesToShowCompact);
+    const remainingCountCompact = fileCount - maxFilesToShowCompact;
+    const hasMoreAbove = showFullList && fileScrollIndex > 0;
+    const hasMoreBelow = showFullList && (fileScrollIndex + maxFilesToShowFull) < filePaths.length;
+    return (React.createElement(Box, { flexDirection: "column", marginX: 1, marginBottom: 1, padding: 1 },
         React.createElement(Box, { marginBottom: 1 },
             React.createElement(Text, { color: "yellow", bold: true }, "\u26A0  File Rollback Confirmation")),
         React.createElement(Box, { marginBottom: 1 },
@@ -37,13 +74,35 @@ export default function FileRollbackConfirmation({ fileCount, onConfirm }) {
                 fileCount,
                 " file",
                 fileCount > 1 ? 's' : '',
-                " that will be rolled back.")),
-        React.createElement(Box, { marginBottom: 1 },
-            React.createElement(Text, { color: "gray", dimColor: true }, "Do you want to rollback the files as well?")),
-        React.createElement(Box, { flexDirection: "column" }, options.map((option, index) => (React.createElement(Box, { key: index },
-            React.createElement(Text, { color: index === selectedIndex ? 'green' : 'white', bold: index === selectedIndex },
-                index === selectedIndex ? '❯  ' : '  ',
-                option.label))))),
-        React.createElement(Box, { marginTop: 1 },
-            React.createElement(Text, { color: "gray", dimColor: true }, "Use \u2191\u2193 to select, Enter to confirm, ESC to cancel"))));
+                " that will be rolled back:")),
+        React.createElement(Box, { flexDirection: "column", marginBottom: 1, marginLeft: 2 },
+            hasMoreAbove && (React.createElement(Text, { color: "gray", dimColor: true },
+                "\u2191 ",
+                fileScrollIndex,
+                " more above...")),
+            displayFiles.map((file, index) => (React.createElement(Text, { key: index, color: "cyan", dimColor: true },
+                "\u2022 ",
+                file))),
+            hasMoreBelow && (React.createElement(Text, { color: "gray", dimColor: true },
+                "\u2193 ",
+                filePaths.length - (fileScrollIndex + maxFilesToShowFull),
+                " more below...")),
+            !showFullList && remainingCountCompact > 0 && (React.createElement(Text, { color: "gray", dimColor: true },
+                "... and ",
+                remainingCountCompact,
+                " more file",
+                remainingCountCompact > 1 ? 's' : ''))),
+        !showFullList && (React.createElement(React.Fragment, null,
+            React.createElement(Box, { marginBottom: 1 },
+                React.createElement(Text, { color: "gray", dimColor: true }, "Do you want to rollback the files as well?")),
+            React.createElement(Box, { flexDirection: "column", marginBottom: 1 }, options.map((option, index) => (React.createElement(Box, { key: index },
+                React.createElement(Text, { color: index === selectedIndex ? 'green' : 'white', bold: index === selectedIndex },
+                    index === selectedIndex ? '❯  ' : '  ',
+                    option.label))))))),
+        React.createElement(Box, null,
+            React.createElement(Text, { color: "gray", dimColor: true }, showFullList
+                ? '↑↓ scroll · Tab back · ESC close'
+                : fileCount > maxFilesToShowCompact
+                    ? `↑↓ select · Tab view all (${fileCount} files) · Enter confirm · ESC cancel`
+                    : '↑↓ select · Enter confirm · ESC cancel'))));
 }

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

@@ -1,7 +1,7 @@
 import React from 'react';
 import { SelectedFile } from '../../utils/fileUtils.js';
 export interface Message {
-    role: 'user' | 'assistant' | 'command';
+    role: 'user' | 'assistant' | 'command' | 'subagent';
     content: string;
     streaming?: boolean;
     discontinued?: boolean;
@@ -39,6 +39,12 @@ export interface Message {
         exitCode?: number;
         command?: string;
     };
+    subAgent?: {
+        agentId: string;
+        agentName: string;
+        isComplete?: boolean;
+    };
+    subAgentInternal?: boolean;
 }
 interface Props {
     messages: Message[];

package/dist/ui/components/MessageList.js

@@ -11,20 +11,31 @@ const MessageList = memo(({ messages, animationFrame, maxMessages = 6 }) => {
             ? 'green'
             : message.role === 'command'
                 ? 'gray'
-                : message.streaming
-                    ? STREAM_COLORS[animationFrame]
-                    : 'cyan';
+                : message.role === 'subagent'
+                    ? 'magenta'
+                    : message.streaming
+                        ? STREAM_COLORS[animationFrame]
+                        : 'cyan';
         return (React.createElement(Box, { key: index },
             React.createElement(Text, { color: iconColor, bold: true }, message.role === 'user'
                 ? '⛇'
                 : message.role === 'command'
                     ? '⌘'
-                    : '❆'),
+                    : message.role === 'subagent'
+                        ? '◈'
+                        : '❆'),
             React.createElement(Box, { marginLeft: 1, marginBottom: 1, flexDirection: "column" }, message.role === 'command' ? (React.createElement(Text, { color: "gray" },
                 "\u2514\u2500 ",
-                message.commandName)) : (React.createElement(React.Fragment, null,
+                message.commandName)) : message.role === 'subagent' ? (React.createElement(React.Fragment, null,
+                React.createElement(Text, { color: "magenta", dimColor: true },
+                    "\u2514\u2500 Sub-Agent: ",
+                    message.subAgent?.agentName,
+                    message.subAgent?.isComplete ? ' ✓' : ' ...'),
+                React.createElement(Box, { marginLeft: 2 },
+                    React.createElement(MarkdownRenderer, { content: message.content || ' ', color: "gray" })))) : (React.createElement(React.Fragment, null,
                 React.createElement(MarkdownRenderer, { content: message.content || ' ', color: message.role === 'user' ? 'gray' : undefined }),
                 (message.systemInfo ||
+                    message.files ||
                     message.files ||
                     message.images) && (React.createElement(Box, { marginTop: 1, flexDirection: "column" },
                     message.systemInfo && (React.createElement(React.Fragment, null,

package/dist/ui/pages/ChatScreen.js

@@ -130,6 +130,20 @@ export default function ChatScreen({ skipWelcome }) {
             clearTimeout(handler);
         };
     }, [terminalWidth, stdout]);
+    // Reload messages from session when remountKey changes (to restore sub-agent messages)
+    useEffect(() => {
+        if (remountKey === 0)
+            return; // Skip initial render
+        const reloadMessages = async () => {
+            const currentSession = sessionManager.getCurrentSession();
+            if (currentSession && currentSession.messages.length > 0) {
+                // Convert session messages back to UI format
+                const uiMessages = convertSessionMessagesToUI(currentSession.messages);
+                setMessages(uiMessages);
+            }
+        };
+        reloadMessages();
+    }, [remountKey]);
     // Use tool confirmation hook
     const { pendingToolConfirmation, requestToolConfirmation, isToolAutoApproved, addMultipleToAlwaysApproved, } = useToolConfirmation();
     // Minimum terminal height required for proper rendering
@@ -249,9 +263,15 @@ export default function ChatScreen({ skipWelcome }) {
         }
         // Show confirmation dialog if there are files to rollback
         if (totalFileCount > 0) {
+            // Get list of files that will be rolled back
+            const currentSession = sessionManager.getCurrentSession();
+            const filePaths = currentSession
+                ? await incrementalSnapshotManager.getFilesToRollback(currentSession.id, selectedIndex)
+                : [];
             snapshotState.setPendingRollback({
                 messageIndex: selectedIndex,
-                fileCount: totalFileCount,
+                fileCount: filePaths.length, // Use actual unique file count
+                filePaths,
             });
         }
         else {
@@ -278,6 +298,11 @@ export default function ChatScreen({ skipWelcome }) {
         snapshotState.setPendingRollback(null);
     };
     const handleRollbackConfirm = (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);
         }
@@ -527,7 +552,7 @@ export default function ChatScreen({ skipWelcome }) {
                                 "\u2022 Working directory: ",
                                 workingDirectory)))),
                 ...messages
-                    .filter(m => !m.streaming && !m.toolPending)
+                    .filter(m => !m.streaming)
                     .map((message, index) => {
                     // Determine tool message type and color
                     let toolStatusColor = 'cyan';
@@ -670,22 +695,6 @@ export default function ChatScreen({ skipWelcome }) {
                                 message.discontinued && (React.createElement(Text, { color: "red", bold: true }, "\u2514\u2500 user discontinue"))))))));
                 }),
             ] }, item => item),
-        messages
-            .filter(m => m.toolPending)
-            .map((message, index) => (React.createElement(Box, { key: `pending-tool-${index}`, marginBottom: 1, paddingX: 1, width: terminalWidth },
-            React.createElement(Text, { color: "yellowBright", bold: true }, "\u2746"),
-            React.createElement(Box, { marginLeft: 1, marginBottom: 1, flexDirection: "column" },
-                React.createElement(Box, { flexDirection: "row" },
-                    React.createElement(MarkdownRenderer, { content: message.content || ' ', color: "yellow" }),
-                    React.createElement(Box, { marginLeft: 1 },
-                        React.createElement(Text, { color: "yellow" },
-                            React.createElement(Spinner, { type: "dots" })))),
-                message.toolDisplay && message.toolDisplay.args.length > 0 && (React.createElement(Box, { flexDirection: "column", marginTop: 1 }, message.toolDisplay.args.map((arg, argIndex) => (React.createElement(Text, { key: argIndex, color: "gray", dimColor: true },
-                    arg.isLast ? '└─' : '├─',
-                    " ",
-                    arg.key,
-                    ": ",
-                    arg.value))))))))),
         (streamingState.isStreaming || isSaving) && !pendingToolConfirmation && (React.createElement(Box, { marginBottom: 1, paddingX: 1, width: terminalWidth },
             React.createElement(Text, { color: ['#FF6EBF', 'green', 'blue', 'cyan', '#B588F8'][streamingState.animationFrame], bold: true }, "\u2746"),
             React.createElement(Box, { marginLeft: 1, marginBottom: 1, flexDirection: "column" }, streamingState.isStreaming ? (React.createElement(React.Fragment, null, streamingState.retryStatus &&
@@ -738,7 +747,7 @@ 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")))),
-        snapshotState.pendingRollback && (React.createElement(FileRollbackConfirmation, { fileCount: snapshotState.pendingRollback.fileCount, onConfirm: handleRollbackConfirm })),
+        snapshotState.pendingRollback && (React.createElement(FileRollbackConfirmation, { fileCount: snapshotState.pendingRollback.fileCount, filePaths: snapshotState.pendingRollback.filePaths || [], onConfirm: handleRollbackConfirm })),
         !pendingToolConfirmation &&
             !isCompressing &&
             !showSessionPanel &&
@@ -752,7 +761,7 @@ export default function ChatScreen({ skipWelcome }) {
                         cacheReadTokens: streamingState.contextUsage.cache_read_input_tokens,
                         cachedTokens: streamingState.contextUsage.cached_tokens,
                     }
-                    : undefined, snapshotFileCount: snapshotState.snapshotFileCount }),
+                    : undefined }),
             vscodeState.vscodeConnectionStatus !== 'disconnected' && (React.createElement(Box, { marginTop: 1 },
                 React.createElement(Text, { color: vscodeState.vscodeConnectionStatus === 'connecting'
                         ? 'yellow'

package/dist/ui/pages/ConfigScreen.js

@@ -3,7 +3,6 @@ import { Box, Text, useInput } from 'ink';
 import Gradient from 'ink-gradient';
 import { Select, Alert, Spinner } from '@inkjs/ui';
 import TextInput from 'ink-text-input';
-import chalk from 'chalk';
 import { getOpenAiConfig, updateOpenAiConfig, validateApiConfig, } from '../../utils/apiConfig.js';
 import { fetchAvailableModels, filterModels, } from '../../api/models.js';
 import { getActiveProfileName, getAllProfiles, switchProfile, createProfile, deleteProfile, saveProfile, } from '../../utils/configManager.js';
@@ -158,34 +157,6 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
             return compactModelName;
         return '';
     };
-    const handleProfileChange = (value) => {
-        if (value === '__CREATE_NEW__') {
-            setProfileMode('creating');
-            setNewProfileName('');
-            return;
-        }
-        if (value === '__DELETE__') {
-            if (activeProfile === 'default') {
-                setErrors(['Cannot delete the default profile']);
-                setIsEditing(false); // Exit editing mode to prevent Select component error
-                return;
-            }
-            setProfileMode('deleting');
-            return;
-        }
-        // Switch profile
-        try {
-            switchProfile(value);
-            loadProfilesAndConfig();
-            setIsEditing(false);
-            setErrors([]);
-        }
-        catch (err) {
-            setErrors([
-                err instanceof Error ? err.message : 'Failed to switch profile',
-            ]);
-        }
-    };
     const handleCreateProfile = () => {
         const cleaned = stripFocusArtifacts(newProfileName).trim();
         if (!cleaned) {
@@ -204,7 +175,9 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
                     basicModel,
                     maxContextTokens,
                     maxTokens,
-                    compactModel: compactModelName ? { modelName: compactModelName } : undefined,
+                    compactModel: compactModelName
+                        ? { modelName: compactModelName }
+                        : undefined,
                 },
             };
             createProfile(cleaned, currentConfig);
@@ -321,6 +294,25 @@ 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) {
@@ -638,20 +630,24 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
                 currentField === 'compactModelName' && 'Compact Model',
                 ":"),
             React.createElement(Box, { marginLeft: 3, marginTop: 1 },
-                currentField === 'profile' && (React.createElement(Select, { options: [
-                        ...profiles.map(p => ({
+                currentField === 'profile' && (React.createElement(Box, { flexDirection: "column" },
+                    profiles.length > 1 && (React.createElement(Text, { color: "gray", dimColor: true }, "Scroll to see more profiles (\u2191\u2193)")),
+                    React.createElement(Select, { options: profiles.map(p => ({
                             label: `${p.displayName}${p.isActive ? ' (Active)' : ''}`,
                             value: p.name,
-                        })),
-                        {
-                            label: chalk.green('+ New Profile'),
-                            value: '__CREATE_NEW__',
-                        },
-                        {
-                            label: chalk.red('🆇 Delete Profile'),
-                            value: '__DELETE__',
-                        },
-                    ], defaultValue: activeProfile, onChange: handleProfileChange })),
+                        })), defaultValue: activeProfile, onChange: value => {
+                            switchProfile(value);
+                            loadProfilesAndConfig();
+                            setIsEditing(false);
+                            setErrors([]);
+                        } }),
+                    React.createElement(Box, { flexDirection: "row", marginTop: 1 },
+                        React.createElement(Box, { marginRight: 2 },
+                            React.createElement(Text, { color: "green" }, "+ New"),
+                            React.createElement(Text, { color: "gray" }, " (n)")),
+                        React.createElement(Box, null,
+                            React.createElement(Text, { color: "red" }, "\uD83C\uDD87 Delete"),
+                            React.createElement(Text, { color: "gray" }, " (d)"))))),
                 currentField === 'requestMethod' && (React.createElement(Select, { options: requestMethodOptions, defaultValue: requestMethod, onChange: value => {
                         setRequestMethod(value);
                         setIsEditing(false);
@@ -669,7 +665,9 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
                         currentField === 'basicModel' ||
                         currentField === 'compactModelName') &&
                         'Type to filter, ↑↓ to select, Enter to confirm, Esc to cancel',
-                    (currentField === 'profile' || currentField === 'requestMethod') &&
+                    currentField === 'profile' &&
+                        '↑↓ to select profile, N to create new, D to delete, Enter to confirm, Esc to cancel',
+                    currentField === 'requestMethod' &&
                         '↑↓ to select, Enter to confirm, Esc to cancel')))) : (React.createElement(Box, { flexDirection: "column" },
             React.createElement(Box, { flexDirection: "column" },
                 React.createElement(Text, { color: currentField === 'profile' ? 'green' : 'white' },
@@ -707,7 +705,7 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
                     "Anthropic Beta:"),
                 React.createElement(Box, { marginLeft: 3 },
                     React.createElement(Text, { color: "gray" },
-                        anthropicBeta ? '☑ Enabled' : '☐ Disabled',
+                        anthropicBeta ? '☒ Enabled' : '☐ Disabled',
                         " (Press Enter to toggle)"))),
             React.createElement(Box, { flexDirection: "column" },
                 React.createElement(Text, { color: currentField === 'advancedModel' ? 'green' : 'white' },
@@ -760,7 +758,10 @@ export default function ConfigScreen({ onBack, onSave, inlineMode = false, }) {
                 currentField === 'compactModelName')) && (React.createElement(Box, { flexDirection: "column", marginTop: 1 }, isEditing ? (React.createElement(Alert, { variant: "info" },
             "Editing mode:",
             ' ',
-            currentField === 'maxContextTokens' || currentField === 'maxTokens'
+            currentField === 'maxContextTokens' ||
+                currentField === 'maxTokens'
                 ? 'Type to edit, Enter to save'
-                : 'Press Enter to save and exit editing')) : (React.createElement(Alert, { variant: "info" }, "Use \u2191\u2193 to navigate, Enter to edit, M for manual input, Ctrl+S or Esc to save"))))));
+                : 'Press Enter to save and exit editing')) : (React.createElement(Alert, { variant: "info" }, currentField === 'profile'
+            ? 'Use ↑↓ to navigate, N to create new profile, D to delete profile, Ctrl+S or Esc to save'
+            : 'Use ↑↓ to navigate, Enter to edit, M for manual input, Ctrl+S or Esc to save'))))));
 }

package/dist/ui/pages/ProxyConfigScreen.js

@@ -98,7 +98,7 @@ export default function ProxyConfigScreen({ onBack, onSave, inlineMode = false }
                         "Enable Proxy:"),
                     React.createElement(Box, { marginLeft: 3 },
                         React.createElement(Text, { color: "gray" },
-                            enabled ? '☑ Enabled' : '☐ Disabled',
+                            enabled ? '☒ Enabled' : '☐ Disabled',
                             " (Press Enter to toggle)")))),
             React.createElement(Box, { marginBottom: 1 },
                 React.createElement(Box, { flexDirection: "column" },

package/dist/ui/pages/SubAgentConfigScreen.d.ts

@@ -0,0 +1,9 @@
+import React from 'react';
+type Props = {
+    onBack: () => void;
+    onSave: () => void;
+    inlineMode?: boolean;
+    agentId?: string;
+};
+export default function SubAgentConfigScreen({ onBack, onSave, inlineMode, agentId, }: Props): React.JSX.Element;
+export {};