matex-cli 1.2.84 → 1.2.87

package/dist/session/agent-session.js

@@ -1,450 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AgentSession = void 0;
-const chalk_1 = __importDefault(require("chalk"));
-const inquirer_1 = __importDefault(require("inquirer"));
-const tui_1 = require("../utils/tui");
-const agent_orchestrator_1 = require("../utils/agent-orchestrator");
-const spinner_1 = require("../utils/spinner");
-const repo_mapper_1 = require("../utils/repo-mapper");
-class AgentSession {
-    constructor(config) {
-        this.lastActivityTime = Date.now();
-        this.isAjaySleeping = false;
-        this.isAjayOnChaiBreak = false;
-        this.chaiBreakEndTime = 0;
-        this.breakInterval = null;
-        this.idleCheckInterval = null;
-        this.client = config.client;
-        this.model = config.model;
-        this.execute = config.execute;
-        this.messages = config.initialMessages;
-        this.broBanter = config.broBanter;
-        this.sleepyAjayProtocol = config.sleepyAjayProtocol;
-        this.currentSessionCwd = config.baseDir;
-        this.messageScrubber = config.messageScrubber;
-        this.auditLogger = config.auditLogger;
-        this.isAugovMode = config.isAugovMode;
-    }
-    async start() {
-        this.startIdleCheck();
-        while (true) {
-            this.lastActivityTime = Date.now();
-            // Premium UI Box (Same as AU-GOV but for Dev)
-            const darkBorder = chalk_1.default.hex('#1E3A8A');
-            const primaryText = chalk_1.default.hex('#3B82F6');
-            const shadow = chalk_1.default.hex('#1e1e1e');
-            console.log(shadow('┌─────────────────────────────────────────────────────────────────────────────┐'));
-            if (this.isAugovMode) {
-                console.log(shadow('│') + chalk_1.default.gray(' Using 1 SECURE node file                          1 AU-GOV validated server ') + shadow('│'));
-            }
-            else {
-                console.log(shadow('│') + chalk_1.default.gray(' Using 1 MATEX_Orchestrator                        1 MCP server              ') + shadow('│'));
-            }
-            console.log(shadow('├─────────────────────────────────────────────────────────────────────────────┤'));
-            const { userInput } = await inquirer_1.default.prompt([
-                {
-                    type: 'input',
-                    name: 'userInput',
-                    message: shadow('│ ') + primaryText('> '),
-                    prefix: ''
-                }
-            ]);
-            if (this.isAugovMode || true) { // Always show status footer in premium mode
-                const homeDirDisplay = this.currentSessionCwd.replace(process.env.HOME || '', '~');
-                console.log(shadow('└─────────────────────────────────────────────────────────────────────────────┘'));
-                console.log(chalk_1.default.gray(`  ${homeDirDisplay.padEnd(30)}  sandbox-exec (minimal)\n`));
-            }
-            this.lastActivityTime = Date.now();
-            if (this.isAjaySleeping) {
-                this.wakeAjayUp();
-            }
-            if (userInput.toLowerCase() === 'exit' || userInput.toLowerCase() === 'quit') {
-                console.log(chalk_1.default.yellow('\n👋 Ending session. Happy coding!\n'));
-                break;
-            }
-            if (!userInput.trim())
-                continue;
-            let processedInput = userInput;
-            if (this.messageScrubber) {
-                processedInput = this.messageScrubber(userInput);
-                if (userInput !== processedInput) {
-                    tui_1.TUI.log(chalk_1.default.yellow('\n 🛡️  Privacy Shield Active: Sensitive Australian data was redacted locally.'));
-                }
-            }
-            if (this.auditLogger) {
-                this.auditLogger.logInteraction('USER', 'INPUT', processedInput);
-            }
-            this.messages.push({ role: 'user', content: processedInput });
-            let loopCount = 0;
-            const MAX_LOOPS = 25;
-            while (loopCount < MAX_LOOPS) {
-                loopCount++;
-                try {
-                    const shouldContinue = await this.agenticLoopPass(loopCount);
-                    if (!shouldContinue)
-                        break;
-                }
-                catch (error) {
-                    spinner_1.spinner.fail('Request failed');
-                    tui_1.TUI.log(chalk_1.default.red(`Error: ${error.message}\n`));
-                    this.messages.pop(); // Pop the prompt so they can retry
-                    break;
-                }
-            }
-        }
-        this.cleanup();
-    }
-    async refreshMap() {
-        try {
-            const freshMapper = new repo_mapper_1.RepoMapper(this.currentSessionCwd);
-            const freshRepoMap = await freshMapper.generateMap(true);
-            // Re-inject map
-            this.messages[0].content = this.messages[0].content.replace(/### 🛠️ CURRENT PROJECT CONTEXT:[\s\S]*$/, `### 🛠️ CURRENT PROJECT CONTEXT:\n${freshRepoMap}`);
-            // Re-inject Sleepy Ajay
-            if (this.sleepyAjayProtocol && !this.messages[0].content.includes("### 💤 SLEEPY AJAY PROTOCOL")) {
-                this.messages[0].content += `\n${this.sleepyAjayProtocol}`;
-            }
-        }
-        catch (e) {
-            // Ignore error
-        }
-    }
-    startChaiBreak() {
-        if (this.isAugovMode)
-            return; // Feature disabled in secure mode
-        this.isAjayOnChaiBreak = true;
-        this.chaiBreakEndTime = Date.now() + 120000; // 2 minutes
-        tui_1.TUI.drawChaiBreakMessage(2);
-        this.breakInterval = setInterval(() => {
-            const timeLeft = Math.ceil((this.chaiBreakEndTime - Date.now()) / 60000);
-            if (timeLeft <= 0) {
-                if (this.breakInterval)
-                    clearInterval(this.breakInterval);
-                this.isAjayOnChaiBreak = false;
-                const complained = Math.random() < 0.2;
-                if (complained) {
-                    tui_1.TUI.drawAjayDialogue("i'm back bros... but the tea was a bit cold today. my mood is slightly cooked. what did i miss?");
-                }
-                else {
-                    tui_1.TUI.drawAjayDialogue("i'm back bros, chai was fire. real kiev vibe. what did i miss?");
-                }
-            }
-            else {
-                tui_1.TUI.drawChaiBreakMessage(timeLeft);
-                const randomBanter = this.broBanter[Math.floor(Math.random() * this.broBanter.length)];
-                const cleanBanter = randomBanter.split(']: ')[1];
-                const agent = randomBanter.split(']: ')[0].replace('[', '');
-                tui_1.TUI.drawSwarmDialogue(agent, cleanBanter);
-            }
-        }, 40000);
-    }
-    startIdleCheck() {
-        if (this.isAugovMode)
-            return; // Strict APS compliance: No sleeping on the job
-        this.idleCheckInterval = setInterval(() => {
-            if (!this.isAjaySleeping && !this.isAjayOnChaiBreak && Date.now() - this.lastActivityTime > 120000) {
-                this.isAjaySleeping = true;
-                tui_1.TUI.drawSleepMessage();
-            }
-        }, 10000);
-    }
-    wakeAjayUp() {
-        if (this.isAugovMode)
-            return; // Feature disabled in secure mode
-        this.isAjaySleeping = false;
-        tui_1.TUI.drawWakeUpMessage();
-        this.messages.push({
-            role: 'user',
-            content: `[SYSTEM: AJAY VAI HAS JUST WOKEN UP. Ajay: Say "oh shit i slept again sorry bros...". Bros: Roast him for being lazy and share common Ajay 'lazy genius' incidents. Then Ajay says he is going for CHAI and triggers the 2-minute break.]`
-        });
-        this.startChaiBreak();
-    }
-    async agenticLoopPass(loopCount) {
-        spinner_1.spinner.start(loopCount > 1 ? 'Analyzing result & Validating...' : 'Thinking...');
-        let fullResponse = '';
-        let buffer = '';
-        let hasStarted = false;
-        let codeLang = 'bash';
-        let technicalBuffer = '';
-        let technicalPath = '';
-        let technicalType = null;
-        let currentAgent = null;
-        let agentBuffer = '';
-        tui_1.TUI.drawStatusBar('Swarm is processing... (Press Enter to stop)');
-        const abortController = new AbortController();
-        let isAborted = false;
-        const streamStartTime = Date.now();
-        const onData = (data) => {
-            if (Date.now() - streamStartTime < 200)
-                return;
-            if (data[0] === 13 || data[0] === 10 || data[0] === 27 || data[0] === 3) {
-                isAborted = true;
-                abortController.abort();
-            }
-        };
-        const isRaw = process.stdin.isRaw;
-        process.stdin.resume();
-        if (process.stdin.setRawMode)
-            process.stdin.setRawMode(true);
-        process.stdin.on('data', onData);
-        try {
-            await this.client.chatStream({
-                messages: this.messages,
-                model: this.model,
-                temperature: 0.3,
-                max_tokens: 8192,
-            }, (chunk) => {
-                if (!hasStarted) {
-                    spinner_1.spinner.stop();
-                    hasStarted = true;
-                    console.log();
-                }
-                buffer += chunk;
-                fullResponse += chunk;
-                const lines = buffer.split('\n');
-                buffer = lines.pop() || '';
-                for (const line of lines) {
-                    const codeBlockMatch = line.match(/```(\w+)?/);
-                    const fileStartMatch = line.match(/<file path="([^"]+)">/i);
-                    const patchStartMatch = line.match(/<<<< SEARCH/i);
-                    const summaryStartMatch = line.match(/<summary>/i);
-                    if (!technicalType && (codeBlockMatch || fileStartMatch || patchStartMatch || summaryStartMatch)) {
-                        if (currentAgent && agentBuffer.trim()) {
-                            if (currentAgent.toLowerCase().includes('ajay vai')) {
-                                tui_1.TUI.drawAjayDialogue(agentBuffer.trim());
-                            }
-                            else {
-                                tui_1.TUI.drawSwarmDialogue(currentAgent, agentBuffer.trim());
-                            }
-                            currentAgent = null;
-                            agentBuffer = '';
-                        }
-                        if (codeBlockMatch) {
-                            technicalType = 'code';
-                            codeLang = (codeBlockMatch[1] || 'bash').toUpperCase();
-                            tui_1.TUI.drawStreamingStart('TECHNICAL BLOCK', codeLang);
-                        }
-                        else if (fileStartMatch) {
-                            technicalType = 'file';
-                            technicalPath = fileStartMatch[1];
-                            tui_1.TUI.drawStreamingStart('NEW FILE', technicalPath);
-                        }
-                        else if (patchStartMatch) {
-                            technicalType = 'patch';
-                            tui_1.TUI.drawStreamingStart('PATCH', 'SURGICAL EDIT');
-                        }
-                        else if (summaryStartMatch) {
-                            technicalType = 'summary';
-                            process.stdout.write(chalk_1.default.magenta('\n [📝] Generating Ajay\'s Work Summary...\n'));
-                        }
-                        continue;
-                    }
-                    const fileEndMatch = line.match(/<\/file>/i);
-                    const patchEndMatch = line.match(/>>>> REPLACE/i);
-                    const summaryEndMatch = line.match(/<\/summary>/i);
-                    const isCodeEnd = technicalType === 'code' && line.trim() === '```';
-                    if (isCodeEnd || fileEndMatch || patchEndMatch || summaryEndMatch) {
-                        const displayContent = technicalBuffer.trim();
-                        if (technicalType === 'summary' || summaryEndMatch) {
-                            tui_1.TUI.drawSummaryBox(agent_orchestrator_1.AgentOrchestrator.cleanSummary(displayContent));
-                        }
-                        else {
-                            tui_1.TUI.drawStreamingEnd();
-                        }
-                        technicalBuffer = '';
-                        technicalType = null;
-                        technicalPath = '';
-                        process.stdout.write('\n');
-                        continue;
-                    }
-                    if (technicalType) {
-                        technicalBuffer += line + '\n';
-                        if (technicalType !== 'summary') {
-                            tui_1.TUI.drawStreamingLine(line);
-                        }
-                        continue;
-                    }
-                    const agentMatch = line.match(/\[\**\s*(Ajay Vai|Sunil Dai|Sandip Dai|Bishal Dai|Narayan Dai|Big Bro)(\s*\([^)]+\))?\s*\**\]?\s*:\s*/i)
-                        || line.match(/\b(Ajay Vai|Sunil Dai|Sandip Dai|Bishal Dai|Narayan Dai|Big Bro)(\s*\([^)]+\))?\s*:\s*/i)
-                        || line.match(/\**Agent:\s*(Ajay Vai|Sunil Dai|Sandip Dai|Bishal Dai|Narayan Dai|Big Bro)(\s*\([^)]+\))?\s*\**\s*:\s*/i)
-                        || line.match(/\*\*(Ajay Vai|Sunil Dai|Sandip Dai|Bishal Dai|Narayan Dai|Big Bro)(\s*\([^)]+\))?\*\*:\s*/i);
-                    if (agentMatch) {
-                        if (currentAgent && currentAgent !== agentMatch[1] && agentBuffer.trim()) {
-                            if (currentAgent.toLowerCase().includes('ajay vai')) {
-                                tui_1.TUI.drawAjayDialogue(agentBuffer.trim());
-                            }
-                            else {
-                                tui_1.TUI.drawSwarmDialogue(currentAgent, agentBuffer.trim());
-                            }
-                        }
-                        let cleanLine = line.replace(/\[\**\s*(Ajay Vai|Sunil Dai|Sandip Dai|Bishal Dai|Narayan Dai|Big Bro)\s*\**\]?\s*:\s*/i, '')
-                            .replace(/\b(Ajay Vai|Sunil Dai|Sandip Dai|Bishal Dai|Narayan Dai|Big Bro)\s*:\s*/i, '');
-                        cleanLine = agent_orchestrator_1.AgentOrchestrator.cleanText(cleanLine);
-                        agentBuffer = cleanLine + '\n';
-                        agentBuffer = agentBuffer.replace(/^\([^)]+\)\s*/, '');
-                        currentAgent = agentMatch[1];
-                    }
-                    else if (currentAgent) {
-                        const trimmedLine = line.trim();
-                        if (trimmedLine.match(/^\([^)]+\)$/))
-                            continue;
-                        agentBuffer += agent_orchestrator_1.AgentOrchestrator.cleanText(line) + '\n';
-                    }
-                    else if (line.trim()) {
-                        process.stdout.write(chalk_1.default.gray(line.trim() + ' '));
-                    }
-                }
-            }, abortController.signal);
-        }
-        catch (streamErr) {
-            if (isAborted || streamErr.name === 'CanceledError' || streamErr.message === 'canceled') {
-                console.log(chalk_1.default.gray('\n\n [🛑] Swarm stopped by brother (Enter pressed).'));
-                if (!hasStarted)
-                    spinner_1.spinner.stop();
-            }
-            else {
-                // 🛡️ API FAILURE FALLBACK 🛡️
-                console.log(chalk_1.default.red(`\n  ❌ Swarm API Error: ${streamErr.message}`));
-                console.log(chalk_1.default.cyan(`  🔄 Engaging DeepSeek Fallback Protocol for API failure...`));
-                this.model = 'matexdeepseek'; // Assuming backend routes this correctly
-                this.messages.push({
-                    role: 'system',
-                    content: `[System Error: The previous agent encountered an API error: ${streamErr.message}. You are DeepSeek, our fallback model. Please restart the previous thought and fulfill the user request.]`
-                });
-                return await this.agenticLoopPass(loopCount); // Retry same loop with DeepSeek
-            }
-        }
-        finally {
-            process.stdin.removeListener('data', onData);
-            if (process.stdin.setRawMode)
-                process.stdin.setRawMode(isRaw);
-            process.stdin.pause();
-            if (technicalType) {
-                const displayContent = technicalBuffer.trim();
-                // We'll let the deepseek fallback handle the UI cleanup if it's broken, 
-                // but if we are just ending normally, we close the box.
-                if (!technicalType) {
-                    // Do nothing, already handled
-                }
-            }
-        }
-        if (!hasStarted && !isAborted)
-            spinner_1.spinner.stop();
-        // 🛡️ DEEPSEEK FALLBACK VERIFICATION 🛡️
-        // If technicalType is still active, it means the agent stopped midway through a block.
-        if (technicalType && !isAborted) {
-            console.log(chalk_1.default.yellow(`\n  ⚠️  Agent hallucinated or disconnected mid-thought (Unclosed ${technicalType}).`));
-            console.log(chalk_1.default.cyan(`  🔄 Engaging DeepSeek Fallback Protocol...`));
-            // Clean up the incomplete block from the UI
-            const displayContent = technicalBuffer.trim();
-            if (technicalType === 'summary') {
-                if (displayContent)
-                    tui_1.TUI.drawSummaryBox(agent_orchestrator_1.AgentOrchestrator.cleanSummary(displayContent));
-            }
-            else {
-                tui_1.TUI.drawStreamingEnd();
-            }
-            // Switch model to deepseek and retry the pass
-            this.model = 'matexdeepseek'; // Assuming backend routes this correctly
-            this.messages.push({
-                role: 'system',
-                content: `[System Error: The previous agent disconnected abruptly while generating a ${technicalType} block. You are DeepSeek, our fallback model. Please continue precisely from where the previous agent left off, or restart the thought/code block properly.]`
-            });
-            // Reset state
-            technicalBuffer = '';
-            technicalType = null;
-            return await this.agenticLoopPass(loopCount); // Retry same loop with DeepSeek
-        }
-        const finalAgent = currentAgent;
-        if (finalAgent && agentBuffer.trim()) {
-            if (finalAgent.toLowerCase().includes('ajay vai')) {
-                tui_1.TUI.drawAjayDialogue(agentBuffer.trim());
-            }
-            else {
-                tui_1.TUI.drawSwarmDialogue(finalAgent, agentBuffer.trim());
-            }
-        }
-        // Handled via the stream loop above. 
-        // No post-stream fallback check to avoid duplicate rendering.
-        if (this.auditLogger) {
-            this.auditLogger.logInteraction('AI_SWARM', 'RESPONSE', fullResponse);
-        }
-        console.log();
-        this.messages.push({ role: 'assistant', content: fullResponse });
-        if (this.execute) {
-            const { executeWithPermission } = await Promise.resolve().then(() => __importStar(require('../utils/command-executor')));
-            const result = await executeWithPermission(fullResponse, this.currentSessionCwd);
-            if (result.newCwd && result.newCwd !== this.currentSessionCwd) {
-                try {
-                    process.chdir(result.newCwd);
-                    this.currentSessionCwd = result.newCwd;
-                    tui_1.TUI.drawStatusBar(`Swarm moved to: ${this.currentSessionCwd}`);
-                    this.messages[0].content = this.messages[0].content.replace(/YOUR ROOT: `[^`]+`/, `YOUR ROOT: \`${this.currentSessionCwd}\``);
-                }
-                catch (e) {
-                    // Ignore chdir failure
-                }
-            }
-            if (result.executed) {
-                if (result.success) {
-                    tui_1.TUI.log(chalk_1.default.gray('\n ↺ Auto-feeding output to Swarm...'));
-                    tui_1.TUI.drawGlowingContainer('TERMINAL OUTPUT', 'stdout', result.output || '(No output)');
-                    this.messages.push({ role: 'user', content: `[Command executed successfully. Output:\n${result.output}]\n\n(System Note: If you have finished the user's request, stop and wait. Do NOT simulate or hallucinate a user response.)` });
-                    return true;
-                }
-                else {
-                    tui_1.TUI.log(chalk_1.default.yellow('\n ↺ Command failed. Auto-feeding error to Swarm...'));
-                    tui_1.TUI.drawGlowingContainer('TERMINAL ERROR', 'stderr', result.error || 'Unknown error');
-                    this.messages.push({ role: 'user', content: `[Command failed with error:\n${result.error}]\n\n(System Note: Please fix this error, or wait for the user if you are stuck. Do NOT simulate or hallucinate a user response.)` });
-                    return true;
-                }
-            }
-        }
-        return false;
-    }
-    cleanup() {
-        if (this.breakInterval)
-            clearInterval(this.breakInterval);
-        if (this.idleCheckInterval)
-            clearInterval(this.idleCheckInterval);
-    }
-}
-exports.AgentSession = AgentSession;
-//# sourceMappingURL=agent-session.js.map