omniquery-ai

OmniQuery AI 是企业级的自然语言转 SQL 助手，专为复杂的关系型数据库架构设计，尤其是 Microsoft SQL Server / AdventureWorks。它由混合向量搜索（Qdrant + FastEmbed）、Groq 上的 Llama 3.3 70B、动态主键/外键图上下文注入，以及自主的自我修复错误纠正循环驱动。该项目针对的是 AdventureWorks 这类拥有 90 多张表和数百列的大型企业数据库，把整个数据库 schema 一股脑塞进 LLM 上下文窗口既低效又昂贵，还会导致严重的幻觉和无效的 JOIN。

OmniQuery AI 使用四阶段混合 RAG 流水线。它把预先计算的一句话业务摘要、列定义以及主键/外键索引到 Qdrant 中。它使用 sentence-transformers/all-MiniLM-L6-v2 来表示语义，使用 Qdrant/bm25 来精确匹配列名/表名。它动态拉取有效的外键关系，使 LLM 能够构造多表 JOIN 语句。如果 SQL Server 返回运行时或语法错误，错误信息会反馈给 LLM，自动修复并重新执行查询，最多重试 3 次。

OmniQuery AI 包含企业级 DDL/DML 护栏，通过在执行前校验 AST 令牌和正则模式，防范 SQL 注入和破坏性语句。它还提供零配置的离线模拟演示模式，让招聘人员、评审者以及 macOS/Linux 用户无需在本地运行真实的 Microsoft SQL Server 实例，就能立即试用该助手。该仓库包含交互式 CLI、非交互式查询运行器、配置、测试和 MIT 许可证。