moth-ai 1.0.3 → 1.0.4

package/README.md

@@ -1,319 +1,90 @@
 # 🦋 Moth AI
 
-[![npm version](https://img.shields.io/npm/v/@kishlay42/moth-ai.svg)](https://www.npmjs.com/package/@kishlay42/moth-ai)
+[![npm version](https://img.shields.io/npm/v/moth-ai.svg)](https://www.npmjs.com/package/moth-ai)
 [![License: ISC](https://img.shields.io/badge/License-ISC-blue.svg)](https://opensource.org/licenses/ISC)
-[![Node.js Version](https://img.shields.io/node/v/@kishlay42/moth-ai.svg)](https://nodejs.org)
 
-**The Intelligent, Local-First CLI Coding Assistant**
+> **The World's First Truly Open CLI Assistant for Local & Open Source Models**
 
-Moth AI is a powerful **terminal-native coding assistant** built for developers who value **privacy, speed, and control**. It lives inside your terminal, understands your project context, and helps you **write, debug, refactor, and reason about code** using both **local and cloud LLMs**.
-
-<img width="1095" height="504" alt="Moth AI Screenshot" src="https://github.com/user-attachments/assets/23b83a6b-2b63-45af-b9ec-a6dcb0a89b2f" />
-
----
-
-## 📦 Installation
-
-### Global Installation (Recommended)
-
-Install Moth AI globally to use it anywhere on your system:
-
-```bash
-npm install -g @kishlay42/moth-ai
-```
-
-After installation, simply run:
-
-```bash
-moth
-```
-
-### Local Installation
-
-Install in a specific project:
-
-```bash
-npm install @kishlay42/moth-ai
-```
-
-Run using npx:
-
-```bash
-npx moth
-```
-
-### Requirements
-
-- **Node.js**: >= 18.0.0
-- **npm**: >= 8.0.0
-
----
-
-## 🚀 Quick Start
-
-1. **Install Moth AI globally:**
-   ```bash
-   npm install -g @kishlay42/moth-ai
-   ```
-
-2. **Add your first LLM profile:**
-   ```bash
-   moth llm add
-   ```
-   
-   Choose from:
-   - **Local models** (via Ollama) - Free, private, offline
-   - **Cloud providers** (OpenAI, Anthropic, Google) - Requires API key
-
-3. **Start chatting:**
-   ```bash
-   moth
-   ```
-
-4. **Use the command palette:**
-   - Press `Ctrl+U` to access all commands
-   - Switch profiles, toggle autopilot, and more
+<img width="953" height="392" alt="Screenshot 2026-01-07 013431" src="https://github.com/user-attachments/assets/55e104e2-8441-4a74-b3f7-c74340d806fa" />
 
 ---
 
-## ✨ Key Features
-
-### 🧠 LLM-Agnostic & Local-First
-
-Use **any LLM**, local or cloud — switch instantly without changing workflows.
-
-- **Local (via Ollama)**  
-  Run models like **Llama 3**, **Mistral**, **Gemma**, and **DeepSeek-Coder** locally  
-  → Zero latency, full privacy, offline-friendly
+## Overview
 
-- **Cloud Providers**  
-  Plug in your own API keys for:
-  - OpenAI (GPT-4 / GPT-4o)
-  - Anthropic (Claude 3.5 Sonnet)
-  - Google (Gemini)
+Moth AI is the **first terminal-native coding assistant** built from the ground up to treat **local and open-source LLMs** as first-class citizens. While others lock you into proprietary models or cloud subscripts, Moth is truly open source—in code, philosophy, and model support.
 
-<img width="1093" height="241" alt="LLM Switching" src="https://github.com/user-attachments/assets/2de67c9d-f562-4ce3-8bc6-51e2066b69ae" />
+It empowers you to **write, debug, refactor, and reason about code** using **your own models** on **your own hardware**. Whether you're running Llama 3 on a MacBook or GPT-4 in the cloud, Moth gives you the same powerful agentic capabilities without the compromise.
 
 ---
 
-### 🤖 Agentic Capabilities
+## 🚀 Key Features
 
-Moth is not just a chatbot — it's an **AI agent**.
+### 🔓 Truly Open & Local-First
+Moth is the only CLI tool designed to democratize AI access.
+- **Local Native:** Optimized deeply for Ollama. Run **Llama 3, Mistral, Gemma, or DeepSeek** locally with zero latency, 100% privacy, and no internet connection required.
+- **Open Source First:** We support any OpenAI-compatible endpoint, making it universally compatible with the open ecosystem of model servers (LM Studio, LocalAI, etc.).
+- **Cloud Optional:** Seamlessly integrate OpenAI (GPT-4), Anthropic (Claude), or Google (Gemini) when you need extra horsepower—but only when *you* choose to.
 
-- **Task Planning** – Break complex goals into executable steps  
-- **File Editing** – Precise diffs, patches, and refactors  
-- **Terminal Control** – Run builds, tests, and Git commands from chat  
-- **Context-Aware** – Understands your project structure and codebase
 
----
-
-### 🛡️ Permission-First by Design
+### 🤖 Agentic Capabilities with Role-Based Modes
+Moth operates in three distinct modes to match your current task intensity:
 
-You stay in control — always.
+1. **🔵 Default Mode:** A balanced assistant that asks for permission before executing sensitive actions.
+2. **� Plan Mode:** Prioritizes detailed architectural planning. Moth creates comprehensive markdown plans for review before writing a single line of code.
+3. **🚀 Autopilot Mode:** For trusted workflows. Moth executes authorized tool calls automatically, streamlining repetitive tasks.
 
-- Explicit approval before file edits or command execution  
-- Granular permissions per action  
-- **Autopilot mode** for trusted workflows  
-- Feedback loop to guide the agent instead of blind execution  
-
----
+<img width="948" height="112" alt="Screenshot 2026-01-07 003822" src="https://github.com/user-attachments/assets/ff14a463-157d-4107-8de1-519257355989" />
 
-### 🎭 Moth Profiles
+<img width="940" height="108" alt="Screenshot 2026-01-07 003744" src="https://github.com/user-attachments/assets/345509dd-a89a-4b78-83df-9c21c51f4e82" />
 
-Save and switch between different AI personalities.
-
-- **Coding Profile** – Optimized for TypeScript / Python  
-- **Architecture Profile** – Reasoning-focused for system design  
-- **Fast Profile** – Lightweight local model for quick answers  
+### 🛡️ Secure & Transparent
+- **Permission-First Architecture:** You approve every significant file edit or shell command.
+- **Context-Aware:** Moth intelligently scans your project structure to provide relevant answers, not hallucinations.
 
 ---
 
-## � CLI Commands
-
-### Main Commands
-
-```bash
-# Start interactive chat
-moth
-
-# Show help
-moth --help
-
-# Display version
-moth --version
-```
+## 📦 Installation
 
-### LLM Profile Management
+### Global Installation (Recommended)
+Install Moth AI globally to access the `moth` command from any directory.
 
 ```bash
-# Add a new LLM profile
-moth llm add
-
-# List all configured profiles
-moth llm list
-
-# Switch to a different profile
-moth llm use
-
-# Remove a profile
-moth llm remove
-```
-
-### Keyboard Shortcuts
-
-- **Ctrl+U** - Open command palette
-- **Ctrl+C** - Exit chat
-- **Arrow Keys** - Navigate command palette
-- **Enter** - Execute selected command
-
----
-
-## ⚙️ Configuration
-
-Moth AI stores configuration in `~/.moth/config.yaml`
-
-### Example Configuration
-
-```yaml
-profiles:
-  - name: "gpt-4"
-    provider: "openai"
-    model: "gpt-4"
-    apiKey: "sk-..."
-    
-  - name: "local-llama"
-    provider: "ollama"
-    model: "llama3"
-    baseUrl: "http://localhost:11434"
-    
-activeProfile: "gpt-4"
+npm install -g moth-ai
 ```
 
-### Setting Up Ollama (Local Models)
-
-1. Install Ollama: https://ollama.ai
-2. Pull a model:
-   ```bash
-   ollama pull llama3
-   ```
-3. Add to Moth:
-   ```bash
-   moth llm add
-   # Select "Ollama" and choose your model
-   ```
-
-### Setting Up Cloud Providers
-
-#### OpenAI
-```bash
-moth llm add
-# Select "OpenAI"
-# Enter your API key from https://platform.openai.com/api-keys
-```
+### Local Installation
+For project-specific constraints.
 
-#### Anthropic (Claude)
 ```bash
-moth llm add
-# Select "Anthropic"
-# Enter your API key from https://console.anthropic.com/
+npm install moth-ai
+npx moth-ai
 ```
 
-#### Google (Gemini)
-```bash
-moth llm add
-# Select "Google"
-# Enter your API key from https://makersuite.google.com/app/apikey
-```
+**Requirements:** Node.js >= 18.0.0
 
 ---
 
-## 💡 Usage Examples
+## ⚡ Quick Start
 
-### Basic Chat
+1.  **Initialize Moth:**
+    ```bash
+    moth
+    ```
 
-```bash
-moth
-> How do I implement a binary search in TypeScript?
-```
+2.  **Configure Your First Profile:**
+    Run the interactive setup wizard to connect your preferred model.
+    ```bash
+    moth llm add
+    ```
+<img width="611" height="379" alt="image" src="https://github.com/user-attachments/assets/a90c54ee-54f7-4391-a29f-f2bdf40d5709" />
 
-### Code Refactoring
-
-```bash
-moth
-> Refactor src/utils.ts to use async/await instead of promises
-```
-
-### Debugging
-
-```bash
-moth
-> Why is my React component re-rendering infinitely?
-```
-
-### Project Analysis
-
-```bash
-moth
-> Analyze the architecture of this project and suggest improvements
-```
+3.  **Execute Commands:**
+    - **Chat:** Simply type your query.
+    - **Command Palette:** Press `Ctrl+U` to manage profiles and settings.
+    - **Switch Modes:** Press `Ctrl+B` to cycle operational modes.
 
 ---
-
-## 🔧 Troubleshooting
-
-### Command not found: moth
-
-Make sure the global npm bin directory is in your PATH:
-
-```bash
-npm config get prefix
-```
-
-Add the bin directory to your PATH in `~/.bashrc` or `~/.zshrc`:
-
-```bash
-export PATH="$PATH:$(npm config get prefix)/bin"
-```
-
-### Ollama connection error
-
-Ensure Ollama is running:
-
-```bash
-ollama serve
-```
-
-### API key errors
-
-Verify your API key is correctly configured:
-
-```bash
-moth llm list
-# Check if your profile shows the correct provider
 ```
 
----
-
-## 📝 License
-
-ISC License - see [LICENSE](LICENSE) file for details
-
----
-
-## 🤝 Contributing
-
-Contributions are welcome! Please feel free to submit a Pull Request.
-
-Repository: https://github.com/kishlay42/Moth-ai
-
----
-
-## 📚 Links
-
-- **npm Package**: https://www.npmjs.com/package/@kishlay42/moth-ai
-- **GitHub**: https://github.com/kishlay42/Moth-ai
-- **Issues**: https://github.com/kishlay42/Moth-ai/issues
-
----
+*Moth AI — Coding at the speed of thought.*
 
-**Made with ❤️ for developers who code in the terminal**

package/dist/agent/orchestrator.js

@@ -32,6 +32,8 @@ export class AgentOrchestrator {
             // In reality, we need to handle the LLM raw output, parse "Thought" and "Tool Call"
             // For now, we will rely on a Structured Output or strict parsing if the Provider supports it.
             // Since we are using Gemini, we can ask for JSON mode or specific formatting.
+            // Show waiting status
+            yield { thought: "Waiting for LLM response..." };
             const responseText = await this.callLLM(messages);
             // Parse response
             let step;
@@ -39,7 +41,7 @@ export class AgentOrchestrator {
                 step = this.parseResponse(responseText);
             }
             catch (e) {
-                yield { thought: "Failed to parse LLM response. Retrying...", toolOutput: `Error: ${e}` };
+                yield { thought: "LLM response received, processing...", toolOutput: `Parsing error: ${e}` };
                 continue;
             }
             this.state.history.push(step);
@@ -59,33 +61,209 @@ export class AgentOrchestrator {
     }
     async buildSystemPrompt(attachedFiles = []) {
         const toolDefs = this.tools.getDefinitions().map(t => `${t.name}: ${t.description} Params: ${JSON.stringify(t.parameters)}`).join('\n');
+        // Automatically scan project context
+        let projectContext = '';
+        try {
+            const contextResult = await this.tools.execute('scan_context', {});
+            const files = contextResult.split('\n').filter((f) => f.trim());
+            // Extract key information
+            const fileCount = files.length - 1; // Subtract header line
+            const extensions = new Set(files.map((f) => {
+                const match = f.match(/\.(\w+)$/);
+                return match ? match[1] : null;
+            }).filter(Boolean));
+            // Identify key files
+            const keyFiles = files.filter((f) => f.includes('package.json') ||
+                f.includes('README') ||
+                f.includes('tsconfig') ||
+                f.includes('.config')).slice(0, 10);
+            projectContext = `
+=== PROJECT CONTEXT ===
+
+**Current Directory:** ${this.root}
+**Total Files:** ${fileCount}
+**Languages/Types:** ${Array.from(extensions).join(', ')}
+
+**Key Files:**
+${keyFiles.slice(1, 11).join('\n')}
+
+**Note:** You have access to all project files. Use read_file, list_dir, or search_files to explore further.
+`;
+        }
+        catch (e) {
+            // If scanning fails, continue without context
+            projectContext = `\n=== PROJECT CONTEXT ===\n**Current Directory:** ${this.root}\n`;
+        }
         let fileContext = '';
         if (attachedFiles.length > 0) {
             fileContext = await this.buildFileContext(attachedFiles);
         }
-        return `You are Moth, an intelligent CLI coding assistant.
-You have access to the following tools:
+        return `You are Moth, an intelligent CLI coding assistant with full access to the project filesystem and tools.
+
+=== YOUR ROLE ===
+You are the DRIVER of this mecha robot (the coding assistant). You have complete control over all tools and actions.
+The user trusts you to make smart decisions about which tools to use and when.
+
+=== AVAILABLE TOOLS ===
 ${toolDefs}
 
+${projectContext}
+
 ${fileContext}
 
-IMPORTANT GUIDELINES:
-1. For general questions, explanations, or code snippets that don't need to be saved, use "finalAnswer". 
-2. Do NOT use "write_to_file" unless the user explicitly asks to save a file or implies a persistent change.
-3. If the user asks for "Hello World code", just show it in the explanation (finalAnswer). Do NOT create a file for it.
-4. Be concise and helpful.
-5. If files are referenced above, use that context to answer questions about them.
+=== TOOL SELECTION GUIDELINES ===
+
+**For File Operations:**
+
+1. **create_file** - Use when:
+   - Creating a brand new file that doesn't exist
+   - User explicitly says "create a new file"
+   - Starting a new component, module, or script
+   - Will FAIL if file already exists (use write_file instead)
+
+2. **write_file** - Use when:
+   - Completely replacing/overwriting an existing file
+   - User asks to "rewrite", "replace", or "change the entire file"
+   - Making major changes where it's easier to rewrite than patch
+   - The file is small (< 100 lines) and changes are extensive
+   - PREFERRED for simple edits - it's more reliable than edit_file
+
+3. **edit_file** - Use when:
+   - Making targeted changes to large files (> 100 lines)
+   - Only modifying specific sections while preserving the rest
+   - User asks to "modify", "update", or "change a specific part"
+   - You need to generate a Unified Diff (git diff format)
+   - REQUIRES: Exact context matching - if unsure, use write_file instead
+
+4. **read_file** - ALWAYS use before editing:
+   - Read the file first to understand its current state
+   - Verify the file exists before trying to edit it
+   - Get the exact content for generating accurate diffs
+
+**For Code Questions & Explanations:**
+
+- If user asks "show me code" or "how do I..." → Use finalAnswer (don't create files)
+- If user asks "create/write/save a file" → Use appropriate file tool
+- If user asks "edit/modify @filename" → Read file first, then use write_file or edit_file
+
+**For Project Understanding:**
+
+- Use scan_context to understand project structure
+- Use search_files to find specific files by name
+- Use search_text to find code patterns or text
+- Use read_file to examine specific files
+
+**For Commands:**
+
+- Use run_command for: npm install, git commands, running tests, etc.
+- Always show the user what command you're running
+- Explain the output in simple terms
+
+=== DECISION MAKING PROCESS ===
+
+When user asks to edit a file:
+1. First, use read_file to see current content
+2. Decide: Is this a small change or complete rewrite?
+   - Small file or major changes? → write_file (easier, more reliable)
+   - Large file with targeted changes? → edit_file (preserves rest of file)
+3. Execute the chosen tool
+4. Confirm what you did in finalAnswer
+
+When user asks a question:
+1. Is this asking for information or asking to DO something?
+   - Information → Use finalAnswer with explanation
+   - Action → Use appropriate tool
+2. If unsure, ask the user to clarify
+
+=== BEST PRACTICES ===
+
+✅ DO:
+- Read files before editing them
+- Use write_file for most edits (it's simpler and more reliable)
+- Explain what you're doing in your "thought"
+- Give clear, helpful finalAnswers
+- Use tools proactively when it makes sense
+
+❌ DON'T:
+- Create files for code examples unless explicitly asked
+- Use edit_file without reading the file first
+- Generate invalid diffs (if unsure, use write_file)
+- Make assumptions - ask if unclear
+
+=== RESPONSE FORMAT ===
 
 Format your response exactly as a JSON object:
+
+For using a tool:
 {
-  "thought": "your reasoning",
+  "thought": "I need to [action] because [reason]. I'll use [tool] to do this.",
   "toolCall": { "name": "tool_name", "arguments": { ... } }
 }
-OR if you are done/replying:
+
+For final answer:
 {
-  "thought": "reasoning",
-  "finalAnswer": "your response/code/explanation"
+  "thought": "I've completed the task. Here's what I did: [summary]",
+  "finalAnswer": "Your detailed response with code/explanation/results"
 }
+
+=== FINAL ANSWER GUIDELINES ===
+
+When you complete a task, NEVER just say "Done". Always provide a comprehensive response that includes:
+
+1. **Summary of what was accomplished:**
+   - "I've created/modified/deleted [X] file(s)"
+   - List each file with its path
+
+2. **File references:**
+   - Use relative paths from project root
+   - Example: "Created \`src/components/Button.tsx\`"
+   - Example: "Modified \`package.json\` to add dependencies"
+
+3. **What changed:**
+   - Briefly explain what was added/modified/removed
+   - Example: "Added a new Button component with TypeScript support"
+   - Example: "Updated the API endpoint to use the new authentication flow"
+
+4. **Next steps or suggestions:**
+   - What the user might want to do next
+   - Example: "You can now import this component with: \`import { Button } from './components/Button'\`"
+   - Example: "Next, you might want to: run \`npm install\` to install the new dependencies"
+
+5. **Testing/verification suggestions:**
+   - How to verify the changes work
+   - Example: "Run \`npm test\` to verify the tests pass"
+   - Example: "Try running the app with \`npm start\`"
+
+**Example of a GOOD final answer:**
+"I've created a new Button component for you!
+
+**Files created:**
+- \`src/components/Button.tsx\` - Main Button component with TypeScript
+- \`src/components/Button.css\` - Styling for the button
+
+**What it includes:**
+- Customizable size (small, medium, large)
+- Multiple variants (primary, secondary, danger)
+- TypeScript props for type safety
+- Accessible with proper ARIA labels
+
+**Next steps:**
+1. Import it in your app: \`import { Button } from './components/Button'\`
+2. Use it: \`<Button variant="primary" size="medium">Click me</Button>\`
+3. Customize the colors in \`Button.css\` to match your theme
+
+**To test:**
+Run \`npm start\` and check the component renders correctly!"
+
+**Example of a BAD final answer:**
+"Done"
+"File created"
+"Task completed"
+
+=== REMEMBER ===
+You are the intelligent driver with full control. Make smart decisions about tools.
+The user trusts you to choose the right approach. When in doubt, use the simpler tool (write_file over edit_file).
+ALWAYS provide detailed, helpful final answers with file references and next steps!
 `;
     }
     collectAttachedFiles(history) {
@@ -131,9 +309,32 @@ OR if you are done/replying:
         return fullText;
     }
     parseResponse(text) {
-        // Clean markdown code blocks if present
-        const jsonText = text.replace(/```json/g, '').replace(/```/g, '').trim();
-        return JSON.parse(jsonText);
+        try {
+            // 1. Try to find JSON in code blocks
+            const jsonMatch = text.match(/```(?:json)?\s*([\s\S]*?)\s*```/);
+            if (jsonMatch) {
+                const jsonContent = jsonMatch[1].trim();
+                return JSON.parse(jsonContent);
+            }
+            // 2. Try to find the first outer-most JSON object
+            // This regex matches from the first '{' to the last '}'
+            const objectMatch = text.match(/\{[\s\S]*\}/);
+            if (objectMatch) {
+                return JSON.parse(objectMatch[0]);
+            }
+            // 3. Fallback: Try parsing the whole text (maybe it's raw JSON)
+            return JSON.parse(text);
+        }
+        catch (e) {
+            // If all parsing fails, but the text contains "Done" or "Completed", return it as a final answer
+            if (text.toLowerCase().includes('done') || text.toLowerCase().includes('completed')) {
+                return {
+                    thought: "LLM responded with plain text indicating completion.",
+                    finalAnswer: text
+                };
+            }
+            throw new Error(`Could not parse JSON from response. Raw text: ${text.slice(0, 100)}...`);
+        }
     }
     async executeTool(name, args) {
         return await this.tools.execute(name, args);

package/dist/config/configManager.js

@@ -19,7 +19,8 @@ export function loadConfig() {
         return {
             profiles: parsed?.profiles || [],
             activeProfile: parsed?.activeProfile,
-            username: parsed?.username
+            username: parsed?.username,
+            mode: parsed?.mode || 'default' // Default to 'default' mode
         };
     }
     catch (e) {
@@ -60,3 +61,7 @@ export function setUsername(config, username) {
     config.username = username;
     return config;
 }
+export function setMode(config, mode) {
+    config.mode = mode;
+    return config;
+}

package/dist/types/modes.js

@@ -0,0 +1,29 @@
+export const MODES = {
+    default: {
+        name: 'default',
+        displayName: 'Default',
+        description: 'Balanced approach - asks for permission',
+        icon: '🔵',
+        color: 'blue',
+        autoApprove: false,
+        emphasizePlanning: false
+    },
+    plan: {
+        name: 'plan',
+        displayName: 'Plan',
+        description: 'Planning-first - creates detailed plans before execution',
+        icon: '📋',
+        color: 'magenta',
+        autoApprove: false,
+        emphasizePlanning: true
+    },
+    autopilot: {
+        name: 'autopilot',
+        displayName: 'Autopilot',
+        description: 'Auto-execute - no permission prompts',
+        icon: '🚀',
+        color: 'yellow',
+        autoApprove: true,
+        emphasizePlanning: false
+    }
+};

package/dist/ui/App.js

@@ -2,7 +2,8 @@ import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import { useState, useEffect } from 'react';
 import { Box, Text, useInput, Newline } from 'ink';
 import * as os from 'os';
-import { loadConfig, saveConfig, addProfile } from '../config/configManager.js';
+import { loadConfig, saveConfig, addProfile, setMode as saveMode } from '../config/configManager.js';
+import { MODES } from '../types/modes.js';
 import { TodoManager } from '../planning/todoManager.js';
 import { AgentOrchestrator } from '../agent/orchestrator.js';
 import { createToolRegistry } from '../tools/factory.js';
@@ -13,6 +14,7 @@ import { WordMoth } from './components/WordMoth.js';
 import { FileChip } from './components/FileChip.js';
 import { FileAutocomplete } from './components/FileAutocomplete.js';
 import { CommandPalette } from './components/CommandPalette.js';
+import { MarkdownText } from './components/MarkdownText.js';
 import { ProfileManager } from './ProfileManager.js';
 import { LLMWizard } from './wizards/LLMWizard.js';
 import { LLMRemover } from './wizards/LLMRemover.js';
@@ -23,9 +25,13 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
     // UX State
     const [showWelcome, setShowWelcome] = useState(true);
     const [isPaused, setIsPaused] = useState(false);
-    const [autopilot, setAutopilot] = useState(false);
     const [isProcessing, setIsProcessing] = useState(false);
     const [thinkingText, setThinkingText] = useState('Sauting...');
+    // Mode State
+    const [mode, setMode] = useState(() => {
+        const config = loadConfig();
+        return config.mode || 'default';
+    });
     // File Reference State
     const [selectedFiles, setSelectedFiles] = useState([]);
     const [showAutocomplete, setShowAutocomplete] = useState(false);
@@ -94,7 +100,8 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
             const root = findProjectRoot() || process.cwd();
             // Permission Callback
             const checkPermission = async (toolName, args) => {
-                if (autopilot) {
+                // Auto-approve if in autopilot mode
+                if (mode === 'autopilot') {
                     return { allowed: true };
                 }
                 return new Promise((resolve) => {
@@ -212,11 +219,11 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
         setShowCommandPalette(false);
         switch (action) {
             case 'autopilot':
-                setAutopilot(prev => !prev);
-                setMessages(prev => [...prev, {
-                        role: 'assistant',
-                        content: `Autopilot ${!autopilot ? 'enabled' : 'disabled'}.`
-                    }]);
+                // Set mode to autopilot (same as Ctrl+B cycling to autopilot)
+                setMode('autopilot');
+                const config = loadConfig();
+                const updatedConfig = saveMode(config, 'autopilot');
+                saveConfig(updatedConfig);
                 break;
             case 'llm-list':
                 setActiveWizard('llm-list');
@@ -270,6 +277,19 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
             setShowCommandPalette(prev => !prev);
             return;
         }
+        // Ctrl+B to cycle modes
+        if (input === 'b' && key.ctrl) {
+            const modes = ['default', 'plan', 'autopilot'];
+            const currentIndex = modes.indexOf(mode);
+            const nextMode = modes[(currentIndex + 1) % modes.length];
+            setMode(nextMode);
+            // Persist mode to config
+            const config = loadConfig();
+            const updatedConfig = saveMode(config, nextMode);
+            saveConfig(updatedConfig);
+            // Mode indicator updates automatically, no need for notification
+            return;
+        }
         // ESC Pause
         if (key.escape && !showAutocomplete) {
             setIsPaused(prev => !prev);
@@ -306,7 +326,11 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
                     pendingPermission.resolve({ allowed: true });
                 }
                 else if (permissionSelection === 1) {
-                    setAutopilot(true);
+                    // Enable autopilot mode
+                    setMode('autopilot');
+                    const config = loadConfig();
+                    const updatedConfig = saveMode(config, 'autopilot');
+                    saveConfig(updatedConfig);
                     pendingPermission.resolve({ allowed: true });
                 }
                 else if (permissionSelection === 2) {
@@ -319,8 +343,11 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
                 pendingPermission.resolve({ allowed: true });
             }
             else if (input === 'b' || input === 'B') {
-                // Yes - autopilot
-                setAutopilot(true);
+                // Yes - enable autopilot mode
+                setMode('autopilot');
+                const config = loadConfig();
+                const updatedConfig = saveMode(config, 'autopilot');
+                saveConfig(updatedConfig);
                 pendingPermission.resolve({ allowed: true });
             }
             else if (input === 'c' || input === 'C') {
@@ -370,7 +397,7 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
                     }]);
             } }));
     }
-    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [command === 'run' && (_jsxs(Box, { flexDirection: "row", paddingX: 1, paddingY: 0, borderStyle: "round", borderColor: "#0192e5", children: [_jsxs(Box, { flexDirection: "column", marginTop: -1, paddingRight: 1, children: [_jsx(WordMoth, { text: "MOTH", big: true }), _jsx(Box, { marginTop: -1, children: _jsx(Text, { dimColor: true, children: "v1.0.0" }) }), _jsxs(Text, { color: "#3EA0C3", children: ["Welcome, ", username || os.userInfo().username] })] }), _jsxs(Box, { flexDirection: "column", alignItems: "flex-start", marginLeft: 2, children: [_jsxs(Text, { color: "green", children: ["Active_AI: ", activeProfile?.name || 'None'] }), _jsxs(Text, { dimColor: true, children: ["Path: ", process.cwd()] }), _jsx(Text, { dimColor: true, children: "Use Ctrl+U to view commands" })] })] })), messages.length > 0 && (_jsx(Box, { flexDirection: "column", marginBottom: 1, children: messages.map((m, i) => (_jsxs(Box, { flexDirection: "row", marginBottom: 1, children: [_jsxs(Text, { color: m.role === 'user' ? 'blue' : 'green', bold: true, children: [m.role === 'user' ? 'You' : 'Moth', ":"] }), _jsxs(Text, { children: [" ", m.content] })] }, i))) })), pendingPermission && (_jsxs(Box, { borderStyle: "double", borderColor: "red", flexDirection: "column", padding: 1, marginBottom: 1, children: [_jsx(Text, { bold: true, color: "red", children: "PERMISSION REQUIRED" }), _jsxs(Text, { children: ["Tool: ", pendingPermission.toolName] }), _jsxs(Text, { children: ["Args: ", JSON.stringify(pendingPermission.args)] }), _jsx(Newline, {}), !feedbackMode ? (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { children: _jsxs(Text, { color: permissionSelection === 0 ? "green" : undefined, bold: permissionSelection === 0, children: [permissionSelection === 0 ? "> " : "  ", " [a] Yes - execute this action"] }) }), _jsx(Text, { children: _jsxs(Text, { color: permissionSelection === 1 ? "green" : undefined, bold: permissionSelection === 1, children: [permissionSelection === 1 ? "> " : "  ", " [b] Yes - enable autopilot (approve all)"] }) }), _jsx(Text, { children: _jsxs(Text, { color: permissionSelection === 2 ? "green" : undefined, bold: permissionSelection === 2, children: [permissionSelection === 2 ? "> " : "  ", " [c] Tell Moth what to do instead"] }) })] })) : (_jsxs(Box, { children: [_jsx(Text, { bold: true, children: "Feedback: " }), _jsx(Text, { children: inputVal })] }))] })), !pendingPermission && (_jsxs(Box, { flexDirection: "column", children: [isProcessing && (_jsxs(Box, { flexDirection: "column", marginBottom: 0, children: [_jsx(Text, { color: "yellow", italic: true, children: thinkingText }), status !== 'Ready' && _jsxs(Text, { color: "gray", dimColor: true, children: ["  ", status] })] })), autopilot && (_jsx(Text, { color: "magenta", children: "AUTOPILOT MODE" })), selectedFiles.length > 0 && (_jsxs(Box, { flexDirection: "column", marginBottom: 1, children: [_jsxs(Box, { flexDirection: "row", children: [_jsxs(Text, { dimColor: true, children: ["Referenced Files (", selectedFiles.length, ") "] }), _jsxs(Text, { color: "gray", dimColor: true, children: ["[Ctrl+O to ", showFileChips ? 'hide' : 'show', "]"] })] }), showFileChips && (_jsx(Box, { flexDirection: "row", flexWrap: "wrap", marginTop: 1, children: selectedFiles.map((file) => (_jsx(FileChip, { filePath: file, onRemove: () => handleFileRemove(file) }, file))) }))] })), _jsxs(Box, { borderStyle: "round", borderColor: isProcessing ? "gray" : "blue", paddingX: 1, children: [_jsx(Text, { color: isProcessing ? "gray" : "cyan", children: '> ' }), _jsx(CustomTextInput, { value: inputVal, onChange: handleInputChange, onSubmit: (val) => {
+    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [command === 'run' && (_jsxs(Box, { flexDirection: "row", paddingX: 1, paddingY: 0, borderStyle: "round", borderColor: "#0192e5", children: [_jsxs(Box, { flexDirection: "column", marginTop: -1, paddingRight: 1, children: [_jsx(WordMoth, { text: "MOTH", big: true }), _jsx(Box, { marginTop: -1, children: _jsx(Text, { dimColor: true, children: "v1.0.4" }) }), _jsxs(Text, { color: "#3EA0C3", children: ["Welcome, ", username || os.userInfo().username] })] }), _jsxs(Box, { flexDirection: "column", alignItems: "flex-start", marginLeft: 2, children: [_jsxs(Text, { color: "green", children: ["Active_AI: ", activeProfile?.name || 'None'] }), _jsxs(Text, { dimColor: true, children: ["Path: ", process.cwd()] }), _jsx(Text, { dimColor: true, children: "Use Ctrl+U to view commands" })] })] })), messages.length > 0 && (_jsx(Box, { flexDirection: "column", marginBottom: 1, children: messages.map((m, i) => (_jsxs(Box, { flexDirection: "column", marginBottom: 1, children: [_jsxs(Text, { color: m.role === 'user' ? 'blue' : 'green', bold: true, children: [m.role === 'user' ? 'You' : 'Moth', ":"] }), m.role === 'assistant' ? (_jsx(MarkdownText, { content: m.content })) : (_jsxs(Text, { children: [" ", m.content] }))] }, i))) })), pendingPermission && (_jsxs(Box, { borderStyle: "double", borderColor: "red", flexDirection: "column", padding: 1, marginBottom: 1, children: [_jsx(Text, { bold: true, color: "red", children: "PERMISSION REQUIRED" }), _jsxs(Text, { children: ["Tool: ", pendingPermission.toolName] }), _jsxs(Text, { children: ["Args: ", JSON.stringify(pendingPermission.args)] }), _jsx(Newline, {}), !feedbackMode ? (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { children: _jsxs(Text, { color: permissionSelection === 0 ? "green" : undefined, bold: permissionSelection === 0, children: [permissionSelection === 0 ? "> " : "  ", " [a] Yes - execute this action"] }) }), _jsx(Text, { children: _jsxs(Text, { color: permissionSelection === 1 ? "green" : undefined, bold: permissionSelection === 1, children: [permissionSelection === 1 ? "> " : "  ", " [b] Yes - enable autopilot (approve all)"] }) }), _jsx(Text, { children: _jsxs(Text, { color: permissionSelection === 2 ? "green" : undefined, bold: permissionSelection === 2, children: [permissionSelection === 2 ? "> " : "  ", " [c] Tell Moth what to do instead"] }) })] })) : (_jsxs(Box, { children: [_jsx(Text, { bold: true, children: "Feedback: " }), _jsx(Text, { children: inputVal })] }))] })), !pendingPermission && (_jsxs(Box, { flexDirection: "column", children: [isProcessing && (_jsxs(Box, { flexDirection: "column", marginBottom: 0, children: [_jsx(Text, { color: "yellow", italic: true, children: thinkingText }), status !== 'Ready' && _jsxs(Text, { color: "gray", dimColor: true, children: ["  ", status] })] })), selectedFiles.length > 0 && (_jsxs(Box, { flexDirection: "column", marginBottom: 1, children: [_jsxs(Box, { flexDirection: "row", children: [_jsxs(Text, { dimColor: true, children: ["Referenced Files (", selectedFiles.length, ") "] }), _jsxs(Text, { color: "gray", dimColor: true, children: ["[Ctrl+O to ", showFileChips ? 'hide' : 'show', "]"] })] }), showFileChips && (_jsx(Box, { flexDirection: "row", flexWrap: "wrap", marginTop: 1, children: selectedFiles.map((file) => (_jsx(FileChip, { filePath: file, onRemove: () => handleFileRemove(file) }, file))) }))] })), _jsxs(Box, { borderStyle: "round", borderColor: isProcessing ? "gray" : "blue", paddingX: 1, children: [_jsx(Text, { color: isProcessing ? "gray" : "cyan", children: '> ' }), _jsx(CustomTextInput, { value: inputVal, onChange: handleInputChange, onSubmit: (val) => {
                                     // Don't submit if autocomplete is open
                                     if (showAutocomplete) {
                                         return;
@@ -383,5 +410,5 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
                                         setInputVal('');
                                         setSelectedFiles([]);
                                     }
-                                }, focus: !isProcessing && !pendingPermission })] }), showAutocomplete && (_jsx(FileAutocomplete, { files: availableFiles, query: autocompleteQuery, selectedIndex: autocompleteIndex, onSelect: handleFileSelect })), showCommandPalette && (_jsx(CommandPalette, { onExecute: executeCommand, onClose: () => setShowCommandPalette(false) })), _jsx(Box, { flexDirection: "row", justifyContent: "flex-end", children: _jsx(Text, { color: "gray", dimColor: true, children: activeProfile?.name }) })] }))] }));
+                                }, focus: !isProcessing && !pendingPermission })] }), showAutocomplete && (_jsx(FileAutocomplete, { files: availableFiles, query: autocompleteQuery, selectedIndex: autocompleteIndex, onSelect: handleFileSelect })), showCommandPalette && (_jsx(CommandPalette, { onExecute: executeCommand, onClose: () => setShowCommandPalette(false) })), _jsxs(Box, { flexDirection: "row", justifyContent: "space-between", children: [_jsxs(Box, { flexDirection: "row", children: [_jsxs(Text, { color: MODES[mode].color, children: [MODES[mode].icon, " ", MODES[mode].displayName.toUpperCase(), " MODE"] }), _jsx(Text, { dimColor: true, children: " | Ctrl+B to switch" })] }), _jsx(Text, { color: "gray", dimColor: true, children: activeProfile?.name })] })] }))] }));
 };

