centaurus-cli 2.9.0 → 2.9.2

package/dist/services/shell-input-agent.js

@@ -0,0 +1,361 @@
+/**
+ * Shell Input Agent Service
+ *
+ * An isolated AI agent that handles shell input prompts during Agent Control Mode.
+ * Completely separate from the main chat agent - has its own conversation history,
+ * tools, and lifecycle. Terminates immediately when the shell command completes.
+ */
+import { aiServiceClient } from './ai-service-client.js';
+import { sessionQuotaManager } from './session-quota-manager.js';
+import { quickLog } from '../utils/conversation-logger.js';
+// ==================== Constants ====================
+const MAX_TURNS_PER_SESSION = 20; // Max AI turns per shell session
+const MODEL = 'gemini-3-flash-preview'; // Fast model for quick responses
+// ==================== Helper Functions ====================
+/**
+ * Convert literal escape sequences (like \n, \r, \t) to actual control characters.
+ * The AI often returns "1\n" where \n is meant to be a newline but arrives as literal chars.
+ */
+function convertEscapeSequences(input) {
+    return input
+        .replace(/\\n/g, '\n') // newline
+        .replace(/\\r/g, '\r') // carriage return
+        .replace(/\\t/g, '\t') // tab
+        .replace(/\\x1b/gi, '\x1b') // escape for arrow keys etc
+        .replace(/\\e/g, '\x1b'); // alternative escape notation
+}
+// ==================== Shell Input System Prompt ====================
+const SHELL_INPUT_SYSTEM_PROMPT = `You are a specialized AI agent that ONLY provides input to an interactive shell command.
+
+CRITICAL RULES:
+1. You may ONLY use the execute_command tool with shell_input parameter
+2. NEVER try to start a new command - only send input to the EXISTING running shell
+3. Respond with ONLY the tool call - no explanations needed
+4. Set SafeToAutoRun: true to avoid approval prompts
+5. For password prompts, DO NOT provide passwords - skip with a neutral response
+6. IMPORTANT: The shell_input value will automatically have Enter pressed after it
+   - Just provide the text you want to type, the system will press Enter
+   - For example: shell_input="1" NOT shell_input="1\\n"
+
+Your ONLY job is to analyze shell output and provide appropriate input when prompted.`;
+// ==================== ShellInputAgent Class ====================
+class ShellInputAgentClass {
+    sessions = new Map();
+    toolRegistry;
+    onShellInput;
+    onToolExecutionUpdate;
+    onPromptAnswered; // Callback when a prompt has been answered
+    processingLock = new Map(); // Prevent concurrent AI calls per shell
+    /**
+     * Initialize the agent with a tool registry
+     */
+    initialize(toolRegistry) {
+        this.toolRegistry = toolRegistry;
+        quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Initialized with tool registry\n`);
+    }
+    /**
+     * Set callback for sending input to shell (bridges to App.tsx's onShellInput)
+     */
+    setOnShellInput(callback) {
+        this.onShellInput = callback;
+    }
+    /**
+     * Set callback for tool execution updates (for UI feedback)
+     */
+    setOnToolExecutionUpdate(callback) {
+        this.onToolExecutionUpdate = callback;
+    }
+    /**
+     * Set callback for when a prompt has been answered
+     * This is used to update the UI to show a tick instead of spinner
+     */
+    setOnPromptAnswered(callback) {
+        this.onPromptAnswered = callback;
+    }
+    /**
+     * Start a new shell input session
+     * @param shellId Unique identifier for this shell session
+     * @param command The command being run
+     * @param cwd Current working directory
+     * @param mainConversation Optional main agent conversation history for context
+     */
+    startSession(shellId, command, cwd, mainConversation) {
+        // Terminate any existing session for this shell
+        this.terminateSession(shellId);
+        const session = {
+            shellId,
+            command,
+            cwd,
+            isActive: true,
+            abortController: new AbortController(),
+            conversationHistory: [],
+            mainConversationContext: mainConversation,
+            turnCount: 0,
+            startTime: new Date()
+        };
+        this.sessions.set(shellId, session);
+        this.processingLock.set(shellId, false);
+        const contextInfo = mainConversation ? ` (with ${mainConversation.length} context messages)` : '';
+        quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Started session for ${shellId}: ${command}${contextInfo}\n`);
+    }
+    /**
+     * Handle an input prompt detected in shell output
+     * Returns quickly - AI call runs in background
+     */
+    async handleInputPrompt(shellId, shellOutput, promptContext, inputType, confidence) {
+        quickLog(`[${new Date().toISOString()}] [ShellInputAgent] handleInputPrompt called for ${shellId}, sessions count: ${this.sessions.size}\n`);
+        const session = this.sessions.get(shellId);
+        if (!session || !session.isActive) {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] No active session for ${shellId}, ignoring prompt\n`);
+            return;
+        }
+        // Check if already processing a prompt for this shell
+        if (this.processingLock.get(shellId)) {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Already processing for ${shellId}, ignoring duplicate\n`);
+            return;
+        }
+        // Set lock
+        this.processingLock.set(shellId, true);
+        // Run AI call in background (don't await)
+        this.processInputPrompt(session, shellOutput, promptContext, inputType, confidence)
+            .catch(error => {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Error processing prompt for ${shellId}: ${error.message}\n`);
+        })
+            .finally(() => {
+            this.processingLock.set(shellId, false);
+        });
+    }
+    /**
+     * Process an input prompt with AI
+     */
+    async processInputPrompt(session, shellOutput, promptContext, inputType, confidence) {
+        quickLog(`[${new Date().toISOString()}] [ShellInputAgent] processInputPrompt entry for ${session.shellId}, isActive=${session.isActive}, turnCount=${session.turnCount}\n`);
+        if (!session.isActive) {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Session not active, returning\n`);
+            return;
+        }
+        // Check turn limit
+        if (session.turnCount >= MAX_TURNS_PER_SESSION) {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Session ${session.shellId} hit max turns\n`);
+            return;
+        }
+        // Check quota
+        if (!sessionQuotaManager.canSendMessage()) {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Session quota exhausted\n`);
+            return;
+        }
+        // Check if tool registry is available
+        if (!this.toolRegistry) {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Tool registry not initialized\n`);
+            return;
+        }
+        session.turnCount++;
+        session.lastInputTime = new Date();
+        // Build the prompt for this input
+        const userPrompt = this.buildInputPrompt(session, shellOutput, promptContext, inputType, confidence);
+        // Add to session's isolated conversation history
+        if (session.conversationHistory.length === 0) {
+            // First message includes system context
+            session.conversationHistory.push({
+                role: 'user',
+                content: SHELL_INPUT_SYSTEM_PROMPT + '\n\n' + userPrompt
+            });
+        }
+        else {
+            session.conversationHistory.push({
+                role: 'user',
+                content: userPrompt
+            });
+        }
+        // Get only execute_command tool schema
+        const executeCommandSchema = this.toolRegistry.getSchemas().filter(t => t.name === 'execute_command');
+        try {
+            let assistantMessage = '';
+            let toolCalls = [];
+            // Increment quota
+            sessionQuotaManager.incrementMessageCount();
+            // Stream AI response with abort signal
+            for await (const chunk of aiServiceClient.streamChat(MODEL, session.conversationHistory, executeCommandSchema, {
+                cwd: session.cwd,
+                platform: process.platform,
+                shell: process.env.SHELL || 'cmd.exe',
+                os: 'windows',
+                homeDir: process.env.USERPROFILE || ''
+            }, undefined, undefined, session.abortController.signal)) {
+                // Check if session was terminated
+                if (!session.isActive) {
+                    quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Session ${session.shellId} terminated during streaming\n`);
+                    return;
+                }
+                if (chunk.type === 'error') {
+                    throw new Error(chunk.message);
+                }
+                if (chunk.type === 'text') {
+                    assistantMessage += chunk.content;
+                }
+                if (chunk.type === 'tool_call') {
+                    const toolCall = chunk.toolCall;
+                    // Parse string arguments if needed
+                    if (toolCall.arguments && typeof toolCall.arguments === 'string') {
+                        try {
+                            toolCall.arguments = JSON.parse(toolCall.arguments);
+                        }
+                        catch (e) {
+                            // Ignore parsing error
+                        }
+                    }
+                    toolCalls.push(toolCall);
+                }
+                if (chunk.type === 'done') {
+                    break;
+                }
+            }
+            // Check again if session is still active
+            if (!session.isActive) {
+                return;
+            }
+            // Process tool calls
+            for (const toolCall of toolCalls) {
+                if (toolCall.name === 'execute_command' && toolCall.arguments?.shell_input) {
+                    const rawShellInput = toolCall.arguments.shell_input;
+                    // Convert escape sequences like \n to actual newlines
+                    let shellInput = convertEscapeSequences(rawShellInput);
+                    // Automatically append Enter (carriage return) if not already present
+                    // This ensures the input is actually submitted to the shell
+                    if (!shellInput.endsWith('\n') && !shellInput.endsWith('\r')) {
+                        shellInput += '\r';
+                    }
+                    // Send input to shell via callback
+                    if (this.onShellInput) {
+                        quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Sending input to ${session.shellId}: "${rawShellInput}" + Enter\n`);
+                        this.onShellInput(session.shellId, shellInput);
+                        // Notify that a prompt has been answered (to update UI spinner -> tick)
+                        if (this.onPromptAnswered) {
+                            this.onPromptAnswered(session.shellId);
+                        }
+                        // Add to conversation history
+                        session.conversationHistory.push({
+                            role: 'assistant',
+                            content: '',
+                            tool_calls: [toolCall]
+                        });
+                        session.conversationHistory.push({
+                            role: 'tool',
+                            tool_call_id: toolCall.id,
+                            content: `Sent input to shell: ${shellInput}`
+                        });
+                    }
+                }
+            }
+        }
+        catch (error) {
+            if (error.name === 'AbortError') {
+                quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Session ${session.shellId} aborted\n`);
+                return;
+            }
+            throw error;
+        }
+    }
+    /**
+     * Build the input prompt for AI
+     */
+    buildInputPrompt(session, shellOutput, promptContext, inputType, confidence) {
+        // Get last 30 lines of output
+        const lastLines = shellOutput.split('\n').slice(-30).join('\n');
+        // Build context from main conversation if available
+        let conversationContext = '';
+        if (session.mainConversationContext && session.mainConversationContext.length > 0) {
+            // Extract user messages for context (most relevant for understanding intent)
+            const userMessages = session.mainConversationContext
+                .filter(msg => msg.role === 'user' && typeof msg.content === 'string')
+                .map(msg => msg.content)
+                .slice(-3) // Last 3 user messages for context
+                .join('\n---\n');
+            if (userMessages) {
+                conversationContext = `
+**User's Original Intent (from main conversation):**
+\`\`\`
+${userMessages.substring(0, 1000)}${userMessages.length > 1000 ? '...[truncated]' : ''}
+\`\`\`
+
+Consider the user's original intent when providing input. If the user mentioned specific values to use, prefer those.
+`;
+            }
+        }
+        return `[SHELL INPUT REQUIRED]
+
+**Running command:** ${session.command}
+**Working directory:** ${session.cwd}
+${conversationContext}
+**Current shell output (last 30 lines):**
+\`\`\`
+${lastLines}
+\`\`\`
+
+**Detected prompt:** ${promptContext}
+**Input type needed:** ${inputType}
+**Confidence:** ${confidence}
+
+Use execute_command with shell_input to respond. Example:
+{
+  "tool": "execute_command",
+  "arguments": {
+    "shell_input": "YOUR_INPUT_HERE",
+    "SafeToAutoRun": true,
+    "CommandLine": "${session.command}",
+    "Cwd": "${session.cwd}"
+  }
+}`;
+    }
+    /**
+     * Terminate a shell input session immediately
+     */
+    terminateSession(shellId) {
+        const session = this.sessions.get(shellId);
+        if (session) {
+            quickLog(`[${new Date().toISOString()}] [ShellInputAgent] Terminating session ${shellId}\n`);
+            // Mark as inactive first
+            session.isActive = false;
+            // Abort any running AI call
+            try {
+                session.abortController.abort();
+            }
+            catch (e) {
+                // Ignore abort errors
+            }
+            // Clean up
+            this.sessions.delete(shellId);
+            this.processingLock.delete(shellId);
+        }
+    }
+    /**
+     * Check if a session is active
+     */
+    isSessionActive(shellId) {
+        const session = this.sessions.get(shellId);
+        return session?.isActive ?? false;
+    }
+    /**
+     * Get session info (for debugging)
+     */
+    getSessionInfo(shellId) {
+        const session = this.sessions.get(shellId);
+        if (!session)
+            return undefined;
+        return {
+            turnCount: session.turnCount,
+            isActive: session.isActive
+        };
+    }
+    /**
+     * Terminate all active sessions
+     */
+    terminateAllSessions() {
+        for (const shellId of this.sessions.keys()) {
+            this.terminateSession(shellId);
+        }
+    }
+}
+// Export singleton
+export const ShellInputAgent = new ShellInputAgentClass();
+//# sourceMappingURL=shell-input-agent.js.map

