MCP Server Airflow Token

Instructions

Implementation Reference

• src/airflow/connection.py:41-70 (handler)

  The async handler function that implements the create_connection tool by building a connection request and calling the Airflow ConnectionApi to create the connection.

  async def create_connection(
      conn_id: str,
      conn_type: str,
      host: Optional[str] = None,
      port: Optional[int] = None,
      login: Optional[str] = None,
      password: Optional[str] = None,
      schema: Optional[str] = None,
      extra: Optional[str] = None,
  ) -> List[Union[types.TextContent, types.ImageContent, types.EmbeddedResource]]:
      connection_request = {
          "connection_id": conn_id,
          "conn_type": conn_type,
      }
      if host is not None:
          connection_request["host"] = host
      if port is not None:
          connection_request["port"] = port
      if login is not None:
          connection_request["login"] = login
      if password is not None:
          connection_request["password"] = password
      if schema is not None:
          connection_request["schema"] = schema
      if extra is not None:
          connection_request["extra"] = extra
  
      response = connection_api.post_connection(connection_request=connection_request)
      return [types.TextContent(type="text", text=str(response.to_dict()))]

• src/airflow/connection.py:11-20 (registration)

  Registers the create_connection tool (along with other connection tools) by including it in the list returned by get_all_functions(), which is used by main.py to add tools to the MCP server.

  def get_all_functions() -> list[tuple[Callable, str, str, bool]]:
      """Return list of (function, name, description, is_read_only) tuples for registration."""
      return [
          (list_connections, "list_connections", "List all connections", True),
          (create_connection, "create_connection", "Create a connection", False),
          (get_connection, "get_connection", "Get a connection by ID", True),
          (update_connection, "update_connection", "Update a connection by ID", False),
          (delete_connection, "delete_connection", "Delete a connection by ID", False),
          (test_connection, "test_connection", "Test a connection", True),
      ]

• src/main.py:90-92 (registration)

  The loop in main() that registers all tools from connection functions (including create_connection) by calling app.add_tool().

  for func, name, description, *_ in functions:
      app.add_tool(func, name=name, description=description)