LLMs之Vanna:Vanna(利用自然语言查询数据库的SQL工具+底层基于RAG)的简介、安装、使用方法之详细攻略

目录

Vanna的简介

1、用户界面

2、RAG vs. Fine-Tuning

3、为什么选择Vanna?

4、扩展Vanna

Vanna的安装和使用方法

1、安装

2、训练

(1)、使用DDL语句训练

(2)、使用文档训练

(3)、使用SQL训练

3、提问问题

Vanna的应用案例

1、基础用法

Vanna的简介

Vanna是一个基于MIT许可的开源Python RAG(检索增强生成)框架,用于SQL生成和相关功能。

Vanna的工作原理分为两个简单步骤:在您的数据上训练一个RAG“模型”,然后提问问题,这将返回可以设置自动在您的数据库上运行的SQL查询。 >> 在您的数据上训练一个RAG“模型”。 >> 提问问题。

如果您不知道什么是RAG,不用担心 - 您不需要知道在底层如何工作就能使用它。您只需要知道您需要“训练”一个模型,它会存储一些元数据,然后用它来“提问”问题。

查看基类以获取有关底层工作原理的更多详细信息。

Vanna.AI是一个基于语言模型的SQL代理,允许用户通过简单的自然语言问题获得数据库中的洞察信息。它提供开源的Python包以及各种前端集成方式,可以部署在自己的基础设施上运行。 >> 系统的准确性取决于提供的训练数据量和质量,更多的数据可以支持复杂数据集提高准确率。 >> 用户数据库内容不会发送到语言模型,只有架构、文档和查询信息存储在元数据层中,从保障数据安全性。 >> 使用越多,模型通过不断增加训练数据会持续提升,实现自我学习效果。 >> 支持许多常见数据库如Snowflake、BigQuery、Postgres,也可以通过连接器支持任意数据库。 >> 提供免费版和付费版两种计划,区别在于查询限制和语言模型版本的不同。 >> 该系统强调以开源方式提供,保障数据安全并支持定制化部署是其一大卖点。

GitHub地址:https://github.com/vanna-ai/vanna

文档地址:Vanna Docs: How It Works

1、用户界面

这是我们使用Vanna构建的一些用户界面。您可以直接使用它们,或将它们作为自定义界面的起点。

2、RAG vs. Fine-Tuning

RAG 微调 可在LLMs之间移植 如果数据变得过时,可以轻松删除训练数据 比微调更便宜 更具未来性 - 如果有更好的LLM出现,可以轻松替换 如果需要最小化提示中的标记,可以选择微调 启动较慢 训练和运行费用昂贵(一般情况下)

3、为什么选择Vanna?

在复杂数据集上具有高准确性 Vanna的能力与您提供的训练数据密切相关。 更多的训练数据对于大型和复杂数据集的准确性更有帮助。 安全和私密 您的数据库内容永远不会发送到LLM或向量数据库。 SQL执行发生在您的本地环境中。 自学习 如果通过Jupyter使用,您可以选择在成功执行的查询上“自动训练”它。 如果通过其他界面使用,您可以要求界面提示用户对结果提供反馈。 正确的问题和SQL对存储供将来参考,使未来的结果更准确。 支持任何SQL数据库 该软件包允许您连接到您可以使用Python连接的任何SQL数据库。 选择您的前端 大多数人从Jupyter Notebook开始。 通过Slackbot、Web应用程序、Streamlit应用程序或自定义前端向最终用户公开。

4、扩展Vanna

Vanna旨在连接到任何数据库、LLM和向量数据库。有一个VannaBase抽象基类定义了一些基本功能。该软件包提供了与OpenAI和ChromaDB一起使用的实现。您可以轻松扩展Vanna以使用自己的LLM或向量数据库。详细信息请参阅文档。

Vanna的安装和使用方法

查看文档以获取有关您所需数据库、LLM等的具体信息。

如果您想在训练后了解其工作方式,可以尝试此Colab笔记本。

1、安装

pip install vanna

有一些可选包可以安装,详细信息请参阅文档。

导入

如果您要自定义LLM或向量数据库,请参阅文档。

import vanna as vn

2、训练

根据您的用例,您可能需要或不需要运行这些vn.train命令。详细信息请参阅文档。

(1)、使用DDL语句训练

DDL语句包含有关数据库中表名、列、数据类型和关系的信息。

vn.train(ddl="""

CREATE TABLE IF NOT EXISTS my-table (

id INT PRIMARY KEY,

name VARCHAR(100),

age INT

)

""")

(2)、使用文档训练

有时您可能希望添加关于业务术语或定义的文档。

vn.train(documentation="Our business defines XYZ as ...")

(3)、使用SQL训练

您还可以将SQL查询添加到训练数据中。如果您已经有一些查询可用,只需从编辑器中复制并粘贴它们即可开始生成新的SQL。

vn.train(sql="SELECT name, age FROM my-table WHERE name = 'John Doe'")

3、提问问题

vn.ask("What are the top 10 customers by sales?")

您将得到SQL查询结果,以及连接到数据库时的表格和自动生成的Plotly图表。

Vanna的应用案例

更新中……

1、基础用法

!pip install vanna

import vanna

from vanna.remote import VannaDefault

vn = VannaDefault(model='chinook', api_key=vanna.get_api_key('my-email@example.com'))

vn.connect_to_sqlite('https://vanna.ai/Chinook.sqlite')

vn.ask('What are the top 10 artists by sales?')

from vanna.flask import VannaFlaskApp

VannaFlaskApp(vn).run()

参考阅读

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: