kyp-mem 0.6.0 → 0.6.4

package/kyp_mem/cli.py

@@ -133,8 +133,24 @@ def _run_init():
     print(f"  {G}✓{R} Vault:  {vault_path}")
     print(f"  {G}✓{R} Config: {CONFIG_FILE}")
     print()
-    print(f"  {Y}Next step:{R} Connect to Claude Code:")
-    print(f"    {Y}kyp-mem setup-claude{R}")
+
+    claude_settings = Path.home() / ".claude" / "settings.json"
+    already_setup = False
+    if claude_settings.exists():
+        try:
+            cs = json.loads(claude_settings.read_text())
+            has_mcp = "kyp-mem" in cs.get("mcpServers", {})
+            has_hooks = any("kyp-mem" in str(h) for h in cs.get("hooks", {}).get("Stop", []))
+            already_setup = has_mcp and has_hooks
+        except (json.JSONDecodeError, KeyError):
+            pass
+
+    if already_setup:
+        print(f"  {G}✓{R} Claude Code already configured (MCP server + hooks)")
+        print(f"  {D}  Restart Claude Code if you changed the vault path.{R}")
+    else:
+        print(f"  {Y}Next step:{R} Connect to Claude Code:")
+        print(f"    {Y}kyp-mem setup-claude --global && kyp-mem install-hooks --global{R}")
     print()
 
 

package/kyp_mem/hooks.py

@@ -3,13 +3,14 @@
 import os
 import sys
 import json
+import subprocess
 from datetime import datetime
 from pathlib import Path
 
 SESSION_DIR = Path.home() / ".kyp-mem" / "sessions"
 CURRENT_SESSION = SESSION_DIR / "current.jsonl"
 
