Pulkit0111/db_rag_mcp

🚀 Natural Language SQL MCP Server v2.0.0

The most comprehensive AI-powered database interface - Chat with your database using plain English!

Transform natural language into SQL queries, visualize data, export results, and manage multiple databases with enterprise-grade features. This advanced MCP server provides a complete database interaction ecosystem with AI-powered intelligence.

✨ What Makes This Special?

This isn't just another SQL translator. It's a complete database interaction platform that combines:

• 🧠 AI-Powered Query Intelligence - Smart suggestions, optimizations, and result explanations
• 🎨 Interactive Data Visualization - Beautiful charts and dashboards with Plotly
• 🔐 Enterprise Security - Full RBAC with user authentication and session management
• 🗄️ Multi-Database Support - PostgreSQL, MySQL, and SQLite
• 📊 Advanced Analytics - Query optimization, performance insights, and trend analysis
• 💾 Multiple Export Formats - CSV, JSON, Excel with metadata
• 🧭 Session Management - Query history, context awareness, and smart suggestions
• ⚡ High Performance - Redis caching, connection pooling, and optimized queries

Perfect for developers, data analysts, business intelligence teams, and enterprises who want to democratize database access!

🏗️ Architecture Overview

┌─────────────────┐    ┌──────────────────┐    ┌─────────────────┐
│   MCP Client    │    │   FastMCP Server │    │   Databases     │
│  (Cursor IDE)   │◄──►│     (38 Tools)   │◄──►│ PostgreSQL/     │
│                 │    │                  │    │ MySQL/SQLite    │
└─────────────────┘    └──────────────────┘    └─────────────────┘
                                │
                       ┌──────────────────┐
                       │  AI Intelligence │
                       │   OpenAI GPT-4   │
                       │  Query Analysis  │
                       │  Optimizations   │
                       └──────────────────┘

🛠️ Complete Feature Set (38 Tools)

🔌 Core Database Operations

• connect_database - Multi-database connection (PostgreSQL/MySQL/SQLite)
• disconnect_database - Safe connection management
• get_connection_status - Real-time connection monitoring

📊 Schema Intelligence

• list_tables - Smart table discovery with caching
• describe_table - Comprehensive schema analysis
• get_database_summary - AI-powered database overview

💬 Natural Language Queries

• query_data - Advanced NL to SQL with caching
• add_data - Intelligent data insertion
• update_data - Smart data modification
• delete_data - Safe data removal with validation

🧠 AI-Powered Query Intelligence

• explain_results - Natural language result explanations
• suggest_related_queries - Context-aware query suggestions
• optimize_query - Performance analysis and recommendations
• improve_query_language - Query phrasing improvements
• analyze_query_intent - Deep intent analysis and insights

📈 Advanced Query Features

• explain_query - Query execution planning and analysis
• query_with_suggestions - Queries with optimization hints
• aggregate_data - Specialized aggregation operations

📚 Session & History Management

• get_query_history - Rich query history with analytics
• repeat_query - One-click query re-execution

🔐 Enterprise Authentication & Security

• authenticate_user - Secure user authentication
• logout_user - Session management
• get_current_user - User profile and permissions
• create_user - User management (Admin)
• list_users - User administration (Admin)
• update_user_role - Role management (Admin)
• deactivate_user - Account management (Admin)
• check_permission - Permission validation

📊 Data Visualization

• create_visualization - Interactive Plotly charts
• recommend_visualizations - AI-suggested chart types
• create_dashboard - Multi-chart dashboards
• export_visualization - Chart export capabilities

💾 Data Export & Reporting

• export_csv - Enhanced CSV export with metadata
• export_json - Structured JSON export
• export_excel - Multi-sheet Excel workbooks
• export_multiple_formats - Bulk export operations

🔧 System & Utilities

• hello - Server connectivity test
• server_info - Comprehensive system status

🚀 Installation & Setup

Prerequisites

• Python 3.9+ (Download)
• Database (PostgreSQL/MySQL/SQLite)
• OpenAI API Key (Get one)
• Redis (optional, for caching) (Install guide)

Step 1: Clone & Install

git clone <your-repo-url>
cd db-rag

# Install all dependencies
pip install -r requirements.txt

# Install additional dependencies
pip install pydantic-settings redis

Step 2: Environment Configuration

