kyp-mem 0.6.7 → 0.6.9

package/kyp_mem/static/index.html

@@ -268,6 +268,9 @@ body.resizing #resize-handle { pointer-events: auto !important; }
 .tree-folder .tf-icon { color: var(--muted); font-size: 11px; }
 .tree-folder .tf-label { flex: 1; color: var(--fg); }
 .tree-folder .tf-count { font-size: var(--fz-xs); color: var(--dim); font-variant-numeric: tabular-nums; }
+.tree-folder .tf-graph-btn { font-size: 11px; color: var(--dim); opacity: 0; transition: opacity 0.15s; padding: 0 2px; }
+.tree-folder:hover .tf-graph-btn { opacity: 1; }
+.tree-folder .tf-graph-btn:hover { color: var(--accent); }
 
 /* Sidebar rows */
 .sidebar-row {
@@ -720,7 +723,7 @@ let _graphData = null;
 let searchTimeout, sessionSearchTimeout;
 
 // ─── Helpers ─────────────────────────────────────────────────────────────────
-async function fetchJSON(url) { const r = await fetch(url); return r.json(); }
+async function fetchJSON(url) { const r = await fetch(url, { cache: 'no-store' }); return r.json(); }
 
 function $(id) { return document.getElementById(id); }
 
@@ -790,7 +793,7 @@ function setView(v) {
 $('view-switch').addEventListener('click', e => {
   if (e.target.dataset.view) {
     setView(e.target.dataset.view);
-    if (view === 'graph') { _graphData = null; renderGraphView(); }
+    if (view === 'graph') { _graphProject = null; _graphData = null; renderGraphView(); }
     else if (view === 'session' && activeSession) loadNote(activeSession);
     else if (view === 'note' && currentPath) loadNote(currentPath);
   }
@@ -886,7 +889,7 @@ function renderSidebar(sessionsData) {
     const group = document.createElement('div');
     const folder = document.createElement('div');
     folder.className = 'tree-folder';
-    folder.innerHTML = `<span class="tf-arrow">▾</span><span class="tf-icon">≡</span><span class="tf-label">${project}</span><span class="tf-count">${sessions.length}</span>`;
+    folder.innerHTML = `<span class="tf-arrow">▾</span><span class="tf-icon">≡</span><span class="tf-label">${project}</span><span class="tf-count">${sessions.length}</span><button class="tf-graph-btn ghost-btn" title="Open graph for ${project}" data-project="${project}">▦</button>`;
 
     const list = document.createElement('div');
     list.style.cssText = 'display:flex;flex-direction:column;gap:1px;padding-left:16px;margin-top:2px;';
@@ -902,6 +905,14 @@ function renderSidebar(sessionsData) {
       list.appendChild(row);
     });
 
+    folder.querySelector('.tf-graph-btn').addEventListener('click', (e) => {
+      e.stopPropagation();
+      _graphProject = project;
+      _graphData = null;
+      setView('graph');
+      renderGraphView();
+    });
+
     folder.addEventListener('click', () => {
       const arrow = folder.querySelector('.tf-arrow');
       const isOpen = arrow.textContent === '▾';
@@ -955,13 +966,14 @@ function renderSidebar(sessionsData) {
 
 function renderProjectTree(container) {
   if (!treeData) return;
-  function walk(node, parent) {
+  function walk(node, parent, depth) {
     if (node.type === 'folder' && node.name !== 'vault') {
       if (node.name === 'Sessions') return;
       const group = document.createElement('div');
       const folder = document.createElement('div');
       folder.className = 'tree-folder';
-      folder.innerHTML = `<span class="tf-arrow">▾</span><span class="tf-icon">≡</span><span class="tf-label">${node.name}</span>`;
+      const isTopLevel = depth === 0;
+      folder.innerHTML = `<span class="tf-arrow">▾</span><span class="tf-icon">≡</span><span class="tf-label">${node.name}</span>${isTopLevel ? `<button class="tf-graph-btn ghost-btn" title="Open graph for ${node.name}" data-project="${node.name}">▦</button>` : ''}`;
       const children = document.createElement('div');
       children.style.cssText = 'display:flex;flex-direction:column;gap:1px;padding-left:16px;margin-top:2px;';
 
@@ -972,10 +984,20 @@ function renderProjectTree(container) {
         children.style.display = isOpen ? 'none' : 'flex';
       });
 
+      if (isTopLevel) {
+        folder.querySelector('.tf-graph-btn').addEventListener('click', (e) => {
+          e.stopPropagation();
+          _graphProject = node.name;
+          _graphData = null;
+          setView('graph');
+          renderGraphView();
+        });
+      }
+
       group.appendChild(folder);
       group.appendChild(children);
       parent.appendChild(group);
-      (node.children || []).forEach(c => walk(c, children));
+      (node.children || []).forEach(c => walk(c, children, depth + 1));
     } else if (node.type === 'note') {
       const row = document.createElement('button');
       row.className = 'sidebar-row';
@@ -985,10 +1007,10 @@ function renderProjectTree(container) {
       row.addEventListener('click', () => { setView('note'); loadNote(node.path); });
       parent.appendChild(row);
     } else if (node.children) {
-      node.children.forEach(c => walk(c, parent));
+      node.children.forEach(c => walk(c, parent, depth));
     }
   }
-  walk(treeData, container);
+  walk(treeData, container, 0);
 }
 
 function renderTagCloud(container) {
@@ -1158,6 +1180,8 @@ function renderRail(note) {
   rail.appendChild(graphCard);
 
   graphCard.querySelector('#rail-graph-expand')?.addEventListener('click', () => {
+    const parts = (note.path || '').split('/');
+    _graphProject = parts.length > 1 ? parts[0] : null;
     _graphData = null;
     setView('graph');
     renderGraphView();
@@ -1445,6 +1469,7 @@ function renderSessionView(note) {
 
 // ─── Graph View ──────────────────────────────────────────────────────────────
 let _graphMode = 'projects';
+let _graphProject = null;
 
 function renderGraphView() {
   const area = $('content-area');
@@ -1457,6 +1482,10 @@ function renderGraphView() {
             <button class="tool-chip graph-mode active" data-mode="projects">projects</button>
             <button class="tool-chip graph-mode" data-mode="sessions">sessions</button>
             <span class="dim tab-nums" id="graph-stats"></span>
+            <span id="graph-project-filter" style="display:none;font-size:var(--fz-sm);color:var(--accent);margin-left:4px">
+              <span id="graph-project-name"></span>
+              <button class="ghost-btn" id="graph-clear-project" style="color:var(--muted);margin-left:4px;font-size:var(--fz-xs)" title="Show all projects">✕</button>
+            </span>
           </div>
           <div class="graph-header-right">
             <button class="tool-chip graph-layout active" data-layout="force">force</button>
@@ -1484,17 +1513,28 @@ function renderGraphView() {
   `;
 
   updateGraphLegend();
+  updateGraphProjectFilter();
 
   const closeBtn = $('graph-close');
   if (closeBtn) {
     closeBtn.addEventListener('click', (e) => {
       e.stopPropagation();
+      _graphProject = null;
       setView('note');
       if (currentPath) loadNote(currentPath);
       else $('content-area').innerHTML = '<div style="padding:40px;color:var(--muted)">Select a note from the sidebar</div>';
     });
   }
 
+  // Clear project filter
+  $('graph-clear-project')?.addEventListener('click', () => {
+    _graphProject = null;
+    _graphData = null;
+    updateGraphProjectFilter();
+    const activeLayout = document.querySelector('.graph-layout.active');
+    buildFullGraph(activeLayout ? activeLayout.dataset.layout : 'force');
+  });
+
   // Mode toggle (projects / sessions)
   document.querySelectorAll('.graph-mode').forEach(btn => {
     btn.addEventListener('click', () => {
@@ -1520,6 +1560,17 @@ function renderGraphView() {
   buildFullGraph('force');
 }
 
+function updateGraphProjectFilter() {
+  const el = $('graph-project-filter');
+  if (!el) return;
+  if (_graphProject) {
+    $('graph-project-name').textContent = `› ${_graphProject}`;
+    el.style.display = 'inline';
+  } else {
+    el.style.display = 'none';
+  }
+}
+
 function updateGraphLegend() {
   const legend = $('graph-legend');
   if (!legend) return;
@@ -1544,7 +1595,9 @@ async function buildFullGraph(layout) {
   wrap.innerHTML = '';
 
   if (!_graphData) {
-    _graphData = await fetchJSON(`/api/graph?kind=${_graphMode}`);
+    let url = `/api/graph?kind=${_graphMode}`;
+    if (_graphProject) url += `&project=${encodeURIComponent(_graphProject)}`;
+    _graphData = await fetchJSON(url);
   }
 
   let nodes = _graphData.nodes.map(n => ({ ...n }));
@@ -1931,30 +1984,26 @@ async function init() {
   const density = localStorage.getItem('kyp-density') || 'regular';
   document.documentElement.setAttribute('data-density', density);
 
-  const [rawTree, sessionsData, stats] = await Promise.all([
-    fetchJSON('/api/tree'), fetchJSON('/api/sessions'), fetchJSON('/api/stats'),
-  ]);
-  treeData = rawTree;
-
-  function walk(node) {
-    if (node.type === 'note') allNotes[node.path] = { title: node.name.replace('.md', ''), tags: node.tags || [] };
-    (node.children || []).forEach(walk);
+  try {
+    const [rawTree, sessionsData, stats] = await Promise.all([
+      fetchJSON('/api/tree'), fetchJSON('/api/sessions'), fetchJSON('/api/stats'),
+    ]);
+    treeData = rawTree;
+
+    function walk(node) {
+      if (node.type === 'note') allNotes[node.path] = { title: node.name.replace('.md', ''), tags: node.tags || [] };
+      (node.children || []).forEach(walk);
+    }
+    walk(treeData);
+
+    renderSidebar(sessionsData);
+    $('stat-notes').textContent = stats.notes;
+    $('stat-folders').textContent = stats.folders;
+    loadTokenEconomics();
+  } catch (e) {
+    console.error('KYP-MEM init failed:', e);
+    $('sidebar-scroll').innerHTML = `<div style="padding:12px;color:#f66;font-size:12px">Init error: ${e.message}<br>Check console for details</div>`;
   }
-  walk(treeData);
-
-  // Enrich notes with tags
-  const promises = Object.keys(allNotes).map(async path => {
-    try {
-      const note = await fetchJSON(`/api/note/${path}`);
-      if (!note.error) allNotes[path] = { title: note.title, tags: note.tags || [] };
-    } catch {}
-  });
-  await Promise.all(promises);
-
-  renderSidebar(sessionsData);
-  $('stat-notes').textContent = stats.notes;
-  $('stat-folders').textContent = stats.folders;
-  loadTokenEconomics();
 
   // Show empty state
   $('content-area').innerHTML = `

package/kyp_mem/ui.py

@@ -16,6 +16,18 @@ def create_app(vault_path: str = None) -> FastAPI:
     vault = Vault(vault_path)
     app = FastAPI(title="KYP-MEM")
 
+    from starlette.middleware.base import BaseHTTPMiddleware
+    from starlette.responses import Response
+
+    class NoCacheMiddleware(BaseHTTPMiddleware):
+        async def dispatch(self, request, call_next):
+            response: Response = await call_next(request)
+            if request.url.path.startswith("/api/"):
+                response.headers["Cache-Control"] = "no-store, no-cache, must-revalidate, max-age=0"
+            return response
+
+    app.add_middleware(NoCacheMiddleware)
+
     @app.get("/")
     def index():
         html_path = Path(__file__).parent / "static" / "index.html"
@@ -23,6 +35,7 @@ def create_app(vault_path: str = None) -> FastAPI:
 
     @app.get("/api/tree")
     def tree():
+        vault.refresh_if_stale()
         return JSONResponse(vault.get_full_tree())
 
     @app.get("/api/stats")
@@ -31,35 +44,38 @@ def create_app(vault_path: str = None) -> FastAPI:
         return JSONResponse(vault.get_stats())
 
     @app.get("/api/graph")
-    def graph(kind: str = "all"):
-        nodes = []
-        edges = []
-        seen_edges = set()
-        for path, note in vault.index.notes.items():
+    def graph(kind: str = "all", project: str = None):
+        vault.refresh_if_stale()
+
+        def include(path: str) -> bool:
+            if project and not path.startswith(project + "/"):
+                return False
             node_kind = "session" if "/Sessions/" in path else "note"
             if kind == "projects" and node_kind == "session":
-                continue
+                return False
             if kind == "sessions" and node_kind == "note":
+                return False
+            return True
+
+        nodes = []
+        for path, note in vault.index.notes.items():
+            if not include(path):
                 continue
+            node_kind = "session" if "/Sessions/" in path else "note"
             nodes.append({"id": path, "title": note.title, "kind": node_kind, "tags": note.tags})
-            for link in (note.links or []):
-                target = None
-                link_lower = link.lower()
-                for p in vault.index.notes:
-                    stem = p.split("/")[-1].replace(".md", "").lower()
-                    if stem == link_lower:
-                        target = p
-                        break
-                if target and target != path:
-                    target_kind = "session" if "/Sessions/" in target else "note"
-                    if kind == "projects" and target_kind == "session":
-                        continue
-                    if kind == "sessions" and target_kind == "note":
-                        continue
-                    key = tuple(sorted([path, target]))
-                    if key not in seen_edges:
-                        seen_edges.add(key)
-                        edges.append({"source": path, "target": target})
+
+        node_ids = {n["id"] for n in nodes}
+        edges = []
+        seen_edges = set()
+        for path in node_ids:
+            for target in vault.index.forward_links.get(path, set()):
+                if target not in node_ids:
+                    continue
+                key = tuple(sorted([path, target]))
+                if key not in seen_edges:
+                    seen_edges.add(key)
+                    edges.append({"source": path, "target": target})
+
         return JSONResponse({"nodes": nodes, "edges": edges})
 
     @app.get("/api/note/{path:path}")

package/kyp_mem/vault.py

@@ -98,6 +98,7 @@ class Index:
     def __init__(self):
         self.notes: dict[str, Note] = {}
         self.backlinks: dict[str, set] = defaultdict(set)
+        self.forward_links: dict[str, set] = defaultdict(set)
         self.tag_index: dict[str, set] = defaultdict(set)
         self._word_index: dict[str, set] = defaultdict(set)
         self._name_to_path: dict[str, str] = {}
@@ -105,6 +106,7 @@ class Index:
     def rebuild(self, notes: dict[str, Note]):
         self.notes = notes
         self.backlinks = defaultdict(set)
+        self.forward_links = defaultdict(set)
         self.tag_index = defaultdict(set)
         self._word_index = defaultdict(set)
         self._name_to_path = {}
@@ -116,8 +118,9 @@ class Index:
         for path, note in notes.items():
             for link in note.links:
                 target = self._name_to_path.get(link.lower())
-                if target:
+                if target and target != path:
                     self.backlinks[target].add(path)
+                    self.forward_links[path].add(target)
 
             for tag in note.tags:
                 self.tag_index[tag.lower()].add(path)
@@ -210,14 +213,23 @@ class Vault:
         init_vector_db(str(self.root))
         self._load_all()
         self._sync_vector_db()
+        self._last_mtime = self._max_mtime()
 
     def _disk_note_paths(self) -> set[str]:
         return {str(f.relative_to(self.root)) for f in self.root.rglob("*.md")}
 
+    def _max_mtime(self) -> float:
+        mtimes = [f.stat().st_mtime for f in self.root.rglob("*.md")]
+        return max(mtimes) if mtimes else 0.0
+
     def refresh_if_stale(self):
-        if self._disk_note_paths() != set(self.index.notes.keys()):
+        current_mtime = self._max_mtime()
+        paths_changed = self._disk_note_paths() != set(self.index.notes.keys())
+        content_changed = current_mtime != self._last_mtime
+        if paths_changed or content_changed:
             self._load_all()
             self._sync_vector_db()
+            self._last_mtime = current_mtime
 
     def _sync_vector_db(self):
         mem = get_session_memory()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kyp-mem",
-  "version": "0.6.7",
+  "version": "0.6.9",
   "description": "Know Your Project — Persistent & Session level knowledge base for AI agents. MCP-powered with wikilinks, backlinks, auto-learning, and neon web UI.",
   "bin": {
     "kyp-mem": "bin/cli.mjs"

package/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kyp-mem"
-version = "0.6.7"
+version = "0.6.9"
 description = "Know Your Project — Persistent knowledge base for AI agents. MCP-powered with wikilinks, backlinks, auto-learning, and neon web UI."
 readme = "README.md"
 license = {text = "MIT"}