LiteCLI中多SQL查询提示问题的分析与解决

在数据库命令行工具LiteCLI的使用过程中，开发团队发现了一个关于LLM(大型语言模型)提示生成SQL查询的有趣问题。当用户请求生成多个SQL查询方案时，模型可能会将所有查询合并到同一个SQL代码块中，这会导致命令行界面无法正确处理多个查询语句。

问题现象

当用户通过\llm+命令请求生成多个SQL查询方案时，例如查找people表中的重复记录，LLM模型(如Deepseek-reasoner)可能会返回类似以下格式的内容：

-- 方法1：分组计数查找重复项
SELECT first_name, last_name, COUNT(*) 
FROM people
GROUP BY first_name, last_name
HAVING COUNT(*) > 1;

-- 方法2：自连接查找重复对
SELECT p1.first_name, p1.last_name
FROM people p1
JOIN people p2
  ON p1.first_name = p2.first_name
  AND p1.last_name = p2.last_name
  AND p1.rowid < p2.rowid;

这种格式会导致LiteCLI将两个查询作为一个整体执行，可能引发语法错误或产生不符合预期的结果。

技术分析

这个问题本质上是一个提示工程(Prompt Engineering)的挑战。LLM模型在生成SQL查询时，如果没有明确的格式指导，倾向于将所有相关查询放在同一个代码块中。这与命令行界面处理SQL语句的预期行为不匹配。

在数据库CLI工具中，通常期望：

1. 每个SQL查询应该独立执行
2. 注释和解释性文字应该与可执行代码分离
3. 多个查询方案应该清晰地区分开来

解决方案

开发团队通过改进提示词工程解决了这个问题。在提示词中明确添加了以下指导原则：

"当用户请求多个SQL查询时，请将它们分别放在独立的代码块中"

这个简单的调整显著改善了模型的行为。改进后，模型会生成如下格式的输出：

-- 方法1：分组计数
SELECT first_name, last_name, COUNT(*) AS duplicate_count
FROM people
GROUP BY first_name, last_name
HAVING COUNT(*) > 1;

-- 方法2：行号标记
WITH cte AS (
    SELECT first_name, last_name,
           ROW_NUMBER() OVER (PARTITION BY first_name, last_name ORDER BY rowid) AS rn
    FROM people
)
SELECT first_name, last_name
FROM cte
WHERE rn > 1;

实现效果

这种改进带来了以下好处：

1. 命令行界面可以正确识别并单独执行每个查询
2. 用户可以选择性地执行感兴趣的查询方案
3. 保持了查询之间的逻辑关联性，同时确保技术上的独立性
4. 提升了用户体验，避免了意外执行多个查询的风险

经验总结

这个案例展示了提示词工程在AI辅助开发工具中的重要性。通过精心设计的提示词，可以引导模型生成更符合技术场景需求的输出格式。对于数据库CLI工具这类技术产品，输出格式的规范性往往与功能正确性同等重要。

开发团队还注意到，即使用户请求多个方案，命令行界面默认只预填充第一个查询也是更合理的用户体验设计。这避免了意外执行大量查询的风险，同时保留了用户查看和选择其他方案的灵活性。