snow-ai 0.2.11 → 0.2.13

package/dist/ui/pages/SystemPromptConfigScreen.js

@@ -4,7 +4,6 @@ import { spawn } from 'child_process';
 import { writeFileSync, readFileSync, existsSync, mkdirSync } from 'fs';
 import { join } from 'path';
 import { homedir, platform } from 'os';
-import { getOpenAiConfig, updateOpenAiConfig, } from '../../utils/apiConfig.js';
 import { SYSTEM_PROMPT } from '../../api/systemPrompt.js';
 const CONFIG_DIR = join(homedir(), '.snow');
 const SYSTEM_PROMPT_FILE = join(CONFIG_DIR, 'system-prompt.txt');
@@ -24,9 +23,17 @@ export default function SystemPromptConfigScreen({ onBack }) {
     useEffect(() => {
         const openEditor = async () => {
             ensureConfigDirectory();
-            // 读取当前配置的自定义系统提示词，如果为空则使用默认系统提示词
-            const config = getOpenAiConfig();
-            const currentPrompt = config.systemPrompt || SYSTEM_PROMPT;
+            // 读取系统提示词文件，如果不存在则使用默认系统提示词
+            let currentPrompt = SYSTEM_PROMPT;
+            if (existsSync(SYSTEM_PROMPT_FILE)) {
+                try {
+                    currentPrompt = readFileSync(SYSTEM_PROMPT_FILE, 'utf8');
+                }
+                catch {
+                    // 读取失败，使用默认
+                    currentPrompt = SYSTEM_PROMPT;
+                }
+            }
             // 写入临时文件供编辑
             writeFileSync(SYSTEM_PROMPT_FILE, currentPrompt, 'utf8');
             const editor = getSystemEditor();
@@ -39,22 +46,28 @@ export default function SystemPromptConfigScreen({ onBack }) {
                 if (existsSync(SYSTEM_PROMPT_FILE)) {
                     try {
                         const editedContent = readFileSync(SYSTEM_PROMPT_FILE, 'utf8');
-                        // 如果编辑后的内容为空或与默认提示词相同，则保存为空（使用默认）
-                        // 否则保存自定义提示词
                         const trimmedContent = editedContent.trim();
                         if (trimmedContent === '' || trimmedContent === SYSTEM_PROMPT.trim()) {
-                            // 保存为空，表示使用默认提示词
-                            updateOpenAiConfig({ systemPrompt: undefined });
-                            console.log('System prompt reset to default. Please use `snow` to restart!');
+                            // 内容为空或与默认相同，删除文件，使用默认提示词
+                            try {
+                                const fs = require('fs');
+                                fs.unlinkSync(SYSTEM_PROMPT_FILE);
+                                console.log('System prompt reset to default. Please use `snow` to restart!');
+                            }
+                            catch {
+                                // 删除失败，保存空内容
+                                writeFileSync(SYSTEM_PROMPT_FILE, '', 'utf8');
+                                console.log('System prompt reset to default. Please use `snow` to restart!');
+                            }
                         }
                         else {
-                            // 保存自定义提示词
-                            updateOpenAiConfig({ systemPrompt: editedContent });
+                            // 保存自定义提示词到文件
+                            writeFileSync(SYSTEM_PROMPT_FILE, editedContent, 'utf8');
                             console.log('Custom system prompt saved successfully! Please use `snow` to restart!');
                         }
                     }
                     catch (error) {
-                        console.error('Failed to read edited content:', error instanceof Error ? error.message : 'Unknown error');
+                        console.error('Failed to save system prompt:', error instanceof Error ? error.message : 'Unknown error');
                     }
                 }
                 process.exit(0);

package/dist/utils/apiConfig.d.ts

@@ -14,7 +14,6 @@ export interface ApiConfig {
     maxTokens?: number;
     compactModel?: CompactModelConfig;
     anthropicBeta?: boolean;
-    systemPrompt?: string;
 }
 export interface MCPServer {
     url?: string;
@@ -37,3 +36,9 @@ export declare function validateApiConfig(config: Partial<ApiConfig>): string[];
 export declare function updateMCPConfig(mcpConfig: MCPConfig): void;
 export declare function getMCPConfig(): MCPConfig;
 export declare function validateMCPConfig(config: Partial<MCPConfig>): string[];
+/**
+ * 读取自定义系统提示词
+ * 如果 system-prompt.txt 文件存在且不为空，返回其内容
+ * 否则返回 undefined (使用默认系统提示词)
+ */
+export declare function getCustomSystemPrompt(): string | undefined;

package/dist/utils/apiConfig.js

@@ -17,6 +17,7 @@ const DEFAULT_MCP_CONFIG = {
     mcpServers: {},
 };
 const CONFIG_DIR = join(homedir(), '.snow');
+const SYSTEM_PROMPT_FILE = join(CONFIG_DIR, 'system-prompt.txt');
 function normalizeRequestMethod(method) {
     if (method === 'chat' || method === 'responses' || method === 'gemini' || method === 'anthropic') {
         return method;
@@ -192,3 +193,26 @@ export function validateMCPConfig(config) {
     }
     return errors;
 }
+/**
+ * 读取自定义系统提示词
+ * 如果 system-prompt.txt 文件存在且不为空，返回其内容
+ * 否则返回 undefined (使用默认系统提示词)
+ */
+export function getCustomSystemPrompt() {
+    ensureConfigDirectory();
+    if (!existsSync(SYSTEM_PROMPT_FILE)) {
+        return undefined;
+    }
+    try {
+        const content = readFileSync(SYSTEM_PROMPT_FILE, 'utf8');
+        // 只有当文件完全为空时才返回 undefined
+        if (content.length === 0) {
+            return undefined;
+        }
+        // 返回原始内容，不做任何处理
+        return content;
+    }
+    catch {
+        return undefined;
+    }
+}

package/dist/utils/commands/ide.js

@@ -3,6 +3,7 @@ import { vscodeConnection } from '../vscodeConnection.js';
 // IDE connection command handler
 registerCommand('ide', {
     execute: async () => {
+        // Check if already connected
         if (vscodeConnection.isConnected()) {
             return {
                 success: true,
@@ -10,15 +11,31 @@ registerCommand('ide', {
                 message: 'Already connected to VSCode editor'
             };
         }
+        // Check if server is already running (but not connected yet)
+        if (vscodeConnection.isServerRunning()) {
+            return {
+                success: true,
+                action: 'info',
+                message: `VSCode connection server is already running on port ${vscodeConnection.getPort()}\nWaiting for VSCode extension to connect...`
+            };
+        }
+        // Start the server
         try {
             await vscodeConnection.start();
             return {
                 success: true,
                 action: 'info',
-                message: `VSCode connection server started on port ${vscodeConnection.getPort()}\nPlease connect from the Snow CLI extension in VSCode`
+                message: `VSCode connection server started on port ${vscodeConnection.getPort()}\nWaiting for VSCode extension to connect...`
             };
         }
         catch (error) {
+            // Handle EADDRINUSE error specifically
+            if (error instanceof Error && 'code' in error && error.code === 'EADDRINUSE') {
+                return {
+                    success: false,
+                    message: `Port ${vscodeConnection.getPort()} is already in use. Please restart Snow CLI to reset the connection.`
+                };
+            }
             return {
                 success: false,
                 message: error instanceof Error ? error.message : 'Failed to start IDE connection'

package/dist/utils/mcpToolsManager.d.ts

@@ -1,6 +1,6 @@
 import { TodoService } from '../mcp/todo.js';
 export interface MCPTool {
-    type: "function";
+    type: 'function';
     function: {
         name: string;
         description: string;

package/dist/utils/mcpToolsManager.js

@@ -42,7 +42,7 @@ function isCacheValid() {
     if (!toolsCache)
         return false;
     const now = Date.now();
-    const isExpired = (now - toolsCache.lastUpdate) > CACHE_DURATION;
+    const isExpired = now - toolsCache.lastUpdate > CACHE_DURATION;
     const configChanged = toolsCache.configHash !== generateConfigHash();
     return !isExpired && !configChanged;
 }
@@ -66,44 +66,44 @@ async function refreshToolsCache() {
     const filesystemServiceTools = filesystemTools.map(tool => ({
         name: tool.name.replace('filesystem_', ''),
         description: tool.description,
-        inputSchema: tool.inputSchema
+        inputSchema: tool.inputSchema,
     }));
     servicesInfo.push({
         serviceName: 'filesystem',
         tools: filesystemServiceTools,
         isBuiltIn: true,
-        connected: true
+        connected: true,
     });
     for (const tool of filesystemTools) {
         allTools.push({
-            type: "function",
+            type: 'function',
             function: {
                 name: `filesystem-${tool.name.replace('filesystem_', '')}`,
                 description: tool.description,
-                parameters: tool.inputSchema
-            }
+                parameters: tool.inputSchema,
+            },
         });
     }
     // Add built-in terminal tools (always available)
     const terminalServiceTools = terminalTools.map(tool => ({
         name: tool.name.replace('terminal_', ''),
         description: tool.description,
-        inputSchema: tool.inputSchema
+        inputSchema: tool.inputSchema,
     }));
     servicesInfo.push({
         serviceName: 'terminal',
         tools: terminalServiceTools,
         isBuiltIn: true,
-        connected: true
+        connected: true,
     });
     for (const tool of terminalTools) {
         allTools.push({
-            type: "function",
+            type: 'function',
             function: {
                 name: `terminal-${tool.name.replace('terminal_', '')}`,
                 description: tool.description,
-                parameters: tool.inputSchema
-            }
+                parameters: tool.inputSchema,
+            },
         });
     }
     // Add built-in TODO tools (always available)
@@ -112,22 +112,22 @@ async function refreshToolsCache() {
     const todoServiceTools = todoTools.map(tool => ({
         name: tool.name.replace('todo-', ''),
         description: tool.description || '',
-        inputSchema: tool.inputSchema
+        inputSchema: tool.inputSchema,
     }));
     servicesInfo.push({
         serviceName: 'todo',
         tools: todoServiceTools,
         isBuiltIn: true,
-        connected: true
+        connected: true,
     });
     for (const tool of todoTools) {
         allTools.push({
-            type: "function",
+            type: 'function',
             function: {
                 name: tool.name,
                 description: tool.description || '',
-                parameters: tool.inputSchema
-            }
+                parameters: tool.inputSchema,
+            },
         });
     }
     // Add user-configured MCP server tools (probe for availability but don't maintain connections)
@@ -140,16 +140,16 @@ async function refreshToolsCache() {
                     serviceName,
                     tools: serviceTools,
                     isBuiltIn: false,
-                    connected: true
+                    connected: true,
                 });
                 for (const tool of serviceTools) {
                     allTools.push({
-                        type: "function",
+                        type: 'function',
                         function: {
                             name: `${serviceName}-${tool.name}`,
                             description: tool.description,
-                            parameters: tool.inputSchema
-                        }
+                            parameters: tool.inputSchema,
+                        },
                     });
                 }
             }
@@ -159,7 +159,7 @@ async function refreshToolsCache() {
                     tools: [],
                     isBuiltIn: false,
                     connected: false,
-                    error: error instanceof Error ? error.message : 'Unknown error'
+                    error: error instanceof Error ? error.message : 'Unknown error',
                 });
             }
         }
@@ -172,7 +172,7 @@ async function refreshToolsCache() {
         tools: allTools,
         servicesInfo,
         lastUpdate: Date.now(),
-        configHash: generateConfigHash()
+        configHash: generateConfigHash(),
     };
 }
 /**
@@ -226,7 +226,7 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
             name: `snow-cli-${serviceName}`,
             version: '1.0.0',
         }, {
-            capabilities: {}
+            capabilities: {},
         });
         // Create transport based on server configuration
         if (server.url) {
@@ -260,11 +260,11 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
                     }
                 }
                 transport = new StreamableHTTPClientTransport(url, {
-                    requestInit: { headers }
+                    requestInit: { headers },
                 });
                 await Promise.race([
                     client.connect(transport),
-                    new Promise((_, reject) => setTimeout(() => reject(new Error('HTTP connection timeout')), timeoutMs))
+                    new Promise((_, reject) => setTimeout(() => reject(new Error('HTTP connection timeout')), timeoutMs)),
                 ]);
             }
             catch (httpError) {
@@ -277,12 +277,12 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
                     name: `snow-cli-${serviceName}`,
                     version: '1.0.0',
                 }, {
-                    capabilities: {}
+                    capabilities: {},
                 });
                 transport = new SSEClientTransport(url);
                 await Promise.race([
                     client.connect(transport),
-                    new Promise((_, reject) => setTimeout(() => reject(new Error('SSE connection timeout')), timeoutMs))
+                    new Promise((_, reject) => setTimeout(() => reject(new Error('SSE connection timeout')), timeoutMs)),
                 ]);
             }
         }
@@ -309,13 +309,13 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
         // Get tools from the service
         const toolsResult = await Promise.race([
             client.listTools(),
-            new Promise((_, reject) => setTimeout(() => reject(new Error('ListTools timeout')), timeoutMs))
+            new Promise((_, reject) => setTimeout(() => reject(new Error('ListTools timeout')), timeoutMs)),
         ]);
-        return toolsResult.tools?.map(tool => ({
+        return (toolsResult.tools?.map(tool => ({
             name: tool.name,
             description: tool.description || '',
-            inputSchema: tool.inputSchema
-        })) || [];
+            inputSchema: tool.inputSchema,
+        })) || []);
     }
     finally {
         try {
@@ -391,8 +391,15 @@ export async function executeMCPTool(toolName, args) {
                 return await filesystemService.getFileInfo(args.filePath);
             case 'edit':
                 return await filesystemService.editFile(args.filePath, args.startLine, args.endLine, args.newContent, args.contextLines);
-            case 'search':
-                return await filesystemService.searchCode(args.query, args.dirPath, args.fileExtensions, args.caseSensitive, args.maxResults);
+            case 'search': {
+                // 兼容性处理：如果 searchMode 不存在或无效，默认使用 'text'
+                const validSearchModes = ['text', 'regex', 'ast'];
+                let searchMode = 'text';
+                if (args.searchMode && validSearchModes.includes(args.searchMode)) {
+                    searchMode = args.searchMode;
+                }
+                return await filesystemService.searchCode(args.query, args.dirPath, args.fileExtensions, args.caseSensitive, args.maxResults, searchMode);
+            }
             default:
                 throw new Error(`Unknown filesystem tool: ${actualToolName}`);
         }
@@ -430,7 +437,7 @@ async function executeOnExternalMCPService(serviceName, server, toolName, args)
             name: `snow-cli-${serviceName}`,
             version: '1.0.0',
         }, {
-            capabilities: {}
+            capabilities: {},
         });
         // Setup transport (similar to getServiceTools)
         let transport;
@@ -450,7 +457,9 @@ async function executeOnExternalMCPService(serviceName, server, toolName, args)
             transport = new StdioClientTransport({
                 command: server.command,
                 args: server.args || [],
-                env: server.env ? { ...process.env, ...server.env } : process.env,
+                env: server.env
+                    ? { ...process.env, ...server.env }
+                    : process.env,
             });
         }
         await client.connect(transport);
@@ -458,7 +467,7 @@ async function executeOnExternalMCPService(serviceName, server, toolName, args)
         // Execute the tool with the original tool name (not prefixed)
         const result = await client.callTool({
             name: toolName,
-            arguments: args
+            arguments: args,
         });
         logger.debug(`result from ${serviceName} tool ${toolName}:`, result);
         return result.content;

package/dist/utils/textBuffer.d.ts

@@ -28,11 +28,16 @@ export declare class TextBuffer {
     private pasteTimer;
     private pastePlaceholderPosition;
     private onUpdateCallback?;
+    private isDestroyed;
     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;
     /**
      * 获取完整文本，包括替换占位符为原始内容

package/dist/utils/textBuffer.js

@@ -79,6 +79,12 @@ export class TextBuffer {
             writable: true,
             value: void 0
         }); // 更新回调函数
+        Object.defineProperty(this, "isDestroyed", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        }); // 标记是否已销毁
         Object.defineProperty(this, "visualLines", {
             enumerable: true,
             configurable: true,
@@ -107,6 +113,19 @@ export class TextBuffer {
         this.onUpdateCallback = onUpdate;
         this.recalculateVisualState();
     }
+    /**
+     * Cleanup method to be called when the buffer is no longer needed
+     */
+    destroy() {
+        this.isDestroyed = true;
+        if (this.pasteTimer) {
+            clearTimeout(this.pasteTimer);
+            this.pasteTimer = null;
+        }
+        this.pasteStorage.clear();
+        this.imageStorage.clear();
+        this.onUpdateCallback = undefined;
+    }
     get text() {
         return this.content;
     }
@@ -136,7 +155,7 @@ export class TextBuffer {
     }
     scheduleUpdate() {
         // Notify external components of updates
-        if (this.onUpdateCallback) {
+        if (!this.isDestroyed && this.onUpdateCallback) {
             this.onUpdateCallback();
         }
     }
@@ -191,7 +210,9 @@ export class TextBuffer {
             this.cursorIndex = this.pastePlaceholderPosition + cpLen(tempPlaceholder);
             // 设置150ms的定时器，如果150ms内没有新数据，则认为粘贴完成
             this.pasteTimer = setTimeout(() => {
-                this.finalizePaste();
+                if (!this.isDestroyed) {
+                    this.finalizePaste();
+                }
             }, 150);
             this.recalculateVisualState();
             this.scheduleUpdate();

package/dist/utils/vscodeConnection.js

@@ -41,6 +41,10 @@ class VSCodeConnectionManager {
             try {
                 this.server = new WebSocketServer({ port: this.port });
                 this.server.on('connection', (ws) => {
+                    // Close old client if exists
+                    if (this.client && this.client !== ws) {
+                        this.client.close();
+                    }
                     this.client = ws;
                     ws.on('message', (message) => {
                         try {
@@ -52,7 +56,12 @@ class VSCodeConnectionManager {
                         }
                     });
                     ws.on('close', () => {
-                        this.client = null;
+                        if (this.client === ws) {
+                            this.client = null;
+                        }
+                    });
+                    ws.on('error', () => {
+                        // Silently handle errors
                     });
                 });
                 this.server.on('listening', () => {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "snow-ai",
-	"version": "0.2.11",
+	"version": "0.2.13",
 	"description": "Intelligent Command Line Assistant powered by AI",
 	"license": "MIT",
 	"bin": {
@@ -40,7 +40,7 @@
 	],
 	"dependencies": {
 		"@anthropic-ai/sdk": "^0.65.0",
-		"@google/generative-ai": "^0.24.1",
+		"@google/genai": "^1.23.0",
 		"@inkjs/ui": "^2.0.0",
 		"@modelcontextprotocol/sdk": "^1.17.3",
 		"chalk-template": "^1.1.2",
@@ -59,6 +59,18 @@
 		"react": "^18.2.0",
 		"string-width": "^7.2.0",
 		"tiktoken": "^1.0.22",
+		"tree-sitter": "^0.25.0",
+		"tree-sitter-c": "^0.24.1",
+		"tree-sitter-c-sharp": "^0.23.1",
+		"tree-sitter-cpp": "^0.23.4",
+		"tree-sitter-go": "^0.25.0",
+		"tree-sitter-java": "^0.23.5",
+		"tree-sitter-javascript": "^0.25.0",
+		"tree-sitter-php": "^0.24.2",
+		"tree-sitter-python": "^0.25.0",
+		"tree-sitter-ruby": "^0.23.1",
+		"tree-sitter-rust": "^0.24.0",
+		"tree-sitter-typescript": "^0.23.2",
 		"ws": "^8.14.2"
 	},
 	"devDependencies": {

package/readme.md

@@ -1,6 +1,17 @@
+<div align="center">
+
+<img src="logo.png" alt="Snow AI CLI Logo" width="200"/>
+
 # snow-ai
 
-> This readme is automatically generated by [create-ink-app](https://github.com/vadimdemedes/create-ink-app)
+**English** | [中文](readme_zh.md)
+
+*An intelligent AI-powered CLI tool for developers*
+
+</div>
+
+---
+
 
 ## Install
 
@@ -21,15 +32,17 @@ $ snow --update
 ## Config example  `./User/.snow/config.json`
 ```json
 {
-  "openai": {
-    "baseUrl": "https://api.openai.com/v1",
+  "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": 200000,
+    "maxContextTokens": 32000, //The maximum context length of the model
+    "maxTokens": 4096, // The maximum generation length of the model
+    "anthropicBeta": false,
     "compactModel": {
-      "baseUrl": "https://api.openai.com/v1",
+      "baseUrl": "https://api.opeai.com/v1",
       "apiKey": "your-api-key",
       "modelName": "gpt-4.1-mini"
     }
@@ -56,7 +69,24 @@ $ npm uninstall --global snow-ai
 * **Agent**
 
 ![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
 
 * **Commands**
 
-![alt text](image-2.png)
+![alt text](image-2.png)
+  - /clear - Create a new session
+
+  - /resume - The recovery history has
+
+  - /mcp - Check the status of MCP service
+
+  - /yolo - Unattended mode, all tools automatically agree to execute
+
+  - /init - Initialize the project and generate the SNOW.md description document
+
+  - /ide - Connect to VSCode, you need to install the plug-in
+
+  - /compact - compress the context into a sentence
+