claude-mpm 4.13.2__py3-none-any.whl → 4.18.2__py3-none-any.whl

claude_mpm/core/output_style_manager.py

@@ -160,203 +160,23 @@ class OutputStyleManager:
 
     def extract_output_style_content(self, framework_loader=None) -> str:
         """
-        Extract output style content from framework instructions.
-
-        This extracts PM delegation behavior, tone, communication standards,
-        response formats, TodoWrite requirements, and workflow rules from:
-        - INSTRUCTIONS.md
-        - BASE_PM.md
+        Read output style content from OUTPUT_STYLE.md.
 
         Args:
-            framework_loader: Optional FrameworkLoader instance to reuse loaded content
+            framework_loader: Optional framework loader (kept for compatibility, not used)
 
         Returns:
-            Formatted output style content in YAML frontmatter + markdown format
+            Complete output style content from file
         """
-        # Build the content sections
-        sections = []
-
-        # Add YAML frontmatter
-        sections.append("---")
-        sections.append("name: Claude MPM")
-        sections.append(
-            "description: Multi-Agent Project Manager orchestration mode for delegation and coordination"
-        )
-        sections.append("---")
-        sections.append("")
-
-        # Header
-        sections.append(
-            "You are Claude Multi-Agent PM, a PROJECT MANAGER whose SOLE PURPOSE is to delegate work to specialized agents."
-        )
-        sections.append("")
-
-        # Extract from INSTRUCTIONS.md
-        if framework_loader and framework_loader.framework_content.get(
-            "framework_instructions"
-        ):
-            instructions = framework_loader.framework_content["framework_instructions"]
-            sections.extend(self._extract_instructions_sections(instructions))
-        else:
-            # Load from file if no framework_loader provided
-            instructions_path = (
-                Path(__file__).parent.parent / "agents" / "INSTRUCTIONS.md"
-            )
-            if instructions_path.exists():
-                instructions = instructions_path.read_text()
-                sections.extend(self._extract_instructions_sections(instructions))
-
-        # Extract from BASE_PM.md
-        if framework_loader and framework_loader.framework_content.get(
-            "base_pm_instructions"
-        ):
-            base_pm = framework_loader.framework_content["base_pm_instructions"]
-            sections.extend(self._extract_base_pm_sections(base_pm))
-        else:
-            # Load from file if no framework_loader provided
-            base_pm_path = Path(__file__).parent.parent / "agents" / "BASE_PM.md"
-            if base_pm_path.exists():
-                base_pm = base_pm_path.read_text()
-                sections.extend(self._extract_base_pm_sections(base_pm))
-
-        return "\n".join(sections)
-
-    def _extract_instructions_sections(self, content: str) -> list:
-        """Extract relevant sections from INSTRUCTIONS.md."""
-        sections = []
-
-        # Extract Primary Directive
-        if "## 🔴 PRIMARY DIRECTIVE" in content:
-            sections.append("## 🔴 PRIMARY DIRECTIVE - MANDATORY DELEGATION 🔴")
-            sections.append("")
-            sections.append(
-                "**YOU ARE STRICTLY FORBIDDEN FROM DOING ANY WORK DIRECTLY.**"
-            )
-            sections.append("")
-            sections.append(
-                "Direct implementation is ABSOLUTELY PROHIBITED unless the user EXPLICITLY overrides with phrases like:"
-            )
-            sections.append('- "do this yourself"')
-            sections.append('- "don\'t delegate"')
-            sections.append('- "implement directly"')
-            sections.append('- "you do it"')
-            sections.append('- "no delegation"')
-            sections.append("")
-
-        # Extract Core Identity and Rules
-        if "## Core Identity" in content:
-            sections.append("## Core Operating Rules")
-            sections.append("")
-            sections.append("**DEFAULT BEHAVIOR - ALWAYS DELEGATE**:")
-            sections.append(
-                "- 🔴 You MUST delegate 100% of ALL work to specialized agents by default"
-            )
-            sections.append(
-                "- 🔴 Direct action is STRICTLY FORBIDDEN without explicit user override"
-            )
-            sections.append(
-                "- 🔴 Even the simplest tasks MUST be delegated - NO EXCEPTIONS"
-            )
-            sections.append("- 🔴 When in doubt, ALWAYS DELEGATE - never act directly")
-            sections.append("")
-            sections.append("**Allowed Tools**:")
-            sections.append("- **Task** for delegation (YOUR PRIMARY FUNCTION)")
-            sections.append("- **TodoWrite** for tracking delegation progress ONLY")
-            sections.append(
-                "- **WebSearch/WebFetch** for gathering context BEFORE delegation"
-            )
-            sections.append(
-                "- **Direct answers** ONLY for questions about PM capabilities"
-            )
-            sections.append("")
-
-        # Extract Communication Standards
-        if "## Communication Standards" in content:
-            sections.append("## Communication Standards")
-            sections.append("")
-            sections.append("- **Tone**: Professional, neutral by default")
-            sections.append('- **Use**: "Understood", "Confirmed", "Noted"')
-            sections.append("- **No simplification** without explicit user request")
-            sections.append("- **No mocks** outside test environments")
-            sections.append("- **Complete implementations** only - no placeholders")
-            sections.append(
-                '- **FORBIDDEN**: Overeager enthusiasm ("Excellent!", "Perfect!", "Amazing!")'
-            )
-            sections.append("")
-
-        # Extract Error Handling
-        if "## Error Handling Protocol" in content:
-            sections.append("## Error Handling Protocol")
-            sections.append("")
-            sections.append("**3-Attempt Process**:")
-            sections.append("1. **First Failure**: Re-delegate with enhanced context")
-            sections.append(
-                '2. **Second Failure**: Mark "ERROR - Attempt 2/3", escalate if needed'
-            )
-            sections.append(
-                "3. **Third Failure**: TodoWrite escalation with user decision required"
-            )
-            sections.append("")
-
-        # Extract Standard Operating Procedure
-        if "## Standard Operating Procedure" in content:
-            sections.append("## Standard Operating Procedure")
-            sections.append("")
-            sections.append("1. **Analysis**: Parse request, assess context (NO TOOLS)")
-            sections.append(
-                "2. **Planning**: Agent selection, task breakdown, priority assignment"
-            )
-            sections.append("3. **Delegation**: Task Tool with enhanced format")
-            sections.append("4. **Monitoring**: Track progress via TodoWrite")
-            sections.append("5. **Integration**: Synthesize results, validate, report")
-            sections.append("")
-
-        return sections
-
-    def _extract_base_pm_sections(self, content: str) -> list:
-        """Extract relevant sections from BASE_PM.md."""
-        sections = []
-
-        # Extract TodoWrite Requirements
-        if "## TodoWrite Framework Requirements" in content:
-            sections.append("## TodoWrite Requirements")
-            sections.append("")
-            sections.append("### Mandatory [Agent] Prefix Rules")
-            sections.append("")
-            sections.append("**ALWAYS use [Agent] prefix for delegated tasks**:")
-            sections.append("- ✅ `[Research] Analyze authentication patterns`")
-            sections.append("- ✅ `[Engineer] Implement user registration`")
-            sections.append("- ✅ `[QA] Test payment flow`")
-            sections.append("- ✅ `[Documentation] Update API docs`")
-            sections.append("")
-            sections.append("**NEVER use [PM] prefix for implementation tasks**")
-            sections.append("")
-            sections.append("### Task Status Management")
-            sections.append("")
-            sections.append("- `pending` - Task not yet started")
-            sections.append(
-                "- `in_progress` - Currently being worked on (ONE at a time)"
-            )
-            sections.append("- `completed` - Task finished successfully")
-            sections.append("")
-
-        # Extract PM Response Format
-        if "## PM Response Format" in content:
-            sections.append("## Response Format")
-            sections.append("")
-            sections.append(
-                "When completing delegations, provide structured summaries including:"
-            )
-            sections.append("- Request summary")
-            sections.append("- Agents used and task counts")
-            sections.append("- Tasks completed with [Agent] prefixes")
-            sections.append("- Files affected across all agents")
-            sections.append("- Blockers encountered and resolutions")
-            sections.append("- Next steps for user")
-            sections.append("- Key information to remember")
-            sections.append("")
-
-        return sections
+        # Always read from the complete OUTPUT_STYLE.md file
+        if self.mpm_output_style_path.exists():
+            content = self.mpm_output_style_path.read_text()
+            self.logger.info(f"Read OUTPUT_STYLE.md directly ({len(content)} chars)")
+            return content
+        # Fallback error
+        error_msg = f"OUTPUT_STYLE.md not found at {self.mpm_output_style_path}"
+        self.logger.error(error_msg)
+        raise FileNotFoundError(error_msg)
 
     def save_output_style(self, content: str) -> Path:
         """

claude_mpm/core/service_registry.py

@@ -10,6 +10,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Type, Union
 from .base_service import BaseService
 from .config import Config
 from .container import DIContainer, ServiceLifetime, get_container
+from .enums import OperationResult
 from .logger import get_logger
 from .shared.config_loader import ConfigLoader
 
@@ -249,7 +250,10 @@ class ServiceRegistry:
                                 "metrics": health.metrics,
                             }
                 except Exception as e:
-                    health_status[name] = {"status": "error", "message": str(e)}
+                    health_status[name] = {
+                        "status": OperationResult.ERROR,
+                        "message": str(e),
+                    }
 
             return health_status
 

claude_mpm/core/types.py

@@ -20,6 +20,8 @@ from datetime import datetime, timezone
 from enum import Enum
 from typing import Any, Dict, List, Optional, Tuple, Union
 
+from .enums import HealthStatus
+
 
 # Service operation results
 @dataclass
@@ -230,15 +232,6 @@ class SocketMessage:
 
 
 # Health monitoring types
-class HealthStatus(Enum):
-    """Service health status levels."""
-
-    HEALTHY = "healthy"
-    DEGRADED = "degraded"
-    UNHEALTHY = "unhealthy"
-    UNKNOWN = "unknown"
-
-
 @dataclass
 class HealthCheck:
     """Health check result."""

claude_mpm/core/typing_utils.py

@@ -32,6 +32,8 @@ from typing import (
 
 from typing_extensions import NotRequired, TypeAlias, TypedDict
 
+from claude_mpm.core.enums import OperationResult, ServiceState
+
 # Generic type variables
 T = TypeVar("T")
 TSession = TypeVar("TSession")  # Generic session type
@@ -42,15 +44,14 @@ TService = TypeVar("TService")  # Generic service type
 PathLike: TypeAlias = Union[str, Path]
 JSONValue: TypeAlias = Union[str, int, float, bool, None, Dict[str, Any], List[Any]]
 JSONDict: TypeAlias = Dict[str, JSONValue]
+
 Headers: TypeAlias = Dict[str, str]
 ErrorCode: TypeAlias = Union[int, str]
 LogLevel: TypeAlias = Literal["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
 
 # Session types
 SessionId: TypeAlias = str
-SessionStatus: TypeAlias = Literal[
-    "initializing", "running", "stopped", "error", "completed"
-]
+SessionStatus: TypeAlias = ServiceState  # Replaced Literal with ServiceState enum
 LaunchMethod: TypeAlias = Literal["exec", "subprocess", "oneshot"]
 
 
@@ -152,7 +153,7 @@ class WebSocketMessage(TypedDict):
 class ClaudeStatus(TypedDict):
     """Claude process status."""
 
-    status: Literal["starting", "running", "stopped", "error"]
+    status: ServiceState  # Replaced Literal with ServiceState enum
     message: str
     timestamp: NotRequired[datetime]
     pid: NotRequired[int]
@@ -163,7 +164,7 @@ class DelegationInfo(TypedDict):
 
     agent: AgentId
     task: str
-    status: Literal["detected", "started", "completed", "failed"]
+    status: OperationResult  # Replaced Literal with OperationResult enum
     timestamp: NotRequired[datetime]
     result: NotRequired[str]
 
@@ -194,7 +195,7 @@ class HookContext(TypedDict):
 
 # Service types
 ServiceName: TypeAlias = str
-ServiceStatus: TypeAlias = Literal["idle", "running", "stopped", "error"]
+ServiceStatus: TypeAlias = ServiceState  # Replaced Literal with ServiceState enum
 
 
 class ServiceConfig(TypedDict):

claude_mpm/dashboard/static/js/dashboard.js

@@ -321,14 +321,6 @@ class Dashboard {
             // Process agent inference for new events
             this.agentInference.processAgentInference();
 
-            // Notify CodeViewer that file operations have been updated
-            // This ensures File Tree tab shows the same data as Files tab
-            if (window.CodeViewer && typeof window.CodeViewer.refreshFromFileToolTracker === 'function') {
-                setTimeout(() => {
-                    window.CodeViewer.refreshFromFileToolTracker();
-                }, 50);
-            }
-
             // Update agent hierarchy with new events
             this.agentHierarchy.updateWithNewEvents(events);
 
@@ -445,12 +437,6 @@ class Dashboard {
             case 'events':
                 // Events tab is handled by EventViewer
                 break;
-            case 'claude-tree':
-                // File Tree tab - trigger CodeViewer rendering
-                if (window.CodeViewer && typeof window.CodeViewer.show === 'function') {
-                    window.CodeViewer.show();
-                }
-                break;
             case 'activity':
                 // Trigger Activity tab rendering through the component
                 // Check if ActivityTree class is available (from built module)

claude_mpm/dashboard/templates/index.html

@@ -33,7 +33,6 @@
     <link rel="stylesheet" href="/static/css/dashboard.css">
     <link rel="stylesheet" href="/static/css/connection-status.css">
     <link rel="stylesheet" href="/static/css/activity.css">
-    <link rel="stylesheet" href="/static/css/code-tree.css">
 
     <!-- Additional styles for file operations -->
     <style>
@@ -258,7 +257,6 @@
                         <a href="#tools" class="tab-button" data-tab="tools">🔧 Tools</a>
                         <a href="#files" class="tab-button" data-tab="files">📁 Files</a>
                         <a href="#activity" class="tab-button" data-tab="activity">🌳 Activity</a>
-                        <a href="#file_tree" class="tab-button" data-tab="claude-tree">📝 File Tree</a>
                     </div>
 
                     <!-- Events Tab -->
@@ -397,15 +395,6 @@
                         </div>
                     </div>
 
-                    <!-- File Tree Tab -->
-                    <div class="tab-content" id="claude-tree-tab">
-                        <div id="claude-tree-container" style="width: 100%; height: 100%; position: relative;">
-                            <!-- File activity tree will be rendered here by code-viewer.js -->
-                            <!-- This container is ISOLATED from other tabs -->
-                        </div>
-                    </div>
-
-
                 </div>
             </div>
         </div>
@@ -489,16 +478,12 @@
             });
         };
         
-        // Load shared services first, then tree modules, then components
+        // Load shared services first, then components
         // Load services sequentially to ensure dependencies are available
         loadModule('/static/js/shared/tooltip-service.js')
             .then(() => loadModule('/static/js/shared/dom-helpers.js'))
             .then(() => loadModule('/static/js/shared/event-bus.js'))
             .then(() => loadModule('/static/js/shared/logger.js'))
-            .then(() => loadModule('/static/js/components/code-tree/tree-utils.js'))
-            .then(() => loadModule('/static/js/components/code-tree/tree-constants.js'))
-            .then(() => loadModule('/static/js/components/code-tree/tree-search.js'))
-            .then(() => loadModule('/static/js/components/code-tree/tree-breadcrumb.js'))
             .then(() => {
                 // CRITICAL: Load socket-client.js FIRST (dependency of socket-manager)
                 return loadModule('/static/js/socket-client.js');
@@ -530,29 +515,15 @@
                 return Promise.all([
                     loadModule('/static/dist/dashboard.js'),  // Use dist version that requires above components
                     loadModule('/static/dist/components/activity-tree.js'),
-                    loadModule('/static/js/components/code-tree.js'),  // TEMPORARY: Direct source for debugging
-                    loadModule('/static/js/components/code-viewer.js').catch(err => {
-                        console.error('[CRITICAL] Failed to load code-viewer.js:', err);
-                        throw err;
-                    }),  // Code viewer now includes file change tracking
                     loadModule('/static/dist/components/file-viewer.js')  // File viewer for viewing file contents
                 ]);
             })
             .then(() => {
             console.log('All dashboard modules loaded successfully');
-            
-            // Debug: Check if CodeViewer loaded
-            if (window.CodeViewer) {
-                console.log('[DEBUG] CodeViewer is available on window object');
-            } else {
-                console.error('[ERROR] CodeViewer NOT FOUND on window object!');
-            }
-            
-            // CodeViewer will auto-initialize and handle tab switching internally
-            
+
             // Browser Log Viewer initialization is now handled by UIStateManager
             // This prevents duplicate event handlers and tab selection conflicts
-            
+
             // Load bulletproof tab isolation fix
             loadModule('/static/js/tab-isolation-fix.js')
                 .then(() => {
@@ -564,15 +535,6 @@
 
             // Hash navigation will handle default tab based on URL
             // If no hash, default will be 'events' as per hashToTab mapping
-            // To start with File Tree, we can set hash if not present
-            setTimeout(() => {
-                if (!window.location.hash) {
-                    console.log('No hash present, setting default to File Tree tab...');
-                    window.location.hash = '#file_tree';
-                } else {
-                    console.log('Hash present:', window.location.hash);
-                }
-            }, 500);
             })
             .catch(error => {
                 console.error('[CRITICAL] Error loading dashboard modules:', error);

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/instruction_reinforcement.py

@@ -10,6 +10,7 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Dict, List, Optional, Tuple
 
+from claude_mpm.core.enums import OperationResult
 from claude_mpm.core.logging_utils import get_logger
 
 logger = get_logger(__name__)
@@ -267,7 +268,7 @@ class InstructionReinforcementHook:
         if not self.violations:
             return {
                 "total_violations": 0,
-                "status": "COMPLIANT",
+                "status": OperationResult.SUCCESS,
                 "message": "No PM delegation violations detected",
             }
 
@@ -276,7 +277,11 @@ class InstructionReinforcementHook:
             vtype = v.violation_type.value
             violation_types[vtype] = violation_types.get(vtype, 0) + 1
 
-        status = "WARNING" if self.violation_count < 3 else "CRITICAL"
+        status = (
+            OperationResult.ERROR
+            if self.violation_count < 3
+            else OperationResult.FAILED
+        )
 
         return {
             "total_violations": self.violation_count,