snow-ai 0.3.2 → 0.3.4

package/dist/utils/sessionManager.js

@@ -2,6 +2,8 @@ import fs from 'fs/promises';
 import path from 'path';
 import os from 'os';
 import { randomUUID } from 'crypto';
+import { summaryAgent } from '../agents/summaryAgent.js';
+import { getTodoService } from './mcpToolsManager.js';
 class SessionManager {
     constructor() {
         Object.defineProperty(this, "sessionsDir", {
@@ -16,21 +18,61 @@ class SessionManager {
             writable: true,
             value: null
         });
+        Object.defineProperty(this, "summaryAbortController", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
+        Object.defineProperty(this, "summaryTimeoutId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
         this.sessionsDir = path.join(os.homedir(), '.snow', 'sessions');
     }
-    async ensureSessionsDir() {
+    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) {
-        return path.join(this.sessionsDir, `${sessionId}.json`);
+    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}`;
+    }
+    /**
+     * Cancel any ongoing summary generation
+     * This prevents wasted resources and race conditions
+     */
+    cancelOngoingSummaryGeneration() {
+        if (this.summaryAbortController) {
+            this.summaryAbortController.abort();
+            this.summaryAbortController = null;
+        }
+        if (this.summaryTimeoutId) {
+            clearTimeout(this.summaryTimeoutId);
+            this.summaryTimeoutId = null;
+        }
     }
     async createNewSession() {
-        await this.ensureSessionsDir();
+        await this.ensureSessionsDir(new Date());
         // 使用 UUID v4 生成唯一会话 ID，避免并发冲突
         const sessionId = randomUUID();
         const session = {
@@ -40,39 +82,86 @@ class SessionManager {
             createdAt: Date.now(),
             updatedAt: Date.now(),
             messages: [],
-            messageCount: 0
+            messageCount: 0,
         };
         this.currentSession = session;
         await this.saveSession(session);
         return session;
     }
     async saveSession(session) {
-        await this.ensureSessionsDir();
-        const sessionPath = this.getSessionPath(session.id);
+        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 sessionPath = this.getSessionPath(sessionId);
-            const data = await fs.readFile(sessionPath, 'utf-8');
+            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) {
-            return null;
+            // 旧格式不存在，搜索日期文件夹
+        }
+        // 在日期文件夹中查找会话
+        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);
-            const sessions = [];
             for (const file of files) {
-                if (file.endsWith('.json')) {
+                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 sessionPath = path.join(this.sessionsDir, file);
-                        const data = await fs.readFile(sessionPath, 'utf-8');
+                        const data = await fs.readFile(filePath, 'utf-8');
                         const session = JSON.parse(data);
                         sessions.push({
                             id: session.id,
@@ -80,7 +169,7 @@ class SessionManager {
                             summary: session.summary,
                             createdAt: session.createdAt,
                             updatedAt: session.updatedAt,
-                            messageCount: session.messageCount
+                            messageCount: session.messageCount,
                         });
                     }
                     catch (error) {
@@ -96,6 +185,35 @@ class SessionManager {
             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: 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();
@@ -142,10 +260,59 @@ class SessionManager {
         this.currentSession.messages.push(message);
         this.currentSession.messageCount = this.currentSession.messages.length;
         this.currentSession.updatedAt = Date.now();
-        // Simple title generation from first user message (no API call)
+        // Generate summary from first user message using summaryAgent (parallel, non-blocking)
         if (this.currentSession.messageCount === 1 && message.role === 'user') {
+            // Set temporary title immediately (synchronous)
             this.currentSession.title = message.content.slice(0, 50);
             this.currentSession.summary = message.content.slice(0, 100);
+            // Cancel any previous summary generation (防呆机制)
+            this.cancelOngoingSummaryGeneration();
+            // Create new AbortController for this summary generation
+            this.summaryAbortController = new AbortController();
+            const currentSessionId = this.currentSession.id;
+            const abortSignal = this.summaryAbortController.signal;
+            // Set timeout to cancel summary generation after 30 seconds (防呆机制)
+            this.summaryTimeoutId = setTimeout(() => {
+                if (this.summaryAbortController) {
+                    console.warn('Summary generation timeout after 30s, aborting...');
+                    this.summaryAbortController.abort();
+                    this.summaryAbortController = null;
+                }
+            }, 30000);
+            // Generate better summary in parallel (non-blocking)
+            // This won't delay the main conversation flow
+            summaryAgent
+                .generateSummary(message.content, abortSignal)
+                .then(summary => {
+                // 防呆检查：确保会话没有被切换，且仍然是第一条消息
+                if (this.currentSession &&
+                    this.currentSession.id === currentSessionId &&
+                    this.currentSession.messageCount === 1) {
+                    // Only update if this is still the first message in the same session
+                    this.currentSession.title = summary;
+                    this.currentSession.summary = summary;
+                    this.saveSession(this.currentSession).catch(error => {
+                        console.error('Failed to save session with generated summary:', error);
+                    });
+                }
+                // Clean up
+                this.cancelOngoingSummaryGeneration();
+            })
+                .catch(error => {
+                // Clean up on error
+                this.cancelOngoingSummaryGeneration();
+                // Silently fail if aborted (expected behavior)
+                if (error.name === 'AbortError' || abortSignal.aborted) {
+                    console.log('Summary generation cancelled (expected)');
+                    return;
+                }
+                // Log other errors - we already have a fallback title/summary
+                console.warn('Summary generation failed, using fallback:', error);
+            });
+        }
+        else if (this.currentSession.messageCount > 1) {
+            // 防呆机制：如果不是第一条消息，取消任何正在进行的摘要生成
+            this.cancelOngoingSummaryGeneration();
         }
         await this.saveSession(this.currentSession);
     }
@@ -153,20 +320,64 @@ class SessionManager {
         return this.currentSession;
     }
     setCurrentSession(session) {
+        // 防呆机制：切换会话时取消正在进行的摘要生成
+        this.cancelOngoingSummaryGeneration();
         this.currentSession = session;
     }
     clearCurrentSession() {
+        // 防呆机制：清除会话时取消正在进行的摘要生成
+        this.cancelOngoingSummaryGeneration();
         this.currentSession = null;
     }
     async deleteSession(sessionId) {
+        let sessionDeleted = false;
+        // 首先尝试删除旧格式（向下兼容）
         try {
-            const sessionPath = this.getSessionPath(sessionId);
-            await fs.unlink(sessionPath);
-            return true;
+            const oldSessionPath = path.join(this.sessionsDir, `${sessionId}.json`);
+            await fs.unlink(oldSessionPath);
+            sessionDeleted = true;
         }
         catch (error) {
-            return false;
+            // 旧格式不存在，搜索日期文件夹
+        }
+        // 在日期文件夹中查找并删除会话
+        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删除失败不影响会话删除结果
+                console.warn(`Failed to delete TODO list for session ${sessionId}:`, error);
+            }
         }
+        return sessionDeleted;
     }
     async truncateMessages(messageCount) {
         if (!this.currentSession) {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "snow-ai",
-	"version": "0.3.2",
+	"version": "0.3.4",
 	"description": "Intelligent Command Line Assistant powered by AI",
 	"license": "MIT",
 	"bin": {

package/readme.md

@@ -6,97 +6,118 @@
 
 **English** | [中文](readme_zh.md)
 
-*An intelligent AI-powered CLI tool for developers*
+_An intelligent AI-powered CLI tool for developers_
 
 </div>
 
 ---
 
-
-## Install
+## Installation
 
 ```bash
-$ npm install --global snow-ai
+$ npm install -g snow-ai
 ```
 
-## Start
-```bash
-$ snow
-```
+You can also clone and build from source: https://github.com/MayDay-wpf/snow-cli
 
-## Update
-```bash
-$ snow --update
-```
+### Install VSCode Extension
 
-## Config example  `./User/.snow/config.json`
-```json
-{
-  "snowcfg": {
-    "baseUrl": "https://api.openai.com/v1",//Gemini：https://generativelanguage.googleapis.com Anthropic：https://api.anthropic.com
-    "apiKey": "your-api-key",
-    "requestMethod": "responses",
-    "advancedModel": "gpt-5-codex",
-    "basicModel": "gpt-5-codex",
-    "maxContextTokens": 32000, //The maximum context length of the model
-    "maxTokens": 4096, // The maximum generation length of the model
-    "anthropicBeta": false,
-    "compactModel": {
-      "baseUrl": "https://api.opeai.com/v1",
-      "apiKey": "your-api-key",
-      "modelName": "gpt-4.1-mini"
-    }
-  }
-}
-```
+- Download [VSIX/snow-cli-x.x.x.vsix](https://github.com/MayDay-wpf/snow-cli/blob/main/VSIX/)
 
-## Uninstall
-```bash
-$ npm uninstall --global snow-ai
-```
+- Open VSCode, click `Extensions` -> `Install from VSIX...` -> select `snow-cli-0.2.6.vsix`
 
-## Install VSCode Extension
+### Install JetBrains Plugin
 
-* download [VSIX/snow-cli-x.x.x.vsix](https://github.com/MayDay-wpf/snow-cli/blob/main/VSIX/)
+- Download [JetBrains/build/distributions](https://github.com/MayDay-wpf/snow-cli/tree/main/JetBrains/build/distributions)
 
-* open VSCode, click `Extensions` -> `Install from VSIX...` -> select `snow-cli-0.2.6.vsix`
+## Available Commands
 
-## Install JetBrains plugin
+- **Start**: `$ snow`
+- **Update**: `$ snow --update`
+- **Version**: `$ snow --version`
+- **Resume**: `$ snow -c` - Restore the latest conversation history (fully compatible with Claude Code)
 
-* download [JetBrains/build/distributions](https://github.com/MayDay-wpf/snow-cli/tree/main/JetBrains/build/distributions)
+## API & Model Settings
 
-* File > Settings > Plugins
+In version `v0.3.2` and later, all official SDKs have been removed (they were too heavy), so the configuration is slightly different. After starting, enter `API & Model Settings` to see the following options:
 
-## Live View
-* **Welcome & Settings**
+- **Profile** - Switch or create new configurations. Snow now supports saving multiple API and model schemes
+- **Base URL** - Request endpoint. Since official SDKs were removed, OpenAI and Anthropic require `/v1` suffix, Gemini requires `/v1beta`
+- **API Key** - Your API key
+- **Request Method** - Choose based on your needs: `Chat Completions`, `Responses`, `Gemini`, or `Anthropic`
+- **Anthropic Beta** - When checked, Anthropic requests will automatically include `beta=true` parameter
+- **Advanced Model**, **Basic Model**, **Compact Model** - Set the high-performance model for tasks, small model for summarization, and compact model for context compression. All three models use the configured `BaseURL` and `API Key`. The system automatically fetches available models from the `/models` endpoint with filtering support. For APIs with incomplete model lists, use `Manual Input (Enter model name)` to specify the model name
+- **Max Context Tokens** - The model's maximum context window, used for calculating context percentage. For example, Gemini typically has 1M context, so enter `1000000`. This parameter only affects UI calculations, not actual model context
+- **Max Tokens** - This is critical and will be directly added to API requests as the `max_tokens` parameter
 
 ![alt text](image.png)
 
-* **Agent**
+## Proxy & Browser Settings
+
+Configure system proxy port and search engine for web search. In most cases, this doesn't need modification as the app will automatically use system proxy. The app automatically detects available search engines (Edge/Chrome) unless you've manually changed their installation paths.
 
 ![alt text](image-1.png)
-* In the middle of the conversation: click ESC to stop AI generation
 
-* When mounting: double-click ESC, view the dialogue recorder, select rollback, including file checkpoints
+## System Prompt Settings
 
-* MacOS:`ctrl + v` Paste image
-* Windows:`alt + v` Paste image
+Customize your system prompt. Note that this supplements Snow's built-in system prompt rather than replacing it. When you set a custom system prompt, Snow's default prompt is downgraded to a user message and appended to the first user message. On Windows, the app automatically opens Notepad; on macOS/Linux, it uses the system's default terminal text editor. After editing and saving, Snow will close and prompt you to restart: `Custom system prompt saved successfully! Please use 'snow' to restart!`
 
+## Custom Headers Settings
 
-* **Commands**
+Add custom request headers. Note that you can only add headers, not override Snow's built-in headers.
+
+## MCP Settings
+
+Configure MCP services. The method is identical to setting system prompts, and the JSON format matches Cursor's format.
+
+## Getting Started - Start Conversation
+
+Once everything is configured, enter the conversation page by clicking `Start`.
+
+- If you launch Snow from VSCode or other editors, Snow will automatically connect to the IDE using the `Snow CLI` plugin. You'll see a connection message. The plugins are published online - search for `Snow CLI` in the plugin marketplace to install.
 
 ![alt text](image-2.png)
-  - /clear - Create a new session
 
-  - /resume - The recovery history has
+### File Selection & Commands
+
+- Use `@` to select files. In VSCode, you can also hold `Shift` and drag files for the same effect
+- Use `/` to view available commands:
+  - `/init` - Build project documentation `SNOW.md`
+  - `/clear` - Create a new session
+  - `/resume` - Restore conversation history
+  - `/mcp` - Check MCP connection status and reconnect
+  - `/yolo` - Unattended mode (all tool calls execute without confirmation - use with caution)
+  - `/ide` - Manually connect to IDE (usually automatic if plugin is installed)
+  - `/compact` - Compress context (rarely used as compression reduces AI quality)
+
+### Keyboard Shortcuts
+
+- **Windows**: `Alt+V` - Paste image; **macOS/Linux**: `Ctrl+V` - Paste image (with prompt)
+- `Ctrl+L` - Clear input from cursor position to the left
+- `Ctrl+R` - Clear input from cursor position to the right
+- `Shift+Tab` - Toggle Yolo mode on/off
+- `ESC` - Stop AI generation
+- **Double-click `ESC`** - Rollback conversation (with file checkpoints)
+
+### Token Usage
 
-  - /mcp - Check the status of MCP service
+The input area displays context usage percentage, token count, cache hit tokens, and cache creation tokens.
 
-  - /yolo - Unattended mode, all tools automatically agree to execute
+![alt text](image-3.png)
 
-  - /init - Initialize the project and generate the SNOW.md description document
+## Snow System Files
 
-  - /ide - Connect to VSCode, you need to install the plug-in
+All Snow files are stored in the `.snow` folder in your user directory. Here's what each file/folder contains:
 
-  - /compact - compress the context into a sentence
+![alt text](image-4.png)
 
+- **log** - Runtime logs (not uploaded anywhere, kept locally for debugging). Safe to delete
+- **profiles** - Multiple configuration files for switching between different API/model setups
+- **sessions** - All conversation history (required for `/resume` and other features, not uploaded)
+- **snapshots** - File snapshots before AI edits (used for rollback). Automatic management, no manual intervention needed
+- **todo** - Persisted todo lists from each conversation (prevents AI from forgetting tasks if app exits unexpectedly)
+- **active-profile.txt** - Identifies the currently active profile (for backward compatibility with early versions)
+- **config.json** - Main API configuration file
+- **custom-headers.json** - Custom request headers
+- **mcp-config.json** - MCP service configuration
+- **system-prompt.txt** - Custom system prompt content