claude-mpm 5.6.4__py3-none-any.whl → 5.6.30__py3-none-any.whl

claude_mpm/commander/frameworks/base.py

@@ -19,6 +19,7 @@ class InstanceInfo:
         pane_target: Tmux pane target (e.g., "%1")
         git_branch: Current git branch if project is a git repo
         git_status: Git status summary if project is a git repo
+        connected: Whether instance has an active adapter connection
 
     Example:
         >>> info = InstanceInfo(
@@ -28,7 +29,8 @@ class InstanceInfo:
         ...     tmux_session="mpm-commander",
         ...     pane_target="%1",
         ...     git_branch="main",
-        ...     git_status="clean"
+        ...     git_status="clean",
+        ...     connected=True
         ... )
     """
 
@@ -39,6 +41,7 @@ class InstanceInfo:
     pane_target: str
     git_branch: Optional[str] = None
     git_status: Optional[str] = None
+    connected: bool = False
 
 
 class BaseFramework(ABC):

claude_mpm/commander/instance_manager.py

@@ -167,6 +167,20 @@ class InstanceManager:
         startup_cmd = framework_obj.get_startup_command(project_path)
         self.orchestrator.send_keys(pane_target, startup_cmd)
 
+        # Create communication adapter for the instance (only for Claude Code for now)
+        # Do this BEFORE creating InstanceInfo so we can set connected=True
+        has_adapter = False
+        if framework == "cc":
+            runtime_adapter = ClaudeCodeAdapter()
+            comm_adapter = ClaudeCodeCommunicationAdapter(
+                orchestrator=self.orchestrator,
+                pane_target=pane_target,
+                runtime_adapter=runtime_adapter,
+            )
+            self._adapters[name] = comm_adapter
+            has_adapter = True
+            logger.debug(f"Created communication adapter for instance '{name}'")
+
         # Create instance info
         instance = InstanceInfo(
             name=name,
@@ -176,22 +190,12 @@ class InstanceManager:
             pane_target=pane_target,
             git_branch=git_branch,
             git_status=git_status,
+            connected=has_adapter,
         )
 
         # Track instance
         self._instances[name] = instance
 
-        # Create communication adapter for the instance (only for Claude Code for now)
-        if framework == "cc":
-            runtime_adapter = ClaudeCodeAdapter()
-            comm_adapter = ClaudeCodeCommunicationAdapter(
-                orchestrator=self.orchestrator,
-                pane_target=pane_target,
-                runtime_adapter=runtime_adapter,
-            )
-            self._adapters[name] = comm_adapter
-            logger.debug(f"Created communication adapter for instance '{name}'")
-
         logger.info(
             f"Started instance '{name}' with framework '{framework}' at {project_path}"
         )
@@ -226,6 +230,7 @@ class InstanceManager:
         # Remove adapter if exists
         if name in self._adapters:
             del self._adapters[name]
+            instance.connected = False
             logger.debug(f"Removed adapter for instance '{name}'")
 
         # Remove from tracking
@@ -335,3 +340,111 @@ class InstanceManager:
             ...         print(chunk, end='')
         """
         return self._adapters.get(name)
