scai 0.1.61 → 0.1.63

package/dist/commands/AskCmd.js

@@ -17,7 +17,7 @@ export async function runAskCommand(query) {
         console.error('❌ No question provided.\n👉 Usage: scai ask "your question"');
         return;
     }
-    console.log(`📁 Using index root: ${getIndexDir}`);
+    console.log(`📁 Using index root: ${getIndexDir()}`);
     console.log(`🔍 Searching for: "${query}"\n`);
     // 🟩 STEP 1: Semantic Search
     const start = Date.now();

package/dist/config.js

@@ -6,6 +6,7 @@ import { getRepoKeyForPath, normalizePath } from './utils/normalizePath.js';
 import chalk from 'chalk';
 const defaultConfig = {
     model: 'llama3',
+    contextLength: 8192,
     language: 'ts',
     indexDir: '',
     githubToken: '',
@@ -17,7 +18,7 @@ function ensureConfigDir() {
         fs.mkdirSync(SCAI_HOME, { recursive: true });
     }
 }
-function readConfig() {
+export function readConfig() {
     try {
         const content = fs.readFileSync(CONFIG_PATH, 'utf-8');
         return { ...defaultConfig, ...JSON.parse(content) };

package/dist/constants.js

@@ -52,7 +52,7 @@ export function getIndexDir() {
 /**
  * Limit for number of related files included in model prompt.
  */
-export const RELATED_FILES_LIMIT = 3;
+export const RELATED_FILES_LIMIT = 5;
 /**
  * Limit for number of candidate files to score.
  */

package/dist/index.js

@@ -1,10 +1,9 @@
-//!/usr/bin/env node
-import { Command } from "commander";
-import { Config } from './config.js';
+#!/usr/bin/env node
 import { createRequire } from 'module';
 const require = createRequire(import.meta.url);
 const { version } = require('../package.json');
-// 🧠 Commands
+import { Command } from "commander";
+import { Config } from './config.js';
 import { suggestCommitMessage } from "./commands/CommitSuggesterCmd.js";
 import { handleRefactor } from "./commands/RefactorCmd.js";
 import { generateTests } from "./commands/TestGenCmd.js";
@@ -27,8 +26,8 @@ import { validateGitHubTokenAgainstRepo } from "./github/githubAuthCheck.js";
 import { checkGit } from "./commands/GitCmd.js";
 import { runSwitchCommand, runInteractiveSwitch } from "./commands/SwitchCmd.js";
 import { execSync } from "child_process";
-import { dirname, resolve } from "path";
 import { fileURLToPath } from "url";
+import { dirname, resolve } from "path";
 // 🎛️ CLI Setup
 const cmd = new Command('scai')
     .version(version)

package/dist/modelSetup.js

@@ -1,96 +1,112 @@
 import { spawn, execSync } from 'child_process';
 import * as readline from 'readline';
-// Port and models
+import * as fs from 'fs';
+import * as path from 'path';
+import chalk from 'chalk';
+import { getDbForRepo } from './db/client.js'; // Ensure this function works correctly
+import { readConfig, writeConfig } from './config.js';
+import { CONFIG_PATH } from './constants.js';
+// Constants
 const MODEL_PORT = 11434;
-const REQUIRED_MODELS = ['llama3', 'mistral']; // Add more if needed
-// Ensure Ollama is running
+const REQUIRED_MODELS = ['llama3', 'mistral']; // Expand as needed
+const isYesMode = process.argv.includes('--yes') || process.env.SCAI_YES === '1';
+// 🧠 Auto init config/db if missing
+export async function autoInitIfNeeded() {
+    const cfg = readConfig();
+    if (!fs.existsSync(CONFIG_PATH)) {
+        console.log(chalk.green('🛠️  Config not found. Initializing...'));
+        writeConfig({}); // This will create config.json with defaults
+    }
+    const activeRepo = cfg.activeRepo && cfg.repos[cfg.activeRepo];
+    if (activeRepo) {
+        const dbPath = path.join(activeRepo.indexDir, 'scai.db');
+        if (!fs.existsSync(dbPath)) {
+            console.log(chalk.green('📦 DB not found. Initializing...'));
+            getDbForRepo(); // This creates the DB
+        }
+    }
+}
+// 🚀 Ensure Ollama server is running
 async function ensureOllamaRunning() {
     try {
         const res = await fetch(`http://localhost:${MODEL_PORT}`);
         if (res.ok) {
             console.log('✅ Ollama is already running.');
+            return;
         }
     }
-    catch (error) {
-        console.error('🟡 Ollama is not running. Starting it in the background...');
-        if (error instanceof Error) {
-            console.error('❌ Error during Ollama health check:', error.message);
-        }
-        else {
-            console.error('❌ Unexpected error during Ollama health check:', error);
-        }
+    catch {
+        // Continue to spawn below
+    }
+    console.log(chalk.yellow('⚙️  Ollama is not running. Starting it in the background...'));
+    try {
         const child = spawn('ollama', ['serve'], {
             detached: true,
             stdio: 'ignore',
             windowsHide: true,
         });
         child.unref();
-        await new Promise((res) => setTimeout(res, 3000)); // Wait a bit for server to be ready
+        await new Promise((res) => setTimeout(res, 3000));
+        console.log('✅ Ollama started.');
+    }
+    catch (err) {
+        console.error('❌ Failed to start Ollama:', err);
+        process.exit(1);
     }
 }
-// Get installed models via ollama list
+// 🧰 List installed models
 async function getInstalledModels() {
     try {
         const result = execSync('ollama list', { encoding: 'utf-8' });
-        const installedModels = result
+        return result
             .split('\n')
-            .map((line) => line.split(/\s+/)[0].split(':')[0]) // Get model name, ignore version (e.g., 'llama3:latest' becomes 'llama3')
-            .filter((model) => REQUIRED_MODELS.includes(model)); // Filter based on required models
-        return installedModels;
+            .map((line) => line.split(/\s+/)[0].split(':')[0])
+            .filter((model) => REQUIRED_MODELS.includes(model));
     }
-    catch (error) {
-        console.error('❌ Failed to fetch installed models:', error instanceof Error ? error.message : error);
+    catch (err) {
+        console.error('❌ Could not fetch installed models:', err);
         return [];
     }
 }
-// Prompt user for input
+// 💬 Prompt user
 function promptUser(question) {
-    const rl = readline.createInterface({
-        input: process.stdin,
-        output: process.stdout,
-    });
-    return new Promise((resolve) => rl.question(question, (answer) => {
+    if (isYesMode)
+        return Promise.resolve('y');
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    return new Promise((resolve) => rl.question(question, (a) => {
         rl.close();
-        resolve(answer.trim());
+        resolve(a.trim());
     }));
 }
-// Ensure all required models are downloaded
+// 📥 Download missing models
 async function ensureModelsDownloaded() {
-    const installedModels = await getInstalledModels();
-    const missingModels = REQUIRED_MODELS.filter((model) => !installedModels.includes(model));
-    if (missingModels.length === 0) {
-        console.log('✅ All required models are already installed.');
+    const installed = await getInstalledModels();
+    const missing = REQUIRED_MODELS.filter((m) => !installed.includes(m));
+    if (!missing.length) {
+        console.log('✅ All required models are installed.');
         return;
     }
-    console.log(`🟡 Missing models: ${missingModels.join(', ')}`);
-    const answer = await promptUser('Do you want to download the missing models now? (y/N): ');
+    console.log(chalk.yellow(`📦 Missing models: ${missing.join(', ')}`));
+    const answer = await promptUser('⬇️  Do you want to download them now? (y/N): ');
     if (answer.toLowerCase() !== 'y') {
-        console.log('🚫 Missing models not downloaded. Exiting.');
+        console.log('🚫 Aborting due to missing models.');
         process.exit(1);
     }
-    for (const model of missingModels) {
+    for (const model of missing) {
         try {
-            console.log(`⬇️ Pulling model: ${model} ...`);
+            console.log(`⬇️  Pulling ${model}...`);
             execSync(`ollama pull ${model}`, { stdio: 'inherit' });
-            console.log(`✅ Successfully pulled ${model}.`);
+            console.log(chalk.green(`✅ Pulled ${model}`));
         }
         catch (err) {
-            console.error(`❌ Failed to pull ${model}:`, err instanceof Error ? err.message : err);
+            console.error(`❌ Failed to pull ${model}:`, err);
             process.exit(1);
         }
     }
 }
-// Initialize the application
+// 🏁 Main bootstrap logic
 export async function bootstrap() {
-    try {
-        // Ensure Ollama is running only once at the start
-        await ensureOllamaRunning();
-        // Ensure models are downloaded once
-        await ensureModelsDownloaded();
-        // Now your CLI logic can proceed here...
-    }
-    catch (error) {
-        console.error('❌ Error during initialization:', error instanceof Error ? error.message : error);
-        process.exit(1);
-    }
+    await autoInitIfNeeded();
+    await ensureOllamaRunning();
+    await ensureModelsDownloaded();
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "scai",
-  "version": "0.1.61",
+  "version": "0.1.63",
   "type": "module",
   "bin": {
     "scai": "./dist/index.js"