snow-ai 0.4.16 → 0.4.18

package/dist/utils/subAgentExecutor.js

@@ -1,434 +0,0 @@
-import { createStreamingAnthropicCompletion } from '../api/anthropic.js';
-import { createStreamingResponse } from '../api/responses.js';
-import { createStreamingGeminiCompletion } from '../api/gemini.js';
-import { createStreamingChatCompletion } from '../api/chat.js';
-import { getSubAgent } from './subAgentConfig.js';
-import { collectAllMCPTools, executeMCPTool } from './mcpToolsManager.js';
-import { getOpenAiConfig } from './apiConfig.js';
-import { sessionManager } from './sessionManager.js';
-/**
- * Execute a sub-agent as a tool
- * @param agentId - The ID of the sub-agent to execute
- * @param prompt - The task prompt to send to the sub-agent
- * @param onMessage - Callback for streaming sub-agent messages (for UI display)
- * @param abortSignal - Optional abort signal
- * @param requestToolConfirmation - Callback to request tool confirmation from user
- * @param isToolAutoApproved - Function to check if a tool is auto-approved
- * @param yoloMode - Whether YOLO mode is enabled (auto-approve all tools)
- * @returns The final result from the sub-agent
- */
-export async function executeSubAgent(agentId, prompt, onMessage, abortSignal, requestToolConfirmation, isToolAutoApproved, yoloMode, addToAlwaysApproved) {
-    try {
-        // Handle built-in agents (hardcoded)
-        let agent;
-        if (agentId === 'agent_explore') {
-            agent = {
-                id: 'agent_explore',
-                name: 'Explore Agent',
-                description: 'Specialized for quickly exploring and understanding codebases. Excels at searching code, finding definitions, analyzing code structure and semantic understanding.',
-                role: 'You are a specialized code exploration agent. Your task is to help users understand codebase structure, locate specific code, and analyze dependencies. Use search and analysis tools to explore code, but do not modify any files or execute commands. Focus on code discovery and understanding.\n\nIMPORTANT: You have NO access to the main conversation history. The prompt provided to you contains ALL the context from the main session. Read it carefully - all file locations, business requirements, constraints, and discovered information are included in the prompt. Do not assume any additional context.',
-                tools: [
-                    // Filesystem read-only tools
-                    'filesystem-read',
-                    // ACE code search tools (core tools)
-                    'ace-find_definition',
-                    'ace-find_references',
-                    'ace-semantic_search',
-                    'ace-text_search',
-                    'ace-file_outline',
-                    // Codebase search tools
-                    'codebase-search',
-                    // Web search for documentation
-                    'websearch-search',
-                    'websearch-fetch',
-                ],
-            };
-        }
-        else if (agentId === 'agent_plan') {
-            agent = {
-                id: 'agent_plan',
-                name: 'Plan Agent',
-                description: 'Specialized for planning complex tasks. Excels at analyzing requirements, exploring existing code, and creating detailed implementation plans.',
-                role: 'You are a specialized task planning agent. Your task is to analyze user requirements, explore existing codebase, identify relevant files and dependencies, and then create detailed implementation plans. Use search and analysis tools to gather information, check diagnostics to understand current state, but do not execute actual modifications. Output clear step-by-step plans including files to modify, suggested implementation approaches, and important considerations.\n\nIMPORTANT: You have NO access to the main conversation history. The prompt provided to you contains ALL the context from the main session. Read it carefully - all requirements, architecture understanding, file locations, constraints, and user preferences are included in the prompt. Do not assume any additional context.',
-                tools: [
-                    // Filesystem read-only tools
-                    'filesystem-read',
-                    // ACE code search tools (planning requires code understanding)
-                    'ace-find_definition',
-                    'ace-find_references',
-                    'ace-semantic_search',
-                    'ace-text_search',
-                    'ace-file_outline',
-                    // IDE diagnostics (understand current issues)
-                    'ide-get_diagnostics',
-                    // Codebase search
-                    'codebase-search',
-                    // Web search for reference
-                    'websearch-search',
-                    'websearch-fetch',
-                ],
-            };
-        }
-        else if (agentId === 'agent_general') {
-            agent = {
-                id: 'agent_general',
-                name: 'General Purpose Agent',
-                description: 'General-purpose multi-step task execution agent. Has complete tool access for code search, file modification, command execution, and various operations.',
-                role: 'You are a general-purpose task execution agent. You can perform various complex multi-step tasks, including searching code, modifying files, executing commands, etc. When given a task, systematically break it down and execute. You have access to all tools and should select appropriate tools as needed to complete tasks efficiently.\n\nIMPORTANT: You have NO access to the main conversation history. The prompt provided to you contains ALL the context from the main session. Read it carefully - all task requirements, file paths, code patterns, dependencies, business logic, constraints, and testing requirements are included in the prompt. Do not assume any additional context.',
-                tools: [
-                    // Filesystem tools (complete access)
-                    'filesystem-read',
-                    'filesystem-create',
-                    'filesystem-edit',
-                    'filesystem-edit_search',
-                    // Terminal tools
-                    'terminal-execute',
-                    // ACE code search tools
-                    'ace-find_definition',
-                    'ace-find_references',
-                    'ace-semantic_search',
-                    'ace-text_search',
-                    'ace-file_outline',
-                    // Web search tools
-                    'websearch-search',
-                    'websearch-fetch',
-                    // IDE diagnostics tools
-                    'ide-get_diagnostics',
-                    // Codebase search tools
-                    'codebase-search',
-                ],
-            };
-        }
-        else {
-            // Get user-configured sub-agent
-            agent = getSubAgent(agentId);
-            if (!agent) {
-                return {
-                    success: false,
-                    result: '',
-                    error: `Sub-agent with ID "${agentId}" not found`,
-                };
-            }
-        }
-        // Get all available tools
-        const allTools = await collectAllMCPTools();
-        // Filter tools based on sub-agent's allowed tools
-        const allowedTools = allTools.filter((tool) => {
-            const toolName = tool.function.name;
-            return agent.tools.some((allowedTool) => {
-                // Normalize both tool names: replace underscores with hyphens for comparison
-                const normalizedToolName = toolName.replace(/_/g, '-');
-                const normalizedAllowedTool = allowedTool.replace(/_/g, '-');
-                // Support both exact match and prefix match (e.g., "filesystem" matches "filesystem-read")
-                return (normalizedToolName === normalizedAllowedTool ||
-                    normalizedToolName.startsWith(`${normalizedAllowedTool}-`));
-            });
-        });
-        if (allowedTools.length === 0) {
-            return {
-                success: false,
-                result: '',
-                error: `Sub-agent "${agent.name}" has no valid tools configured`,
-            };
-        }
-        // Build conversation history for sub-agent
-        // Append role to prompt if configured
-        let finalPrompt = prompt;
-        if (agent.role) {
-            finalPrompt = `${prompt}\n\n${agent.role}`;
-        }
-        const messages = [
-            {
-                role: 'user',
-                content: finalPrompt,
-            },
-        ];
-        // Stream sub-agent execution
-        let finalResponse = '';
-        let hasError = false;
-        let errorMessage = '';
-        let totalUsage;
-        // Local session-approved tools for this sub-agent execution
-        // This ensures tools approved during execution are immediately recognized
-        const sessionApprovedTools = new Set();
-        // eslint-disable-next-line no-constant-condition
-        while (true) {
-            // Check abort signal before streaming
-            if (abortSignal?.aborted) {
-                // Send done message to mark completion (like normal tool abort)
-                if (onMessage) {
-                    onMessage({
-                        type: 'sub_agent_message',
-                        agentId: agent.id,
-                        agentName: agent.name,
-                        message: {
-                            type: 'done',
-                        },
-                    });
-                }
-                return {
-                    success: false,
-                    result: finalResponse,
-                    error: 'Sub-agent execution aborted',
-                };
-            }
-            // Get API configuration
-            const config = getOpenAiConfig();
-            const currentSession = sessionManager.getCurrentSession();
-            const model = config.advancedModel || 'gpt-5';
-            // Call API with sub-agent's tools - choose API based on config
-            const stream = config.requestMethod === 'anthropic'
-                ? createStreamingAnthropicCompletion({
-                    model,
-                    messages,
-                    temperature: 0,
-                    max_tokens: config.maxTokens || 4096,
-                    tools: allowedTools,
-                    sessionId: currentSession?.id,
-                    disableThinking: true, // Sub-agents 不使用 Extended Thinking
-                }, abortSignal)
-                : config.requestMethod === 'gemini'
-                    ? createStreamingGeminiCompletion({
-                        model,
-                        messages,
-                        temperature: 0,
-                        tools: allowedTools,
-                    }, abortSignal)
-                    : config.requestMethod === 'responses'
-                        ? createStreamingResponse({
-                            model,
-                            messages,
-                            temperature: 0,
-                            tools: allowedTools,
-                            prompt_cache_key: currentSession?.id,
-                        }, abortSignal)
-                        : createStreamingChatCompletion({
-                            model,
-                            messages,
-                            temperature: 0,
-                            tools: allowedTools,
-                        }, abortSignal);
-            let currentContent = '';
-            let toolCalls = [];
-            for await (const event of stream) {
-                // Forward message to UI (but don't save to main conversation)
-                if (onMessage) {
-                    onMessage({
-                        type: 'sub_agent_message',
-                        agentId: agent.id,
-                        agentName: agent.name,
-                        message: event,
-                    });
-                }
-                // Capture usage from stream events
-                if (event.type === 'usage' && event.usage) {
-                    const eventUsage = event.usage;
-                    if (!totalUsage) {
-                        totalUsage = {
-                            inputTokens: eventUsage.prompt_tokens || 0,
-                            outputTokens: eventUsage.completion_tokens || 0,
-                            cacheCreationInputTokens: eventUsage.cache_creation_input_tokens,
-                            cacheReadInputTokens: eventUsage.cache_read_input_tokens,
-                        };
-                    }
-                    else {
-                        // Accumulate usage if there are multiple rounds
-                        totalUsage.inputTokens += eventUsage.prompt_tokens || 0;
-                        totalUsage.outputTokens += eventUsage.completion_tokens || 0;
-                        if (eventUsage.cache_creation_input_tokens) {
-                            totalUsage.cacheCreationInputTokens =
-                                (totalUsage.cacheCreationInputTokens || 0) +
-                                    eventUsage.cache_creation_input_tokens;
-                        }
-                        if (eventUsage.cache_read_input_tokens) {
-                            totalUsage.cacheReadInputTokens =
-                                (totalUsage.cacheReadInputTokens || 0) +
-                                    eventUsage.cache_read_input_tokens;
-                        }
-                    }
-                }
-                if (event.type === 'content' && event.content) {
-                    currentContent += event.content;
-                }
-                else if (event.type === 'tool_calls' && event.tool_calls) {
-                    toolCalls = event.tool_calls;
-                }
-            }
-            if (hasError) {
-                return {
-                    success: false,
-                    result: finalResponse,
-                    error: errorMessage,
-                };
-            }
-            // Add assistant response to conversation
-            if (currentContent || toolCalls.length > 0) {
-                const assistantMessage = {
-                    role: 'assistant',
-                    content: currentContent || '',
-                };
-                if (toolCalls.length > 0) {
-                    assistantMessage.tool_calls = toolCalls;
-                }
-                messages.push(assistantMessage);
-                finalResponse = currentContent;
-            }
-            // If no tool calls, we're done
-            if (toolCalls.length === 0) {
-                break;
-            }
-            // Check tool approvals before execution
-            const approvedToolCalls = [];
-            const rejectedToolCalls = [];
-            for (const toolCall of toolCalls) {
-                const toolName = toolCall.function.name;
-                let args;
-                try {
-                    args = JSON.parse(toolCall.function.arguments);
-                }
-                catch (e) {
-                    args = {};
-                }
-                // Check if tool needs confirmation
-                let needsConfirmation = true;
-                // In YOLO mode, auto-approve all tools
-                if (yoloMode) {
-                    needsConfirmation = false;
-                }
-                // Check if tool is in auto-approved list (global or session)
-                else if (sessionApprovedTools.has(toolName) ||
-                    (isToolAutoApproved && isToolAutoApproved(toolName))) {
-                    needsConfirmation = false;
-                }
-                if (needsConfirmation && requestToolConfirmation) {
-                    // Request confirmation from user
-                    const confirmation = await requestToolConfirmation(toolName, args);
-                    if (confirmation === 'reject' ||
-                        (typeof confirmation === 'object' &&
-                            confirmation.type === 'reject_with_reply')) {
-                        rejectedToolCalls.push(toolCall);
-                        continue;
-                    }
-                    // If approve_always, add to both global and session lists
-                    if (confirmation === 'approve_always') {
-                        // Add to local session set (immediate effect)
-                        sessionApprovedTools.add(toolName);
-                        // Add to global list (persistent across sub-agent calls)
-                        if (addToAlwaysApproved) {
-                            addToAlwaysApproved(toolName);
-                        }
-                    }
-                }
-                approvedToolCalls.push(toolCall);
-            }
-            // Handle rejected tools
-            if (rejectedToolCalls.length > 0) {
-                // Send done message to mark completion when tools are rejected
-                if (onMessage) {
-                    onMessage({
-                        type: 'sub_agent_message',
-                        agentId: agent.id,
-                        agentName: agent.name,
-                        message: {
-                            type: 'done',
-                        },
-                    });
-                }
-                return {
-                    success: false,
-                    result: finalResponse,
-                    error: `User rejected tool execution: ${rejectedToolCalls
-                        .map(tc => tc.function.name)
-                        .join(', ')}`,
-                };
-            }
-            // Execute approved tool calls
-            const toolResults = [];
-            for (const toolCall of approvedToolCalls) {
-                // Check abort signal before executing each tool
-                if (abortSignal?.aborted) {
-                    // Send done message to mark completion
-                    if (onMessage) {
-                        onMessage({
-                            type: 'sub_agent_message',
-                            agentId: agent.id,
-                            agentName: agent.name,
-                            message: {
-                                type: 'done',
-                            },
-                        });
-                    }
-                    return {
-                        success: false,
-                        result: finalResponse,
-                        error: 'Sub-agent execution aborted during tool execution',
-                    };
-                }
-                try {
-                    const args = JSON.parse(toolCall.function.arguments);
-                    const result = await executeMCPTool(toolCall.function.name, args, abortSignal);
-                    const toolResult = {
-                        role: 'tool',
-                        tool_call_id: toolCall.id,
-                        content: JSON.stringify(result),
-                    };
-                    toolResults.push(toolResult);
-                    // Send tool result to UI
-                    if (onMessage) {
-                        onMessage({
-                            type: 'sub_agent_message',
-                            agentId: agent.id,
-                            agentName: agent.name,
-                            message: {
-                                type: 'tool_result',
-                                tool_call_id: toolCall.id,
-                                tool_name: toolCall.function.name,
-                                content: JSON.stringify(result),
-                            },
-                        });
-                    }
-                }
-                catch (error) {
-                    const errorResult = {
-                        role: 'tool',
-                        tool_call_id: toolCall.id,
-                        content: `Error: ${error instanceof Error ? error.message : 'Tool execution failed'}`,
-                    };
-                    toolResults.push(errorResult);
-                    // Send error result to UI
-                    if (onMessage) {
-                        onMessage({
-                            type: 'sub_agent_message',
-                            agentId: agent.id,
-                            agentName: agent.name,
-                            message: {
-                                type: 'tool_result',
-                                tool_call_id: toolCall.id,
-                                tool_name: toolCall.function.name,
-                                content: `Error: ${error instanceof Error
-                                    ? error.message
-                                    : 'Tool execution failed'}`,
-                            },
-                        });
-                    }
-                }
-            }
-            // Add tool results to conversation
-            messages.push(...toolResults);
-            // Continue to next iteration if there were tool calls
-            // The loop will continue until no more tool calls
-        }
-        return {
-            success: true,
-            result: finalResponse,
-            usage: totalUsage,
-        };
-    }
-    catch (error) {
-        return {
-            success: false,
-            result: '',
-            error: error instanceof Error ? error.message : 'Unknown error',
-        };
-    }
-}

