kyp-mem 0.6.1 → 0.6.5

package/kyp_mem/hooks.py

@@ -3,6 +3,7 @@
 import os
 import sys
 import json
+import subprocess
 from datetime import datetime
 from pathlib import Path
 
@@ -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,20 @@ 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)
+
+        parts.append("")
+        parts.append("**INSTRUCTION:** Display the session summaries and token savings above to the user as your first message at session start. Format it cleanly.")
+
         output = "\n".join(parts)
+
         try:
             _record_injection(project_name, len(output))
         except Exception:
@@ -128,7 +174,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 +199,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)
@@ -454,9 +500,11 @@ Return ONLY this format (no preamble):
 Raw session data:
 {raw_note}"""
 
+        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,
+            capture_output=True, text=True, timeout=120, env=env,
         )
         if result.returncode == 0 and result.stdout.strip():
             return result.stdout.strip()
@@ -466,7 +514,7 @@ Raw session data:
 
 
 def handle_stop():
-    if not CURRENT_SESSION.exists():
+    if _is_subprocess() or not CURRENT_SESSION.exists():
         return
 
     text = CURRENT_SESSION.read_text().strip()
@@ -642,6 +690,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)
 
@@ -710,8 +762,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

@@ -1042,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;
@@ -1370,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}

package/kyp_mem/ui.py

@@ -27,6 +27,7 @@ def create_app(vault_path: str = None) -> FastAPI:
 
     @app.get("/api/stats")
     def stats():
+        vault.refresh_if_stale()
         return JSONResponse(vault.get_stats())
 
     @app.get("/api/graph")
@@ -199,6 +200,7 @@ def create_app(vault_path: str = None) -> FastAPI:
 
     @app.get("/api/sessions")
     def list_sessions(project: str = ""):
+        vault.refresh_if_stale()
         sessions = {}
         for path, note in vault.index.notes.items():
             if "/Sessions/" not in path and not path.startswith("Sessions/"):

package/kyp_mem/vault.py

@@ -211,6 +211,14 @@ class Vault:
         self._load_all()
         self._sync_vector_db()
 
+    def _disk_note_paths(self) -> set[str]:
+        return {str(f.relative_to(self.root)) for f in self.root.rglob("*.md")}
+
+    def refresh_if_stale(self):
+        if self._disk_note_paths() != set(self.index.notes.keys()):
+            self._load_all()
+            self._sync_vector_db()
+
     def _sync_vector_db(self):
         mem = get_session_memory()
         for path, note in self.index.notes.items():

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kyp-mem",
-  "version": "0.6.1",
+  "version": "0.6.5",
   "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.5"
 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"}