Vanna – 开源AI检索生成框架，自动生成精确的SQL查询-爱论文

Vanna是什么

Vanna是开源的Python RAG（Retrieval-Augmented Generation）框架，能帮助用户基于大型语言模型（LLMs）为其数据库生成精确的SQL查询。Vanna用两步简单流程操作：首先在用户数据上训练RAG模型，然后基于提问生成可在数据库上执行的SQL查询。Vanna支持多种LLMs、向量数据库和SQL数据库，提供高准确性查询，同时确保数据库内容安全私密，不外泄。Vanna支持自定义前端界面，支持用户用反馈自我学习，提高查询准确性。

Vanna的主要功能

SQL查询生成：基于大型语言模型（LLMs）根据用户的问题自动生成SQL查询。检索增强：基于检索增强技术（RAG）结合生成模型，提高查询的准确性。多数据库支持：支持连接多种SQL数据库，如PostgreSQL、MySQL等。多LLMs支持：兼容多个LLMs，包括OpenAI、Anthropic等。向量存储集成：与多种向量数据库集成，如AzureSearch、PgVector等。用户界面：提供多种用户界面选项，包括Jupyter Notebook、Streamlit、Flask和Slack。

Vanna的技术原理

检索增强生成（RAG）：基于RAG技术，一种结合检索（从大量数据中检索相关信息）和生成（基于检索到的信息生成输出）的方法。让Vanna更准确地理解和响应用户的问题。大型语言模型（LLMs）：Vanna依赖于LLMs理解和生成自然语言。模型经过预训练，能理解语言的复杂性和上下文。向量数据库：Vanna用向量数据库存储和检索与用户问题相关的信息。数据库能高效地处理和比较大量的数据点。训练与微调：用户基于提供DDL语句、文档和SQL查询训练Vanna，使其更好地理解和响应特定于业务的查询。自动化SQL执行：Vanna生成的SQL查询能自动在用户的数据库上执行，无需人工干预。

Vanna的项目地址

项目官网：vanna.ai/docsGitHub仓库：https://github.com/vanna-ai/vanna

Vanna的应用场景

数据分析师：数据分析师快速生成复杂的SQL查询，来分析大量数据，无需手动编写每个查询。业务智能（BI）工具：在BI工具中集成Vanna，支持用户用自然语言提问获取数据洞察，增强数据的可访问性和分析能力。客户支持系统：基于集成Vanna，客户支持系统自动从数据库中检索相关信息，快速响应客户的查询和问题。自动化报告：企业自动化生成日常或周期性的报告，减少手动数据处理和报告编写的工作量。数据科学项目：数据科学家快速原型化和测试新的数据分析想法，用自然语言交互加速数据探索过程。