package/dist/services/shell-input-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shell-input-agent.js","sourceRoot":"","sources":["../../src/services/shell-input-agent.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,eAAe,EAAqC,MAAM,wBAAwB,CAAC;AAI5F,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,EAAE,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAiB3D,sDAAsD;AAEtD,MAAM,qBAAqB,GAAG,EAAE,CAAC,CAAE,iCAAiC;AACpE,MAAM,KAAK,GAAG,wBAAwB,CAAC,CAAE,iCAAiC;AAE1E,6DAA6D;AAE7D;;;GAGG;AACH,SAAS,sBAAsB,CAAC,KAAa;IACzC,OAAO,KAAK;SACP,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAI,UAAU;SACnC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAI,kBAAkB;SAC3C,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAI,MAAM;SAC/B,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAE,4BAA4B;SACxD,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAI,8BAA8B;AACnE,CAAC;AAED,sEAAsE;AAEtE,MAAM,yBAAyB,GAAG;;;;;;;;;;;;sFAYoD,CAAC;AAEvF,kEAAkE;AAElE,MAAM,oBAAoB;IACd,QAAQ,GAAmC,IAAI,GAAG,EAAE,CAAC;IACrD,YAAY,CAAgB;IAC5B,YAAY,CAA4C;IACxD,qBAAqB,CAAyB;IAC9C,gBAAgB,CAA6B,CAAE,2CAA2C;IAC1F,cAAc,GAAyB,IAAI,GAAG,EAAE,CAAC,CAAE,wCAAwC;IAEnG;;OAEG;IACH,UAAU,CAAC,YAA0B;QACjC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,sDAAsD,CAAC,CAAC;IACjG,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,QAAkD;QAC9D,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,wBAAwB,CAAC,QAA+B;QACpD,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC;IAC1C,CAAC;IAED;;;OAGG;IACH,mBAAmB,CAAC,QAAmC;QACnD,IAAI,CAAC,gBAAgB,GAAG,QAAQ,CAAC;IACrC,CAAC;IAED;;;;;;OAMG;IACH,YAAY,CAAC,OAAe,EAAE,OAAe,EAAE,GAAW,EAAE,gBAA8B;QACtF,gDAAgD;QAChD,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAE/B,MAAM,OAAO,GAAsB;YAC/B,OAAO;YACP,OAAO;YACP,GAAG;YACH,QAAQ,EAAE,IAAI;YACd,eAAe,EAAE,IAAI,eAAe,EAAE;YACtC,mBAAmB,EAAE,EAAE;YACvB,uBAAuB,EAAE,gBAAgB;YACzC,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,IAAI,IAAI,EAAE;SACxB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACpC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAExC,MAAM,WAAW,GAAG,gBAAgB,CAAC,CAAC,CAAC,UAAU,gBAAgB,CAAC,MAAM,oBAAoB,CAAC,CAAC,CAAC,EAAE,CAAC;QAClG,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,2CAA2C,OAAO,KAAK,OAAO,GAAG,WAAW,IAAI,CAAC,CAAC;IAC3H,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,iBAAiB,CACnB,OAAe,EACf,WAAmB,EACnB,aAAqB,EACrB,SAAiB,EACjB,UAAkB;QAElB,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,oDAAoD,OAAO,qBAAqB,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC;QAE7I,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YAChC,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,6CAA6C,OAAO,qBAAqB,CAAC,CAAC;YAChH,OAAO;QACX,CAAC;QAED,sDAAsD;QACtD,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACnC,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,8CAA8C,OAAO,wBAAwB,CAAC,CAAC;YACpH,OAAO;QACX,CAAC;QAED,WAAW;QACX,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAEvC,0CAA0C;QAC1C,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,CAAC;aAC9E,KAAK,CAAC,KAAK,CAAC,EAAE;YACX,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,mDAAmD,OAAO,KAAK,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC;QAC3H,CAAC,CAAC;aACD,OAAO,CAAC,GAAG,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACX,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAC5B,OAA0B,EAC1B,WAAmB,EACnB,aAAqB,EACrB,SAAiB,EACjB,UAAkB;QAElB,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,oDAAoD,OAAO,CAAC,OAAO,cAAc,OAAO,CAAC,QAAQ,eAAe,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC;QAE5K,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACpB,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,qDAAqD,CAAC,CAAC;YAC5F,OAAO;QACX,CAAC;QAED,mBAAmB;QACnB,IAAI,OAAO,CAAC,SAAS,IAAI,qBAAqB,EAAE,CAAC;YAC7C,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,+BAA+B,OAAO,CAAC,OAAO,kBAAkB,CAAC,CAAC;YACvG,OAAO;QACX,CAAC;QAED,cAAc;QACd,IAAI,CAAC,mBAAmB,CAAC,cAAc,EAAE,EAAE,CAAC;YACxC,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,+CAA+C,CAAC,CAAC;YACtF,OAAO;QACX,CAAC;QAED,sCAAsC;QACtC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACrB,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,qDAAqD,CAAC,CAAC;YAC5F,OAAO;QACX,CAAC;QAED,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,OAAO,CAAC,aAAa,GAAG,IAAI,IAAI,EAAE,CAAC;QAEnC,kCAAkC;QAClC,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QAErG,iDAAiD;QACjD,IAAI,OAAO,CAAC,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3C,wCAAwC;YACxC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC7B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,yBAAyB,GAAG,MAAM,GAAG,UAAU;aAC3D,CAAC,CAAC;QACP,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC7B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,UAAU;aACtB,CAAC,CAAC;QACP,CAAC;QAED,uCAAuC;QACvC,MAAM,oBAAoB,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC;QAEtG,IAAI,CAAC;YACD,IAAI,gBAAgB,GAAG,EAAE,CAAC;YAC1B,IAAI,SAAS,GAAU,EAAE,CAAC;YAE1B,kBAAkB;YAClB,mBAAmB,CAAC,qBAAqB,EAAE,CAAC;YAE5C,uCAAuC;YACvC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,eAAe,CAAC,UAAU,CAChD,KAAK,EACL,OAAO,CAAC,mBAAmB,EAC3B,oBAAoB,EACpB;gBACI,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,SAAS;gBACrC,EAAE,EAAE,SAAS;gBACb,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE;aACzC,EACD,SAAS,EACT,SAAS,EACT,OAAO,CAAC,eAAe,CAAC,MAAM,CACjC,EAAE,CAAC;gBACA,kCAAkC;gBAClC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;oBACpB,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,+BAA+B,OAAO,CAAC,OAAO,gCAAgC,CAAC,CAAC;oBACrH,OAAO;gBACX,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACnC,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,gBAAgB,IAAI,KAAK,CAAC,OAAO,CAAC;gBACtC,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;oBAChC,mCAAmC;oBACnC,IAAI,QAAQ,CAAC,SAAS,IAAI,OAAO,QAAQ,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;wBAC/D,IAAI,CAAC;4BACD,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;wBACxD,CAAC;wBAAC,OAAO,CAAC,EAAE,CAAC;4BACT,uBAAuB;wBAC3B,CAAC;oBACL,CAAC;oBACD,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC7B,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,MAAM;gBACV,CAAC;YACL,CAAC;YAED,yCAAyC;YACzC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACpB,OAAO;YACX,CAAC;YAED,qBAAqB;YACrB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBAC/B,IAAI,QAAQ,CAAC,IAAI,KAAK,iBAAiB,IAAI,QAAQ,CAAC,SAAS,EAAE,WAAW,EAAE,CAAC;oBACzE,MAAM,aAAa,GAAG,QAAQ,CAAC,SAAS,CAAC,WAAW,CAAC;oBACrD,sDAAsD;oBACtD,IAAI,UAAU,GAAG,sBAAsB,CAAC,aAAa,CAAC,CAAC;oBAEvD,sEAAsE;oBACtE,4DAA4D;oBAC5D,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC3D,UAAU,IAAI,IAAI,CAAC;oBACvB,CAAC;oBAED,mCAAmC;oBACnC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;wBACpB,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,wCAAwC,OAAO,CAAC,OAAO,MAAM,aAAa,aAAa,CAAC,CAAC;wBAC9H,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;wBAE/C,wEAAwE;wBACxE,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;4BACxB,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;wBAC3C,CAAC;wBAED,8BAA8B;wBAC9B,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC;4BAC7B,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE,EAAE;4BACX,UAAU,EAAE,CAAC,QAAQ,CAAC;yBACzB,CAAC,CAAC;wBACH,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC;4BAC7B,IAAI,EAAE,MAAM;4BACZ,YAAY,EAAE,QAAQ,CAAC,EAAE;4BACzB,OAAO,EAAE,wBAAwB,UAAU,EAAE;yBAChD,CAAC,CAAC;oBACP,CAAC;gBACL,CAAC;YACL,CAAC;QAEL,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YAClB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC9B,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,+BAA+B,OAAO,CAAC,OAAO,YAAY,CAAC,CAAC;gBACjG,OAAO;YACX,CAAC;YACD,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED;;OAEG;IACK,gBAAgB,CACpB,OAA0B,EAC1B,WAAmB,EACnB,aAAqB,EACrB,SAAiB,EACjB,UAAkB;QAElB,8BAA8B;QAC9B,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhE,oDAAoD;QACpD,IAAI,mBAAmB,GAAG,EAAE,CAAC;QAC7B,IAAI,OAAO,CAAC,uBAAuB,IAAI,OAAO,CAAC,uBAAuB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChF,6EAA6E;YAC7E,MAAM,YAAY,GAAG,OAAO,CAAC,uBAAuB;iBAC/C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC;iBACrE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,OAAiB,CAAC;iBACjC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAE,mCAAmC;iBAC9C,IAAI,CAAC,SAAS,CAAC,CAAC;YAErB,IAAI,YAAY,EAAE,CAAC;gBACf,mBAAmB,GAAG;;;EAGpC,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE;;;;CAIrF,CAAC;YACU,CAAC;QACL,CAAC;QAED,OAAO;;uBAEQ,OAAO,CAAC,OAAO;yBACb,OAAO,CAAC,GAAG;EAClC,mBAAmB;;;EAGnB,SAAS;;;uBAGY,aAAa;yBACX,SAAS;kBAChB,UAAU;;;;;;;;sBAQN,OAAO,CAAC,OAAO;cACvB,OAAO,CAAC,GAAG;;EAEvB,CAAC;IACC,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,OAAe;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,OAAO,EAAE,CAAC;YACV,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,2CAA2C,OAAO,IAAI,CAAC,CAAC;YAE7F,yBAAyB;YACzB,OAAO,CAAC,QAAQ,GAAG,KAAK,CAAC;YAEzB,4BAA4B;YAC5B,IAAI,CAAC;gBACD,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YACpC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,sBAAsB;YAC1B,CAAC;YAED,WAAW;YACX,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC9B,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,OAAe;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,OAAO,OAAO,EAAE,QAAQ,IAAI,KAAK,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,OAAe;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAC/B,OAAO;YACH,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,QAAQ,EAAE,OAAO,CAAC,QAAQ;SAC7B,CAAC;IACN,CAAC;IAED;;OAEG;IACH,oBAAoB;QAChB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;IACL,CAAC;CACJ;AAED,mBAAmB;AACnB,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,oBAAoB,EAAE,CAAC"}

