agent_manager_docstring.txt

AgentManager Class Docstrings
=============================

Class: AgentManager
-------------------
"""Manages the lifecycle and execution of agents within the orchestrator system.

The AgentManager is responsible for registering agents, tracking their status,
and coordinating task execution across multiple agents. It provides a centralized
interface for agent management and maintains state information for all registered
agents.

Attributes:
    event_bus (EventBus): The event bus instance for inter-component communication.
    agents (Dict[str, BaseAgent]): Dictionary mapping agent names to agent instances.
    status (Dict[str, AgentStatusEnum]): Dictionary tracking the current status of each agent.
"""

Method: __init__
----------------
"""Initialize the AgentManager with an event bus.

Args:
    event_bus (EventBus): The event bus instance used for communication
        between components in the orchestrator system.
"""

Method: register_agent
----------------------
"""Register a new agent with the manager.

Adds the agent to the internal registry and sets its initial status to idle.
The agent is identified by its name attribute and can be referenced for
task execution.

Args:
    agent (BaseAgent): The agent instance to register. Must have a valid
        name attribute that will be used as the unique identifier.
"""

Method: run_task
----------------
"""Execute a task using the specified agent.

Coordinates the execution of a task by the named agent, including status
tracking and event emission. The method handles the complete lifecycle
of task execution from start to completion or error.

Args:
    agent_name (str): The name of the registered agent to execute the task.
    task (Task): The task object containing all necessary information
        for task execution.

Returns:
    AgentOutput: The result of the task execution, containing any output
        data, artifacts, or results produced by the agent.

Raises:
    KeyError: If the specified agent_name is not found in the registered agents.
    Exception: Any exception raised during task execution is re-raised after
        updating the agent status and emitting error events.
"""