claude-memory-agent 2.0.1 → 2.2.0

package/hooks/session_end_hook.py

@@ -0,0 +1,149 @@
+#!/usr/bin/env python3
+"""
+SessionEnd hook for Claude Code.
+
+Called when a Claude Code session ends. Performs final memory extraction
+from the conversation transcript and cleans up the cursor file for
+this session.
+
+This script:
+  1. Reads hook JSON from stdin (session_id, transcript_path, etc.)
+  2. Runs final extraction via extract_memories.py (with is_session_end=True)
+  3. Optionally invokes the existing session_end.py for full session wrapup
+  4. Cleans up cursor state for this session
+  5. Exits 0 on success OR failure (never blocks session teardown)
+
+Timing budget: < 5 seconds total.
+"""
+
+import sys
+import json
+import os
+import time
+
+# Ensure the hooks directory is on the path
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+
+def main():
+    start = time.time()
+
+    try:
+        # Read hook data from stdin
+        hook_data = {}
+        if not sys.stdin.isatty():
+            raw = sys.stdin.read()
+            if raw.strip():
+                hook_data = json.loads(raw)
+
+        # Ensure hook_event_name is set
+        if "hook_event_name" not in hook_data:
+            hook_data["hook_event_name"] = "SessionEnd"
+
+        session_id = hook_data.get("session_id", "")
+        transcript_path = hook_data.get("transcript_path", "")
+        project_path = hook_data.get("cwd") or hook_data.get("project_path", "")
+
+        # ---------------------------------------------------------------
+        # Step 1: Extract memories from transcript (final pass)
+        # ---------------------------------------------------------------
+        if transcript_path:
+            from extract_memories import run_extraction
+
+            results = run_extraction(
+                session_id=session_id,
+                transcript_path=transcript_path,
+                project_path=project_path,
+                is_session_end=True,  # This will clean up the cursor after extraction
+            )
+
+            elapsed_extract = round(time.time() - start, 2)
+            print(
+                f"[SessionEnd] Extraction complete: "
+                f"extracted={results['extracted']} stored={results['stored']} "
+                f"errors={results['errors']} time={elapsed_extract}s",
+                file=sys.stderr,
+            )
+        else:
+            print("[SessionEnd] No transcript_path provided, skipping extraction.", file=sys.stderr)
+            # Still clean up cursor if session_id is present
+            if session_id:
+                try:
+                    from extract_memories import cleanup_cursor
+                    cleanup_cursor(session_id)
+                except ImportError:
+                    pass
+
+        # ---------------------------------------------------------------
+        # Step 2: Trigger the existing session_end.py wrapup logic
+        # (summarization, daily log, MEMORY.md sync, flush)
+        # Only if we have time left in our budget
+        # ---------------------------------------------------------------
+        remaining = 5.0 - (time.time() - start)
+        if remaining > 1.0 and session_id:
+            try:
+                _trigger_session_wrapup(session_id, project_path, timeout=remaining - 0.5)
+            except Exception as e:
+                print(f"[SessionEnd] Session wrapup failed (non-fatal): {e}", file=sys.stderr)
+
+        elapsed_total = round(time.time() - start, 2)
+        print(f"[SessionEnd] Complete. Total time: {elapsed_total}s", file=sys.stderr)
+
+    except Exception as e:
+        elapsed = round(time.time() - start, 2)
+        print(f"[SessionEnd] Error (non-fatal): {e} [{elapsed}s]", file=sys.stderr)
+
+    # Always exit 0 - never block session end
+    sys.exit(0)
+
+
+def _trigger_session_wrapup(session_id: str, project_path: str, timeout: float = 3.0):
+    """
+    Trigger the existing session_end.py summarization via the memory agent API.
+    This calls key skills: daily_log_append_session, sync_memory_md, pre_compaction_flush.
+    Uses a single lightweight API call rather than the full async pipeline.
+    """
+    import urllib.request
+    import urllib.error
+
+    memory_agent_url = os.getenv("MEMORY_AGENT_URL", "http://localhost:8102")
+    api_key = os.getenv("MEMORY_API_KEY", "")
+
+    # Call the pre_compaction_flush skill as a lightweight session wrapup
+    payload = {
+        "jsonrpc": "2.0",
+        "method": "tasks/send",
+        "params": {
+            "message": {"parts": [{"type": "text", "text": ""}]},
+            "metadata": {
+                "skill_id": "pre_compaction_flush",
+                "params": {
+                    "project_path": project_path,
+                    "session_id": session_id,
+                }
+            }
+        },
+        "id": f"session-end-flush-{session_id}"
+    }
+
+    headers = {"Content-Type": "application/json"}
+    if api_key:
+        headers["X-Memory-Key"] = api_key
+
+    try:
+        data = json.dumps(payload).encode("utf-8")
+        req = urllib.request.Request(
+            f"{memory_agent_url}/a2a",
+            data=data,
+            headers=headers,
+            method="POST"
+        )
+        with urllib.request.urlopen(req, timeout=min(timeout, 3.0)) as resp:
+            if resp.status == 200:
+                print(f"[SessionEnd] Flush triggered successfully.", file=sys.stderr)
+    except (urllib.error.URLError, urllib.error.HTTPError, OSError, TimeoutError) as e:
+        print(f"[SessionEnd] Flush API call failed: {e}", file=sys.stderr)
+
+
+if __name__ == "__main__":
+    main()