+
+    async def rename_instance(self, old_name: str, new_name: str) -> bool:
+        """Rename an instance.
+
+        Args:
+            old_name: Current instance name
+            new_name: New instance name
+
+        Returns:
+            True if renamed successfully
+
+        Raises:
+            InstanceNotFoundError: If old_name doesn't exist
+            InstanceAlreadyExistsError: If new_name already exists
+
+        Example:
+            >>> manager = InstanceManager(orchestrator)
+            >>> await manager.rename_instance("myapp", "myapp-v2")
+            True
+        """
+        # Validate old_name exists
+        if old_name not in self._instances:
+            raise InstanceNotFoundError(old_name)
+
+        # Validate new_name doesn't exist
+        if new_name in self._instances:
+            raise InstanceAlreadyExistsError(new_name)
+
+        # Get instance and update name
+        instance = self._instances[old_name]
+        instance.name = new_name
+
+        # Update _instances dict (remove old key, add new)
+        del self._instances[old_name]
+        self._instances[new_name] = instance
+
+        # Update _adapters dict if exists
+        if old_name in self._adapters:
+            adapter = self._adapters[old_name]
+            del self._adapters[old_name]
+            self._adapters[new_name] = adapter
+            logger.debug(f"Moved adapter from '{old_name}' to '{new_name}'")
+
+        logger.info(f"Renamed instance from '{old_name}' to '{new_name}'")
+
+        return True
+
+    async def close_instance(self, name: str) -> bool:
+        """Close and remove an instance.
+
+        Alias for stop_instance that provides clearer semantics for closing.
+
+        Args:
+            name: Instance name to close
+
+        Returns:
+            True if closed successfully
+
+        Raises:
+            InstanceNotFoundError: If instance not found
+
+        Example:
+            >>> manager = InstanceManager(orchestrator)
+            >>> await manager.close_instance("myapp")
+            True
+        """
+        return await self.stop_instance(name)
+
+    async def disconnect_instance(self, name: str) -> bool:
+        """Disconnect from an instance without closing it.
+
+        The instance keeps running but we stop communication.
+        Removes the adapter while keeping the instance tracked.
+
+        Args:
+            name: Instance name to disconnect from
+
+        Returns:
+            True if disconnected successfully
+
+        Raises:
+            InstanceNotFoundError: If instance not found
+
+        Example:
+            >>> manager = InstanceManager(orchestrator)
+            >>> await manager.disconnect_instance("myapp")
+            True
+            >>> # Instance still running, but no adapter connection
+            >>> adapter = manager.get_adapter("myapp")
+            >>> print(adapter)
+            None
+        """
+        # Validate instance exists
+        if name not in self._instances:
+            raise InstanceNotFoundError(name)
+
+        instance = self._instances[name]
+
+        # Remove adapter if exists (but keep instance)
+        if name in self._adapters:
+            # Could add cleanup here if adapter has resources to close
+            del self._adapters[name]
+            instance.connected = False
+            logger.info(f"Disconnected from instance '{name}' (instance still running)")
+        else:
+            logger.debug(f"No adapter to disconnect for instance '{name}'")
+
+        return True

claude_mpm/commander/memory/__init__.py

@@ -0,0 +1,45 @@
+"""Conversation memory system for Commander.
+
+This module provides semantic search, storage, and context compression
+for all Claude Code instance conversations.
+
+Key Components:
+    - ConversationStore: CRUD operations for conversations
+    - EmbeddingService: Generate vector embeddings
+    - SemanticSearch: Query conversations semantically
+    - ContextCompressor: Summarize conversations for context
+    - EntityExtractor: Extract files, functions, errors
+
+Example:
+    >>> from claude_mpm.commander.memory import (
+    ...     ConversationStore,
+    ...     EmbeddingService,
+    ...     SemanticSearch,
+    ...     ContextCompressor,
+    ... )
+    >>> store = ConversationStore()
+    >>> embeddings = EmbeddingService()
+    >>> search = SemanticSearch(store, embeddings)
+    >>> results = await search.search("how did we fix the login bug?")
+"""
+
+from .compression import ContextCompressor
+from .embeddings import EmbeddingService
+from .entities import Entity, EntityExtractor, EntityType
+from .integration import MemoryIntegration
+from .search import SearchResult, SemanticSearch
+from .store import Conversation, ConversationMessage, ConversationStore
+
+__all__ = [
+    "ContextCompressor",
+    "Conversation",
+    "ConversationMessage",
+    "ConversationStore",
+    "EmbeddingService",
+    "Entity",
+    "EntityExtractor",
+    "EntityType",
+    "MemoryIntegration",
+    "SearchResult",
+    "SemanticSearch",
+]

claude_mpm/commander/memory/compression.py

