可从公众号“AI早高峰”看全文
https://mp.weixin..com/s/Vs0iqZVFphc-QCsr3X1ywA
跟数据库对话机器人对话，我可不止一个问题。
可能基于第一句问话，还有第二句、第三句问话。。。第N句对话。所以本文测试了多轮对话功能。
单轮对话的环境搭建参考博客 Text-to-SQL 工具Vanna + MySQL本地部署 | 数据库对话机器人
我的数据是这样
1. 基础配置
import vanna from vanna.remote import VannaDefault from vanna.flask import VannaFlaskApp import pandas as pd import mysql.connector api_key = '7acxxx68c' vanna_model_name = 'community' vn = VannaDefault(model=vanna_model_name, api_key=api_key) def run_sql(sql: str) -> pd.DataFrame: cnx = mysql.connector.connect(user='root',password='',host='localhost',database='Community') cursor = cnx.cursor() cursor.execute(sql) result = cursor.fetchall() columns = cursor.column_names # print('columns:',columns) df = pd.DataFrame(result, columns=columns) return df # 将函数设置到vn.run_sql中 vn.run_sql = run_sql vn.run_sql_is_set = True 讯享网
2. 多轮生成SQL
- 用
vn.generate生成SQL
使用方法看官方文档
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容，请联系我们，一经查实，本站将立刻删除。
如需转载请保留出处：https://51itzy.com/kjqy/69771.html