cloving 0.2.1 → 0.3.1

package/dist/managers/ChatManager.js

@@ -4,18 +4,34 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const copy_paste_1 = __importDefault(require("copy-paste"));
-const path_1 = __importDefault(require("path"));
-const fs_1 = __importDefault(require("fs"));
 const readline_1 = __importDefault(require("readline"));
 const child_process_1 = require("child_process");
+const colors_1 = __importDefault(require("colors"));
+const CommitManager_1 = __importDefault(require("./CommitManager"));
 const ChunkManager_1 = __importDefault(require("./ChunkManager"));
 const ReviewManager_1 = __importDefault(require("./ReviewManager"));
 const cloving_gpt_1 = __importDefault(require("../cloving_gpt"));
-const git_utils_1 = require("../utils/git_utils");
 const string_utils_1 = require("../utils/string_utils");
+const config_utils_1 = require("../utils/config_utils");
 const command_utils_1 = require("../utils/command_utils");
-const PREAMBLE = `When generating code, don't apologize and wherever possible include filenames in bold with paths to the code files mentioned and do not be lazy and ask me to keep the existing code or show things like previous code remains unchanged, always include existing code in the response.`;
+const specialCommands = [
+    'save',
+    'commit',
+    'copy',
+    'review',
+    'add <file-path>',
+    'find <file-name>',
+    'rm <pattern>',
+    'ls <pattern>',
+    'git <command>',
+    'help',
+    'exit'
+];
 class ChatManager {
+    /**
+     * Creates an instance of ChatManager.
+     * @param {ClovingGPTOptions} options - Configuration options for the ChatManager.
+     */
     constructor(options) {
         this.options = options;
         this.chatHistory = [];
@@ -32,78 +48,134 @@ class ChatManager {
         this.rl = readline_1.default.createInterface({
             input: process.stdin,
             output: process.stdout,
-            prompt: 'cloving> ',
+            prompt: colors_1.default.green.bold('cloving> '),
             historySize: 1000,
         });
         this.chunkManager = new ChunkManager_1.default();
     }
+    /**
+     * Initializes the ChatManager by loading context files, displaying a welcome message,
+     * and setting up event listeners.
+     * @returns {Promise<void>}
+     */
     async initialize() {
-        console.log('Welcome to Cloving REPL. Type "exit" to quit.');
+        await this.loadContextFiles();
+        console.log(`\n🍀 🍀 🍀 ${colors_1.default.bold('Welcome to Cloving REPL')} 🍀 🍀 🍀\n`);
+        this.displayAvailableCommands();
+        console.log('\nWhat would you like to do?');
         this.rl.prompt();
         this.setupEventListeners();
-        await this.loadContextFiles();
     }
+    displayAvailableCommands() {
+        console.log(`Type a freeform request or question to interact with your Cloving AI pair programmer.\n`);
+        console.log('Available special commands:');
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('save'))}             Save all the changes from the last response to files`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('commit'))}           Commit the changes to git with an AI-generated message that you can edit`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('copy'))}             Copy the last response to clipboard`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('review'))}           Start a code review`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('find <file-name>'))} Find and add files matching the name to the chat context (supports * for glob matching)`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('add <file-path>'))}  Add a file to the chat context (supports * for glob matching)`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('rm <pattern>'))}     Remove files from the chat context (supports * for glob matching)`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('ls <pattern>'))}     List files in the chat context (supports * for glob matching)`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('git <command>'))}    Run a git command`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('help'))}             Display this help message`);
+        console.log(` - ${colors_1.default.yellow(colors_1.default.bold('exit'))}             Quit this session`);
+    }
+    /**
+     * Sets up event listeners for the readline interface and process input.
+     * @private
+     */
     setupEventListeners() {
         this.rl.on('line', this.handleLine.bind(this));
         this.rl.on('close', this.handleClose.bind(this));
         process.stdin.on('keypress', this.handleKeypress.bind(this));
     }
