claude-mpm 4.7.5__py3-none-any.whl → 4.7.7__py3-none-any.whl

claude_mpm/VERSION

@@ -1 +1 @@
-4.7.5
+4.7.7

claude_mpm/cli/commands/configure.py

@@ -374,7 +374,9 @@ class ConfigureCommand(BaseCommand):
         self.console.print(menu_panel)
         self.console.print()
 
-        return Prompt.ask("[bold cyan]Select an option[/bold cyan]", default="q")
+        choice = Prompt.ask("[bold cyan]Select an option[/bold cyan]", default="q")
+        # Strip whitespace to handle leading/trailing spaces
+        return choice.strip().lower()
 
     def _manage_agents(self) -> None:
         """Agent management interface."""
@@ -388,12 +390,38 @@ class ConfigureCommand(BaseCommand):
 
             # Show agent menu
             self.console.print("\n[bold]Agent Management Options:[/bold]")
-            self.console.print("  [cyan][e][/cyan] Enable an agent")
-            self.console.print("  [cyan][d][/cyan] Disable an agent")
-            self.console.print("  [cyan][c][/cyan] Customize agent template")
-            self.console.print("  [cyan][v][/cyan] View agent details")
-            self.console.print("  [cyan][r][/cyan] Reset agent to defaults")
-            self.console.print("  [cyan][b][/cyan] Back to main menu")
+
+            # Use Text objects to properly display shortcuts with styling
+            text_e = Text("  ")
+            text_e.append("[e]", style="cyan bold")
+            text_e.append(" Enable an agent")
+            self.console.print(text_e)
+
+            text_d = Text("  ")
+            text_d.append("[d]", style="cyan bold")
+            text_d.append(" Disable an agent")
+            self.console.print(text_d)
+
+            text_c = Text("  ")
+            text_c.append("[c]", style="cyan bold")
+            text_c.append(" Customize agent template")
+            self.console.print(text_c)
+
+            text_v = Text("  ")
+            text_v.append("[v]", style="cyan bold")
+            text_v.append(" View agent details")
+            self.console.print(text_v)
+
+            text_r = Text("  ")
+            text_r.append("[r]", style="cyan bold")
+            text_r.append(" Reset agent to defaults")
+            self.console.print(text_r)
+
+            text_b = Text("  ")
+            text_b.append("[b]", style="cyan bold")
+            text_b.append(" Back to main menu")
+            self.console.print(text_b)
+
             self.console.print()
 
             choice = Prompt.ask("[bold cyan]Select an option[/bold cyan]", default="b")
@@ -425,7 +453,7 @@ class ConfigureCommand(BaseCommand):
         table.add_column("ID", style="dim", width=3)
         table.add_column("Name", style="cyan", width=22)
         table.add_column("Status", width=12)
-        table.add_column("Description", style="white", width=45)
+        table.add_column("Description", style="bold cyan", width=45)
         table.add_column("Model/Tools", style="dim", width=20)
 
         for idx, agent in enumerate(agents, 1):
@@ -456,12 +484,11 @@ class ConfigureCommand(BaseCommand):
                 except Exception:
                     tools_display = "Default"
 
-            # Truncate description for table display
-            desc_display = (
-                agent.description[:42] + "..."
-                if len(agent.description) > 42
-                else agent.description
-            )
+            # Truncate description for table display with bright styling
+            if len(agent.description) > 42:
+                desc_display = f"[cyan]{agent.description[:42]}[/cyan][dim]...[/dim]"
+            else:
+                desc_display = f"[cyan]{agent.description}[/cyan]"
 
             table.add_row(str(idx), agent.name, status, desc_display, tools_display)
 
@@ -612,14 +639,41 @@ class ConfigureCommand(BaseCommand):
         # Editing options
         self.console.print("[bold]Editing Options:[/bold]")
         if not is_system:
