Text2SQLの例

このデモでは、自然言語の質問をSQLクエリに変換し、TiDBに対して実行するAI搭載インターフェースの構築方法を紹介しますpytidb （TiDBの公式Python SDK）、OpenAI GPT、Streamlitを使用して構築されており、平易な英語でデータベースにクエリを実行できます。

前提条件

始める前に、次のものがあることを確認してください。

• Python (>=3.10) : パイソン 3.10 以降のバージョンをインストールします。
• TiDB Cloud Starter クラスター: TiDB Cloudに無料の TiDB クラスターを作成できます。
• OpenAI API キー: オープンAIから OpenAI API キーを取得します。

実行方法

ステップ1. pytidbリポジトリのクローンを作成する

git clone https://github.com/pingcap/pytidb.git
cd pytidb/examples/text2sql/

ステップ2. 必要なパッケージをインストールする

python -m venv .venv
source .venv/bin/activate
pip install -r reqs.txt

ステップ3. Streamlitアプリを実行する

streamlit run app.py

ステップ4. アプリを使用する

ブラウザを開いてhttp://localhost:8501アクセスします。

1. 左のサイドバーにOpenAI APIキーを入力してください
2. 左側のサイドバーにTiDB接続文字列を入力します。例: mysql+pymysql://root@localhost:4000/test

関連リソース

• ソースコード: GitHubでビュー