scai 0.1.114 → 0.1.115

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.114",
+  "version": "0.1.115",
   "type": "module",
   "bin": {
     "scai": "./dist/index.js"