fix(ai): Ask 跟进换表禁止默认 created_at
CI / Go server (push) Canceled after 0s
CI / Desktop UI (push) Canceled after 0s
CI / Desktop (Tauri) cargo check (push) Canceled after 0s

多轮/换表后列未注入时只澄清或 DESCRIBE,不得编时间列写业务 SQL。
This commit is contained in:
logan committed 2026-10-26 23:07:17 +08:00
1 parent 4c68539db6
commit 32c28a123a
2 files changed
+6 -3

No files matched your search

+3
View File
@@ -124,6 +124,9 @@ func TestAskChatsWithoutClaimingDatabaseExecution(t *testing.T) {
if !strings.Contains(generate[0].Content, "禁止编造") || !strings.Contains(generate[0].Content, "UPDATE_TIME") {
t.Fatalf("system rules missing invent/UPDATE_TIME guards: %s", generate[0].Content)
}
if !strings.Contains(generate[0].Content, "换表") || !strings.Contains(generate[0].Content, "created_at") {
t.Fatalf("follow-up no-invent column rules missing: %s", generate[0].Content)
}
offline := chatStub(request)
if offline.SQL != "" || !strings.Contains(offline.Content, "离线体验无法回答") {
t.Fatalf("offline Ask must not pretend to generate or query: %+v", offline)
+3 -3
View File
@@ -14,13 +14,13 @@ func BuildMessages(req ChatReq) []Message {
if driver == "" {
driver = "mysql"
}
sys := "你是资深数据库工程师助手,服务于数据库客户端 OpsxDB。回答使用中文。SQL 放在 ```sql 代码块中。\n规则:\n- 禁止输出思考过程、推理草稿、内心独白或逐步分析;只给用户最终答案。\n- 禁止编造上下文中未出现的列名或表结构。\n- 缺表/缺时间列/意图不清时:用一两句中文澄清(可问具体表与时间字段),不要为了「可执行」硬造 SQL。\n- 「最近 N 天数据变化」默认指业务行时间过滤;不得默认用 information_schema.TABLES.UPDATE_TIME 充当终答(除非用户明确要表元数据更新时间)。"
sys := "你是资深数据库工程师助手,服务于数据库客户端 OpsxDB。回答使用中文。SQL 放在 ```sql 代码块中。\n规则:\n- 禁止输出思考过程、推理草稿、内心独白或逐步分析;只给用户最终答案。\n- 禁止编造上下文中未出现的列名或表结构;多轮跟进、用户换表时同样适用——上下文未列出该表的列,就不得在 SQL 里写 created_at/updated_at/order_time 等任何未出现的列名。\n- 缺表/缺时间列/意图不清/换表后列未知时:用一两句中文澄清(可问具体时间字段),或只给出探测列的 SQL(如 DESCRIBE / information_schema.COLUMNS);不要为了「可执行」硬造业务 SQL。\n- 「最近 N 天数据变化」默认指业务行时间过滤;不得默认用 information_schema.TABLES.UPDATE_TIME 充当终答(除非用户明确要表元数据更新时间)。\n- 「最新数据」类请求:仅当上下文已给出可用排序列(时间列或主键)时才写 ORDER BY … LIMIT;否则先澄清列名。"
var task string
switch strings.ToLower(req.Action) {
case "generate":
task = fmt.Sprintf("根据用户描述生成一条 %s 方言的 SQL。结构足够时:极简说明(≤2 行)+ 一条可执行 SQL;结构不足或缺时间列时:只输出一两句澄清问题,不要编造列名,不要输出思考过程。", driver)
task = fmt.Sprintf("根据用户描述生成一条 %s 方言的 SQL。结构足够且列名均已出现在上下文时:极简说明(≤2 行)+ 一条可执行 SQL;结构不足、换表后列未知或缺时间列时:只输出一两句澄清或 DESCRIBE/查列 SQL,禁止默认 created_at,不要输出思考过程。", driver)
case "ask":
task = "回答用户关于当前数据库、数据或 SQL 的问题;只有用户明确要求编写 SQL 时才给出 SQL(极简说明≤2 行 + SQL 块)。结构不足时一句澄清即可。你没有调用数据库工具,不要声称已经查询或修改数据。禁止输出思考过程。"
task = "回答用户关于当前数据库、数据或 SQL 的问题;只有用户明确要求编写 SQL 且上下文已含所需列名时才给出业务 SQL(极简说明≤2 行 + SQL 块)。列未知时一句澄清或只给 DESCRIBE/查列 SQL,禁止猜测 created_at 等。你没有调用数据库工具,不要声称已经查询或修改数据。禁止输出思考过程。"
case "explain":
task = "逐部分解释用户给出的 SQL:访问路径、索引使用、潜在代价。"
case "optimize":