Legion Database

🚀 Database MCP Server (Powered by Legion AI)

A server that helps users access and query database data and integrates with the Model Context Protocol (MCP) Python SDK using the Legion Query Runner.

🚀 Quick Start

The Database MCP Server enables users to access and query database data. It integrates with the Legion Query Runner and the Model Context Protocol (MCP) Python SDK, providing a seamless experience for database operations.

✨ Features

• Access databases using the Legion Query Runner.
• Support for the Model Context Protocol (MCP).
• Expose database operations as MCP resources, tools, and prompts.
• Multiple deployment options (standalone MCP server, FastAPI integration).
• Query execution and result processing.
• Flexible configuration methods: environment variables, command-line arguments, or MCP settings JSON.

📦 Supported Databases

We use the Legion Query Runner library as the connector. For more detailed information, please refer to its API documentation.

📚 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 and Configuration

Required Parameters

All installation methods require two parameters:

• DB_TYPE: The database type code (see the table above).
• DB_CONFIG: The database configuration information.

Installation Methods

Install with UV

uv pip install package_name

Install with PIP

pip install package_name

💻 Running the Server

Development Mode

uv config.yaml

Production Mode

python -m module_name

💻 Configuration Methods

Configuration via Environment Variables

Set the following environment variables:

export DB_TYPE="pg"
export DB_CONFIG="your_config_here"

Configuration via Command-Line Arguments

Specify the parameters when running:

--db_type pg --db_config "your_config_here"

💻 MCP Features

Resources

• Execute a query: execute_query
• Return JSON results: execute_query_json
• Get table column names: get_table_columns
• Get column types: get_table_types
• Get query history: get_query_history

Tools

• Generate an SQL query: sql_query
• Explain an SQL query: explain_query
• Optimize an SQL query: optimize_query

💻 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 uses the GPL license.