claude-mpm 3.1.3__py3-none-any.whl → 3.2.1__py3-none-any.whl

claude_mpm/services/agent_modification_tracker.py

@@ -28,8 +28,9 @@ import os
 import shutil
 import time
 import uuid
-from dataclasses import asdict
+from dataclasses import dataclass, field, asdict
 from datetime import datetime
+from enum import Enum
 from pathlib import Path
 from typing import Dict, List, Optional, Any, Callable, Tuple, Set
 
@@ -39,25 +40,105 @@ from watchdog.events import FileSystemEventHandler, FileSystemEvent
 from claude_mpm.core.base_service import BaseService
 from claude_mpm.services.shared_prompt_cache import SharedPromptCache
 from claude_mpm.services.agent_registry import AgentRegistry
-from claude_mpm.models.modification import (
-    ModificationType,
-    ModificationTier,
-    AgentModification,
-    ModificationHistory
-)
 from claude_mpm.utils.path_operations import path_ops
 from claude_mpm.utils.config_manager import ConfigurationManager
 
-# Backward compatibility exports
-# These allow existing code to import models from this module
-__all__ = [
-    'ModificationType',
-    'ModificationTier',
-    'AgentModification', 
-    'ModificationHistory',
-    'AgentModificationTracker',
-    'AgentFileSystemHandler'
-]
+
+# ============================================================================
+# Data Models
+# ============================================================================
+
+class ModificationType(Enum):
+    """Types of agent modifications."""
+    CREATE = "create"
+    MODIFY = "modify"
+    DELETE = "delete"
+    MOVE = "move"
+    RESTORE = "restore"
+
+
+class ModificationTier(Enum):
+    """Agent hierarchy tiers for modification tracking."""
+    PROJECT = "project"
+    USER = "user"
+    SYSTEM = "system"
+
+
+@dataclass
+class AgentModification:
+    """Agent modification record with comprehensive metadata."""
+    
+    modification_id: str
+    agent_name: str
+    modification_type: ModificationType
+    tier: ModificationTier
+    file_path: str
+    timestamp: float
+    user_id: Optional[str] = None
+    modification_details: Dict[str, Any] = field(default_factory=dict)
+    file_hash_before: Optional[str] = None
+    file_hash_after: Optional[str] = None
+    file_size_before: Optional[int] = None
+    file_size_after: Optional[int] = None
+    backup_path: Optional[str] = None
+    validation_status: str = "pending"
+    validation_errors: List[str] = field(default_factory=list)
+    related_modifications: List[str] = field(default_factory=list)
+    metadata: Dict[str, Any] = field(default_factory=dict)
+    
+    @property
+    def modification_datetime(self) -> datetime:
+        """Get modification timestamp as datetime."""
+        return datetime.fromtimestamp(self.timestamp)
+    
+    @property
+    def age_seconds(self) -> float:
+        """Get age of modification in seconds."""
+        return time.time() - self.timestamp
+    
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary for serialization."""
+        data = asdict(self)
+        data['modification_type'] = self.modification_type.value
+        data['tier'] = self.tier.value
+        return data
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> 'AgentModification':
+        """Create from dictionary."""
+        data['modification_type'] = ModificationType(data['modification_type'])
+        data['tier'] = ModificationTier(data['tier'])
+        return cls(**data)
+
+
+@dataclass
+class ModificationHistory:
+    """Complete modification history for an agent."""
+    
+    agent_name: str
+    modifications: List[AgentModification] = field(default_factory=list)
+    current_version: Optional[str] = None
+    total_modifications: int = 0
+    first_seen: Optional[float] = None
+    last_modified: Optional[float] = None
+    
+    def add_modification(self, modification: AgentModification) -> None:
+        """Add a modification to history."""
+        self.modifications.append(modification)
+        self.total_modifications += 1
+        self.last_modified = modification.timestamp
+        
+        if self.first_seen is None:
+            self.first_seen = modification.timestamp
+    
+    def get_recent_modifications(self, hours: int = 24) -> List[AgentModification]:
+        """Get modifications within specified hours."""
+        cutoff = time.time() - (hours * 3600)
+        return [mod for mod in self.modifications if mod.timestamp >= cutoff]
+    
+    def get_modifications_by_type(self, mod_type: ModificationType) -> List[AgentModification]:
+        """Get modifications by type."""
+        return [mod for mod in self.modifications if mod.modification_type == mod_type]
 
 
 # ============================================================================

claude_mpm/services/agent_persistence_service.py

@@ -13,22 +13,42 @@ DESIGN DECISION: Creating a minimal stub because:
 - Allows for future extension if needed
 """
 