package/dist/services/sub-agent-manager.d.ts

@@ -0,0 +1,139 @@
+/**
+ * Sub-Agent Manager Service
+ *
+ * Manages the lifecycle of background sub-agents that can be spawned
+ * by the main agent to handle delegated tasks.
+ */
+import { Message as AIMessage } from './ai-service-client.js';
+import { ToolRegistry } from '../tools/registry.js';
+export interface SubAgentConfig {
+    prompt: string;
+    context: string;
+    workingDirectory: string;
+    complexity: number;
+}
+export interface FileOperation {
+    type: 'create' | 'edit' | 'delete';
+    path: string;
+    originalContent?: string;
+    newContent?: string;
+    diff?: string;
+    reasonText?: string;
+    timestamp: Date;
+}
+export interface ToolExecution {
+    toolName: string;
+    arguments: Record<string, any>;
+    reasonText?: string;
+    result: string;
+    success: boolean;
+    timestamp: Date;
+}
+export type SubAgentStatus = 'pending' | 'running' | 'completed' | 'failed' | 'terminated';
+export interface SubAgent {
+    id: string;
+    status: SubAgentStatus;
+    prompt: string;
+    context: string;
+    workingDirectory: string;
+    complexity: number;
+    model: string;
+    startTime: Date;
+    endTime?: Date;
+    conversationHistory: AIMessage[];
+    turnCount: number;
+    fileOperations: FileOperation[];
+    toolHistory: ToolExecution[];
+    result?: string;
+    error?: string;
+    isRead: boolean;
+    abortController?: AbortController;
+}
+declare class SubAgentManagerClass {
+    private subAgents;
+    private toolRegistry?;
+    private onSubAgentCountChange?;
+    private onToolExecutionUpdate?;
+    /**
+     * Initialize the manager with a tool registry
+     */
+    initialize(toolRegistry: ToolRegistry): void;
+    /**
+     * Set callback for sub-agent count changes
+     */
+    setOnSubAgentCountChange(callback: (count: number) => void): void;
+    /**
+     * Set callback for tool execution updates (for logging)
+     */
+    setOnToolExecutionUpdate(callback: (update: any) => void): void;
+    /**
+     * Notify listeners about sub-agent count change
+     */
+    private notifyCountChange;
+    /**
+     * Select model based on complexity
+     */
+    private selectModel;
+    /**
+     * Generate system prompt for sub-agent
+     */
+    private generateSystemPrompt;
+    /**
+     * Spawn a new sub-agent
+     */
+    spawnSubAgent(config: SubAgentConfig): Promise<{
+        success: boolean;
+        agentId?: string;
+        error?: string;
+    }>;
+    /**
+     * Run the sub-agent's agent loop
+     */
+    private runSubAgentLoop;
+    /**
+     * Track file operations made by sub-agent
+     */
+    private trackFileOperation;
+    /**
+     * Get a sub-agent by ID
+     */
+    getSubAgent(agentId: string): SubAgent | undefined;
+    /**
+     * Get all sub-agents
+     */
+    getAllSubAgents(): SubAgent[];
+    /**
+     * Get running sub-agents
+     */
+    getRunningSubAgents(): SubAgent[];
+    /**
+     * Get completed but unread sub-agents
+     */
+    getCompletedUnreadSubAgents(): SubAgent[];
+    /**
+     * Terminate a sub-agent
+     */
+    terminateSubAgent(agentId: string): {
+        success: boolean;
+        error?: string;
+    };
+    /**
+     * Get formatted diff output for a sub-agent
+     */
+    getSubAgentDiff(agentId: string): string;
+    /**
+     * Mark sub-agent results as read
+     */
+    markSubAgentRead(agentId: string): void;
+    /**
+     * Format duration between two dates
+     */
+    private formatDuration;
+    /**
+     * Clean up old completed sub-agents (call periodically)
+     */
+    cleanup(): void;
+}
+export declare const SubAgentManager: SubAgentManagerClass;
+export {};
+//# sourceMappingURL=sub-agent-manager.d.ts.map

