Airflow + Augment Code: Complete MCP Integration
Airflow is a MCP server that A MCP Server that connects to Apache Airflow using official python client..
When integrated with Augment Code, you can:
- List all available DAGs in Airflow
- Manually trigger a DAG run
- Get DAG run history and status
This guide provides step-by-step instructions to set up Airflow in Augment Code, including configuration, examples, and troubleshooting.
What You'll Achieve
After completing this setup:
- Airflow will be fully integrated and operational
- You can use Airflow tools directly in Augment Code
- All Airflow capabilities will be available for your workflows
- Access to 5 different tools
Prerequisites
Before starting, ensure you have:
- Airflow webserver host URL
- Airflow authentication username
- Airflow authentication password
- Augment Code installed and configured
- Compatible operating system (Windows, macOS, Linux)
Installation
Step 1: Install Airflow
Configuration
Step 2: Configure Augment Code
- Open Augment Code settings
- Navigate to MCP server configuration
- Add Airflow server with appropriate settings
- Save and restart if needed
Examples
Once configured, you can use Airflow in Augment Code:
Pipeline Monitoring
undefined
Ask Augment Code: "Check status of all DAGs and recent failures"
Expected Result: undefined
Manual DAG Execution
undefined
Ask Augment Code: "Trigger monthly_report DAG with date parameter"
Expected Result: undefined
Troubleshooting Failed Tasks
undefined
Ask Augment Code: "Show failed tasks in data_pipeline and their error logs"
Expected Result: undefined
Testing Your Setup
- Launch Augment Code
- Verify Airflow is available in the tools list
- Test basic Airflow functionality
Troubleshooting
Common Issues
Installation Failed
Symptoms: npm install errors, Command not found
Cause: Node.js version incompatibility or PATH issues
Solution:
- Verify Node.js version 16 or higher is installed
- Check npm installation and PATH configuration
- Try using yarn instead of npm
- Clear npm cache and reinstall
MCP Server Not Found
Symptoms: Server command not found, Configuration errors
Cause: Server not installed or not in PATH
Solution:
- Verify MCP server installation
- Use absolute path to server executable
- Check server executable permissions
- Test server command independently
Repository Analysis Fails
Symptoms: Analysis errors, No improvements suggested
Cause: MCP server integration or codebase issues
Solution:
- Check repository is properly initialized
- Verify MCP server is responding correctly
- Review analysis logs for specific errors
- Start with smaller codebase for testing
Airflow not appearing in Augment Code
Symptoms: Server not listed, Tools not available
Cause: Configuration or installation issue
Solution:
- Verify configuration syntax
- Check Airflow installation
- Restart Augment Code
- Check logs for error messages
Next Steps
Now that Airflow is integrated with Augment Code:
- Explore all Airflow capabilities through Augment Code
- Check out other MCP servers that work with Augment Code
- Join the MCP community for tips and support
- Consider contributing to Airflow development
Need Help?
- Search for Airflow documentation
- Check the Augment Code MCP guide
- Join the MCP community discussions