Database MCP Server

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