promptgraph-mcp 1.0.2 → 1.0.3

package/index.js

@@ -1,137 +1,137 @@
-#!/usr/bin/env node
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
-import { search, getContext, getCallers, getCallees, getImpact, listAll } from './search.js';
-import { indexAll } from './indexer.js';
-import { startWatcher } from './watcher.js';
-import { promptConfig } from './config.js';
-
-const args = process.argv.slice(2);
-
-if (args[0] === 'help' || args[0] === '--help' || args[0] === '-h') {
-  console.log(`
-PromptGraph — semantic skill router for Claude Code
-
-Usage:
-  promptgraph-mcp init      First-time setup + index all skills
-  promptgraph-mcp reindex   Re-index all skills
-  promptgraph-mcp           Start MCP server (used by Claude)
-  promptgraph-mcp help      Show this help
-
-GitHub: https://github.com/NeiP4n/promptgraph
-npm:    https://npmjs.com/package/promptgraph-mcp
-  `);
-  process.exit(0);
-}
-
-if (args[0] === 'init') {
-  const config = await promptConfig();
-  console.log('\nIndexing skills...');
-  await indexAll();
-  console.log('\nDone! Add to Claude Code settings.json:');
-  console.log(JSON.stringify({
-    mcpServers: {
-      promptgraph: {
-        command: 'npx',
-        args: ['promptgraph-mcp'],
-      }
-    }
-  }, null, 2));
-  process.exit(0);
-}
-
-if (args[0] === 'reindex') {
-  console.log('[PromptGraph] Reindexing...');
-  await indexAll();
-  process.exit(0);
-}
-
-const server = new Server(
-  { name: 'promptgraph', version: '1.0.0' },
-  { capabilities: { tools: {} } }
-);
-
-server.setRequestHandler(ListToolsRequestSchema, async () => ({
-  tools: [
-    {
-      name: 'pg_search',
-      description: 'Search skills by task description. Returns top relevant skills with scores.',
-      inputSchema: {
-        type: 'object',
-        properties: {
-          query: { type: 'string', description: 'Task or topic to search for' },
-          top_k: { type: 'number', description: 'Number of results (default 5)' },
-        },
-        required: ['query'],
-      },
-    },
-    {
-      name: 'pg_list',
-      description: 'List all indexed skills.',
-      inputSchema: { type: 'object', properties: {} },
-    },
-    {
-      name: 'pg_context',
-      description: 'Get full context for a skill: description, content, callers, callees.',
-      inputSchema: {
-        type: 'object',
-        properties: { name: { type: 'string' } },
-        required: ['name'],
-      },
-    },
-    {
-      name: 'pg_callers',
-      description: 'Get skills that call/reference this skill.',
-      inputSchema: {
-        type: 'object',
-        properties: { name: { type: 'string' } },
-        required: ['name'],
-      },
-    },
-    {
-      name: 'pg_callees',
-      description: 'Get skills that this skill calls/references.',
-      inputSchema: {
-        type: 'object',
-        properties: { name: { type: 'string' } },
-        required: ['name'],
-      },
-    },
-    {
-      name: 'pg_impact',
-      description: 'Get all skills that would be affected if this skill changes.',
-      inputSchema: {
-        type: 'object',
-        properties: { name: { type: 'string' } },
-        required: ['name'],
-      },
-    },
-  ],
-}));
-
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-  const { name, arguments: args } = request.params;
-
-  try {
-    let result;
-    switch (name) {
-      case 'pg_search': result = await search(args.query, args.top_k || 5); break;
-      case 'pg_list': result = listAll(); break;
-      case 'pg_context': result = getContext(args.name); break;
-      case 'pg_callers': result = getCallers(args.name); break;
-      case 'pg_callees': result = getCallees(args.name); break;
-      case 'pg_impact': result = getImpact(args.name); break;
-      default: throw new Error(`Unknown tool: ${name}`);
-    }
-    return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
-  } catch (e) {
-    return { content: [{ type: 'text', text: `Error: ${e.message}` }], isError: true };
-  }
-});
-
-startWatcher();
-
-const transport = new StdioServerTransport();
-await server.connect(transport);
-console.error('[PromptGraph] MCP server running');
+#!/usr/bin/env node
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { search, getContext, getCallers, getCallees, getImpact, listAll } from './search.js';
+import { indexAll } from './indexer.js';
+import { startWatcher } from './watcher.js';
+import { promptConfig } from './config.js';
+
+const args = process.argv.slice(2);
+
+if (args[0] === 'help' || args[0] === '--help' || args[0] === '-h') {
+  console.log(`
+PromptGraph — semantic skill router for Claude Code
+
+Usage:
+  promptgraph-mcp init      First-time setup + index all skills
+  promptgraph-mcp reindex   Re-index all skills
+  promptgraph-mcp           Start MCP server (used by Claude)
+  promptgraph-mcp help      Show this help
+
+GitHub: https://github.com/NeiP4n/promptgraph
+npm:    https://npmjs.com/package/promptgraph-mcp
+  `);
+  process.exit(0);
+}
+
+if (args[0] === 'init') {
+  const config = await promptConfig();
+  console.log('\nIndexing skills...');
+  await indexAll();
+  console.log('\nDone! Add to Claude Code settings.json:');
+  console.log(JSON.stringify({
+    mcpServers: {
+      promptgraph: {
+        command: 'npx',
+        args: ['promptgraph-mcp'],
+      }
+    }
+  }, null, 2));
+  process.exit(0);
+}
+
+if (args[0] === 'reindex') {
+  console.log('[PromptGraph] Reindexing...');
+  await indexAll();
+  process.exit(0);
+}
+
+const server = new Server(
+  { name: 'promptgraph', version: '1.0.0' },
+  { capabilities: { tools: {} } }
+);
+
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+  tools: [
+    {
+      name: 'pg_search',
+      description: 'Search skills by task description. Returns top relevant skills with scores.',
+      inputSchema: {
+        type: 'object',
+        properties: {
+          query: { type: 'string', description: 'Task or topic to search for' },
+          top_k: { type: 'number', description: 'Number of results (default 5)' },
+        },
+        required: ['query'],
+      },
+    },
+    {
+      name: 'pg_list',
+      description: 'List all indexed skills.',
+      inputSchema: { type: 'object', properties: {} },
+    },
+    {
+      name: 'pg_context',
+      description: 'Get full context for a skill: description, content, callers, callees.',
+      inputSchema: {
+        type: 'object',
+        properties: { name: { type: 'string' } },
+        required: ['name'],
+      },
+    },
+    {
+      name: 'pg_callers',
+      description: 'Get skills that call/reference this skill.',
+      inputSchema: {
+        type: 'object',
+        properties: { name: { type: 'string' } },
+        required: ['name'],
+      },
+    },
+    {
+      name: 'pg_callees',
+      description: 'Get skills that this skill calls/references.',
+      inputSchema: {
+        type: 'object',
+        properties: { name: { type: 'string' } },
+        required: ['name'],
+      },
+    },
+    {
+      name: 'pg_impact',
+      description: 'Get all skills that would be affected if this skill changes.',
+      inputSchema: {
+        type: 'object',
+        properties: { name: { type: 'string' } },
+        required: ['name'],
+      },
+    },
+  ],
+}));
+
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { name, arguments: args } = request.params;
+
+  try {
+    let result;
+    switch (name) {
+      case 'pg_search': result = await search(args.query, args.top_k || 5); break;
+      case 'pg_list': result = listAll(); break;
+      case 'pg_context': result = getContext(args.name); break;
+      case 'pg_callers': result = getCallers(args.name); break;
+      case 'pg_callees': result = getCallees(args.name); break;
+      case 'pg_impact': result = getImpact(args.name); break;
+      default: throw new Error(`Unknown tool: ${name}`);
+    }
+    return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
+  } catch (e) {
+    return { content: [{ type: 'text', text: `Error: ${e.message}` }], isError: true };
+  }
+});
+
+startWatcher();
+
+const transport = new StdioServerTransport();
+await server.connect(transport);
+console.error('[PromptGraph] MCP server running');

