promptgraph-mcp 2.9.26 → 2.9.28

package/indexer.js

@@ -19,7 +19,7 @@ function sanitizePath(filePath) {
   return path.resolve(filePath);
 }
 
-export async function indexBatch(db, skills, { fast = false, silent = false } = {}) {
+export async function indexBatch(db, skills, { fast = false, silent = false, onEmbed = null } = {}) {
   const upsertSkill = db.prepare(`
     INSERT INTO skills (id, name, description, path, source, content, hash, version, author, license, updated_at, downloads, verified)
     VALUES (@id, @name, @description, @path, @source, @content, @hash, @version, @author, @license, @updated_at, @downloads, @verified)
@@ -72,7 +72,9 @@ export async function indexBatch(db, skills, { fast = false, silent = false } =
     const total = texts.length;
     if (!silent) process.stdout.write('\n  ⠋ Preparing model...\n');
     let embedStart = null;
-    const embeddings = await embedBatch(texts, silent ? null : (done, tot) => {
+    const embeddings = await embedBatch(texts, (done, tot) => {
+      if (onEmbed) { onEmbed(done, tot); return; }
+      if (silent) return;
       if (!embedStart) {
         process.stdout.write('\x1b[1A\x1b[2K');
         embedStart = Date.now();
@@ -209,14 +211,23 @@ export async function indexAll({ fast = false } = {}) {
       if (!isSkillFile(file, raw)) { skipped++; count++; continue; }
 
       if (!processedStart) processedStart = Date.now();
+      count++;
+      if (count % 50 === 0) {
+        const elapsed = Date.now() - start;
+        const eta = processedCount > 0 && processedStart
+          ? Math.round((total - count) * (Date.now() - processedStart) / processedCount / 1000)
+          : elapsed > 500 ? Math.round((total - count) * elapsed / count / 1000) : '?';
+        progress(count, total, { skipped, eta, errors });
+      }
       const parsed = parseSkillFile(file, source, { raw });
       batch.push({ ...parsed, hash });
 
       if (batch.length >= BATCH_SIZE) {
         const filtered = await filterWithClassifier(batch);
         classifierRemoved += batch.length - filtered.length;
-        await indexBatch(db, filtered, { fast, silent: true });
-        count += filtered.length;
+        await indexBatch(db, filtered, { fast, silent: true, onEmbed: (done, tot) => {
+          process.stdout.write(`\r  ⠋ Embedding ${done}/${tot} chunks...   `);
+        }});
         processedCount += filtered.length;
         batch = [];
         const eta = processedCount > 0 ? Math.round((total - count) * (Date.now() - processedStart) / processedCount / 1000) : '?';
@@ -241,8 +252,9 @@ export async function indexAll({ fast = false } = {}) {
   if (batch.length > 0) {
     const filtered = await filterWithClassifier(batch);
     classifierRemoved += batch.length - filtered.length;
-    await indexBatch(db, filtered, { fast, silent: true });
-    count += filtered.length;
+    await indexBatch(db, filtered, { fast, silent: true, onEmbed: (done, tot) => {
+      process.stdout.write(`\r  ⠋ Embedding ${done}/${tot} chunks...   `);
+    }});
   }
 
   progress(total, total, { skipped, errors });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "promptgraph-mcp",
-  "version": "2.9.26",
+  "version": "2.9.28",
   "files": [
     "*.js",
     "commands/",