snow-ai 0.3.36 → 0.4.0

package/dist/utils/subAgentExecutor.js

@@ -19,21 +19,103 @@ import { sessionManager } from './sessionManager.js';
  */
 export async function executeSubAgent(agentId, prompt, onMessage, abortSignal, requestToolConfirmation, isToolAutoApproved, yoloMode, addToAlwaysApproved) {
     try {
-        // Get sub-agent configuration
-        const agent = getSubAgent(agentId);
-        if (!agent) {
-            return {
-                success: false,
-                result: '',
-                error: `Sub-agent with ID "${agentId}" not found`,
+        // 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 => {
+            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, '-');
@@ -65,6 +147,7 @@ export async function executeSubAgent(agentId, prompt, onMessage, abortSignal, r
         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();
@@ -137,6 +220,33 @@ export async function executeSubAgent(agentId, prompt, onMessage, abortSignal, r
                         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;
                 }
@@ -193,7 +303,9 @@ export async function executeSubAgent(agentId, prompt, onMessage, abortSignal, r
                 if (needsConfirmation && requestToolConfirmation) {
                     // Request confirmation from user
                     const confirmation = await requestToolConfirmation(toolName, args);
-                    if (confirmation === 'reject') {
+                    if (confirmation === 'reject' ||
+                        (typeof confirmation === 'object' &&
+                            confirmation.type === 'reject_with_reply')) {
                         rejectedToolCalls.push(toolCall);
                         continue;
                     }
@@ -309,6 +421,7 @@ export async function executeSubAgent(agentId, prompt, onMessage, abortSignal, r
         return {
             success: true,
             result: finalResponse,
+            usage: totalUsage,
         };
     }
     catch (error) {

package/dist/utils/toolExecutor.d.ts

@@ -1,4 +1,5 @@
 import type { SubAgentMessage } from './subAgentExecutor.js';
+import type { ConfirmationResult } from '../ui/components/ToolConfirmation.js';
 export interface ToolCall {
     id: string;
     type: 'function';
@@ -14,7 +15,7 @@ export interface ToolResult {
 }
 export type SubAgentMessageCallback = (message: SubAgentMessage) => void;
 export interface ToolConfirmationCallback {
-    (toolCall: ToolCall, batchToolNames?: string, allTools?: ToolCall[]): Promise<string>;
+    (toolCall: ToolCall, batchToolNames?: string, allTools?: ToolCall[]): Promise<ConfirmationResult>;
 }
 export interface ToolApprovalChecker {
     (toolName: string): boolean;

package/dist/utils/toolExecutor.js

@@ -81,8 +81,7 @@ function getToolResourceType(toolName) {
     // Each file is a separate resource
     if (toolName === 'filesystem-edit' ||
         toolName === 'filesystem-edit_search' ||
-        toolName === 'filesystem-create' ||
-        toolName === 'filesystem-delete') {
+        toolName === 'filesystem-create') {
         return 'filesystem'; // Will be further refined by file path
     }
     // Other tools are independent

package/dist/utils/usageLogger.js

@@ -7,9 +7,24 @@ let writeQueue = Promise.resolve();
 async function getActiveProfile() {
     try {
         const homeDir = os.homedir();
-        const profilePath = path.join(homeDir, '.snow', 'active-profile.txt');
-        const profileName = await fs.readFile(profilePath, 'utf-8');
-        return profileName.trim();
+        const jsonPath = path.join(homeDir, '.snow', 'active-profile.json');
+        const legacyPath = path.join(homeDir, '.snow', 'active-profile.txt');
+        // Try JSON format first
+        try {
+            const fileContent = await fs.readFile(jsonPath, 'utf-8');
+            const data = JSON.parse(fileContent.trim());
+            return data.activeProfile || 'default';
+        }
+        catch {
+            // Fallback to legacy .txt format if JSON doesn't exist
+            try {
+                const profileName = await fs.readFile(legacyPath, 'utf-8');
+                return profileName.trim() || 'default';
+            }
+            catch {
+                return 'default';
+            }
+        }
     }
     catch (error) {
         return 'default';

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "snow-ai",
-	"version": "0.3.36",
+	"version": "0.4.0",
 	"description": "Intelligent Command Line Assistant powered by AI",
 	"license": "MIT",
 	"bin": {
@@ -65,6 +65,7 @@
 	"devDependencies": {
 		"@sindresorhus/tsconfig": "^3.0.1",
 		"@types/diff": "^7.0.2",
+		"@types/prettier": "^2.7.3",
 		"@types/react": "^18.0.32",
 		"@types/ws": "^8.5.8",
 		"@vdemedes/prettier-config": "^2.0.1",