保姆级教程:用AI实时查询数据库,自动生成可视化图表|Dify工作流

摘要

针对业务人员不会写SQL的痛点,提供Dify工作流教程:通过AI自动查询数据库并生成可视化图表,结合知识库提升查询准确率,解决数据安全与效率问题。

现在基本上每个上点规模的公司都会有数据库,但想要让业务同学自己去写 SQL 取数,其实是不现实的,很多小白会觉得 SQL 甚至比 Python 还要难学。

现在,不怕了,我们可以用 AI 来帮我们完成数据库的查询,甚至生成文字洞察和可视化图表。

之前我写过一篇教程,用 MCP 的方式,让 Cursor 自动查询数据库并完成表结构的修改等操作。

但说实话,MCP 的方式目前还是很不稳定的,不适合用在实际工作中。

更不用说把 Excel 直接扔给 AI 做分析的方式:1. 数据安全问题;2. 出来的结果准确率很低。

怎么办?

最好的方式就是在本地部署工作流,直接对接数据库**,通过给 SQL 代码然后执行的方式来查询,准确率最高。**

但数据库如果量比较大,又怎么才能让 AI 知道我表结构,并给出准确代码呢?答案是搭建数据库表结构的知识库

以下就是具体怎么做的教程。

效果及逻辑如下:

给定一个查询需求后,AI 会到知识库查询表结构,并据此生成 SQL 语句,执行后,调用 Echarts 图表的能力,以及数据科学家的 Agent,最终生成一个图文并茂的可视化查询结果。

dify-ai-sql-visualization-workflow

文末****附工作流 DSL 文件获取方式

数据准备

我准备了一个数据库,里面包含了销售表、产品表、地区表,可以实现复杂的多表查询,也是在实际业务中很常见的需求场景。

dify-ai-sql-visualization-workflow

新增节点

在 Dify 里做 SQL 查询,逻辑是:

  1. 先用 AI 生成 SQL 代码

  2. 执行 SQL 代码

这里可以直接搜索 database 工具,里面就包含了这两个功能:Text to SQLSQL Execute

dify-ai-sql-visualization-workflow
dify-ai-sql-visualization-workflow

去授权,按以下形式输入你的数据库地址和账号密码,以 mysql 为例:

mysql+pymysql://root:8455be@152.12.12.12:3306/bgggtest

OK,接下来我们就来具体实现 SQL 查询。

模块一:SQL 查询

为了让大家能理解我搭建这个工作流的逻辑,我逐步演示给大家看。同时,这也是从 0 开始搭建一个 AI 工作流的方式:小步快跑、不断测试。

坐稳扶好,马上出发!!!

假设 SQL 查询有分级。。。

第 1 级 青铜:指定表格、字段名称查询

这也是最简单的,如图,直接告诉 AI 我要查哪个表的什么字段的内容

dify-ai-sql-visualization-workflow

就能得到如下结果,这步是为了来验证 AI 的运行是否有问题。

## 得到的查询语句
{
  "text": "SELECT * FROM 销售表 LIMIT 5",
  "files": [],
  "json": []
}
## 查询结果
{
  "text": "",
  "files": [],
  "json": [
    {
      "result": [
        {
          "id": "1028",
          "sku": "1012551685",
          "spu": "101255",
          "区域 Id": "中南-0091",
          "商品件数": "4",
          "客户 Id": "10150",
          "订单 Id": "CN-2020-3064620",
          "订单日期": "2020-05-24",
          "订单金额": "213"
        },
        {
          "id": "1322",
          "sku": "1005590887",
          "spu": "100559",
          "区域 Id": "华东-0013",
          "商品件数": "3",
          "客户 Id": "10150",
          "订单 Id": "CN-2021-1086295",
          "订单日期": "2021-01-09",
          "订单金额": "1748.01"
        }
      ]
    }
  ]
}

第 2 级 白银:在提示词里给表结构

我们不可能说每个查询语句都自己写清楚表格和字段(如果是这样的话我自己都写完 SQL 了 还要 AI 干嘛)

所以这里,我们再加一步,把数据库的表结构全部导出来给到 AI

那问题来了,怎么拿到数据库表结构呢?

我的方案是用 Navicat,链接数据库后,如图,右键 转储SQL文件-仅结构

dify-ai-sql-visualization-workflow
dify-ai-sql-visualization-workflow

就能得到以下的建表语句:

dify-ai-sql-visualization-workflow

接下来要做的就很简单了:把提示词放到生成 SQL 语句的提示词里即可。这里就不重复演示了。

做了一下测试,继续验证数据的正确性 统计每天销售额

