MCP Server for Apache Airflow

Instructions

Implementation Reference

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

  The main handler function for the 'delete_dataset_queued_events' tool. It accepts a dataset URI and optional 'before' timestamp, constructs kwargs, calls the underlying DatasetApi to delete queued events, and returns the response as text content.

  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 'delete_dataset_queued_events' tool within the get_all_functions() list, including the function reference, name, description, and mutability flag.

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