promptgraph-mcp 2.6.4 → 2.6.6

package/github-import.js

@@ -429,36 +429,50 @@ function detectSkillsDirLocal(repoRoot) {
   return { dir: repoRoot, label: '(root)', sparse: false };
 }
 
-// If the embedding classifier is trained, remove files classified as non-skills
+// Remove files classified as non-skills by embedding classifier (only if model is trained)
 async function classifierCleanup(dest) {
   const mdFiles = globSync(`${dest}/**/*.md`);
   if (mdFiles.length === 0) return;
 
   const parsed = [];
   const fileMap = [];
+  let heuristicRemoved = 0;
+
   for (const fp of mdFiles) {
     try {
       const raw = fs.readFileSync(fp, 'utf8');
       if (!isSkillFile(fp, raw)) continue;
+      if (!seemsLikeSkill(raw)) {
+        try { fs.unlinkSync(fp); heuristicRemoved++; } catch {}
+        continue;
+      }
       parsed.push(parseSkillFile(fp, '', { raw }));
       fileMap.push(fp);
     } catch {}
   }
 
-  if (parsed.length === 0) return;
+  if (heuristicRemoved > 0) {
+    console.log(`Removed ${heuristicRemoved} files by content heuristic`);
+  }
+
+  if (parsed.length === 0) {
+    if (heuristicRemoved > 0) removeEmptyDirs(dest);
+    return;
+  }
 
   const filtered = await filterWithClassifier(parsed);
   const keptPaths = new Set(filtered.map(s => s.path));
 
-  let removed = 0;
+  let classifierRemoved = 0;
   for (const fp of fileMap) {
     if (!keptPaths.has(fp)) {
-      try { fs.unlinkSync(fp); removed++; } catch {}
+      try { fs.unlinkSync(fp); classifierRemoved++; } catch {}
     }
   }
 
-  if (removed > 0) {
-    console.log(`Removed ${removed} files classified as non-skills`);
+  const totalRemoved = heuristicRemoved + classifierRemoved;
+  if (totalRemoved > 0) {
+    console.log(`Removed ${totalRemoved} non-skill files (heuristic: ${heuristicRemoved}, classifier: ${classifierRemoved})`);
     removeEmptyDirs(dest);
   }
 }
@@ -566,6 +580,17 @@ export async function importFromGitHub(repoUrl) {
 
   await classifierCleanup(dest);
 
+  // Update skill count cache with real survivor count
+  const realCount = globSync(`${dest}/**/*.md`).length;
+  const cacheKey = url.replace(/\.git$/, '');
+  const cachePath = path.join(PROMPTGRAPH_DIR, 'skill-counts.json');
+  try {
+    fs.mkdirSync(path.dirname(cachePath), { recursive: true });
+    const cache = JSON.parse(fs.readFileSync(cachePath, 'utf8') || '{}');
+    cache[cacheKey] = { count: realCount, ts: Date.now() };
+    fs.writeFileSync(cachePath, JSON.stringify(cache, null, 2));
+  } catch {}
+
   const { dir: localDir, label: localLabel } = detectSkillsDirLocal(dest);
   // Prefer the known skillsSubdir (from API detection or sparse patterns) as the
   // canonical skills directory — it's more accurate than local heuristics for

package/marketplace.js

@@ -262,6 +262,15 @@ async function countRepoSkills(repoUrl) {
   } catch { return null; }
 }
 
+// Count real .md files on disk for an installed bundle (always correct)
+function localSkillCount(repoUrl) {
+  const repoName = repoUrl.replace(/^https?:\/\/github\.com\//, '').replace(/\.git$/, '').replace('/', '-');
+  const dest = path.join(SKILLS_STORE_DIR, 'github', repoName);
+  if (!fs.existsSync(dest)) return null;
+  const files = globSync(`${dest}/**/*.md`);
+  return files.length;
+}
+
 export async function browseBundles(topK = 20) {
   try {
     const text = await fetchText(REGISTRY_URL);
@@ -273,20 +282,33 @@ export async function browseBundles(topK = 20) {
 
     await Promise.all(bundles.map(async b => {
       if (!b.repo_url) return;
+
+      // 1. If installed locally — count real files on disk (always correct)
+      const local = localSkillCount(b.repo_url);
+      if (local !== null) {
+        if (b.skillCount !== local) {
+          b.skillCount = local;
+          cache[b.repo_url] = { count: local, ts: now };
+          changed = true;
+        }
+        return;
+      }
+
+      // 2. Not installed — use cached API count if fresh
       const cached = cache[b.repo_url];
-      // Use cached count if fresh, else fetch from API
       if (cached && (now - cached.ts) < SKILL_COUNT_TTL) {
         b.skillCount = cached.count;
+        return;
+      }
+
+      // 3. Fetch from GitHub API
+      const count = await countRepoSkills(b.repo_url);
+      if (count !== null) {
+        b.skillCount = count;
+        cache[b.repo_url] = { count, ts: now };
+        changed = true;
       } else {
-        const count = await countRepoSkills(b.repo_url);
-        if (count !== null) {
-          b.skillCount = count;
-          cache[b.repo_url] = { count, ts: now };
-          changed = true;
-        } else {
-          // API failed — use stale cache if exists, else keep registry value as fallback
-          b.skillCount = cached?.count ?? b.skillCount ?? 0;
-        }
+        b.skillCount = cached?.count ?? b.skillCount ?? 0;
       }
     }));
 

package/package.json

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