Create a comprehensive .env file:

# ====================================
# DATABASE CONFIGURATION
# ====================================
DB_HOST=localhost
DB_PORT=5432
DB_USERNAME=postgres
DB_PASSWORD=your_password
DB_DATABASE=your_database
DB_TYPE=postgresql

# ====================================
# AI CONFIGURATION  
# ====================================
LLM_API_KEY=sk-your-openai-key-here
LLM_MODEL=gpt-4o-mini
LLM_MAX_TOKENS=1000
LLM_TEMPERATURE=0.1

# ====================================
# SERVER CONFIGURATION
# ====================================
MCP_SERVER_NAME=Natural Language SQL Server
MCP_HOST=127.0.0.1
MCP_PORT=8000
MCP_TRANSPORT=http

# ====================================
# FEATURE FLAGS
# ====================================
ENABLE_AUTHENTICATION=false
ENABLE_QUERY_CACHING=true
ENABLE_QUERY_HISTORY=true
ENABLE_SMART_SUGGESTIONS=true
ENABLE_VISUALIZATION=true

# ====================================
# PERFORMANCE & CACHING
# ====================================
CACHE_REDIS_URL=redis://localhost:6379
CACHE_TTL=300
QUERY_TIMEOUT=30
MAX_RESULT_ROWS=1000

# ====================================
# ENVIRONMENT
# ====================================
ENVIRONMENT=development
DEBUG=false

Step 3: Launch Server

python src/server.py

Expected startup output:

============================================================
🚀 NATURAL LANGUAGE SQL MCP SERVER v2.0.0
============================================================
✅ Configuration loaded successfully
   Database: postgresql at localhost:5432
   LLM Model: gpt-4o-mini

🔧 Feature Status:
   Authentication: ❌ Disabled
   Query Caching: ✅ Enabled
   Query History: ✅ Enabled
   AI Suggestions: ✅ Enabled
   Visualizations: ✅ Enabled

🔨 Tools Registered: 38 tools available

📊 Supported Databases: PostgreSQL, MySQL, SQLite
🤖 AI Features: OpenAI GPT-4o-mini (default)
📈 Visualization: Plotly-based interactive charts
💾 Export Formats: CSV, JSON, Excel
============================================================

📡 Starting Natural Language SQL Server with STDIO transport
   Ready for MCP client connections
============================================================

🔧 Integration with Cursor IDE

MCP Server Configuration

Add to your Cursor MCP settings:

{
  "mcpServers": {
    "natural-language-sql": {
      "name": "Natural Language SQL Server v2.0",
      "command": "python",
      "args": ["src/server.py"],
      "cwd": "/path/to/db-rag",
      "env": {
        "PYTHONPATH": "/path/to/db-rag"
      },
      "description": "Advanced AI-powered database interface with 38 tools",
      "enabled": true
    }
  }
}

Quick Start Conversation

You: Connect to my database and show me what tables I have

AI: I'll connect to your database and show you the available tables.
[Uses connect_database and list_tables tools]
Connected! You have 15 tables: users, orders, products, categories...

You: Show me sales trends for the last 3 months with a chart

AI: I'll create a visualization of your sales trends.
[Uses query_data and create_visualization tools]  
Here's an interactive line chart showing your sales growth...

You: Export this data to Excel with detailed formatting

AI: I'll export the sales data to Excel with metadata.
[Uses export_excel tool]
Exported 1,247 rows to sales_trends_20241220_143022.xlsx...

You: What other insights can you find in this data?

AI: Let me analyze the query results and suggest related insights.
[Uses explain_results and suggest_related_queries tools]
Based on your data, I found 3 key insights and suggest 5 related questions...

🎯 Advanced Use Cases

📊 Business Intelligence

# Revenue Analysis Dashboard
"Create a dashboard showing monthly revenue, top products, and customer segments"

# Performance Optimization  
"Analyze my slowest queries and suggest optimizations"

# Automated Reporting
"Export quarterly sales data to Excel with charts and pivot tables"

🔍 Data Exploration

# AI-Powered Discovery
"What interesting patterns do you see in my customer data?"

# Smart Suggestions
"Based on my order history, what questions should I ask next?"

# Context-Aware Analysis  
"Compare this month's performance with historical trends"

🛡️ Enterprise Security

# User Management
"Create analyst users with read-only permissions"

