Files
DB-GPT/pilot/connections/rdbms/mysql.py
2023-05-29 19:32:20 +08:00

19 lines
394 B
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import pymysql
from pilot.connections.rdbms.rdbms_connect import RDBMSDatabase
class MySQLConnect(RDBMSDatabase):
"""Connect MySQL Database fetch MetaData For LLM Prompt
Args:
Usage:
"""
type:str = "MySQL"
connect_url = "mysql+pymysql://"
default_db = ["information_schema", "performance_schema", "sys", "mysql"]