Text2SQL の例
このデモでは、自然言語の質問を SQL クエリに変換し、TiDB に対して実行する AI 搭載インターフェースの構築方法を紹介します。pytidb（TiDB の公式 Python SDK）、OpenAI GPT、Streamlit を使用して構築されており、平易な英語でデータベースをクエリできます。
前提条件
開始する前に、以下を確認してください。
- Python (>=3.10): Python 3.10 以降のバージョンをインストールします。
- TiDB Cloud Starter インスタンス: TiDB Cloud で無料の TiDB Cloud Starter インスタンスを作成できます。
- OpenAI APIキー: OpenAI から OpenAI APIキーを取得します。
実行方法
Step 1. pytidb リポジトリをクローンする
git clone https://github.com/pingcap/pytidb.git
cd pytidb/examples/text2sql/
Step 2. 必要なパッケージをインストールする
python -m venv .venv
source .venv/bin/activate
pip install -r reqs.txt
Step 3. Streamlit アプリを実行する
streamlit run app.py
Step 4. アプリを使用する
ブラウザを開き、http://localhost:8501 にアクセスします。
- 左側のサイドバーに OpenAI APIキーを入力します
- 左側のサイドバーに TiDB 接続文字列を入力します。例:
mysql+pymysql://root@localhost:4000/test
関連リソース
- ソースコード: GitHub で見る