MCP Server Airflow Token

Instructions

Implementation Reference

• src/airflow/provider.py:18-41 (handler)

  The main asynchronous handler function that implements the logic for the 'get_providers' tool. It constructs parameters from inputs, calls the external ProviderApi, and formats the response as MCP TextContent.
  async def get_providers( limit: Optional[int] = None, offset: Optional[int] = None, ) -> List[Union[types.TextContent, types.ImageContent, types.EmbeddedResource]]: """ Get a list of providers. Args: limit: The numbers of items to return. offset: The number of items to skip before starting to collect the result set. Returns: A list of providers with their details. """ # Build parameters dictionary kwargs: Dict[str, Any] = {} if limit is not None: kwargs["limit"] = limit if offset is not None: kwargs["offset"] = offset response = provider_api.get_providers(**kwargs) return [types.TextContent(type="text", text=str(response.to_dict()))]
• src/airflow/provider.py:11-16 (registration)

  Module-level registration function that returns the tuple defining the 'get_providers' tool for inclusion in the MCP toolset.
  def get_all_functions() -> list[tuple[Callable, str, str, bool]]: """Return list of (function, name, description, is_read_only) tuples for registration.""" return [ (get_providers, "get_providers", "Get a list of loaded providers", True), ]
• src/main.py:90-92 (registration)

  The generic loop in main() that registers all collected tools (including 'get_providers' from providers) to the MCP app using app.add_tool.
  for func, name, description, *_ in functions: app.add_tool(func, name=name, description=description)
• src/main.py:22-38 (registration)

  Central mapping that links APIType.PROVIDER to the provider module's get_all_functions, enabling registration of 'get_providers'.
  APITYPE_TO_FUNCTIONS = { APIType.CONFIG: get_config_functions, APIType.CONNECTION: get_connection_functions, APIType.DAG: get_dag_functions, APIType.DAGRUN: get_dagrun_functions, APIType.DAGSTATS: get_dagstats_functions, APIType.DATASET: get_dataset_functions, APIType.EVENTLOG: get_eventlog_functions, APIType.IMPORTERROR: get_importerror_functions, APIType.MONITORING: get_monitoring_functions, APIType.PLUGIN: get_plugin_functions, APIType.POOL: get_pool_functions, APIType.PROVIDER: get_provider_functions, APIType.TASKINSTANCE: get_taskinstance_functions, APIType.VARIABLE: get_variable_functions, APIType.XCOM: get_xcom_functions, }
• src/main.py:16-16 (registration)

  Import of the provider module's get_all_functions, which includes 'get_providers', into the main registration system.
  from src.airflow.provider import get_all_functions as get_provider_functions