package/dist/services/sub-agent-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sub-agent-manager.d.ts","sourceRoot":"","sources":["../../src/services/sub-agent-manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAmB,OAAO,IAAI,SAAS,EAAe,MAAM,wBAAwB,CAAC;AAE5F,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AASpD,MAAM,WAAW,cAAc;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,aAAa;IAC1B,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,QAAQ,CAAC;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,IAAI,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,IAAI,CAAC;CACnB;AAED,MAAM,MAAM,cAAc,GAAG,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,YAAY,CAAC;AAE3F,MAAM,WAAW,QAAQ;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,cAAc,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,IAAI,CAAC;IAChB,OAAO,CAAC,EAAE,IAAI,CAAC;IAGf,mBAAmB,EAAE,SAAS,EAAE,CAAC;IACjC,SAAS,EAAE,MAAM,CAAC;IAGlB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,WAAW,EAAE,aAAa,EAAE,CAAC;IAG7B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IAGf,MAAM,EAAE,OAAO,CAAC;IAChB,eAAe,CAAC,EAAE,eAAe,CAAC;CACrC;AAsCD,cAAM,oBAAoB;IACtB,OAAO,CAAC,SAAS,CAAoC;IACrD,OAAO,CAAC,YAAY,CAAC,CAAe;IACpC,OAAO,CAAC,qBAAqB,CAAC,CAA0B;IACxD,OAAO,CAAC,qBAAqB,CAAC,CAAwB;IAEtD;;OAEG;IACH,UAAU,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI;IAK5C;;OAEG;IACH,wBAAwB,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAKjE;;OAEG;IACH,wBAAwB,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,IAAI,GAAG,IAAI;IAI/D;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAazB;;OAEG;IACH,OAAO,CAAC,WAAW;IAInB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAO5B;;OAEG;IACG,aAAa,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAyE5G;;OAEG;YACW,eAAe;IAwM7B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAuB1B;;OAEG;IACH,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS;IAIlD;;OAEG;IACH,eAAe,IAAI,QAAQ,EAAE;IAI7B;;OAEG;IACH,mBAAmB,IAAI,QAAQ,EAAE;IAIjC;;OAEG;IACH,2BAA2B,IAAI,QAAQ,EAAE;IAOzC;;OAEG;IACH,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IAkBxE;;OAEG;IACH,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IA4FxC;;OAEG;IACH,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAOvC;;OAEG;IACH,OAAO,CAAC,cAAc;IAYtB;;OAEG;IACH,OAAO,IAAI,IAAI;CASlB;AAGD,eAAO,MAAM,eAAe,sBAA6B,CAAC"}