@indiccoder/mentis-cli 1.0.4 → 1.0.8

package/dist/index.js

@@ -3,6 +3,12 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const ReplManager_1 = require("./repl/ReplManager");
 async function main() {
+    if (process.argv.includes('update')) {
+        const { UpdateManager } = require('./utils/UpdateManager');
+        const updater = new UpdateManager();
+        await updater.checkAndPerformUpdate(true);
+        return;
+    }
     const repl = new ReplManager_1.ReplManager();
     await repl.start();
 }

package/dist/repl/ReplManager.js

@@ -123,26 +123,25 @@ class ReplManager {
         // console.log(chalk.dim(`Initialized ${provider} client with model ${model}`));
     }
     async start() {
-        UIManager_1.UIManager.displayLogo();
-        UIManager_1.UIManager.displayWelcome();
+        UIManager_1.UIManager.renderDashboard({
+            model: this.currentModelName,
+            mode: this.mode,
+            cwd: process.cwd()
+        });
         // Load History
         let commandHistory = [];
         if (fs.existsSync(HISTORY_FILE)) {
             try {
-                commandHistory = fs.readFileSync(HISTORY_FILE, 'utf-8').split('\n').filter(Boolean).reverse(); // readline expects newest first? No, newest is usually 0? Check.
-                // readline.history is [newest, ..., oldest]
-                // If I read from file where newest is at bottom (standard append), I need to reverse it.
-                // Let's assume standard file: line 1 (old), line 2 (new).
-                // So split -> reverse -> history.
+                commandHistory = fs.readFileSync(HISTORY_FILE, 'utf-8').split('\n').filter(Boolean).reverse();
             }
             catch (e) { }
         }
         while (true) {
-            UIManager_1.UIManager.printSeparator();
-            // console.log(chalk.dim(`  /help for help | Model: ${chalk.cyan(this.currentModelName)}`));
-            // Removed redundancy to keep CLI clean, prompt has info? No, prompt is minimal.
-            const modeLabel = this.mode === 'PLAN' ? chalk_1.default.magenta('PLAN') : chalk_1.default.blue('BUILD');
-            const promptText = `${modeLabel} ${chalk_1.default.cyan('>')}`;
+            // Minimalist Separator
+            console.log(chalk_1.default.gray('────────────────────────────────────────────────────────────────────────────────'));
+            // Hint (Claude style puts it below, we put it above for standard terminal compatibility)
+            console.log(chalk_1.default.dim('  ? for shortcuts'));
+            const promptText = `> `; // Clean prompt
             // Use readline for basic input to support history
             const answer = await new Promise((resolve) => {
                 const rl = readline.createInterface({
@@ -150,7 +149,7 @@ class ReplManager {
                     output: process.stdout,
                     history: commandHistory,
                     historySize: 1000,
-                    prompt: promptText + ' '
+                    prompt: promptText
                 });
                 rl.prompt();
                 rl.on('line', (line) => {
@@ -192,6 +191,7 @@ class ReplManager {
                 console.log('  /help    - Show this help message');
                 console.log('  /clear   - Clear chat history');
                 console.log('  /exit    - Exit the application');
+                console.log('  /update  - Check for and install updates');
                 console.log('  /config  - Configure settings');
                 console.log('  /add <file> - Add file to context');
                 console.log('  /drop <file> - Remove file from context');
@@ -273,6 +273,11 @@ class ReplManager {
                 console.log(chalk_1.default.green('Session saved. Goodbye!'));
                 process.exit(0);
                 break;
+            case '/update':
+                const UpdateManager = require('../utils/UpdateManager').UpdateManager;
+                const updater = new UpdateManager();
+                await updater.checkAndPerformUpdate(true);
+                break;
             default:
                 console.log(chalk_1.default.red(`Unknown command: ${command}`));
         }
@@ -518,43 +523,53 @@ class ReplManager {
     }
     async handleModelCommand(args) {
         const config = this.configManager.getConfig();
-        const provider = config.defaultProvider || 'ollama';
-        // If argument provided, use it directly
+        const currentProvider = config.defaultProvider || 'ollama';
+        // Direct argument: /model gpt-4o (updates active provider's model)
         if (args.length > 0) {
             const modelName = args[0];
             const updates = {};
-            updates[provider] = { ...(config[provider] || {}), model: modelName };
+            updates[currentProvider] = { ...(config[currentProvider] || {}), model: modelName };
             this.configManager.updateConfig(updates);
             this.initializeClient(); // Re-init with new model
-            console.log(chalk_1.default.green(`\nModel set to ${chalk_1.default.bold(modelName)} for ${provider}!`));
+            console.log(chalk_1.default.green(`\nModel set to ${chalk_1.default.bold(modelName)} for ${currentProvider}!`));
             return;
         }
+        // Interactive Mode: Streamlined Provider -> Model Flow
+        console.log(chalk_1.default.cyan('Configure Model & Provider'));
+        const { provider } = await inquirer_1.default.prompt([
+            {
+                type: 'list',
+                name: 'provider',
+                message: 'Select Provider:',
+                choices: ['Gemini', 'Ollama', 'OpenAI', 'GLM'],
+                default: currentProvider.charAt(0).toUpperCase() + currentProvider.slice(1) // Capitalize for default selection
+            }
+        ]);
+        const selectedProvider = provider.toLowerCase();
         let models = [];
-        if (provider === 'gemini') {
+        if (selectedProvider === 'gemini') {
             models = ['gemini-2.5-flash', 'gemini-1.5-pro', 'gemini-1.0-pro', 'Other...'];
         }
-        else if (provider === 'ollama') {
-            models = ['llama3:latest', 'deepseek-r1:latest', 'mistral:latest', 'Other...'];
+        else if (selectedProvider === 'ollama') {
+            models = ['llama3:latest', 'deepseek-r1:latest', 'mistral:latest', 'qwen2.5-coder', 'Other...'];
         }
-        else if (provider === 'openai') {
+        else if (selectedProvider === 'openai') {
             models = ['gpt-4o', 'gpt-4o-mini', 'gpt-4-turbo', 'Other...'];
         }
-        else if (provider === 'glm') {
+        else if (selectedProvider === 'glm') {
             models = ['glm-4.6', 'glm-4-plus', 'glm-4', 'glm-4-air', 'glm-4-flash', 'Other...'];
         }
-        else if (provider === 'anthropic') {
-            models = ['claude-3-5-sonnet-20241022', 'claude-3-opus-20240229', 'claude-3-sonnet-20240229', 'claude-3-haiku-20240307', 'glm-4.6', 'Other...'];
-        }
         else {
             models = ['Other...'];
         }
-        console.log(chalk_1.default.blue(`Configuring model for active provider: ${chalk_1.default.bold(provider)}`));
         let { model } = await inquirer_1.default.prompt([
             {
                 type: 'list',
                 name: 'model',
-                message: 'Select Model:',
+                message: `Select Model for ${provider}:`,
                 choices: models,
+                // Try to find current model in list to set default
+                default: config[selectedProvider]?.model
             }
         ]);
         if (model === 'Other...') {
@@ -565,11 +580,33 @@ class ReplManager {
                 }]);
             model = customModel;
         }
+        // Check for missing API Key (except for Ollama)
+        let newApiKey = undefined;
+        const currentKey = config[selectedProvider]?.apiKey;
+        if (selectedProvider !== 'ollama' && !currentKey) {
+            console.log(chalk_1.default.yellow(`\n⚠️ No API Key found for ${provider}.`));
+            const { apiKey } = await inquirer_1.default.prompt([{
+                    type: 'password',
+                    name: 'apiKey',
+                    message: `Enter API Key for ${provider} (or leave empty to skip):`,
+                    mask: '*'
+                }]);
+            if (apiKey && apiKey.trim()) {
+                newApiKey = apiKey.trim();
+            }
+        }
         const updates = {};
-        updates[provider] = { ...(config[provider] || {}), model: model };
+        updates.defaultProvider = selectedProvider;
+        updates[selectedProvider] = {
+            ...(config[selectedProvider] || {}),
+            model: model
+        };
+        if (newApiKey) {
+            updates[selectedProvider].apiKey = newApiKey;
+        }
         this.configManager.updateConfig(updates);
         this.initializeClient();
-        console.log(chalk_1.default.green(`\nModel set to ${model} for ${provider}!`));
+        console.log(chalk_1.default.green(`\nSwitched to ${chalk_1.default.bold(provider)} (${model})!`));
     }
     async handleConnectCommand(args) {
         if (args.length < 1) {

package/dist/ui/UIManager.js

@@ -12,26 +12,64 @@ class UIManager {
     static displayLogo() {
         console.clear();
         const logoText = figlet_1.default.textSync('MENTIS', {
-            font: 'ANSI Shadow', // Use a block-like font
+            font: 'ANSI Shadow',
             horizontalLayout: 'default',
             verticalLayout: 'default',
             width: 100,
             whitespaceBreak: true,
         });
         console.log(gradient_string_1.default.pastel.multiline(logoText));
-        console.log(chalk_1.default.gray('  v1.0.0 - AI Coding Agent'));
+        console.log(chalk_1.default.gray('  v1.0.5 - AI Coding Agent'));
         console.log('');
     }
-    static displayWelcome() {
-        console.log((0, boxen_1.default)(`${chalk_1.default.bold('Welcome to Mentis-CLI')}\n\n` +
-            `• Type ${chalk_1.default.cyan('/help')} for commands.\n` +
-            `• Type ${chalk_1.default.cyan('/config')} to setup your model.\n` +
-            `• Start typing to chat with your agent.`, {
+    static renderDashboard(config) {
+        const { model, cwd } = config;
+        const version = 'v1.0.8';
+        // Layout: Left (Status/Welcome) | Right (Tips/Activity)
+        // Total width ~80 chars. 
+        // Left ~45, Right ~30.
+        const pad = (str, width) => str + ' '.repeat(Math.max(0, width - str.length));
+        const logo = gradient_string_1.default.pastel.multiline(figlet_1.default.textSync('MENTIS', { font: 'Small' }));
+        const logoLines = logo.split('\n');
+        // Tips Column
+        const tips = [
+            chalk_1.default.bold('Tips for getting started'),
+            chalk_1.default.dim('Run /init to scaffold project'),
+            chalk_1.default.dim('Run /model to switch AI'),
+            chalk_1.default.dim('Run /help for full list')
+        ];
+        // Combine Logo (Left) and Tips (Right)
+        let body = '';
+        for (let i = 0; i < Math.max(logoLines.length, tips.length); i++) {
+            const left = logoLines[i] || ''; // Logo line
+            const right = tips[i] || ''; // Tip line
+            // Need to strip ansi to calc padding? simple padding might break with ansi.
+            // Let's just create two distinct blocks and join them?
+            // Complex with boxen.
+            // Let's stick to vertical stack if side-by-side matches ansi poorly.
+            // Actually, let's just use the previous cleaner vertical stack but wider.
+            // User liked the previous one "this is exellent", just wanted input box.
+            // So I will keep the Dashboard mostly same, maybe just widen it.
+        }
+        // Re-using the clean layout but ensuring no "undefined" or weird overlaps
+        const title = ` Mentis-CLI ${version} `;
+        const content = ` ${chalk_1.default.bold('Welcome back!')}\n\n` +
+            `${logo}\n\n` +
+            ` ${chalk_1.default.dim('Model:')} ${chalk_1.default.cyan(model)}\n` +
+            ` ${chalk_1.default.dim('Dir:')}   ${chalk_1.default.dim(cwd)}\n\n` +
+            `${chalk_1.default.gray('────────────────────────────────────────────────────────────────')}\n` +
+            ` ${chalk_1.default.dim('Tips: /help • /config • /mcp • Esc to cancel')}`;
+        console.log((0, boxen_1.default)(content, {
             padding: 1,
-            margin: 1,
+            margin: 0,
             borderStyle: 'round',
             borderColor: 'cyan',
+            title: title,
+            titleAlignment: 'left',
+            dimBorder: true,
+            width: 80
         }));
+        console.log('');
     }
     static printSeparator() {
         console.log(chalk_1.default.gray('──────────────────────────────────────────────────'));

package/dist/utils/UpdateManager.js

@@ -0,0 +1,81 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpdateManager = void 0;
+const child_process_1 = require("child_process");
+const util_1 = require("util");
+const path_1 = __importDefault(require("path"));
+const fs_1 = __importDefault(require("fs"));
+const chalk_1 = __importDefault(require("chalk"));
+const ora_1 = __importDefault(require("ora"));
+const inquirer_1 = __importDefault(require("inquirer"));
+const execAsync = (0, util_1.promisify)(child_process_1.exec);
+class UpdateManager {
+    constructor() {
+        const packageJsonPath = path_1.default.join(__dirname, '../../package.json');
+        try {
+            const packageJson = JSON.parse(fs_1.default.readFileSync(packageJsonPath, 'utf-8'));
+            this.packageName = packageJson.name;
+            this.currentVersion = packageJson.version;
+        }
+        catch (e) {
+            // Fallback if running from a context where package.json isn't found easily (e.g. global install oddities)
+            // But usually this works relative to dist/utils/
+            this.packageName = '@indiccoder/mentis-cli';
+            this.currentVersion = '0.0.0';
+        }
+    }
+    async checkAndPerformUpdate(interactive = true) {
+        const spinner = (0, ora_1.default)('Checking for updates...').start();
+        try {
+            // Check latest version from NPM registry
+            const { stdout } = await execAsync(`npm view ${this.packageName} version`);
+            const latestVersion = stdout.trim();
+            if (latestVersion === this.currentVersion) {
+                spinner.succeed(chalk_1.default.green(`You are on the latest version (${this.currentVersion}).`));
+                return;
+            }
+            spinner.info(chalk_1.default.blue(`Update available: ${this.currentVersion} -> ${chalk_1.default.bold(latestVersion)}`));
+            if (!interactive) {
+                // If running in non-interactive mode (e.g. auto-check prompt), maybe just log it.
+                // But for explicit 'update' command, we usually assume interactive or force.
+                console.log(chalk_1.default.yellow(`Run 'mentis update' or '/update' inside the tool to upgrade.`));
+                return;
+            }
+            const { confirm } = await inquirer_1.default.prompt([{
+                    type: 'confirm',
+                    name: 'confirm',
+                    message: `Do you want to install v${latestVersion} now?`,
+                    default: true
+                }]);
+            if (confirm) {
+                await this.installUpdate(latestVersion);
+            }
+            else {
+                console.log(chalk_1.default.yellow('Update skipped.'));
+            }
+        }
+        catch (error) {
+            spinner.fail(chalk_1.default.red('Failed to check for updates.'));
+            if (process.env.DEBUG)
+                console.error(error);
+        }
+    }
+    async installUpdate(version) {
+        const spinner = (0, ora_1.default)(`Installing ${this.packageName}@${version}...`).start();
+        try {
+            await execAsync(`npm install -g ${this.packageName}@latest`);
+            spinner.succeed(chalk_1.default.green('Update completed successfully!'));
+            console.log(chalk_1.default.cyan('Please restart Mentis to use the new version.'));
+            process.exit(0);
+        }
+        catch (error) {
+            spinner.fail(chalk_1.default.red('Update failed.'));
+            console.error(chalk_1.default.red('Error details:'), error.message);
+            console.log(chalk_1.default.yellow(`Try running: npm install -g ${this.packageName}@latest`));
+        }
+    }
+}
+exports.UpdateManager = UpdateManager;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@indiccoder/mentis-cli",
-  "version": "1.0.4",
+  "version": "1.0.8",
   "publishConfig": {
     "access": "public"
   },

package/scripts/test_config_update.ts

@@ -0,0 +1,63 @@
+import { ConfigManager } from '../src/config/ConfigManager';
+import chalk from 'chalk';
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+
+async function testConfig() {
+    console.log(chalk.cyan('🧪 Testing ConfigManager...'));
+
+    const configManager = new ConfigManager();
+    const initialConfig = configManager.getConfig();
+    console.log('Initial Config:', initialConfig);
+
+    // 1. Test updating active provider
+    console.log(chalk.yellow('\n1. Setting active provider to "gemini"'));
+    configManager.updateConfig({ defaultProvider: 'gemini' });
+
+    let currentConfig = configManager.getConfig();
+    if (currentConfig.defaultProvider === 'gemini') {
+        console.log(chalk.green('✅ Provider set successfully.'));
+    } else {
+        console.log(chalk.red(`❌ Failed to set provider. Got: ${currentConfig.defaultProvider}`));
+    }
+
+    // 2. Test updating model for a provider
+    console.log(chalk.yellow('\n2. Updating Model for "gemini"'));
+    const newModel = 'gemini-1.5-pro-test';
+
+    const updates: any = {};
+    updates['gemini'] = { ...(currentConfig.gemini || {}), model: newModel };
+    configManager.updateConfig(updates);
+
+    currentConfig = configManager.getConfig();
+    if (currentConfig.gemini?.model === newModel) {
+        console.log(chalk.green(`✅ Model updated successfully to ${newModel}`));
+    } else {
+        console.log(chalk.red(`❌ Failed to update model. Got: ${currentConfig.gemini?.model}`));
+    }
+
+    // 3. Test API Key update
+    console.log(chalk.yellow('\n3. Updating API Key for "gemini"'));
+    const newKey = 'test-key-123';
+
+    updates['gemini'] = { ...(currentConfig.gemini || {}), apiKey: newKey };
+    configManager.updateConfig(updates); // Should merge with model update from previous step effectively if we pull fresh?
+    // Actually our test logic pulled currentConfig above, so it preserves 'model'.
+
+    currentConfig = configManager.getConfig();
+    if (currentConfig.gemini?.apiKey === newKey && currentConfig.gemini?.model === newModel) {
+        console.log(chalk.green('✅ API Key updated successfully (and model preserved).'));
+    } else {
+        console.log(chalk.red('❌ Failed to update API Key or verify persistence.'));
+        console.log(currentConfig.gemini);
+    }
+
+    // Restore original config to be nice
+    console.log(chalk.yellow('\nRestoring original config (defaultProvider)...'));
+    configManager.updateConfig({ defaultProvider: initialConfig.defaultProvider });
+
+    console.log(chalk.cyan('\n🏁 Config Tests Completed.'));
+}
+
+testConfig();

package/src/index.ts

@@ -2,6 +2,13 @@
 import { ReplManager } from './repl/ReplManager';
 
 async function main() {
+    if (process.argv.includes('update')) {
+        const { UpdateManager } = require('./utils/UpdateManager');
+        const updater = new UpdateManager();
+        await updater.checkAndPerformUpdate(true);
+        return;
+    }
+
     const repl = new ReplManager();
     await repl.start();
 }

package/src/repl/ReplManager.ts

@@ -99,28 +99,28 @@ export class ReplManager {
     }
 
     public async start() {
-        UIManager.displayLogo();
-        UIManager.displayWelcome();
+        UIManager.renderDashboard({
+            model: this.currentModelName,
+            mode: this.mode,
+            cwd: process.cwd()
+        });
 
         // Load History
         let commandHistory: string[] = [];
         if (fs.existsSync(HISTORY_FILE)) {
             try {
-                commandHistory = fs.readFileSync(HISTORY_FILE, 'utf-8').split('\n').filter(Boolean).reverse(); // readline expects newest first? No, newest is usually 0? Check.
-                // readline.history is [newest, ..., oldest]
-                // If I read from file where newest is at bottom (standard append), I need to reverse it.
-                // Let's assume standard file: line 1 (old), line 2 (new).
-                // So split -> reverse -> history.
+                commandHistory = fs.readFileSync(HISTORY_FILE, 'utf-8').split('\n').filter(Boolean).reverse();
             } catch (e) { }
         }
 
         while (true) {
-            UIManager.printSeparator();
-            // console.log(chalk.dim(`  /help for help | Model: ${chalk.cyan(this.currentModelName)}`));
-            // Removed redundancy to keep CLI clean, prompt has info? No, prompt is minimal.
+            // Minimalist Separator
+            console.log(chalk.gray('────────────────────────────────────────────────────────────────────────────────'));
+
+            // Hint (Claude style puts it below, we put it above for standard terminal compatibility)
+            console.log(chalk.dim('  ? for shortcuts'));
 
-            const modeLabel = this.mode === 'PLAN' ? chalk.magenta('PLAN') : chalk.blue('BUILD');
-            const promptText = `${modeLabel} ${chalk.cyan('>')}`;
+            const promptText = `> `;  // Clean prompt
 
             // Use readline for basic input to support history
             const answer = await new Promise<string>((resolve) => {
@@ -129,7 +129,7 @@ export class ReplManager {
                     output: process.stdout,
                     history: commandHistory,
                     historySize: 1000,
-                    prompt: promptText + ' '
+                    prompt: promptText
                 });
 
                 rl.prompt();
@@ -178,6 +178,7 @@ export class ReplManager {
                 console.log('  /help    - Show this help message');
                 console.log('  /clear   - Clear chat history');
                 console.log('  /exit    - Exit the application');
+                console.log('  /update  - Check for and install updates');
                 console.log('  /config  - Configure settings');
                 console.log('  /add <file> - Add file to context');
                 console.log('  /drop <file> - Remove file from context');
@@ -257,6 +258,11 @@ export class ReplManager {
                 console.log(chalk.green('Session saved. Goodbye!'));
                 process.exit(0);
                 break;
+            case '/update':
+                const UpdateManager = require('../utils/UpdateManager').UpdateManager;
+                const updater = new UpdateManager();
+                await updater.checkAndPerformUpdate(true);
+                break;
             default:
                 console.log(chalk.red(`Unknown command: ${command}`));
         }
@@ -533,42 +539,55 @@ export class ReplManager {
 
     private async handleModelCommand(args: string[]) {
         const config = this.configManager.getConfig();
-        const provider = config.defaultProvider || 'ollama';
+        const currentProvider = config.defaultProvider || 'ollama';
 
-        // If argument provided, use it directly
+        // Direct argument: /model gpt-4o (updates active provider's model)
         if (args.length > 0) {
             const modelName = args[0];
             const updates: any = {};
-            updates[provider] = { ...((config as any)[provider] || {}), model: modelName };
+            updates[currentProvider] = { ...((config as any)[currentProvider] || {}), model: modelName };
             this.configManager.updateConfig(updates);
             this.initializeClient(); // Re-init with new model
-            console.log(chalk.green(`\nModel set to ${chalk.bold(modelName)} for ${provider}!`));
+            console.log(chalk.green(`\nModel set to ${chalk.bold(modelName)} for ${currentProvider}!`));
             return;
         }
 
+        // Interactive Mode: Streamlined Provider -> Model Flow
+        console.log(chalk.cyan('Configure Model & Provider'));
+
+        const { provider } = await inquirer.prompt([
+            {
+                type: 'list',
+                name: 'provider',
+                message: 'Select Provider:',
+                choices: ['Gemini', 'Ollama', 'OpenAI', 'GLM'],
+                default: currentProvider.charAt(0).toUpperCase() + currentProvider.slice(1) // Capitalize for default selection
+            }
+        ]);
+
+        const selectedProvider = provider.toLowerCase();
+
         let models: string[] = [];
-        if (provider === 'gemini') {
+        if (selectedProvider === 'gemini') {
             models = ['gemini-2.5-flash', 'gemini-1.5-pro', 'gemini-1.0-pro', 'Other...'];
-        } else if (provider === 'ollama') {
-            models = ['llama3:latest', 'deepseek-r1:latest', 'mistral:latest', 'Other...'];
-        } else if (provider === 'openai') {
+        } else if (selectedProvider === 'ollama') {
+            models = ['llama3:latest', 'deepseek-r1:latest', 'mistral:latest', 'qwen2.5-coder', 'Other...'];
+        } else if (selectedProvider === 'openai') {
             models = ['gpt-4o', 'gpt-4o-mini', 'gpt-4-turbo', 'Other...'];
-        } else if (provider === 'glm') {
+        } else if (selectedProvider === 'glm') {
             models = ['glm-4.6', 'glm-4-plus', 'glm-4', 'glm-4-air', 'glm-4-flash', 'Other...'];
-        } else if (provider === 'anthropic') {
-            models = ['claude-3-5-sonnet-20241022', 'claude-3-opus-20240229', 'claude-3-sonnet-20240229', 'claude-3-haiku-20240307', 'glm-4.6', 'Other...'];
         } else {
             models = ['Other...'];
         }
 
-        console.log(chalk.blue(`Configuring model for active provider: ${chalk.bold(provider)}`));
-
         let { model } = await inquirer.prompt([
             {
                 type: 'list',
                 name: 'model',
-                message: 'Select Model:',
+                message: `Select Model for ${provider}:`,
                 choices: models,
+                // Try to find current model in list to set default
+                default: (config as any)[selectedProvider]?.model
             }
         ]);
 
@@ -581,12 +600,37 @@ export class ReplManager {
             model = customModel;
         }
 
+        // Check for missing API Key (except for Ollama)
+        let newApiKey = undefined;
+        const currentKey = (config as any)[selectedProvider]?.apiKey;
+
+        if (selectedProvider !== 'ollama' && !currentKey) {
+            console.log(chalk.yellow(`\n⚠️ No API Key found for ${provider}.`));
+            const { apiKey } = await inquirer.prompt([{
+                type: 'password',
+                name: 'apiKey',
+                message: `Enter API Key for ${provider} (or leave empty to skip):`,
+                mask: '*'
+            }]);
+            if (apiKey && apiKey.trim()) {
+                newApiKey = apiKey.trim();
+            }
+        }
+
         const updates: any = {};
-        updates[provider] = { ...((config as any)[provider] || {}), model: model };
+        updates.defaultProvider = selectedProvider;
+        updates[selectedProvider] = {
+            ...((config as any)[selectedProvider] || {}),
+            model: model
+        };
+
+        if (newApiKey) {
+            updates[selectedProvider].apiKey = newApiKey;
+        }
 
         this.configManager.updateConfig(updates);
         this.initializeClient();
-        console.log(chalk.green(`\nModel set to ${model} for ${provider}!`));
+        console.log(chalk.green(`\nSwitched to ${chalk.bold(provider)} (${model})!`));
     }
 
     private async handleConnectCommand(args: string[]) {

package/src/ui/UIManager.ts

@@ -7,34 +7,80 @@ export class UIManager {
     public static displayLogo() {
         console.clear();
         const logoText = figlet.textSync('MENTIS', {
-            font: 'ANSI Shadow', // Use a block-like font
+            font: 'ANSI Shadow',
             horizontalLayout: 'default',
             verticalLayout: 'default',
             width: 100,
             whitespaceBreak: true,
         });
         console.log(gradient.pastel.multiline(logoText));
-        console.log(chalk.gray('  v1.0.0 - AI Coding Agent'));
+        console.log(chalk.gray('  v1.0.5 - AI Coding Agent'));
         console.log('');
     }
 
-    public static displayWelcome() {
+    public static renderDashboard(config: { model: string, mode: string, cwd: string }) {
+        const { model, cwd } = config;
+        const version = 'v1.0.8';
+
+        // Layout: Left (Status/Welcome) | Right (Tips/Activity)
+        // Total width ~80 chars. 
+        // Left ~45, Right ~30.
+
+        const pad = (str: string, width: number) => str + ' '.repeat(Math.max(0, width - str.length));
+
+        const logo = gradient.pastel.multiline(figlet.textSync('MENTIS', { font: 'Small' }));
+        const logoLines = logo.split('\n');
+
+        // Tips Column
+        const tips = [
+            chalk.bold('Tips for getting started'),
+            chalk.dim('Run /init to scaffold project'),
+            chalk.dim('Run /model to switch AI'),
+            chalk.dim('Run /help for full list')
+        ];
+
+        // Combine Logo (Left) and Tips (Right)
+        let body = '';
+        for (let i = 0; i < Math.max(logoLines.length, tips.length); i++) {
+            const left = logoLines[i] || ''; // Logo line
+            const right = tips[i] || '';     // Tip line
+            // Need to strip ansi to calc padding? simple padding might break with ansi.
+            // Let's just create two distinct blocks and join them?
+            // Complex with boxen.
+            // Let's stick to vertical stack if side-by-side matches ansi poorly.
+            // Actually, let's just use the previous cleaner vertical stack but wider.
+            // User liked the previous one "this is exellent", just wanted input box.
+            // So I will keep the Dashboard mostly same, maybe just widen it.
+        }
+
+        // Re-using the clean layout but ensuring no "undefined" or weird overlaps
+        const title = ` Mentis-CLI ${version} `;
+
+        const content =
+            ` ${chalk.bold('Welcome back!')}\n\n` +
+            `${logo}\n\n` +
+            ` ${chalk.dim('Model:')} ${chalk.cyan(model)}\n` +
+            ` ${chalk.dim('Dir:')}   ${chalk.dim(cwd)}\n\n` +
+            `${chalk.gray('────────────────────────────────────────────────────────────────')}\n` +
+            ` ${chalk.dim('Tips: /help • /config • /mcp • Esc to cancel')}`;
+
         console.log(
-            boxen(
-                `${chalk.bold('Welcome to Mentis-CLI')}\n\n` +
-                `• Type ${chalk.cyan('/help')} for commands.\n` +
-                `• Type ${chalk.cyan('/config')} to setup your model.\n` +
-                `• Start typing to chat with your agent.`,
-                {
-                    padding: 1,
-                    margin: 1,
-                    borderStyle: 'round',
-                    borderColor: 'cyan',
-                }
-            )
+            boxen(content, {
+                padding: 1,
+                margin: 0,
+                borderStyle: 'round',
+                borderColor: 'cyan',
+                title: title,
+                titleAlignment: 'left',
+                dimBorder: true,
+                width: 80
+            })
         );
+        console.log('');
     }
 
+
+
     public static printSeparator() {
         console.log(chalk.gray('──────────────────────────────────────────────────'));
     }

package/src/utils/UpdateManager.ts

@@ -0,0 +1,83 @@
+import { exec } from 'child_process';
+import { promisify } from 'util';
+import path from 'path';
+import fs from 'fs';
+import chalk from 'chalk';
+import ora from 'ora';
+import inquirer from 'inquirer';
+
+const execAsync = promisify(exec);
+
+export class UpdateManager {
+    private packageName: string;
+    private currentVersion: string;
+
+    constructor() {
+        const packageJsonPath = path.join(__dirname, '../../package.json');
+        try {
+            const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf-8'));
+            this.packageName = packageJson.name;
+            this.currentVersion = packageJson.version;
+        } catch (e) {
+            // Fallback if running from a context where package.json isn't found easily (e.g. global install oddities)
+            // But usually this works relative to dist/utils/
+            this.packageName = '@indiccoder/mentis-cli';
+            this.currentVersion = '0.0.0';
+        }
+    }
+
+    public async checkAndPerformUpdate(interactive: boolean = true) {
+        const spinner = ora('Checking for updates...').start();
+
+        try {
+            // Check latest version from NPM registry
+            const { stdout } = await execAsync(`npm view ${this.packageName} version`);
+            const latestVersion = stdout.trim();
+
+            if (latestVersion === this.currentVersion) {
+                spinner.succeed(chalk.green(`You are on the latest version (${this.currentVersion}).`));
+                return;
+            }
+
+            spinner.info(chalk.blue(`Update available: ${this.currentVersion} -> ${chalk.bold(latestVersion)}`));
+
+            if (!interactive) {
+                // If running in non-interactive mode (e.g. auto-check prompt), maybe just log it.
+                // But for explicit 'update' command, we usually assume interactive or force.
+                console.log(chalk.yellow(`Run 'mentis update' or '/update' inside the tool to upgrade.`));
+                return;
+            }
+
+            const { confirm } = await inquirer.prompt([{
+                type: 'confirm',
+                name: 'confirm',
+                message: `Do you want to install v${latestVersion} now?`,
+                default: true
+            }]);
+
+            if (confirm) {
+                await this.installUpdate(latestVersion);
+            } else {
+                console.log(chalk.yellow('Update skipped.'));
+            }
+
+        } catch (error: any) {
+            spinner.fail(chalk.red('Failed to check for updates.'));
+            if (process.env.DEBUG) console.error(error);
+        }
+    }
+
+    private async installUpdate(version: string) {
+        const spinner = ora(`Installing ${this.packageName}@${version}...`).start();
+        try {
+            await execAsync(`npm install -g ${this.packageName}@latest`);
+            spinner.succeed(chalk.green('Update completed successfully!'));
+            console.log(chalk.cyan('Please restart Mentis to use the new version.'));
+            process.exit(0);
+        } catch (error: any) {
+            spinner.fail(chalk.red('Update failed.'));
+            console.error(chalk.red('Error details:'), error.message);
+            console.log(chalk.yellow(`Try running: npm install -g ${this.packageName}@latest`));
+        }
+    }
+}