+    /**
+     * Loads context files specified in the options or defaults to the primary language's directory.
+     * @private
+     * @returns {Promise<void>}
+     */
     async loadContextFiles() {
-        let files = this.options.files || '.';
-        let expandedFiles = [];
-        for (const file of files) {
-            const filePath = path_1.default.resolve(file);
-            if (await fs_1.default.promises.stat(filePath).then(stat => stat.isDirectory()).catch(() => false)) {
-                const dirFiles = await (0, command_utils_1.getAllFilesInDirectory)(filePath);
-                expandedFiles = expandedFiles.concat(dirFiles.map(f => path_1.default.relative(process.cwd(), f)));
-            }
-            else {
-                expandedFiles.push(path_1.default.relative(process.cwd(), filePath));
-            }
+        const config = (0, config_utils_1.getClovingConfig)();
+        const primaryLanguage = config.languages.find(lang => lang.primary);
+        const defaultDirectory = primaryLanguage ? primaryLanguage.directory : '.';
+        const testingDirectories = config.testingFrameworks?.map(framework => framework.directory) || [];
+        const testingDirectory = testingDirectories[0];
+        let files = this.options.files || [defaultDirectory, testingDirectory].filter(Boolean);
+        if (files.length > 0) {
+            console.log(`\nBuilding chat session context...\n`);
         }
-        files = expandedFiles;
         for (const file of files) {
-            const filePath = path_1.default.resolve(file);
-            if (await fs_1.default.promises.stat(filePath).then(stat => stat.isFile()).catch(() => false)) {
-                const content = await fs_1.default.promises.readFile(filePath, 'utf-8');
-                this.contextFiles[file] = content;
-            }
+            // Skip if the file is not a string
+            if (!file)
+                continue;
+            const previousCount = Object.keys(this.contextFiles).length;
+            this.contextFiles = await (0, command_utils_1.addFileOrDirectoryToContext)(file, this.contextFiles, this.options);
+            const newCount = Object.keys(this.contextFiles).length;
+            const addedCount = newCount - previousCount;
+            const totalTokens = this.calculateTotalTokens();
+            console.log(colors_1.default.cyan(`📁 Loaded context from: ${colors_1.default.bold(file)}`));
+            console.log(colors_1.default.green(`   ✅ Added ${addedCount} file(s) to context`));
+            console.log(colors_1.default.yellow(`   📊 Total tokens in context: ${totalTokens.toLocaleString()}\n`));
         }
     }
+    calculateTotalTokens() {
+        return Object.values(this.contextFiles).reduce((total, content) => {
+            return total + Math.ceil(content.length / 4);
+        }, 0);
+    }
+    /**
+     * Handles each line of input from the user.
+     * @private
+     * @param {string} line - The input line from the user.
+     * @returns {Promise<void>}
+     */
     async handleLine(line) {
         if (this.isProcessing) {
             return;
         }
+        const trimmedLine = line.trim();
+        if (this.handleMultilineInput(trimmedLine)) {
+            return;
+        }
+        if (trimmedLine === '') {
+            this.displayPrompt();
+            return;
+        }
+        this.updateCommandHistory(trimmedLine);
+        if (this.handleExitCommand(trimmedLine)) {
+            return;
+        }
+        await this.handleCommand(trimmedLine);
+    }
+    handleMultilineInput(line) {
         if (this.isMultilineMode) {
-            if (line.trim() === '```') {
+            if (line === '```') {
                 this.isMultilineMode = false;
-                line = this.multilineInput;
+                this.handleCommand(this.multilineInput);
                 this.multilineInput = '';
             }
             else {
                 this.multilineInput += line + '\n';
                 this.rl.prompt();
-                return;
             }
+            return true;
         }
-        else if (line.trim() === '```') {
+        else if (line === '```') {
             this.isMultilineMode = true;
             this.multilineInput = '';
             console.log('Entering multiline mode. Type ``` on a new line to end.\n');
             this.rl.prompt();
-            return;
-        }
-        const trimmedLine = line.trim();
-        if (trimmedLine === '') {
-            this.rl.prompt();
-            return;
+            return true;
         }
-        this.updateCommandHistory(trimmedLine);
-        if (trimmedLine.toLowerCase() === 'exit') {
+        return false;
+    }
+    handleExitCommand(command) {
+        if (command.toLowerCase() === 'exit') {
             this.rl.close();
-            return;
+            return true;
         }
-        await this.handleCommand(trimmedLine);
+        return false;
     }
     updateCommandHistory(command) {
         if (this.commandHistory[0] !== command) {
@@ -114,8 +186,33 @@ class ChatManager {
         }
         this.historyIndex = -1;
     }
+    /**
+     * Handles various commands entered by the user.
+     * This function acts as a command router, delegating to specific handlers based on the input.
+     *
+     * @private
+     * @param {string} command - The command entered by the user.
+     * @returns {Promise<void>}
+     *
+     * @description
+     * Supported commands:
+     * - 'copy': Copies the last response to clipboard.
+     * - 'save': Saves changes to files.
+     * - 'commit': Commits changes to git with an AI-generated message.
+     * - 'ls': Lists files in the context (equivalent to 'ls *').
+     * - 'rm': Removes all files from the context (equivalent to 'rm *').
+     * - 'review': Starts a code review.
+     * - 'add <file-path>': Adds a file to the context.
+     * - 'rm <pattern>': Removes files matching the pattern from the context.
+     * - 'ls <pattern>': Lists files in the context matching the pattern.
+     * - 'git <command>': Executes a git command.
+     * - Any other input is processed as a user query to the AI.
+     */
     async handleCommand(command) {
         switch (command) {
+            case 'help':
+                this.displayHelp();
+                break;
             case 'copy':
                 await this.handleCopy();
                 break;
@@ -125,11 +222,29 @@ class ChatManager {
             case 'commit':
                 await this.handleCommit();
                 break;
+            case 'ls':
+                this.handleList("ls *");
+                break;
+            case 'rm':
+                await this.handleRemove("rm *");
+                break;
             case 'review':
                 await this.handleReview();
                 break;
             default:
-                if (this.isGitCommand(command)) {
+                if (this.isAddCommand(command)) {
+                    await this.handleAdd(command);
+                }
+                else if (this.isFindCommand(command)) {
+                    await this.handleFind(command);
+                }
+                else if (this.isRemoveCommand(command)) {
+                    await this.handleRemove(command);
+                }
+                else if (this.isListCommand(command)) {
+                    this.handleList(command);
+                }
+                else if (this.isGitCommand(command)) {
                     this.executeGitCommand(command);
                 }
                 else {
@@ -137,11 +252,144 @@ class ChatManager {
                 }
         }
     }
+    isFindCommand(command) {
+        const parts = command.trim().split(/\s+/);
+        return parts.length === 2 && parts[0] === 'find';
+    }
+    async handleFind(command) {
+        const fileName = command.slice(5).trim();
+        if (fileName) {
+            try {
+                const foundFiles = await this.findFiles(fileName);
+                if (foundFiles.length === 0) {
+                    console.log(`No files found matching ${colors_1.default.bold(colors_1.default.red(fileName))}.`);
+                }
+                else {
+                    for (const filePath of foundFiles) {
+                        this.contextFiles = await (0, command_utils_1.addFileOrDirectoryToContext)(filePath, this.contextFiles, this.options);
+                        const content = this.contextFiles[filePath];
+                        const tokenEstimate = this.estimateTokens(content);
+                        console.log(`\nAdded ${colors_1.default.bold(colors_1.default.green(filePath))} to this chat session's context (${colors_1.default.yellow(`~${tokenEstimate.toLocaleString()} tokens`)})`);
+                    }
+                    const totalTokens = this.calculateTotalTokens();
+                    console.log(colors_1.default.yellow(`\n📊 Total tokens in context now: ${totalTokens.toLocaleString()}\n`));
+                }
+            }
+            catch (error) {
+                console.error(`Failed to find and add files matching ${colors_1.default.bold(colors_1.default.red(fileName))}:`, error);
+            }
+        }
+        else {
+            console.log('No file name provided.');
+        }
+        this.rl.prompt();
+    }
+    async findFiles(fileName) {
+        const { exec } = require('child_process');
+        return new Promise((resolve, reject) => {
+            exec(`find . -name "${fileName}"`, (error, stdout) => {
+                if (error) {
+                    return reject(error);
+                }
+                const files = stdout.split('\n').filter((filePath) => filePath.trim() !== '')
+                    .map((filePath) => filePath.startsWith('./') ? filePath.slice(2) : filePath);
+                resolve(files);
+            });
+        });
+    }
+    displayHelp() {
+        console.log('');
+        this.displayAvailableCommands();
+        console.log('\nFor any other input, I will process it as a request or question.\n');
+        this.rl.prompt();
+    }
+    isAddCommand(command) {
+        const parts = command.trim().split(/\s+/);
+        return parts.length === 2 && parts[0] === 'add';
+    }
+    async handleAdd(command) {
+        const filePath = command.slice(4).trim();
+        if (filePath) {
+            try {
+                this.contextFiles = await (0, command_utils_1.addFileOrDirectoryToContext)(filePath, this.contextFiles, this.options);
+                console.log(`\nAdded ${colors_1.default.bold(colors_1.default.green(filePath))} to this chat session's context`);
+            }
+            catch (error) {
+                console.error(`Failed to add ${colors_1.default.bold(colors_1.default.red(filePath))} to this chat session's context:`, error);
+            }
+        }
+        else {
+            console.log('No file path provided.');
+        }
+        this.rl.prompt();
+    }
+    isRemoveCommand(command) {
+        return command.startsWith('rm ');
+    }
+    // Function to convert glob-like pattern to a regular expression
+    globToRegExp(glob) {
+        const escapedGlob = glob.replace(/[-/\\^$+?.()|[\]{}]/g, '\\$&');
+        const regexString = escapedGlob.replace(/\*/g, '.*');
+        return new RegExp(`^${regexString}$`);
+    }
+    // Function to filter paths using the pattern
+    filterPaths(paths, pattern) {
+        const regex = this.globToRegExp(pattern);
+        return paths.filter(path => regex.test(path));
+    }
+    async handleRemove(command) {
+        const pattern = command.slice(3).trim();
+        if (pattern) {
+            const matchedFiles = this.filterPaths(Object.keys(this.contextFiles), pattern);
+            if (matchedFiles.length > 0) {
+                matchedFiles.forEach(filePath => {
+                    delete this.contextFiles[filePath];
+                    console.log(`Removed ${colors_1.default.bold(colors_1.default.green(filePath))} from this chat session's context files`);
+                });
+            }
+            else {
+                console.log(`No files matching pattern "${colors_1.default.bold(pattern)}" found in this chat session's context files, try running ${colors_1.default.yellow('ls')} to see the list of files.`);
+            }
+        }
+        else {
+            console.log('No pattern provided.');
+        }
+        this.rl.prompt();
+    }
+    isListCommand(command) {
+        return command.startsWith('ls ');
+    }
+    estimateTokens(text) {
+        // A simple estimation: 1 token ≈ 4 characters
+        return Math.ceil(text.length / 4);
+    }
+    handleList(command) {
+        const pattern = command.slice(3).trim();
+        const fileNames = Object.keys(this.contextFiles);
+        const matchedFiles = pattern ? this.filterPaths(fileNames, pattern) : fileNames;
+        if (matchedFiles.length > 0) {
+            console.log(`\nFiles in the current chat session context:`);
+            let totalTokens = 0;
+            matchedFiles.forEach(fileName => {
+                const content = this.contextFiles[fileName];
+                const tokenEstimate = this.estimateTokens(content);
+                totalTokens += tokenEstimate;
+                console.log(` - ${colors_1.default.bold(colors_1.default.green(fileName))} (${colors_1.default.yellow(`~${tokenEstimate.toLocaleString()} tokens`)})`);
+            });
+            console.log(`\nTotal files: ${matchedFiles.length}`);
+            console.log(`Total estimated tokens: ${colors_1.default.yellow(totalTokens.toLocaleString())}\n`);
+        }
+        else {
+            console.log('No files currently in context.');
+        }
+        this.rl.prompt();
+    }
     async handleCopy() {
         const lastResponse = this.chatHistory.filter(msg => msg.role === 'assistant').pop();
         if (lastResponse) {
             copy_paste_1.default.copy(lastResponse.content, () => {
                 console.info('Last response copied to clipboard.');
+                this.rl.prompt();
             });
         }
         else {
@@ -149,22 +397,30 @@ class ChatManager {
         }
     }
     async handleReview() {
-        this.gpt.stream = false;
-        const reviewManager = new ReviewManager_1.default(this.gpt, this.options);
+        this.options.stream = false;
+        const reviewManager = new ReviewManager_1.default(this.options);
         await reviewManager.review();
-        this.gpt.stream = true;
+        this.options.stream = true;
+        this.rl.prompt();
+    }
+    async handleCommit() {
+        this.options.stream = false;
+        const commitManager = new CommitManager_1.default(this.options);
+        await commitManager.commit();
+        this.options.stream = true;
+        this.rl.prompt();
     }
     async handleSave() {
-        const lastResponse = this.chatHistory.filter(msg => msg.role === 'assistant').pop();
+        const lastResponse = this.chatHistory.slice().reverse().find(msg => msg.role === 'assistant');
         if (lastResponse) {
-            const fileContents = (0, string_utils_1.extractFilesAndContent)(lastResponse.content);
-            if (Object.keys(fileContents).length > 0) {
-                await (0, string_utils_1.saveGeneratedFiles)(fileContents);
-                console.info('Files have been saved.');
+            const currentNewBlocks = (0, string_utils_1.extractCurrentNewBlocks)(lastResponse.content);
+            if (Object.keys(currentNewBlocks).length > 0) {
+                await (0, string_utils_1.applyAndSaveCurrentNewBlocks)(currentNewBlocks);
+                console.info(`\n${colors_1.default.bold('save')} has finished\n`);
                 this.rl.prompt();
             }
             else {
-                console.info('No files found to save in the last response.');
+                console.info('No changes found to save in the last response.');
                 this.rl.prompt();
             }
         }
@@ -173,32 +429,6 @@ class ChatManager {
             this.rl.prompt();
         }
     }
-    async handleCommit() {
-        const diff = (0, child_process_1.execSync)('git diff HEAD').toString().trim();
-        if (!diff) {
-            console.error('No changes to commit.');
-            this.rl.prompt();
-            return;
-        }
-        this.prompt = (0, git_utils_1.generateCommitMessagePrompt)(diff);
-        this.gpt.stream = false;
-        const rawCommitMessage = await this.gpt.generateText({ prompt: this.prompt });
-        this.gpt.stream = true;
-        const commitMessage = (0, string_utils_1.extractMarkdown)(rawCommitMessage);
-        const tempCommitFilePath = path_1.default.join('.git', 'SUGGESTED_COMMIT_EDITMSG');
-        fs_1.default.writeFileSync(tempCommitFilePath, commitMessage);
-        try {
-            (0, child_process_1.execFileSync)('git', ['commit', '-a', '--edit', '--file', tempCommitFilePath], { stdio: 'inherit' });
-        }
-        catch (commitError) {
-            console.error('Commit was canceled or failed.');
-        }
-        fs_1.default.unlink(tempCommitFilePath, (err) => {
-            if (err)
-                throw err;
-        });
-        this.rl.prompt();
-    }
     isGitCommand(command) {
         return command.split(' ').length <= 3 && command.startsWith('git ');
     }
@@ -211,6 +441,18 @@ class ChatManager {
         }
         this.rl.prompt();
     }
+    /**
+     * Processes user input by sending it to the AI model and handling the response.
+     *
+     * This method manages the interaction with the AI model by:
+     * 1. Checking if a request is already being processed.
+     * 2. Initializing the chat history and generating a prompt.
+     * 3. Streaming the response from the AI model.
+     * 4. Handling the response stream and updating the chat history.
+     *
+     * @param {string} input - The user's input or request to be processed.
+     * @returns {Promise<void>} - A promise that resolves when the processing is complete.
+     */
     async processUserInput(input) {
         if (this.isProcessing) {
             console.log('Please wait for the current request to complete.');
@@ -219,73 +461,122 @@ class ChatManager {
         this.isProcessing = true;
         this.chunkManager = new ChunkManager_1.default();
         try {
-            this.chatHistory.push({ role: 'user', content: input });
+            this.initializeChatHistory(input);
             this.prompt = this.generatePrompt(input);
-            const responseStream = await this.gpt.streamText({ prompt: this.prompt });
+            const responseStream = await this.gpt.streamText({ prompt: this.prompt, messages: this.chatHistory });
             let accumulatedContent = '';
-            this.chunkManager.on('content', (buffer) => {
-                let convertedStream = this.gpt.convertStream(buffer);
-                while (convertedStream !== null) {
-                    const { output, lastChar } = convertedStream;
-                    process.stdout.write(output);
-                    accumulatedContent += output;
-                    this.chunkManager.clearBuffer(lastChar);
-                    buffer = buffer.slice(lastChar);
-                    convertedStream = this.gpt.convertStream(buffer);
-                }
-            });
-            responseStream.data.on('data', (chunk) => {
-                const chunkString = chunk.toString();
-                this.chunkManager.addChunk(chunkString);
-            });
-            responseStream.data.on('end', () => {
-                this.chatHistory.push({ role: 'assistant', content: accumulatedContent.trim() });
-                this.isProcessing = false;
-                process.stdout.write('\n');
-                this.rl.prompt();
-            });
-            responseStream.data.on('error', (error) => {
-                console.error('Error streaming response:', error);
-                this.isProcessing = false;
-                process.stdout.write('\n');
-                this.rl.prompt();
-            });
+            this.handleResponseStream(responseStream, accumulatedContent);
         }
         catch (err) {
-            const error = err;
-            let errorMessage = error.message || 'An error occurred.';
-            const errorNumber = error.response?.status || 'unknown';
-            // switch case
-            switch (errorNumber) {
-                case 400:
-                    errorMessage = "Invalid model or prompt size too large. Try specifying fewer files.";
-                    break;
-                case 403:
-                    errorMessage = "Inactive subscription or usage limit reached";
-                    break;
-                case 429:
-                    errorMessage = "Rate limit error";
-                    break;
-                case 500:
-                    errorMessage = "Internal server error";
-                    break;
+            this.handleError(err);
+        }
+    }
+    initializeChatHistory(input) {
+        if (this.chatHistory.length === 0) {
+            const systemPrompt = (0, command_utils_1.generateCodegenPrompt)(this.contextFiles);
+            this.chatHistory.push({ role: 'user', content: systemPrompt });
+            this.chatHistory.push({ role: 'assistant', content: 'What would you like to do?' });
+        }
+        this.chatHistory.push({ role: 'user', content: input });
+    }
+    handleResponseStream(responseStream, accumulatedContent) {
+        this.chunkManager.on('content', (buffer) => {
+            let convertedStream = this.gpt.convertStream(buffer);
+            while (convertedStream !== null) {
+                const { output, lastChar } = convertedStream;
+                process.stdout.write(output);
+                accumulatedContent += output;
+                this.chunkManager.clearBuffer(lastChar);
+                buffer = buffer.slice(lastChar);
+                convertedStream = this.gpt.convertStream(buffer);
             }
-            // get token estimate for prompt
-            const promptTokens = Math.ceil(this.prompt.length / 4).toLocaleString();
-            console.error(`Error processing a ${promptTokens} token prompt:`, errorMessage, `(${errorNumber})\n`);
+        });
+        responseStream.data.on('data', (chunk) => {
+            const chunkString = chunk.toString();
+            this.chunkManager.addChunk(chunkString);
+        });
+        responseStream.data.on('end', () => {
+            this.finalizeResponse(accumulatedContent);
+        });
+        responseStream.data.on('error', (error) => {
+            console.error('Error streaming response:', error);
             this.isProcessing = false;
+            process.stdout.write('\n');
             this.rl.prompt();
+        });
+    }
+    finalizeResponse(accumulatedContent) {
+        this.chatHistory.push({ role: 'assistant', content: accumulatedContent.trim() });
+        this.isProcessing = false;
+        process.stdout.write(`
+
+  You can follow up with another request or:
+   - type ${colors_1.default.yellow(colors_1.default.bold('"save"'))} to save all the changes to files
+   - type ${colors_1.default.yellow(colors_1.default.bold('"commit"'))} to commit the changes to git with a AI-generated message
+   - type ${colors_1.default.yellow(colors_1.default.bold('"copy"'))} to copy the last response to clipboard
+   - type ${colors_1.default.yellow(colors_1.default.bold('"review"'))} to start a code review
+   - type ${colors_1.default.yellow(colors_1.default.bold('"add <file-path>"'))} to add a file to the chat context
+   - type ${colors_1.default.yellow(colors_1.default.bold('"rm <pattern>"'))} to remove files from the chat context
+   - type ${colors_1.default.yellow(colors_1.default.bold('"ls <pattern>"'))} to list files in the chat context
+   - type ${colors_1.default.yellow(colors_1.default.bold('"git <command>"'))} to run a git command
+   - type ${colors_1.default.yellow(colors_1.default.bold('"exit"'))} to quit this session
+  `);
+        this.rl.prompt();
+    }
+    handleError(err) {
+        const error = err;
+        let errorMessage = error.message || 'An error occurred.';
+        const errorNumber = error.response?.status || 'unknown';
+        switch (errorNumber) {
+            case 400:
+                errorMessage = "Invalid model or prompt size too large. Try specifying fewer files.";
+                break;
+            case 403:
+                errorMessage = "Inactive subscription or usage limit reached";
+                break;
+            case 429:
+                errorMessage = "Rate limit error";
+                break;
+            case 500:
+                errorMessage = "Internal server error";
+                break;
         }
+        const promptTokens = Math.ceil(this.prompt.length / 4).toLocaleString();
+        console.error(`Error processing a ${promptTokens} token prompt:`, errorMessage, `(${errorNumber})\n`);
+        this.isProcessing = false;
+        this.rl.prompt();
     }
-    generatePrompt(input) {
+    /**
+     * Generates a comprehensive prompt for the AI model.
+     *
+     * This function constructs a detailed prompt that includes:
+     * 1. The user's current request
+     * 2. Contents of all context files
+     * 3. Full chat history (excluding the current request)
+     * 4. The current request repeated at the end
+     *
+     * @private
+     * @param {string} prompt - The user's current request or input.
+     * @returns {string} A formatted string containing the complete prompt for the AI.
+     *
+     * @description
+     * The generated prompt follows this structure:
+     * 1. "### Request" section with the current prompt
+     * 2. Contents of all context files, each prefixed with its file name
+     * 3. "### Full Chat History Context" section with all previous messages
+     * 4. "### Current Request" section repeating the current prompt
+     *
+     * This structure provides the AI with comprehensive context for generating
+     * accurate and relevant responses.
+     */
+    generatePrompt(prompt) {
         const contextFileContents = Object.keys(this.contextFiles)
             .map((file) => `### Contents of ${file}\n\n${this.contextFiles[file]}\n\n`)
             .join('\n');
         const allButLast = this.chatHistory.slice(0, -1).map(msg => `${msg.role.toUpperCase()}: ${msg.content}`).join('\n\n');
-        const lastUserMessage = this.chatHistory.slice(-1).map(msg => msg.content).join('\n\n');
         return `### Request
 
-${lastUserMessage}
+${prompt}
 
 ${contextFileContents}
 
@@ -295,14 +586,25 @@ ${allButLast}
 
 ### Current Request
 
-${PREAMBLE}
+${prompt}
 
-${lastUserMessage}`;
+### Note
+
+Whenever possible, break up the changes into pieces and make sure every change is in its own CURRENT/NEW block.`;
     }
     handleClose() {
         console.log('Goodbye!');
         process.exit(0);
     }
+    /**
+     * Displays the prompt in green and bold.
+     * This method should be used instead of directly calling this.rl.prompt().
+     * @private
+     */
+    displayPrompt() {
+        this.rl.setPrompt(colors_1.default.green.bold('cloving> '));
+        this.rl.prompt();
+    }
     handleKeypress(_, key) {
         if (key && (key.name === 'up' || key.name === 'down')) {
             if (key.name === 'up' && this.historyIndex < this.commandHistory.length - 1) {
@@ -319,6 +621,14 @@ ${lastUserMessage}`;
                 this.rl.write(null, { ctrl: true, name: 'u' });
             }
         }
+        else if (key && key.name === 'tab') {
+            const line = this.rl.line.trim();
+            const hits = specialCommands.filter((command) => command.startsWith(line));
+            if (hits.length > 0) {
+                this.rl.write(null, { ctrl: true, name: 'u' });
+                this.rl.write(hits[0]);
+            }
+        }
     }
 }
 exports.default = ChatManager;