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

feat(core): APP use new SDK component (#1050)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-01-10 10:39:04 +08:00
committed by GitHub
parent e11b72c724
commit fa8b5b190c
242 changed files with 2768 additions and 2163 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
dbgpt/util/_db_migration_utils.py
View File

@@ -1,12 +1,12 @@
from typing import Optional
import os
import logging
from sqlalchemy import Engine, text
from sqlalchemy.orm import Session, DeclarativeMeta
from alembic import command
from alembic.util.exc import CommandError
from alembic.config import Config as AlembicConfig
import os
from typing import Optional
from alembic import command
from alembic.config import Config as AlembicConfig
from alembic.util.exc import CommandError
from sqlalchemy import Engine, text
from sqlalchemy.orm import DeclarativeMeta, Session
logger = logging.getLogger(__name__)
@@ -67,8 +67,8 @@ def create_migration_script(
Returns:
The path of the generated migration script.
"""
from alembic.script import ScriptDirectory
from alembic.runtime.migration import MigrationContext
from alembic.script import ScriptDirectory
# Check if the database is up-to-date
script_dir = ScriptDirectory.from_config(alembic_cfg)
@@ -244,8 +244,8 @@ def _check_database_migration_status(alembic_cfg: AlembicConfig, engine: Engine)
Raises:
Exception: If the database is not at the latest revision.
"""
from alembic.script import ScriptDirectory
from alembic.runtime.migration import MigrationContext
from alembic.script import ScriptDirectory
script = ScriptDirectory.from_config(alembic_cfg)