12.2Kstar!开源AI检索生成框架,自动生成精确的SQL查询!
Vanna是开源的Python RAG( Retrieval-Augmented Generation)框架,能帮助用户基于大型语言模型(LLMs)为其数据库生成精确的SQL查询。Vanna用两步简单流程操作:首先在用户数据上训练RAG模型,然后基于提问生成可在数据库上执行的SQL查询。Vanna支持多种LLMs、向量数据库和SQL数据库,提供高准确性查询,同时确保数据库内容安全私密,不外泄。Vanna支持自定义前端界面,支持用户用反馈自我学习,提高查询准确性。
主要功能:
SQL查询生成:基于大型语言模型(LLMs)根据用户的问题自动生成SQL查询。
检索增强:基于检索增强技术(RAG)结合生成模型,提高查询的准确性。
多数据库支持:支持连接多种SQL数据库,如PostgreSQL、MySQL等。
多LLMs支持:兼容多个LLMs,包括OpenAl、Anthropic等。
向量存储集成:与多种向量数据库集成,如AzureSearch、PgVector等
用户界面:提供多种用户界面选项,包括Jupyter Notebook、Streamlit、Flask和Slack。
GitHub地址:
https://github.com/vanna-ai/vanna
项目官网:https://vanna.ai/docs/?utm_source=ai-bot.cn