-from typing import Optional, Any
+from dataclasses import dataclass
+from enum import Enum
+from typing import Optional, Any, Dict
 import time
 
-from claude_mpm.models.persistence import (
-    PersistenceStrategy,
-    PersistenceOperation,
-    PersistenceRecord
-)
 
-# Backward compatibility exports
-__all__ = [
-    'PersistenceStrategy',
-    'PersistenceOperation',
-    'PersistenceRecord',
-    'AgentPersistenceService'
-]
+class PersistenceStrategy(Enum):
+    """Agent persistence strategies."""
+    USER_OVERRIDE = "user_override"
+    PROJECT_SPECIFIC = "project_specific"
+    SYSTEM_DEFAULT = "system_default"
+
+
+class PersistenceOperation(Enum):
+    """Persistence operation types."""
+    CREATE = "create"
+    UPDATE = "update"
+    DELETE = "delete"
+    BACKUP = "backup"
+    RESTORE = "restore"
+
+
+@dataclass
+class PersistenceRecord:
+    """Record of a persistence operation."""
+    operation_id: str
+    operation_type: PersistenceOperation
+    agent_name: str
+    source_tier: Any
+    target_tier: Optional[Any] = None
+    strategy: Optional[PersistenceStrategy] = None
+    success: bool = True
+    timestamp: float = 0.0
+    file_path: Optional[str] = None
+    error_message: Optional[str] = None
+    metadata: Dict[str, Any] = None
 
 
 class AgentPersistenceService:

claude_mpm/services/agent_registry.py

@@ -25,35 +25,36 @@ import hashlib
 import logging
 from pathlib import Path
 from typing import Dict, List, Optional, Set, Tuple, Any, Union
-from dataclasses import asdict
+from dataclasses import dataclass, field, asdict
 from datetime import datetime
 from enum import Enum
 
-from claude_mpm.models.registry import AgentTier, AgentRegistryMetadata
-from claude_mpm.models.common import (
-    CORE_AGENT_TYPES,
-    SPECIALIZED_AGENT_TYPES,
-    ALL_AGENT_TYPES,
-    AGENT_FILE_EXTENSIONS,
-    AGENT_IGNORE_PATTERNS
-)
-
 logger = logging.getLogger(__name__)
 
-# Backward compatibility exports
-# Export AgentRegistryMetadata as AgentMetadata for existing code
-AgentMetadata = AgentRegistryMetadata
 
-__all__ = [
-    'AgentTier',
-    'AgentType',
-    'AgentMetadata',  # Alias for AgentRegistryMetadata
-    'AgentRegistryMetadata',
-    'AgentRegistry'
-]
+# ============================================================================
+# Constants and Types
+# ============================================================================
+
+CORE_AGENT_TYPES = {
+    'engineer', 'architect', 'qa', 'security', 'documentation',
+    'ops', 'data', 'research', 'version_control'
+}
+
+SPECIALIZED_AGENT_TYPES = {
+    'pm_orchestrator', 'frontend', 'backend', 'devops', 'ml',
+    'database', 'api', 'mobile', 'cloud', 'testing'
+}
+
+ALL_AGENT_TYPES = CORE_AGENT_TYPES | SPECIALIZED_AGENT_TYPES
+
+
+class AgentTier(Enum):
+    """Agent hierarchy tiers."""
+    USER = "user"
+    SYSTEM = "system"
 
 
-# Keep local AgentType enum for internal classification logic
 class AgentType(Enum):
     """Agent classification types."""
     CORE = "core"