-            self.console.print("  [cyan][1][/cyan] Edit in external editor")
-            self.console.print("  [cyan][2][/cyan] Add/modify a field")
-            self.console.print("  [cyan][3][/cyan] Remove a field")
-            self.console.print("  [cyan][4][/cyan] Reset to defaults")
+            text_1 = Text("  ")
+            text_1.append("[1]", style="cyan bold")
+            text_1.append(" Edit in external editor")
+            self.console.print(text_1)
+
+            text_2 = Text("  ")
+            text_2.append("[2]", style="cyan bold")
+            text_2.append(" Add/modify a field")
+            self.console.print(text_2)
+
+            text_3 = Text("  ")
+            text_3.append("[3]", style="cyan bold")
+            text_3.append(" Remove a field")
+            self.console.print(text_3)
+
+            text_4 = Text("  ")
+            text_4.append("[4]", style="cyan bold")
+            text_4.append(" Reset to defaults")
+            self.console.print(text_4)
         else:
-            self.console.print("  [cyan][1][/cyan] Create customized copy")
-            self.console.print("  [cyan][2][/cyan] View full template")
-        self.console.print("  [cyan][b][/cyan] Back")
+            text_1 = Text("  ")
+            text_1.append("[1]", style="cyan bold")
+            text_1.append(" Create customized copy")
+            self.console.print(text_1)
+
+            text_2 = Text("  ")
+            text_2.append("[2]", style="cyan bold")
+            text_2.append(" View full template")
+            self.console.print(text_2)
+
+        text_b = Text("  ")
+        text_b.append("[b]", style="cyan bold")
+        text_b.append(" Back")
+        self.console.print(text_b)
+
         self.console.print()
 
         choice = Prompt.ask("[bold cyan]Select an option[/bold cyan]", default="b")
@@ -821,6 +875,61 @@ class ConfigureCommand(BaseCommand):
         with self.console.pager():
             self.console.print(syntax)
 
+    def _reset_agent_defaults(self, agents: List[AgentConfig]) -> None:
+        """Reset an agent to default enabled state and remove custom template.
+
+        This method:
+        - Prompts for agent ID
+        - Resets agent to enabled state
+        - Removes any custom template overrides
+        - Shows success/error messages
+        """
+        agent_id = Prompt.ask("Enter agent ID to reset to defaults")
+
+        try:
+            idx = int(agent_id) - 1
+            if 0 <= idx < len(agents):
+                agent = agents[idx]
+
+                # Confirm the reset action
+                if not Confirm.ask(
+                    f"[yellow]Reset '{agent.name}' to defaults? This will:[/yellow]\n"
+                    "  - Enable the agent\n"
+                    "  - Remove custom template (if any)\n"
+                    "[yellow]Continue?[/yellow]"
+                ):
+                    self.console.print("[yellow]Reset cancelled.[/yellow]")
+                    Prompt.ask("Press Enter to continue")
+                    return
+
+                # Enable the agent
+                self.agent_manager.set_agent_enabled(agent.name, True)
+
+                # Remove custom template if exists
+                template_path = self._get_agent_template_path(agent.name)
+                if template_path.exists() and not str(template_path).startswith(
+                    str(self.agent_manager.templates_dir)
+                ):
+                    # This is a custom template, remove it
+                    template_path.unlink(missing_ok=True)
+                    self.console.print(
+                        f"[green]✓ Removed custom template for '{agent.name}'[/green]"
+                    )
+
+                self.console.print(
+                    f"[green]✓ Agent '{agent.name}' reset to defaults![/green]"
+                )
+                self.console.print(
+                    "[dim]Agent is now enabled with system template.[/dim]"
+                )
+            else:
+                self.console.print("[red]Invalid agent ID.[/red]")
+
+        except ValueError:
+            self.console.print("[red]Invalid input. Please enter a number.[/red]")
+
+        Prompt.ask("Press Enter to continue")
+
     def _view_agent_details(self, agents: List[AgentConfig]) -> None:
         """View detailed information about an agent."""
         agent_id = Prompt.ask("Enter agent ID to view")
@@ -917,11 +1026,32 @@ class ConfigureCommand(BaseCommand):
 
             # Show behavior menu
             self.console.print("\n[bold]Options:[/bold]")
