scai 0.1.66 → 0.1.67

package/dist/commands/AskCmd.js

@@ -7,7 +7,8 @@ import { generate } from '../lib/generate.js';
 import { buildContextualPrompt } from '../utils/buildContextualPrompt.js';
 import { generateFocusedFileTree } from '../utils/fileTree.js';
 import { log } from '../utils/log.js';
-import { PROMPT_LOG_PATH, SCAI_HOME, RELATED_FILES_LIMIT, MAX_SUMMARY_LINES, getIndexDir } from '../constants.js';
+import { PROMPT_LOG_PATH, SCAI_HOME, RELATED_FILES_LIMIT, MAX_SUMMARY_LINES, getIndexDir, MAX_FUNCTION_LINES } from '../constants.js';
+import chalk from 'chalk';
 export async function runAskCommand(query) {
     if (!query) {
         query = await promptOnce('💬 Ask your question:\n');
@@ -81,10 +82,15 @@ export async function runAskCommand(query) {
     try {
         code = fs.readFileSync(filepath, 'utf-8');
         const topFileId = topFile.id;
-        topFunctions = allFunctionsMap[topFileId]?.map(fn => ({
-            name: fn.name,
-            content: fn.content
-        })) || [];
+        topFunctions = allFunctionsMap[topFileId]?.map(fn => {
+            const content = fn.content
+                ? fn.content.split('\n').slice(0, MAX_FUNCTION_LINES).join('\n')
+                : '(No content available)';
+            return {
+                name: fn.name,
+                content,
+            };
+        }) || [];
     }
     catch (err) {
         console.warn(`⚠️ Failed to read or analyze top file (${filepath}):`, err);
@@ -96,10 +102,15 @@ export async function runAskCommand(query) {
         if (summary) {
             summary = summary.split('\n').slice(0, MAX_SUMMARY_LINES).join('\n');
         }
-        const functions = allFunctionsMap[fileId]?.map(fn => ({
-            name: fn.name,
-            content: fn.content || '(No content available)', // Assuming content is available
-        })) || [];
+        const functions = allFunctionsMap[fileId]?.map(fn => {
+            const content = fn.content
+                ? fn.content.split('\n').slice(0, MAX_FUNCTION_LINES).join('\n')
+                : '(No content available)';
+            return {
+                name: fn.name,
+                content,
+            };
+        }) || [];
         return {
             path: file.path,
             summary,
@@ -115,6 +126,8 @@ export async function runAskCommand(query) {
         console.warn('⚠️ Could not generate file tree:', e);
     }
     // 🟩 STEP 7: Build prompt
+    console.log(chalk.blueBright('\n📦 Building contextual prompt...'));
+    console.log(chalk.gray(`[runAskCommand] Calling buildContextualPrompt()`));
     const promptContent = buildContextualPrompt({
         baseInstruction: query,
         code,
@@ -124,6 +137,8 @@ export async function runAskCommand(query) {
         projectFileTree: fileTree || undefined,
         fileFunctions,
     });
+    console.log(chalk.greenBright('✅ Prompt built successfully.'));
+    console.log(chalk.cyan(`[runAskCommand] Prompt token estimate: ~${Math.round(promptContent.length / 4)} tokens`));
     // 🟩 STEP 8: Save prompt
     try {
         if (!fs.existsSync(SCAI_HOME))

package/dist/constants.js

@@ -61,3 +61,7 @@ export const CANDIDATE_LIMIT = 100;
  * Limit number of summary lines
  */
 export const MAX_SUMMARY_LINES = 12;
+/**
+ * Limit number of function content
+ */
+export const MAX_FUNCTION_LINES = 12;

package/dist/utils/buildContextualPrompt.js

@@ -1,33 +1,56 @@
-export function buildContextualPrompt({ baseInstruction, code, summary, functions, relatedFiles, projectFileTree }) {
+// File: src/utils/buildContextualPrompt.ts
+import chalk from 'chalk';
+function estimateTokenCount(text) {
+    return Math.round(text.length / 4); // simple heuristic approximation
+}
+export function buildContextualPrompt({ baseInstruction, code, summary, functions, relatedFiles, projectFileTree, }) {
     const parts = [baseInstruction];
     if (summary) {
         parts.push(`📄 File Summary:\n${summary}`);
     }
     if (functions?.length) {
-        // Display each function's name and content
         const formattedFunctions = functions
             .map(fn => `• ${fn.name}:\n${fn.content}`)
-            .join('\n\n'); // Adds a line break between each function
+            .join('\n\n');
         parts.push(`🔧 Functions:\n${formattedFunctions}`);
     }
     else {
-        console.log(`🔧 No functions found `);
+        console.log(chalk.gray(`[buildContextualPrompt]`) +
+            chalk.yellow(` ⚠️ No functions found in top file.`));
     }
     if (relatedFiles?.length) {
-        // Include functions from related files
         const formattedRelatedFiles = relatedFiles
             .map(f => {
             const relatedFunctions = f.functions
                 .map(fn => `  • ${fn.name}:\n    ${fn.content}`)
-                .join('\n\n'); // Adds a line break between related file functions
+                .join('\n\n');
             return `• ${f.path}: ${f.summary}\n${relatedFunctions}`;
         })
-            .join('\n\n'); // Adds a line break between related files
+            .join('\n\n');
         parts.push(`📚 Related Files:\n${formattedRelatedFiles}`);
     }
     if (projectFileTree) {
         parts.push(`📁 Project File Structure:\n\`\`\`\n${projectFileTree.trim()}\n\`\`\``);
     }
     parts.push(`\n--- CODE START ---\n${code}\n--- CODE END ---`);
-    return parts.join('\n\n');
+    const prompt = parts.join('\n\n');
+    const tokenEstimate = estimateTokenCount(prompt);
+    // 🔵 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(', ')));
+    console.log(labelColor('🔍 Key:'), contentColor('[buildContextualPrompt]\n'));
+    return prompt;
 }

package/package.json

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