@@ -62,6 +63,44 @@ class AgentType(Enum):
     UNKNOWN = "unknown"
 
 
+# ============================================================================
+# Data Models
+# ============================================================================
+
+@dataclass
+class AgentMetadata:
+    """Complete metadata for discovered agent."""
+    name: str
+    path: str
+    tier: AgentTier
+    agent_type: AgentType
+    description: str = ""
+    version: str = "0.0.0"
+    dependencies: List[str] = field(default_factory=list)
+    capabilities: List[str] = field(default_factory=list)
+    created_at: float = field(default_factory=time.time)
+    last_modified: float = field(default_factory=time.time)
+    file_size: int = 0
+    checksum: str = ""
+    is_valid: bool = True
+    validation_errors: List[str] = field(default_factory=list)
+    metadata: Dict[str, Any] = field(default_factory=dict)
+    
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary for serialization."""
+        data = asdict(self)
+        data['tier'] = self.tier.value
+        data['agent_type'] = self.agent_type.value
+        return data
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> 'AgentMetadata':
+        """Create from dictionary."""
+        data['tier'] = AgentTier(data['tier'])
+        data['agent_type'] = AgentType(data['agent_type'])
+        return cls(**data)
+
+
 # ============================================================================
 # Main Registry Class
 # ============================================================================
@@ -80,7 +119,7 @@ class AgentRegistry:
         self.model_selector = model_selector
         
         # Registry storage
-        self.registry: Dict[str, AgentRegistryMetadata] = {}
+        self.registry: Dict[str, AgentMetadata] = {}
         self.discovery_paths: List[Path] = []
         
         # Cache configuration
@@ -89,8 +128,8 @@ class AgentRegistry:
         self.cache_prefix = "agent_registry"
         
         # Discovery configuration
-        self.file_extensions = AGENT_FILE_EXTENSIONS
-        self.ignore_patterns = AGENT_IGNORE_PATTERNS
+        self.file_extensions = {'.md', '.json', '.yaml', '.yml'}
+        self.ignore_patterns = {'__pycache__', '.git', 'node_modules', '.pytest_cache'}
         
         # Statistics
         self.discovery_stats = {
@@ -130,7 +169,7 @@ class AgentRegistry:
     # Discovery Methods
     # ========================================================================
     
-    def discover_agents(self, force_refresh: bool = False) -> Dict[str, AgentRegistryMetadata]:
+    def discover_agents(self, force_refresh: bool = False) -> Dict[str, AgentMetadata]:
         """
         Discover all available agents across configured paths.
         
@@ -230,7 +269,7 @@ class AgentRegistry:
         
         return name
     
-    def _create_agent_metadata(self, file_path: Path, agent_name: str, tier: AgentTier) -> AgentRegistryMetadata:
+    def _create_agent_metadata(self, file_path: Path, agent_name: str, tier: AgentTier) -> AgentMetadata:
         """Create agent metadata from file."""
         # Get file stats
         stat = file_path.stat()
@@ -285,11 +324,11 @@ class AgentRegistry:
         except Exception as e:
             logger.warning(f"Failed to parse {file_path}: {e}")
         
