promptgraph-mcp 1.5.0 → 1.5.2

package/README.md

@@ -19,7 +19,7 @@ Claude Code loads all `.md` files from `~/.claude/commands/` into the system pro
   ...            ← 40+ skills, NOT loaded into context
 ```
 
-When you ask Claude a question, it calls `pg_search("your task")` → finds the right skill via vector search → reads only that file. **One skill loaded instead of forty.**
+When you ask Claude a question, it calls `pg_search("your task")` → finds the right skill via vector search → returns the path. Claude then reads only that file instead of having all skills preloaded in context.
 
 ## Features
 

package/cli.js

@@ -1,6 +1,7 @@
 import chalk from 'chalk';
 import ora from 'ora';
 import boxen from 'boxen';
+import fs from 'fs';
 
 export const colors = {
   primary: chalk.hex('#7C3AED'),
@@ -12,10 +13,17 @@ export const colors = {
   bold: chalk.bold,
 };
 
+function getVersion() {
+  try {
+    const pkg = JSON.parse(fs.readFileSync(new URL('./package.json', import.meta.url), 'utf8'));
+    return pkg.version;
+  } catch { return ''; }
+}
+
 export function banner() {
   console.log(
     boxen(
-      colors.primary.bold('PromptGraph') + '  ' + colors.muted('v' + (await getVersion())) + '\n' +
+      colors.primary.bold('PromptGraph') + '  ' + colors.muted('v' + getVersion()) + '\n' +
       colors.muted('Semantic skill router for Claude Code'),
       {
         padding: { top: 0, bottom: 0, left: 2, right: 2 },
@@ -27,14 +35,6 @@ export function banner() {
   );
 }
 
-async function getVersion() {
-  try {
-    const { createRequire } = await import('module');
-    const require = createRequire(import.meta.url);
-    return require('./package.json').version;
-  } catch { return ''; }
-}
-
 export function spinner(text) {
   return ora({
     text: colors.muted(text),

package/db.js

@@ -5,9 +5,13 @@ import fs from 'fs';
 
 const DB_PATH = path.join(os.homedir(), '.claude', '.promptgraph', 'promptgraph.db');
 
+let _db = null;
+
 export function getDb() {
+  if (_db) return _db;
   fs.mkdirSync(path.dirname(DB_PATH), { recursive: true });
   const db = new Database(DB_PATH);
+  _db = db;
   db.pragma('journal_mode = WAL');
 
   db.exec(`

package/indexer.js

@@ -1,7 +1,7 @@
 import { globSync } from 'glob';
 import { createHash } from 'crypto';
 import fs from 'fs';
-import { parseSkillFile } from './parser.js';
+import { parseSkillFile, isSkillFile } from './parser.js';
 import { embedBatch, BATCH_SIZE } from './embedder.js';
 import { getDb, skillId } from './db.js';
 import { loadConfig } from './config.js';
