diff --git a/db/__init__.py b/db/__init__.py index e570f0f..aeed376 100644 --- a/db/__init__.py +++ b/db/__init__.py @@ -1,3 +1,3 @@ -from .init_db import get_engine, initialize_db +from .init_db import get_engine, initialize_sqlite_db -__all__ = ['get_engine', 'initialize_db'] \ No newline at end of file +__all__ = ['get_engine', 'initialize_sqlite_db'] \ No newline at end of file diff --git a/db/init_db.py b/db/init_db.py index caf9274..f05df0b 100644 --- a/db/init_db.py +++ b/db/init_db.py @@ -37,7 +37,7 @@ def get_engine(workdir: str) -> Engine: _engine = create_engine(db_url) return _engine -def initialize_db(engine: Engine) -> None: +def initialize_sqlite_db(engine: Engine) -> None: """ 初始化数据库,创建所有表结构,并插入初始数据(如果不存在)。 @@ -76,4 +76,4 @@ if __name__ == "__main__": # 获取数据库引擎 engine = get_engine(workdir) # 初始化数据库 - initialize_db(engine) \ No newline at end of file + initialize_sqlite_db(engine) \ No newline at end of file diff --git a/main.py b/main.py index ac64130..b220894 100644 --- a/main.py +++ b/main.py @@ -1,11 +1,11 @@ import gradio as gr from frontend.setting_page import setting_page from frontend import chat_page,setting_page,train_page,dataset_page -from db import initialize_db +from db import initialize_sqlite_db from global_var import sql_engine if __name__ == "__main__": - initialize_db(sql_engine) + initialize_sqlite_db(sql_engine) with gr.Blocks() as app: gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")