snow-ai 0.4.16 → 0.4.18

package/dist/utils/sessionManager.js

@@ -1,371 +0,0 @@
-import fs from 'fs/promises';
-import path from 'path';
-import os from 'os';
-import { randomUUID } from 'crypto';
-import { getTodoService } from './mcpToolsManager.js';
-import { logger } from './logger.js';
-import { summaryAgent } from '../agents/summaryAgent.js';
-class SessionManager {
-    constructor() {
-        Object.defineProperty(this, "sessionsDir", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "currentSession", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
-        this.sessionsDir = path.join(os.homedir(), '.snow', 'sessions');
-    }
-    async ensureSessionsDir(date) {
-        try {
-            await fs.mkdir(this.sessionsDir, { recursive: true });
-            if (date) {
-                const dateFolder = this.formatDateForFolder(date);
-                const sessionDir = path.join(this.sessionsDir, dateFolder);
-                await fs.mkdir(sessionDir, { recursive: true });
-            }
-        }
-        catch (error) {
-            // Directory already exists or other error
-        }
-    }
-    getSessionPath(sessionId, date) {
-        const sessionDate = date || new Date();
-        const dateFolder = this.formatDateForFolder(sessionDate);
-        const sessionDir = path.join(this.sessionsDir, dateFolder);
-        return path.join(sessionDir, `${sessionId}.json`);
-    }
-    formatDateForFolder(date) {
-        const year = date.getFullYear();
-        const month = String(date.getMonth() + 1).padStart(2, '0');
-        const day = String(date.getDate()).padStart(2, '0');
-        return `${year}-${month}-${day}`;
-    }
-    /**
-     * Clean title by removing newlines and extra spaces
-     */
-    cleanTitle(title) {
-        return title
-            .replace(/[\r\n]+/g, ' ') // Replace newlines with space
-            .replace(/\s+/g, ' ') // Replace multiple spaces with single space
-            .trim(); // Remove leading/trailing spaces
-    }
-    async createNewSession() {
-        await this.ensureSessionsDir(new Date());
-        // 使用 UUID v4 生成唯一会话 ID，避免并发冲突
-        const sessionId = randomUUID();
-        const session = {
-            id: sessionId,
-            title: 'New Chat',
-            summary: '',
-            createdAt: Date.now(),
-            updatedAt: Date.now(),
-            messages: [],
-            messageCount: 0,
-        };
-        this.currentSession = session;
-        await this.saveSession(session);
-        return session;
-    }
-    async saveSession(session) {
-        const sessionDate = new Date(session.createdAt);
-        await this.ensureSessionsDir(sessionDate);
-        const sessionPath = this.getSessionPath(session.id, sessionDate);
-        await fs.writeFile(sessionPath, JSON.stringify(session, null, 2));
-    }
-    async loadSession(sessionId) {
-        // 首先尝试从旧格式加载（向下兼容）
-        try {
-            const oldSessionPath = path.join(this.sessionsDir, `${sessionId}.json`);
-            const data = await fs.readFile(oldSessionPath, 'utf-8');
-            const session = JSON.parse(data);
-            this.currentSession = session;
-            return session;
-        }
-        catch (error) {
-            // 旧格式不存在，搜索日期文件夹
-        }
-        // 在日期文件夹中查找会话
-        try {
-            const session = await this.findSessionInDateFolders(sessionId);
-            if (session) {
-                this.currentSession = session;
-                return session;
-            }
-        }
-        catch (error) {
-            // 搜索失败
-        }
-        return null;
-    }
-    async findSessionInDateFolders(sessionId) {
-        try {
-            const files = await fs.readdir(this.sessionsDir);
-            for (const file of files) {
-                const filePath = path.join(this.sessionsDir, file);
-                const stat = await fs.stat(filePath);
-                if (stat.isDirectory() && /^\d{4}-\d{2}-\d{2}$/.test(file)) {
-                    // 这是日期文件夹，查找会话文件
-                    const sessionPath = path.join(filePath, `${sessionId}.json`);
-                    try {
-                        const data = await fs.readFile(sessionPath, 'utf-8');
-                        const session = JSON.parse(data);
-                        return session;
-                    }
-                    catch (error) {
-                        // 文件不存在或读取失败，继续搜索
-                        continue;
-                    }
-                }
-            }
-        }
-        catch (error) {
-            // 目录读取失败
-        }
-        return null;
-    }
-    async listSessions() {
-        await this.ensureSessionsDir();
-        const sessions = [];
-        try {
-            // 首先处理新的日期文件夹结构
-            const files = await fs.readdir(this.sessionsDir);
-            for (const file of files) {
-                const filePath = path.join(this.sessionsDir, file);
-                const stat = await fs.stat(filePath);
-                if (stat.isDirectory() && /^\d{4}-\d{2}-\d{2}$/.test(file)) {
-                    // 这是日期文件夹，读取其中的会话文件
-                    await this.readSessionsFromDir(filePath, sessions);
-                }
-                else if (file.endsWith('.json')) {
-                    // 这是旧格式的会话文件（向下兼容）
-                    try {
-                        const data = await fs.readFile(filePath, 'utf-8');
-                        const session = JSON.parse(data);
-                        sessions.push({
-                            id: session.id,
-                            title: this.cleanTitle(session.title),
-                            summary: session.summary,
-                            createdAt: session.createdAt,
-                            updatedAt: session.updatedAt,
-                            messageCount: session.messageCount,
-                        });
-                    }
-                    catch (error) {
-                        // Skip invalid session files
-                        continue;
-                    }
-                }
-            }
-            // Sort by updatedAt (newest first)
-            return sessions.sort((a, b) => b.updatedAt - a.updatedAt);
-        }
-        catch (error) {
-            return [];
-        }
-    }
-    async readSessionsFromDir(dirPath, sessions) {
-        try {
-            const files = await fs.readdir(dirPath);
-            for (const file of files) {
-                if (file.endsWith('.json')) {
-                    try {
-                        const sessionPath = path.join(dirPath, file);
-                        const data = await fs.readFile(sessionPath, 'utf-8');
-                        const session = JSON.parse(data);
-                        sessions.push({
-                            id: session.id,
-                            title: this.cleanTitle(session.title),
-                            summary: session.summary,
-                            createdAt: session.createdAt,
-                            updatedAt: session.updatedAt,
-                            messageCount: session.messageCount,
-                        });
-                    }
-                    catch (error) {
-                        // Skip invalid session files
-                        continue;
-                    }
-                }
-            }
-        }
-        catch (error) {
-            // Skip directory if it can't be read
-        }
-    }
-    async addMessage(message) {
-        if (!this.currentSession) {
-            this.currentSession = await this.createNewSession();
-        }
-        // Check if this exact message already exists to prevent duplicates
-        // For assistant messages with tool_calls, also compare tool_call_id to ensure uniqueness
-        const existingMessage = this.currentSession.messages.find(m => {
-            if (m.role !== message.role)
-                return false;
-            if (m.content !== message.content)
-                return false;
-            if (Math.abs(m.timestamp - message.timestamp) >= 5000)
-                return false;
-            // If both messages have tool_calls, compare tool call IDs
-            if (m.tool_calls && message.tool_calls) {
-                // Create sets of tool call IDs for comparison
-                const existingIds = new Set(m.tool_calls.map(tc => tc.id));
-                const newIds = new Set(message.tool_calls.map(tc => tc.id));
-                // If IDs are different, these are different messages
-                if (existingIds.size !== newIds.size)
-                    return false;
-                for (const id of newIds) {
-                    if (!existingIds.has(id))
-                        return false;
-                }
-            }
-            else if (m.tool_calls || message.tool_calls) {
-                // One has tool_calls, the other doesn't - different messages
-                return false;
-            }
-            // If both have tool_call_id (tool response), compare them
-            if (m.tool_call_id && message.tool_call_id) {
-                return m.tool_call_id === message.tool_call_id;
-            }
-            else if (m.tool_call_id || message.tool_call_id) {
-                // One has tool_call_id, the other doesn't - different messages
-                return false;
-            }
-            return true;
-        });
-        if (existingMessage) {
-            return; // Don't add duplicate message
-        }
-        this.currentSession.messages.push(message);
-        this.currentSession.messageCount = this.currentSession.messages.length;
-        this.currentSession.updatedAt = Date.now();
-        // Generate simple title and summary from first user message
-        if (this.currentSession.messageCount === 1 && message.role === 'user') {
-            // Use first 50 chars as title, first 100 chars as summary
-            const title = message.content.slice(0, 50) + (message.content.length > 50 ? '...' : '');
-            const summary = message.content.slice(0, 100) + (message.content.length > 100 ? '...' : '');
-            this.currentSession.title = this.cleanTitle(title);
-            this.currentSession.summary = this.cleanTitle(summary);
-        }
-        // After the first complete conversation exchange (user + assistant), generate AI summary
-        // Only run once when messageCount becomes 2 and the second message is from assistant
-        if (this.currentSession.messageCount === 2 && message.role === 'assistant') {
-            // Run summary generation in background without blocking
-            this.generateAndUpdateSummary().catch(error => {
-                logger.error('Failed to generate conversation summary:', error);
-            });
-        }
-        await this.saveSession(this.currentSession);
-    }
-    /**
-     * Generate AI-powered summary for the first conversation exchange
-     * This runs in the background without blocking the main flow
-     */
-    async generateAndUpdateSummary() {
-        if (!this.currentSession || this.currentSession.messages.length < 2) {
-            return;
-        }
-        try {
-            // Extract first user and assistant messages
-            const firstUserMessage = this.currentSession.messages.find(m => m.role === 'user');
-            const firstAssistantMessage = this.currentSession.messages.find(m => m.role === 'assistant');
-            if (!firstUserMessage || !firstAssistantMessage) {
-                logger.warn('Summary agent: Could not find first user/assistant messages');
-                return;
-            }
-            // Generate summary using summary agent
-            const result = await summaryAgent.generateSummary(firstUserMessage.content, firstAssistantMessage.content);
-            if (result) {
-                // Update session with generated summary
-                this.currentSession.title = result.title;
-                this.currentSession.summary = result.summary;
-                // Save updated session
-                await this.saveSession(this.currentSession);
-                logger.info('Summary agent: Successfully updated session summary', {
-                    sessionId: this.currentSession.id,
-                    title: result.title,
-                    summary: result.summary
-                });
-            }
-        }
-        catch (error) {
-            // Silently fail - don't disrupt main conversation flow
-            logger.error('Summary agent: Failed to generate summary', error);
-        }
-    }
-    getCurrentSession() {
-        return this.currentSession;
-    }
-    setCurrentSession(session) {
-        this.currentSession = session;
-    }
-    clearCurrentSession() {
-        this.currentSession = null;
-    }
-    async deleteSession(sessionId) {
-        let sessionDeleted = false;
-        // 首先尝试删除旧格式（向下兼容）
-        try {
-            const oldSessionPath = path.join(this.sessionsDir, `${sessionId}.json`);
-            await fs.unlink(oldSessionPath);
-            sessionDeleted = true;
-        }
-        catch (error) {
-            // 旧格式不存在，搜索日期文件夹
-        }
-        // 在日期文件夹中查找并删除会话
-        if (!sessionDeleted) {
-            try {
-                const files = await fs.readdir(this.sessionsDir);
-                for (const file of files) {
-                    const filePath = path.join(this.sessionsDir, file);
-                    const stat = await fs.stat(filePath);
-                    if (stat.isDirectory() && /^\d{4}-\d{2}-\d{2}$/.test(file)) {
-                        // 这是日期文件夹，查找会话文件
-                        const sessionPath = path.join(filePath, `${sessionId}.json`);
-                        try {
-                            await fs.unlink(sessionPath);
-                            sessionDeleted = true;
-                            break;
-                        }
-                        catch (error) {
-                            // 文件不存在，继续搜索
-                            continue;
-                        }
-                    }
-                }
-            }
-            catch (error) {
-                // 目录读取失败
-            }
-        }
-        // 如果会话删除成功，同时删除对应的TODO列表
-        if (sessionDeleted) {
-            try {
-                const todoService = getTodoService();
-                await todoService.deleteTodoList(sessionId);
-            }
-            catch (error) {
-                // TODO删除失败不影响会话删除结果
-                logger.warn(`Failed to delete TODO list for session ${sessionId}:`, error);
-            }
-        }
-        return sessionDeleted;
-    }
-    async truncateMessages(messageCount) {
-        if (!this.currentSession) {
-            return;
-        }
-        // Truncate messages array to specified count
-        this.currentSession.messages = this.currentSession.messages.slice(0, messageCount);
-        this.currentSession.messageCount = this.currentSession.messages.length;
-        this.currentSession.updatedAt = Date.now();
-        await this.saveSession(this.currentSession);
-    }
-}
-export const sessionManager = new SessionManager();

package/dist/utils/subAgentConfig.d.ts

@@ -1,50 +0,0 @@
-export interface SubAgent {
-    id: string;
-    name: string;
-    description: string;
-    role?: string;
-    tools: string[];
-    createdAt: string;
-    updatedAt: string;
-    builtin?: boolean;
-}
-export interface SubAgentsConfig {
-    agents: SubAgent[];
-}
-/**
- * Get user-configured sub-agents only (exported for MCP tool generation)
- */
-export declare function getUserSubAgents(): SubAgent[];
-/**
- * Get all sub-agents (built-in + user-configured)
- */
-export declare function getSubAgents(): SubAgent[];
-/**
- * Get a sub-agent by ID (checks both built-in and user-configured)
- */
-export declare function getSubAgent(id: string): SubAgent | null;
-/**
- * Create a new sub-agent (user-configured only)
- */
-export declare function createSubAgent(name: string, description: string, tools: string[], role?: string): SubAgent;
-/**
- * Update an existing sub-agent (only user-configured agents can be updated)
- */
-export declare function updateSubAgent(id: string, updates: {
-    name?: string;
-    description?: string;
-    role?: string;
-    tools?: string[];
-}): SubAgent | null;
-/**
- * Delete a sub-agent (only user-configured agents can be deleted)
- */
-export declare function deleteSubAgent(id: string): boolean;
-/**
- * Validate sub-agent data
- */
-export declare function validateSubAgent(data: {
-    name: string;
-    description: string;
-    tools: string[];
-}): string[];

package/dist/utils/subAgentConfig.js

@@ -1,221 +0,0 @@
-import { existsSync, readFileSync, writeFileSync, mkdirSync } from 'fs';
-import { join } from 'path';
-import { homedir } from 'os';
-const CONFIG_DIR = join(homedir(), '.snow');
-const SUB_AGENTS_CONFIG_FILE = join(CONFIG_DIR, 'sub-agents.json');
-/**
- * Built-in sub-agents (hardcoded, always available)
- */
-const BUILTIN_AGENTS = [
-    {
-        id: 'agent_explore',
-        name: 'Explore Agent',
-        description: 'Specialized for quickly exploring and understanding codebases. Excels at searching code, finding definitions, analyzing code structure and dependencies. Read-only operations.',
-        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.',
-        tools: [
-            'filesystem-read',
-            'ace-find_definition',
-            'ace-find_references',
-            'ace-semantic_search',
-            'ace-text_search',
-            'ace-file_outline',
-            'codebase-search',
-            'websearch-search',
-            'websearch-fetch',
-        ],
-        createdAt: '2024-01-01T00:00:00.000Z',
-        updatedAt: '2024-01-01T00:00:00.000Z',
-        builtin: true,
-    },
-    {
-        id: 'agent_plan',
-        name: 'Plan Agent',
-        description: 'Specialized for planning complex tasks. Analyzes requirements, explores code, identifies relevant files, and creates detailed implementation plans. Read-only operations.',
-        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.',
-        tools: [
-            'filesystem-read',
-            'ace-find_definition',
-            'ace-find_references',
-            'ace-semantic_search',
-            'ace-text_search',
-            'ace-file_outline',
-            'ide-get_diagnostics',
-            'codebase-search',
-            'websearch-search',
-            'websearch-fetch',
-        ],
-        createdAt: '2024-01-01T00:00:00.000Z',
-        updatedAt: '2024-01-01T00:00:00.000Z',
-        builtin: true,
-    },
-    {
-        id: 'agent_general',
-        name: 'General Purpose Agent',
-        description: 'General-purpose multi-step task execution agent. Has complete tool access for searching, modifying files, and executing commands. Best for complex tasks requiring actual 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.',
-        tools: [
-            'filesystem-read',
-            'filesystem-create',
-            'filesystem-edit',
-            'filesystem-edit_search',
-            'terminal-execute',
-            'ace-find_definition',
-            'ace-find_references',
-            'ace-semantic_search',
-            'ace-text_search',
-            'ace-file_outline',
-            'websearch-search',
-            'websearch-fetch',
-            'ide-get_diagnostics',
-            'codebase-search',
-        ],
-        createdAt: '2024-01-01T00:00:00.000Z',
-        updatedAt: '2024-01-01T00:00:00.000Z',
-        builtin: true,
-    },
-];
-function ensureConfigDirectory() {
-    if (!existsSync(CONFIG_DIR)) {
-        mkdirSync(CONFIG_DIR, { recursive: true });
-    }
-}
-function generateId() {
-    return `agent_${Date.now()}_${Math.random().toString(36).substring(2, 9)}`;
-}
-/**
- * Get user-configured sub-agents only (exported for MCP tool generation)
- */
-export function getUserSubAgents() {
-    try {
-        ensureConfigDirectory();
-        if (!existsSync(SUB_AGENTS_CONFIG_FILE)) {
-            return [];
-        }
-        const configData = readFileSync(SUB_AGENTS_CONFIG_FILE, 'utf8');
-        const config = JSON.parse(configData);
-        return config.agents || [];
-    }
-    catch (error) {
-        console.error('Failed to load sub-agents:', error);
-        return [];
-    }
-}
-/**
- * Get all sub-agents (built-in + user-configured)
- */
-export function getSubAgents() {
-    const userAgents = getUserSubAgents();
-    // Return built-in agents first, then user-configured agents
-    return [...BUILTIN_AGENTS, ...userAgents];
-}
-/**
- * Get a sub-agent by ID (checks both built-in and user-configured)
- */
-export function getSubAgent(id) {
-    const agents = getSubAgents();
-    return agents.find(agent => agent.id === id) || null;
-}
-/**
- * Save user-configured sub-agents only (never saves built-in agents)
- */
-function saveSubAgents(agents) {
-    try {
-        ensureConfigDirectory();
-        // Filter out built-in agents (should never be saved to config)
-        const userAgents = agents.filter(agent => !agent.builtin);
-        const config = { agents: userAgents };
-        const configData = JSON.stringify(config, null, 2);
-        writeFileSync(SUB_AGENTS_CONFIG_FILE, configData, 'utf8');
-    }
-    catch (error) {
-        throw new Error(`Failed to save sub-agents: ${error}`);
-    }
-}
-/**
- * Create a new sub-agent (user-configured only)
- */
-export function createSubAgent(name, description, tools, role) {
-    const userAgents = getUserSubAgents();
-    const now = new Date().toISOString();
-    const newAgent = {
-        id: generateId(),
-        name,
-        description,
-        role,
-        tools,
-        createdAt: now,
-        updatedAt: now,
-        builtin: false,
-    };
-    userAgents.push(newAgent);
-    saveSubAgents(userAgents);
-    return newAgent;
-}
-/**
- * Update an existing sub-agent (only user-configured agents can be updated)
- */
-export function updateSubAgent(id, updates) {
-    // Prevent updating built-in agents
-    const agent = getSubAgent(id);
-    if (agent?.builtin) {
-        throw new Error('Cannot update built-in agents');
-    }
-    const userAgents = getUserSubAgents();
-    const index = userAgents.findIndex(agent => agent.id === id);
-    if (index === -1) {
-        return null;
-    }
-    const existingAgent = userAgents[index];
-    if (!existingAgent) {
-        return null;
-    }
-    const updatedAgent = {
-        id: existingAgent.id,
-        name: updates.name ?? existingAgent.name,
-        description: updates.description ?? existingAgent.description,
-        role: updates.role ?? existingAgent.role,
-        tools: updates.tools ?? existingAgent.tools,
-        createdAt: existingAgent.createdAt,
-        updatedAt: new Date().toISOString(),
-        builtin: false,
-    };
-    userAgents[index] = updatedAgent;
-    saveSubAgents(userAgents);
-    return updatedAgent;
-}
-/**
- * Delete a sub-agent (only user-configured agents can be deleted)
- */
-export function deleteSubAgent(id) {
-    // Prevent deleting built-in agents
-    const agent = getSubAgent(id);
-    if (agent?.builtin) {
-        throw new Error('Cannot delete built-in agents');
-    }
-    const userAgents = getUserSubAgents();
-    const filteredAgents = userAgents.filter(agent => agent.id !== id);
-    if (filteredAgents.length === userAgents.length) {
-        return false; // Agent not found
-    }
-    saveSubAgents(filteredAgents);
-    return true;
-}
-/**
- * Validate sub-agent data
- */
-export function validateSubAgent(data) {
-    const errors = [];
-    if (!data.name || data.name.trim().length === 0) {
-        errors.push('Agent name is required');
-    }
-    if (data.name && data.name.length > 100) {
-        errors.push('Agent name must be less than 100 characters');
-    }
-    if (data.description && data.description.length > 500) {
-        errors.push('Description must be less than 500 characters');
-    }
-    if (!data.tools || data.tools.length === 0) {
-        errors.push('At least one tool must be selected');
-    }
-    return errors;
-}

package/dist/utils/subAgentExecutor.d.ts

@@ -1,40 +0,0 @@
-import type { ConfirmationResult } from '../ui/components/ToolConfirmation.js';
-export interface SubAgentMessage {
-    type: 'sub_agent_message';
-    agentId: string;
-    agentName: string;
-    message: any;
-}
-export interface TokenUsage {
-    inputTokens: number;
-    outputTokens: number;
-    cacheCreationInputTokens?: number;
-    cacheReadInputTokens?: number;
-}
-export interface SubAgentResult {
-    success: boolean;
-    result: string;
-    error?: string;
-    usage?: TokenUsage;
-}
-export interface ToolConfirmationCallback {
-    (toolName: string, toolArgs: any): Promise<ConfirmationResult>;
-}
-export interface ToolApprovalChecker {
-    (toolName: string): boolean;
-}
-export interface AddToAlwaysApprovedCallback {
-    (toolName: string): void;
-}
-/**
- * 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 declare function executeSubAgent(agentId: string, prompt: string, onMessage?: (message: SubAgentMessage) => void, abortSignal?: AbortSignal, requestToolConfirmation?: ToolConfirmationCallback, isToolAutoApproved?: ToolApprovalChecker, yoloMode?: boolean, addToAlwaysApproved?: AddToAlwaysApprovedCallback): Promise<SubAgentResult>;