@kishlay42/moth-ai 1.0.2 → 1.0.3

package/README.md

@@ -1,17 +1,85 @@
 # 🦋 Moth AI
 
+[![npm version](https://img.shields.io/npm/v/@kishlay42/moth-ai.svg)](https://www.npmjs.com/package/@kishlay42/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**
 
-<img width="1095" height="504" alt="Screenshot 2025-12-27 015039" src="https://github.com/user-attachments/assets/23b83a6b-2b63-45af-b9ec-a6dcb0a89b2f" />
+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
 
-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**.
+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
 
 ---
 
 ## ✨ Key Features
 
 ### 🧠 LLM-Agnostic & Local-First
+
 Use **any LLM**, local or cloud — switch instantly without changing workflows.
 
 - **Local (via Ollama)**  
@@ -24,32 +92,34 @@ Use **any LLM**, local or cloud — switch instantly without changing workflows.
   - Anthropic (Claude 3.5 Sonnet)
   - Google (Gemini)
 
-<img width="1093" height="241" alt="Screenshot 2025-12-27 014805" src="https://github.com/user-attachments/assets/2de67c9d-f562-4ce3-8bc6-51e2066b69ae" />
-
-_Easily switch between local and cloud models_
+<img width="1093" height="241" alt="LLM Switching" src="https://github.com/user-attachments/assets/2de67c9d-f562-4ce3-8bc6-51e2066b69ae" />
 
 ---
 
 ### 🤖 Agentic Capabilities
-Moth is not just a chatbot — it’s an **AI agent**.
+
+Moth is not just a chatbot — it's an **AI agent**.
 
 - **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
+
 You stay in control — always.
 
 - Explicit approval before file edits or command execution  
 - Granular permissions per action  
-- Autopilot mode for trusted workflows  
+- **Autopilot mode** for trusted workflows  
 - Feedback loop to guide the agent instead of blind execution  
 
 ---
 
 ### 🎭 Moth Profiles
+
 Save and switch between different AI personalities.
 
 - **Coding Profile** – Optimized for TypeScript / Python  
@@ -57,17 +127,193 @@ Save and switch between different AI personalities.
 - **Fast Profile** – Lightweight local model for quick answers  
 
 ---
-## 🚀 Installation
 
-### Global (recommended)
+## � CLI Commands
+
+### Main Commands
+
 ```bash
-npm install -g @kishlay42/moth-ai
+# Start interactive chat
+moth
+
+# Show help
+moth --help
+
+# Display version
+moth --version
 ```
 
-### Project-local
+### LLM Profile Management
+
+```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
 ```
-npm install @kishlay42/moth-ai
-npx moth
+
+### 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"
+```
+
+### 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
+```
+
+#### Anthropic (Claude)
+```bash
+moth llm add
+# Select "Anthropic"
+# Enter your API key from https://console.anthropic.com/
+```
+
+#### Google (Gemini)
+```bash
+moth llm add
+# Select "Google"
+# Enter your API key from https://makersuite.google.com/app/apikey
+```
+
+---
+
+## 💡 Usage Examples
+
+### Basic Chat
+
+```bash
+moth
+> How do I implement a binary search in TypeScript?
+```
+
+### 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
 ```
 
+---
+
+## 🔧 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
+
+---
 
+**Made with ❤️ for developers who code in the terminal**

package/dist/agent/orchestrator.js

@@ -1,10 +1,15 @@
+import { formatFileForContext, truncateFileContent } from '../utils/fileUtils.js';
+import * as fs from 'fs/promises';
+import * as path from 'path';
 export class AgentOrchestrator {
     config;
     state;
     tools;
-    constructor(config, registry) {
+    root;
+    constructor(config, registry, root = process.cwd()) {
         this.config = config;
         this.tools = registry;
+        this.root = root;
         this.state = {
             history: [],
             maxSteps: config.maxSteps || 10
@@ -13,8 +18,10 @@ export class AgentOrchestrator {
     async *run(prompt, history = []) {
         let currentPrompt = prompt;
         for (let i = 0; i < this.state.maxSteps; i++) {
-            // Construct system prompt with tools
-            const systemPrompt = this.buildSystemPrompt();
+            // Collect all attached files from history
+            const allAttachedFiles = this.collectAttachedFiles(history);
+            // Construct system prompt with tools and file context
+            const systemPrompt = await this.buildSystemPrompt(allAttachedFiles);
             // Full context: System -> History -> Current State
             const messages = [
                 { role: 'user', content: systemPrompt }, // In many APIs system prompt is special, here we use user role as generic fallback or modify Client to handle system
@@ -50,17 +57,24 @@ export class AgentOrchestrator {
         }
         return "Max steps reached.";
     }
-    buildSystemPrompt() {
+    async buildSystemPrompt(attachedFiles = []) {
         const toolDefs = this.tools.getDefinitions().map(t => `${t.name}: ${t.description} Params: ${JSON.stringify(t.parameters)}`).join('\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:
 ${toolDefs}
 
+${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.
 
 Format your response exactly as a JSON object:
 {
@@ -74,6 +88,36 @@ OR if you are done/replying:
 }
 `;
     }
+    collectAttachedFiles(history) {
+        const files = new Set();
+        for (const msg of history) {
+            if (msg.attachedFiles) {
+                msg.attachedFiles.forEach(f => files.add(f));
+            }
+        }
+        return Array.from(files);
+    }
+    async buildFileContext(filePaths) {
+        const fileContents = ['=== Referenced Files ===\n'];
+        for (const filePath of filePaths) {
+            try {
+                const fullPath = path.join(this.root, filePath);
+                // Security: Prevent breaking out of root
+                if (!fullPath.startsWith(this.root)) {
+                    fileContents.push(`File: ${filePath}\nError: Access denied (outside project root)\n`);
+                    continue;
+                }
+                const content = await fs.readFile(fullPath, 'utf-8');
+                const truncated = truncateFileContent(content, 500);
+                const formatted = formatFileForContext(filePath, truncated);
+                fileContents.push(formatted);
+            }
+            catch (e) {
+                fileContents.push(`File: ${filePath}\nError: ${e.message}\n`);
+            }
+        }
+        return fileContents.join('\n');
+    }
     async callLLM(messages) {
         // Direct integration with LLM Client
         // This assumes Client has a simple chat interface returning string

package/dist/index.js

@@ -114,11 +114,12 @@ program.command('use')
     let config = loadConfig();
     if (!name) {
         console.log("No active profile selected. Please select one:");
-        render(React.createElement(ProfileManager, {
+        const { unmount } = render(React.createElement(ProfileManager, {
             config,
             onSelect: (selected) => {
                 // onSelect saves internally
-                process.exit(0);
+                unmount();
+                setTimeout(() => startChatSession(), 500);
             }
         }));
         return;
@@ -139,9 +140,12 @@ program.command('remove')
     .action(async (name) => {
     let config = loadConfig();
     if (!name) {
-        render(React.createElement(LLMRemover, {
+        const { unmount } = render(React.createElement(LLMRemover, {
             config,
-            onExit: () => process.exit(0)
+            onExit: () => {
+                unmount();
+                setTimeout(() => startChatSession(), 500);
+            }
         }));
         return;
     }
@@ -160,7 +164,7 @@ program.command('add')
     .action(async (options) => {
     // Re-use exactly the same logic as llm add
     if (!options.name && !options.provider && !options.model) {
-        render(React.createElement(LLMWizard, {
+        const { unmount } = render(React.createElement(LLMWizard, {
             onComplete: async (resultConfig) => {
                 let config = loadConfig();
                 config = addProfile(config, resultConfig);
@@ -170,12 +174,14 @@ program.command('add')
                 config = setActiveProfile(config, resultConfig.name);
                 saveConfig(config);
                 console.log(`Profile '${resultConfig.name}' created.`);
-                // Chain to main chat
-                process.exit(0); // For now, just exit as per current wizard
+                // Redirect to chat
+                unmount();
+                setTimeout(() => startChatSession(), 500);
             },
             onCancel: () => {
                 console.log('Setup cancelled.');
-                process.exit(0);
+                unmount();
+                setTimeout(() => startChatSession(), 500);
             }
         }));
         return;
@@ -228,7 +234,7 @@ llm.command('add')
     .action(async (options) => {
     // If no options provided, launch wizard
     if (!options.name && !options.provider && !options.model) {
-        render(React.createElement(LLMWizard, {
+        const { unmount } = render(React.createElement(LLMWizard, {
             onComplete: async (resultConfig) => {
                 let config = loadConfig();
                 config = addProfile(config, resultConfig);
@@ -240,11 +246,13 @@ llm.command('add')
                 }
                 saveConfig(config);
                 console.log(`Profile '${resultConfig.name}' added successfully!`);
-                process.exit(0);
+                unmount();
+                setTimeout(() => startChatSession(), 500);
             },
             onCancel: () => {
                 console.log('Setup cancelled.');
-                process.exit(0);
+                unmount();
+                setTimeout(() => startChatSession(), 500);
             }
         }));
         return;
@@ -298,9 +306,12 @@ llm.command('remove')
     .action(async (name) => {
     let config = loadConfig();
     if (!name) {
-        render(React.createElement(LLMRemover, {
+        const { unmount } = render(React.createElement(LLMRemover, {
             config,
-            onExit: () => process.exit(0)
+            onExit: () => {
+                unmount();
+                setTimeout(() => startChatSession(), 500);
+            }
         }));
         return;
     }

package/dist/ui/App.js

@@ -2,12 +2,20 @@ 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 { TodoManager } from '../planning/todoManager.js';
 import { AgentOrchestrator } from '../agent/orchestrator.js';
 import { createToolRegistry } from '../tools/factory.js';
 import { findProjectRoot } from '../utils/paths.js';
-import TextInput from 'ink-text-input';
+import { ProjectScanner } from '../context/scanner.js';
+import { CustomTextInput } from './components/CustomTextInput.js';
 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 { ProfileManager } from './ProfileManager.js';
+import { LLMWizard } from './wizards/LLMWizard.js';
+import { LLMRemover } from './wizards/LLMRemover.js';
 export const App = ({ command, args, todoManager: propTodoManager, username }) => {
     const [messages, setMessages] = useState([]);
     const [inputVal, setInputVal] = useState('');
@@ -18,6 +26,17 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
     const [autopilot, setAutopilot] = useState(false);
     const [isProcessing, setIsProcessing] = useState(false);
     const [thinkingText, setThinkingText] = useState('Sauting...');
+    // File Reference State
+    const [selectedFiles, setSelectedFiles] = useState([]);
+    const [showAutocomplete, setShowAutocomplete] = useState(false);
+    const [autocompleteQuery, setAutocompleteQuery] = useState('');
+    const [autocompleteIndex, setAutocompleteIndex] = useState(0);
+    const [availableFiles, setAvailableFiles] = useState([]);
+    const [showFileChips, setShowFileChips] = useState(false); // Collapsed by default
+    // Command Palette State
+    const [showCommandPalette, setShowCommandPalette] = useState(false);
+    // Active Wizard State (for command palette execution)
+    const [activeWizard, setActiveWizard] = useState(null);
     // Effect for cycling thinking text
     useEffect(() => {
         if (isProcessing) {
@@ -28,6 +47,21 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
             return () => clearInterval(interval);
         }
     }, [isProcessing]);
+    // Load available files on mount
+    useEffect(() => {
+        const loadFiles = async () => {
+            try {
+                const root = findProjectRoot() || process.cwd();
+                const scanner = new ProjectScanner(root);
+                const files = await scanner.scan();
+                setAvailableFiles(files);
+            }
+            catch (e) {
+                console.error('Failed to load files:', e);
+            }
+        };
+        loadFiles();
+    }, []);
     // Permission State
     const [pendingPermission, setPendingPermission] = useState(null);
     const [feedbackMode, setFeedbackMode] = useState(false);
@@ -45,12 +79,16 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
             runAgent(initialPrompt);
         }
     }, [initialPrompt]);
-    const runAgent = async (userPrompt) => {
+    const runAgent = async (userPrompt, attachedFiles) => {
         if (showWelcome)
             setShowWelcome(false);
         setIsProcessing(true);
-        // Create new user message
-        const userMsg = { role: 'user', content: userPrompt };
+        // Create new user message with attached files
+        const userMsg = {
+            role: 'user',
+            content: userPrompt,
+            attachedFiles: attachedFiles && attachedFiles.length > 0 ? attachedFiles : undefined
+        };
         setMessages(prev => [...prev, userMsg]);
         try {
             const root = findProjectRoot() || process.cwd();
@@ -76,7 +114,7 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
                 model: client,
                 tools: registry.getDefinitions(),
                 maxSteps: 10
-            }, registry);
+            }, registry, root);
             let finalAnswer = "";
             // Pass accumulated history to the agent
             // 'messages' state contains history prior to this turn.
@@ -111,9 +149,129 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
             setIsProcessing(false);
         }
     };
+    // Handle input changes to detect @ for autocomplete
+    const handleInputChange = (value) => {
+        setInputVal(value);
+        // Extract and track file references from @ mentions
+        const referencedFiles = extractFileReferences(value);
+        setSelectedFiles(referencedFiles);
+        // Detect @ symbol for file autocomplete
+        const atIndex = value.lastIndexOf('@');
+        if (atIndex !== -1) {
+            const afterAt = value.slice(atIndex + 1);
+            // Close autocomplete if there's a space after @ (end of filename)
+            if (afterAt.includes(' ')) {
+                setShowAutocomplete(false);
+                return;
+            }
+            const query = afterAt;
+            // Only show autocomplete if @ is at start or after a space
+            if (atIndex === 0 || value[atIndex - 1] === ' ') {
+                setAutocompleteQuery(query);
+                setShowAutocomplete(true);
+                setAutocompleteIndex(0);
+            }
+        }
+        else {
+            setShowAutocomplete(false);
+        }
+    };
+    // Get filtered files for autocomplete
+    const getFilteredFiles = () => {
+        return availableFiles
+            .filter(file => {
+            const lowerFile = file.toLowerCase();
+            const lowerQuery = autocompleteQuery.toLowerCase();
+            return lowerFile.includes(lowerQuery);
+        })
+            .filter(file => !selectedFiles.includes(file)) // Exclude already selected
+            .slice(0, 10);
+    };
+    const handleFileSelect = (file) => {
+        // Replace the entire input value to ensure cursor ends up at the end
+        const atIndex = inputVal.lastIndexOf('@');
+        if (atIndex !== -1) {
+            const beforeAt = inputVal.slice(0, atIndex);
+            // Set the complete new value - this places cursor at the end automatically
+            const newInput = beforeAt + '@' + file + ' ';
+            setInputVal(newInput);
+        }
+        setShowAutocomplete(false);
+        setAutocompleteQuery('');
+    };
+    const handleFileRemove = (file) => {
+        setSelectedFiles(prev => prev.filter(f => f !== file));
+    };
+    // Extract file references from input text
+    const extractFileReferences = (text) => {
+        const atMatches = text.match(/@[\w\/.\-]+/g) || [];
+        return atMatches.map(match => match.slice(1)); // Remove @ prefix
+    };
+    // Execute command from command palette
+    const executeCommand = (action) => {
+        setShowCommandPalette(false);
+        switch (action) {
+            case 'autopilot':
+                setAutopilot(prev => !prev);
+                setMessages(prev => [...prev, {
+                        role: 'assistant',
+                        content: `Autopilot ${!autopilot ? 'enabled' : 'disabled'}.`
+                    }]);
+                break;
+            case 'llm-list':
+                setActiveWizard('llm-list');
+                break;
+            case 'llm-add':
+                setActiveWizard('llm-add');
+                break;
+            case 'llm-use':
+                setActiveWizard('llm-use');
+                break;
+            case 'llm-remove':
+                setActiveWizard('llm-remove');
+                break;
+            case 'help':
+                setMessages(prev => [...prev, {
+                        role: 'assistant',
+                        content: 'Moth AI - Your terminal coding assistant.\n\nKeyboard Shortcuts:\n- Ctrl+U: Open command palette\n- Ctrl+O: Toggle file references\n- Esc: Pause/Resume\n\nFor full help, run: moth --help'
+                    }]);
+                break;
+        }
+    };
     useInput((input, key) => {
+        // Autocomplete Navigation
+        if (showAutocomplete && !pendingPermission) {
+            const filteredFiles = getFilteredFiles();
+            if (key.escape) {
+                setShowAutocomplete(false);
+                return;
+            }
+            if (key.upArrow) {
+                setAutocompleteIndex(prev => (prev - 1 + filteredFiles.length) % filteredFiles.length);
+                return;
+            }
+            if (key.downArrow) {
+                setAutocompleteIndex(prev => (prev + 1) % filteredFiles.length);
+                return;
+            }
+            // Use Enter to select file from autocomplete
+            if (key.return && filteredFiles.length > 0) {
+                handleFileSelect(filteredFiles[autocompleteIndex]);
+                return;
+            }
+        }
+        // Ctrl+O to toggle file chips visibility
+        if (input === 'o' && key.ctrl) {
+            setShowFileChips(prev => !prev);
+            return;
+        }
+        // Ctrl+U to toggle command palette
+        if (input === 'u' && key.ctrl) {
+            setShowCommandPalette(prev => !prev);
+            return;
+        }
         // ESC Pause
-        if (key.escape) {
+        if (key.escape && !showAutocomplete) {
             setIsPaused(prev => !prev);
             setStatus(prev => prev === 'Paused' ? 'Resumed' : 'Paused');
             return;
@@ -173,10 +331,57 @@ export const App = ({ command, args, todoManager: propTodoManager, username }) =
         }
     });
     // --- RENDER ---
-    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [(command === 'run' && showWelcome && messages.length === 0) && (_jsxs(Box, { flexDirection: "column", paddingX: 1, paddingBottom: 1, paddingTop: 0, borderStyle: "round", borderColor: "#0192e5", alignItems: "flex-start", children: [_jsx(WordMoth, { text: "MOTH", big: true }), _jsxs(Box, { flexDirection: "column", alignItems: "flex-start", marginTop: -1, children: [_jsx(Text, { dimColor: true, children: "v1.0.0" }), _jsxs(Text, { color: "#3EA0C3", bold: true, children: ["Welcome back, ", username || os.userInfo().username, "."] }), _jsxs(Text, { color: "green", children: ["Active AI: ", activeProfile?.name || 'None'] }), _jsxs(Text, { dimColor: true, children: ["Path: ", process.cwd()] }), _jsx(Text, { dimColor: true, children: "Run \"moth --help\" to view all 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" })), _jsxs(Box, { borderStyle: "round", borderColor: isProcessing ? "gray" : "blue", paddingX: 1, children: [_jsx(Text, { color: isProcessing ? "gray" : "cyan", children: '> ' }), _jsx(TextInput, { value: inputVal, onChange: setInputVal, onSubmit: (val) => {
+    // Render wizards if active
+    if (activeWizard === 'llm-list' || activeWizard === 'llm-use') {
+        const config = loadConfig();
+        return (_jsx(ProfileManager, { config: config, onSelect: (profile) => {
+                setActiveWizard(null);
+                setMessages(prev => [...prev, {
+                        role: 'assistant',
+                        content: `Switched to profile '${profile.name}'.`
+                    }]);
+            } }));
+    }
+    if (activeWizard === 'llm-add') {
+        return (_jsx(LLMWizard, { onComplete: (newProfile) => {
+                const config = loadConfig();
+                const updatedConfig = addProfile(config, newProfile);
+                saveConfig(updatedConfig);
+                setActiveWizard(null);
+                setMessages(prev => [...prev, {
+                        role: 'assistant',
+                        content: `Profile '${newProfile.name}' added successfully!`
+                    }]);
+            }, onCancel: () => {
+                setActiveWizard(null);
+                setMessages(prev => [...prev, {
+                        role: 'assistant',
+                        content: 'Cancelled.'
+                    }]);
+            } }));
+    }
+    if (activeWizard === 'llm-remove') {
+        const config = loadConfig();
+        return (_jsx(LLMRemover, { config: config, onExit: () => {
+                setActiveWizard(null);
+                setMessages(prev => [...prev, {
+                        role: 'assistant',
+                        content: 'Returned to chat.'
+                    }]);
+            } }));
+    }
+    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) => {
+                                    // Don't submit if autocomplete is open
+                                    if (showAutocomplete) {
+                                        return;
+                                    }
                                     if (val.trim() && !isProcessing) {
-                                        runAgent(val);
+                                        // Extract file references from @ mentions in text
+                                        const referencedFiles = extractFileReferences(val);
+                                        const allFiles = [...new Set([...selectedFiles, ...referencedFiles])];
+                                        runAgent(val, allFiles.length > 0 ? allFiles : undefined);
                                         setInputVal('');
+                                        setSelectedFiles([]);
                                     }
-                                }, focus: !isProcessing && !pendingPermission })] }), _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) })), _jsx(Box, { flexDirection: "row", justifyContent: "flex-end", children: _jsx(Text, { color: "gray", dimColor: true, children: activeProfile?.name }) })] }))] }));
 };

package/dist/ui/components/CommandPalette.js

@@ -0,0 +1,29 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useState } from 'react';
+import { Box, Text, useInput } from 'ink';
+const commands = [
+    { name: 'LLM: List Profiles', description: 'Show all configured LLM profiles', action: 'llm-list' },
+    { name: 'LLM: Add Profile', description: 'Add a new LLM profile', action: 'llm-add' },
+    { name: 'LLM: Switch Profile', description: 'Switch to a different LLM profile', action: 'llm-use' },
+    { name: 'LLM: Remove Profile', description: 'Remove an LLM profile', action: 'llm-remove' },
+    { name: 'Toggle Autopilot', description: 'Enable/disable autopilot mode', action: 'autopilot' },
+    { name: 'Show Help', description: 'Display help information', action: 'help' },
+];
+export const CommandPalette = ({ onExecute, onClose }) => {
+    const [selectedIndex, setSelectedIndex] = useState(0);
+    useInput((input, key) => {
+        if (key.upArrow) {
+            setSelectedIndex(prev => Math.max(0, prev - 1));
+        }
+        else if (key.downArrow) {
+            setSelectedIndex(prev => Math.min(commands.length - 1, prev + 1));
+        }
+        else if (key.return) {
+            onExecute(commands[selectedIndex].action);
+        }
+        else if (key.escape || (input === 'u' && key.ctrl)) {
+            onClose();
+        }
+    });
+    return (_jsx(Box, { flexDirection: "column", children: _jsxs(Box, { borderStyle: "round", borderColor: "cyan", flexDirection: "column", paddingX: 2, paddingY: 1, children: [_jsx(Text, { bold: true, color: "cyan", children: "Command Palette (Ctrl+U to close)" }), _jsx(Text, { dimColor: true, children: "Select a command and press Enter" }), _jsx(Box, { marginTop: 1, flexDirection: "column", children: commands.map((cmd, index) => (_jsxs(Box, { marginY: 0, children: [_jsxs(Text, { color: index === selectedIndex ? 'green' : undefined, bold: index === selectedIndex, children: [index === selectedIndex ? '> ' : '  ', cmd.name] }), _jsxs(Text, { dimColor: true, children: [" - ", cmd.description] })] }, cmd.action))) })] }) }));
+};

package/dist/ui/components/CustomTextInput.js

@@ -0,0 +1,75 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useState, useEffect } from 'react';
+import { Box, Text, useInput } from 'ink';
+export const CustomTextInput = ({ value: externalValue, onChange, onSubmit, focus = true, placeholder = '' }) => {
+    const [internalValue, setInternalValue] = useState(externalValue);
+    const [cursorPos, setCursorPos] = useState(externalValue.length);
+    // Sync with external value changes (e.g., from file path insertion)
+    useEffect(() => {
+        setInternalValue(externalValue);
+        // Place cursor at end when value changes externally
+        setCursorPos(externalValue.length);
+    }, [externalValue]);
+    useInput((input, key) => {
+        if (!focus)
+            return;
+        // Submit on Enter
+        if (key.return) {
+            onSubmit(internalValue);
+            return;
+        }
+        // Backspace - delete character before cursor
+        if (key.backspace) {
+            if (cursorPos > 0) {
+                const newValue = internalValue.slice(0, cursorPos - 1) + internalValue.slice(cursorPos);
+                setInternalValue(newValue);
+                setCursorPos(cursorPos - 1);
+                onChange(newValue);
+            }
+            return;
+        }
+        // Delete - delete character after cursor
+        if (key.delete) {
+            if (cursorPos < internalValue.length) {
+                const newValue = internalValue.slice(0, cursorPos) + internalValue.slice(cursorPos + 1);
+                setInternalValue(newValue);
+                onChange(newValue);
+            }
+            return;
+        }
+        // Left arrow - move cursor left
+        if (key.leftArrow) {
+            setCursorPos(Math.max(0, cursorPos - 1));
+            return;
+        }
+        // Right arrow - move cursor right
+        if (key.rightArrow) {
+            setCursorPos(Math.min(internalValue.length, cursorPos + 1));
+            return;
+        }
+        // Home - move to start
+        if (key.home) {
+            setCursorPos(0);
+            return;
+        }
+        // End - move to end
+        if (key.end) {
+            setCursorPos(internalValue.length);
+            return;
+        }
+        // Regular character input (ignore ctrl/meta combinations)
+        if (input && !key.ctrl && !key.meta && input.length === 1) {
+            const newValue = internalValue.slice(0, cursorPos) + input + internalValue.slice(cursorPos);
+            setInternalValue(newValue);
+            setCursorPos(cursorPos + 1);
+            onChange(newValue);
+            return;
+        }
+    }, { isActive: focus });
+    // Render the input with cursor
+    const displayValue = internalValue;
+    const beforeCursor = displayValue.slice(0, cursorPos);
+    const cursorChar = displayValue[cursorPos] || ' '; // Always show space if nothing there
+    const afterCursor = displayValue.slice(cursorPos + 1);
+    return (_jsxs(Box, { children: [_jsx(Text, { children: beforeCursor }), _jsx(Text, { inverse: true, children: cursorChar }), _jsx(Text, { children: afterCursor })] }));
+};

package/dist/ui/components/FileAutocomplete.js

@@ -0,0 +1,16 @@
+import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+import { Box, Text } from 'ink';
+export const FileAutocomplete = ({ files, query, selectedIndex, onSelect }) => {
+    // Filter files based on query
+    const filteredFiles = files
+        .filter(file => {
+        const lowerFile = file.toLowerCase();
+        const lowerQuery = query.toLowerCase();
+        return lowerFile.includes(lowerQuery);
+    })
+        .slice(0, 10); // Limit to 10 results
+    if (filteredFiles.length === 0) {
+        return (_jsx(Box, { borderStyle: "round", borderColor: "yellow", flexDirection: "column", paddingX: 1, children: _jsxs(Text, { dimColor: true, children: ["No files found matching \"", query, "\""] }) }));
+    }
+    return (_jsxs(Box, { borderStyle: "round", borderColor: "yellow", flexDirection: "column", paddingX: 1, children: [_jsx(Text, { bold: true, color: "yellow", children: "\u2191\u2193 navigate, Enter inserts path, Esc cancels" }), filteredFiles.map((file, index) => (_jsx(Box, { marginLeft: 1, children: _jsxs(Text, { color: index === selectedIndex ? 'green' : undefined, bold: index === selectedIndex, children: [index === selectedIndex ? '> ' : '  ', file] }) }, file)))] }));
+};

package/dist/ui/components/FileChip.js

@@ -0,0 +1,8 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { Box, Text } from 'ink';
+import * as path from 'path';
+export const FileChip = ({ filePath, onRemove }) => {
+    const basename = path.basename(filePath);
+    const dirname = path.dirname(filePath);
+    return (_jsxs(Box, { borderStyle: "round", borderColor: "cyan", paddingX: 1, marginRight: 1, children: [_jsx(Text, { color: "cyan", bold: true, children: basename }), _jsxs(Text, { dimColor: true, children: [" ", dirname !== '.' ? `(${dirname})` : ''] }), _jsx(Text, { color: "red", bold: true, children: " [x]" })] }));
+};

package/dist/utils/fileUtils.js

@@ -0,0 +1,67 @@
+import * as path from 'path';
+/**
+ * Filter files by query using simple substring matching
+ */
+export function fuzzyMatchFiles(query, files) {
+    const lowerQuery = query.toLowerCase();
+    return files.filter(file => file.toLowerCase().includes(lowerQuery));
+}
+/**
+ * Truncate file content to a maximum number of lines
+ */
+export function truncateFileContent(content, maxLines = 500) {
+    const lines = content.split('\n');
+    if (lines.length <= maxLines) {
+        return content;
+    }
+    const truncated = lines.slice(0, maxLines).join('\n');
+    return `${truncated}\n\n... (truncated ${lines.length - maxLines} lines)`;
+}
+/**
+ * Format file for LLM context with proper delimiters
+ */
+export function formatFileForContext(filePath, content) {
+    const ext = getFileExtension(filePath);
+    const language = getLanguageFromExtension(ext);
+    return `File: ${filePath}\n\`\`\`${language}\n${content}\n\`\`\`\n`;
+}
+/**
+ * Get file extension from path
+ */
+export function getFileExtension(filePath) {
+    return path.extname(filePath).slice(1);
+}
+/**
+ * Map file extension to language identifier for syntax highlighting
+ */
+function getLanguageFromExtension(ext) {
+    const languageMap = {
+        'ts': 'typescript',
+        'tsx': 'typescript',
+        'js': 'javascript',
+        'jsx': 'javascript',
+        'py': 'python',
+        'java': 'java',
+        'cpp': 'cpp',
+        'c': 'c',
+        'cs': 'csharp',
+        'go': 'go',
+        'rs': 'rust',
+        'rb': 'ruby',
+        'php': 'php',
+        'swift': 'swift',
+        'kt': 'kotlin',
+        'md': 'markdown',
+        'json': 'json',
+        'yaml': 'yaml',
+        'yml': 'yaml',
+        'xml': 'xml',
+        'html': 'html',
+        'css': 'css',
+        'scss': 'scss',
+        'sql': 'sql',
+        'sh': 'bash',
+        'bash': 'bash',
+    };
+    return languageMap[ext] || ext;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kishlay42/moth-ai",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Local, LLM-agnostic code intelligence CLI",
   "main": "dist/index.js",
   "bin": {