promptgraph-mcp 1.5.25 → 1.5.26

package/index.js

@@ -12,7 +12,7 @@ const args = process.argv.slice(2);
 const rawBin = process.argv[1]?.split(/[\\/]/).pop()?.replace(/\.js$/, '');
 const bin = (rawBin && rawBin !== 'index') ? rawBin : 'pg';
 
-const KNOWN_COMMANDS = new Set(['init', 'reindex', 'import', 'setup', 'validate', 'marketplace', 'doctor', 'help', '--help', '-h']);
+const KNOWN_COMMANDS = new Set(['init', 'reindex', 'import', 'setup', 'validate', 'marketplace', 'doctor', 'search', 'help', '--help', '-h']);
 
 function showHelp() {
   console.log(
@@ -26,6 +26,7 @@ function showHelp() {
   const cmds = [
     ['init',                'First-time setup + index all skills'],
     ['reindex',             'Re-index all skills'],
+    ['search <query>',      'Search skills from the terminal'],
     ['import <owner/repo>', 'Import skills from GitHub'],
     ['marketplace [page]',  'Browse the community skill registry'],
     ['validate <file.md>',  'Validate a skill before publishing'],
@@ -213,6 +214,27 @@ if (args[0] === 'validate') {
   }
 }
 
+if (args[0] === 'search') {
+  const query = args.slice(1).join(' ');
+  if (!query) { error('Usage: ' + bin + ' search <query>'); process.exit(1); }
+  const { search: searchSkills } = await import('./search.js');
+  const spin = (await import('./cli.js')).spinner('Searching...');
+  spin.start();
+  const results = await searchSkills(query, 10);
+  spin.stop();
+  if (!results.length) { info('No results for: ' + query); process.exit(0); }
+  const purple = chalk.hex('#7C3AED');
+  console.log();
+  results.forEach((s, i) => {
+    const score = chalk.dim((s.score * 100).toFixed(0) + '%');
+    console.log('  ' + chalk.dim(String(i + 1) + '.') + ' ' + chalk.bold.white(s.name) + '  ' + score);
+    console.log('     ' + chalk.dim(s.description || ''));
+    console.log('     ' + purple(s.source) + '  ' + chalk.dim(s.path));
+    console.log();
+  });
+  process.exit(0);
+}
+
 if (args[0] === 'import') {
   const { importFromGitHub } = await import('./github-import.js');
   await importFromGitHub(args[1]);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "promptgraph-mcp",
-  "version": "1.5.25",
+  "version": "1.5.26",
   "main": "index.js",
   "type": "module",
   "bin": {

package/pg-hook.js

@@ -1,6 +1,8 @@
 #!/usr/bin/env node
-import { embed, cosineSimilarity } from './embedder.js';
+import { embed } from './embedder.js';
 import { getDb } from './db.js';
+import { annSearch } from './ann.js';
+import { cosineSimilarity } from './embedder.js';
 
 let input = '';
 process.stdin.on('data', d => input += d);
@@ -13,19 +15,34 @@ process.stdin.on('end', async () => {
     const queryVec = await embed(prompt);
     const db = getDb();
 
-    // search over chunks, deduplicate by skill
-    const chunks = db.prepare('SELECT skill_id, embedding FROM chunks').all();
-    const bestBySkill = new Map();
-    for (const chunk of chunks) {
-      const score = cosineSimilarity(queryVec, JSON.parse(chunk.embedding));
-      const prev = bestBySkill.get(chunk.skill_id);
-      if (!prev || score > prev) bestBySkill.set(chunk.skill_id, score);
-    }
+    // Use ANN index (fast). Fall back to brute-force only if ANN unavailable.
+    let topIds;
+    const annResults = await annSearch(queryVec, 15);
 
-    const topIds = [...bestBySkill.entries()]
-      .sort((a, b) => b[1] - a[1])
-      .slice(0, 3)
-      .filter(([, score]) => score > 0.55);
+    if (annResults && annResults.length > 0) {
+      const bestBySkill = new Map();
+      for (const r of annResults) {
+        const prev = bestBySkill.get(r.skill_id);
+        if (!prev || r.score > prev) bestBySkill.set(r.skill_id, r.score);
+      }
+      topIds = [...bestBySkill.entries()]
+        .sort((a, b) => b[1] - a[1])
+        .slice(0, 3)
+        .filter(([, score]) => score > 0.55);
+    } else {
+      // Fallback: brute-force (only before first reindex)
+      const chunks = db.prepare('SELECT skill_id, embedding FROM chunks').all();
+      const bestBySkill = new Map();
+      for (const chunk of chunks) {
+        const score = cosineSimilarity(queryVec, JSON.parse(chunk.embedding));
+        const prev = bestBySkill.get(chunk.skill_id);
+        if (!prev || score > prev) bestBySkill.set(chunk.skill_id, score);
+      }
+      topIds = [...bestBySkill.entries()]
+        .sort((a, b) => b[1] - a[1])
+        .slice(0, 3)
+        .filter(([, score]) => score > 0.55);
+    }
 
     if (topIds.length === 0) process.exit(0);
 

package/search.js

@@ -108,7 +108,7 @@ export function getImpact(nameOrId) {
     const callers = db.prepare('SELECT from_skill FROM edges WHERE to_skill = ?').all(cur).map(r => r.from_skill);
     queue.push(...callers);
   }
-  visited.delete(id);
+  visited.delete(res.id);
   return [...visited];
 }
 

package/watcher.js

@@ -25,8 +25,11 @@ export function startWatcher() {
 }
 
 function getSource(filePath, config) {
-  for (const { dir, source } of config.sources) {
-    if (filePath.startsWith(dir)) return source;
+  const normFile = path.resolve(filePath);
+  // Sort longest-first so skills-store/marketplace wins over skills-store
+  const sorted = [...config.sources].sort((a, b) => b.dir.length - a.dir.length);
+  for (const { dir, source } of sorted) {
+    if (normFile.startsWith(path.resolve(dir))) return source;
   }
   return 'unknown';
 }