Sample For Bedrock Integration With Database MCP Server

🚀 亞馬遜Bedrock Converse API與數據庫MCP服務器集成示例

本示例展示瞭如何藉助模型上下文協議（Model Context Protocol，簡稱MCP），將AWS Bedrock的對話式人工智能功能與關係型數據庫進行集成。具體而言，這是一個利用Bedrock開展自然語言查詢數據庫的示例。

🚀 快速開始

具體實現步驟

1. 克隆代碼倉庫

使用以下命令克隆示例代碼：

git clone https://github.com/your-username/chat-with-database.git

2. 安裝依賴項

運行以下命令安裝所需的Python包：

pip install -r requirements.txt

3. 配置數據庫連接

修改config/database_config.py文件，填寫你的數據庫信息：

DATABASE_TYPE = 'sqlite'  # 或者 'postgres'
SQLITE_DB_NAME = 'mymcpdb.db'  # SQLite默認數據庫名稱
POSTGRES_CONFIG = {
    'host': 'your-rds-endpoint',
    'database': 'your-database-name',
    'user': 'your-db-user',
    'password': 'your-db-password',
    'ssl_cert_path': '/path/to/global-bundle.pem'
}

4. 啟動應用

運行以下命令啟動程序：

cd src
python3 chat.py --db-type [sqlite/postgres]

✨ 主要特性

架構圖

組件說明

1. Amazon Bedrock Converse API：用於處理對話式交互的人工智能服務。
2. Database MCP Server：連接和管理數據庫的服務器，支持SQLite和PostgreSQL。
3. SSL Certificate Path：安全證書路徑，用於加密通信。

💻 使用示例

示例查詢

• 列出所有用戶：

  查詢users表中的所有記錄。
  

• 按價格排序產品：

  將products表中的產品按價格從高到低排序。
  

📚 詳細文檔

成本估算

1. RDS PostgreSQL數據庫：根據實例類型和使用時長計費，具體費用可參考Amazon RDS定價。

2. Bedrock Claude模型調用API：費用根據調用次數和所選模型版本決定，詳細信息請查看Bedrock定價。

3. Guardrail服務：提供額外的安全防護功能，費用參考Guardrail定價.

補充說明

測試數據表結構

用戶表

CREATE TABLE users (
    id SERIAL PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    email VARCHAR(255) UNIQUE NOT NULL,
    created_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP
);

示例數據插入

INSERT INTO users (name, email) VALUES
    ('John Doe', 'john.doe@example.com'),
    ('Jane Smith', 'jane.smith@example.com'),
    ('Bob Wilson', 'bob.wilson@example.com'),
    ('Alice Johnson', 'alice.j@example.com'),
    ('Tom Brown', 'tom.brown@example.com');

產品表

CREATE TABLE products (
    product_id SERIAL PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    price DECIMAL(10,2) NOT NULL
);

INSERT INTO products (name, price) VALUES
    ('Amazon Echo Dot (5th Gen)', 49.99),
    ('Kindle Paperwhite 8GB', 139.99),
    ('Fire TV Stick 4K', 49.99),
    ('Ring Video Doorbell', 99.99),
    ('Amazon Smart Plug', 24.99);

這些表和數據僅用於測試目的，實際使用時可根據需求進行調整。