claude-mpm 4.1.26__py3-none-any.whl → 5.0.9__py3-none-any.whl

claude_mpm/hooks/kuzu_response_hook.py

@@ -0,0 +1,179 @@
+"""
+Kuzu-Memory Response Learning Hook
+===================================
+
+Captures assistant responses and extracts learnings to store in kuzu-memory.
+This completes the bidirectional enrichment cycle:
+- KuzuMemoryHook enriches prompts with memories (READ)
+- KuzuResponseHook stores learnings from responses (WRITE)
+
+WHY: To automatically capture and persist important information from agent
+responses, enabling continuous learning across conversations.
+
+DESIGN DECISIONS:
+- Priority 80 to run late after main processing
+- Reuses KuzuMemoryHook's storage methods for consistency
+- Graceful degradation if kuzu-memory is not available
+- Extracts structured learnings using patterns and AI
+"""
+
+import logging
+from typing import Any, Optional
+
+from claude_mpm.hooks.base_hook import HookContext, HookResult, PostDelegationHook
+from claude_mpm.hooks.kuzu_memory_hook import get_kuzu_memory_hook
+
+logger = logging.getLogger(__name__)
+
+
+class KuzuResponseHook(PostDelegationHook):
+    """
+    Hook that captures agent responses and stores learnings in kuzu-memory.
+
+    This hook:
+    1. Processes agent responses after delegation completes
+    2. Extracts important learnings and information
+    3. Stores memories in kuzu-memory for future retrieval
+    4. Tags memories for better categorization
+    """
+
+    def __init__(self):
+        """Initialize the kuzu-memory response learning hook."""
+        super().__init__(name="kuzu_response_learner", priority=80)
+
+        # Reuse the kuzu-memory hook instance for storage
+        self.kuzu_hook = get_kuzu_memory_hook()
+        self.enabled = self.kuzu_hook.enabled
+
+        if not self.enabled:
+            logger.info(
+                "Kuzu-memory response hook disabled (kuzu-memory not available)"
+            )
+        else:
+            logger.info("Kuzu-memory response learning hook enabled")
+
+    def validate(self, context: HookContext) -> bool:
+        """
+        Validate if hook should process this context.
+
+        Args:
+            context: Hook context to validate
+
+        Returns:
+            True if hook should execute
+        """
+        if not self.enabled:
+            return False
+
+        # Check base validation (enabled, correct hook type, has result)
+        if not super().validate(context):
+            return False
+
+        # Must have result data to extract learnings from
+        result_data = context.data.get("result")
+        if not result_data:
+            return False
+
+        return True
+
+    def execute(self, context: HookContext) -> HookResult:
+        """
+        Extract and store learnings from agent responses.
+
+        Args:
+            context: Hook context containing response data
+
+        Returns:
+            HookResult with success status and metadata
+        """
+        if not self.enabled:
+            return HookResult(success=True, data=context.data, modified=False)
+
+        try:
+            # Extract response content from various possible formats
+            result_data = context.data.get("result", {})
+            response_content = self._extract_response_content(result_data)
+
+            if not response_content:
+                logger.debug("No response content found for learning extraction")
+                return HookResult(success=True, data=context.data, modified=False)
+
+            # Extract and store learnings
+            count = self.kuzu_hook.extract_and_store_learnings(response_content)
+
+            if count > 0:
+                logger.info(f"Stored {count} learnings from agent response")
+                return HookResult(
+                    success=True,
+                    data=context.data,
+                    modified=False,
+                    metadata={"learnings_stored": count, "memory_backend": "kuzu"},
+                )
+
+            return HookResult(success=True, data=context.data, modified=False)
+
+        except Exception as e:
+            logger.error(f"Error in kuzu response hook: {e}")
+            # Don't fail the operation if learning extraction fails
+            return HookResult(
+                success=True,
+                data=context.data,
+                modified=False,
+                error=f"Learning extraction failed: {e}",
+            )
+
+    def _extract_response_content(self, result_data: Any) -> Optional[str]:
+        """
+        Extract response content from various result formats.
+
+        Args:
+            result_data: Result data in various possible formats
+
+        Returns:
+            Extracted response content as string, or None
+        """
+        if not result_data:
+            return None
+
+        # Handle dict format
+        if isinstance(result_data, dict):
+            # Try common response fields
+            for field in ["content", "text", "response", "output", "message"]:
+                if field in result_data:
+                    content = result_data[field]
+                    if isinstance(content, str):
+                        return content
+                    if isinstance(content, dict):
+                        # Recursively extract from nested dict
+                        return self._extract_response_content(content)
+
+            # If dict has no recognizable fields, try converting to string
+            return str(result_data)
+
+        # Handle string format
+        if isinstance(result_data, str):
+            return result_data
+
+        # Handle list format (concatenate items)
+        if isinstance(result_data, list):
+            items = []
+            for item in result_data:
+                extracted = self._extract_response_content(item)
+                if extracted:
+                    items.append(extracted)
+            return "\n\n".join(items) if items else None
+
+        # Fallback to string conversion
+        return str(result_data) if result_data else None
+
+
+# Create a singleton instance
+_kuzu_response_hook = None
+
+
+def get_kuzu_response_hook() -> KuzuResponseHook:
+    """Get the singleton kuzu-memory response hook instance."""
+    global _kuzu_response_hook
+    if _kuzu_response_hook is None:
+        _kuzu_response_hook = KuzuResponseHook()
+    return _kuzu_response_hook

claude_mpm/hooks/memory_integration_hook.py

@@ -16,7 +16,7 @@ import re
 from typing import Dict, List
 
 from claude_mpm.core.config import Config
