Vanna是一个基于RAG(检索增强生成)的开源Python框架,专门用于SQL生成和相关功能。它允许用户通过自然语言与SQL数据库进行交互,将文本查询准确转换为SQL语句。

安装Vanna:
pip install vanna
导入并初始化:
from vanna.openai.openai_chat import OpenAI_Chat from vanna.chromadb.chromadb_vector import ChromaDB_VectorStore class MyVanna(ChromaDB_VectorStore, OpenAI_Chat): def __init__(self, config=None): ChromaDB_VectorStore.__init__(self, config=config) OpenAI_Chat.__init__(self, config=config) vn = MyVanna(config={'api_key': 'sk-...', 'model': 'gpt-4-'})
训练模型:
vn.train(ddl="CREATE TABLE ...") vn.train(documentation="业务定义...") vn.train(sql="SELECT ...")
提问并获取SQL:
vn.ask("销售额前10的客户是谁?")

Vanna提供了多种用户界面选择:
通过以上资源,读者可以全面了解Vanna,并开始将其应用于自己的项目中。无论是数据分析师、开发人员还是数据科学家,Vanna都能帮助您更高效地与SQL数据库交互,释放数据的潜力。🚀📊