refactor(db): 重命名数据库初始化函数以明确其适用范围
- 将 initialize_db 函数重命名为 initialize_sqlite_db,以明确该函数专用于 SQLite 数据库 - 更新相关模块和文件中的引用,以确保代码一致性 - 此修改旨在提高代码的可读性和维护性,特别是未来可能接入多种数据库时
This commit is contained in:
parent
b52ca9b1af
commit
70b64dc3d3
@ -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']
|
||||
__all__ = ['get_engine', 'initialize_sqlite_db']
|
@ -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)
|
||||
initialize_sqlite_db(engine)
|
4
main.py
4
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("# 基于文档驱动的自适应编码大模型微调框架")
|
||||
|
Loading…
x
Reference in New Issue
Block a user