claude-mpm 3.9.9__py3-none-any.whl → 4.0.3__py3-none-any.whl

claude_mpm/cli/commands/aggregate.py

@@ -1,3 +1,5 @@
+from pathlib import Path
+
 """CLI commands for the Event Aggregator service.
 
 WHY: Provides command-line interface for managing the event aggregator service
@@ -6,41 +8,38 @@ that captures Socket.IO events and saves them as structured session documents.
 
 import json
 import sys
-from pathlib import Path
-from typing import Optional
 
+from ...core.logger import get_logger
+from ...models.agent_session import AgentSession
 from ...services.event_aggregator import (
+    aggregator_status,
     get_aggregator,
     start_aggregator,
     stop_aggregator,
-    aggregator_status
 )
-from ...models.agent_session import AgentSession
-from ...core.logger import get_logger
-
 
 logger = get_logger("cli.aggregate")
 
 
 def aggregate_command(args):
     """Main entry point for aggregate commands.
-    
+
     WHY: Routes subcommands to appropriate handlers for managing the
     event aggregator service.
     """
     subcommand = args.aggregate_subcommand
-    
-    if subcommand == 'start':
+
+    if subcommand == "start":
         return start_command(args)
-    elif subcommand == 'stop':
+    elif subcommand == "stop":
         return stop_command(args)
-    elif subcommand == 'status':
+    elif subcommand == "status":
         return status_command(args)
-    elif subcommand == 'sessions':
+    elif subcommand == "sessions":
         return sessions_command(args)
-    elif subcommand == 'view':
+    elif subcommand == "view":
         return view_command(args)
-    elif subcommand == 'export':
+    elif subcommand == "export":
         return export_command(args)
     else:
         print(f"Unknown subcommand: {subcommand}", file=sys.stderr)
@@ -49,18 +48,19 @@ def aggregate_command(args):
 
 def start_command(args):
     """Start the event aggregator service.
-    
+
     WHY: Starts capturing events from the Socket.IO dashboard server
     for building complete session documents.
     """
     print("Starting Event Aggregator service...")
-    
+
     # Check if Socket.IO server is running
     import socket
+
     try:
         with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
             s.settimeout(1)
-            result = s.connect_ex(('127.0.0.1', 8765))
+            result = s.connect_ex(("127.0.0.1", 8765))
             if result != 0:
                 print("Warning: Socket.IO server not detected on port 8765")
                 print("The aggregator requires the dashboard server to be running.")
@@ -69,32 +69,35 @@ def start_command(args):
                     return 1
     except Exception as e:
         logger.error(f"Error checking server status: {e}")
-    
+
     # Start the aggregator
     if start_aggregator():
         print("✅ Event Aggregator started successfully")
         print(f"Capturing events from localhost:8765")
         print(f"Sessions will be saved to: .claude-mpm/sessions/")
-        
+
         # Show initial status
         status = aggregator_status()
         print(f"\nStatus:")
         print(f"  Connected: {status['connected']}")
         print(f"  Active sessions: {status['active_sessions']}")
-        
+
         if args.daemon:
-            print("\nAggregator running in background. Use 'claude-mpm aggregate stop' to stop it.")
+            print(
+                "\nAggregator running in background. Use 'claude-mpm aggregate stop' to stop it."
+            )
         else:
             print("\nPress Ctrl+C to stop the aggregator...")
             try:
                 # Keep running until interrupted
                 import time
+
                 while True:
                     time.sleep(1)
             except KeyboardInterrupt:
                 print("\nStopping aggregator...")
                 stop_aggregator()
-        
+
         return 0
     else:
         print("❌ Failed to start Event Aggregator")
@@ -104,42 +107,43 @@ def start_command(args):
 
 def stop_command(args):
     """Stop the event aggregator service.
