local-knowledge-graph 1.6.1 → 1.8.0

package/public/index.html

@@ -33,6 +33,7 @@
         <div data-tab="entity" class="active">实体</div>
         <div data-tab="relation">关系</div>
         <div data-tab="search">检索</div>
+        <div data-tab="mcp">MCP</div>
         <div data-tab="log">日志</div>
         <div data-tab="version">版本</div>
       </div>
@@ -63,11 +64,27 @@
           <input id="r-name" placeholder="例如：位于">
           <label>大类（对应线型/颜色）</label>
           <select id="r-category"></select>
+          <label>置信度</label>
+          <select id="r-confidence">
+            <option value="确证">确证（有明确依据）</option>
+            <option value="推测">推测（合理推断）</option>
+            <option value="存疑">存疑（待核实）</option>
+          </select>
+          <label>来源引用（URL / 文献+页码，可选）</label>
+          <input id="r-source-ref" placeholder="例如：《明史》卷304">
           <label>终点实体</label>
           <select id="r-target"></select>
           <div class="row"><button class="primary" id="r-submit">添加</button></div>
         </div>
-        <div class="card"><h3>关系列表</h3><div id="r-list"></div></div>
+        <div class="card"><h3>关系列表</h3>
+          <div class="conf-filter" id="conf-filter">
+            <span class="cf-chip active" data-c="">全部</span>
+            <span class="cf-chip" data-c="确证">确证</span>
+            <span class="cf-chip" data-c="推测">推测</span>
+            <span class="cf-chip" data-c="存疑">存疑</span>
+          </div>
+          <div id="r-list"></div>
+        </div>
       </div>
 
       <div class="tabbody" id="tab-search">
@@ -103,6 +120,19 @@
           <div class="row"><button class="primary" id="c-run">执行</button><span id="c-hint" style="font-size:11px;color:#8fa3c0;align-self:center">支持 contains / = 与 类型:互动|归属</span></div>
           <div id="c-results"></div>
         </div>
+        <div class="card">
+          <h3>文档批量入图（md / txt / pdf）</h3>
+          <div class="row" style="gap:12px">
+            <button class="primary" id="ing-btn">选择文档</button>
+            <label style="display:flex;align-items:center;gap:4px;font-size:12px;cursor:pointer">
+              <input type="checkbox" id="ing-auto" style="width:auto"> 跳过审核直接入库
+            </label>
+            <span id="ing-status" style="font-size:11px;color:#8fa3c0;align-self:center"></span>
+          </div>
+          <div id="ing-tasks"></div>
+          <div class="hint-text">LLM逐片抽取候选三元组 → 审核勾选 → 确认入库（先自动打保存点）。新实体来源标记"文档"，关系自带来源引用"《文档》片段n"。勾选"跳过审核"后抽取完成即自动全量入库。</div>
+          <input type="file" id="ing-file" accept=".md,.markdown,.txt,.pdf" style="display:none">
+        </div>
       </div>
 
       <div class="tabbody" id="tab-log">
@@ -136,6 +166,60 @@
         </div>
         <div class="card"><h3>提交历史（与操作日志双向绑定）</h3><div id="v-list"></div></div>
       </div>
