Text2SQLの例

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

前提条件

始める前に、以下のものを用意してください。

Python (>=3.10) : Python 3.10以降のバージョンをインストールしてください。
TiDB Cloud Starterインスタンス: TiDB Cloudで無料のTiDB Cloud Starterインスタンスを作成できます。
OpenAI API キー: OpenAIから 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にアクセスしてください。

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

ソースコード： GitHubでビュー