{
  "text": "SELECT `订单日期`, SUM(`订单金额`) AS '每日销售金额' FROM `销售表` GROUP BY `订单日期` LIMIT 5",
  "files": [],
  "json": []
}

{
  "text": "",
  "files": [],
  "json": [
    {
      "result": [
        {
          "每日销售金额": 19747.02,
          "订单日期": "2020-05-24"
        },
        {
          "每日销售金额": 3707.9900000000002,
          "订单日期": "2021-01-09"
        },
        {
          "每日销售金额": 11604.099999999999,
          "订单日期": "2021-02-11"
        },
        {
          "每日销售金额": 20188.98,
          "订单日期": "2021-06-11"
        },
        {
          "每日销售金额": 47586.95000000001,
          "订单日期": "2021-05-06"
        }
      ]
    }
  ]
}

数据没问题,继续升级。

dify-ai-sql-visualization-workflow

至此,我们已经能满足很多场景的 SQL 查询了,但可能还无法应对实际业务中的需求,因为实际业务中的数据表会很多很多、表结构也复杂,不太可能把整个数据库表结构都直接放到提示词里,否则会导致上下文过长。

怎么办呢?

第 3 级 黄金:把表结构放到知识库里,调用 RAG 查询

如果可以让 AI 在查询的时候,顺手去查表结构,就好了!!!这就是解决方案。

新建数据库表结构知识库

首先,在 Dify 创建知识库,如下图:

dify-ai-sql-visualization-workflow

还记得之前我们导出来的数据库结构的文件吗,把它们放到 txt 文件里,然后上传。

dify-ai-sql-visualization-workflow

知识库设置关键点:

  1. 如下图,正常来说按默认配置就好了,预览块看右边的分块结果,要确保一个Chunk一个表

dify-ai-sql-visualization-workflow

  1. 索引方式要选 高质量 的,否则永远查不出来结果

例如下面左边就是经济(无法击中)、右边是高质量(能准确查到)

dify-ai-sql-visualization-workflow
dify-ai-sql-visualization-workflow

  1. 检索设置中的** Top K 可以根据数据复杂程度选高一点,正常来说一次查询最多就 3-5 个表做连接**。如果再多的话,或许就不太适合用 AI 了。

dify-ai-sql-visualization-workflow

新建知识检索节点

回到我们的工作流,在开始的之后,新建一个知识检索,也就是说,在用户对话的时候,就根据用户的需求,提前先到知识库里找好需要用到的表结构,喂给 AI。

dify-ai-sql-visualization-workflow

此时,因为我们需要用到知识检索的结果,就要用用 LLM 大模型节点来生成,只需要设置好提示词,以及把检索到 的表结构通过上下文的形式放进去即可:

dify-ai-sql-visualization-workflow

完整提示词:

【角色设定】
您是一位资深数据库专家,擅长根据业务需求将自然语言精准转换为符合规范的SQL语句。您具有以下核心能力:
1. 精通关系型数据库设计,尤其熟悉MySQL语法
2. 深入理解当前业务系统的表结构
[上下文]
3. 能准确识别业务需求中的隐藏条件
4. 熟悉SQL性能优化和事务控制

【输入处理流程】
1. 结构确认:先验证涉及的表是否存在,检查字段名是否准确
2. 关联分析:确认多表关联时的连接条件是否正确
3. 条件推导:将模糊的业务描述转化为精确的WHERE条件
4. 异常处理:识别可能存在的逻辑矛盾或性能陷阱

【输出要求】
输出sql中不要添加任何注释信息
采用如下格式返回结果:
[符合ANSI标准的SQL语句]

【示例】
用户问题:"查询北京用户最近一个月购买过智能手机的订单"
输出:
SELECT o.order_no, u.username, oi.total_price, o.order_time 
FROM orders o
JOIN users u ON o.user_id = u.id
JOIN order_items oi ON o.id = oi.order_id
JOIN products p ON oi.product_id = p.id
WHERE u.address LIKE '北京市%'
  AND p.product_name = '智能手机'
  AND o.order_time >= CURRENT_DATE - INTERVAL '1 month'
  AND o.status NOT IN ('已取消') 
ORDER BY o.order_time DESC;

新增一个数据处理节点

但注意,大模型跑出来的 SQL 语句里面有很多换位符等需要清洗,这里我们简单加个 Python 代码执行

import re
def main(arg: str) -> dict:
    result = re.sub(r'```sql\n|```', '', arg)
    return {
        "result": result
    }

测试运行结果

目前的工作流和对话效果如图所示,也就完成了我们第一个模块:text2SQL

dify-ai-sql-visualization-workflow