+
+      <div class="tabbody" id="tab-mcp">
+        <div class="card">
+          <h3>MCP 服务（外部 Agent 接入）</h3>
+          <div class="row" style="gap:16px;flex-wrap:wrap">
+            <label style="display:flex;align-items:center;gap:6px;cursor:pointer">
+              <input type="checkbox" id="mcp-toggle" style="width:auto"> 启用 MCP 服务
+            </label>
+            <label style="display:flex;align-items:center;gap:6px;cursor:pointer">
+              <input type="checkbox" id="mcp-readonly" style="width:auto"> 只读模式（禁用外部写入）
+            </label>
+          </div>
+          <div id="mcp-cfg" style="display:none">
+            <div class="row" style="align-items:center;gap:8px;flex-wrap:wrap;margin-top:8px">
+              <span style="font-size:12px;color:#8fa3c0">端点</span>
+              <code id="mcp-endpoint" style="font-size:11px;word-break:break-all"></code>
+            </div>
+            <div class="row" style="align-items:center;gap:8px;flex-wrap:wrap">
+              <span style="font-size:12px;color:#8fa3c0">令牌</span>
+              <code id="mcp-token" style="font-size:11px;word-break:break-all"></code>
+              <button id="mcp-copy-token" title="复制令牌">复制</button>
+              <button id="mcp-regen" title="重新生成后旧令牌立即失效">重新生成</button>
+            </div>
+            <div class="hint-text">外部 Agent 凭令牌访问。令牌泄露时点击"重新生成"即刻吊销旧令牌。</div>
+
+            <h3 style="margin-top:14px">客户端一键配置</h3>
+            <label>通用 Streamable HTTP（Cline / opencode / 通用 MCP 客户端）</label>
+            <div class="row" style="align-items:flex-start;gap:8px">
+              <pre id="mcp-snippet-http" class="mcp-snippet"></pre>
+              <button id="mcp-copy-http">复制</button>
+            </div>
+            <label>Claude Desktop / Cursor（claude_desktop_config.json / mcp.json）</label>
+            <div class="row" style="align-items:flex-start;gap:8px">
+              <pre id="mcp-snippet-desktop" class="mcp-snippet"></pre>
+              <button id="mcp-copy-desktop">复制</button>
+            </div>
+            <label>本机 stdio（与网页同机时，免令牌）</label>
+            <div class="row" style="align-items:flex-start;gap:8px">
+              <pre id="mcp-snippet-stdio" class="mcp-snippet"></pre>
+              <button id="mcp-copy-stdio">复制</button>
+            </div>
+            <div class="hint-text">stdio 方式命令：npx -y local-knowledge-graph --mcp（数据目录默认 ~/.local-knowledge-graph，可用环境变量 KG_DATA_DIR 指定其他目录）。远程 Agent 请使用 HTTP 方式并保管好令牌。</div>
+          </div>
+          <div id="mcp-status" class="hint-text"></div>
+        </div>
+        <div class="card">
+          <h3>外部 Agent 可用的工具（12 个）</h3>
+          <div class="hint-text">
+            查询：kg_stats / kg_list_entities / kg_get_entity / kg_get_graph / kg_ego / kg_search / kg_cypher / kg_inference / kg_path / kg_digest / kg_export_rdf<br>
+            写入：kg_apply_ops（add_entity / add_relation / update_entity / update_relation / delete_entity / delete_relation，自动记操作日志+Git保存点）<br>
+            实时同步：任一写入方修改图谱后，页面会在 3 秒内自动刷新。
+          </div>
+        </div>
+      </div>
     </div>
 
     <div id="canvas-wrap">

package/public/style.css

