list_tasks
Retrieve detailed task information for a specific DAG within Apache Airflow. The tool extracts task lists, including IDs and counts, to simplify DAG inspection and task monitoring.
Instructions
[Tool Role]: Lists all tasks for a specific DAG.
Args: dag_id: The DAG ID to get tasks for
Returns: List of tasks with detailed task information: dag_id, tasks, total_tasks
Input Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| dag_id | Yes |
Implementation Reference
- The core handler function for the 'list_tasks' tool. It makes an API request to retrieve all tasks for a given DAG ID and returns the JSON response.@mcp.tool() async def list_tasks(dag_id: str) -> Dict[str, Any]: """[Tool Role]: Lists all tasks within the specified DAG.""" if not dag_id: raise ValueError("dag_id must not be empty") resp = await airflow_request("GET", f"/dags/{dag_id}/tasks") resp.raise_for_status() return resp.json()
- src/mcp_airflow_api/tools/v1_tools.py:13-27 (registration)Registration entry point for v1 API tools. Sets the v1-specific HTTP request function and calls register_common_tools(mcp), which defines and registers the list_tasks handler.def register_tools(mcp): """Register v1 tools by importing common tools with v1 request function.""" logger.info("Initializing MCP server for Airflow API v1") logger.info("Loading Airflow API v1 tools (Airflow 2.x)") # Set the global request function to v1 common_tools.airflow_request = airflow_request_v1 # Register all 56 common tools (includes management tools) common_tools.register_common_tools(mcp) # V1 has no exclusive tools - all tools are shared with v2 logger.info("Registered all Airflow API v1 tools (56 tools: 43 core + 13 management tools)")
- src/mcp_airflow_api/tools/v2_tools.py:14-25 (registration)Registration entry point for v2 API tools. Sets the v2-specific HTTP request function and calls register_common_tools(mcp), which defines and registers the list_tasks handler.def register_tools(mcp): """Register v2 tools: common tools + v2-exclusive asset tools.""" logger.info("Initializing MCP server for Airflow API v2") logger.info("Loading Airflow API v2 tools (Airflow 3.0+)") # Set the global request function to v2 common_tools.airflow_request = airflow_request_v2 # Register all 43 common tools common_tools.register_common_tools(mcp)
- src/mcp_airflow_api/mcp_main.py:263-272 (registration)Top-level registration in mcp_main.py where API version is detected and the appropriate v1_tools.register_tools or v2_tools.register_tools is called to register all tools including list_tasks.if api_version == "v1": logger.info("Loading Airflow API v1 tools (Airflow 2.x)") from mcp_airflow_api.tools import v1_tools v1_tools.register_tools(mcp_instance) elif api_version == "v2": logger.info("Loading Airflow API v2 tools (Airflow 3.0+)") from mcp_airflow_api.tools import v2_tools v2_tools.register_tools(mcp_instance) else: raise ValueError(f"Unsupported API version: {api_version}. Use 'v1' or 'v2'")