quackstack 1.0.0 → 1.0.2

package/dist/commands/ingest.js

@@ -11,31 +11,35 @@ export async function ingest(rootDir, projectName, silent = false) {
     if (!silent)
         console.log(`Found ${files.length} files to process`);
     let processedCount = 0;
-    for (const filePath of files) {
-        try {
-            const content = fs.readFileSync(filePath, "utf-8");
-            const chunks = chunkCode(content, filePath);
-            for (const chunk of chunks) {
-                const embedding = await aiClient.getEmbeddings(chunk.content);
-                await saveToDB({
-                    content: chunk.content,
-                    embedding,
-                    filePath,
-                    projectName,
-                    language: path.extname(filePath),
-                    functionName: chunk.functionName,
-                    lineStart: chunk.lineStart,
-                    lineEnd: chunk.lineEnd,
-                });
+    const BATCH_SIZE = 10;
+    for (let i = 0; i < files.length; i += BATCH_SIZE) {
+        const batch = files.slice(i, i + BATCH_SIZE);
+        await Promise.all(batch.map(async (filePath) => {
+            try {
+                const content = fs.readFileSync(filePath, "utf-8");
+                const chunks = chunkCode(content, filePath);
+                for (const chunk of chunks) {
+                    const embedding = await aiClient.getEmbeddings(chunk.content);
+                    await saveToDB({
+                        content: chunk.content,
+                        embedding,
+                        filePath,
+                        projectName,
+                        language: path.extname(filePath),
+                        functionName: chunk.functionName,
+                        lineStart: chunk.lineStart,
+                        lineEnd: chunk.lineEnd,
+                    });
+                }
+                processedCount++;
+                if (!silent && processedCount % 10 === 0) {
+                    console.log(`Processed ${processedCount}/${files.length} files...`);
+                }
             }
-            processedCount++;
-            if (!silent && processedCount % 10 === 0) {
-                console.log(`Processed ${processedCount}/${files.length} files...`);
+            catch (error) {
+                console.error(`Error processing ${filePath}:`, error);
             }
-        }
-        catch (error) {
-            console.error(`Error processing ${filePath}:`, error);
-        }
+        }));
     }
     if (!silent)
         console.log(`Done! Processed ${processedCount} files.`);

package/dist/commands/search.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
-import { client } from "../lib/database.js";
 import { aiClient } from "../lib/ai-provider.js";
+import { client } from "../lib/database.js";
 function cosineSim(a, b) {
     let dot = 0, normA = 0, normB = 0;
     for (let i = 0; i < a.length; i++) {
@@ -15,16 +15,22 @@ export async function search(query, projectName) {
     const snippets = await client.codeSnippet.findMany({
         where: { projectName },
     });
-    const ranked = snippets
-        .map(snippet => ({
+    const scored = snippets.map(snippet => ({
         id: snippet.id,
         content: snippet.content,
         filePath: snippet.filePath,
         functionName: snippet.functionName,
         score: cosineSim(queryEmbedding, snippet.embedding),
-    }))
-        .sort((a, b) => b.score - a.score)
-        .slice(0, 5);
+    }));
+    scored.sort((a, b) => b.score - a.score);
+    const seenFiles = new Set();
+    const ranked = scored.filter(item => {
+        if (seenFiles.has(item.filePath)) {
+            return false;
+        }
+        seenFiles.add(item.filePath);
+        return true;
+    }).slice(0, 5);
     const context = ranked
         .map((r, i) => `[${i + 1}] ${r.filePath}${r.functionName ? ` (${r.functionName})` : ""}\n${r.content}`)
         .join("\n\n---\n\n");

package/dist/lib/ai-provider.js

@@ -2,7 +2,7 @@ import OpenAI from "openai";
 import Anthropic from "@anthropic-ai/sdk";
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import dotenv from "dotenv";
-dotenv.config();
+dotenv.config({ quiet: true });
 export class AIClient {
     provider;
     openai;

package/dist/repl.js

@@ -25,23 +25,24 @@ export async function startREPL(forceReindex = false) {
     const rl = readline.createInterface({
         input: process.stdin,
         output: process.stdout,
-        prompt: "🐥 Quack! How can I help? > "
+        terminal: true
     });
     console.log("💡 Tip: Press Ctrl+C to exit\n");
-    rl.prompt();
-    rl.on("line", async (input) => {
-        const trimmed = input.trim();
-        if (!trimmed) {
-            rl.prompt();
-            return;
-        }
-        try {
-            const { answer, sources } = await search(trimmed, PROJECT_NAME);
-            console.log(`\n${answer}\n`);
-            await new Promise((resolve) => {
+    const askQuestion = () => {
+        rl.question("🐥 Quack! How can I help? > ", async (input) => {
+            const trimmed = input.trim();
+            if (!trimmed) {
+                askQuestion();
+                return;
+            }
+            try {
+                rl.pause();
+                const { answer, sources } = await search(trimmed, PROJECT_NAME);
+                console.log(`\n${answer}\n`);
                 const detailRL = readline.createInterface({
                     input: process.stdin,
-                    output: process.stdout
+                    output: process.stdout,
+                    terminal: true
                 });
                 detailRL.question("💡 Want more details? (y/n) > ", (ans) => {
                     if (ans.toLowerCase() === "y") {
@@ -54,15 +55,18 @@ export async function startREPL(forceReindex = false) {
                     }
                     detailRL.close();
                     console.log();
-                    resolve();
+                    rl.resume();
+                    askQuestion();
                 });
-            });
-        }
-        catch (error) {
-            console.error("❌ Error:", error instanceof Error ? error.message : "Unknown error");
-        }
-        rl.prompt();
-    });
+            }
+            catch (error) {
+                console.error("❌ Error:", error instanceof Error ? error.message : "Unknown error");
+                rl.resume();
+                askQuestion();
+            }
+        });
+    };
+    askQuestion();
     rl.on("close", () => {
         console.log("\n👋 Happy coding!");
         process.exit(0);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "quackstack",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Your cracked unpaid intern for all things codebase related! AI-powered codebase search and Q&A.",
   "type": "module",
   "main": "dist/cli.cjs",
@@ -11,7 +11,8 @@
     "dist",
     "prisma",
     "README.md",
-    "LICENSE"
+    "LICENSE",
+    ".npmrc"
   ],
   "publishConfig": {
     "access": "public"
@@ -20,7 +21,7 @@
     "build": "tsc -b",
     "dev": "tsc -b && node dist/cli.cjs",
     "prepare": "prisma generate",
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "postinstall": "prisma generate"
   },
   "keywords": [
     "cli",
@@ -57,13 +58,13 @@
     "chalk-animation": "^2.0.3",
     "commander": "^14.0.1",
     "dotenv": "^17.2.3",
-    "openai": "^6.0.1"
+    "openai": "^6.0.1",
+    "prisma": "^6.16.3"
   },
   "devDependencies": {
     "@types/babel__traverse": "^7.28.0",
     "@types/chalk-animation": "^1.6.3",
     "@types/node": "^20.0.0",
-    "prisma": "^6.16.3",
     "typescript": "^5.9.3"
   }
-}
+}