| 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) |
| |
