OpenAI

apache-airflow-providers-openai

OpenAI provider for Apache Airflow. Enables interaction with OpenAI APIs for text generation, embeddings, and other AI-powered workflows directly from Airflow DAGs.

Works with Airflow 2.11+

Install:

pip install apache-airflow-providers-openai==1.7.4

Version:

Airflow

2.11+

Python

>=3.10

Dependencies (3)

Show all Hide

apache-airflow>=2.11.0 apache-airflow-providers-common-compat>=1.12.0 openai[datalib]>=1.66.0

Connections (1)

Modules

OpenAIEmbeddingOperator

Operator that accepts input text to generate OpenAI embeddings using the specified model.

airflow.providers.openai.operators.openai.OpenAIEmbeddingOperator

View Docs Source

OpenAITriggerBatchOperator

Operator that triggers an OpenAI Batch API endpoint and waits for the batch to complete.

airflow.providers.openai.operators.openai.OpenAITriggerBatchOperator

View Docs Source

OpenAIHook

Use OpenAI SDK to interact with OpenAI APIs.

airflow.providers.openai.hooks.openai.OpenAIHook

View Docs Source

OpenAIBatchTrigger

Triggers OpenAI Batch API.

airflow.providers.openai.triggers.openai.OpenAIBatchTrigger

View Docs Source