-    
+
     WHY: Gracefully stops the aggregator and saves any active sessions.
     """
     print("Stopping Event Aggregator service...")
-    
+
     # Get status before stopping
     status = aggregator_status()
-    active_sessions = status.get('active_sessions', 0)
-    
+    active_sessions = status.get("active_sessions", 0)
+
     if active_sessions > 0:
         print(f"Saving {active_sessions} active session(s)...")
-    
+
     stop_aggregator()
     print("✅ Event Aggregator stopped")
-    
+
     # Show final statistics
-    if status['total_events'] > 0:
+    if status["total_events"] > 0:
         print(f"\nStatistics:")
         print(f"  Total events captured: {status['total_events']}")
         print(f"  Sessions completed: {status['sessions_completed']}")
         print(f"  Events by type:")
-        for event_type, count in sorted(status['events_by_type'].items(), 
-                                       key=lambda x: x[1], reverse=True)[:5]:
+        for event_type, count in sorted(
+            status["events_by_type"].items(), key=lambda x: x[1], reverse=True
+        )[:5]:
             print(f"    {event_type}: {count}")
-    
+
     return 0
 
 
 def status_command(args):
     """Show status of the event aggregator service.
-    
+
     WHY: Provides visibility into what the aggregator is doing and
     what it has captured.
     """
     status = aggregator_status()
-    
+
     print("Event Aggregator Status")
     print("=" * 50)
     print(f"Running: {status['running']}")
@@ -148,37 +152,38 @@ def status_command(args):
     print(f"Save directory: {status['save_directory']}")
     print()
     print(f"Active sessions: {status['active_sessions']}")
-    if status['active_session_ids']:
-        for sid in status['active_session_ids']:
+    if status["active_session_ids"]:
+        for sid in status["active_session_ids"]:
             print(f"  - {sid}")
     print()
     print(f"Sessions completed: {status['sessions_completed']}")
     print(f"Total events captured: {status['total_events']}")
-    
-    if status['events_by_type']:
+
+    if status["events_by_type"]:
         print("\nTop event types:")
-        for event_type, count in sorted(status['events_by_type'].items(), 
-                                       key=lambda x: x[1], reverse=True)[:10]:
+        for event_type, count in sorted(
+            status["events_by_type"].items(), key=lambda x: x[1], reverse=True
+        )[:10]:
             print(f"  {event_type:30s} {count:6d}")
-    
+
     return 0
 
 
 def sessions_command(args):
     """List captured sessions.
-    
+
     WHY: Shows what sessions have been captured for analysis.
     """
     aggregator = get_aggregator()
     sessions = aggregator.list_sessions(limit=args.limit)
-    
+
     if not sessions:
         print("No sessions found")
         return 0
-    
+
     print(f"Recent Sessions (showing {len(sessions)} of {args.limit} max)")
     print("=" * 80)
-    
+
     for session in sessions:
         print(f"\n📁 {session['file']}")
         print(f"   Session ID: {session['session_id']}")
@@ -187,37 +192,37 @@ def sessions_command(args):
         print(f"   Events: {session['events']}")
         print(f"   Delegations: {session['delegations']}")
         print(f"   Prompt: {session['initial_prompt']}")
-    
+
     print(f"\nUse 'claude-mpm aggregate view <session_id>' to view details")
-    
+
     return 0
 
 
 def view_command(args):
     """View details of a specific session.
