commitmind 1.0.3 → 1.1.0

package/README.md

@@ -96,6 +96,23 @@ This will:
 
 ---
 
+
+## Model Configuration
+
+CommitMind allows you to choose and persist your preferred local AI model.
+
+### First-time setup
+
+On the first run, CommitMind will automatically prompt you to select an installed Ollama model.  
+Your choice will be saved and used for future commits.
+
+---
+
+### Change model anytime
+
+```bash
+aic model set
+
 ##  Example Workflow
 
 ```bash
@@ -106,6 +123,18 @@ aic push main
 ```
 
 ---
+### View current model
+```bash
+aic model get
+```
+
+#### Override model (optional)
+
+You can also specify a model directly:
+
+```bash
+aic auto --model mistral
+```
 
 ##  Why CommitMind?
 

package/dist/config.js

@@ -0,0 +1,21 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { homedir } from "node:os";
+import path from "node:path";
+const CONFIG_DIR = path.join(homedir(), '.commitmind');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+export function getConfig() {
+    try {
+        if (!existsSync(CONFIG_FILE))
+            return null;
+        return JSON.parse(readFileSync(CONFIG_FILE, 'utf-8'));
+    }
+    catch (err) {
+        return null;
+    }
+}
+export function saveConfig(config) {
+    if (!existsSync(CONFIG_DIR)) {
+        mkdirSync(CONFIG_DIR);
+    }
+    writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
+}

package/dist/index.js

@@ -1,30 +1,71 @@
 #!/usr/bin/env node
 import { stageAll, getDiff, commit, push } from "./git.js";
 import { generateCommit } from "./ollama.js";
+import { getConfig, saveConfig } from "./config.js";
+import { selectModel } from "./selectModel.js";
+async function resolveModel(cliModel) {
+    // 1. CLI flag wins (old users safe)
+    if (cliModel) {
+        saveConfig({ model: cliModel });
+        return cliModel;
+    }
+    const config = getConfig();
+    if (config?.model)
+        return config.model;
+    if (!process.stdout.isTTY) {
+        return "llama3.1:latest";
+    }
+    console.log("No model configured.");
+    return await selectModel();
+}
 async function main() {
-    const command = process.argv[2];
+    const args = process.argv.slice(2);
+    const command = args[0];
     if (!command) {
         console.log(`
 Usage:
 aic auto
 aic push <branch>
 
+Model:
+aic model set
+aic model get
+
 Examples:
 aic auto
 aic push main
 `);
         return;
     }
+    if (command === "model") {
+        const sub = args[1];
+        if (sub === "set") {
+            await selectModel();
+            return;
+        }
+        if (sub === "get") {
+            const config = getConfig();
+            console.log("Current model:", config?.model || "Not set");
+            return;
+        }
+    }
+    // -------- model flag --------
+    let cliModel;
+    const modelIndex = args.indexOf("--model");
+    if (modelIndex !== -1 && args[modelIndex + 1]) {
+        cliModel = args[modelIndex + 1];
+    }
+    const model = await resolveModel(cliModel);
     console.log("📦 Staging changes...");
     await stageAll();
     const diff = await getDiff();
-    console.log(" Generating commit...");
-    const message = await generateCommit(diff);
-    console.log("✅ Commit:", message);
+    console.log(`Generating commit using ${model}...`);
+    const message = await generateCommit(diff, model);
+    console.log("Commit:", message);
     await commit(message);
     if (command === "push") {
-        const branch = process.argv[3] || "main";
-        console.log(` Pushing to ${branch}...`);
+        const branch = args[1] || "main";
+        console.log(`Pushing to ${branch}...`);
         await push(branch);
     }
 }

package/dist/ollama.js

@@ -1,4 +1,4 @@
-export async function generateCommit(diff) {
+export async function generateCommit(diff, model = 'llama3.1:latest') {
     if (!diff) {
         console.log(`No changes done`);
         process.exit(0);
@@ -48,14 +48,20 @@ Now generate the commit message.
 Diff:
 ${diff}
 `;
-    const res = await fetch('http://localhost:11434/api/generate', {
-        method: 'POST',
-        body: JSON.stringify({
-            model: 'llama3.1:latest',
-            prompt,
-            stream: false,
-        })
-    });
-    const data = await res.json();
-    return data.response.trim();
+    try {
+        const res = await fetch('http://localhost:11434/api/generate', {
+            method: 'POST',
+            body: JSON.stringify({
+                model,
+                prompt,
+                stream: false,
+            })
+        });
+        const data = (await res.json());
+        return data.response.trim();
+    }
+    catch (err) {
+        console.log(`Failed to connect to Ollama`);
+        process.exit(1);
+    }
 }

package/dist/selectModel.js

@@ -0,0 +1,40 @@
+import { saveConfig } from "./config.js";
+import readline from "readline";
+async function getOllamaModels() {
+    try {
+        const res = await fetch('http://localhost:11434/api/tags');
+        const data = await res.json();
+        return data.models.map((m) => m.name);
+    }
+    catch (err) {
+        console.log(`Ollama is not running ${err}`);
+        process.exit(1);
+    }
+}
+export async function selectModel() {
+    const models = await getOllamaModels();
+    if (!models.length) {
+        console.log(`No Ollama models found ! `);
+        process.exit(1);
+    }
+    console.log('Choose an Ollama model : ');
+    models.forEach((m, i) => console.log(`${i + 1}.${m}`));
+    const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    const choice = await new Promise((res) => {
+        rl.question('Enter number : ', (ans) => {
+            rl.close();
+            res(Number(ans));
+        });
+    });
+    const selected = models[choice - 1];
+    if (!selected) {
+        console.log(`Invalid Choice`);
+        process.exit(1);
+    }
+    saveConfig({ model: selected });
+    console.log(`Model saved : ${selected}`);
+    return selected;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "commitmind",
-  "version": "1.0.3",
+  "version": "1.1.0",
   "author": "Nabin Sharma",
   "bin": {
     "aic": "./dist/index.js",
@@ -24,7 +24,6 @@
     "typescript": "^5.9.3"
   },
   "dependencies": {
-    "commitmind": "^1.0.0",
     "readline": "^1.3.0"
   }
 }

package/src/selectModel.ts

@@ -1,5 +1,5 @@
 
-import { saveConfig } from "./config";
+import { saveConfig } from "./config.js";
 import readline from "readline";
 
 async function getOllamaModels():Promise<string[]> {