跳到主要内容
版本:V1.0.0

seekdb MCP Server 与 Augment Code 集成

MCP(Model Context Protocol) 是 Anthropic 公司于 2024 年 11 月推出并开源,旨在实现大语言模型与外部工具或数据源交互的协议。通过 MCP,用户不需要将大模型的输出手动复制执行,大模型可以直接指挥工具执行相应的动作(Action)。

MCP Server 通过 MCP 协议提供了大模型与 seekdb 交互的能力,可以执行 SQL 语句。通过合适的客户端可以快速搭建项目原型,已在 github 上开源。

Augment Code 是开发人员 AI 平台,它可以帮助您理解代码、调试问题,并更快地发布,因为它了解您的代码库。使用聊天、下一步编辑和代码完成来完成更多的工作。

本文将展示如何使用 Augment Code,展示如何通过 seekdb MCP Server 快速构建一个后端应用。

前提条件

  • 您已完成部署 seekdb。

  • 安装 Python 3.11 及以上版本 和相应 pip。如果您的机器上 Python 版本较低,可以使用 Miniconda 来创建新的 Python 3.11 及以上的环境,具体可参考 Miniconda 安装指南

  • 安装 Python 包管理器 uv。安装完成后,可使用 uv --version 命令验证安装是否成功:

    pip install uv
    uv --version
  • 使用 Visual Studio Code 安装 Augment Code 插件,插件名称为 Augment Code

    1

步骤一:获取数据库连接信息

联系 seekdb 部署人员或者管理员获取相应的数据库连接串,例如:

mysql -h$host -P$port -u$user_name -p$password -D$database_name

参数说明:

  • $host:提供 seekdb 连接 IP 地址。

  • $port:提供 seekdb 连接端口,默认是 2881。

  • $database_name:需要访问的数据库名称。

    提示

    连接的用户需要拥有该数据库的 CREATEINSERTDROPSELECT 权限。

  • $user_name:提供数据库连接账户。

  • $password:提供账户密码。

步骤二:配置 seekdb MCP Server

这里以 Visual Studio Code 为例,展示如何配置 seekdb MCP Server。

安装 seekdb MCP Server

执行下面的命令进行安装:

pip install seekdb-mcp-server

配置 seekdb 服务器环境变量到 .env 文件中

在当前目录下创建一个名为 .env 的文件,并添加以下内容:

vi .env

SEEKDB_HOST=localhost # 数据库主机
SEEKDB_PORT=2881 # 数据库端口(默认:2881)
SEEKDB_USER=your_username
SEEKDB_PASSWORD=your_password
SEEKDB_DATABASE=your_database

以 SSE 模式启动

env $(cat .env | xargs) uvx seekdb-mcp-server --transport sse --port 8000 --host 0.0.0.0

添加并配置 MCP Servers

  1. 登录 Augment Code。

    2

  2. 在 Augment Code 中配置 seekdb MCP Server。

    3

  3. 点击 Import from json,填写 mcp 配置文件。

    4

    5

  4. 填写配置文件,点击确认。

    {
    "mcpServers": {
    "seekdb": {
    "autoApprove": [],
    "disabled": false,
    "timeout": 60,
    "type": "sse",
    "url": "http://ip:port/sse"
    }
    }
    }

    Augment Code 展示当前 seekdb mcp server 中的工具即配置成功。