@indiccoder/mentis-cli 1.0.3 → 1.0.5

package/README.md

@@ -5,26 +5,30 @@ Mentis is a powerful terminal-based AI coding assistant that lives in your comma
 
 ## ✨ Features
 
-*   **🤖 Multi-Model Support**: Switch seamlessly between **Gemini**, **Ollama** (Local), and **OpenAI**.
+*   **🤖 Multi-Model Support**: Switch seamlessly between **Gemini**, **Ollama** (Local), **OpenAI**, and **GLM-4.6** (Z.AI Coding).
 *   **🛠️ Agentic Capabilities**: Mentis can read, write, list files, and search your codebase to understand context.
-*   **🌐 Web Intelligence**: Hybrid **Google** + **DuckDuckGo** search for documentation and error fixing (Zero-Config).
-*   **🗺️ Smart Context**: Automatically maps your repository structure for better spatial awareness.
-*   **💰 Cost Awareness**: Real-time token usage and cost estimation displayed after every response.
-*   **🧠 Persistent Memory**: Auto-saves sessions (`/resume`) and supports manual checkpoints (`/checkpoint`).
-*   **🔌 MCP Support**: Full [Model Context Protocol](https://github.com/modelcontextprotocol) client. Connect external tools like databases or memory servers.
-*   **🔍 Codebase Search**: Built-in `grep` tool to find code across your project.
-*   **🐚 Shell Integration**: Run shell commands (`/run`) and commit changes (`/commit`) directly from the agent.
-*   **🎨 Premium UI**: Beautiful terminal interface with modes (`PLAN` vs `BUILD`).
+*   **🌐 Web Intelligence**: Hybrid search for documentation and error fixing.
+*   **🗺️ Smart Context**: Automatically maps your repository structure.
+*   **🧩 Interactive & Robust CLI**:
+    *   **History**: Navigate previous commands with Up/Down arrows (persistent across sessions).
+    *   **Cancellation**: Press `Esc` to instantly stop generation.
+    *   **Markdown**: Beautiful syntax highlighting in terminal output.
+*   **💰 Cost Awareness**: Real-time token usage and cost tracking.
+*   **🧠 Persistent Memory**: Auto-saves sessions (`/resume`) and checkpoints (`/checkpoint`).
+*   **🔌 MCP Support**: Full [Model Context Protocol](https://github.com/modelcontextprotocol) client.
+*   **🔍 Codebase Search**: Built-in `grep` tool.
+*   **🐚 Shell Integration**: Run shell commands and commit changes directly.
 
 ## 🚀 Installation
 
-### From GitHub (Recommended)
-You can install Mentis directly from the repository, ensuring you always get the latest version (bypassing NPM delays).
+### Using NPM (Recommended)
+You can install Mentis globally directly from the NPM registry:
 
 ```bash
-npm install -g git+https://github.com/CoderVLSI/Mentis-CLI.git
+npm install -g @indiccoder/mentis-cli
 ```
 
+
 ### From Source (Dev)
 ```bash
 git clone https://github.com/CoderVLSI/Mentis-CLI.git

package/dist/repl/ReplManager.js

@@ -142,7 +142,8 @@ class ReplManager {
             // 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('>')}`;
+            const modelInfo = this.currentModelName ? ` (${this.currentModelName})` : '';
+            const promptText = `${modeLabel}${chalk_1.default.dim(modelInfo)} ${chalk_1.default.cyan('>')}`;
             // Use readline for basic input to support history
             const answer = await new Promise((resolve) => {
                 const rl = readline.createInterface({
@@ -518,43 +519,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 +576,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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@indiccoder/mentis-cli",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "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/repl/ReplManager.ts

@@ -120,7 +120,8 @@ export class ReplManager {
             // Removed redundancy to keep CLI clean, prompt has info? No, prompt is minimal.
 
             const modeLabel = this.mode === 'PLAN' ? chalk.magenta('PLAN') : chalk.blue('BUILD');
-            const promptText = `${modeLabel} ${chalk.cyan('>')}`;
+            const modelInfo = this.currentModelName ? ` (${this.currentModelName})` : '';
+            const promptText = `${modeLabel}${chalk.dim(modelInfo)} ${chalk.cyan('>')}`;
 
             // Use readline for basic input to support history
             const answer = await new Promise<string>((resolve) => {
@@ -533,42 +534,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 +595,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[]) {