Database MCP Server (由 Legion AI 提供)

一个服务器，旨在帮助人们使用 Legion Query Runner 访问和查询数据库中的数据，并集成了模型上下文协议 (MCP) Python SDK。

Start Generation Here

此工具由 Legion AI 提供。要使用功能齐全且功能强大的 AI 数据分析工具，请访问该网站。

End Generation Here

功能

• 通过 Legion Query Runner 访问数据库
• 支持 AI 助手的模型上下文协议 (MCP)
• 将数据库操作公开为 MCP 资源、工具和提示
• 多种部署选项（独立 MCP 服务器、FastAPI 集成）
• 查询执行和结果处理
• 通过环境变量、命令行参数或 MCP 设置 JSON 进行灵活配置

支持的数据库

数据库	DB_TYPE 代码
PostgreSQL	pg
Redshift	redshift
CockroachDB	cockroach
MySQL	mysql
RDS MySQL	rds_mysql
Microsoft SQL Server	mssql
Big Query	bigquery
Oracle DB	oracle
SQLite	sqlite

我们使用 Legion Query Runner 库作为连接器。您可以在他们的 api doc 上找到更多信息。

什么是 MCP？

模型上下文协议 (MCP) 是一种用于在 AI 应用程序中维护上下文的规范。此服务器使用 MCP Python SDK 来：

• 将数据库操作公开为 AI 助手的工具
• 提供数据库模式和元数据作为资源
• 为数据库操作生成有用的提示
• 实现与数据库的有状态交互

安装与配置

必需参数

所有安装方法都需要两个参数：

• DB_TYPE: 数据库类型代码（参见上表）
• DB_CONFIG: 数据库连接的 JSON 配置字符串

DB_CONFIG 格式因数据库类型而异。有关特定于数据库的配置详细信息，请参阅 API 文档。

安装方法

选项 1：使用 UV（推荐）

当使用 uv 时，不需要特定的安装。我们将使用 uvx 直接运行 database-mcp。

UV 配置示例：

替换 DB_TYPE 和 DB_CONFIG 为您的连接信息。
{
    "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": []
      }
    }
}

选项 2：使用 PIP

通过 pip 安装：

pip install database-mcp

PIP 配置示例：

{
  "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\"}"
      }
    }
  }
}

运行服务器

开发模式

mcp dev mcp_server.py

生产模式

python mcp_server.py

配置方法

环境变量

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

命令行参数

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

或者使用 UV：

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

公开的 MCP 功能

资源

资源	描述
schema://all	获取完整的数据库模式

工具

工具	描述
execute_query	执行 SQL 查询并将结果作为 markdown 表格返回
execute_query_json	执行 SQL 查询并将结果作为 JSON 返回
get_table_columns	获取特定表的列名
get_table_types	获取特定表的列类型
get_query_history	获取最近的查询历史

提示

提示	描述
sql_query	创建针对数据库的 SQL 查询
explain_query	解释 SQL 查询的作用
optimize_query	优化 SQL 查询以获得更好的性能

开发

测试

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

发布

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

许可证

此存储库在 GPL 许可下