@@ -51,8 +51,10 @@ async function indexBatch(db, skills) {
       upsertSkill.run({ id, name: skill.name, description: skill.description, path: skill.path, source: skill.source, content: skill.content, hash: skill.hash || null });
       deleteChunks.run(id);
       deleteEdges.run(id);
-      for (const called of skill.calls) {
-        upsertEdge.run(id, called);
+      for (const calledName of skill.calls) {
+        // try to resolve to a real skill id, fallback to bare name
+        const resolved = db.prepare("SELECT id FROM skills WHERE name = ? ORDER BY id LIMIT 1").get(calledName);
+        upsertEdge.run(id, resolved ? resolved.id : calledName);
       }
     }
     for (let i = 0; i < allChunks.length; i++) {
@@ -88,6 +90,7 @@ export async function indexAll() {
   let skipped = 0;
   for (const { file, source } of allFiles) {
     try {
+      if (!isSkillFile(file)) { skipped++; count++; continue; }
       const hash = fileHash(file);
       const parsed = parseSkillFile(file, source);
       const id = skillId(source, parsed.name);

package/package.json

@@ -1,9 +1,10 @@
 {
   "name": "promptgraph-mcp",
-  "version": "1.5.0",
+  "version": "1.5.2",
   "main": "index.js",
   "type": "module",
   "bin": {
+    "pg": "./index.js",
     "promptgraph": "./index.js",
     "promptgraph-mcp": "./index.js"
   },

package/parser.js

@@ -2,9 +2,18 @@ import fs from 'fs';
 import path from 'path';
 import matter from 'gray-matter';
 
-const SKILL_REF_RE = /\/([a-z0-9][a-z0-9-]+)/g;
+// match /skill-name but not URLs (http://, https://, etc.)
+const SKILL_REF_RE = /(?<!https?:|ftp:)(?<![a-zA-Z0-9])\/([a-z0-9][a-z0-9-]{2,})/g;
 
-export function parseSkillFile(filePath, source) {
+// files that are likely not skills
+const SKIP_FILENAMES = new Set(['readme', 'changelog', 'license', 'contributing', 'code-of-conduct', 'security', 'authors', 'credits']);
+
+export function isSkillFile(filePath) {
+  const base = filePath.split(/[\\/]/).pop().replace(/\.md$/i, '').toLowerCase();
+  return !SKIP_FILENAMES.has(base);
+}
+
+export function parseSkillFile(filePath, source, opts = {}) {
   const raw = fs.readFileSync(filePath, 'utf8');
 
   let name, description, content;

package/pg-hook.js

@@ -2,7 +2,6 @@
 import { embed, cosineSimilarity } from './embedder.js';
 import { getDb } from './db.js';
 
-const chunks = [];
 let input = '';
 process.stdin.on('data', d => input += d);
 process.stdin.on('end', async () => {
@@ -13,18 +12,27 @@ process.stdin.on('end', async () => {
 
     const queryVec = await embed(prompt);
     const db = getDb();
-    const skills = db.prepare('SELECT name, description, path, embedding FROM skills').all();
-
-    const results = skills
-      .map(s => ({
-        name: s.name,
-        description: s.description,
-        path: s.path,
-        score: cosineSimilarity(queryVec, JSON.parse(s.embedding)),
-      }))
-      .sort((a, b) => b.score - a.score)
+
+    // 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);
+    }
+
+    const topIds = [...bestBySkill.entries()]
+      .sort((a, b) => b[1] - a[1])
       .slice(0, 3)
-      .filter(s => s.score > 0.55);
+      .filter(([, score]) => score > 0.55);
+
+    if (topIds.length === 0) process.exit(0);
+
+    const results = topIds.map(([id, score]) => {
+      const skill = db.prepare('SELECT name, description, path FROM skills WHERE id = ?').get(id);
+      return skill ? { ...skill, score } : null;
+    }).filter(Boolean);
 
     if (results.length === 0) process.exit(0);
 

package/search.js

@@ -45,9 +45,10 @@ export async function search(query, topK = 5) {
   return [...bestBySkill.entries()]
     .sort((a, b) => b[1] - a[1])
     .slice(0, topK)
+    // apply same rating boost as ANN path
     .map(([id, score]) => {
       const skill = db.prepare('SELECT id, name, description, path, source FROM skills WHERE id = ?').get(id);
-      return skill ? { ...skill, score } : null;
+      return skill ? { ...skill, score: applyRatingBoost(db, id, score) } : null;
     })
     .filter(Boolean);
 }

package/watcher.js

@@ -1,50 +1,61 @@
-import chokidar from 'chokidar';
-import path from 'path';
-import os from 'os';
-import { indexFile } from './indexer.js';
-import { getDb } from './db.js';
-
-const SOURCES = [
-  { dir: path.join(os.homedir(), '.claude', 'skills-store'), source: 'skills-store' },
-  { dir: path.join(os.homedir(), '.claude', 'skills'), source: 'skills' },
-];
-
-export function startWatcher() {
-  const paths = SOURCES.map(s => s.dir);
-
-  const watcher = chokidar.watch(paths, {
-    ignored: /[/\\]\./,
-    persistent: true,
-    ignoreInitial: true,
-    awaitWriteFinish: { stabilityThreshold: 500 },
-  });
-
-  watcher.on('add', filePath => reindex(filePath));
-  watcher.on('change', filePath => reindex(filePath));
-  watcher.on('unlink', filePath => {
-    const name = path.basename(filePath, '.md');
-    const db = getDb();
-    db.prepare('DELETE FROM skills WHERE name = ?').run(name);
-    db.prepare('DELETE FROM edges WHERE from_skill = ? OR to_skill = ?').run(name, name);
-    console.error(`[PromptGraph] Removed: ${name}`);
-  });
-
-  console.error('[PromptGraph] Watcher started');
-}
-
-function getSource(filePath) {
-  for (const { dir, source } of SOURCES) {
-    if (filePath.startsWith(dir)) return source;
-  }
-  return 'unknown';
-}
-
-async function reindex(filePath) {
-  if (!filePath.endsWith('.md')) return;
-  try {
-    await indexFile(filePath, getSource(filePath));
-    console.error(`[PromptGraph] Reindexed: ${path.basename(filePath)}`);
-  } catch (e) {
-    console.error(`[PromptGraph] Error reindexing ${filePath}: ${e.message}`);
-  }
-}
+import chokidar from 'chokidar';
+import path from 'path';
+import os from 'os';
+import { indexFile } from './indexer.js';
+import { getDb, skillId } from './db.js';
+import { parseSkillFile } from './parser.js';
+
+const SOURCES = [
+  { dir: path.join(os.homedir(), '.claude', 'skills-store'), source: 'skills-store' },
+  { dir: path.join(os.homedir(), '.claude', 'skills'), source: 'skills' },
+];
+
+export function startWatcher() {
+  const paths = SOURCES.map(s => s.dir);
+
+  const watcher = chokidar.watch(paths, {
+    ignored: /[/\\]\./,
+    persistent: true,
+    ignoreInitial: true,
+    awaitWriteFinish: { stabilityThreshold: 500 },
+  });
+
+  watcher.on('add', filePath => reindex(filePath));
+  watcher.on('change', filePath => reindex(filePath));
+  watcher.on('unlink', filePath => remove(filePath));
+
+  console.error('[PromptGraph] Watcher started');
+}
+
+function getSource(filePath) {
+  for (const { dir, source } of SOURCES) {
+    if (filePath.startsWith(dir)) return source;
+  }
+  return 'unknown';
+}
+
+function remove(filePath) {
+  if (!filePath.endsWith('.md')) return;
+  try {
+    const source = getSource(filePath);
+    const name = path.basename(filePath, '.md');
+    const id = skillId(source, name);
+    const db = getDb();
+    db.prepare('DELETE FROM skills WHERE id = ?').run(id);
+    db.prepare('DELETE FROM chunks WHERE skill_id = ?').run(id);
+    db.prepare('DELETE FROM edges WHERE from_skill = ? OR to_skill = ?').run(id, id);
+    console.error(`[PromptGraph] Removed: ${id}`);
+  } catch (e) {
+    console.error(`[PromptGraph] Error removing ${filePath}: ${e.message}`);
+  }
+}
+
+async function reindex(filePath) {
+  if (!filePath.endsWith('.md')) return;
+  try {
+    await indexFile(filePath, getSource(filePath));
+    console.error(`[PromptGraph] Reindexed: ${path.basename(filePath)}`);
+  } catch (e) {
+    console.error(`[PromptGraph] Error reindexing ${filePath}: ${e.message}`);
+  }
+}