scai 0.1.67 → 0.1.68

package/dist/utils/buildContextualPrompt.js

@@ -1,5 +1,6 @@
 // File: src/utils/buildContextualPrompt.ts
 import chalk from 'chalk';
+import path from 'path';
 function estimateTokenCount(text) {
     return Math.round(text.length / 4); // simple heuristic approximation
 }
@@ -36,21 +37,39 @@ export function buildContextualPrompt({ baseInstruction, code, summary, function
     const prompt = parts.join('\n\n');
     const tokenEstimate = estimateTokenCount(prompt);
     // 🔵 Colorized diagnostic output
+    // 🔵 Colorized diagnostic output
     const header = chalk.bgBlue.white.bold(' [SCAI] Prompt Overview ');
     const labelColor = chalk.cyan;
     const contentColor = chalk.gray;
     console.log('\n' + header);
     console.log(labelColor('🔢 Token Estimate:'), contentColor(`${tokenEstimate.toLocaleString()} tokens`));
-    console.log(labelColor('🧩 Prompt Sections:'));
-    console.log(contentColor([
-        summary ? '📄 Summary' : null,
-        functions?.length ? '🔧 Functions' : null,
-        relatedFiles?.length ? '📚 Related Files' : null,
-        projectFileTree ? '📁 File Tree' : null,
-        '📦 Code',
-    ]
-        .filter(Boolean)
-        .join(', ')));
+    // 📄 Summary
+    if (summary) {
+        console.log(labelColor('📄 Summary:'), contentColor(`${estimateTokenCount(summary).toLocaleString()} tokens`));
+    }
+    // 🔧 Functions
+    if (functions?.length) {
+        const fnCount = functions.length;
+        const fnTokens = functions.reduce((sum, f) => sum + estimateTokenCount(f.content), 0);
+        console.log(labelColor(`🔧 Functions (${fnCount}):`), contentColor(`${fnTokens.toLocaleString()} tokens`));
+    }
+    // 📚 Related Files
+    if (relatedFiles?.length) {
+        const relCount = relatedFiles.length;
+        const relTokens = relatedFiles.reduce((sum, f) => sum + estimateTokenCount(f.summary), 0);
+        console.log(labelColor(`📚 Related Files (${relCount}):`), contentColor(`${relTokens.toLocaleString()} tokens`));
+        // Optional: Show top 3 file names
+        const fileList = relatedFiles.slice(0, 3).map(f => `- ${path.basename(f.path)}`).join('\n');
+        if (fileList)
+            console.log(contentColor(fileList + (relCount > 3 ? `\n  ...+${relCount - 3} more` : '')));
+    }
+    // 📁 File Tree
+    if (projectFileTree) {
+        console.log(labelColor('📁 File Tree:'), contentColor(`${estimateTokenCount(projectFileTree).toLocaleString()} tokens`));
+    }
+    // 📦 Code Section
+    console.log(labelColor('📦 Code:'), contentColor(`${estimateTokenCount(prompt).toLocaleString()} tokens`));
+    // 📌 Key
     console.log(labelColor('🔍 Key:'), contentColor('[buildContextualPrompt]\n'));
     return prompt;
 }

package/dist/utils/fileTree.js

@@ -1,17 +1,38 @@
 import fs from 'fs';
 import path from 'path';
 import { getIndexDir } from '../constants.js';
-export function generateFocusedFileTree(focusPath, maxDepth = 2) {
+/**
+ * Generate a reduced file tree centered around the focus path, including nearby sibling folders.
+ */
+export function generateFocusedFileTree(focusPath, maxDepth = 2, siblingWindow = 2) {
     const absoluteFocus = path.resolve(focusPath);
-    const parentDir = path.dirname(absoluteFocus);
+    const fileOrDir = fs.statSync(absoluteFocus);
+    const targetDir = fileOrDir.isDirectory()
+        ? absoluteFocus
+        : path.dirname(absoluteFocus);
+    const parentDir = path.dirname(targetDir);
     const indexDir = getIndexDir();
-    const relativeTitle = path.relative(indexDir, parentDir).replace(/\\/g, '/');
-    const tree = generateFileTree(parentDir, maxDepth, absoluteFocus);
-    return `📂 ${relativeTitle || '.'}\n${tree}`;
+    const siblings = fs
+        .readdirSync(parentDir, { withFileTypes: true })
+        .filter(entry => entry.isDirectory())
+        .sort((a, b) => a.name.localeCompare(b.name));
+    const focusIndex = siblings.findIndex(entry => path.resolve(path.join(parentDir, entry.name)) === path.resolve(targetDir));
+    const start = Math.max(0, focusIndex - siblingWindow);
+    const end = Math.min(siblings.length, focusIndex + siblingWindow + 1);
+    const nearbySiblings = siblings.slice(start, end);
+    const relativeTitle = path.relative(indexDir, parentDir).replace(/\\/g, '/') || '.';
+    let output = `📂 ${relativeTitle}\n`;
+    nearbySiblings.forEach(entry => {
+        const siblingPath = path.join(parentDir, entry.name);
+        const isFocusDir = path.resolve(siblingPath) === path.resolve(targetDir);
+        const tree = generateFileTree(siblingPath, maxDepth - 1, isFocusDir ? absoluteFocus : undefined, '│   ');
+        output += `${isFocusDir ? '➡️ ' : ''}${entry.name}/\n${tree}`;
+    });
+    return output;
 }
 function generateFileTree(dir, depth, highlightPath, prefix = '') {
     if (depth < 0)
-        return ''; // Stop at maxDepth
+        return '';
     let output = '';
     const entries = fs.readdirSync(dir, { withFileTypes: true });
     const sorted = entries.sort((a, b) => Number(b.isDirectory()) - Number(a.isDirectory()));
@@ -20,13 +41,11 @@ function generateFileTree(dir, depth, highlightPath, prefix = '') {
         const connector = isLast ? '└── ' : '├── ';
         const fullPath = path.join(dir, entry.name);
         const isHighlighted = highlightPath && path.resolve(fullPath) === path.resolve(highlightPath);
-        // If it is a directory, recurse and reduce depth
         if (entry.isDirectory()) {
             output += `${prefix}${connector}${entry.name}/\n`;
             output += generateFileTree(fullPath, depth - 1, highlightPath, prefix + (isLast ? '    ' : '│   '));
         }
         else {
-            // Highlight the file if it matches the focusPath
             const name = isHighlighted ? `➡️ ${entry.name}` : entry.name;
             output += `${prefix}${connector}${name}\n`;
         }

package/package.json

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