-            self.console.print("  [cyan][1][/cyan] Edit identity configuration")
-            self.console.print("  [cyan][2][/cyan] Edit workflow configuration")
-            self.console.print("  [cyan][3][/cyan] Import behavior file")
-            self.console.print("  [cyan][4][/cyan] Export behavior file")
-            self.console.print("  [cyan][b][/cyan] Back to main menu")
+
+            text_1 = Text("  ")
+            text_1.append("[1]", style="cyan bold")
+            text_1.append(" Edit identity configuration")
+            self.console.print(text_1)
+
+            text_2 = Text("  ")
+            text_2.append("[2]", style="cyan bold")
+            text_2.append(" Edit workflow configuration")
+            self.console.print(text_2)
+
+            text_3 = Text("  ")
+            text_3.append("[3]", style="cyan bold")
+            text_3.append(" Import behavior file")
+            self.console.print(text_3)
+
+            text_4 = Text("  ")
+            text_4.append("[4]", style="cyan bold")
+            text_4.append(" Export behavior file")
+            self.console.print(text_4)
+
+            text_b = Text("  ")
+            text_b.append("[b]", style="cyan bold")
+            text_b.append(" Back to main menu")
+            self.console.print(text_b)
+
             self.console.print()
 
             choice = Prompt.ask("[bold cyan]Select an option[/bold cyan]", default="b")
@@ -1208,9 +1338,22 @@ class ConfigureCommand(BaseCommand):
         self.console.print(table)
         self.console.print("\n[bold]Commands:[/bold]")
         self.console.print("  Enter service IDs to toggle (e.g., '1,3' or '1-4')")
-        self.console.print("  [cyan][a][/cyan] Enable all")
-        self.console.print("  [cyan][n][/cyan] Disable all")
-        self.console.print("  [cyan][b][/cyan] Back to previous menu")
+
+        text_a = Text("  ")
+        text_a.append("[a]", style="cyan bold")
+        text_a.append(" Enable all")
+        self.console.print(text_a)
+
+        text_n = Text("  ")
+        text_n.append("[n]", style="cyan bold")
+        text_n.append(" Disable all")
+        self.console.print(text_n)
+
+        text_b = Text("  ")
+        text_b.append("[b]", style="cyan bold")
+        text_b.append(" Back to previous menu")
+        self.console.print(text_b)
+
         self.console.print()
 
         choice = Prompt.ask("[bold cyan]Toggle services[/bold cyan]", default="b")
@@ -1275,9 +1418,22 @@ class ConfigureCommand(BaseCommand):
         self.console.print(table)
         self.console.print("\n[bold]Commands:[/bold]")
         self.console.print("  Enter service IDs to toggle (e.g., '1,3' or '1-4')")
-        self.console.print("  [cyan][a][/cyan] Enable all")
-        self.console.print("  [cyan][n][/cyan] Disable all")
-        self.console.print("  [cyan][b][/cyan] Back to previous menu")
+
+        text_a = Text("  ")
+        text_a.append("[a]", style="cyan bold")
+        text_a.append(" Enable all")
+        self.console.print(text_a)
+
+        text_n = Text("  ")
+        text_n.append("[n]", style="cyan bold")
+        text_n.append(" Disable all")
+        self.console.print(text_n)
+
+        text_b = Text("  ")
+        text_b.append("[b]", style="cyan bold")
+        text_b.append(" Back to previous menu")
+        self.console.print(text_b)
+
         self.console.print()
 
         choice = Prompt.ask("[bold cyan]Toggle services[/bold cyan]", default="b")
@@ -1331,7 +1487,7 @@ class ConfigureCommand(BaseCommand):
             table.add_column("ID", style="dim", width=5)
             table.add_column("Agent", style="cyan", width=25)
             table.add_column("Status", width=15)
-            table.add_column("Description", style="white", width=45)
+            table.add_column("Description", style="bold cyan", width=45)
 
             for idx, agent in enumerate(agents, 1):
                 # Agent is ENABLED if NOT in disabled list
