Sample For Bedrock Integration With Database MCP Server

🚀 アマゾンBedrock Converse APIとデータベースMCPサーバーの統合サンプル

このサンプルは、AWS Bedrockの対話型人工知能機能とリレーショナルデータベースをモデルコンテキストプロトコル（Model Context Protocol、略称MCP）を通じて統合する方法を示しています。具体的には、Bedrockを使用して自然言語でデータベースをクエリするサンプルです。

🚀 クイックスタート

アーキテクチャ図

コンポーネント説明

1. Amazon Bedrock Converse API：対話型インタラクションを処理する人工知能サービスです。
2. Database MCP Server：データベースの接続と管理を行うサーバーで、SQLiteとPostgreSQLをサポートしています。
3. SSL Certificate Path：通信を暗号化するためのセキュリティ証明書のパスです。

具体的な実装手順

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]

サンプルクエリ

全ユーザーの一覧表示

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);

これらのテーブルとデータはテスト目的でのみ使用され、実際の使用時には必要に応じて調整できます。