https://github.com/vojay-dev/pydantic-airflow-agent

https://github.com/vojay-dev/pydantic-airflow-agent

Last synced: about 1 month ago
JSON representation

• Host: GitHub
• URL: https://github.com/vojay-dev/pydantic-airflow-agent
• Owner: vojay-dev
• Created: 2024-12-23T15:36:16.000Z (5 months ago)
• Default Branch: main
• Last Pushed: 2025-02-27T16:40:47.000Z (3 months ago)
• Last Synced: 2025-03-27T21:52:15.703Z (about 2 months ago)
• Language: Python
• Size: 669 KB
• Stars: 16
• Watchers: 1
• Forks: 7
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# PydanticAI Airflow Agent Demo

A demo project showing how to build an AI agent that interacts with Apache Airflow using natural language

queries. Built with PydanticAI and Gemini 2.0.

> "What's the status of our payment report DAG?" - Now you can simply ask your agent! 🚀

![AI Airflow Agent](doc/ai-agent-aviation.jpg)

## Overview

This project demonstrates how to:

- Create a PydanticAI agent that interacts with Airflow's API

- Handle natural language queries about DAG statuses

- Return structured, type-safe responses

- Implement function tools for models to retrieve extra information

- Use Gemini 2.0 for enhanced language understanding

## Prerequisites

- Python 3.12

- Poetry

- Docker

- Astro CLI

- Google Cloud credentials (for Gemini 2.0)

## Quick Start

```bash

# Install dependencies

poetry install

# Start Airflow

make airflow-start

# Run the agent

make run-agent

```