@@ -0,0 +1,347 @@
+"""Context compression and conversation summarization.
+
+Compresses long conversations into concise summaries for efficient context
+loading when resuming sessions or searching past work.
+"""
+
+import logging
+from typing import List, Optional
+
+from ..llm.openrouter_client import OpenRouterClient
+from .store import Conversation, ConversationMessage
+
+logger = logging.getLogger(__name__)
+
+
+class ContextCompressor:
+    """Compress conversations into summaries for context loading.
+
+    Uses cheap LLM (mistral-small) to generate summaries of conversations
+    and compress multiple conversations into context strings.
+
+    Attributes:
+        client: OpenRouterClient for LLM requests
+        summary_threshold: Minimum messages to trigger summarization
+        max_context_tokens: Maximum tokens for compressed context
+
+    Example:
+        >>> compressor = ContextCompressor(client)
+        >>> summary = await compressor.summarize(messages)
+        >>> context = await compressor.compress_for_context(
+        ...     conversations,
+        ...     max_tokens=4000
+        ... )
+    """
+
+    def __init__(
+        self,
+        client: OpenRouterClient,
+        summary_threshold: int = 10,
+        max_context_tokens: int = 4000,
+    ):
+        """Initialize context compressor.
+
+        Args:
+            client: OpenRouterClient for LLM requests
+            summary_threshold: Minimum messages to summarize
+            max_context_tokens: Maximum tokens for context string
+        """
+        self.client = client
+        self.summary_threshold = summary_threshold
+        self.max_context_tokens = max_context_tokens
+
+        logger.info(
+            "ContextCompressor initialized (threshold: %d msgs, max_tokens: %d)",
+            summary_threshold,
+            max_context_tokens,
+        )
+
+    async def summarize(
+        self,
+        messages: List[ConversationMessage],
+        focus: Optional[str] = None,
+    ) -> str:
+        """Generate summary of conversation messages.
+
+        Args:
+            messages: List of messages to summarize
+            focus: Optional focus area (e.g., "bug fixes", "API changes")
+
+        Returns:
+            Concise summary (2-4 sentences)
+
+        Example:
+            >>> summary = await compressor.summarize(messages)
+            >>> print(summary)
+            "Fixed login authentication bug in src/auth.py by updating token validation..."
+        """
+        if len(messages) < 2:
+            # Too short to summarize
+            return messages[0].content if messages else ""
+
+        # Build conversation text
+        conversation_text = self._format_messages(messages)
+
+        # Build summarization prompt
+        if focus:
+            prompt = f"""Summarize the following conversation, focusing on: {focus}
+
+Conversation:
+{conversation_text}
+
+Provide a concise summary (2-4 sentences) that captures:
+1. What was the main task or problem
+2. What actions were taken
+3. What was the outcome or current status
+4. Any important files, functions, or errors mentioned
+
+Summary:"""
+        else:
+            prompt = f"""Summarize the following conversation in 2-4 sentences.
+
+Conversation:
+{conversation_text}
+
+Focus on:
+1. What was the main task or problem
+2. What actions were taken
+3. What was the outcome or current status
+
+Summary:"""
+
+        messages_for_llm = [{"role": "user", "content": prompt}]
+
+        system = (
+            "You are a technical summarization assistant. "
+            "Provide clear, concise summaries of development conversations. "
+            "Focus on actionable information and key outcomes."
+        )
+
+        summary = await self.client.chat(messages_for_llm, system=system)
+        logger.debug(
+            "Generated summary (%d chars) from %d messages", len(summary), len(messages)
+        )
+
+        return summary.strip()
+
+    async def compress_for_context(
+        self,
+        conversations: List[Conversation],
+        max_tokens: Optional[int] = None,
+        prioritize_recent: bool = True,
+    ) -> str:
+        """Compress multiple conversations into context string.
+
+        Prioritizes recent conversations and uses summaries for older ones
+        to fit within token budget.
+
+        Args:
+            conversations: List of conversations to compress
+            max_tokens: Maximum tokens (default: self.max_context_tokens)
+            prioritize_recent: Whether to prioritize recent conversations
+
+        Returns:
+            Compressed context string ready for LLM input
+
+        Example:
+            >>> context = await compressor.compress_for_context(
+            ...     conversations,
+            ...     max_tokens=4000
+            ... )
+            >>> print(f"Context: {len(context)} chars")
+        """
+        if max_tokens is None:
+            max_tokens = self.max_context_tokens
+
+        # Sort by recency if prioritizing
+        if prioritize_recent:
+            conversations = sorted(
+                conversations, key=lambda c: c.updated_at, reverse=True
+            )
+
+        # Build context incrementally
+        context_parts = []
+        current_tokens = 0
+
+        for conv in conversations:
+            # Use summary if available, else generate one
+            if conv.summary:
+                summary_text = conv.summary
+            elif len(conv.messages) >= self.summary_threshold:
+                # Generate summary on-the-fly
+                summary_text = await self.summarize(conv.messages)
+            else:
+                # Use full conversation for short ones
+                summary_text = conv.get_full_text()
+
+            # Format conversation section
+            section = self._format_conversation_section(conv, summary_text)
+            section_tokens = len(section) // 4  # Rough approximation
+
+            # Check if adding this would exceed budget
+            if current_tokens + section_tokens > max_tokens:
+                # Try to fit summary only
+                short_summary = summary_text.split(". ")[0] + "."
+                short_section = self._format_conversation_section(conv, short_summary)
+                short_tokens = len(short_section) // 4
+
+                if current_tokens + short_tokens <= max_tokens:
+                    context_parts.append(short_section)
+                    current_tokens += short_tokens
+                else:
+                    # Can't fit any more, stop
+                    break
+            else:
+                context_parts.append(section)
+                current_tokens += section_tokens
+
+        context = "\n\n---\n\n".join(context_parts)
+
+        logger.info(
+            "Compressed %d conversations into context (%d chars, ~%d tokens)",
+            len(context_parts),
+            len(context),
+            current_tokens,
+        )
+
+        return context
+
+    def needs_summarization(self, messages: List[ConversationMessage]) -> bool:
+        """Check if conversation needs summarization.
+
+        Args:
+            messages: List of messages to check
+
+        Returns:
+            True if message count exceeds threshold
+
+        Example:
+            >>> if compressor.needs_summarization(messages):
+            ...     summary = await compressor.summarize(messages)
+        """
+        return len(messages) >= self.summary_threshold
+
+    def _format_messages(
+        self,
+        messages: List[ConversationMessage],
+        max_messages: Optional[int] = None,
+    ) -> str:
+        """Format messages as text for summarization.
+
+        Args:
+            messages: Messages to format
+            max_messages: Maximum messages to include
+
+        Returns:
+            Formatted conversation text
+        """
+        if max_messages:
+            messages = messages[:max_messages]
+
+        lines = []
+        for msg in messages:
+            # Format: ROLE: content
+            lines.append(f"{msg.role.upper()}: {msg.content}")
+
+        return "\n\n".join(lines)
+
+    def _format_conversation_section(
+        self, conversation: Conversation, summary: str
+    ) -> str:
+        """Format conversation section for context string.
+
+        Args:
+            conversation: Conversation to format
+            summary: Summary or full text
+
+        Returns:
+            Formatted section with metadata
+        """
+        # Format timestamp
+        timestamp = conversation.updated_at.strftime("%Y-%m-%d %H:%M")
+
+        # Build section
+        return f"""## Conversation: {conversation.id}
+**Project:** {conversation.project_id}
+**Instance:** {conversation.instance_name}
+**Updated:** {timestamp}
+**Messages:** {conversation.message_count}
+
+{summary}"""
+
+    async def auto_summarize_conversation(
+        self, conversation: Conversation
+    ) -> Optional[str]:
+        """Automatically summarize conversation if needed.
+
+        Checks if conversation needs summarization and generates one if so.
+        Updates the conversation's summary field but does NOT save to store.
+
+        Args:
+            conversation: Conversation to summarize
+
+        Returns:
+            Summary if generated, None if not needed
+
+        Example:
+            >>> summary = await compressor.auto_summarize_conversation(conv)
+            >>> if summary:
+            ...     conv.summary = summary
+            ...     await store.save(conv)
+        """
+        if not self.needs_summarization(conversation.messages):
+            logger.debug(
+                "Conversation %s too short to summarize (%d messages)",
+                conversation.id,
+                len(conversation.messages),
+            )
+            return None
+
+        if conversation.summary:
+            logger.debug("Conversation %s already has summary", conversation.id)
+            return conversation.summary
+
+        # Generate summary
+        summary = await self.summarize(conversation.messages)
+        logger.info("Auto-generated summary for conversation %s", conversation.id)
+
+        return summary
+
+    async def update_summary_if_stale(
+        self,
+        conversation: Conversation,
+        message_threshold: int = 5,
+    ) -> Optional[str]:
+        """Update summary if conversation has grown significantly.
+
+        Args:
+            conversation: Conversation to check
+            message_threshold: New messages required to trigger update
+
+        Returns:
+            Updated summary if regenerated, None otherwise
+
+        Example:
+            >>> updated = await compressor.update_summary_if_stale(conv)
+            >>> if updated:
+            ...     conv.summary = updated
+            ...     await store.save(conv)
+        """
+        if not conversation.summary:
+            # No existing summary, generate one
+            return await self.auto_summarize_conversation(conversation)
+
+        # Check if conversation has grown significantly
+        # (Simple heuristic: if more than threshold messages since last summarization)
+        # In practice, you'd track when summary was generated
+        if len(conversation.messages) < self.summary_threshold + message_threshold:
+            return None
+
+        # Regenerate summary
+        logger.info(
+            "Regenerating stale summary for conversation %s (%d messages)",
+            conversation.id,
+            len(conversation.messages),
+        )
+
+        return await self.summarize(conversation.messages)