升级测试需求,做多表查询:查询各地区销售数据

得到的 SQL 代码:

SELECT t2.`地区`, SUM(CAST(t1.`订单金额` AS DECIMAL(18, 2))) AS `销售合计` 
  FROM `销售表` AS t1
  INNER JOIN `地区表` AS t2 ON t1.`区域 Id` = t2.`区域 Id`
  GROUP BY t2.`地区`;

最终运行结果(节选示例):

{
  "text": "",
  "files": [],
  "json": [
    {
      "result": [
        {
          "地区": "华东",
          "销售合计": "2574033.54"
        },
        {
          "地区": "西南",
          "销售合计": "598791.29"
        },
        {
          "地区": "东北",
          "销售合计": "1420192.51"
        }
      ]
    }
  ]
}

结果没问题的 ✅

dify-ai-sql-visualization-workflow

番外:调用 Agent 能力

本来,我想测试 Dify 用 Agent 的查 SQL 的能力。

逻辑是:让 Agent 自行去决定调用什么工具,也就是说前面我们的工作流,把生成代码、执行代码的部分删掉,改为 Agent 的工具,来执行。

结果不知道是 bug 还是什么问题,一直跑不出来。

虽然工作流看上去比较简单了,但实际上工作量也没少多少,还依赖大模型的工具调用能力,不确定性比较高,不适合在实际业务中使用。大家可以自己试一下。

模块二:可视化图表

接下来完成可视化图表的部分。

1.0 青铜

首先,新增插件 Echarts图表生成

dify-ai-sql-visualization-workflow

但这个工具目前只有三种图可以用:线性图(折线图)、柱状图、饼图。正常来说也是最常用的。

dify-ai-sql-visualization-workflow

继续来看我怎么从 0 开始创建这个工作流的。

刚开始,我肯定不知道这个组件怎么用,所以要先看看这个图表的逻辑是什么样的,如下图,随便填数据

dify-ai-sql-visualization-workflow

得到的图表是这样的,我们就明白了组件里的参数怎么填:

  • 标题:很好理解

  • 数据:就是 y 轴上的数据

  • X 轴:就是 x 轴上的标签或者数据显示

dify-ai-sql-visualization-workflow

柱状图和饼图都是一样的数据结构:

dify-ai-sql-visualization-workflow
dify-ai-sql-visualization-workflow

OK,测试后,我们就明白了:现在的问题就变成了怎么把 sql 查询出来的结果,转成Echarts需要的数据结构?

有两种方法可以解决

方法一:用 Dify 内置工具****参数提取器

适用场景:这个方法本质还是用 AI 去提取数据,所以适合数据量比较小的时候,上下文不大,才不容易出错。

dify-ai-sql-visualization-workflow

参考提示词:

## 任务
从JSON数据中提取出不同的数据参数用于生成echarts图表

## 要求
1. 根据数据起一个标题并赋值给title
2.  根据数据判断要只做的echarts图表类型提取y轴的数据赋值给ydata每个文本用单引号引起之间用 "," 分隔
3. 把x轴的数据赋值给xdate每个文本用单引号引起之间用 ";" 分隔

## 示例
给定的JSON数据

{
"text": "销售日期 | 日销售总额 \n———-|——–\n2021-05-01|18819.99\n2021-05-02|3292.00 \n2021-05-04|37564.94\n2021-05-05|3607.01",
"files": [],
"json": []
}


抽取的结果:

title: 2021 年 5 月 1 每日销售
ydata: 18819.99;3292;37564.939999999995;3607.01
xdata: 2021-05-01;2021-05-02;2021-05-04;2021-05-05


需要处理的数据:

然后再把提取器的数据给到图表中引用即可,如图:

dify-ai-sql-visualization-workflow

方法二:AI 编程,写一个脚本从 JSON 中提取出数据。

适用场景:查询出来的数据量比较大的时候,用代码处理不容易出错。

首先要调整一个地方,就是 SQL Execute****这个节点是可以定义输出格式的,原先我们一直用的是 Markdown,好处是在最后的对话中能直接呈现表格,但不利于我们解析数据。要改成 JSON

dify-ai-sql-visualization-workflow

现在看下这个节点输出的 SQL 结果长什么样:

计算2021年5月的每日销售

就得到以下的 JSON 数据格式(节选示意)

{
  "text": "",
  "files": [],
  "json": [
    {
      "result": [
        {
          "每日销售额": 18819.99,
          "订单日期": "2021-05-01"
        },
        {
          "每日销售额": 3292,
          "订单日期": "2021-05-02"
        },
        {
          "每日销售额": 37564.939999999995,
          "订单日期": "2021-05-04"
        },
        {
          "每日销售额": 3607.01,
          "订单日期": "2021-05-05"
        }
      ]
    }
  ]
}

