SQL_MCP_Server

SQL_MCP_Server

7
databasesai_chatbot

SQLGenius is an AI-powered SQL assistant designed to help users interact with BigQuery databases through natural language queries. It leverages Vertex AI's Gemini Pro and Streamlit for intuitive data exploration and SQL execution. Its standout features include a natural language to SQL conversion, interactive UI, and robust security measures.

SQLGenius - AI-Powered SQL Assistant

SQLGenius is an intelligent SQL assistant that helps you query your BigQuery database using natural language. Built with MCP (Model Context Protocol), Vertex AI's Gemini Pro, and Streamlit.

🌟 Features

  • Natural language to SQL conversion using Gemini Pro
  • Interactive Streamlit UI with multiple tabs
  • Real-time query execution and visualization
  • Database schema explorer
  • Query history tracking
  • Safe query validation
  • BigQuery integration
  • MCP-based architecture

🎥 Demo

Watch SQLGenius in action! Here's a quick demo of how to use the application:

In this demo, you can see:

  1. Natural language query conversion to SQL
  2. Interactive data visualization
  3. Schema exploration
  4. Query history tracking

🎮 Usage

  1. Start the application:

  2. The MCP server will start automatically when the Streamlit app launches

  3. Use the tabs to:

    • Ask natural language questions about your data
    • Write SQL queries directly
    • Explore your database schema

📊 Interface Tabs

💬 Natural Language Query

Ask questions in plain English and get SQL results:

  • "Show me the top 5 customers by revenue"
  • "What products have the highest sales in January?"
  • "How many orders were placed last month?"

📊 SQL Query

Write and execute SQL queries directly.

📋 Database Explorer

  • Browse available tables
  • View table schemas
  • See sample data from any table

🔒 Security Features

  • Only SELECT queries are permitted
  • Query validation to prevent dangerous operations
  • Secure credential management
  • Error handling and input validation

🛠️ Architecture

SQLGenius uses the Model Context Protocol (MCP) to expose tools that enable:

  1. Natural Language Processing: Convert English questions to SQL
  2. Data Exploration: Fetch schema information and sample data
  3. SQL Execution: Run validated queries against your database

The architecture consists of:

  • MCP Server: Handles DB connection and provides tools
  • Streamlit Frontend: User interface for interacting with the system
  • Vertex AI (Gemini Pro): Powers natural language understanding
  • BigQuery: Executes SQL queries on your data

📝 MCP Tools

The following MCP tools are available:

  1. execute_nl_query: Execute a natural language query
  2. execute_sql_query: Execute a raw SQL query
  3. list_tables: List all available tables
  4. get_table_schema: Get schema for a specific table

🤝 Contributing

Contributions are welcome! Please feel free to submit a Pull Request.