AI integrations¶

AI / LLM¶

These integrations allow you to interact with AI and LLM services.

Common AI¶

Operators:: airflow.providers.common.ai.operators.agent, airflow.providers.common.ai.operators.llm, airflow.providers.common.ai.operators.llm_file_analysis, airflow.providers.common.ai.operators.llm_branch, airflow.providers.common.ai.operators.llm_sql, airflow.providers.common.ai.operators.llm_schema_compare, airflow.providers.common.ai.operators.document_loader, airflow.providers.common.ai.operators.llamaindex_embedding, airflow.providers.common.ai.operators.llamaindex_retrieval.
Guides:: AgentOperator & @task.agent, LLMOperator, LLMFileAnalysisOperator & @task.llm_file_analysis, LLMBranchOperator, LLMSQLQueryOperator, LLMSchemaCompareOperator, DocumentLoaderOperator.
Provider:: apache-airflow-providers-common-ai
Product documentation:: Common AI

LangChain¶

Hooks:: airflow.providers.common.ai.hooks.langchain.
Provider:: apache-airflow-providers-common-ai
Product documentation:: LangChain

LlamaIndex¶

Hooks:: airflow.providers.common.ai.hooks.llamaindex.
Guides:: LlamaIndex LlamaIndexEmbeddingOperator, LlamaIndex LlamaIndexRetrievalOperator.
Provider:: apache-airflow-providers-common-ai
Product documentation:: LlamaIndex

MCP Server¶

Hooks:: airflow.providers.common.ai.hooks.mcp.
Provider:: apache-airflow-providers-common-ai
Product documentation:: MCP Server

OpenAI¶

Operators:: airflow.providers.openai.operators.openai.
Hooks:: airflow.providers.openai.hooks.openai.
Guides:: OpenAIEmbeddingOperator.
Provider:: apache-airflow-providers-openai
Product documentation:: OpenAI

Pydantic AI¶

Hooks:: airflow.providers.common.ai.hooks.pydantic_ai.
Provider:: apache-airflow-providers-common-ai
Product documentation:: Pydantic AI

Was this entry helpful?