Dify+test2data实现自然语言查询数据库

5 阅读1分钟

先展示chatflow工程图 image.png

实现关键

Dify 查询 MySQL 的关键:

  1. 需要给大模型表结构信息:这样 AI 才能生成正确的查询语句,查询出对应的表信息。
  2. 大模型需要具备查询数据库的能力:使用 Dify 的数据库查询工具text2data来实现。

具体实现

具体实现步骤如下:

  1. 在 Dify 应用市场安装数据库连接插件text2data
  2. 创建一个Chatflow 应用。
  3. 添加知识检索节点,将表结构给大模型。
  4. 添加 LLM节点,引用模型,添加系统描述。
  5. 添加参数提取器节点,添加提取参数
  6. 添加text2data(rookie_excute_sql)工具。
  7. 添加LLM节点,添加系统描述。
  8. 添加直接回复,回复内容为LLM2的text。

LLM系统描述

image.png

参数提取器配置

image.png

text2data配置

主要是选择待执行的SQL:为参数提取器的配置的SQL image.png

LLM2系统描述

image.png

最终效果

image.png