kyp-mem 0.2.2 → 0.4.0

package/kyp_mem/server.py

@@ -1,6 +1,7 @@
 """KYP-MEM MCP server — headless knowledge base for AI agents."""
 
 import json
+from datetime import datetime
 from mcp.server.fastmcp import FastMCP
 from .config import get_vault_path
 from .vault import Vault
@@ -12,13 +13,18 @@ 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 +32,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)
@@ -164,3 +192,106 @@ def kyp_stats() -> str:
         f"  Links: {s['links']}\n"
         f"  Backlinks: {s['backlinks']}"
     )
+
+
+@mcp.tool()
+def kyp_session_create(project: str, summary: str = "", investigated: str = "", learned: str = "", completed: str = "", next_steps: str = "") -> str:
+    """Create a structured session note. Project is required. Sections accept markdown text."""
+    session_id = datetime.now().strftime("%Y-%m-%d_%H%M%S")
+    parts = [f"# Session {session_id}", ""]
+    parts.append(f"**Project:** {project}")
+    parts.append("")
+    parts.append("## Summary")
+    parts.append(summary or "")
+    parts.append("")
+    parts.append("## INVESTIGATED")
+    parts.append(investigated or "")
+    parts.append("")
+    parts.append("## LEARNED")
+    parts.append(learned or "")
+    parts.append("")
+    parts.append("## COMPLETED")
+    parts.append(completed or "")
+    parts.append("")
+    parts.append("## NEXT STEPS")
+    parts.append(next_steps or "")
+
+    content = "\n".join(parts)
+    tags = ["session", "manual", project.lower().replace(" ", "-")]
+    path = f"{project}/Sessions/{session_id}.md"
+    vault.write_note(path, content, tags, {})
+    return f"Created session: {path}"
+
+
+@mcp.tool()
+def kyp_sessions(project: str = "", limit: int = 10) -> str:
+    """List sessions, optionally filtered by project. Shows most recent first."""
+    sessions = []
+    for path, note in vault.index.notes.items():
+        if "/Sessions/" not in path and not path.startswith("Sessions/"):
+            continue
+        if project and not path.lower().startswith(project.lower() + "/"):
+            continue
+        sessions.append((path, note))
+    sessions.sort(key=lambda s: s[0], reverse=True)
+    sessions = sessions[:limit]
+    if not sessions:
+        return "No sessions found." + (f" (project filter: {project})" if project else "")
+    lines = ["Sessions:", ""]
+    for path, note in sessions:
+        tags = f" [{', '.join(note.tags)}]" if note.tags else ""
+        date = note.created or note.updated or ""
+        lines.append(f"  {date} — {note.title} ({path}){tags}")
+    return "\n".join(lines)
+
+
+@mcp.tool()
+def kyp_project_context(project: str) -> str:
+    """Get full project context: knowledge base + recent session summaries. Call this at session start to understand project history, avoid repeating past work, and prevent hallucination."""
+    parts = []
+
+    knowledge_path = f"{project}/Knowledge.md"
+    knowledge = vault.read(knowledge_path)
+    if knowledge:
+        parts.append("=== PROJECT KNOWLEDGE ===")
+        parts.append(knowledge.content)
+        parts.append("")
+
+    project_notes = []
+    for path, note in vault.index.notes.items():
+        if path.startswith(f"{project}/") and "/Sessions/" not in path and path != knowledge_path:
+            project_notes.append((path, note))
+
+    if project_notes:
+        parts.append("=== PROJECT NOTES ===")
+        for path, note in sorted(project_notes):
+            parts.append(f"\n--- {note.title} ({path}) ---")
+            preview = note.content.strip().split("\n")
+            parts.append("\n".join(preview[:10]))
+            if len(preview) > 10:
+                parts.append("...")
+        parts.append("")
+
+    sessions = []
+    for path, note in vault.index.notes.items():
+        if path.startswith(f"{project}/Sessions/"):
+            sessions.append((path, note))
+    sessions.sort(key=lambda s: s[0], reverse=True)
+    recent = sessions[:5]
+
+    if recent:
+        parts.append(f"=== RECENT SESSIONS ({len(recent)} of {len(sessions)}) ===")
+        for path, note in recent:
+            parts.append(f"\n--- {note.title} ---")
+            content = note.content
+            timeline_idx = content.find("## Timeline")
+            if timeline_idx > 0:
+                parts.append(content[:timeline_idx].strip())
+            else:
+                parts.append(content[:500])
+        parts.append("")
+
+    if not parts:
+        return f"No context found for project '{project}'. Create a Knowledge.md to get started."
+
+    return "\n".join(parts)