拓冰建站拓冰建站
首页 / 资讯中心 / 正文

不写 SQL 也能查库:Vanna 2.0 自然语言转 SQL 上手全记录

不写 SQL 也能查库Vanna 2.0 自然语言转 SQL 上手全记录【免费下载链接】vanna Chat with your SQL database . Accurate Text-to-SQL Generation via LLMs using Agentic Retrieval .项目地址: https://gitcode.com/GitHub_Trending/va/vanna运营在群里问上个月销售额 Top 10 的客户是谁你要么手写 SQL 截图丢回去要么打开 BI 面板翻半天。Vanna 2.0 想替代的就是这个环节把自然语言问题交给大模型翻译成 SQL执行后把表格、图表流式吐回给提问的人。一句话定位它是库不是服务Vanna 是一个 Python 框架核心链路是自然语言 → SQL → 数据答案靠 RAG 检索把表结构、文档、历史 SQL 塞进提示词来提高生成准确率。它适合两类人想给产品加自然语言查数入口的开发者需要按用户身份过滤数据的多租户或企业团队。注意它不是开箱即用的 SaaS——pip install vanna之后它是一个嵌进你自己 FastAPI/Flask 应用的库。15 行代码跑通最小示例装包后把 SQLite 查询工具注册进 Agent 即可LLM 可换成 OpenAI、Anthropic、Ollama 等十几种实现# pip install vanna from vanna import Agent, User from vanna.core.registry import ToolRegistry from vanna.tools import RunSqlTool from vanna.integrations.sqlite import SqliteRunner from vanna.integrations.anthropic import AnthropicLlmService tools ToolRegistry() tools.register(RunSqlTool(sql_runnerSqliteRunner(Chinook.sqlite))) agent Agent(llm_serviceAnthropicLlmService(), tool_registrytools) user User(idu1, usernamealice, permissions[]) async for comp in agent.send_message( useruser, message列出数据库里所有的表, conversation_idc1): print(comp)跑完你会在终端依次看到思考进度、生成的 SQL 代码块、查询结果表最后是一段自然语言总结。整个过程里 LLM 并没有背下你的库而是一次次调用 run_sql 工具。为什么生成的 SQL 比裸大模型靠谱它解决大模型不知道你的表长什么样的问题做法是训练 检索两段式先把建表 DDL、字段文档、参考 SQL 语句生成嵌入存进向量库提问时按语义搜出最相关的表和示例拼进提示词再让模型写 SQL。表越多、业务黑话越重这一步收益越大——只把 schema 丢给模型裸跑和喂了上下文示例再跑准确率差距在仓库的论文报告里有实测对比可看 papers/ai-sql-accuracy-2023-08-17.md。别问向量检索细节问就是黑盒你只需要把 DDL 和常跑的 SQL 喂给它训练。让不同角色看到不同数据这是 Vanna 2.0 相对 1.x 最实际的改动用户身份贯穿系统提示词、工具执行、SQL 结果三层。你只需要实现一个 UserResolver从 JWT 或 cookie 里解析出用户和所属组class MyUserResolver(UserResolver): async def resolve_user(self, request_context): data self.decode_jwt(request_context.get_header(Authorization)) return User(iddata[id], emaildata[email], group_membershipsdata[groups])group_memberships是权限的钥匙工具按组做访问检查没权限的工具对该用户直接不可见SQL 工具按用户权限做行级过滤管理员才看得到 SQL 代码块每次查询还会留下审计记录。按用户限流、配额控制则挂在生命周期钩子上钩子基类在 src/vanna/core/lifecycle/base.py。走查一遍 接进现有网站后端用 FastAPI 时把 Vanna 的聊天路由注册到你已有的 app 上就得到一个/api/vanna/v2/chat_sse流式端点。前端一行搞定支持 React、Vue 或原生 HTMLvanna-chat sse-endpoint/api/vanna/v2/chat_sse/vanna-chat之后用户体验是问一句按销售额给我 Top 10 客户页面先流式显示处理进度然后出现可交互的数据表最后跟一张 Plotly 柱状图。仓库里这张图就是可视化工具对 Chinook 示例库跑出来的产出边界什么场景合适什么场景别用它适合内部数据分析入口、多租户 SaaS 的按身份过滤数据、以及已有认证体系想快速挂一个聊天查数界面的团队。不太适合的超复杂多表关联报表准确率取决于你训练的示例质量业务语义越重越要喂足够的参考 SQL高并发在线服务每次提问都现场生成并执行 SQL不是预编译查询以及只想要个聊天框、不想维护 Python 服务的场景——前端组件再省后端 Agent 服务还是得你自己跑。具体支持哪些数据库和 LLM看 src/vanna/integrations/ 目录下的集成清单。如果你从 0.x 升级先读 MIGRATION_GUIDE.md 里的 LegacyVannaAdapter 过渡方案。【免费下载链接】vanna Chat with your SQL database . Accurate Text-to-SQL Generation via LLMs using Agentic Retrieval .项目地址: https://gitcode.com/GitHub_Trending/va/vanna创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
分享:

看完干货,该让你的企业上线了

免费需求沟通 · 48 小时内出具建站方案 · 河南本地可上门