package/dist/utils/terminal.d.ts

@@ -1,5 +0,0 @@
-type WritableStreamLike = Pick<NodeJS.WriteStream, 'write'> | {
-    write: (data: string) => unknown;
-};
-export declare function resetTerminal(stream?: WritableStreamLike): void;
-export {};

package/dist/utils/terminal.js

@@ -1,13 +0,0 @@
-export function resetTerminal(stream) {
-    const target = stream ?? process.stdout;
-    if (!target || typeof target.write !== 'function') {
-        return;
-    }
-    // RIS (Reset to Initial State) clears scrollback and resets terminal modes
-    target.write('\x1bc');
-    target.write('\x1B[3J\x1B[2J\x1B[H');
-    // Re-enable focus reporting immediately after terminal reset
-    // The RIS command (\x1bc) disables focus reporting, so we must re-enable it
-    // This ensures focus state tracking continues to work after /clear command
-    target.write('\x1b[?1004h');
-}

package/dist/utils/textBuffer.d.ts

@@ -1,99 +0,0 @@
-export interface Viewport {
-    width: number;
-    height: number;
-}
-/**
- * 统一的占位符类型，用于大文本粘贴和图片
- */
-export interface Placeholder {
-    id: string;
-    content: string;
-    type: 'text' | 'image';
-    charCount: number;
-    index: number;
-    placeholder: string;
-    mimeType?: string;
-}
-/**
- * 图片数据类型（向后兼容）
- */
-export interface ImageData {
-    id: string;
-    data: string;
-    mimeType: string;
-    index: number;
-    placeholder: string;
-}
-export declare class TextBuffer {
-    private content;
-    private cursorIndex;
-    private viewport;
-    private placeholderStorage;
-    private textPlaceholderCounter;
-    private imagePlaceholderCounter;
-    private onUpdateCallback?;
-    private isDestroyed;
-    private tempPastingPlaceholder;
-    private visualLines;
-    private visualLineStarts;
-    private visualCursorPos;
-    private preferredVisualCol;
-    constructor(viewport: Viewport, onUpdate?: () => void);
-    /**
-     * Cleanup method to be called when the buffer is no longer needed
-     */
-    destroy(): void;
-    get text(): string;
-    /**
-     * 获取完整文本，包括替换占位符为原始内容（仅文本类型）
-     */
-    getFullText(): string;
-    get visualCursor(): [number, number];
-    getCursorPosition(): number;
-    setCursorPosition(position: number): void;
-    get viewportVisualLines(): string[];
-    get maxWidth(): number;
-    private scheduleUpdate;
-    setText(text: string): void;
-    insert(input: string): void;
-    /**
-     * 插入临时"粘贴中"占位符，用于大文本粘贴时的用户反馈
-     */
-    insertPastingIndicator(): void;
-    private insertPlainText;
-    backspace(): void;
-    delete(): void;
-    moveLeft(): void;
-    moveRight(): void;
-    moveUp(): void;
-    moveDown(): void;
-    /**
-     * Update the viewport dimensions, useful for terminal resize handling.
-     */
-    updateViewport(viewport: Viewport): void;
-    /**
-     * Get the character and its visual info at cursor position for proper rendering.
-     */
-    getCharAtCursor(): {
-        char: string;
-        isWideChar: boolean;
-    };
-    private clampCursorIndex;
-    private recalculateVisualState;
-    private wrapLineToWidth;
-    private computeVisualCursorFromIndex;
-    private moveCursorToVisualRow;
-    private recomputeVisualCursorOnly;
-    /**
-     * 插入图片数据（使用统一的占位符系统）
-     */
-    insertImage(base64Data: string, mimeType: string): void;
-    /**
-     * 获取所有图片数据（还原为 data URL 格式）
-     */
-    getImages(): ImageData[];
-    /**
-     * 清除所有图片
-     */
-    clearImages(): void;
-}