github地址:github.com/gaojunjie03…
gitee地址: gitee.com/gaojunjie03…
这是我刚入门 AI 大模型后的第一个开源项目。它或许还比较简单,但每一个 star 对我来说都非常重要。感谢你的支持!
数据库智能查询系统。支持数据库元数据管理、表结构映射、向量化入库、以及基于大语言模型的自然语言智能查询,能够自动生成 SQL 并可视化查询结果。
功能特点
- 数据库元数据获取:支持 MySQL、Oracle、PostgreSQL、达梦、Kingbase 数据库。
- 智能表结构映射:根据用户问题提取中文表描述,自动生成真实数据库表名与字段映射。
- 向量化知识库构建:将表描述向量化并存储至 Milvus,支持语义检索。
- 自然语言智能查询:用户用自然语言提问,系统自动生成 SQL 并返回查询结果。
- 查询结果可视化:支持 Markdown 表格渲染及统计图表生成。
- 多阶段异常处理:自动处理 SQL 执行异常、系统异常及函数调用异常,保证查询稳定性。
- 安全机制:自动拒绝可能破坏数据的操作(新增、修改、删除数据或表)。
技术特点
- 嵌入式向量模型:采用 [Sentence-Transformers] 的 Qwen3-Embedding-0.6B 模型(本地部署)进行表结构向量化。
- 语言生成模型:使用 GPT-OSS-20B模型(本地部署)处理自然语言理解与 SQL 生成。
- 消息流式解析:采用 OpenAI Harmony 格式及 StreamableParser 技术,实现高效流式生成和增量输出。
- 数据库与向量数据库支持:MySQL、Oracle、PostgreSQL、达梦、Kingbase;向量数据存储在 Milvus,支持快速语义检索。
- 前端交互:基于 Gradio Blocks + ChatInterface。
- 安全策略:系统层面过滤高风险数据库操作,确保查询安全性。