-MIN_ACTIONS = 3
+MIN_ACTIONS = 5
 CHARS_PER_TOKEN = 4
 
 COMMAND_OUTPUT_ESTIMATES = {
@@ -73,9 +74,51 @@ def _record_injection(project, chars):
     _save_token_stats(stats)
 
 
+def _is_subprocess():
+    return os.environ.get("KYP_MEM_SUMMARIZING") == "1"
+
+
+def _extract_session_summary(content, max_chars=800):
+    import re
+    parts = []
+    for heading in ("Summary", "LEARNED", "COMPLETED"):
+        m = re.search(rf"(?:^|\n)##\s+{re.escape(heading)}\s*\n(.*?)(?=\n##\s|\Z)", content, re.DOTALL)
+        if m:
+            text = m.group(1).strip()
+            if heading == "Summary":
+                parts.append(text)
+            else:
+                parts.append(f"**{heading.title()}:** {text[:250]}")
+    return "\n".join(parts)[:max_chars] if parts else content[:200]
+
+
+def _build_stats_line(project_name, injected_chars, session_ids):
+    try:
+        stats = _load_token_stats()
+        project_sessions = [s for s in stats.get("sessions", []) if s.get("project") == project_name]
+        if not project_sessions:
+            return None
+        matched = [s for s in project_sessions if s.get("id") in session_ids]
+        exploration_tokens = sum(s.get("exploration_tokens", 0) for s in matched)
+        injected_tokens = injected_chars // CHARS_PER_TOKEN
+        if exploration_tokens == 0:
+            return None
+        saved = exploration_tokens - injected_tokens
+        if saved <= 0:
+            return None
+        return (
+            f"---\n"
+            f"*kyp-mem saved ~{saved:,} tokens this session (injected ~{injected_tokens:,} tokens instead of re-exploring ~{exploration_tokens:,})*"
+        )
+    except Exception:
+        return None
+
+
 def handle_session_start():
     """Inject recent session memory into the conversation at session start."""
     sys.stdin.read()
+    if _is_subprocess():
+        return
 
     cwd = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
     project_name = Path(cwd).name
@@ -98,8 +141,7 @@ def handle_session_start():
             return
 
         parts = [f"# [kyp-mem] {project_name} — Recent Sessions"]
-        parts.append(f"Use `kyp_search` or `kyp_project_context` for architecture/project knowledge on demand.")
-        parts.append("")
+        parts.append(f"Use `kyp_search` or `kyp_project_context` for architecture/project knowledge on demand.\n")
 
         parts.append(f"## Last {len(sessions)} Sessions")
         for sp in sessions:
@@ -107,16 +149,16 @@ def handle_session_start():
             if not note:
                 continue
             parts.append(f"### {note.title}")
-            content = note.content
-            timeline_idx = content.find("## TIMELINE")
-            if timeline_idx < 0:
-                timeline_idx = content.find("## Timeline")
-            if timeline_idx > 0:
-                content = content[:timeline_idx].strip()
-            parts.append(content)
+            summary = _extract_session_summary(note.content)
+            parts.append(summary)
             parts.append("")
 
+        session_ids = {Path(sp).stem for sp in sessions}
+        stats_line = _build_stats_line(project_name, len("\n".join(parts)), session_ids)
+        if stats_line:
+            parts.append(stats_line)
         output = "\n".join(parts)
+
         try:
             _record_injection(project_name, len(output))
         except Exception:
@@ -128,7 +170,7 @@ def handle_session_start():
 
 def handle_user_prompt():
     raw = sys.stdin.read().strip()
-    if not raw:
+    if not raw or _is_subprocess():
         return
     try:
         data = json.loads(raw)
@@ -153,7 +195,7 @@ def handle_user_prompt():
 
 def handle_post_tool_use():
     raw = sys.stdin.read().strip()
-    if not raw:
+    if not raw or _is_subprocess():
         return
     try:
         data = json.loads(raw)
@@ -405,17 +447,19 @@ def _build_next_steps(files_edited, files_created, commands_classified):
 
 
 def _summarize_with_claude(raw_note, project_name):
-    """Use Claude to rewrite session sections in plain, human-readable language."""
+    """Use Claude CLI to rewrite session sections — uses existing Claude Code auth."""
     try:
-        from .config import get_session_model
-        import anthropic
+        import shutil
+        claude_bin = shutil.which("claude")
+        if not claude_bin:
+            return None
 
+        from .config import get_session_model
         model = get_session_model()
-        client = anthropic.Anthropic()
 
         prompt = f"""Rewrite this raw coding session into a structured summary. A future AI agent reads this to pick up where you left off — be precise and technical.
 
-You have: user prompts (the objectives), a timeline of file edits/reads/commands, and raw section data. Synthesize into a dense, specific narrative.
+You have: user prompts (the objectives), a timeline of file edits/reads/commands with their actual content and output. Synthesize into a dense, specific narrative.
 
 ## Format rules
 
@@ -452,18 +496,21 @@ Return ONLY this format (no preamble):
 Raw session data:
 {raw_note}"""
 
-        response = client.messages.create(
-            model=model,
-            max_tokens=1024,
-            messages=[{"role": "user", "content": prompt}],
+        env = os.environ.copy()
+        env["KYP_MEM_SUMMARIZING"] = "1"
+        result = subprocess.run(
+            [claude_bin, "-p", prompt, "--max-turns", "1", "--model", model],
+            capture_output=True, text=True, timeout=120, env=env,
         )
-        return response.content[0].text.strip()
+        if result.returncode == 0 and result.stdout.strip():
+            return result.stdout.strip()
+        return None
     except Exception:
         return None
 
 
 def handle_stop():
-    if not CURRENT_SESSION.exists():
+    if _is_subprocess() or not CURRENT_SESSION.exists():
         return
 
     text = CURRENT_SESSION.read_text().strip()
@@ -639,6 +686,10 @@ def handle_stop():
     except Exception:
         pass
 
+    # Delete session file BEFORE summarization so the spawned claude subprocess
+    # doesn't pollute it via hooks writing back into current.jsonl
+    CURRENT_SESSION.unlink(missing_ok=True)
+
     # Try Claude summarization, fall back to raw sections
     summarized = _summarize_with_claude(raw_note, project_name)
 
@@ -707,8 +758,6 @@ def handle_stop():
     vault = Vault(get_vault_path())
     vault.write_note(f"{project_name}/Sessions/{session_id}.md", content, tags, {})
 
-    CURRENT_SESSION.unlink(missing_ok=True)
-
 
 def main():
     if len(sys.argv) > 1 and sys.argv[1] == "stop":

package/kyp_mem/static/index.html

@@ -475,8 +475,9 @@ body.resizing #resize-handle { pointer-events: auto !important; }
 .graph-header {
   display: flex; align-items: center; justify-content: space-between;
   padding: 10px 14px; border-bottom: 1px solid var(--line);
-  background: var(--bg-2); gap: 12px;
+  background: var(--bg-2); gap: 12px; position: relative; z-index: 10;
 }
+.graph-header .ghost-btn { cursor: pointer; }
 .graph-header-left { display: flex; align-items: center; gap: 10px; font-size: var(--fz-sm); white-space: nowrap; }
 .graph-header-right { display: flex; gap: 6px; flex-shrink: 0; }
 .tool-chip {
@@ -1041,6 +1042,18 @@ function openSession(path) {
   loadNote(path);
 }
 
+async function deleteSession(path) {
+  if (!confirm('Delete this session?')) return;
+  const res = await fetch(`/api/note/${path}`, { method: 'DELETE' });
+  const data = await res.json();
+  if (data.ok) {
+    activeSession = null;
+    currentPath = null;
+    $('content-area').innerHTML = '<div class="dim" style="padding:40px;text-align:center">Session deleted</div>';
+    refreshAll();
+  }
+}
+
 // ─── Load Note ───────────────────────────────────────────────────────────────
 async function loadNote(path) {
   currentPath = path;
@@ -1369,7 +1382,10 @@ function renderSessionView(note) {
           ${tagsHtml}
           <span class="dim" style="font-size:var(--fz-xs);margin-left:8px">${note.created || sessionFile}</span>
         </div>
-        <button class="ghost-btn" onclick="openEditor('${note.path}')">⌥ edit</button>
+        <div style="display:flex;gap:6px">
+          <button class="ghost-btn" onclick="openEditor('${note.path}')">⌥ edit</button>
+          <button class="ghost-btn" style="color:var(--muted)" onclick="deleteSession('${note.path}')">✕ delete</button>
+        </div>
       </div>
       <h1 style="margin:12px 0 4px;font-size:calc(var(--fz-xl) + 6px);font-weight:500;letter-spacing:-0.01em">
         Session ${sessionFile}
@@ -1428,6 +1444,8 @@ function renderSessionView(note) {
 }
 
 // ─── Graph View ──────────────────────────────────────────────────────────────
+let _graphMode = 'projects';
+
 function renderGraphView() {
   const area = $('content-area');
   area.innerHTML = `
@@ -1436,24 +1454,19 @@ function renderGraphView() {
         <div class="graph-header">
           <div class="graph-header-left">
             <span class="acc">▦</span>
-            <span style="font-weight:500">vault graph</span>
+            <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>
           </div>
           <div class="graph-header-right">
-            <button class="tool-chip active">force</button>
-            <button class="tool-chip">radial</button>
-            <button class="tool-chip">time</button>
+            <button class="tool-chip graph-layout active" data-layout="force">force</button>
+            <button class="tool-chip graph-layout" data-layout="radial">radial</button>
+            <button class="tool-chip graph-layout" data-layout="time">time</button>
             <button class="ghost-btn" id="graph-close">✕</button>
           </div>
         </div>
         <div class="graph-svg-wrap" id="graph-svg-wrap"></div>
-        <div class="graph-legend">
-          <div class="graph-legend-title">legend</div>
-          <div class="graph-legend-item"><i style="width:8px;height:8px;border-radius:999px;background:var(--accent)"></i>notes</div>
-          <div class="graph-legend-item"><i style="width:8px;height:8px;border-radius:999px;background:var(--panel-2);border:1px solid var(--muted)"></i>sessions</div>
-          <div class="graph-legend-item"><i style="width:12px;height:1px;background:var(--line-2)"></i>link</div>
-          <div class="graph-legend-item"><i style="width:12px;height:0;border-top:1px dashed var(--line-2)"></i>session ref</div>
-        </div>
+        <div class="graph-legend" id="graph-legend"></div>
       </section>
       <aside class="graph-rail" id="graph-rail">
         <div class="rail-card">
@@ -1470,36 +1483,88 @@ function renderGraphView() {
     </div>
   `;
 
-  $('graph-close').addEventListener('click', () => setView('note'));
+  updateGraphLegend();
 
-  // Layout toggle buttons
-  const chips = document.querySelectorAll('.graph-header-right .tool-chip');
-  chips.forEach(chip => {
+  const closeBtn = $('graph-close');
+  if (closeBtn) {
+    closeBtn.addEventListener('click', (e) => {
+      e.stopPropagation();
+      setView('note');
+      if (currentPath) loadNote(currentPath);
+      else $('content-area').innerHTML = '<div style="padding:40px;color:var(--muted)">Select a note from the sidebar</div>';
+    });
+  }
+
+  // Mode toggle (projects / sessions)
+  document.querySelectorAll('.graph-mode').forEach(btn => {
+    btn.addEventListener('click', () => {
+      document.querySelectorAll('.graph-mode').forEach(b => b.classList.remove('active'));
+      btn.classList.add('active');
+      _graphMode = btn.dataset.mode;
+      _graphData = null;
+      updateGraphLegend();
+      const activeLayout = document.querySelector('.graph-layout.active');
+      buildFullGraph(activeLayout ? activeLayout.dataset.layout : 'force');
+    });
+  });
+
+  // Layout toggle (force / radial / time)
+  document.querySelectorAll('.graph-layout').forEach(chip => {
     chip.addEventListener('click', () => {
-      chips.forEach(c => c.classList.remove('active'));
+      document.querySelectorAll('.graph-layout').forEach(c => c.classList.remove('active'));
       chip.classList.add('active');
-      buildFullGraph(chip.textContent.trim());
+      buildFullGraph(chip.dataset.layout);
     });
   });
 
   buildFullGraph('force');
 }
 
+function updateGraphLegend() {
+  const legend = $('graph-legend');
+  if (!legend) return;
+  if (_graphMode === 'sessions') {
+    legend.innerHTML = `
+      <div class="graph-legend-title">sessions</div>
+      <div class="graph-legend-item"><i style="width:8px;height:8px;border-radius:999px;background:var(--panel-2);border:1px solid var(--muted)"></i>session</div>
+      <div class="graph-legend-item"><i style="width:12px;height:0;border-top:1px dashed var(--line-2)"></i>temporal link</div>
+    `;
+  } else {
+    legend.innerHTML = `
+      <div class="graph-legend-title">projects</div>
+      <div class="graph-legend-item"><i style="width:8px;height:8px;border-radius:999px;background:var(--accent)"></i>notes</div>
+      <div class="graph-legend-item"><i style="width:12px;height:1px;background:var(--line-2)"></i>wikilink</div>
+    `;
+  }
+}
+
 async function buildFullGraph(layout) {
   const wrap = $('graph-svg-wrap');
   if (!wrap) return;
   wrap.innerHTML = '';
 
   if (!_graphData) {
-    _graphData = await fetchJSON('/api/graph');
+    _graphData = await fetchJSON(`/api/graph?kind=${_graphMode}`);
+  }
+
+  let nodes = _graphData.nodes.map(n => ({ ...n }));
+  let links = _graphData.edges.map(e => ({ ...e }));
+
+  // For sessions mode, create temporal links between consecutive sessions
+  if (_graphMode === 'sessions' && links.length === 0 && nodes.length > 1) {
+    const sorted = [...nodes].sort((a, b) => a.id.localeCompare(b.id));
+    for (let i = 1; i < sorted.length; i++) {
+      links.push({ source: sorted[i - 1].id, target: sorted[i].id });
+    }
   }
 
-  const nodes = _graphData.nodes.map(n => ({ ...n }));
-  const links = _graphData.edges.map(e => ({ ...e }));
   const nodeMap = new Map(nodes.map(n => [n.id, n]));
 
   $('graph-stats').textContent = `${nodes.length}n · ${links.length}e`;
-  if (nodes.length === 0) return;
+  if (nodes.length === 0) {
+    wrap.innerHTML = '<div style="display:flex;align-items:center;justify-content:center;height:100%;color:var(--muted);font-size:var(--fz-sm)">No nodes to display</div>';
+    return;
+  }
 
   const rect = wrap.getBoundingClientRect();
   const W = rect.width || 800;
@@ -1568,36 +1633,44 @@ async function buildFullGraph(layout) {
     const s = typeof l.source === 'object' ? l.source : nodeMap.get(l.source);
     const t = typeof l.target === 'object' ? l.target : nodeMap.get(l.target);
     if (!s || !t) return;
-    const isSession = s.kind === 'session' || t.kind === 'session';
+    const isTemporal = _graphMode === 'sessions';
     edgeGroup.append('line')
       .attr('x1', s.x).attr('y1', s.y).attr('x2', t.x).attr('y2', t.y)
-      .attr('stroke', 'var(--line-2)').attr('stroke-width', 1.2)
-      .attr('stroke-opacity', isSession ? 0.35 : 0.6)
-      .attr('stroke-dasharray', isSession ? '4 3' : '')
+      .attr('stroke', 'var(--line-2)').attr('stroke-width', isTemporal ? 1 : 1.2)
+      .attr('stroke-opacity', isTemporal ? 0.3 : 0.6)
+      .attr('stroke-dasharray', isTemporal ? '4 3' : '')
       .attr('stroke-linecap', 'round');
   });
 
   // Draw nodes
   const nodeGroup = g.append('g');
   nodes.forEach(n => {
-    const r = n.kind === 'session' ? 6 : 10;
+    const isPrimary = _graphMode === 'sessions' ? n.kind === 'session' : n.kind === 'note';
+    const r = isPrimary ? 10 : 6;
     const ng = nodeGroup.append('g').style('cursor', 'pointer');
 
-    if (n.kind === 'session') {
-      ng.append('circle').attr('cx', n.x).attr('cy', n.y).attr('r', r)
-        .attr('fill', 'var(--bg-2)').attr('stroke', 'var(--muted)').attr('stroke-width', 1.2);
-    } else {
+    if (isPrimary) {
       ng.append('circle').attr('cx', n.x).attr('cy', n.y).attr('r', r + 4)
         .attr('fill', 'var(--accent)').attr('fill-opacity', 0.12);
       ng.append('circle').attr('cx', n.x).attr('cy', n.y).attr('r', r)
         .attr('fill', 'var(--panel-2)').attr('stroke', 'var(--accent)').attr('stroke-width', 1.5);
+    } else {
+      ng.append('circle').attr('cx', n.x).attr('cy', n.y).attr('r', r)
+        .attr('fill', 'var(--bg-2)').attr('stroke', 'var(--muted)').attr('stroke-width', 1.2);
+    }
+
+    let label = n.title;
+    if (_graphMode === 'sessions' && n.kind === 'session') {
+      const m = n.id.match(/(\d{4}-\d{2}-\d{2})_(\d{2})(\d{2})/);
+      label = m ? `${m[1]} ${m[2]}:${m[3]}` : n.title;
     }
+    if (label.length > 22) label = label.substring(0, 20) + '…';
 
     ng.append('text').attr('x', n.x).attr('y', n.y + r + 14)
       .attr('text-anchor', 'middle').attr('font-size', 11)
-      .attr('fill', n.kind === 'session' ? 'var(--dim)' : 'var(--muted)').attr('opacity', 0.9)
+      .attr('fill', isPrimary ? 'var(--muted)' : 'var(--dim)').attr('opacity', 0.9)
       .style('font-family', 'JetBrains Mono, monospace')
-      .text(n.title.length > 20 ? n.title.substring(0, 18) + '…' : n.title);
+      .text(label);
 
     ng.on('click', () => {
       updateGraphRail(n, links, nodes);

package/kyp_mem/ui.py

@@ -30,13 +30,17 @@ def create_app(vault_path: str = None) -> FastAPI:
         return JSONResponse(vault.get_stats())
 
     @app.get("/api/graph")
-    def graph():
+    def graph(kind: str = "all"):
         nodes = []
         edges = []
         seen_edges = set()
         for path, note in vault.index.notes.items():
-            kind = "session" if "/Sessions/" in path else "note"
-            nodes.append({"id": path, "title": note.title, "kind": kind, "tags": note.tags})
+            node_kind = "session" if "/Sessions/" in path else "note"
+            if kind == "projects" and node_kind == "session":
+                continue
+            if kind == "sessions" and node_kind == "note":
+                continue
+            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()
@@ -46,6 +50,11 @@ def create_app(vault_path: str = None) -> FastAPI:
                         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)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kyp-mem",
-  "version": "0.6.0",
+  "version": "0.6.4",
   "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.4.2"
+version = "0.6.4"
 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"}