database-mcp

Database MCP Server (by Legion AI)

A server that helps people access and query data in databases using the Legion Query Runner with integration of the Model Context Protocol (MCP) Python SDK.

Start Generation Here

This tool is provided by Legion AI. To use the full-fledged and fully powered AI data analytics tool, please visit the site.

End Generation Here

Features

Database access via Legion Query Runner
Model Context Protocol (MCP) support for AI assistants
Expose database operations as MCP resources, tools, and prompts
Multiple deployment options (standalone MCP server, FastAPI integration)
Query execution and result handling
Flexible configuration via environment variables, command-line arguments, or MCP settings JSON

Supported Databases

Database	DB_TYPE code
PostgreSQL	pg
Redshift	redshift
CockroachDB	cockroach
MySQL	mysql
RDS MySQL	rds_mysql
Microsoft SQL Server	mssql
Big Query	bigquery
Oracle DB	oracle
SQLite	sqlite

We use Legion Query Runner library as connectors. You can find more info on their api doc.

What is MCP?

The Model Context Protocol (MCP) is a specification for maintaining context in AI applications. This server uses the MCP Python SDK to:

Expose database operations as tools for AI assistants
Provide database schemas and metadata as resources
Generate useful prompts for database operations
Enable stateful interactions with databases

Installation & Configuration

Required Parameters

Two parameters are required for all installation methods:

DB_TYPE: The database type code (see table above)
DB_CONFIG: A JSON configuration string for database connection

The DB_CONFIG format varies by database type. See the API documentation for database-specific configuration details.

Installation Methods

Option 1: Using UV (Recommended)

When using uv, no specific installation is needed. We will use uvx to directly run database-mcp.

UV Configuration Example:

REPLACE DB_TYPE and DB_CONFIG with your connection info.
{
    "mcpServers": {
      "database-mcp": {
        "command": "uvx",
        "args": [
          "database-mcp"
        ],
        "env": {
          "DB_TYPE": "pg",
          "DB_CONFIG": "{\"host\":\"localhost\",\"port\":5432,\"user\":\"user\",\"password\":\"pw\",\"dbname\":\"dbname\"}"
        },
        "disabled": true,
        "autoApprove": []
      }
    }
}

Option 2: Using PIP

Install via pip:

pip install database-mcp

PIP Configuration Example:

{
  "mcpServers": {
    "database": {
      "command": "python",
      "args": [
        "-m", "database_mcp", 
        "--repository", "path/to/git/repo"
      ],
      "env": {
        "DB_TYPE": "pg",
        "DB_CONFIG": "{\"host\":\"localhost\",\"port\":5432,\"user\":\"user\",\"password\":\"pw\",\"dbname\":\"dbname\"}"
      }
    }
  }
}

Running the Server

Development Mode

mcp dev mcp_server.py

Production Mode

python mcp_server.py

Configuration Methods

Environment Variables

export DB_TYPE="pg"  # or mysql, postgresql, etc.
export DB_CONFIG='{"host":"localhost","port":5432,"user":"username","password":"password","dbname":"database_name"}'
mcp dev mcp_server.py

Command Line Arguments

python mcp_server.py --db-type pg --db-config '{"host":"localhost","port":5432,"user":"username","password":"password","dbname":"database_name"}'

Or with UV:

uv mcp_server.py --db-type pg --db-config '{"host":"localhost","port":5432,"user":"username","password":"password","dbname":"database_name"}'

Exposed MCP Capabilities

Resources

Resource	Description
`schema://all`	Get the complete database schema

Tools

Tool	Description
`execute_query`	Execute a SQL query and return results as a markdown table
`execute_query_json`	Execute a SQL query and return results as JSON
`get_table_columns`	Get column names for a specific table
`get_table_types`	Get column types for a specific table
`get_query_history`	Get the recent query history

Prompts

Prompt	Description
`sql_query`	Create an SQL query against the database
`explain_query`	Explain what a SQL query does
`optimize_query`	Optimize a SQL query for better performance

Development

Testing

uv pip install -e ".[dev]"
pytest

Publishing

rm -rf dist/ build/ *.egg-info/ && python -m build
python -m build
python -m twine upload dist/*

License

This repository is licensed under GPL