@@ -19,6 +19,18 @@
   header .badge.ok { color: #7ee787; border-color: #2ea04366; }
   header .badge.off { color: #f0883e; border-color: #f0883e66; }
 
+  /* 置信度徽章与过滤条 */
+  .conf-badge { font-size: 10px; border: 1px solid; border-radius: 8px; padding: 0 6px; margin-left: 5px; vertical-align: 1px; }
+  .conf-filter { display: flex; gap: 5px; margin-bottom: 8px; flex-wrap: wrap; }
+  .cf-chip { font-size: 11px; color: #8fa3c0; border: 1px solid #2a3a5c; border-radius: 10px; padding: 2px 10px; cursor: pointer; user-select: none; }
+  .cf-chip.active { color: #7fd1ff; border-color: #58a6ff; background: #16233f; }
+  .alias-chip { display: inline-block; font-size: 11px; color: #cfe2ff; border: 1px solid #2a3a5c; border-radius: 8px; padding: 0 6px; margin-right: 4px; }
+  .alias-chip i { font-style: normal; color: #f0883e; cursor: pointer; margin-left: 3px; }
+  .p-path { margin-bottom: 10px; border-top: 1px dashed #24344f; padding-top: 6px; }
+  .p-path:first-child { border-top: none; padding-top: 0; }
+  .p-head2 { font-size: 11.5px; color: #7fd1ff; margin-bottom: 4px; display: flex; align-items: center; gap: 8px; }
+  .ask-path-row { line-height: 1.7; }
+
   button {
     background: #1b2b4d; color: #cfe2ff; border: 1px solid #2f4a7a; border-radius: 5px;
     padding: 5px 12px; cursor: pointer; font-size: 12px; font-family: inherit;
@@ -484,3 +496,20 @@
   html[data-theme="light"] #file-menu button:active { background: #eaf3ff; }
   html[data-theme="light"] .m-btn { background: #eaf3ff; }
   html[data-theme="light"] #m-legend { background: rgba(255, 255, 255, 0.92); }
+
+/* 浅色主题：置信度/别名/路径新增元素适配 */
+html[data-theme="light"] .cf-chip { color: #5a6b85; border-color: #c9d6e8; }
+html[data-theme="light"] .cf-chip.active { color: #1f6feb; border-color: #1f6feb; background: #e8f1ff; }
+html[data-theme="light"] .alias-chip { color: #2c3e57; border-color: #c9d6e8; }
+html[data-theme="light"] .alias-chip i { color: #c44f27; }
+html[data-theme="light"] .p-path { border-top-color: #dfe7f2; }
+html[data-theme="light"] .p-head2 { color: #1a5dc8; }
+
+/* MCP 设置片段 */
+.mcp-snippet {
+  flex: 1; margin: 0; padding: 8px 10px; font-size: 10.5px; line-height: 1.5;
+  background: #0d1526; color: #9fb6d9; border: 1px solid #24324d; border-radius: 8px;
+  white-space: pre-wrap; word-break: break-all; max-height: 110px; overflow: auto;
+  font-family: ui-monospace, SFMono-Regular, Menlo, Consolas, monospace;
+}
+html[data-theme="light"] .mcp-snippet { background: #f4f7fc; color: #33445e; border-color: #d8e2f0; }

package/server.js

@@ -2,6 +2,7 @@
 
 const path = require('path');
 const fs = require('fs');
+const crypto = require('crypto');
 const express = require('express');
 const db = require('./lib/db');
 const git = require('./lib/git');
@@ -13,6 +14,30 @@ const embeddings = require('./lib/embeddings');
 const askLib = require('./lib/ask');
 const updater = require('./lib/updater');
 const importer = require('./lib/importer');
+const bus = require('./lib/bus');
+const { createMcpHandler } = require('./mcp/http');
+
+// ---------- 图谱变更总线：db 写函数统一包装，任一写入即时广播 ----------
+for (const fn of ['addEntity', 'updateEntity', 'deleteEntity', 'addRelation', 'updateRelation', 'deleteRelation', 'addAlias', 'removeAlias', 'deleteEntityImage', 'applyAgentOps', 'undoLast', 'logRestore']) {
+  const orig = db[fn];
+  if (typeof orig === 'function') {
+    db[fn] = function (...args) {
+      const r = orig.apply(this, args);
+      bus.emit('graph-changed', { via: fn });
+      return r;
+    };
+  }
+}
+
+function mcpServerVersion() {
+  try { return require('./package.json').version || '1.0.0'; } catch (_) { return '1.0.0'; }
+}
+const mcpHandler = createMcpHandler({
+  isEnabled: () => embeddings.loadSettings().mcp_enabled === true,
+  getToken: () => embeddings.loadSettings().mcp_token || '',
+  isReadonly: () => embeddings.loadSettings().mcp_readonly === true,
+  serverVersion: mcpServerVersion(),
+});
 
 const PORT = Number(process.env.PORT || 3000);
 const app = express();
@@ -43,7 +68,27 @@ function agentSessionInfo() {
 
 // ---------- 实体 ----------
 api.get('/entities', (req, res) => res.json(db.listEntities()));
-api.post('/entities', (req, res) => res.json(db.addEntity(req.body, '手工')));
+api.post('/entities', (req, res) => {
+  try {
+    const body = req.body || {};
+    const name = body.name;
+    // 同名消歧：主名或别名占用时，按 merge_into / force 分流
+    if (typeof name === 'string' && name.trim()) {
+      const conflicts = db.findNameConflicts(name);
+      if (conflicts.length) {
+        const mergeInto = Number(body.merge_into);
+        if (body.merge_into !== undefined && Number.isInteger(mergeInto) && mergeInto > 0) {
+          const alias = db.addAlias(mergeInto, name.trim(), '手工');
+          return res.json({ merged: true, alias, entity: db.getEntity(mergeInto) });
+        }
+        if (body.force !== true && body.force !== 1 && body.force !== '1') {
+          return res.status(409).json({ error: `名称"${name.trim()}"已被占用，可并入现有实体、改用限定名或强制创建`, conflicts });
+        }
+      }
+    }
+    res.json(db.addEntity(body, '手工'));
+  } catch (e) { res.status(e.status || 500).json({ error: e.message, conflicts: e.conflicts }); }
+});
 api.put('/entities/:id', (req, res) => res.json(db.updateEntity(Number(req.params.id), req.body, '手工')));
 api.delete('/entities/:id', (req, res) => {
   const r = db.deleteEntity(Number(req.params.id), '手工');
@@ -81,6 +126,80 @@ api.get('/graph/path', (req, res) => {
 });
 
 // ---------- 撤销最近操作（快照逆向写入） ----------
+api.get('/aliases', (req, res) => res.json(db.aliasMap()));
+
+api.get('/aliases/records', (req, res) => res.json(db.aliasRecords()));
+
+api.post('/aliases', (req, res) => {
+  try {
+    const { entity_id, alias } = req.body || {};
+    res.json(db.addAlias(Number(entity_id), alias, '手工'));
+  } catch (e) { res.status(e.status || 500).json({ error: e.message, conflicts: e.conflicts }); }
+});
+
+api.delete('/aliases/:id', (req, res) => {
+  try { res.json(db.removeAlias(Number(req.params.id), '手工')); }
+  catch (e) { res.status(e.status || 500).json({ error: e.message }); }
+});
+
+api.get('/graph/paths', (req, res) => {
+  try {
+    const fromId = db.resolveKey(req.query.from);
+    const toId = db.resolveKey(req.query.to);
+    let maxHops = 4;
+    if (req.query.max !== undefined && String(req.query.max).trim() !== '') {
+      maxHops = Number(req.query.max);
+      if (!Number.isInteger(maxHops)) return res.status(400).json({ error: 'max必须为整数' });
+    }
+    res.json(db.findPaths(fromId, toId, { maxHops }));
+  } catch (e) { res.status(e.status || 500).json({ error: e.message, candidates: e.candidates }); }
+});
+
+// ---------- 相似实体（语义Top-K，结构降级） ----------
+api.get('/similar/:id', async (req, res) => {
+  try {
+    const k = req.query.k !== undefined ? Number(req.query.k) : 8;
+    res.json(await require('./lib/similar').similarEntities(Number(req.params.id), Number.isInteger(k) ? k : 8));
+  } catch (e) { res.status(e.status || 500).json({ error: e.message }); }
+});
+
+// ---------- 文档批量入图 ----------
+const ingest = require('./lib/ingest');
+ingest.loadPersisted();
+
+api.post('/ingest', async (req, res) => {
+  if (agentBusy) return res.status(429).json({ error: '已有AI任务执行中（问答、导入或智能检索），请稍后再试' });
+  const { filename, content_b64, auto_commit } = req.body || {};
+  if (!filename || !content_b64) return res.status(400).json({ error: 'filename 与 content_b64 必填' });
+  let buf;
+  try { buf = Buffer.from(content_b64, 'base64'); } catch (_) { return res.status(400).json({ error: 'base64解码失败' }); }
+  agentBusy = true;
+  try {
+    res.json(await ingest.createTask(String(filename), buf, {
+      autoCommit: auto_commit === true || auto_commit === 1 || auto_commit === '1',
+      onSettled: () => { agentBusy = false; },
+    }));
+  } catch (e) {
+    agentBusy = false;
+    res.status(e.status || 500).json({ error: e.message });
+  }
+});
+
+api.get('/ingest/tasks', (req, res) => res.json(ingest.listTasks()));
+api.get('/ingest/:id', (req, res) => {
+  const t = ingest.getTask(req.params.id);
+  if (!t) return res.status(404).json({ error: '任务不存在' });
+  res.json(t);
+});
+api.post('/ingest/:id/commit', (req, res) => {
+  try { res.json(ingest.commitTask(req.params.id, (req.body || {}).selected || null, '手工')); }
+  catch (e) { res.status(e.status || 500).json({ error: e.message }); }
+});
+api.delete('/ingest/:id', (req, res) => {
+  try { res.json(ingest.deleteTask(req.params.id)); }
+  catch (e) { res.status(e.status || 500).json({ error: e.message }); }
+});
+
 api.post('/undo', (req, res) => {
   try { res.json({ ok: true, ...db.undoLast('手工'), counts: db.counts(), version: db.getVersion() }); }
   catch (e) { res.status(e.status || 500).json({ error: e.message }); }
@@ -351,6 +470,55 @@ api.post('/git/restore', (req, res) => {
   res.json(r);
 });
 
+// ---------- MCP 外部接入设置 ----------
+api.get('/mcp/settings', (req, res) => {
+  const s = embeddings.loadSettings();
+  res.json({
+    enabled: s.mcp_enabled === true,
+    readonly: s.mcp_readonly === true,
+    token: s.mcp_token || '',
+    endpoint: `/mcp`,
+    version: mcpServerVersion(),
+  });
+});
+
+api.put('/mcp/settings', (req, res) => {
+  try {
+    const patch = {};
+    const b = req.body || {};
+    if (typeof b.enabled === 'boolean') patch.mcp_enabled = b.enabled;
+    if (typeof b.readonly === 'boolean') patch.mcp_readonly = b.readonly;
+    const cur = embeddings.loadSettings();
+    if (patch.mcp_enabled === true && !cur.mcp_token) patch.mcp_token = crypto.randomBytes(24).toString('hex');
+    const s = embeddings.saveSettings(patch);
+    res.json({ enabled: s.mcp_enabled === true, readonly: s.mcp_readonly === true, token: s.mcp_token || '', endpoint: `/mcp`, version: mcpServerVersion() });
+  } catch (e) { res.status(400).json({ error: e.message }); }
+});
+
+api.post('/mcp/token/regen', (req, res) => {
+  const token = crypto.randomBytes(24).toString('hex');
+  embeddings.saveSettings({ mcp_token: token });
+  res.json({ token });
+});
+
+// ---------- 实时同步：SSE 事件流 ----------
+api.get('/events', (req, res) => {
+  res.writeHead(200, {
+    'Content-Type': 'text/event-stream',
+    'Cache-Control': 'no-cache',
+    Connection: 'keep-alive',
+    'X-Accel-Buffering': 'no',
+  });
+  res.write('retry: 3000\n\n');
+  const send = (type, payload) => {
+    try { res.write(`event: ${type}\ndata: ${JSON.stringify(payload || {})}\n\n`); } catch (_) { /* 连接已断 */ }
+  };
+  send('hello', { ts: Date.now() });
+  const unsub = bus.on((type, payload) => send(type, payload));
+  const heartbeat = setInterval(() => { try { res.write(': ping\n\n'); } catch (_) { /* 忽略 */ } }, 25000);
+  req.on('close', () => { clearInterval(heartbeat); unsub(); });
+});
+
 // ---------- OpenCode自然语言指令 ----------
 // AI任务互斥：OpenCode进程最长5分钟，防止并发请求叠加进程互抢SQLite与内存
 let agentBusy = false;
@@ -517,6 +685,9 @@ api.post('/agent/doc', async (req, res) => {
   } finally { agentBusy = false; }
 });
 
+// MCP 端点（置于 /api 之前；内部有开关与令牌校验）
+app.use('/mcp', mcpHandler);
+
 app.use('/api', api);
 
 // 统一错误处理
@@ -623,3 +794,15 @@ function scheduleRestart() {
     }
   });
 })(0);
+
+// ---------- 跨进程变更探测：外部进程（stdio MCP/CLI）写库时兜底广播 ----------
+let lastSignature = '';
+function detectExternalChange() {
+  try {
+    const sig = JSON.stringify([db.counts(), db.maxLogId()]);
+    if (lastSignature && sig !== lastSignature) bus.emit('graph-changed', { via: 'external' });
+    lastSignature = sig;
+  } catch (_) { /* 数据库短暂不可用时跳过本轮 */ }
+}
+setInterval(detectExternalChange, 3000).unref();
+