-        return AgentRegistryMetadata(
+        return AgentMetadata(
             name=agent_name,
             path=str(file_path),
             tier=tier,
-            agent_type=agent_type.value,  # Convert enum to string
+            agent_type=agent_type,
             description=description,
             version=version,
             capabilities=capabilities,
@@ -340,7 +379,7 @@ class AgentRegistry:
     def _apply_tier_precedence(self) -> None:
         """Apply tier precedence rules to discovered agents."""
         # Group agents by name
-        agents_by_name: Dict[str, List[AgentRegistryMetadata]] = {}
+        agents_by_name: Dict[str, List[AgentMetadata]] = {}
         
         for agent in self.registry.values():
             if agent.name not in agents_by_name:
@@ -364,7 +403,7 @@ class AgentRegistry:
     # Validation Methods
     # ========================================================================
     
-    def _validate_agent(self, metadata: AgentRegistryMetadata) -> bool:
+    def _validate_agent(self, metadata: AgentMetadata) -> bool:
         """Validate agent metadata and file."""
         errors = []
         
@@ -398,7 +437,7 @@ class AgentRegistry:
     # Cache Methods
     # ========================================================================
     
-    def _get_cached_registry(self) -> Optional[Dict[str, AgentRegistryMetadata]]:
+    def _get_cached_registry(self) -> Optional[Dict[str, AgentMetadata]]:
         """Get registry from cache if available."""
         if not self.cache_service:
             return None
@@ -411,7 +450,7 @@ class AgentRegistry:
                 # Deserialize metadata
                 registry = {}
                 for name, data in cached_data.items():
-                    registry[name] = AgentRegistryMetadata.from_dict(data)
+                    registry[name] = AgentMetadata.from_dict(data)
                 return registry
         
         except Exception as e:
@@ -452,7 +491,7 @@ class AgentRegistry:
     # Query Methods
     # ========================================================================
     
-    def get_agent(self, name: str) -> Optional[AgentRegistryMetadata]:
+    def get_agent(self, name: str) -> Optional[AgentMetadata]:
         """Get metadata for a specific agent."""
         # Ensure registry is populated
         if not self.registry:
@@ -461,7 +500,7 @@ class AgentRegistry:
         return self.registry.get(name)
     
     def list_agents(self, tier: Optional[AgentTier] = None, 
-                   agent_type: Optional[str] = None) -> List[AgentRegistryMetadata]:
+                   agent_type: Optional[AgentType] = None) -> List[AgentMetadata]:
         """List agents with optional filtering."""
         # Ensure registry is populated
         if not self.registry:
@@ -485,19 +524,19 @@ class AgentRegistry:
         
         return sorted(self.registry.keys())
     
-    def get_core_agents(self) -> List[AgentRegistryMetadata]:
+    def get_core_agents(self) -> List[AgentMetadata]:
         """Get all core framework agents."""
-        return self.list_agents(agent_type=AgentType.CORE.value)
+        return self.list_agents(agent_type=AgentType.CORE)
     
-    def get_specialized_agents(self) -> List[AgentRegistryMetadata]:
+    def get_specialized_agents(self) -> List[AgentMetadata]:
         """Get all specialized agents."""
-        return self.list_agents(agent_type=AgentType.SPECIALIZED.value)
+        return self.list_agents(agent_type=AgentType.SPECIALIZED)
     
-    def get_custom_agents(self) -> List[AgentRegistryMetadata]:
+    def get_custom_agents(self) -> List[AgentMetadata]:
         """Get all custom user-defined agents."""
-        return self.list_agents(agent_type=AgentType.CUSTOM.value)
+        return self.list_agents(agent_type=AgentType.CUSTOM)
     
-    def search_agents(self, query: str) -> List[AgentRegistryMetadata]:
+    def search_agents(self, query: str) -> List[AgentMetadata]:
         """Search agents by name or description."""
         if not self.registry:
             self.discover_agents()
@@ -540,7 +579,7 @@ class AgentRegistry:
         
         # Count by type
         for agent in self.registry.values():
-            agent_type = agent.agent_type  # Already a string
+            agent_type = agent.agent_type.value
             stats['agents_by_type'][agent_type] = stats['agents_by_type'].get(agent_type, 0) + 1
         
         # Validation stats
@@ -628,7 +667,7 @@ class AgentRegistry:
         
         # Import agents
         for name, agent_data in data.get('agents', {}).items():
-            self.registry[name] = AgentRegistryMetadata.from_dict(agent_data)
+            self.registry[name] = AgentMetadata.from_dict(agent_data)
         
         # Cache imported registry
         if self.cache_enabled: