snow-ai 0.3.36 → 0.4.0

package/dist/hooks/useConversation.js

@@ -37,7 +37,7 @@ export async function handleConversationWithTools(options) {
     const mcpTools = await collectAllMCPTools();
     // Build conversation history with TODO context as pinned user message
     let conversationMessages = [
-        { role: 'system', content: getSystemPrompt(mcpTools) },
+        { role: 'system', content: getSystemPrompt() },
     ];
     // If there are TODOs, add pinned context message at the front
     if (existingTodoList && existingTodoList.todos.length > 0) {
@@ -430,32 +430,67 @@ export async function handleConversationWithTools(options) {
                         const firstTool = sensitiveTools[0];
                         const allTools = sensitiveTools.length > 1 ? sensitiveTools : undefined;
                         const confirmation = await requestToolConfirmation(firstTool, undefined, allTools);
-                        if (confirmation === 'reject') {
+                        if (confirmation === 'reject' ||
+                            (typeof confirmation === 'object' &&
+                                confirmation.type === 'reject_with_reply')) {
                             setMessages(prev => prev.filter(msg => !msg.toolPending));
+                            const rejectMessage = typeof confirmation === 'object'
+                                ? `Tool execution rejected by user: ${confirmation.reason}`
+                                : 'Error: Tool execution rejected by user';
+                            // Create UI messages for rejected tools
+                            const rejectedToolUIMessages = [];
                             for (const toolCall of sensitiveTools) {
                                 const rejectionMessage = {
                                     role: 'tool',
                                     tool_call_id: toolCall.id,
-                                    content: 'Error: Tool execution rejected by user',
+                                    content: rejectMessage,
                                 };
                                 conversationMessages.push(rejectionMessage);
                                 saveMessage(rejectionMessage).catch(error => {
                                     console.error('Failed to save tool rejection message:', error);
                                 });
-                            }
-                            setMessages(prev => [
-                                ...prev,
-                                {
+                                // Add UI message for each rejected tool
+                                const toolDisplay = formatToolCallMessage(toolCall);
+                                const statusIcon = '✗';
+                                let statusText = '';
+                                if (typeof confirmation === 'object' && confirmation.reason) {
+                                    statusText = `\n  └─ Rejection reason: ${confirmation.reason}`;
+                                }
+                                else {
+                                    statusText = `\n  └─ ${rejectMessage}`;
+                                }
+                                rejectedToolUIMessages.push({
                                     role: 'assistant',
-                                    content: 'Tool call rejected, session ended',
+                                    content: `${statusIcon} ${toolDisplay.toolName}${statusText}`,
                                     streaming: false,
-                                },
-                            ]);
-                            if (options.setIsStreaming) {
-                                options.setIsStreaming(false);
+                                });
+                            }
+                            // Add rejected tool messages to UI
+                            if (rejectedToolUIMessages.length > 0) {
+                                setMessages(prev => [...prev, ...rejectedToolUIMessages]);
+                            }
+                            // If reject_with_reply, continue the conversation instead of ending
+                            if (typeof confirmation === 'object' &&
+                                confirmation.type === 'reject_with_reply') {
+                                // Continue to next iteration - AI will see the rejection message and respond
+                                continue;
+                            }
+                            else {
+                                // Original reject behavior - end session
+                                setMessages(prev => [
+                                    ...prev,
+                                    {
+                                        role: 'assistant',
+                                        content: 'Tool call rejected, session ended',
+                                        streaming: false,
+                                    },
+                                ]);
+                                if (options.setIsStreaming) {
+                                    options.setIsStreaming(false);
+                                }
+                                freeEncoder();
+                                return { usage: accumulatedUsage };
                             }
-                            freeEncoder();
-                            return { usage: accumulatedUsage };
                         }
                         // Approved, add sensitive tools to approved list
                         approvedTools.push(...sensitiveTools);
@@ -467,32 +502,67 @@ export async function handleConversationWithTools(options) {
                         ? toolsNeedingConfirmation
                         : undefined;
                     const confirmation = await requestToolConfirmation(firstTool, undefined, allTools);
-                    if (confirmation === 'reject') {
+                    if (confirmation === 'reject' ||
+                        (typeof confirmation === 'object' &&
+                            confirmation.type === 'reject_with_reply')) {
                         setMessages(prev => prev.filter(msg => !msg.toolPending));
+                        const rejectMessage = typeof confirmation === 'object'
+                            ? `Tool execution rejected by user: ${confirmation.reason}`
+                            : 'Error: Tool execution rejected by user';
+                        // Create UI messages for rejected tools
+                        const rejectedToolUIMessages = [];
                         for (const toolCall of toolsNeedingConfirmation) {
                             const rejectionMessage = {
                                 role: 'tool',
                                 tool_call_id: toolCall.id,
-                                content: 'Error: Tool execution rejected by user',
+                                content: rejectMessage,
                             };
                             conversationMessages.push(rejectionMessage);
                             saveMessage(rejectionMessage).catch(error => {
                                 console.error('Failed to save tool rejection message:', error);
                             });
-                        }
-                        setMessages(prev => [
-                            ...prev,
-                            {
+                            // Add UI message for each rejected tool
+                            const toolDisplay = formatToolCallMessage(toolCall);
+                            const statusIcon = '✗';
+                            let statusText = '';
+                            if (typeof confirmation === 'object' && confirmation.reason) {
+                                statusText = `\n  └─ Rejection reason: ${confirmation.reason}`;
+                            }
+                            else {
+                                statusText = `\n  └─ ${rejectMessage}`;
+                            }
+                            rejectedToolUIMessages.push({
                                 role: 'assistant',
-                                content: 'Tool call rejected, session ended',
+                                content: `${statusIcon} ${toolDisplay.toolName}${statusText}`,
                                 streaming: false,
-                            },
-                        ]);
-                        if (options.setIsStreaming) {
-                            options.setIsStreaming(false);
+                            });
+                        }
+                        // Add rejected tool messages to UI
+                        if (rejectedToolUIMessages.length > 0) {
+                            setMessages(prev => [...prev, ...rejectedToolUIMessages]);
+                        }
+                        // If reject_with_reply, continue the conversation instead of ending
+                        if (typeof confirmation === 'object' &&
+                            confirmation.type === 'reject_with_reply') {
+                            // Continue to next iteration - AI will see the rejection message and respond
+                            continue;
+                        }
+                        else {
+                            // Original reject behavior - end session
+                            setMessages(prev => [
+                                ...prev,
+                                {
+                                    role: 'assistant',
+                                    content: 'Tool call rejected, session ended',
+                                    streaming: false,
+                                },
+                            ]);
+                            if (options.setIsStreaming) {
+                                options.setIsStreaming(false);
+                            }
+                            freeEncoder();
+                            return { usage: accumulatedUsage };
                         }
-                        freeEncoder();
-                        return { usage: accumulatedUsage };
                     }
                     // If approved_always, add ALL these tools to both global and session-approved sets
                     if (confirmation === 'approve_always') {
@@ -535,7 +605,7 @@ export async function handleConversationWithTools(options) {
                                             name: toolCall.function.name,
                                             arguments: toolArgs,
                                         },
-                                        toolDisplay,
+                                        // Don't include toolDisplay for sub-agent tools to avoid showing parameters
                                         toolCallId: toolCall.id,
                                         toolPending: true,
                                         subAgent: {
@@ -711,7 +781,9 @@ export async function handleConversationWithTools(options) {
                     break;
                 }
                 // 在工具执行完成后、发送结果到AI前，检查是否需要压缩
-                if (options.getCurrentContextPercentage &&
+                const config = getOpenAiConfig();
+                if (config.enableAutoCompress !== false &&
+                    options.getCurrentContextPercentage &&
                     shouldAutoCompress(options.getCurrentContextPercentage())) {
                     try {
                         // 显示压缩提示消息
@@ -763,12 +835,24 @@ export async function handleConversationWithTools(options) {
                             const isError = result.content.startsWith('Error:');
                             const statusIcon = isError ? '✗' : '✓';
                             const statusText = isError ? `\n  └─ ${result.content}` : '';
+                            // Parse sub-agent result to extract usage information
+                            let usage = undefined;
+                            if (!isError) {
+                                try {
+                                    const subAgentResult = JSON.parse(result.content);
+                                    usage = subAgentResult.usage;
+                                }
+                                catch (e) {
+                                    // Ignore parsing errors
+                                }
+                            }
                             resultMessages.push({
                                 role: 'assistant',
                                 content: `${statusIcon} ${toolCall.function.name}${statusText}`,
                                 streaming: false,
                                 // Pass the full result.content for ToolResultPreview to parse
                                 toolResult: !isError ? result.content : undefined,
+                                subAgentUsage: usage,
                             });
                             // Save the tool result to conversation history
                             conversationMessages.push(result);
@@ -852,7 +936,9 @@ export async function handleConversationWithTools(options) {
                     const pendingMessages = options.getPendingMessages();
                     if (pendingMessages.length > 0) {
                         // 检查 token 占用，如果 >= 80% 先执行自动压缩
-                        if (options.getCurrentContextPercentage &&
+                        const config = getOpenAiConfig();
+                        if (config.enableAutoCompress !== false &&
+                            options.getCurrentContextPercentage &&
                             shouldAutoCompress(options.getCurrentContextPercentage())) {
                             try {
                                 // 显示压缩提示消息

package/dist/hooks/useGlobalExit.js

@@ -1,6 +1,8 @@
 import { useInput } from 'ink';
 import { useState } from 'react';
+import { useI18n } from '../i18n/index.js';
 export function useGlobalExit(onNotification) {
+    const { t } = useI18n();
     const [lastCtrlCTime, setLastCtrlCTime] = useState(0);
     const ctrlCTimeout = 1000; // 1 second timeout for double Ctrl+C
     useInput((input, key) => {
@@ -16,13 +18,13 @@ export function useGlobalExit(onNotification) {
                 if (onNotification) {
                     onNotification({
                         show: true,
-                        message: 'Press Ctrl+C again to exit'
+                        message: t.hooks.pressCtrlCAgain,
                     });
                     // Hide notification after timeout
                     setTimeout(() => {
                         onNotification({
                             show: false,
-                            message: ''
+                            message: '',
                         });
                     }, ctrlCTimeout);
                 }

package/dist/hooks/useStreamingState.d.ts

@@ -6,6 +6,13 @@ export type RetryStatus = {
     remainingSeconds?: number;
     errorMessage?: string;
 };
+export type CodebaseSearchStatus = {
+    isSearching: boolean;
+    attempt: number;
+    maxAttempts: number;
+    currentTopN: number;
+    message: string;
+};
 export declare function useStreamingState(): {
     isStreaming: boolean;
     setIsStreaming: import("react").Dispatch<import("react").SetStateAction<boolean>>;
@@ -21,4 +28,6 @@ export declare function useStreamingState(): {
     retryStatus: RetryStatus | null;
     setRetryStatus: import("react").Dispatch<import("react").SetStateAction<RetryStatus | null>>;
     animationFrame: number;
+    codebaseSearchStatus: CodebaseSearchStatus | null;
+    setCodebaseSearchStatus: import("react").Dispatch<import("react").SetStateAction<CodebaseSearchStatus | null>>;
 };

package/dist/hooks/useStreamingState.js

@@ -9,6 +9,7 @@ export function useStreamingState() {
     const [timerStartTime, setTimerStartTime] = useState(null);
     const [retryStatus, setRetryStatus] = useState(null);
     const [animationFrame, setAnimationFrame] = useState(0);
+    const [codebaseSearchStatus, setCodebaseSearchStatus] = useState(null);
     // Animation for streaming/saving indicator
     useEffect(() => {
         if (!isStreaming)
@@ -98,5 +99,7 @@ export function useStreamingState() {
         retryStatus,
         setRetryStatus,
         animationFrame,
+        codebaseSearchStatus,
+        setCodebaseSearchStatus,
     };
 }

package/dist/i18n/I18nContext.d.ts

@@ -0,0 +1,14 @@
+import React, { ReactNode } from 'react';
+import type { Language, TranslationKeys } from './types.js';
+type I18nContextType = {
+    language: Language;
+    setLanguage: (lang: Language) => void;
+    t: TranslationKeys;
+};
+type Props = {
+    children: ReactNode;
+    defaultLanguage?: Language;
+};
+export declare function I18nProvider({ children, defaultLanguage }: Props): React.JSX.Element;
+export declare function useI18n(): I18nContextType;
+export {};

package/dist/i18n/I18nContext.js

@@ -0,0 +1,24 @@
+import React, { createContext, useState, useCallback } from 'react';
+import { translations } from './translations.js';
+import { getCurrentLanguage, setCurrentLanguage, } from '../utils/languageConfig.js';
+const I18nContext = createContext(undefined);
+export function I18nProvider({ children, defaultLanguage }) {
+    // Load saved language on mount or use default
+    const [language, setLanguageState] = useState(() => {
+        return defaultLanguage || getCurrentLanguage();
+    });
+    const setLanguage = useCallback((lang) => {
+        setLanguageState(lang);
+        setCurrentLanguage(lang); // Persist to file system
+    }, []);
+    // Get translations for current language
+    const t = translations[language];
+    return (React.createElement(I18nContext.Provider, { value: { language, setLanguage, t } }, children));
+}
+export function useI18n() {
+    const context = React.useContext(I18nContext);
+    if (!context) {
+        throw new Error('useI18n must be used within I18nProvider');
+    }
+    return context;
+}

package/dist/i18n/index.d.ts

@@ -0,0 +1,3 @@
+export { I18nProvider, useI18n } from './I18nContext.js';
+export type { Language, TranslationKeys, Translations } from './types.js';
+export { translations } from './translations.js';

package/dist/i18n/index.js

@@ -0,0 +1,2 @@
+export { I18nProvider, useI18n } from './I18nContext.js';
+export { translations } from './translations.js';

package/dist/i18n/lang/en.d.ts

@@ -0,0 +1,2 @@
+import type { TranslationKeys } from '../types.js';
+export declare const en: TranslationKeys;