package/dist/ui/components/MarkdownText.js

@@ -0,0 +1,130 @@
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
+import { Box, Text } from 'ink';
+/**
+ * Renders markdown-style text with colors in the terminal
+ * Supports:
+ * - # Headings (cyan, bold)
+ * - ## Subheadings (blue, bold)
+ * - ### Sub-subheadings (green)
+ * - **bold** text
+ * - `code` inline (yellow)
+ * - ```code blocks``` (yellow, dimmed)
+ * - Tables (formatted)
+ * - Bullet points
+ */
+export const MarkdownText = ({ content }) => {
+    const lines = content.split('\n');
+    const elements = [];
+    let inCodeBlock = false;
+    let codeBlockContent = [];
+    lines.forEach((line, index) => {
+        // Code block detection
+        if (line.trim().startsWith('```')) {
+            if (inCodeBlock) {
+                // End of code block - render accumulated content
+                elements.push(_jsx(Box, { flexDirection: "column", marginY: 1, paddingX: 2, borderStyle: "single", borderColor: "yellow", children: codeBlockContent.map((codeLine, i) => (_jsx(Text, { color: "yellow", dimColor: true, children: codeLine }, i))) }, `code-${index}`));
+                codeBlockContent = [];
+                inCodeBlock = false;
+            }
+            else {
+                // Start of code block
+                inCodeBlock = true;
+                const lang = line.trim().slice(3);
+                if (lang) {
+                    elements.push(_jsxs(Text, { color: "gray", dimColor: true, children: ["[", lang, "]"] }, `lang-${index}`));
+                }
+            }
+            return;
+        }
+        // Inside code block
+        if (inCodeBlock) {
+            codeBlockContent.push(line);
+            return;
+        }
+        // Main heading (# )
+        if (line.startsWith('# ')) {
+            elements.push(_jsx(Text, { color: "cyan", bold: true, children: line.slice(2) }, index));
+            return;
+        }
+        // Subheading (## )
+        if (line.startsWith('## ')) {
+            elements.push(_jsx(Text, { color: "blue", bold: true, children: line.slice(3) }, index));
+            return;
+        }
+        // Sub-subheading (### )
+        if (line.startsWith('### ')) {
+            elements.push(_jsx(Text, { color: "green", children: line.slice(4) }, index));
+            return;
+        }
+        // Table detection (lines with | characters)
+        if (line.includes('|') && line.trim().startsWith('|')) {
+            const cells = line.split('|').filter(c => c.trim()).map(c => c.trim());
+            // Check if it's a separator line (|---|---|)
+            if (cells.every(c => /^-+$/.test(c))) {
+                elements.push(_jsx(Text, { color: "gray", dimColor: true, children: line }, index));
+                return;
+            }
+            // Regular table row
+            elements.push(_jsx(Box, { flexDirection: "row", children: cells.map((cell, i) => (_jsx(Text, { color: i === 0 ? "cyan" : undefined, children: cell.padEnd(20, ' ') }, i))) }, index));
+            return;
+        }
+        // Bullet points
+        if (line.trim().startsWith('* ') || line.trim().startsWith('- ')) {
+            const indent = line.search(/[*-]/);
+            const content = line.slice(indent + 2);
+            elements.push(_jsxs(Text, { children: [' '.repeat(indent), _jsx(Text, { color: "yellow", children: "\u2022 " }), renderInlineFormatting(content)] }, index));
+            return;
+        }
+        // Numbered lists
+        if (/^\s*\d+\.\s/.test(line)) {
+            const match = line.match(/^(\s*)(\d+)\.\s(.*)$/);
+            if (match) {
+                const [, indent, num, content] = match;
+                elements.push(_jsxs(Text, { children: [indent, _jsxs(Text, { color: "cyan", children: [num, ". "] }), renderInlineFormatting(content)] }, index));
+                return;
+            }
+        }
+        // Empty lines
+        if (line.trim() === '') {
+            elements.push(_jsx(Text, { children: " " }, index));
+            return;
+        }
+        // Regular text with inline formatting
+        elements.push(_jsx(Text, { children: renderInlineFormatting(line) }, index));
+    });
+    return _jsx(Box, { flexDirection: "column", children: elements });
+};
+/**
+ * Renders inline markdown formatting like **bold**, `code`, etc.
+ */
+function renderInlineFormatting(text) {
+    const parts = [];
+    let currentIndex = 0;
+    let key = 0;
+    // Match patterns: `code`, **bold**
+    const pattern = /(`[^`]+`)|(\*\*[^*]+\*\*)/g;
+    let match;
+    while ((match = pattern.exec(text)) !== null) {
+        // Add text before the match
+        if (match.index > currentIndex) {
+            parts.push(_jsx(Text, { children: text.slice(currentIndex, match.index) }, key++));
+        }
+        // Add the formatted match
+        if (match[1]) {
+            // Code (backticks)
+            const code = match[1].slice(1, -1);
+            parts.push(_jsx(Text, { color: "yellow", backgroundColor: "black", children: code }, key++));
+        }
+        else if (match[2]) {
+            // Bold
+            const bold = match[2].slice(2, -2);
+            parts.push(_jsx(Text, { bold: true, children: bold }, key++));
+        }
+        currentIndex = match.index + match[0].length;
+    }
+    // Add remaining text
+    if (currentIndex < text.length) {
+        parts.push(_jsx(Text, { children: text.slice(currentIndex) }, key++));
+    }
+    return parts.length > 0 ? _jsx(_Fragment, { children: parts }) : text;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "moth-ai",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Local, LLM-agnostic code intelligence CLI",
   "main": "dist/index.js",
   "bin": {
@@ -13,6 +13,7 @@
     "build": "tsc",
     "prepublishOnly": "npm run build",
     "start": "node dist/index.js",
+    "test": "node --import tsx --test tests/**/*.test.ts",
     "lint": "eslint src/**/*.ts",
     "format": "prettier --write \"src/**/*.ts\""
   },
@@ -50,6 +51,7 @@
     "eslint-plugin-prettier": "^5.5.4",
     "prettier": "^3.7.4",
     "ts-node": "^10.9.2",
+    "tsx": "^4.19.2",
     "typescript": "^5.9.3"
   },
   "dependencies": {