@@ -1341,11 +1497,13 @@ class ConfigureCommand(BaseCommand):
                     if is_enabled
                     else "[red]✗ Disabled[/red]"
                 )
-                desc_display = (
-                    agent.description[:42] + "..."
-                    if len(agent.description) > 42
-                    else agent.description
-                )
+                # Format description with bright styling
+                if len(agent.description) > 42:
+                    desc_display = (
+                        f"[cyan]{agent.description[:42]}[/cyan][dim]...[/dim]"
+                    )
+                else:
+                    desc_display = f"[cyan]{agent.description}[/cyan]"
                 table.add_row(str(idx), agent.name, status, desc_display)
 
             self.console.print(table)

claude_mpm/hooks/__init__.py

@@ -1,13 +1,19 @@
 """Hook system for claude-mpm."""
 
 from .base_hook import BaseHook, HookContext, HookResult, HookType
+from .kuzu_enrichment_hook import KuzuEnrichmentHook, get_kuzu_enrichment_hook
 from .kuzu_memory_hook import KuzuMemoryHook, get_kuzu_memory_hook
+from .kuzu_response_hook import KuzuResponseHook, get_kuzu_response_hook
 
 __all__ = [
     "BaseHook",
     "HookContext",
     "HookResult",
     "HookType",
+    "KuzuEnrichmentHook",
     "KuzuMemoryHook",
+    "KuzuResponseHook",
+    "get_kuzu_enrichment_hook",
     "get_kuzu_memory_hook",
+    "get_kuzu_response_hook",
 ]

claude_mpm/hooks/kuzu_enrichment_hook.py

