Airflow + Klavis AI: Complete MCP Integration
Airflow is a MCP server that A MCP Server that connects to Apache Airflow using official python client..
When integrated with Klavis AI, 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 Klavis AI, 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 Klavis AI
- 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
- Klavis AI installed and configured
- Compatible operating system (Cloud (klavis.ai), Self-Hosted (Docker), Python SDK, TypeScript SDK)
Installation
Step 1: Install Airflow
Configuration
Step 2: Configure Klavis AI
- Open Klavis AI 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 Klavis AI:
Pipeline Monitoring
undefined
Ask Klavis AI: "Check status of all DAGs and recent failures"
Expected Result: undefined
Manual DAG Execution
undefined
Ask Klavis AI: "Trigger monthly_report DAG with date parameter"
Expected Result: undefined
Troubleshooting Failed Tasks
undefined
Ask Klavis AI: "Show failed tasks in data_pipeline and their error logs"
Expected Result: undefined
Testing Your Setup
- Launch Klavis AI
- Verify Airflow is available in the tools list
- Test basic Airflow functionality
Troubleshooting
Common Issues
API Authentication Failed
Symptoms: 401 Unauthorized errors, Invalid API key message
Cause: Missing or incorrect API key
Solution:
- Verify API key is correctly copied from klavis.ai dashboard
- Check API key is set in environment variable or code
- Ensure API key has not expired
- Regenerate API key if necessary
OAuth Authorization Errors
Symptoms: OAuth redirect failures, Authentication loop, Token refresh errors
Cause: OAuth credentials not configured or expired
Solution:
- Complete OAuth authorization flow for required services
- Check redirect URLs are correctly configured
- Verify OAuth app credentials in service provider settings
- Contact Klavis support for OAuth troubleshooting
Tool Call Failures
Symptoms: Tool not found errors, Execution timeouts, Rate limit errors
Cause: Tool not properly configured or service rate limits exceeded
Solution:
- Verify tool name matches available tools list
- Check service-specific rate limits and quotas
- Ensure required parameters are provided correctly
- Review Klavis API logs for detailed error messages
Context Window Overload
Symptoms: Too many tools error, Context length exceeded, Slow responses
Cause: Too many tools exposed to AI agent at once
Solution:
- Use Strata Progressive Discovery to expose tools hierarchically
- Reduce number of simultaneous MCP servers in Strata configuration
- Filter tools based on current task context
- Enable intelligent tool filtering in Strata settings
Docker Deployment Issues
Symptoms: Container fails to start, Port conflicts, Connection refused
Cause: Docker configuration or port binding errors
Solution:
- Ensure Docker is installed and running
- Check port 5000 is not already in use (or change port mapping)
- Verify Docker image pulled successfully
- Check container logs: docker logs <container-id>
Airflow not appearing in Klavis AI
Symptoms: Server not listed, Tools not available
Cause: Configuration or installation issue
Solution:
- Verify configuration syntax
- Check Airflow installation
- Restart Klavis AI
- Check logs for error messages
Next Steps
Now that Airflow is integrated with Klavis AI:
- Explore all Airflow capabilities through Klavis AI
- Check out other MCP servers that work with Klavis AI
- Join the MCP community for tips and support
- Consider contributing to Airflow development
Need Help?
- Search for Airflow documentation
- Check the Klavis AI MCP guide
- Join the MCP community discussions