package/indexer.js

@@ -1,65 +1,65 @@
-import { globSync } from 'glob';
-import { parseSkillFile } from './parser.js';
-import { embed } from './embedder.js';
-import { getDb } from './db.js';
-import { loadConfig } from './config.js';
-
-export async function indexAll() {
-  const config = loadConfig();
-  const db = getDb();
-
-  const upsert = db.prepare(`
-    INSERT INTO skills (name, description, path, source, content, embedding)
-    VALUES (@name, @description, @path, @source, @content, @embedding)
-    ON CONFLICT(name) DO UPDATE SET
-      description = excluded.description,
-      path = excluded.path,
-      source = excluded.source,
-      content = excluded.content,
-      embedding = excluded.embedding
-  `);
-  const upsertEdge = db.prepare(`INSERT OR IGNORE INTO edges (from_skill, to_skill) VALUES (?, ?)`);
-  db.prepare('DELETE FROM edges').run();
-
-  let count = 0;
-  for (const { dir, source } of config.sources) {
-    const files = globSync(`${dir}/**/*.md`);
-    for (const file of files) {
-      try {
-        const skill = parseSkillFile(file, source);
-        const embedding = await embed(skill.name + ' ' + skill.description + ' ' + skill.content.slice(0, 500));
-        upsert.run({ ...skill, embedding: JSON.stringify(embedding) });
-        for (const called of skill.calls) {
-          upsertEdge.run(skill.name, called);
-        }
-        count++;
-        process.stdout.write(`\r  Indexed: ${count} skills`);
-      } catch (e) {
-        console.error(`\n  Error indexing ${file}: ${e.message}`);
-      }
-    }
-  }
-  console.log(`\n  Done. ${count} skills indexed.`);
-}
-
-export async function indexFile(filePath, source) {
-  const db = getDb();
-  const skill = parseSkillFile(filePath, source);
-  const embedding = await embed(skill.name + ' ' + skill.description + ' ' + skill.content.slice(0, 500));
-  db.prepare(`
-    INSERT INTO skills (name, description, path, source, content, embedding)
-    VALUES (@name, @description, @path, @source, @content, @embedding)
-    ON CONFLICT(name) DO UPDATE SET
-      description = excluded.description,
-      path = excluded.path,
-      source = excluded.source,
-      content = excluded.content,
-      embedding = excluded.embedding
-  `).run({ ...skill, embedding: JSON.stringify(embedding) });
-
-  db.prepare('DELETE FROM edges WHERE from_skill = ?').run(skill.name);
-  const upsertEdge = db.prepare('INSERT OR IGNORE INTO edges (from_skill, to_skill) VALUES (?, ?)');
-  for (const called of skill.calls) {
-    upsertEdge.run(skill.name, called);
-  }
-}
+import { globSync } from 'glob';
+import { parseSkillFile } from './parser.js';
+import { embed } from './embedder.js';
+import { getDb } from './db.js';
+import { loadConfig } from './config.js';
+
+export async function indexAll() {
+  const config = loadConfig();
+  const db = getDb();
+
+  const upsert = db.prepare(`
+    INSERT INTO skills (name, description, path, source, content, embedding)
+    VALUES (@name, @description, @path, @source, @content, @embedding)
+    ON CONFLICT(name) DO UPDATE SET
+      description = excluded.description,
+      path = excluded.path,
+      source = excluded.source,
+      content = excluded.content,
+      embedding = excluded.embedding
+  `);
+  const upsertEdge = db.prepare(`INSERT OR IGNORE INTO edges (from_skill, to_skill) VALUES (?, ?)`);
+  db.prepare('DELETE FROM edges').run();
+
+  let count = 0;
+  for (const { dir, source } of config.sources) {
+    const files = globSync(`${dir}/**/*.md`);
+    for (const file of files) {
+      try {
+        const skill = parseSkillFile(file, source);
+        const embedding = await embed(skill.name + ' ' + skill.description + ' ' + skill.content.slice(0, 500));
+        upsert.run({ ...skill, embedding: JSON.stringify(embedding) });
+        for (const called of skill.calls) {
+          upsertEdge.run(skill.name, called);
+        }
+        count++;
+        process.stdout.write(`\r  Indexed: ${count} skills`);
+      } catch (e) {
+        console.error(`\n  Error indexing ${file}: ${e.message}`);
+      }
+    }
+  }
+  console.log(`\n  Done. ${count} skills indexed.`);
+}
+
+export async function indexFile(filePath, source) {
+  const db = getDb();
+  const skill = parseSkillFile(filePath, source);
+  const embedding = await embed(skill.name + ' ' + skill.description + ' ' + skill.content.slice(0, 500));
+  db.prepare(`
+    INSERT INTO skills (name, description, path, source, content, embedding)
+    VALUES (@name, @description, @path, @source, @content, @embedding)
+    ON CONFLICT(name) DO UPDATE SET
+      description = excluded.description,
+      path = excluded.path,
+      source = excluded.source,
+      content = excluded.content,
+      embedding = excluded.embedding
+  `).run({ ...skill, embedding: JSON.stringify(embedding) });
+
+  db.prepare('DELETE FROM edges WHERE from_skill = ?').run(skill.name);
+  const upsertEdge = db.prepare('INSERT OR IGNORE INTO edges (from_skill, to_skill) VALUES (?, ?)');
+  for (const called of skill.calls) {
+    upsertEdge.run(skill.name, called);
+  }
+}

