MCP Server Airflow Token

Instructions

Implementation Reference

• src/airflow/dataset.py:162-171 (handler)

  The main handler function that implements the tool logic. It calls the Airflow DatasetApi to delete queued events for a given dataset URI, optionally before a certain timestamp.

  async def delete_dataset_queued_events(
      uri: str,
      before: Optional[str] = None,
  ) -> List[Union[types.TextContent, types.ImageContent, types.EmbeddedResource]]:
      kwargs: Dict[str, Any] = {}
      if before is not None:
          kwargs["before"] = before
  
      response = dataset_api.delete_dataset_queued_events(uri=uri, **kwargs)
      return [types.TextContent(type="text", text=str(response.to_dict()))]

• src/airflow/dataset.py:34-38 (registration)

  Registration of the tool in the get_all_functions() list, which is used to register MCP tools.

      delete_dataset_queued_events,
      "delete_dataset_queued_events",
      "Delete queued Dataset events for a Dataset",
      False,
  ),