Amazon Bedrock Converse API and Database MCP Server Integration
2025.04.04
2
Python数据库集成生成式AI自然语言处理AWS服务集成数据库开发效率
该示例展示了如何将AWS Bedrock的对话式AI能力与关系型数据库通过Model Context Protocol (MCP) Server架构进行集成。它利用Bedrock的Converse API实现与数据库的自然语言交互（仅查询模式），而MCP Server则处理数据库操作和安全协议的复杂性。此架构为开发者提供了一个参考实现，用于将生成式AI能力整合到数据库应用中。
View on GitHub
Overview
基本能力
产品定位
- 集成AWS Bedrock的生成式AI能力与关系型数据库
- 通过自然语言查询数据库（只读模式）
- 为现有应用添加生成式AI功能
核心功能
- 集成RDS PostgreSQL和SQLite与AWS Bedrock的基础模型
- 通过Bedrock Converse API与MCP Server集成
- 使用Bedrock基础模型进行自然语言数据库查询
- 通过MCP Server实现安全高效的数据库操作
- 提供跨数据库的生成式AI解决方案基础
适用场景
- 需要自然语言查询数据库的应用
- 希望为现有数据库应用添加AI能力的开发者
- 需要安全高效数据库操作的场景
工具列表
- AWS Bedrock Converse API：管理上下文、记忆和状态，实现实时对话体验
- AWS Bedrock模型：如Anthropic Claude Sonnet 3.5等LLM
- AWS Bedrock Guardrails：实现符合应用需求和负责任AI政策的保障措施
- RDS PostgreSQL：用于数据库查询
常见问题解答
- 成本问题：使用RDS PostgreSQL数据库、Bedrock模型调用API和Guardrails会产生费用
- 权限问题：数据库用户需要配置适当的只读权限
- 区域问题：需要在使用Bedrock模型时注意区域可用性
使用教程
使用依赖
- 安装Node.js（v10.13.0或更高版本）
- 安装AWS CDK
npm install -g aws-cdk
- 配置AWS CLI
aws configure
- 创建并激活虚拟环境
python3 -m venv .venv
source .venv/bin/activate
安装教程
- 克隆仓库
git clone https://github.com/aws-samples/sample-for-bedrock-integration-with-database-mcp-server
- 安装依赖
pip install -r requirements.txt
- 安装PostgreSQL MCP服务器
npm install -g @modelcontextprotocol/server-postgres
pip install mcp psycopg2-binary
调试方式
- 修改dbconfig.ini文件配置数据库连接信息
- 运行应用程序
cd src
python3 chat.py --db-type [sqlite / postgres]
- 使用自然语言查询数据库，例如：
- "List all products that are available in products table"
- "List of products which cost above 50.00"