一、项目环境搭建

1、部署 openEuler 服务器,关闭 SELinux、配置可用 yum 源,编译安装 Python3.11.9

(1)部署 openEuler 服务器

(2)关闭 SELinux

(3)配置可用 yum

(4)编译安装 Python3.11.9

2、安装 MySQL8.0.45,创建 testdb 库、order_info 订单业务表,导入20 条标准测试订单数据;

(1)测试MySQL

(2)创建 testdb 库、order_info 订单业务表,导入20 条标准测试订单数据

测试

3、登录并开通腾讯云 TokenHub 大模型聚合平台,申请 API 密钥,领取免费额度,编写.env 配置文件并加固文件权限;

(1)获取API

(2)新建目录及脚本

4、安装项目全部 Python 依赖包,测试数据库连通性、大模型接口连通性。

二、脚本开发

1、mysql_client.py:封装 MySQL 连接、查询与执行计划获取,拦截危险 SQL 保障数据库安全。

2、prompts.py:统一存放生成 SQL、调优分析两套 AI 提示词,并提供提取纯净 SQL 的工具函数。

3、main.py:项目总调度,串联数据库、大模型、提示词,提供终端菜单交互,实现完整业务逻辑。

4、web_main.pyStreamlit 可视化网页界面,做输入校验、图形化展示结果,复用 main 核心逻辑。

5、.env:存放数据库、腾讯云 TokenHub 大模型密钥等配置,分离敏感信息与业务代码。

三、功能测试

1、多业务自然语言用例全覆盖测试,校验生成 SQL 安全、逻辑准确,模糊匹配无 bug

2、性能调优模块测试,校验 EXPLAIN 解析完整、输出索引与改写方案可直接执行;

3、边界场景测试:无匹配用户 / 商品、空白输入、超长复合需求,程序无崩溃报错;

4、安全测试:验证不会生成任何 DELETE/UPDATE/ALTER 等危险 SQL,密钥无明文打印。

测试:

web可视化:

Logo

openEuler 是由开放原子开源基金会孵化的全场景开源操作系统项目,面向数字基础设施四大核心场景(服务器、云计算、边缘计算、嵌入式),全面支持 ARM、x86、RISC-V、loongArch、PowerPC、SW-64 等多样性计算架构

更多推荐