Airflow Summit 2026 is coming August 31 - September 2 in Austin, TX. Register now to secure your spot!

apache-airflow-providers-common-ai

apache-airflow-providers-common-ai package

AI/LLM hooks and operators for Airflow pipelines using pydantic-ai.

Release: 0.6.0

Provider package

This package is for the common.ai provider. All classes for this package are included in the airflow.providers.common.ai python package.

Installation

You can install this package on top of an existing Airflow installation via pip install apache-airflow-providers-common-ai. For the minimum Airflow version supported, see Requirements below.

Requirements

The minimum Apache Airflow version supported by this provider distribution is 3.0.0.

PIP package

Version required

apache-airflow

>=3.0.0

apache-airflow-providers-common-compat

>=1.15.0

apache-airflow-providers-standard

>=1.12.1

pydantic-ai-slim

>=2.0.0

Optional cross provider package dependencies

Those are dependencies that might be needed in order to use all the features of the package. You need to install the specified provider distributions in order to use them.

You can install such cross-provider dependencies when installing from PyPI. For example:

pip install apache-airflow-providers-common-ai[common.sql]

Dependent package

Extra

apache-airflow-providers-common-sql

common.sql

apache-airflow-providers-git

git

Optional dependencies

These extras install optional third-party libraries that enable additional features of the provider. Install them when installing from PyPI. For example:

pip install apache-airflow-providers-common-ai[anthropic]

Extra

Dependencies

anthropic

pydantic-ai-slim[anthropic]

bedrock

pydantic-ai-slim[bedrock]

google

pydantic-ai-slim[google]

openai

pydantic-ai-slim[openai]

mcp

pydantic-ai-slim[mcp]

code-mode

pydantic-ai-harness[codemode]>=0.3.0

skills

apache-airflow-providers-git>=0.4.0, pydantic-ai-skills>=0.11.0

avro

fastavro>=1.10.0; python_version < "3.14", fastavro>=1.12.1; python_version >= "3.14"

parquet

pyarrow>=18.0.0; python_version < '3.14', pyarrow>=22.0.0; python_version >= '3.14'

sql

apache-airflow-providers-common-sql, sqlglot>=30.0.0

common.sql

apache-airflow-providers-common-sql

langchain

langchain>=1.0.0

llamaindex

llama-index-core>=0.13.0, llama-index-embeddings-openai>=0.6.0, llama-index-llms-openai>=0.6.0

pdf

pypdf>=4.0.0

docx

python-docx>=1.0.0

git

apache-airflow-providers-git

Downloading official packages

You can download officially released packages and verify their checksums and signatures from the Official Apache Download site

Was this entry helpful?