aiecs 1.3.8__py3-none-any.whl → 1.4.1__py3-none-any.whl

aiecs/domain/agent/integration/role_config.py

@@ -0,0 +1,217 @@
+"""
+Role-Based Configuration
+
+Load agent configuration from role templates.
+"""
+
+import logging
+import yaml
+from typing import Dict, Any, Optional
+from pathlib import Path
+
+from ..models import AgentConfiguration
+
+logger = logging.getLogger(__name__)
+
+
+class RoleConfiguration:
+    """
+    Manages role-based agent configurations.
+
+    Example:
+        role_config = RoleConfiguration.load_from_file("roles/developer.yaml")
+        agent_config = role_config.to_agent_config()
+    """
+
+    def __init__(
+        self,
+        role_name: str,
+        goal: Optional[str] = None,
+        backstory: Optional[str] = None,
+        domain_knowledge: Optional[str] = None,
+        llm_model: Optional[str] = None,
+        temperature: float = 0.7,
+        max_tokens: Optional[int] = None,
+        tools: Optional[list] = None,
+        **kwargs
+    ):
+        """
+        Initialize role configuration.
+
+        Args:
+            role_name: Role name
+            goal: Agent goal
+            backstory: Agent backstory
+            domain_knowledge: Domain knowledge
+            llm_model: LLM model to use
+            temperature: LLM temperature
+            max_tokens: Max tokens
+            tools: List of tool names
+            **kwargs: Additional configuration
+        """
+        self.role_name = role_name
+        self.goal = goal
+        self.backstory = backstory
+        self.domain_knowledge = domain_knowledge
+        self.llm_model = llm_model
+        self.temperature = temperature
+        self.max_tokens = max_tokens
+        self.tools = tools or []
+        self.additional_config = kwargs
+
+    def to_agent_config(self) -> AgentConfiguration:
+        """
+        Convert to AgentConfiguration.
+
+        Returns:
+            AgentConfiguration instance
+        """
+        return AgentConfiguration(
+            goal=self.goal,
+            backstory=self.backstory,
+            domain_knowledge=self.domain_knowledge,
+            llm_model=self.llm_model,
+            temperature=self.temperature,
+            max_tokens=self.max_tokens,
+            **self.additional_config
+        )
+
+    @classmethod
+    def load_from_file(cls, file_path: str) -> "RoleConfiguration":
+        """
+        Load role configuration from YAML file.
+
+        Args:
+            file_path: Path to YAML file
+
+        Returns:
+            RoleConfiguration instance
+        """
+        path = Path(file_path)
+        
+        if not path.exists():
+            raise FileNotFoundError(f"Role configuration file not found: {file_path}")
+
+        with open(path, 'r') as f:
+            data = yaml.safe_load(f)
+
+        logger.info(f"Loaded role configuration from {file_path}")
+        return cls(**data)
+
+    @classmethod
+    def load_from_dict(cls, data: Dict[str, Any]) -> "RoleConfiguration":
+        """
+        Load role configuration from dictionary.
+
+        Args:
+            data: Configuration dictionary
+
+        Returns:
+            RoleConfiguration instance
+        """
+        return cls(**data)
+
+    def merge_with(self, other: "RoleConfiguration") -> "RoleConfiguration":
+        """
+        Merge with another configuration (other takes precedence).
+
+        Args:
+            other: Other configuration
+
+        Returns:
+            New merged RoleConfiguration
+        """
+        merged_data = {
+            "role_name": other.role_name or self.role_name,
+            "goal": other.goal or self.goal,
+            "backstory": other.backstory or self.backstory,
+            "domain_knowledge": other.domain_knowledge or self.domain_knowledge,
+            "llm_model": other.llm_model or self.llm_model,
+            "temperature": other.temperature if other.temperature != 0.7 else self.temperature,
+            "max_tokens": other.max_tokens or self.max_tokens,
+            "tools": other.tools if other.tools else self.tools,
+            **{**self.additional_config, **other.additional_config},
+        }
+        return RoleConfiguration(**merged_data)
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary."""
+        return {
+            "role_name": self.role_name,
+            "goal": self.goal,
+            "backstory": self.backstory,
+            "domain_knowledge": self.domain_knowledge,
+            "llm_model": self.llm_model,
+            "temperature": self.temperature,
+            "max_tokens": self.max_tokens,
+            "tools": self.tools,
+            **self.additional_config,
+        }
+
+
+def load_role_config(
+    role_name: str,
+    base_path: str = "./agent_roles"
+) -> RoleConfiguration:
+    """
+    Load role configuration by name.
+
+    Args:
+        role_name: Role name
+        base_path: Base directory for role configurations
+
+    Returns:
+        RoleConfiguration instance
+    """
+    file_path = Path(base_path) / f"{role_name}.yaml"
+    return RoleConfiguration.load_from_file(str(file_path))
+
+
+# Predefined role templates
+ROLE_TEMPLATES = {
+    "developer": {
+        "role_name": "developer",
+        "goal": "Write clean, efficient, and maintainable code",
+        "backstory": "You are an experienced software developer with expertise in multiple programming languages and best practices",
+        "domain_knowledge": "Software development, design patterns, testing, debugging",
+        "temperature": 0.3,
+    },
+    "researcher": {
+        "role_name": "researcher",
+        "goal": "Gather, analyze, and synthesize information from various sources",
+        "backstory": "You are a meticulous researcher skilled at finding and evaluating information",
+        "domain_knowledge": "Research methodologies, critical analysis, information synthesis",
+        "temperature": 0.5,
+    },
+    "analyst": {
+        "role_name": "analyst",
+        "goal": "Analyze data and provide actionable insights",
+        "backstory": "You are a data analyst with strong analytical and problem-solving skills",
+        "domain_knowledge": "Data analysis, statistics, visualization, interpretation",
+        "temperature": 0.4,
+    },
+    "creative": {
+        "role_name": "creative",
+        "goal": "Generate creative and innovative ideas",
+        "backstory": "You are a creative thinker with a knack for innovative solutions",
+        "domain_knowledge": "Creative thinking, brainstorming, innovation",
+        "temperature": 0.9,
+    },
+}
+
+
+def get_role_template(role_name: str) -> RoleConfiguration:
+    """
+    Get predefined role template.
+
+    Args:
+        role_name: Role name
+
+    Returns:
+        RoleConfiguration instance
+    """
+    if role_name not in ROLE_TEMPLATES:
+        raise ValueError(f"Unknown role template: {role_name}")
+
+    return RoleConfiguration.load_from_dict(ROLE_TEMPLATES[role_name])
+

aiecs/domain/agent/lifecycle.py

@@ -0,0 +1,298 @@
+"""
+Agent Lifecycle Management
+
+Manages agent lifecycle transitions and state management.
+"""
+
+import logging
+from typing import Dict, Any, Optional, List
+from datetime import datetime
+
+from .base_agent import BaseAIAgent
+from .models import AgentState
+from .registry import AgentRegistry, get_global_registry
+from .exceptions import (
+    InvalidStateTransitionError,
+    AgentInitializationError,
+    AgentNotFoundError,
+)
+
+logger = logging.getLogger(__name__)
+
+
+class AgentLifecycleManager:
+    """
+    Manages agent lifecycle: creation, initialization, activation, deactivation, shutdown.
+
+    Ensures proper state transitions and coordinates with registry.
+    """
+
+    def __init__(self, registry: Optional[AgentRegistry] = None):
+        """
+        Initialize lifecycle manager.
+
+        Args:
+            registry: Optional custom registry (uses global if not provided)
+        """
+        self.registry = registry or get_global_registry()
+        logger.info("AgentLifecycleManager initialized")
+
+    async def create_and_initialize(self, agent: BaseAIAgent) -> BaseAIAgent:
+        """
+        Register and initialize an agent.
+
+        Args:
+            agent: Agent to initialize
+
+        Returns:
+            Initialized agent
+
+        Raises:
+            AgentInitializationError: If initialization fails
+        """
+        try:
+            # Register agent
+            self.registry.register(agent)
+            logger.info(f"Agent {agent.agent_id} registered")
+
+            # Initialize agent
+            await agent.initialize()
+            logger.info(f"Agent {agent.agent_id} initialized")
+
+            return agent
+
+        except Exception as e:
+            logger.error(f"Failed to create and initialize agent {agent.agent_id}: {e}")
+            
+            # Cleanup on failure
+            try:
+                if self.registry.exists(agent.agent_id):
+                    self.registry.unregister(agent.agent_id)
+            except Exception as cleanup_error:
+                logger.error(f"Cleanup failed: {cleanup_error}")
+
+            raise AgentInitializationError(
+                f"Agent initialization failed: {str(e)}",
+                agent_id=agent.agent_id
+            )
+
+    async def activate(self, agent_id: str) -> None:
+        """
+        Activate an agent.
+
+        Args:
+            agent_id: Agent identifier
+
+        Raises:
+            AgentNotFoundError: If agent not found
+            InvalidStateTransitionError: If activation not allowed
+        """
+        agent = self.registry.get(agent_id)
+        
+        try:
+            await agent.activate()
+            logger.info(f"Agent {agent_id} activated")
+        except Exception as e:
+            logger.error(f"Failed to activate agent {agent_id}: {e}")
+            raise
+
+    async def deactivate(self, agent_id: str) -> None:
+        """
+        Deactivate an agent.
+
+        Args:
+            agent_id: Agent identifier
+
+        Raises:
+            AgentNotFoundError: If agent not found
+            InvalidStateTransitionError: If deactivation not allowed
+        """
+        agent = self.registry.get(agent_id)
+        
+        try:
+            await agent.deactivate()
+            logger.info(f"Agent {agent_id} deactivated")
+        except Exception as e:
+            logger.error(f"Failed to deactivate agent {agent_id}: {e}")
+            raise
+
+    async def shutdown(self, agent_id: str, unregister: bool = True) -> None:
+        """
+        Shutdown an agent.
+
+        Args:
+            agent_id: Agent identifier
+            unregister: Whether to unregister after shutdown
+
+        Raises:
+            AgentNotFoundError: If agent not found
+        """
+        agent = self.registry.get(agent_id)
+        
+        try:
+            await agent.shutdown()
+            logger.info(f"Agent {agent_id} shut down")
+
+            if unregister:
+                self.registry.unregister(agent_id)
+                logger.info(f"Agent {agent_id} unregistered")
+
+        except Exception as e:
+            logger.error(f"Failed to shutdown agent {agent_id}: {e}")
+            raise
+
+    async def restart(self, agent_id: str) -> None:
+        """
+        Restart an agent (deactivate → initialize → activate).
+
+        Args:
+            agent_id: Agent identifier
+
+        Raises:
+            AgentNotFoundError: If agent not found
+        """
+        agent = self.registry.get(agent_id)
+        
+        try:
+            # Deactivate
+            if agent.state in [AgentState.ACTIVE, AgentState.BUSY]:
+                await agent.deactivate()
+
+            # Re-initialize
+            await agent.initialize()
+
+            # Re-activate
+            await agent.activate()
+
+            logger.info(f"Agent {agent_id} restarted")
+
+        except Exception as e:
+            logger.error(f"Failed to restart agent {agent_id}: {e}")
+            raise
+
+    async def shutdown_all(self) -> Dict[str, Any]:
+        """
+        Shutdown all registered agents.
+
+        Returns:
+            Dictionary with shutdown results
+        """
+        results = {
+            "success": [],
+            "failed": [],
+            "total": self.registry.count(),
+        }
+
+        agents = self.registry.list_all()
+        
+        for agent in agents:
+            try:
+                await self.shutdown(agent.agent_id, unregister=True)
+                results["success"].append(agent.agent_id)
+            except Exception as e:
+                logger.error(f"Failed to shutdown agent {agent.agent_id}: {e}")
+                results["failed"].append({
+                    "agent_id": agent.agent_id,
+                    "error": str(e),
+                })
+
+        logger.info(
+            f"Shutdown all agents: {len(results['success'])} succeeded, "
+            f"{len(results['failed'])} failed"
+        )
+
+        return results
+
+    def get_agent_status(self, agent_id: str) -> Dict[str, Any]:
+        """
+        Get agent status information.
+
+        Args:
+            agent_id: Agent identifier
+
+        Returns:
+            Status dictionary
+
+        Raises:
+            AgentNotFoundError: If agent not found
+        """
+        agent = self.registry.get(agent_id)
+        
+        return {
+            "agent_id": agent.agent_id,
+            "name": agent.name,
+            "type": agent.agent_type.value,
+            "state": agent.state.value,
+            "version": agent.version,
+            "created_at": agent.created_at.isoformat() if agent.created_at else None,
+            "last_active_at": agent.last_active_at.isoformat() if agent.last_active_at else None,
+            "current_task_id": agent._current_task_id,
+            "metrics": {
+                "total_tasks_executed": agent.get_metrics().total_tasks_executed,
+                "successful_tasks": agent.get_metrics().successful_tasks,
+                "failed_tasks": agent.get_metrics().failed_tasks,
+                "average_execution_time": agent.get_metrics().average_execution_time,
+                "total_tool_calls": agent.get_metrics().total_tool_calls,
+            },
+        }
+
+    def list_agent_statuses(
+        self,
+        agent_type: Optional[str] = None,
+        state: Optional[str] = None
+    ) -> "List[Dict[str, Any]]":
+        """
+        List agent statuses with optional filtering.
+
+        Args:
+            agent_type: Optional filter by agent type
+            state: Optional filter by state
+
+        Returns:
+            List of status dictionaries
+        """
+        agents = self.registry.list_all()
+
+        # Filter by type
+        if agent_type:
+            from .models import AgentType
+            try:
+                type_enum = AgentType(agent_type)
+                agents = [a for a in agents if a.agent_type == type_enum]
+            except ValueError:
+                logger.warning(f"Invalid agent type: {agent_type}")
+
+        # Filter by state
+        if state:
+            from .models import AgentState
+            try:
+                state_enum = AgentState(state)
+                agents = [a for a in agents if a.state == state_enum]
+            except ValueError:
+                logger.warning(f"Invalid state: {state}")
+
+        return [self.get_agent_status(a.agent_id) for a in agents]
+
+
+# Global lifecycle manager
+_global_lifecycle_manager: Optional[AgentLifecycleManager] = None
+
+
+def get_global_lifecycle_manager() -> AgentLifecycleManager:
+    """
+    Get or create global lifecycle manager.
+
+    Returns:
+        Global AgentLifecycleManager instance
+    """
+    global _global_lifecycle_manager
+    if _global_lifecycle_manager is None:
+        _global_lifecycle_manager = AgentLifecycleManager()
+    return _global_lifecycle_manager
+
+
+def reset_global_lifecycle_manager() -> None:
+    """Reset global lifecycle manager (primarily for testing)."""
+    global _global_lifecycle_manager
+    _global_lifecycle_manager = None
+