github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-04 10:00:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:meta_data mysql ddl bug

Browse Source
This commit is contained in:
aries_ckt 2023-10-18 00:06:23 +08:00
parent f981831352
commit 96ae5006fc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/base_modules/meta_data/meta_data.py
View File

@ -7,7 +7,7 @@ import fnmatch
from datetime import datetime from datetime import datetime
from typing import Optional, Type, TypeVar from typing import Optional, Type, TypeVar
from sqlalchemy import create_engine, DateTime, String, func, MetaData from sqlalchemy import create_engine, DateTime, String, func, MetaData, DDL
from sqlalchemy.exc import OperationalError from sqlalchemy.exc import OperationalError
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm import Mapped from sqlalchemy.orm import Mapped
@ -47,7 +47,7 @@ if CFG.LOCAL_DB_TYPE == "mysql":
try: try:
# try to connect # try to connect
with engine_temp.connect() as conn: with engine_temp.connect() as conn:
conn.execute(f"CREATE DATABASE IF NOT EXISTS {db_name}") conn.execute(DDL(f"CREATE DATABASE IF NOT EXISTS {db_name}"))
print(f"Already connect '{db_name}'") print(f"Already connect '{db_name}'")
except OperationalError as e: except OperationalError as e: