diff --git a/server/internal/ai/ai_test.go b/server/internal/ai/ai_test.go index 1288e71..88e562d 100644 --- a/server/internal/ai/ai_test.go +++ b/server/internal/ai/ai_test.go @@ -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) diff --git a/server/internal/ai/prompt.go b/server/internal/ai/prompt.go index d943503..6f54c86 100644 --- a/server/internal/ai/prompt.go +++ b/server/internal/ai/prompt.go @@ -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":