kyp-mem 0.2.1 → 0.3.0

package/kyp_mem/hooks.py

@@ -0,0 +1,126 @@
+"""KYP-MEM session hooks — compile captured tool activity into vault notes."""
+
+import sys
+import json
+from datetime import datetime
+from pathlib import Path
+
+SESSION_DIR = Path.home() / ".kyp-mem" / "sessions"
+CURRENT_SESSION = SESSION_DIR / "current.jsonl"
+
+MIN_ACTIONS = 3
+
+
+def handle_stop():
+    if not CURRENT_SESSION.exists():
+        return
+
+    text = CURRENT_SESSION.read_text().strip()
+    if not text:
+        CURRENT_SESSION.unlink(missing_ok=True)
+        return
+
+    entries = []
+    for line in text.split("\n"):
+        try:
+            entries.append(json.loads(line))
+        except json.JSONDecodeError:
+            continue
+
+    if not entries:
+        CURRENT_SESSION.unlink(missing_ok=True)
+        return
+
+    write_actions = [e for e in entries if e.get("action") in ("edit", "create", "command")]
+    if len(write_actions) < MIN_ACTIONS:
+        CURRENT_SESSION.unlink(missing_ok=True)
+        return
+
+    project_dir = entries[0].get("cwd", "unknown")
+    project_name = Path(project_dir).name
+    session_id = datetime.now().strftime("%Y-%m-%d_%H%M%S")
+
+    files_edited = set()
+    files_created = set()
+    commands = []
+    timeline = []
+
+    for e in entries:
+        ts_raw = e.get("ts", "")
+        ts = ts_raw[11:19] if len(ts_raw) >= 19 else ""
+        action = e.get("action", "")
+
+        if action == "edit":
+            fp = e.get("file", "")
+            files_edited.add(fp)
+            timeline.append(f"  {ts} — Edit `{Path(fp).name}`")
+        elif action == "create":
+            fp = e.get("file", "")
+            files_created.add(fp)
+            timeline.append(f"  {ts} — Write `{Path(fp).name}`")
+        elif action == "command":
+            cmd = e.get("command", "")
+            commands.append(cmd)
+            short = cmd[:80] + "..." if len(cmd) > 80 else cmd
+            timeline.append(f"  {ts} — `{short}`")
+
+    parts = [f"# Session {session_id}", ""]
+    parts.append(f"**Project:** `{project_dir}`")
+    parts.append(f"**Actions:** {len(entries)} total, {len(write_actions)} substantive")
+    parts.append("")
+
+    if files_edited:
+        parts.append("## Files Modified")
+        for f in sorted(files_edited):
+            parts.append(f"- `{f}`")
+        parts.append("")
+
+    if files_created:
+        parts.append("## Files Created")
+        for f in sorted(files_created):
+            parts.append(f"- `{f}`")
+        parts.append("")
+
+    if commands:
+        parts.append("## Commands Run")
+        for cmd in commands[:25]:
+            short = cmd[:120] + "..." if len(cmd) > 120 else cmd
+            parts.append(f"- `{short}`")
+        parts.append("")
+
+    if timeline:
+        parts.append("## Timeline")
+        for line in timeline[:40]:
+            parts.append(line)
+        if len(timeline) > 40:
+            parts.append(f"  ... and {len(timeline) - 40} more actions")
+
+    content = "\n".join(parts)
+    tags = ["session", "auto-captured", project_name]
+
+    from .config import get_vault_path
+    from .vault import Vault
+
+    vault = Vault(get_vault_path())
+    vault.write_note(f"Sessions/{session_id}.md", content, tags, {})
+
+    CURRENT_SESSION.unlink(missing_ok=True)
+
+
+def main():
+    if len(sys.argv) > 1 and sys.argv[1] == "stop":
+        handle_stop()
+    else:
+        raw = sys.stdin.read().strip()
+        if not raw:
+            return
+        try:
+            data = json.loads(raw)
+        except json.JSONDecodeError:
+            return
+        if "stop_reason" in data:
+            handle_stop()
+
+
+if __name__ == "__main__":
+    main()

package/kyp_mem/server.py

@@ -7,18 +7,23 @@ from .vault import Vault
 
 vault = Vault(get_vault_path())
 
-mcp = FastMCP("kyp-mem", description="Know Your Project — headless knowledge base for AI agents")
+mcp = FastMCP("kyp-mem")
 
 
 @mcp.tool()
 def kyp_list(path: str = "") -> str:
-    """List notes and folders in the vault. Pass a folder path to list its contents, or empty for root."""
+    """List notes and folders in the vault. Shows inline tags for quick navigation. Pass a folder path or empty for root."""
     tree = vault.list_tree(path)
     lines = []
     for f in tree["folders"]:
         lines.append(f"  {f}/")
     for n in tree["notes"]:
-        lines.append(f"  {n}")
+        rel = f"{path}/{n}" if path else n
+        note = vault.index.notes.get(rel)
+        if note and note.tags:
+            lines.append(f"  {n}  [{', '.join(note.tags)}]")
+        else:
+            lines.append(f"  {n}")
     if not lines:
         lines.append("(empty vault)")
     header = f"Vault: {path or '/'}"
@@ -26,24 +31,46 @@ def kyp_list(path: str = "") -> str:
 
 
 @mcp.tool()
-def kyp_read(path: str) -> str:
-    """Read a note by path (e.g. 'Hedge Engine/Configuration.md'). Returns content + properties + backlinks + related notes."""
+def kyp_read(path: str, full: bool = False) -> str:
+    """Read a note. Returns brief summary by default (title, tags, preview, links). Set full=True for complete content."""
     note = vault.read(path)
     if not note:
         return f"Not found: {path}"
 
+    if not full:
+        parts = [f"# {note.title}"]
+        if note.tags:
+            parts.append(f"tags: {', '.join(note.tags)}")
+        if note.created:
+            parts.append(f"created: {note.created}")
+
+        lines = [l for l in note.content.strip().split("\n") if l.strip() and not l.startswith("# ")]
+        preview = "\n".join(lines[:6])
+        if len(lines) > 6:
+            preview += "\n..."
+        parts.append("")
+        parts.append(preview)
+
+        backlinks = vault.get_backlinks(path)
+        outlinks = note.links
+        if outlinks:
+            parts.append(f"\nlinks: {', '.join(f'[[{l}]]' for l in outlinks)}")
+        if backlinks:
+            parts.append(f"backlinks: {', '.join(f'[[{b.replace('.md', '')}]]' for b in backlinks)}")
+
+        return "\n".join(parts)
+
     parts = [f"# {note.title}", ""]
 
     if note.tags or note.properties or note.created:
-        parts.append("**Properties:**")
         if note.tags:
-            parts.append(f"  tags: {', '.join(note.tags)}")
+            parts.append(f"tags: {', '.join(note.tags)}")
         if note.created:
-            parts.append(f"  created: {note.created}")
+            parts.append(f"created: {note.created}")
         if note.updated:
-            parts.append(f"  updated: {note.updated}")
+            parts.append(f"updated: {note.updated}")
         for k, v in note.properties.items():
-            parts.append(f"  {k}: {v}")
+            parts.append(f"{k}: {v}")
         parts.append("")
 
     parts.append(note.content)