from .agent import Agent
from ..actions.task_planning import TaskPlanning
from ..prompts.task_planner import TASK_PLANNER


class TaskPlanner(Agent):
    """An agent responsible for planning and decomposing high-level tasks into smaller sub-tasks.
    
    The TaskPlanner agent analyzes complex goals and breaks them down into a structured
    sequence of smaller, more manageable tasks. It serves as a critical component in the
    workflow by creating execution plans that other specialized agents can follow.
    
    Attributes:
        name (str): Name of the task planner agent, defaults to the value in TASK_PLANNER
        description (str): Description of the agent's purpose and capabilities, defaults to the value in TASK_PLANNER
        system_prompt (str): System prompt guiding the agent's behavior, defaults to the value in TASK_PLANNER
        actions (List[Action]): List of actions the agent can perform, defaults to [TaskPlanning()]
    """
    def __init__(self, **kwargs):

        name = kwargs.pop("name") if "name" in kwargs else TASK_PLANNER["name"]
        description = kwargs.pop("description") if "description" in kwargs else TASK_PLANNER["description"]
        system_prompt = kwargs.pop("system_prompt") if "system_prompt" in kwargs else TASK_PLANNER["system_prompt"]
        actions = kwargs.pop("actions") if "actions" in kwargs else [TaskPlanning()]
        super().__init__(name=name, description=description, system_prompt=system_prompt, actions=actions, **kwargs)
    
    @property
    def task_planning_action_name(self):
        """Get the name of the TaskPlanning action associated with this agent.
        
        Returns:
            The name of the TaskPlanning action in this agent's action registry
        """
        return self.get_action_name(action_cls=TaskPlanning)