promptgraph-mcp 1.0.2 → 1.1.0

package/platform.js

@@ -0,0 +1,98 @@
+import os from 'os';
+import path from 'path';
+import fs from 'fs';
+
+const HOME = os.homedir();
+
+export const PLATFORMS = {
+  'claude-code': {
+    name: 'Claude Code',
+    configPath: path.join(HOME, '.claude', 'settings.json'),
+    addMcp: (config, serverPath) => {
+      const json = readJson(config.configPath);
+      json.mcpServers = json.mcpServers || {};
+      json.mcpServers.promptgraph = { command: 'npx', args: ['promptgraph-mcp'] };
+      writeJson(config.configPath, json);
+    },
+  },
+  'claude-desktop': {
+    name: 'Claude Desktop',
+    configPath: getClaudeDesktopConfig(),
+    addMcp: (config, serverPath) => {
+      const json = readJson(config.configPath);
+      json.mcpServers = json.mcpServers || {};
+      json.mcpServers.promptgraph = { command: 'npx', args: ['promptgraph-mcp'] };
+      writeJson(config.configPath, json);
+    },
+  },
+  'cline': {
+    name: 'Cline (VS Code)',
+    configPath: path.join(HOME, '.vscode', 'mcp.json'),
+    addMcp: (config, serverPath) => {
+      const json = readJson(config.configPath) || { servers: {} };
+      json.servers = json.servers || {};
+      json.servers.promptgraph = { command: 'npx', args: ['promptgraph-mcp'] };
+      fs.mkdirSync(path.dirname(config.configPath), { recursive: true });
+      writeJson(config.configPath, json);
+    },
+  },
+  'codex': {
+    name: 'OpenAI Codex CLI',
+    configPath: path.join(HOME, '.codex', 'config.json'),
+    addMcp: (config, serverPath) => {
+      const json = readJson(config.configPath) || {};
+      json.mcpServers = json.mcpServers || {};
+      json.mcpServers.promptgraph = { command: 'npx', args: ['promptgraph-mcp'] };
+      fs.mkdirSync(path.dirname(config.configPath), { recursive: true });
+      writeJson(config.configPath, json);
+    },
+  },
+  'cursor': {
+    name: 'Cursor',
+    configPath: path.join(HOME, '.cursor', 'mcp.json'),
+    addMcp: (config, serverPath) => {
+      const json = readJson(config.configPath) || { mcpServers: {} };
+      json.mcpServers.promptgraph = { command: 'npx', args: ['promptgraph-mcp'] };
+      fs.mkdirSync(path.dirname(config.configPath), { recursive: true });
+      writeJson(config.configPath, json);
+    },
+  },
+  'windsurf': {
+    name: 'Windsurf',
+    configPath: path.join(HOME, '.codeium', 'windsurf', 'mcp_config.json'),
+    addMcp: (config, serverPath) => {
+      const json = readJson(config.configPath) || { mcpServers: {} };
+      json.mcpServers.promptgraph = { command: 'npx', args: ['promptgraph-mcp'] };
+      fs.mkdirSync(path.dirname(config.configPath), { recursive: true });
+      writeJson(config.configPath, json);
+    },
+  },
+};
+
+export function detectPlatforms() {
+  return Object.entries(PLATFORMS)
+    .filter(([, p]) => p.configPath && fs.existsSync(path.dirname(p.configPath)))
+    .map(([id, p]) => ({ id, ...p }));
+}
+
+function getClaudeDesktopConfig() {
+  if (process.platform === 'win32') {
+    const base = process.env.LOCALAPPDATA || '';
+    const packages = path.join(base, 'Packages');
+    if (fs.existsSync(packages)) {
+      const claudeDir = fs.readdirSync(packages).find(d => d.startsWith('Claude_'));
+      if (claudeDir) return path.join(packages, claudeDir, 'LocalCache', 'Roaming', 'Claude', 'claude_desktop_config.json');
+    }
+  }
+  if (process.platform === 'darwin') return path.join(HOME, 'Library', 'Application Support', 'Claude', 'claude_desktop_config.json');
+  return path.join(HOME, '.config', 'Claude', 'claude_desktop_config.json');
+}
+
+function readJson(filePath) {
+  try { return JSON.parse(fs.readFileSync(filePath, 'utf8')); } catch { return {}; }
+}
+
+function writeJson(filePath, data) {
+  fs.mkdirSync(path.dirname(filePath), { recursive: true });
+  fs.writeFileSync(filePath, JSON.stringify(data, null, 2));
+}

package/search.js

@@ -4,18 +4,26 @@ import { getDb } from './db.js';
 export async function search(query, topK = 5) {
   const db = getDb();
   const queryVec = await embed(query);
-  const skills = db.prepare('SELECT name, description, path, source, embedding FROM skills').all();
-
-  return skills
-    .map(skill => ({
-      name: skill.name,
-      description: skill.description,
-      path: skill.path,
-      source: skill.source,
-      score: cosineSimilarity(queryVec, JSON.parse(skill.embedding)),
-    }))
-    .sort((a, b) => b.score - a.score)
-    .slice(0, topK);
+
+  // RAG: search over chunks, deduplicate by skill
+  const chunks = db.prepare('SELECT skill_name, 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_name);
+    if (!prev || score > prev) bestBySkill.set(chunk.skill_name, score);
+  }
+
+  const skillNames = [...bestBySkill.entries()]
+    .sort((a, b) => b[1] - a[1])
+    .slice(0, topK)
+    .map(([name]) => name);
+
+  return skillNames.map(name => {
+    const skill = db.prepare('SELECT name, description, path, source FROM skills WHERE name = ?').get(name);
+    return { ...skill, score: bestBySkill.get(name) };
+  });
 }
 
 export function getContext(name) {
@@ -24,7 +32,7 @@ export function getContext(name) {
   if (!skill) return null;
   const callees = db.prepare('SELECT to_skill FROM edges WHERE from_skill = ?').all(name).map(r => r.to_skill);
   const callers = db.prepare('SELECT from_skill FROM edges WHERE to_skill = ?').all(name).map(r => r.from_skill);
-  return { ...skill, embedding: undefined, callees, callers };
+  return { ...skill, callees, callers };
 }
 
 export function getCallers(name) {

package/watcher.js

@@ -1,50 +1,50 @@
-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 } 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}`);
+  }
+}