# Audit Trail
"Show me all database modifications in the last week"

# Permission Management
"What databases can the current user access?"

🏆 Key Advantages

🚀 Performance & Scalability

• Redis caching - Query results and schema cached for speed
• Connection pooling - Efficient database resource management
• Async operations - Non-blocking I/O for better throughput
• Smart optimization - AI-powered query performance suggestions

🔒 Enterprise Security

• Role-Based Access Control (RBAC) - Fine-grained permissions
• Session management - Secure user authentication
• SQL injection prevention - Parameterized queries
• Audit logging - Complete activity tracking

🧠 AI Intelligence

• Context awareness - Learns from query history
• Smart suggestions - Proactive query recommendations
• Result explanation - Natural language insights
• Query optimization - Performance improvement hints

📈 Rich Visualizations

• Interactive charts - Plotly-powered visualizations
• Smart recommendations - AI suggests best chart types
• Dashboard creation - Multi-chart dashboards
• Export capabilities - Charts as PNG, SVG, PDF

🔧 Developer Experience

• 38 comprehensive tools - Everything you need in one place
• Excellent error handling - User-friendly error messages
• Comprehensive documentation - Every tool documented
• Easy integration - Works with any MCP client

🎛️ Configuration Options

Feature Flags

Control exactly which features are enabled:

ENABLE_AUTHENTICATION=true    # User authentication
ENABLE_QUERY_CACHING=true     # Redis caching  
ENABLE_QUERY_HISTORY=true     # Session history
ENABLE_SMART_SUGGESTIONS=true # AI suggestions
ENABLE_VISUALIZATION=true     # Chart generation

Performance Tuning

CACHE_TTL=300                 # Cache timeout (seconds)
QUERY_TIMEOUT=30              # Query timeout (seconds)  
MAX_RESULT_ROWS=1000          # Maximum rows returned

Database Support

DB_TYPE=postgresql            # postgresql, mysql, sqlite

🏗️ Database Support Matrix

🐛 Troubleshooting

Common Issues

Server Won't Start?

# Check Python version
python --version  # Must be 3.9+

# Install missing dependencies
pip install -r requirements.txt
pip install pydantic-settings

# Check configuration
python -c "from src.core.config import config; print('Config OK')"

Database Connection Issues?

# Test database connection
python -c "
from src.database import create_database_manager
import asyncio
async def test():
    db = create_database_manager('postgresql', {
        'host': 'localhost', 'port': 5432, 
        'username': 'postgres', 'password': 'password', 
        'database': 'testdb'
    })
    print('Connected:', await db.connect())
asyncio.run(test())
"

AI Features Not Working?

• Verify OpenAI API key is valid
• Check API quota and billing
• Test with simple queries first

Visualizations Not Generated?

• Ensure matplotlib/plotly are installed
• Check data format and column types
• Try with smaller datasets first

📊 Performance Benchmarks

🛣️ Roadmap & Future Features

Phase 3 (Planned)

• 🌐 Web Interface - Browser-based query interface
• 📱 Mobile API - REST API for mobile applications
• 🔄 Real-time Sync - Live data synchronization
• 🤖 Advanced AI - Custom model training
• 📊 More Databases - MongoDB, Cassandra support

Phase 4 (Future)

• ☁️ Cloud Deployment - AWS/GCP/Azure support
• 🔐 SSO Integration - SAML/OAuth support
• 📈 Advanced Analytics - ML-powered insights
• 🌍 Multi-language - Support for multiple languages

🤝 Contributing

We welcome contributions! Areas where you can help:

• 🐛 Bug fixes and testing
• 📚 Documentation improvements
• 🔧 New database adapters
• 🎨 UI/UX enhancements
• 🧪 Test coverage expansion

📄 License

This project is licensed under the MIT License - see the file for details.

🎉 Ready to Transform Your Database Experience?

This isn't just a tool—it's a complete database interaction revolution. With 38 powerful tools, enterprise-grade security, AI intelligence, and beautiful visualizations, you're equipped to handle any data challenge.

Start your journey today:

git clone <your-repo-url>
cd db-rag
pip install -r requirements.txt
python src/server.py

Join thousands of developers, analysts, and enterprises who've revolutionized their database interactions! 🚀

Natural Language SQL MCP Server v2.0.0 - Making databases accessible to everyone ✨