diff --git a/db/__init__.py b/db/__init__.py index aeed376..8fe1308 100644 --- a/db/__init__.py +++ b/db/__init__.py @@ -1,3 +1,3 @@ -from .init_db import get_engine, initialize_sqlite_db +from .init_db import get_sqlite_engine, initialize_sqlite_db -__all__ = ['get_engine', 'initialize_sqlite_db'] \ No newline at end of file +__all__ = ['get_sqlite_engine', 'initialize_sqlite_db'] \ No newline at end of file diff --git a/db/init_db.py b/db/init_db.py index f05df0b..2e49dc5 100644 --- a/db/init_db.py +++ b/db/init_db.py @@ -14,7 +14,7 @@ from schema.dataset_generation import APIProvider # 全局变量,用于存储数据库引擎实例 _engine: Optional[Engine] = None -def get_engine(workdir: str) -> Engine: +def get_sqlite_engine(workdir: str) -> Engine: """ 获取数据库引擎实例。如果引擎尚未创建,则创建一个新的引擎并返回。 @@ -74,6 +74,6 @@ if __name__ == "__main__": # 定义工作目录路径 workdir = os.path.join(os.path.dirname(__file__), "..", "workdir") # 获取数据库引擎 - engine = get_engine(workdir) + engine = get_sqlite_engine(workdir) # 初始化数据库 initialize_sqlite_db(engine) \ No newline at end of file diff --git a/frontend/setting_page.py b/frontend/setting_page.py index 437063f..a2b0885 100644 --- a/frontend/setting_page.py +++ b/frontend/setting_page.py @@ -1,7 +1,7 @@ import gradio as gr from typing import List from sqlmodel import Session, select -from db import get_engine +from db import get_sqlite_engine from schema import APIProvider import os from global_var import sql_engine diff --git a/global_var.py b/global_var.py index 7a494b3..ee0ef40 100644 --- a/global_var.py +++ b/global_var.py @@ -1,3 +1,3 @@ -from db import get_engine +from db import get_sqlite_engine -sql_engine = get_engine("workdir") \ No newline at end of file +sql_engine = get_sqlite_engine("workdir") \ No newline at end of file