AI SQL Assistant

A natural language interface for interacting with SQLite databases, powered by Anthropic’s Claude AI.

Overview

This application allows users to interact with SQLite databases using natural language. Instead of writing SQL queries manually, users can ask questions or give instructions in plain English, and the AI will generate and execute the appropriate SQL queries, returning formatted results.

Features

• Natural Language Queries: Ask questions about your data in plain English
• SQL Query Generation: Automatically generates SQL queries based on user input
• Query Execution: Executes the generated queries against a connected SQLite database
• Result Formatting: Displays query results in a readable format
• Error Handling: Provides clear explanations when queries fail
• Example Queries: Includes sample queries to help users get started

Architecture

The application consists of three main components:

1. Web Interface (app.py): A Gradio web interface that allows users to interact with the AI
2. MCP Server (mcp_server.py): A server that handles SQL query execution
3. MCP Client (mcp_client.py): A client that facilitates communication between the web interface and the MCP server

Prerequisites

• Python 3.8+
• An Anthropic API key (to access Claude)

Installation

1. Clone this repository:

   git clone https://github.com/yourusername/ai-sql-assistant.git
   cd ai-sql-assistant
   

2. Install dependencies:

   pip install -r requirements.txt
   

3. Create a .env file in the project root with your Anthropic API key:

   ANTHROPIC_API_KEY=your_api_key_here
   

Usage

1. Start the application:

   python app.py
   

2. Open your web browser and navigate to:

   http://localhost:7860
   

3. Type in a natural language query and press Enter or click Submit.

Sample Queries

Here are some example queries you can try:

• “List all tables in the database”
• “Show me all users”
• “Count how many products cost more than $300”
• “What’s the total value of all products in stock?”
• “Insert a new product called ‘Tablet’ with price $499.99 and stock 5”
• “Update the price of Laptop to $1099.99”

Sample Database

The application creates a sample SQLite database with the following tables:

Users Table

• id: INTEGER (Primary Key)
• name: TEXT
• email: TEXT
• created_at: TIMESTAMP

Products Table

• id: INTEGER (Primary Key)
• name: TEXT
• price: REAL
• stock: INTEGER

Technical Details

MCP (Model Completion Protocol)

The application uses MCP to manage communication between Claude and the SQLite database. MCP allows Claude to call functions (tools) and receive results, creating an interactive feedback loop.

Anthropic Claude

The application uses Anthropic’s Claude 3.7 Sonnet model to:

1. Understand natural language queries
2. Generate appropriate SQL queries
3. Explain query results to users

SQLite

The application uses SQLite for database operations. SQLite is:

• Lightweight
• Serverless
• Self-contained
• Zero-configuration

Customization

Connecting to Your Own Database

To use your own SQLite database instead of the sample one:

1. Replace the database initialization in app.py with your own database file path
2. Update the sample table documentation in the UI to match your schema

Modifying the System Prompt

The system prompt that guides Claude’s behavior can be modified in the ChatProcessor class in app.py.

Limitations

• Currently only supports SQLite databases
• Complex queries might require refinement
• Limited error recovery for malformed queries

Contributing

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

License

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