package/package.json

@@ -1,42 +1,42 @@
-{
-  "name": "promptgraph-mcp",
-  "version": "1.0.2",
-  "main": "index.js",
-  "type": "module",
-  "bin": {
-    "promptgraph": "./index.js",
-    "promptgraph-mcp": "./index.js"
-  },
-  "scripts": {
-    "start": "node index.js",
-    "init": "node index.js init",
-    "reindex": "node index.js reindex"
-  },
-  "keywords": [
-    "claude",
-    "claude-code",
-    "mcp",
-    "ai",
-    "skills",
-    "embeddings"
-  ],
-  "author": "NeiP4n",
-  "license": "MIT",
-  "description": "Semantic skill router for Claude Code — load only the skills you need, save 20k+ tokens per session",
-  "repository": {
-    "type": "git",
-    "url": "https://github.com/NeiP4n/promptgraph.git"
-  },
-  "homepage": "https://github.com/NeiP4n/promptgraph#readme",
-  "engines": {
-    "node": ">=18"
-  },
-  "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.29.0",
-    "better-sqlite3": "^12.10.0",
-    "chokidar": "^5.0.0",
-    "fastembed": "^2.1.0",
-    "glob": "^13.0.6",
-    "gray-matter": "^4.0.3"
-  }
-}
+{
+  "name": "promptgraph-mcp",
+  "version": "1.0.3",
+  "main": "index.js",
+  "type": "module",
+  "bin": {
+    "promptgraph": "./index.js",
+    "promptgraph-mcp": "./index.js"
+  },
+  "scripts": {
+    "start": "node index.js",
+    "init": "node index.js init",
+    "reindex": "node index.js reindex"
+  },
+  "keywords": [
+    "claude",
+    "claude-code",
+    "mcp",
+    "ai",
+    "skills",
+    "embeddings"
+  ],
+  "author": "NeiP4n",
+  "license": "MIT",
+  "description": "Semantic skill router for Claude Code — load only the skills you need, save 20k+ tokens per session",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/NeiP4n/promptgraph.git"
+  },
+  "homepage": "https://github.com/NeiP4n/promptgraph#readme",
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.29.0",
+    "better-sqlite3": "^12.10.0",
+    "chokidar": "^5.0.0",
+    "fastembed": "^2.1.0",
+    "glob": "^13.0.6",
+    "gray-matter": "^4.0.3"
+  }
+}