-    
+
     WHY: Allows detailed inspection of what happened during a session.
     """
     aggregator = get_aggregator()
-    
+
     # Load the session
     session = aggregator.load_session(args.session_id)
-    
+
     if not session:
         print(f"Session not found: {args.session_id}")
         print("Use 'claude-mpm aggregate sessions' to list available sessions")
         return 1
-    
+
     print(f"Session: {session.session_id}")
     print("=" * 80)
     print(f"Start: {session.start_time}")
     print(f"End: {session.end_time or 'In progress'}")
     print(f"Working directory: {session.working_directory}")
     print(f"Launch method: {session.launch_method}")
-    
+
     if session.git_branch:
         print(f"Git branch: {session.git_branch}")
-    
+
     print(f"\nInitial prompt:")
     print("-" * 40)
     if session.initial_prompt:
@@ -226,7 +231,7 @@ def view_command(args):
             print("...")
     else:
         print("(No prompt captured)")
-    
+
     print(f"\nMetrics:")
     print("-" * 40)
     metrics = session.metrics
@@ -234,24 +239,24 @@ def view_command(args):
     print(f"Delegations: {metrics.total_delegations}")
     print(f"Tool calls: {metrics.total_tool_calls}")
     print(f"File operations: {metrics.total_file_operations}")
-    
+
     if metrics.session_duration_ms:
         duration_sec = metrics.session_duration_ms / 1000
         print(f"Duration: {duration_sec:.1f} seconds")
-    
+
     if metrics.agents_used:
         print(f"\nAgents used: {', '.join(sorted(metrics.agents_used))}")
-    
+
     if metrics.tools_used:
         print(f"\nTools used: {', '.join(sorted(metrics.tools_used))}")
-    
+
     if metrics.files_modified:
         print(f"\nFiles modified ({len(metrics.files_modified)}):")
         for filepath in sorted(metrics.files_modified)[:10]:
             print(f"  - {filepath}")
         if len(metrics.files_modified) > 10:
             print(f"  ... and {len(metrics.files_modified) - 10} more")
-    
+
     if session.delegations:
         print(f"\nDelegations ({len(session.delegations)}):")
         print("-" * 40)
@@ -267,196 +272,172 @@ def view_command(args):
             print(f"   Status: {'✅ Success' if delegation.success else '❌ Failed'}")
             if delegation.error:
                 print(f"   Error: {delegation.error}")
-    
+
     if args.show_events:
         print(f"\nEvents ({len(session.events)}):")
         print("-" * 40)
-        for event in session.events[:args.event_limit]:
+        for event in session.events[: args.event_limit]:
             print(f"{event.timestamp} [{event.category.value:10s}] {event.event_type}")
             if args.verbose:
                 print(f"  Agent: {event.agent_context or 'N/A'}")
                 if event.correlation_id:
                     print(f"  Correlation: {event.correlation_id}")
-    
+
     if session.final_response and not args.no_response:
         print(f"\nFinal response:")
         print("-" * 40)
         print(session.final_response[:1000])
         if len(session.final_response) > 1000:
             print("...")
-    
+
     return 0
 
 
 def export_command(args):
     """Export a session to a file.
-    
+
     WHY: Allows sessions to be exported for external analysis or sharing.
     """
     aggregator = get_aggregator()
-    
+
     # Load the session
     session = aggregator.load_session(args.session_id)
-    
+
     if not session:
         print(f"Session not found: {args.session_id}")
         return 1
-    
+
     # Determine output file
     if args.output:
         output_path = Path(args.output)
     else:
         output_path = Path(f"session_{session.session_id[:8]}_export.json")
-    
+
     # Export based on format
-    if args.format == 'json':
+    if args.format == "json":
         # Full JSON export
-        with open(output_path, 'w') as f:
+        with open(output_path, "w") as f:
             json.dump(session.to_dict(), f, indent=2)
         print(f"✅ Exported session to {output_path}")
-        
-    elif args.format == 'summary':
+
+    elif args.format == "summary":
         # Summary export
         summary = {
-            'session_id': session.session_id,
-            'start_time': session.start_time,
-            'end_time': session.end_time,
-            'working_directory': session.working_directory,
-            'initial_prompt': session.initial_prompt,
-            'final_response': session.final_response,
-            'metrics': session.metrics.to_dict(),
-            'delegations_summary': [
+            "session_id": session.session_id,
+            "start_time": session.start_time,
+            "end_time": session.end_time,
+            "working_directory": session.working_directory,
+            "initial_prompt": session.initial_prompt,
+            "final_response": session.final_response,
+            "metrics": session.metrics.to_dict(),
+            "delegations_summary": [
                 {
-                    'agent': d.agent_type,
-                    'task': d.task_description,
-                    'duration_ms': d.duration_ms,
-                    'success': d.success,
-                    'tools_used': len(d.tool_operations),
-                    'files_changed': len(d.file_changes)
+                    "agent": d.agent_type,
+                    "task": d.task_description,
+                    "duration_ms": d.duration_ms,
+                    "success": d.success,
+                    "tools_used": len(d.tool_operations),
+                    "files_changed": len(d.file_changes),
                 }
                 for d in session.delegations
-            ]
+            ],
         }
-        
-        with open(output_path, 'w') as f:
+
+        with open(output_path, "w") as f:
             json.dump(summary, f, indent=2)
         print(f"✅ Exported session summary to {output_path}")
-    
-    elif args.format == 'events':
+
+    elif args.format == "events":
         # Events-only export
         events_data = [e.to_dict() for e in session.events]
-        
-        with open(output_path, 'w') as f:
+
+        with open(output_path, "w") as f:
             json.dump(events_data, f, indent=2)
         print(f"✅ Exported {len(events_data)} events to {output_path}")
-    
+
     return 0
 
 
 def add_aggregate_parser(subparsers):
     """Add the aggregate command parser.