package/hooks/session_start.py

@@ -1,227 +1,227 @@
-#!/usr/bin/env python3
-"""Session start hook - auto-loads relevant context.
-
-This hook runs when a Claude Code session starts and:
-- Loads project info and preferences
-- Retrieves recent decisions and patterns
-- Gets unresolved items from previous sessions
-- Injects relevant context into the session
-- Loads daily logs (Moltbot-inspired)
-- Loads MEMORY.md core facts (Moltbot-inspired)
-
-Configure in Claude Code settings:
-{
-  "hooks": {
-    "SessionStart": ["python /path/to/session_start.py"]
-  }
-}
-
-Output is printed to stdout and injected into Claude's context.
-"""
-import os
-import sys
-import json
-import asyncio
-from datetime import datetime
-from pathlib import Path
-from typing import Dict, Any, List, Optional
-
-sys.path.insert(0, str(Path(__file__).parent.parent))
-
-import httpx
-
-MEMORY_AGENT_URL = os.getenv("MEMORY_AGENT_URL", "http://localhost:8102")
-API_KEY = os.getenv("MEMORY_API_KEY", "")
-
-
-async def call_memory_skill(skill_id: str, params: Dict[str, Any]) -> Optional[Dict[str, Any]]:
-    """Call a memory agent skill."""
-    headers = {"Content-Type": "application/json"}
-    if API_KEY:
-        headers["X-Memory-Key"] = API_KEY
-
-    payload = {
-        "jsonrpc": "2.0",
-        "method": "skills/call",
-        "params": {
-            "skill_id": skill_id,
-            "params": params
-        },
-        "id": f"session-start-{datetime.now().isoformat()}"
-    }
-
-    try:
-        async with httpx.AsyncClient(timeout=10.0) as client:
-            response = await client.post(
-                f"{MEMORY_AGENT_URL}/a2a",
-                json=payload,
-                headers=headers
-            )
-            if response.status_code == 200:
-                data = response.json()
-                return data.get("result", {}).get("result", {})
-    except Exception as e:
-        pass
-    return None
-
-
-async def load_session_context(project_path: str) -> str:
-    """Load all relevant context for a session start."""
-    context_parts = []
-
-    # ============================================================
-    # MOLTBOT-INSPIRED: Load MEMORY.md first (core facts)
-    # ============================================================
-    memory_md = await call_memory_skill("read_memory_md", {
-        "project_path": project_path
-    })
-
-    if memory_md and memory_md.get("exists"):
-        context_parts.append("## Core Facts (from MEMORY.md)")
-        # Include the summary or first part of content
-        content = memory_md.get("content", "")
-        # Truncate if too long
-        if len(content) > 2000:
-            content = content[:2000] + "\n...(truncated)"
-        context_parts.append(content)
-
-    # ============================================================
-    # MOLTBOT-INSPIRED: Load recent daily logs
-    # ============================================================
-    daily_logs = await call_memory_skill("daily_log_read", {
-        "project_path": project_path,
-        "days": 2,
-        "max_chars": 3000
-    })
-
-    if daily_logs and daily_logs.get("logs"):
-        context_parts.append("\n## Recent Activity (from Daily Logs)")
-        for log in daily_logs["logs"]:
-            log_date = log.get("date", "Unknown")
-            log_content = log.get("content", "")
-            # Show just the highlights, not full content
-            if len(log_content) > 1500:
-                log_content = log_content[:1500] + "\n...(truncated)"
-            context_parts.append(f"\n### {log_date}")
-            context_parts.append(log_content)
-
-    # ============================================================
-    # ORIGINAL MEMORY SYSTEM CONTEXT
-    # ============================================================
-
-    # 1. Get project info
-    project_info = await call_memory_skill("get_project_context", {
-        "project_path": project_path,
-        "limit": 5
-    })
-
-    if project_info and project_info.get("project"):
-        proj = project_info["project"]
-        context_parts.append(f"\n## Project: {proj.get('name', project_path)}")
-        if proj.get("tech_stack"):
-            context_parts.append(f"Tech Stack: {', '.join(proj['tech_stack'])}")
-        if proj.get("conventions"):
-            context_parts.append(f"Conventions: {json.dumps(proj['conventions'], indent=2)}")
-
-    # 2. Get recent decisions
-    decisions = await call_memory_skill("semantic_search", {
-        "query": "decision architecture approach",
-        "project_path": project_path,
-        "type": "decision",
-        "limit": 5
-    })
-
-    if decisions and decisions.get("results"):
-        context_parts.append("\n## Recent Decisions")
-        for d in decisions["results"][:3]:
-            context_parts.append(f"- {d['content'][:150]}")
-
-    # 3. Get recent errors (to avoid repeating)
-    errors = await call_memory_skill("semantic_search", {
-        "query": "error bug fix problem",
-        "project_path": project_path,
-        "type": "error",
-        "success_only": True,  # Only get solved errors
-        "limit": 5
-    })
-
-    if errors and errors.get("results"):
-        context_parts.append("\n## Past Errors & Solutions")
-        for e in errors["results"][:3]:
-            context_parts.append(f"- {e['content'][:150]}")
-
-    # 4. Get session handoff (unresolved items)
-    handoff = await call_memory_skill("get_session_handoff", {
-        "project_path": project_path,
-        "include_last_n_sessions": 2
-    })
-
-    if handoff:
-        if handoff.get("unresolved_questions"):
-            context_parts.append("\n## Unresolved from Previous Sessions")
-            for q in handoff["unresolved_questions"][:3]:
-                context_parts.append(f"- {q}")
-
-        if handoff.get("recent_summaries"):
-            context_parts.append("\n## Recent Session Summaries")
-            for s in handoff["recent_summaries"][:2]:
-                context_parts.append(f"- {s.get('summary', '')[:200]}")
-
-    # 5. Get relevant patterns
-    patterns = await call_memory_skill("search_patterns", {
-        "query": "common patterns solutions",
-        "limit": 3
-    })
-
-    if patterns and patterns.get("patterns"):
-        context_parts.append("\n## Useful Patterns")
-        for p in patterns["patterns"][:2]:
-            context_parts.append(f"- **{p['name']}**: {p['solution'][:100]}")
-
-    # 6. Check for anchor conflicts
-    conflicts = await call_memory_skill("get_unresolved_conflicts", {
-        "project_path": project_path,
-        "limit": 3
-    })
-
-    if conflicts and conflicts.get("conflicts"):
-        context_parts.append("\n## Unresolved Fact Conflicts")
-        for c in conflicts["conflicts"]:
-            context_parts.append(f"- {c.get('anchor1_summary', '')} vs {c.get('anchor2_summary', '')}")
-
-    if context_parts:
-        return "\n".join(context_parts)
-    return ""
-
-
-async def main():
-    # Get project path from environment or current directory
-    project_path = os.getenv("PROJECT_PATH") or os.getcwd()
-
-    # Try to read from stdin if available
-    try:
-        if not sys.stdin.isatty():
-            data = sys.stdin.read()
-            if data:
-                hook_data = json.loads(data)
-                project_path = hook_data.get("project_path", project_path)
-    except:
-        pass
-
-    context = await load_session_context(project_path)
-
-    if context:
-        # Output context for Claude to see
-        print("\n<memory-context>")
-        print("# Loaded from Memory System")
-        print(context)
-        print("</memory-context>\n")
-    else:
-        print("\n<memory-context>")
-        print("# No prior context found for this project")
-        print("Starting fresh session.")
-        print("</memory-context>\n")
-
-
-if __name__ == "__main__":
-    asyncio.run(main())
+#!/usr/bin/env python3
+"""Session start hook - auto-loads relevant context.
+
+This hook runs when a Claude Code session starts and:
+- Loads project info and preferences
+- Retrieves recent decisions and patterns
+- Gets unresolved items from previous sessions
+- Injects relevant context into the session
+- Loads daily logs (Moltbot-inspired)
+- Loads MEMORY.md core facts (Moltbot-inspired)
+
+Configure in Claude Code settings:
+{
+  "hooks": {
+    "SessionStart": ["python /path/to/session_start.py"]
+  }
+}
+
+Output is printed to stdout and injected into Claude's context.
+"""
+import os
+import sys
+import json
+import asyncio
+from datetime import datetime
+from pathlib import Path
+from typing import Dict, Any, List, Optional
+
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+import httpx
+
+MEMORY_AGENT_URL = os.getenv("MEMORY_AGENT_URL", "http://localhost:8102")
+API_KEY = os.getenv("MEMORY_API_KEY", "")
+
+
+async def call_memory_skill(skill_id: str, params: Dict[str, Any]) -> Optional[Dict[str, Any]]:
+    """Call a memory agent skill."""
+    headers = {"Content-Type": "application/json"}
+    if API_KEY:
+        headers["X-Memory-Key"] = API_KEY
+
+    payload = {
+        "jsonrpc": "2.0",
+        "method": "skills/call",
+        "params": {
+            "skill_id": skill_id,
+            "params": params
+        },
+        "id": f"session-start-{datetime.now().isoformat()}"
+    }
+
+    try:
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            response = await client.post(
+                f"{MEMORY_AGENT_URL}/a2a",
+                json=payload,
+                headers=headers
+            )
+            if response.status_code == 200:
+                data = response.json()
+                return data.get("result", {}).get("result", {})
+    except Exception as e:
+        pass
+    return None
+
+
+async def load_session_context(project_path: str) -> str:
+    """Load all relevant context for a session start."""
+    context_parts = []
+
+    # ============================================================
+    # MOLTBOT-INSPIRED: Load MEMORY.md first (core facts)
+    # ============================================================
+    memory_md = await call_memory_skill("read_memory_md", {
+        "project_path": project_path
+    })
+
+    if memory_md and memory_md.get("exists"):
+        context_parts.append("## Core Facts (from MEMORY.md)")
+        # Include the summary or first part of content
+        content = memory_md.get("content", "")
+        # Truncate if too long
+        if len(content) > 2000:
+            content = content[:2000] + "\n...(truncated)"
+        context_parts.append(content)
+
+    # ============================================================
+    # MOLTBOT-INSPIRED: Load recent daily logs
+    # ============================================================
+    daily_logs = await call_memory_skill("daily_log_read", {
+        "project_path": project_path,
+        "days": 2,
+        "max_chars": 3000
+    })
+
+    if daily_logs and daily_logs.get("logs"):
+        context_parts.append("\n## Recent Activity (from Daily Logs)")
+        for log in daily_logs["logs"]:
+            log_date = log.get("date", "Unknown")
+            log_content = log.get("content", "")
+            # Show just the highlights, not full content
+            if len(log_content) > 1500:
+                log_content = log_content[:1500] + "\n...(truncated)"
+            context_parts.append(f"\n### {log_date}")
+            context_parts.append(log_content)
+
+    # ============================================================
+    # ORIGINAL MEMORY SYSTEM CONTEXT
+    # ============================================================
+
+    # 1. Get project info
+    project_info = await call_memory_skill("get_project_context", {
+        "project_path": project_path,
+        "limit": 5
+    })
+
+    if project_info and project_info.get("project"):
+        proj = project_info["project"]
+        context_parts.append(f"\n## Project: {proj.get('name', project_path)}")
+        if proj.get("tech_stack"):
+            context_parts.append(f"Tech Stack: {', '.join(proj['tech_stack'])}")
+        if proj.get("conventions"):
+            context_parts.append(f"Conventions: {json.dumps(proj['conventions'], indent=2)}")
+
+    # 2. Get recent decisions
+    decisions = await call_memory_skill("semantic_search", {
+        "query": "decision architecture approach",
+        "project_path": project_path,
+        "type": "decision",
+        "limit": 5
+    })
+
+    if decisions and decisions.get("results"):
+        context_parts.append("\n## Recent Decisions")
+        for d in decisions["results"][:3]:
+            context_parts.append(f"- {d['content'][:150]}")
+
+    # 3. Get recent errors (to avoid repeating)
+    errors = await call_memory_skill("semantic_search", {
+        "query": "error bug fix problem",
+        "project_path": project_path,
+        "type": "error",
+        "success_only": True,  # Only get solved errors
+        "limit": 5
+    })
+
+    if errors and errors.get("results"):
+        context_parts.append("\n## Past Errors & Solutions")
+        for e in errors["results"][:3]:
+            context_parts.append(f"- {e['content'][:150]}")
+
+    # 4. Get session handoff (unresolved items)
+    handoff = await call_memory_skill("get_session_handoff", {
+        "project_path": project_path,
+        "include_last_n_sessions": 2
+    })
+
+    if handoff:
+        if handoff.get("unresolved_questions"):
+            context_parts.append("\n## Unresolved from Previous Sessions")
+            for q in handoff["unresolved_questions"][:3]:
+                context_parts.append(f"- {q}")
+
+        if handoff.get("recent_summaries"):
+            context_parts.append("\n## Recent Session Summaries")
+            for s in handoff["recent_summaries"][:2]:
+                context_parts.append(f"- {s.get('summary', '')[:200]}")
+
+    # 5. Get relevant patterns
+    patterns = await call_memory_skill("search_patterns", {
+        "query": "common patterns solutions",
+        "limit": 3
+    })
+
+    if patterns and patterns.get("patterns"):
+        context_parts.append("\n## Useful Patterns")
+        for p in patterns["patterns"][:2]:
+            context_parts.append(f"- **{p['name']}**: {p['solution'][:100]}")
+
+    # 6. Check for anchor conflicts
+    conflicts = await call_memory_skill("get_unresolved_conflicts", {
+        "project_path": project_path,
+        "limit": 3
+    })
+
+    if conflicts and conflicts.get("conflicts"):
+        context_parts.append("\n## Unresolved Fact Conflicts")
+        for c in conflicts["conflicts"]:
+            context_parts.append(f"- {c.get('anchor1_summary', '')} vs {c.get('anchor2_summary', '')}")
+
+    if context_parts:
+        return "\n".join(context_parts)
+    return ""
+
+
+async def main():
+    # Get project path from environment or current directory
+    project_path = os.getenv("PROJECT_PATH") or os.getcwd()
+
+    # Try to read from stdin if available
+    try:
+        if not sys.stdin.isatty():
+            data = sys.stdin.read()
+            if data:
+                hook_data = json.loads(data)
+                project_path = hook_data.get("project_path", project_path)
+    except:
+        pass
+
+    context = await load_session_context(project_path)
+
+    if context:
+        # Output context for Claude to see
+        print("\n<memory-context>")
+        print("# Loaded from Memory System")
+        print(context)
+        print("</memory-context>\n")
+    else:
+        print("\n<memory-context>")
+        print("# No prior context found for this project")
+        print("Starting fresh session.")
+        print("</memory-context>\n")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())