-from claude_mpm.core.logger import get_logger
+from claude_mpm.core.logging_utils import get_logger
 from claude_mpm.core.shared.config_loader import ConfigLoader
 from claude_mpm.hooks.base_hook import (
     HookContext,

claude_mpm/hooks/session_resume_hook.py

@@ -0,0 +1,121 @@
+"""Session Resume Startup Hook.
+
+WHY: This hook automatically checks for paused sessions on PM startup and displays
+resume context to help users continue their work seamlessly.
+
+DESIGN DECISIONS:
+- Runs automatically on PM startup
+- Non-blocking: doesn't prevent PM from starting if check fails
+- Displays context to stdout for user visibility
+- Integrates with existing session pause/resume infrastructure
+"""
+
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+from claude_mpm.core.logger import get_logger
+from claude_mpm.services.cli.session_resume_helper import SessionResumeHelper
+
+logger = get_logger(__name__)
+
+
+class SessionResumeStartupHook:
+    """Hook for automatic session resume detection on PM startup."""
+
+    def __init__(self, project_path: Optional[Path] = None):
+        """Initialize the session resume hook.
+
+        Args:
+            project_path: Project root path (default: current directory)
+        """
+        self.project_path = project_path or Path.cwd()
+        self.resume_helper = SessionResumeHelper(self.project_path)
+        self._session_displayed = False
+
+    def on_pm_startup(self) -> Optional[Dict[str, Any]]:
+        """Execute on PM startup to check for paused sessions.
+
+        Returns:
+            Session data if paused session found, None otherwise
+        """
+        try:
+            # Check if we already displayed a session in this process
+            if self._session_displayed:
+                logger.debug("Session already displayed, skipping")
+                return None
+
+            # Check for paused sessions
+            session_data = self.resume_helper.check_and_display_resume_prompt()
+
+            if session_data:
+                self._session_displayed = True
+                logger.info("Paused session context displayed to user")
+
+            return session_data
+
+        except Exception as e:
+            logger.error(f"Failed to check for paused sessions: {e}", exc_info=True)
+            return None
+
+    def get_session_count(self) -> int:
+        """Get count of paused sessions.
+
+        Returns:
+            Number of paused sessions
+        """
+        try:
+            return self.resume_helper.get_session_count()
+        except Exception as e:
+            logger.error(f"Failed to get session count: {e}")
+            return 0
+
+    def clear_displayed_session(self, session_data: Dict[str, Any]) -> bool:
+        """Clear a session after it has been displayed and user has acknowledged.
+
+        Args:
+            session_data: Session data to clear
+
+        Returns:
+            True if successfully cleared, False otherwise
+        """
+        try:
+            return self.resume_helper.clear_session(session_data)
+        except Exception as e:
+            logger.error(f"Failed to clear session: {e}")
+            return False
+
+
+# Global hook instance
+_session_resume_hook: Optional[SessionResumeStartupHook] = None
+
+
+def get_session_resume_hook(
+    project_path: Optional[Path] = None,
+) -> SessionResumeStartupHook:
+    """Get or create the global session resume hook instance.
+
+    Args:
+        project_path: Project root path (default: current directory)
+
+    Returns:
+        SessionResumeStartupHook instance
+    """
+    global _session_resume_hook
+
+    if _session_resume_hook is None:
+        _session_resume_hook = SessionResumeStartupHook(project_path)
+        logger.debug("Created session resume hook instance")
+
+    return _session_resume_hook
+
+
+def trigger_session_resume_check() -> Optional[Dict[str, Any]]:
+    """Trigger a session resume check (convenience function).
+
+    This is the main entry point for PM startup integration.
+
+    Returns:
+        Session data if found, None otherwise
+    """
+    hook = get_session_resume_hook()
+    return hook.on_pm_startup()

claude_mpm/hooks/templates/pre_tool_use_simple.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env python3
+"""
+Simple PreToolUse Hook Example for Claude Code v2.0.30+
+
+This is a minimal example showing the basic structure of a PreToolUse hook
+that can modify tool inputs.
+
+To use this hook:
+1. Copy this file to ~/.claude-mpm/hooks/
+2. Make it executable: chmod +x ~/.claude-mpm/hooks/pre_tool_use_simple.py
+3. Add to ~/.claude/settings.json:
+
+{
+  "hooks": {
+    "PreToolUse": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "/Users/YOUR_USERNAME/.claude-mpm/hooks/pre_tool_use_simple.py",
+            "modifyInput": true
+          }
+        ]
+      }
+    ]
+  }
+}
+"""
+
+import json
+import sys
+
+
+def main():
+    """Process PreToolUse event and optionally modify input."""
+    try:
+        # Read event from stdin
+        event_data = sys.stdin.read()
+        if not event_data.strip():
+            print(json.dumps({"action": "continue"}))
+            return
+
+        event = json.loads(event_data)
+        tool_name = event.get("tool_name", "")
+        tool_input = event.get("tool_input", {})
+
+        # Example: Add line numbers to all Grep operations
+        if tool_name == "Grep" and "-n" not in tool_input:
+            modified_input = tool_input.copy()
+            modified_input["-n"] = True
+            print(json.dumps({"action": "continue", "tool_input": modified_input}))
+            return
+
+        # Example: Block operations on .env files
+        if tool_name in ["Write", "Edit", "Read"]:
+            file_path = tool_input.get("file_path", "")
+            if ".env" in file_path:
+                print(
+                    json.dumps(
+                        {
+                            "action": "block",
+                            "message": "Access to .env file blocked for security",
+                        }
+                    )
+                )
+                return
+
+        # Default: continue without modification
+        print(json.dumps({"action": "continue"}))
+
+    except Exception:
+        # Always continue on error to avoid blocking Claude
+        print(json.dumps({"action": "continue"}))
+
+
+if __name__ == "__main__":
+    main()