scai 0.1.113 → 0.1.115

package/dist/CHANGELOG.md

@@ -189,4 +189,8 @@ Type handling with the module pipeline
 4. Updated Edge/Table schema for better query performance.
 5. Update package-lock.json to caniuse-lite@1.0.30001741.
 6. Enable execution of  as an executable file in the scripts.
-7. Remove context failure if models not installed. Add ability to set global model.
+7. Remove context failure if models not installed. Add ability to set global model.
+
+## 2025-09-13
+
+• Improve robustness of context update logic

package/dist/config.js

@@ -151,7 +151,7 @@ export const Config = {
                 ? chalk.green(`✅ ${key} (active)`)
                 : chalk.white(`  ${key}`);
             console.log(`- ${label}`);
-            console.log(`    ↳ indexDir: ${r.indexDir}`);
+            console.log(`    ↳ indexDir: ${isActive ? chalk.yellow(r.indexDir) : r.indexDir}`);
         }
     },
     getGitHubToken() {

package/dist/context.js

@@ -1,7 +1,6 @@
 // context.ts
 import { readConfig, writeConfig } from "./config.js";
 import { normalizePath } from "./utils/contentUtils.js";
-import { getHashedRepoKey } from "./utils/repoKey.js";
 import { getDbForRepo, getDbPathForRepo } from "./db/client.js";
 import fs from "fs";
 import chalk from "chalk";
@@ -23,32 +22,39 @@ function modelExists(model) {
 export async function updateContext() {
     const cwd = normalizePath(process.cwd());
     const cfg = readConfig();
+    // 🔎 Look up existing repo by indexDir
     let repoKey = Object.keys(cfg.repos || {}).find((key) => normalizePath(cfg.repos[key]?.indexDir || "") === cwd);
-    let isNewRepo = false;
     if (!repoKey) {
-        repoKey = getHashedRepoKey(cwd);
-        if (!cfg.repos[repoKey])
-            cfg.repos[repoKey] = {};
-        cfg.repos[repoKey].indexDir = cwd;
-        isNewRepo = true;
+        const err = new Error();
+        err.message =
+            chalk.red("❌ Current directory is not in your list of indexed repos. See the 'scai index list' command.") +
+                "\n" +
+                chalk.blueBright("  ↳ To get the full benefit of an indexed repo, navigate to your repository root folder.") +
+                "\n\n" +
+                chalk.greenBright("  Run ") +
+                chalk.cyan("scai index set") +
+                chalk.greenBright(" command. This enables Scai to index the repo, find git credentials etc.\n");
+        throw err;
     }
     const activeRepoChanged = cfg.activeRepo !== repoKey;
     cfg.activeRepo = repoKey;
     writeConfig(cfg);
     const repoCfg = cfg.repos[repoKey];
     let ok = true;
-    if (isNewRepo || activeRepoChanged) {
+    if (activeRepoChanged) {
         console.log(chalk.yellow("\n🔁 Updating context...\n"));
         console.log(`✅ Active repo: ${chalk.green(repoKey)}`);
         console.log(`✅ Index dir: ${chalk.cyan(repoCfg.indexDir || cwd)}`);
     }
+    // 🔑 Token check
     const token = repoCfg.githubToken || cfg.githubToken;
     if (!token) {
         console.log(`ℹ️ No GitHub token found. You can set one with: ${chalk.bold(chalk.bgGreen("scai auth set"))}`);
     }
-    else if (isNewRepo || activeRepoChanged) {
+    else if (activeRepoChanged) {
         console.log(`✅ GitHub token present`);
     }
+    // 💾 DB check
     const dbPath = getDbPathForRepo();
     if (!fs.existsSync(dbPath)) {
         console.log(chalk.yellow(`📦 Initializing DB at ${dbPath}`));
@@ -59,7 +65,7 @@ export async function updateContext() {
             ok = false;
         }
     }
-    else if (isNewRepo || activeRepoChanged) {
+    else if (activeRepoChanged) {
         console.log(chalk.green("✅ Database present"));
     }
     // 🧠 Model check

package/dist/index.js

@@ -232,10 +232,10 @@ index
 index
     .command('set [dir]')
     .description('Set and activate index directory')
-    .action(async (dir = process.cwd()) => await withContext(async () => {
-    Config.setIndexDir(dir);
+    .action(async (dir = process.cwd()) => {
+    await Config.setIndexDir(dir);
     Config.show();
-}));
+});
 index
     .command('list')
     .description('List all indexed repositories')
@@ -336,9 +336,13 @@ cmd.addHelpText('after', `
 
 💡 Use with caution and expect possible changes or instability.
 `);
-cmd.parse(process.argv);
 async function withContext(action) {
     const ok = await updateContext();
     //if (!ok) process.exit(1);
     await action();
 }
+// 👇 this should be the very last line
+cmd.parseAsync(process.argv).catch((err) => {
+    console.error(err.message); // only show our styled message
+    process.exit(1);
+});

package/dist/lib/generate.js

@@ -6,9 +6,6 @@ export async function generate(input) {
     const model = Config.getModel();
     const contextLength = readConfig().contextLength ?? 8192;
     let prompt = input.content;
-    if (prompt.length > contextLength) {
-        console.warn(`⚠️ Warning: Input prompt length (${prompt.length}) exceeds model context length (${contextLength}).`);
-    }
     const spinner = new Spinner(`🧠 Thinking with ${model}...`);
     spinner.start();
     try {

package/dist/utils/buildContextualPrompt.js

@@ -6,33 +6,51 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
     const log = (...args) => console.log("[buildContextualPrompt]", ...args);
     const promptSections = [];
     const seenPaths = new Set();
-    // ✂️ Word-based summarizer with progressive shortening
+    // --- Utility: Summarize text to a few words ---
     function summarizeForPrompt(summary, maxWords = 30) {
         if (!summary)
             return undefined;
         const words = summary.split(/\s+/);
-        if (words.length <= maxWords)
-            return summary.trim();
-        return words.slice(0, maxWords).join(" ") + " …";
+        return words.length <= maxWords ? summary.trim() : words.slice(0, maxWords).join(" ") + " …";
     }
     // --- Step 1: Top file summary ---
     if (topFile.summary) {
         promptSections.push(`**Top file:** ${topFile.path}\n${topFile.summary}`);
         seenPaths.add(topFile.path);
     }
-    // --- Step 2: KG entities/tags for top file ---
-    const topEntitiesStmt = db.prepare(`
+    // ===========================
+    // SECTION A: Database queries
+    // ===========================
+    // --- Step 2a: KG entities/tags ---
+    const topEntitiesRows = db.prepare(`
     SELECT et.entity_type, et.entity_id, tm.name AS tag
     FROM entity_tags et
     JOIN tags_master tm ON et.tag_id = tm.id
     WHERE et.entity_id = ?
-  `);
-    const topEntitiesRows = topEntitiesStmt.all(topFile.id);
+  `).all(topFile.id);
     if (topEntitiesRows.length > 0) {
         const tags = topEntitiesRows.map(r => `- **${r.entity_type}**: ${r.tag}`);
         promptSections.push(`**Knowledge Graph context for ${topFile.path}:**\n${tags.join("\n")}`);
     }
-    // --- Step 3: Recursive KG traversal ---
+    // --- Step 2b: Functions in this file ---
+    const functionRows = db
+        .prepare(`SELECT name, start_line, end_line, content FROM functions WHERE file_id = ? ORDER BY start_line`)
+        .all(topFile.id);
+    const FUNCTION_LIMIT = 15;
+    const hasMoreFunctions = functionRows.length > FUNCTION_LIMIT;
+    const functionsSummary = functionRows.slice(0, FUNCTION_LIMIT).map(f => {
+        const lines = f.content?.split("\n").map(l => l.trim()).filter(Boolean) || ["[no content]"];
+        const preview = lines.slice(0, 3) // first 3 lines
+            .map(l => l.slice(0, 200) + (l.length > 200 ? "…" : ""))
+            .join(" | ");
+        return `- ${f.name || "[anonymous]"} (lines ${f.start_line}-${f.end_line}) — ${preview}`;
+    });
+    if (functionsSummary.length) {
+        promptSections.push(`**Functions in ${topFile.path} (showing ${functionsSummary.length}${hasMoreFunctions ? ` of ${functionRows.length}` : ""}):**\n${functionsSummary.join("\n")}`);
+    }
+    // ===============================
+    // SECTION B: Graph / KG traversal
+    // ===============================
     const kgRelatedStmt = db.prepare(`
     SELECT DISTINCT f.id, f.path, f.summary
     FROM edges e
@@ -42,18 +60,10 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
       AND e.source_id = ?
   `);
     function getRelatedKGFiles(fileId, visited = new Set()) {
-        if (visited.has(fileId)) {
-            log(`🔹 Already visited fileId ${fileId}, skipping`);
+        if (visited.has(fileId))
             return [];
-        }
         visited.add(fileId);
         const rows = kgRelatedStmt.all(fileId);
-        if (rows.length === 0) {
-            log(`⚠️ No edges found for fileId ${fileId}`);
-        }
-        else {
-            log(`🔹 Found ${rows.length} related files for fileId ${fileId}:`, rows.map(r => r.path));
-        }
         let results = [];
         for (const row of rows) {
             results.push(row);
@@ -62,12 +72,9 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
         return results;
     }
     function buildFileTree(file, depth, visited = new Set()) {
-        log(`buildFileTree - file=${file.path}, depth=${depth}`);
-        if (visited.has(file.id)) {
+        if (visited.has(file.id))
             return { id: file.id.toString(), path: file.path };
-        }
         visited.add(file.id);
-        // progressively shorten summaries, drop at depth <= 1
         const maxWordsByDepth = depth >= 3 ? 30 : depth === 2 ? 15 : 0;
         const node = {
             id: file.id.toString(),
@@ -78,7 +85,6 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
             const relatedFiles = getRelatedKGFiles(file.id, visited)
                 .map(f => ({ id: f.id, path: f.path, summary: f.summary }))
                 .slice(0, 5); // cap children
-            log(`File ${file.path} has ${relatedFiles.length} related files`);
             const relatedNodes = relatedFiles.map(f => buildFileTree(f, depth - 1, visited));
             if (relatedNodes.length)
                 node.related = relatedNodes;
@@ -86,12 +92,10 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
         return node;
     }
     const kgTree = buildFileTree({ id: topFile.id, path: topFile.path, summary: topFile.summary }, kgDepth);
-    const kgJson = JSON.stringify(kgTree, null, 2);
-    promptSections.push(`**KG-Related Files (JSON tree, depth ${kgDepth}):**\n\`\`\`json\n${kgJson}\n\`\`\``);
-    // --- Step 4: File tree (shallow, depth 2) ---
-    let fileTree = "";
+    promptSections.push(`**KG-Related Files (JSON tree, depth ${kgDepth}):**\n\`\`\`json\n${JSON.stringify(kgTree, null, 2)}\n\`\`\``);
+    // --- Step 3: File tree (shallow) ---
     try {
-        fileTree = generateFocusedFileTree(topFile.path, 2);
+        const fileTree = generateFocusedFileTree(topFile.path, 2);
         if (fileTree) {
             promptSections.push(`**Focused File Tree (depth 2):**\n\`\`\`\n${fileTree}\n\`\`\``);
         }
@@ -99,21 +103,18 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
     catch (e) {
         console.warn("⚠️ Could not generate file tree:", e);
     }
-    // --- Step 5: Code snippet ---
-    // Only include raw code if no summary exists, or if the query explicitly asks for it
+    // --- Step 4: Optional code snippet ---
     const MAX_LINES = 50;
     const queryNeedsCode = /\b(code|implementation|function|snippet)\b/i.test(query);
     if ((!topFile.summary || queryNeedsCode) && topFile.code) {
         const lines = topFile.code.split("\n").slice(0, MAX_LINES);
         let snippet = lines.join("\n");
-        if (topFile.code.split("\n").length > MAX_LINES) {
+        if (topFile.code.split("\n").length > MAX_LINES)
             snippet += "\n... [truncated]";
-        }
         promptSections.push(`**Code Context (first ${MAX_LINES} lines):**\n\`\`\`\n${snippet}\n\`\`\``);
     }
-    // --- Step 6: User query ---
+    // --- Step 5: User query ---
     promptSections.push(`**Query:** ${query}`);
-    // --- Step 7: Combine prompt ---
     const promptText = promptSections.join("\n\n---\n\n");
     log("✅ Contextual prompt built for:", topFile.path);
     log("📄 Prompt preview:\n", promptText + "\n");

package/package.json

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