Text-to-SQL 工具Vanna + MySQL本地部署 | 数据库对话机器人

import os
from dotenv import load_dotenv
from vann.vann_api import Vanna
 
# 加载环境变量
load_dotenv()
 
# 获取环境变量
MYSQL_USER = os.getenv('MYSQL_USER')
MYSQL_PASSWORD = os.getenv('MYSQL_PASSWORD')
MYSQL_HOST = os.getenv('MYSQL_HOST')
MYSQL_DB = os.getenv('MYSQL_DB')
 
# 初始化Vanna并连接到MySQL数据库
vanna = Vanna(
    user=MYSQL_USER,
    password=MYSQL_PASSWORD,
    host=MYSQL_HOST,
    database=MYSQL_DB
)
 
# 示例SQL查询
query = "SELECT * FROM your_table WHERE condition = 'value';"
 
# 执行SQL查询
result = vanna.execute_query(query)
 
# 打印查询结果
print(result)

这段代码展示了如何使用Vanna库连接到MySQL数据库并执行一个简单的SQL查询。首先，它会从环境变量中加载数据库的连接信息。然后，它会初始化Vanna实例并使用这些连接信息来连接到数据库。最后，它会执行一个SQL查询并打印出结果。这个例子简单明了地展示了如何将Vanna与MySQL结合使用，并且是构建对话机器人或其他需要与数据库交互的应用程序的基础。