create_connection
Configure and establish connections in Apache Airflow by specifying connection parameters like host, port, and credentials for data pipeline integrations.
Instructions
Create a connection
Input Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| conn_id | Yes | ||
| conn_type | Yes | ||
| host | No | ||
| port | No | ||
| login | No | ||
| password | No | ||
| schema | No | ||
| extra | No |
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)