MCP Server Airflow Token

Instructions

Implementation Reference

• src/airflow/variable.py:54-56 (handler)

  The main handler function for the 'get_variable' tool, which retrieves an Airflow variable by key using the VariableApi and returns its dict as text content.
  async def get_variable(key: str) -> List[Union[types.TextContent, types.ImageContent, types.EmbeddedResource]]: response = variable_api.get_variable(variable_key=key) return [types.TextContent(type="text", text=str(response.to_dict()))]
• src/airflow/variable.py:11-19 (registration)

  The function that returns the list of tool registrations, including the specific tuple for 'get_variable'.
  def get_all_functions() -> list[tuple[Callable, str, str, bool]]: """Return list of (function, name, description, is_read_only) tuples for registration.""" return [ (list_variables, "list_variables", "List all variables", True), (create_variable, "create_variable", "Create a variable", False), (get_variable, "get_variable", "Get a variable by key", True), (update_variable, "update_variable", "Update a variable by key", False), (delete_variable, "delete_variable", "Delete a variable by key", False), ]
• src/main.py:90-92 (registration)

  Loop that registers each tool from the variable module (and others) by calling app.add_tool() with the function, name, and description.
  for func, name, description, *_ in functions: app.add_tool(func, name=name, description=description)
• src/main.py:36-36 (registration)

  Mapping of APIType.VARIABLE to the get_variable_functions which provides the tool list including 'get_variable'.
  APIType.VARIABLE: get_variable_functions,
• src/main.py:18-18 (registration)

  Import of the function provider for variable tools in main.py.
  from src.airflow.variable import get_all_functions as get_variable_functions