claude-mpm 4.15.6__py3-none-any.whl → 4.21.3__py3-none-any.whl

claude_mpm/hooks/__init__.py

@@ -12,6 +12,11 @@ from .failure_learning import (
 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
+from .session_resume_hook import (
+    SessionResumeStartupHook,
+    get_session_resume_hook,
+    trigger_session_resume_check,
+)
 
 __all__ = [
     "BaseHook",
@@ -24,10 +29,13 @@ __all__ = [
     "KuzuMemoryHook",
     "KuzuResponseHook",
     "LearningExtractionHook",
+    "SessionResumeStartupHook",
     "get_failure_detection_hook",
     "get_fix_detection_hook",
     "get_kuzu_enrichment_hook",
     "get_kuzu_memory_hook",
     "get_kuzu_response_hook",
     "get_learning_extraction_hook",
+    "get_session_resume_hook",
+    "trigger_session_resume_check",
 ]

claude_mpm/hooks/claude_hooks/response_tracking.py

@@ -268,7 +268,10 @@ class ResponseTrackingManager:
     def track_stop_response(
         self, event: dict, session_id: str, metadata: dict, pending_prompts: dict
     ):
-        """Track response for stop events."""
+        """Track response for stop events.
+
+        Captures Claude API stop_reason and usage data for context management.
+        """
         if not (self.response_tracking_enabled and self.response_tracker):
             return
 
@@ -294,6 +297,37 @@ class ResponseTrackingManager:
                 # Add prompt timestamp to metadata
                 metadata["prompt_timestamp"] = prompt_data.get("timestamp")
 
+                # Capture Claude API stop_reason if available
+                if "stop_reason" in event:
+                    metadata["stop_reason"] = event["stop_reason"]
+                    if DEBUG:
+                        print(
+                            f"  - Captured stop_reason: {event['stop_reason']}",
+                            file=sys.stderr,
+                        )
+
+                # Capture Claude API usage data if available
+                if "usage" in event:
+                    usage_data = event["usage"]
+                    metadata["usage"] = {
+                        "input_tokens": usage_data.get("input_tokens", 0),
+                        "output_tokens": usage_data.get("output_tokens", 0),
+                        "cache_creation_input_tokens": usage_data.get(
+                            "cache_creation_input_tokens", 0
+                        ),
+                        "cache_read_input_tokens": usage_data.get(
+                            "cache_read_input_tokens", 0
+                        ),
+                    }
+                    if DEBUG:
+                        total_tokens = usage_data.get(
+                            "input_tokens", 0
+                        ) + usage_data.get("output_tokens", 0)
+                        print(
+                            f"  - Captured usage: {total_tokens} total tokens",
+                            file=sys.stderr,
+                        )
+
                 # Track the main Claude response
                 file_path = self.response_tracker.track_response(
                     agent_name="claude_main",

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/models/resume_log.py

@@ -0,0 +1,340 @@
+"""Resume Log Data Model.
+
+This module defines the data structure for session resume logs that enable
+seamless context restoration when Claude hits token limits.
+
+Design Philosophy:
+- Target 10k tokens maximum per resume log
+- Human-readable markdown format
+- Structured sections with token budgets
+- Optimized for Claude consumption on session resume
+"""
+
+from dataclasses import dataclass, field
+from datetime import datetime, timezone
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+from claude_mpm.core.logging_utils import get_logger
+
+logger = get_logger(__name__)
+
+
+@dataclass
+class ContextMetrics:
+    """Context window usage metrics."""
+
+    total_budget: int = 200000
+    used_tokens: int = 0
+    remaining_tokens: int = 0
+    percentage_used: float = 0.0
+    stop_reason: Optional[str] = None
+    model: str = "claude-sonnet-4.5"
+    session_id: str = ""
+    timestamp: str = field(
+        default_factory=lambda: datetime.now(timezone.utc).isoformat()
+    )
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "total_budget": self.total_budget,
+            "used_tokens": self.used_tokens,
+            "remaining_tokens": self.remaining_tokens,
+            "percentage_used": self.percentage_used,
+            "stop_reason": self.stop_reason,
+            "model": self.model,
+            "session_id": self.session_id,
+            "timestamp": self.timestamp,
+        }
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "ContextMetrics":
+        """Create from dictionary."""
+        return cls(
+            total_budget=data.get("total_budget", 200000),
+            used_tokens=data.get("used_tokens", 0),
+            remaining_tokens=data.get("remaining_tokens", 0),
+            percentage_used=data.get("percentage_used", 0.0),
+            stop_reason=data.get("stop_reason"),
+            model=data.get("model", "claude-sonnet-4.5"),
+            session_id=data.get("session_id", ""),
+            timestamp=data.get("timestamp", datetime.now(timezone.utc).isoformat()),
+        )
+
+
+@dataclass
+class ResumeLog:
+    """Resume log containing all information needed to restore session context.
+
+    Token Budget Distribution (10k tokens total):
+    - Context Metrics: 500 tokens
+    - Mission Summary: 1,000 tokens
+    - Accomplishments: 2,000 tokens
+    - Key Findings: 2,500 tokens
+    - Decisions & Rationale: 1,500 tokens
+    - Next Steps: 1,500 tokens
+    - Critical Context: 1,000 tokens
+    """
+
+    # Session identification
+    session_id: str
+    previous_session_id: Optional[str] = None
+    created_at: str = field(
+        default_factory=lambda: datetime.now(timezone.utc).isoformat()
+    )
+
+    # Context metrics
+    context_metrics: ContextMetrics = field(default_factory=ContextMetrics)
+
+    # Core content sections (with token budgets)
+    mission_summary: str = ""  # 1,000 tokens - What was the overall goal?
+    accomplishments: List[str] = field(
+        default_factory=list
+    )  # 2,000 tokens - What was completed?
+    key_findings: List[str] = field(
+        default_factory=list
+    )  # 2,500 tokens - What was discovered?
+    decisions_made: List[Dict[str, str]] = field(
+        default_factory=list
+    )  # 1,500 tokens - What choices were made and why?
+    next_steps: List[str] = field(
+        default_factory=list
+    )  # 1,500 tokens - What needs to happen next?
+    critical_context: Dict[str, Any] = field(
+        default_factory=dict
+    )  # 1,000 tokens - Essential state/data
+
+    # Metadata
+    files_modified: List[str] = field(default_factory=list)
+    agents_used: Dict[str, int] = field(default_factory=dict)
+    errors_encountered: List[str] = field(default_factory=list)
+    warnings: List[str] = field(default_factory=list)
+
+    def to_markdown(self) -> str:
+        """Generate markdown format for Claude consumption.
+
+        Returns:
+            Markdown-formatted resume log
+        """
+        sections = []
+
+        # Header
+        sections.append(f"# Session Resume Log: {self.session_id}\n")
+        sections.append(f"**Created**: {self.created_at}")
+        if self.previous_session_id:
+            sections.append(f"**Previous Session**: {self.previous_session_id}")
+        sections.append("")
+
+        # Context Metrics (500 tokens)
+        sections.append("## Context Metrics\n")
+        sections.append(f"- **Model**: {self.context_metrics.model}")
+        sections.append(
+            f"- **Tokens Used**: {self.context_metrics.used_tokens:,} / {self.context_metrics.total_budget:,}"
+        )
+        sections.append(
+            f"- **Percentage**: {self.context_metrics.percentage_used:.1f}%"
+        )
+        sections.append(
+            f"- **Remaining**: {self.context_metrics.remaining_tokens:,} tokens"
+        )
+        if self.context_metrics.stop_reason:
+            sections.append(f"- **Stop Reason**: {self.context_metrics.stop_reason}")
+        sections.append("")
+
+        # Mission Summary (1,000 tokens)
+        sections.append("## Mission Summary\n")
+        sections.append(
+            self.mission_summary
+            if self.mission_summary
+            else "_No mission summary provided_"
+        )
+        sections.append("")
+
+        # Accomplishments (2,000 tokens)
+        sections.append("## Accomplishments\n")
+        if self.accomplishments:
+            for i, item in enumerate(self.accomplishments, 1):
+                sections.append(f"{i}. {item}")
+        else:
+            sections.append("_No accomplishments recorded_")
+        sections.append("")
+
+        # Key Findings (2,500 tokens)
+        sections.append("## Key Findings\n")
+        if self.key_findings:
+            for i, finding in enumerate(self.key_findings, 1):
+                sections.append(f"{i}. {finding}")
+        else:
+            sections.append("_No key findings recorded_")
+        sections.append("")
+
+        # Decisions & Rationale (1,500 tokens)
+        sections.append("## Decisions & Rationale\n")
+        if self.decisions_made:
+            for i, decision in enumerate(self.decisions_made, 1):
+                decision_text = decision.get("decision", "")
+                rationale = decision.get("rationale", "")
+                sections.append(f"{i}. **Decision**: {decision_text}")
+                if rationale:
+                    sections.append(f"   **Rationale**: {rationale}")
+        else:
+            sections.append("_No decisions recorded_")
+        sections.append("")
+
+        # Next Steps (1,500 tokens)
+        sections.append("## Next Steps\n")
+        if self.next_steps:
+            for i, step in enumerate(self.next_steps, 1):
+                sections.append(f"{i}. {step}")
+        else:
+            sections.append("_No next steps defined_")
+        sections.append("")
+
+        # Critical Context (1,000 tokens)
+        sections.append("## Critical Context\n")
+        if self.critical_context:
+            for key, value in self.critical_context.items():
+                sections.append(f"- **{key}**: {value}")
+        else:
+            sections.append("_No critical context preserved_")
+        sections.append("")
+
+        # Metadata
+        sections.append("## Session Metadata\n")
+        if self.files_modified:
+            sections.append(f"**Files Modified** ({len(self.files_modified)}):")
+            for file in self.files_modified[:20]:  # Limit to first 20
+                sections.append(f"- {file}")
+            if len(self.files_modified) > 20:
+                sections.append(f"- ... and {len(self.files_modified) - 20} more")
+            sections.append("")
+
+        if self.agents_used:
+            sections.append("**Agents Used**:")
+            for agent, count in self.agents_used.items():
+                sections.append(f"- {agent}: {count} delegations")
+            sections.append("")
+
+        if self.errors_encountered:
+            sections.append(f"**Errors** ({len(self.errors_encountered)}):")
+            for error in self.errors_encountered[:5]:  # Limit to first 5
+                sections.append(f"- {error}")
+            sections.append("")
+
+        if self.warnings:
+            sections.append(f"**Warnings** ({len(self.warnings)}):")
+            for warning in self.warnings[:5]:  # Limit to first 5
+                sections.append(f"- {warning}")
+            sections.append("")
+
+        return "\n".join(sections)
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary for JSON serialization."""
+        return {
+            "session_id": self.session_id,
+            "previous_session_id": self.previous_session_id,
+            "created_at": self.created_at,
+            "context_metrics": self.context_metrics.to_dict(),
+            "mission_summary": self.mission_summary,
+            "accomplishments": self.accomplishments,
+            "key_findings": self.key_findings,
+            "decisions_made": self.decisions_made,
+            "next_steps": self.next_steps,
+            "critical_context": self.critical_context,
+            "files_modified": self.files_modified,
+            "agents_used": self.agents_used,
+            "errors_encountered": self.errors_encountered,
+            "warnings": self.warnings,
+        }
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "ResumeLog":
+        """Create from dictionary."""
+        context_metrics_data = data.get("context_metrics", {})
+        context_metrics = ContextMetrics.from_dict(context_metrics_data)
+
+        return cls(
+            session_id=data.get("session_id", ""),
+            previous_session_id=data.get("previous_session_id"),
+            created_at=data.get("created_at", datetime.now(timezone.utc).isoformat()),
+            context_metrics=context_metrics,
+            mission_summary=data.get("mission_summary", ""),
+            accomplishments=data.get("accomplishments", []),
+            key_findings=data.get("key_findings", []),
+            decisions_made=data.get("decisions_made", []),
+            next_steps=data.get("next_steps", []),
+            critical_context=data.get("critical_context", {}),
+            files_modified=data.get("files_modified", []),
+            agents_used=data.get("agents_used", {}),
+            errors_encountered=data.get("errors_encountered", []),
+            warnings=data.get("warnings", []),
+        )
+
+    def save(self, storage_dir: Optional[Path] = None) -> Path:
+        """Save resume log to markdown file.
+
+        Args:
+            storage_dir: Directory to save the log (default: .claude-mpm/resume-logs)
+
+        Returns:
+            Path to saved file
+        """
+        if storage_dir is None:
+            storage_dir = Path.home() / ".claude-mpm" / "resume-logs"
+
+        storage_dir.mkdir(parents=True, exist_ok=True)
+
+        # Generate filename
+        file_path = storage_dir / f"session-{self.session_id}.md"
+
+        try:
+            # Write markdown file
+            markdown_content = self.to_markdown()
+            file_path.write_text(markdown_content, encoding="utf-8")
+
+            logger.info(f"Resume log saved: {file_path}")
+            return file_path
+
+        except Exception as e:
+            logger.error(f"Failed to save resume log: {e}")
+            raise
+
+    @classmethod
+    def load(
+        cls, session_id: str, storage_dir: Optional[Path] = None
+    ) -> Optional["ResumeLog"]:
+        """Load resume log from file.
+
+        Args:
+            session_id: Session ID to load
+            storage_dir: Directory to load from (default: .claude-mpm/resume-logs)
+
+        Returns:
+            ResumeLog instance or None if not found
+        """
+        if storage_dir is None:
+            storage_dir = Path.home() / ".claude-mpm" / "resume-logs"
+
+        file_path = storage_dir / f"session-{session_id}.md"
+
+        if not file_path.exists():
+            logger.debug(f"Resume log not found: {file_path}")
+            return None
+
+        try:
+            # For now, we just return the markdown content
+            # In the future, could parse markdown back to structured data
+            _ = file_path.read_text(encoding="utf-8")
+            logger.info(f"Resume log loaded: {file_path}")
+
+            # Return a basic ResumeLog with the markdown content embedded
+            return cls(
+                session_id=session_id,
+                mission_summary=f"Loaded from previous session. See full context in {file_path}",
+            )
+
+        except Exception as e:
+            logger.error(f"Failed to load resume log: {e}")
+            return None

claude_mpm/services/agents/auto_config_manager.py

@@ -678,7 +678,7 @@ class AutoConfigManagerService(BaseService, IAutoConfigManager):
                     agent_id, agent_name, success=True
                 )
                 deployed.append(agent_id)
-                self.logger.info(f"Successfully deployed agent: {agent_id}")
+                self.logger.debug(f"Successfully deployed agent: {agent_id}")
 
             except Exception as e:
                 self.logger.error(

claude_mpm/services/agents/deployment/agent_configuration_manager.py

@@ -71,7 +71,7 @@ class AgentConfigurationManager:
             # Cache the result
             self._base_agent_cache = (base_agent_data, base_agent_version)
 
-            self.logger.info(f"Loaded base agent from {self.base_agent_path}")
+            self.logger.debug(f"Loaded base agent from {self.base_agent_path}")
             return self._base_agent_cache
 
         except Exception as e:

claude_mpm/services/agents/deployment/agent_record_service.py

@@ -107,7 +107,7 @@ class AgentRecordService(BaseService):
                     record = self._deserialize_record(record_data)
                     records[agent_name] = record
 
-                self.logger.info(f"Loaded {len(records)} agent records")
+                self.logger.debug(f"Loaded {len(records)} agent records")
             else:
                 self.logger.debug("No existing records file found")
 

claude_mpm/services/agents/deployment/agent_validator.py

@@ -329,10 +329,26 @@ class AgentValidator:
             "type": "agent",  # Default type
         }
 
-        # Extract from YAML frontmatter
+        # Extract ONLY from YAML frontmatter (between --- markers)
         lines = content.split("\n")
+        in_frontmatter = False
+        frontmatter_ended = False
+
         for line in lines:
             stripped_line = line.strip()
+
+            # Track frontmatter boundaries
+            if stripped_line == "---":
+                if not in_frontmatter:
+                    in_frontmatter = True
+                    continue
+                frontmatter_ended = True
+                break  # Stop parsing after frontmatter ends
+
+            # Only parse within frontmatter
+            if not in_frontmatter or frontmatter_ended:
+                continue
+
             if stripped_line.startswith("name:"):
                 agent_info["name"] = stripped_line.split(":", 1)[1].strip().strip("\"'")
             elif stripped_line.startswith("description:"):

claude_mpm/services/agents/deployment/async_agent_deployment.py

@@ -224,7 +224,7 @@ class AsyncAgentDeploymentService:
 
         elapsed = (time.time() - start_time) * 1000
         self._metrics["time_saved_ms"] += max(0, (len(directories) * 75) - elapsed)
-        self.logger.info(f"Discovered agents in {elapsed:.1f}ms (parallel scan)")
+        self.logger.debug(f"Discovered agents in {elapsed:.1f}ms (parallel scan)")
 
         return discovered
 

claude_mpm/services/agents/deployment/local_template_deployment.py

@@ -90,7 +90,7 @@ class LocalTemplateDeploymentService:
                 logger.error(f"Failed to deploy local template {agent_id}: {e}")
                 results["errors"].append(f"{agent_id}: {e}")
 
-        logger.info(
+        logger.debug(
             f"Local template deployment: deployed={len(results['deployed'])}, "
             f"updated={len(results['updated'])}, skipped={len(results['skipped'])}, "
             f"errors={len(results['errors'])}"

claude_mpm/services/agents/local_template_manager.py

@@ -182,7 +182,7 @@ class LocalAgentTemplateManager:
             self._discover_templates_in_dir(self.user_agents_dir, "user")
 
         self._cache_valid = True
-        logger.info(f"Discovered {len(self._template_cache)} local agent templates")
+        logger.debug(f"Discovered {len(self._template_cache)} local agent templates")
 
         return self._template_cache
 

claude_mpm/services/agents/recommender.py

@@ -226,6 +226,53 @@ class AgentRecommenderService(BaseService, IAgentRecommender):
         if max_agents is not None:
             recommendations = recommendations[:max_agents]
 
+        # Check if toolchain is unknown and we have no recommendations
+        if not recommendations and toolchain.primary_language.lower() == "unknown":
+            self.logger.info("Toolchain unknown - applying default configuration")
+
+            # Get default configuration
+            default_config = self._capabilities_config.get("default_configuration", {})
+            if default_config.get("enabled", False):
+                default_agents = default_config.get("agents", [])
+                default_confidence = default_config.get("min_confidence", 0.7)
+
+                for default_agent in default_agents:
+                    agent_id = default_agent.get("agent_id")
+
+                    # Skip if agent doesn't exist in capabilities
+                    if agent_id not in agent_configs:
+                        self.logger.warning(f"Default agent not found: {agent_id}")
+                        continue
+
+                    agent_config = agent_configs[agent_id]
+                    capabilities = self.get_agent_capabilities(agent_id)
+
+                    recommendation = AgentRecommendation(
+                        agent_id=agent_id,
+                        agent_name=agent_config.get("name", agent_id),
+                        confidence_score=default_confidence,
+                        match_reasons=[
+                            "Default configuration applied - toolchain detection unsuccessful",
+                            default_agent.get("reasoning", "General-purpose agent"),
+                        ],
+                        concerns=[
+                            "Consider manually selecting specialized agents for better results"
+                        ],
+                        capabilities=capabilities,
+                        deployment_priority=default_agent.get("priority", 10),
+                        configuration_hints={"default_deployment": True},
+                        metadata={
+                            "is_default": True,
+                            "specialization": agent_config.get("specialization"),
+                            "auto_deploy": agent_config.get("auto_deploy", True),
+                        },
+                    )
+                    recommendations.append(recommendation)
+
+                self.logger.info(
+                    f"Applied {len(recommendations)} default agent recommendations"
+                )
+
         self.logger.info(
             f"Generated {len(recommendations)} agent recommendations "
             f"for project: {toolchain.project_path}"