airflow-ai-sdk
An SDK for working with LLMs and AI Agents from Apache Airflow, based on Pydantic AI
File Explorer
Download Latest Version (.zip)- ci.yaml
- release.yaml
- dependabot.yml
- agent.py
- branch.py
- embed.py
- llm.py
- base.py
- tool.py
- agent.py
- embed.py
- llm.py
- llm_branch.py
- __init__.py
- airflow.py
- index.md
- agent.md
- branch.md
- embed.md
- llm.md
- base.md
- tool.md
- agent.md
- embed.md
- llm.md
- llm_branch.md
- airflow.md
- README.md
- features.md
- index.md
- README.md
- usage.md
- config.yaml
- dag_integrity_exceptions.txt
- test_dag_integrity_default.py
- custom_model.py
- deep_research.py
- email_generation.py
- example_dag.py
- github_changelog.py
- product_feedback_summarization.py
- sentiment_classification.py
- support_ticket_routing.py
- text_embedding.py
- .dockerignore
- .gitignore
- docker-compose.yaml
- Dockerfile
- packages.txt
- README.md
- requirements.txt
- generate_interface_docs.py
- test_agent.py
- test_embed.py
- test_llm.py
- test_llm_branch.py
- __init__.py
- test_airflow_imports.py
- .gitignore
- .pre-commit-config.yaml
- .python-version
- AGENTS.md
- LICENSE
- pyproject.toml
- README.md
- Tiltfile
- uv.lock
๐ Installation Guide
1. Get the code
git clone https://github.com/astronomer/airflow-ai-sdk
Downloads the entire project code from GitHub to your computer.
cd airflow-ai-sdk
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install airflow-ai-sdk[openai]
Installs the package published on PyPI directly โ no need to clone the source.
After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.
Pulled directly from this repo's README.
3. Docker
EasyPrerequisites
- Git Needed to download the project code from GitHub.
- Docker Desktop Needed to build and run containers. Install it and keep it running in the background.
docker compose -f examples/docker-compose.yaml up -d --build
Runs the command against the services defined in the compose file.
Run docker compose ps to check the containers are Up. If the README mentions a port, open http://localhost:PORT in your browser.
4. Python
EasyPrerequisites
pip install airflow-ai-sdk[openai]
Installs the package published on PyPI directly โ no need to clone the source.
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)