docker+ollama+flask+mysql实现本地数据库读取操作

1.安装mysql

安装mysql后,使用自带的数据库表

2. 使用flask,创建查询接口

文件目录

config.py,配置数据库相关信息

python 复制代码
MYSQL_HOST = "127.0.0.1"
MYSQL_PORT = 3306
MYSQL_USER = "root"
MYSQL_PASSWD = "123456"
MYSQL_DB = "world"

mysql_operate.py:创建mysql操作类

python 复制代码
import pymysql
from config.config import MYSQL_HOST, MYSQL_PORT, MYSQL_USER, MYSQL_PASSWD, MYSQL_DB

class MysqlDb():
    def __init__(self, host, port, user, password, db):
        self.conn = pymysql.connect(host=host,
                                    port=port,
                                    user=user,
                                    passwd=password,
                                    db=db)
        self.cursor = self.conn.cursor()

    def select_db(self, sql):
        self.conn.ping(reconnect=True)
        self.cursor.execute(sql)
        data = self.cursor.fetchall()
        return data

    def __del__(self):
        self.cursor.close()
        self.conn.close()

    def execute_db(self, sql):
        try:
            self.conn.ping(reconnect=True)
            self.cursor.execute(sql)
            self.conn.commit()
            return "插入成功"
        except Exception as e:
            self.conn.rollback()
            return "error"

db = MysqlDb(MYSQL_HOST, MYSQL_PORT, MYSQL_USER, MYSQL_PASSWD, MYSQL_DB)

app.py

python 复制代码
from flask import Flask
from flask import jsonify
from flask import request
from common import mysql_operate

app = Flask(__name__)

@app.route('/')
def index():
    return 'hello mysql'


@app.route('/query')
def query():  # put application's code here
    # sql = "SELECT * FROM city"
    data = request.get_json()
    if not data or 'sql' not in data:
        return jsonify({"error": "sql语句是必须的"})
    sql = data['sql']
    data = mysql_operate.db.select_db(sql)
    return jsonify(data)


if __name__ == '__main__':
    app.run()

3. 在dify中创建sql工作流

1.创建空白工作流

2.在开始模块中创建sql变量

3.创建http请求并修改相关参数

修改请求方式未post

修改请求地址,注意 127.0.0.1需要更改为host.docker.internal

求改headers

修改json数据
3.进行模板转换

4.添加结束节点

选择上一步模板转换中的输出output,如果没有3,则直接选择http body也可以

4.发布为工具

4.创建agent

1.创建agent

2.输入提示词

bash 复制代码
## 角色
你是一个专业的数据分析专家,并且精通根据用户的问题,数据表接口进行生成sql语句进行查询数据库
## 背景
根据用户问题,精确确定用户需求,首先查看数据表中的数据表结构,然后编写准确的并且可以直接执行的sql语句进行统计和查询数据库中的数据,并且返回相关数据,对于用户的工作效率提升十分重要
## 任务
根据用户的相关问题,首先查询知识库查看大概的数据表结构,然后编写可执行的mysql语句并且调用'查询数据库'工具进行查询数据库,用于最后满足用户的需求。对于用户需要查询数据库的需求务必调用'查询数据库'工具。
## mysql语法提示
如果需要子查询的话需要用如下语法
select * from table where name="beijing"
## 首先进行下面的逻辑推理过程
1.查询知识库了解数据库表结构
2.梳理和理解用户需求
3.分析这个需求需要编写多少sql语句
4.分析如何编写sql语句使得以最少的sql查询次数进行查询数据库
5.尝试编写仅仅一条或者5条内sql进行查询数据库
6.判断sql语句是否符合mysql语句的语法
## 提示
调用查询数据库工具时,直接输入组装好的sql语句即可,不用输入字典

3.添加知识库和工具

相关推荐
积跬步,慕至千里7 分钟前
clickhouse数据库表和doris数据库表迁移starrocks数据库时建表注意事项总结
数据库·clickhouse
艾伦_耶格宇16 分钟前
【docker】-1 docker简介
运维·docker·容器
极限实验室33 分钟前
搭建持久化的 INFINI Console 与 Easysearch 容器环境
数据库
白仑色1 小时前
Oracle PL/SQL 编程基础详解(从块结构到游标操作)
数据库·oracle·数据库开发·存储过程·plsql编程
IvanCodes2 小时前
二、Docker安装部署教程
docker·容器
程序猿小D2 小时前
[附源码+数据库+毕业论文]基于Spring+MyBatis+MySQL+Maven+jsp实现的个人财务管理系统,推荐!
java·数据库·mysql·spring·毕业论文·ssm框架·个人财务管理系统
钢铁男儿3 小时前
C# 接口(什么是接口)
java·数据库·c#
__风__4 小时前
PostgreSQL kv(jsonb)存储
数据库·postgresql
轩情吖4 小时前
Qt的第一个程序(2)
服务器·数据库·qt·qt creator·qlineedit·hello world·编辑框
Databend4 小时前
Databend 产品月报(2025年6月)
数据库