-    
+
     WHY: Integrates the aggregator commands into the main CLI system.
     """
     aggregate_parser = subparsers.add_parser(
-        'aggregate',
-        help='Manage event aggregator for capturing agent sessions'
+        "aggregate", help="Manage event aggregator for capturing agent sessions"
     )
-    
+
     aggregate_subparsers = aggregate_parser.add_subparsers(
-        dest='aggregate_subcommand',
-        help='Aggregator subcommands'
+        dest="aggregate_subcommand", help="Aggregator subcommands"
     )
-    
+
     # Start command
     start_parser = aggregate_subparsers.add_parser(
-        'start',
-        help='Start the event aggregator service'
+        "start", help="Start the event aggregator service"
     )
     start_parser.add_argument(
-        '--daemon', '-d',
-        action='store_true',
-        help='Run in background as daemon'
+        "--daemon", "-d", action="store_true", help="Run in background as daemon"
     )
     start_parser.add_argument(
-        '--force', '-f',
-        action='store_true',
-        help='Start even if Socket.IO server not detected'
+        "--force",
+        "-f",
+        action="store_true",
+        help="Start even if Socket.IO server not detected",
     )
-    
+
     # Stop command
-    aggregate_subparsers.add_parser(
-        'stop',
-        help='Stop the event aggregator service'
-    )
-    
+    aggregate_subparsers.add_parser("stop", help="Stop the event aggregator service")
+
     # Status command
     aggregate_subparsers.add_parser(
-        'status',
-        help='Show aggregator status and statistics'
+        "status", help="Show aggregator status and statistics"
     )
-    
+
     # Sessions command
     sessions_parser = aggregate_subparsers.add_parser(
-        'sessions',
-        help='List captured sessions'
+        "sessions", help="List captured sessions"
     )
     sessions_parser.add_argument(
-        '--limit', '-l',
+        "--limit",
+        "-l",
         type=int,
         default=10,
-        help='Maximum number of sessions to show (default: 10)'
+        help="Maximum number of sessions to show (default: 10)",
     )
-    
+
     # View command
     view_parser = aggregate_subparsers.add_parser(
-        'view',
-        help='View details of a specific session'
-    )
-    view_parser.add_argument(
-        'session_id',
-        help='Session ID or prefix to view'
+        "view", help="View details of a specific session"
     )
+    view_parser.add_argument("session_id", help="Session ID or prefix to view")
     view_parser.add_argument(
-        '--show-events', '-e',
-        action='store_true',
-        help='Show event list'
+        "--show-events", "-e", action="store_true", help="Show event list"
     )
     view_parser.add_argument(
-        '--event-limit',
+        "--event-limit",
         type=int,
         default=50,
-        help='Maximum events to show (default: 50)'
+        help="Maximum events to show (default: 50)",
     )
     view_parser.add_argument(
-        '--verbose', '-v',
-        action='store_true',
-        help='Show verbose event details'
+        "--verbose", "-v", action="store_true", help="Show verbose event details"
     )
     view_parser.add_argument(
-        '--no-response',
-        action='store_true',
-        help='Don\'t show final response'
+        "--no-response", action="store_true", help="Don't show final response"
     )
-    
+
     # Export command
     export_parser = aggregate_subparsers.add_parser(
-        'export',
-        help='Export a session to file'
+        "export", help="Export a session to file"
     )
+    export_parser.add_argument("session_id", help="Session ID or prefix to export")
+    export_parser.add_argument("--output", "-o", help="Output file path")
     export_parser.add_argument(
-        'session_id',
-        help='Session ID or prefix to export'
+        "--format",
+        "-f",
+        choices=["json", "summary", "events"],
+        default="json",
+        help="Export format (default: json)",
     )
-    export_parser.add_argument(
-        '--output', '-o',
-        help='Output file path'
-    )
-    export_parser.add_argument(
-        '--format', '-f',
-        choices=['json', 'summary', 'events'],
-        default='json',
-        help='Export format (default: json)'
-    )
-    
-    aggregate_parser.set_defaults(func=aggregate_command)
+
+    aggregate_parser.set_defaults(func=aggregate_command)