【Sqlboys的福音】录入你的表结构，让Vanna.AI为你编写业务SQL

【Sqlboys的福音】录入你的表结构，让Vanna.AI为你编写业务SQL

sqlai.py

import pandas as pd
import mysql.connector
from vanna.remote import VannaDefault
vn = VannaDefault(model='yourModel', api_key='xxxx')
conn_details = {
  'host': '127.0.0.1',
  'user': 'admin',
  'password': '123456', 
  'database': 'test'
}

def run_sql(sql: str) -> pd.DataFrame:
  conn = mysql.connector.connect(**conn_details)
  df = pd.read_sql(sql, conn)
  conn.close()
  return df

vn.run_sql = run_sql
vn.run_sql_is_set = True

from vanna.flask import VannaFlaskApp
app = VannaFlaskApp(vn)
app.run()

开源https://vanna.ai/人工智能，注册一个账号，输入你自己的模型，和api_key，复制粘贴上面的代码运行。

然后打开浏览器http://yourIP:8084/，如下图所示：

第一步，点击Training Data手工录入你的表结构

第二步，用中文描述业务，AI会根据你的表结构自动输出SQL，并执行SQL。