在Dify中需通过JSON Schema强制LLM输出结构化SQL、添加JSON解析与SQL语法校验节点、启用SQL沙箱执行(只读连接+仅允许SELECT),实现安全自动执行。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

在Dify中让大模型生成的SQL语句自动通过结构化校验并安全执行,避免直接运行未验证的SQL导致数据库误操作或注入风险。
配置LLM输出为JSON Schema格式
进入Dify应用编辑页→知识库/工作流/模型配置 → 找到「提示词」或「模型参数」区域 → 在系统提示词末尾追加明确约束:
请严格按以下JSON Schema输出结果,字段名、类型、嵌套结构必须完全一致,不得添加额外字段或注释:{"type":"object","properties":{"sql":{"type":"string"},"description":{"type":"string"},"parameters":{"type":"array","items":{"type":"string"}}},"required":["sql","description"]}
这一步强制模型只输出可解析的JSON对象,跳过自由文本解释。若不加该约束,模型可能返回“我已生成SQL:SELECT * FROM users;”这类混合内容,后续无法自动提取sql字段。
添加JSON解析与SQL结构校验节点
在Dify工作流中新增一个「代码」节点(Code Node):
方法一:使用内置JSON解析器
选择「JSON Schema Validator」工具 → 粘贴上一步定义的schema → 输入变量设为{{llm_output}} → 勾选「失败时中断流程」。
方法二:自定义Python脚本校验
输入代码如下:
import json
data = json.loads({{llm_output}})
if not isinstance(data.get("sql"), str) or not data.get("sql").strip().upper().startswith(("SELECT", "INSERT", "UPDATE", "DELETE")):
raise ValueError("SQL字段缺失或不以合法关键词开头")
print(json.dumps({"valid": True, "sql": data["sql"], "params": data.get("parameters", [])}))
【必须确保输入变量名与上游LLM节点输出字段名完全一致,例如有些版本输出为{{llm.text}}而非{{llm_output}}】
启用沙箱环境执行SQL
第一步:在Dify后台 → 设置 → 安全 → 启用「SQL沙箱执行」开关
第二步:为当前应用绑定一个只读数据库连接(如PostgreSQL只读用户、MySQL无写权限账号)
第三步:在工作流中添加「Database Query」节点 → 选择已配置的沙箱数据源 → SQL字段填入{{parsed_sql.sql}}(即上一节点输出的合法SQL)→ 设置超时为3000ms → 开启「仅允许SELECT语句」选项
这一步会拦截所有非查询类SQL,即使JSON校验通过了INSERT语句,沙箱也会拒绝执行。沙箱不依赖模型判断,而是由数据库连接权限+语法白名单双重控制。
第四步:勾选「返回执行计划而非结果集」用于调试阶段,确认SQL能被数据库解析且不触发全表扫描警告。