根据这个输入,就可以让 Deepseek V3 帮我们写一个 Python 语句了

dify-ai-sql-visualization-workflow

大家也可以直接用我的:

import json
def main(arg: str) -> dict:

    data = arg

    # 获取result列表
    result_list = data[0]['result']

    result = str(result_list)

    ydata = ";".join([str(list(item.values())[0]) for item in result_list])
    xdata = ";".join([list(item.values())[1] for item in result_list])

    return {
        "ydata": ydata,
        "xdata": xdata
    }

这里的一个麻烦的地方就在于 SQL 运行后的名字的不确定,所以就不能直接用名字去获取,例如 item[‘日期’]

而是改用数字的形式 例如 list(item.values())[0] 这样

同时,用代码的方式,标题就要再用一个 AI 节点来生成了,相对来说比较麻烦。

除非是流程非常确定的情况下,否则不推荐使用这种方式。

这样我们就初步完成了可视化图表的模块。

为了让结果更丰富,我们可以新增一个 数据科学家 的 AI 节点,让 AI 做一段定性的文字分析,参考提示词:

# 角色 
 数据科学家 
 ## 注意 
 1. 激励模型深入思考角色配置细节,确保任务完成。 
 2. 专家设计应考虑使用者的需求和关注点。 
 3. 使用情感提示的方法来强调角色的意义和情感层面。 
 ## 性格类型指标 
 INTJ(内向直觉思维判断型) 
 ## 背景 
 作为数据科学家,这个角色旨在为用户提供数据分析、模型构建和预测的专业知识。专家通过分析大量数据,帮助用户发现数据背后的模式和趋势,从而为决策提供支持。 
 ## 约束条件 
 - 必须遵循数据隐私和安全原则 
 - 需要保持客观和中立,避免个人偏见影响分析结果 
 ## 定义 
 数据科学:数据科学是一门跨学科领域,涉及使用统计学、机器学习、数据挖掘和可视化技术来分析和解释数据,以发现模式和做出预测。 
 ## 目标 
 1. 提供准确的数据分析和洞察 
 2. 构建有效的预测模型 
 3. 帮助用户做出基于数据的决策 
 ## Skills 
 1. 统计学和机器学习知识 
 2. 数据处理和分析能力 
 3. 编程和算法实现能力 
 4. 数据可视化技能 
 ## 音调 
 专业、客观、细致、耐心 
 ## 价值观 
 - 追求数据的真实性和准确性 
 - 以数据驱动的决策为依据 
 - 尊重数据的隐私和安全 
 # Initialization 
 您好,作为数据科学家,我将严格遵循工作流程,为您提供深入的数据分析和专业的建议。让我们开始吧,共同探索数据背后的奥秘。

这样就像模像样了。

dify-ai-sql-visualization-workflow

接下来解决另一个问题:

前面我们做的案例是 线性图表 的,难道接下来要对柱状图、饼图都做一样的流程吗?如果后面有了更多图表,岂不是很麻烦?

2.0 白银

所以,我们就需要在前面新增一个 判断用哪种图片 的节点,然后让 AI 自动选择好后,呈现出合适的可视化类型。

怎么做呢?

还记得我们的参数提取器吗,它本身就是一个 AI 的运行逻辑,所以我们可以在里面新增一个 ChartType 的参数,同时在指令里也给出对应的要求和示例,如下图:

dify-ai-sql-visualization-workflow

然后在 参数提取器 的后面新增 条件分支,如下图,一个类型对应一个图表,而每个图表都是引用一样的参数即可。

dify-ai-sql-visualization-workflow

最后,为了让 AI 回复部分用同样的参数,我们需要新增一个会话变量 chart_var

然后在每个图表后面都新增一个 变量赋值 的节点,把对应的图表结果都赋值给它,这样直接回复的部分直接引用这个 chart_var 就好了。

dify-ai-sql-visualization-workflow

最终效果还是不错的。

dify-ai-sql-visualization-workflow

此 Dify 工作流 DSL 文件获取方式:

关注公众号「饼干哥哥 AGI」,后台回复关键词「Dify」即可




👤 关于作者:饼干哥哥 & NGS
我是饼干哥哥,数据分析师、AI 博主,和出海业务专家朋友创立了公司 NGS NextGrowthSail,专注 AI 在出海营销场景下的落地。上周我们内部复盘自动化内容营销工作流数据时,发现如果用上这种AI驱动数据库查询技术,客户的数据分析效率能提升30%以上。

发表评论