@@ -0,0 +1,263 @@
+"""
+Kuzu-Memory Pre-Delegation Enrichment Hook
+==========================================
+
+Enriches agent delegation context with relevant memories from kuzu-memory
+before the agent receives the task. This is the READ side of bidirectional
+enrichment.
+
+WHY: Agents need access to relevant historical knowledge when performing tasks.
+This hook retrieves memories from kuzu-memory and injects them into the
+delegation context.
+
+DESIGN DECISIONS:
+- Priority 10 to run early, before other context modifications
+- Reuses KuzuMemoryHook's retrieval methods for consistency
+- Injects memories as a dedicated section in agent context
+- Falls back gracefully if kuzu-memory is not available
+"""
+
+import logging
+from typing import Any, Dict, Optional
+
+from claude_mpm.hooks.base_hook import HookContext, HookResult, PreDelegationHook
+from claude_mpm.hooks.kuzu_memory_hook import get_kuzu_memory_hook
+
+logger = logging.getLogger(__name__)
+
+
+class KuzuEnrichmentHook(PreDelegationHook):
+    """
+    Hook that enriches agent delegation context with kuzu-memory.
+
+    This hook:
+    1. Extracts the task/prompt from delegation context
+    2. Retrieves relevant memories from kuzu-memory
+    3. Injects memories into agent context
+    4. Formats memories for optimal agent understanding
+    """
+
+    def __init__(self):
+        """Initialize the kuzu-memory enrichment hook."""
+        super().__init__(name="kuzu_memory_enrichment", priority=10)
+
+        # Reuse the kuzu-memory hook instance for retrieval
+        self.kuzu_hook = get_kuzu_memory_hook()
+        self.enabled = self.kuzu_hook.enabled
+
+        if not self.enabled:
+            logger.info(
+                "Kuzu-memory enrichment hook disabled (kuzu-memory not available)"
+            )
+        else:
+            logger.info("Kuzu-memory enrichment 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 agent)
+        if not super().validate(context):
+            return False
+
+        # Must have agent and context data
+        if not context.data.get("agent"):
+            return False
+
+        return True
+
+    def execute(self, context: HookContext) -> HookResult:
+        """
+        Enrich delegation context with relevant memories.
+
+        Args:
+            context: Hook context containing delegation data
+
+        Returns:
+            HookResult with enriched context
+        """
+        if not self.enabled:
+            return HookResult(success=True, data=context.data, modified=False)
+
+        try:
+            # Extract query for memory retrieval
+            query = self._extract_query_from_context(context.data)
+
+            if not query:
+                logger.debug("No query extracted from context for memory retrieval")
+                return HookResult(success=True, data=context.data, modified=False)
+
+            # Retrieve relevant memories
+            memories = self.kuzu_hook._retrieve_memories(query)
+
+            if not memories:
+                logger.debug("No relevant memories found")
+                return HookResult(success=True, data=context.data, modified=False)
+
+            # Enrich context with memories
+            enriched_data = self._enrich_delegation_context(
+                context.data, memories, context.data.get("agent", "Agent")
+            )
+
+            logger.info(
+                f"Enriched delegation context with {len(memories)} memories for {context.data.get('agent')}"
+            )
+
+            return HookResult(
+                success=True,
+                data=enriched_data,
+                modified=True,
+                metadata={
+                    "memories_added": len(memories),
+                    "memory_source": "kuzu",
+                    "agent": context.data.get("agent"),
+                },
+            )
+
+        except Exception as e:
+            logger.error(f"Error in kuzu enrichment hook: {e}")
+            # Don't fail the delegation if memory enrichment fails
+            return HookResult(
+                success=True,
+                data=context.data,
+                modified=False,
+                error=f"Memory enrichment failed: {e}",
+            )
+
+    def _extract_query_from_context(self, data: Dict[str, Any]) -> Optional[str]:
+        """
+        Extract query text for memory retrieval.
+
+        Args:
+            data: Delegation context data
+
+        Returns:
+            Query string or None
+        """
+        # Try various context fields
+        delegation_context = data.get("context", {})
+
+        # Handle string context
+        if isinstance(delegation_context, str):
+            return delegation_context
+
+        # Handle dict context
+        if isinstance(delegation_context, dict):
+            # Try common fields
+            for field in ["prompt", "task", "query", "user_request", "description"]:
+                if field in delegation_context:
+                    value = delegation_context[field]
+                    if isinstance(value, str):
+                        return value
+
+            # If no specific field, join all string values
+            text_parts = [
+                str(v) for v in delegation_context.values() if isinstance(v, str)
+            ]
+            if text_parts:
+                return " ".join(text_parts)
+
+        # Fallback: try to get task or instruction directly
+        if "task" in data and isinstance(data["task"], str):
+            return data["task"]
+
+        if "instruction" in data and isinstance(data["instruction"], str):
+            return data["instruction"]
+
+        return None
+
+    def _enrich_delegation_context(
+        self, original_data: Dict[str, Any], memories: list, agent_name: str
+    ) -> Dict[str, Any]:
+        """
+        Enrich delegation context with memories.
+
+        Args:
+            original_data: Original delegation data
+            memories: Retrieved memories
+            agent_name: Name of the agent
+
+        Returns:
+            Enriched delegation data
+        """
+        # Format memories
+        memory_section = self._format_memory_section(memories, agent_name)
+
+        # Create enriched data
+        enriched_data = original_data.copy()
+
+        # Get existing context
+        delegation_context = enriched_data.get("context", {})
+        if isinstance(delegation_context, str):
+            delegation_context = {"prompt": delegation_context}
+
+        # Add memory section
+        if isinstance(delegation_context, dict):
+            # Prepend memory section to context
+            delegation_context["kuzu_memories"] = memory_section
+
+            # If there's a main prompt/task, prepend memory note
+            for field in ["prompt", "task", "instruction"]:
+                if field in delegation_context and isinstance(
+                    delegation_context[field], str
+                ):
+                    delegation_context[field] = (
+                        f"{memory_section}\n\n{delegation_context[field]}"
+                    )
+                    break
+        else:
+            # If context is not dict, create new dict with memory
+            delegation_context = {
+                "kuzu_memories": memory_section,
+                "original_context": delegation_context,
+            }
+
+        enriched_data["context"] = delegation_context
+        enriched_data["_kuzu_enriched"] = True
+
+        return enriched_data
+
+    def _format_memory_section(self, memories: list, agent_name: str) -> str:
+        """
+        Format memories into a readable section.
+
+        Args:
+            memories: List of memory dictionaries
+            agent_name: Name of the agent
+
+        Returns:
+            Formatted memory section
+        """
+        memory_text = self.kuzu_hook._format_memories(memories)
+
+        return f"""
+=== RELEVANT KNOWLEDGE FROM KUZU MEMORY ===
+{agent_name}, you have access to these relevant memories from the knowledge graph:
+
+{memory_text}
+
+INSTRUCTIONS: Review these memories before proceeding. Apply learned patterns and avoid known mistakes.
+Use this knowledge to provide more informed and contextual responses.
+===========================================
+"""
+
+
+# Create a singleton instance
+_kuzu_enrichment_hook = None
+
+
+def get_kuzu_enrichment_hook() -> KuzuEnrichmentHook:
+    """Get the singleton kuzu-memory enrichment hook instance."""
+    global _kuzu_enrichment_hook
+    if _kuzu_enrichment_hook is None:
+        _kuzu_enrichment_hook = KuzuEnrichmentHook()
+    return _kuzu_enrichment_hook

claude_mpm/hooks/kuzu_memory_hook.py

@@ -157,9 +157,13 @@ class KuzuMemoryHook(SubmitHook):
             List of relevant memory dictionaries
         """
         try:
-            # Use kuzu-memory recall command
+            # Type narrowing: ensure kuzu_memory_cmd is not None before using
+            if self.kuzu_memory_cmd is None:
+                return []
+
+            # Use kuzu-memory recall command (v1.2.7+ syntax)
             result = subprocess.run(
-                [self.kuzu_memory_cmd, "recall", query, "--format", "json"],
+                [self.kuzu_memory_cmd, "memory", "recall", query, "--format", "json"],
                 capture_output=True,
                 text=True,
                 timeout=5,
@@ -168,10 +172,21 @@ class KuzuMemoryHook(SubmitHook):
             )
 
             if result.returncode == 0 and result.stdout:
-                memories = json.loads(result.stdout)
-                return memories if isinstance(memories, list) else []
-
-        except (subprocess.TimeoutExpired, json.JSONDecodeError, Exception) as e:
+                try:
+                    # Parse JSON with strict=False to handle control characters
+                    data = json.loads(result.stdout, strict=False)
+                    # v1.2.7 returns dict with 'memories' key, not array
+                    if isinstance(data, dict):
+                        memories = data.get("memories", [])
+                    else:
+                        memories = data if isinstance(data, list) else []
+                    return memories
+                except json.JSONDecodeError as e:
+                    logger.warning(f"Failed to parse kuzu-memory JSON output: {e}")
+                    logger.debug(f"Raw output: {result.stdout[:200]}")
+                    return []  # Graceful fallback
+
+        except (subprocess.TimeoutExpired, Exception) as e:
             logger.debug(f"Memory retrieval failed: {e}")
 
         return []
@@ -255,12 +270,12 @@ Note: Use the memories above to provide more informed and contextual responses.
         Returns:
             True if storage was successful
         """
-        if not self.enabled:
+        if not self.enabled or self.kuzu_memory_cmd is None:
             return False
 
         try:
-            # Use kuzu-memory remember command (synchronous)
-            cmd = [self.kuzu_memory_cmd, "remember", content]
+            # Use kuzu-memory store command (v1.2.7+ syntax)
+            cmd = [self.kuzu_memory_cmd, "memory", "store", content]
 
             # Execute store command in project directory
             result = subprocess.run(
@@ -273,8 +288,10 @@ Note: Use the memories above to provide more informed and contextual responses.
             )
 
             if result.returncode == 0:
-                logger.debug(f"Stored memory: {content[:50]}...")
+                logger.debug(f"Stored memory in kuzu: {content[:50]}...")
                 return True
+            logger.warning(f"Failed to store memory in kuzu: {result.stderr}")
+            return False
 
         except Exception as e:
             logger.error(f"Failed to store memory: {e}")