package/parser.js

@@ -1,47 +1,47 @@
-import fs from 'fs';
-import path from 'path';
-
-const SKILL_REF_RE = /\/([a-z0-9][a-z0-9-]+)/g;
-const FRONTMATTER_RE = /^---\n([\s\S]*?)\n---/;
-
-export function parseSkillFile(filePath, source) {
-  const raw = fs.readFileSync(filePath, 'utf8');
-
-  let name = null;
-  let description = null;
-  let content = raw;
-
-  const fm = raw.match(FRONTMATTER_RE);
-  if (fm) {
-    try {
-      const nameMatch = fm[1].match(/^name:\s*(.+)$/m);
-      const descMatch = fm[1].match(/^description:\s*(.+)$/m);
-      if (nameMatch) name = nameMatch[1].trim().replace(/^["']|["']$/g, '');
-      if (descMatch) description = descMatch[1].trim().replace(/^["']|["']$/g, '');
-    } catch {}
-  }
-
-  name = name || path.basename(filePath, '.md');
-  description = description || extractFirstParagraph(raw);
-
-  const calls = new Set();
-  for (const match of raw.matchAll(SKILL_REF_RE)) {
-    const ref = match[1];
-    if (ref !== name && ref.length > 2) calls.add(ref);
-  }
-
-  return {
-    name,
-    description,
-    path: filePath,
-    source,
-    content: raw,
-    calls: [...calls],
-  };
-}
-
-function extractFirstParagraph(content) {
-  const lines = content.replace(FRONTMATTER_RE, '').split('\n')
-    .filter(l => l.trim() && !l.startsWith('#') && !l.startsWith('---'));
-  return lines[0]?.trim().slice(0, 200) || '';
-}
+import fs from 'fs';
+import path from 'path';
+
+const SKILL_REF_RE = /\/([a-z0-9][a-z0-9-]+)/g;
+const FRONTMATTER_RE = /^---\n([\s\S]*?)\n---/;
+
+export function parseSkillFile(filePath, source) {
+  const raw = fs.readFileSync(filePath, 'utf8');
+
+  let name = null;
+  let description = null;
+  let content = raw;
+
+  const fm = raw.match(FRONTMATTER_RE);
+  if (fm) {
+    try {
+      const nameMatch = fm[1].match(/^name:\s*(.+)$/m);
+      const descMatch = fm[1].match(/^description:\s*(.+)$/m);
+      if (nameMatch) name = nameMatch[1].trim().replace(/^["']|["']$/g, '');
+      if (descMatch) description = descMatch[1].trim().replace(/^["']|["']$/g, '');
+    } catch {}
+  }
+
+  name = name || path.basename(filePath, '.md');
+  description = description || extractFirstParagraph(raw);
+
+  const calls = new Set();
+  for (const match of raw.matchAll(SKILL_REF_RE)) {
+    const ref = match[1];
+    if (ref !== name && ref.length > 2) calls.add(ref);
+  }
+
+  return {
+    name,
+    description,
+    path: filePath,
+    source,
+    content: raw,
+    calls: [...calls],
+  };
+}
+
+function extractFirstParagraph(content) {
+  const lines = content.replace(FRONTMATTER_RE, '').split('\n')
+    .filter(l => l.trim() && !l.startsWith('#') && !l.startsWith('---'));
+  return lines[0]?.trim().slice(0, 200) || '';
+}

package/pg-hook.js

@@ -0,0 +1,46 @@
+#!/usr/bin/env node
+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 () => {
+  try {
+    const json = JSON.parse(input);
+    const prompt = json?.prompt || json?.user_prompt || '';
+    if (!prompt || prompt.length < 5) process.exit(0);
+
+    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)
+      .slice(0, 3)
+      .filter(s => s.score > 0.55);
+
+    if (results.length === 0) process.exit(0);
+
+    const context = [
+      '## Relevant skills found by PromptGraph',
+      ...results.map(s => `- **${s.name}** (score: ${s.score.toFixed(2)}): ${s.description}\n  path: ${s.path}`),
+      '\nIf any skill matches the task — Read its file and follow its instructions.',
+    ].join('\n');
+
+    console.log(JSON.stringify({
+      hookSpecificOutput: {
+        hookEventName: 'UserPromptSubmit',
+        additionalContext: context,
+      }
+    }));
+  } catch {
+    process.exit(0);
+  }
+});