From 70b64dc3d3db659127fa79c0a0e49cf771ca6469 Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Tue, 8 Apr 2025 14:16:12 +0800
Subject: [PATCH] =?UTF-8?q?refactor(db):=20=E9=87=8D=E5=91=BD=E5=90=8D?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93=E5=88=9D=E5=A7=8B=E5=8C=96=E5=87=BD?=
 =?UTF-8?q?=E6=95=B0=E4=BB=A5=E6=98=8E=E7=A1=AE=E5=85=B6=E9=80=82=E7=94=A8?=
 =?UTF-8?q?=E8=8C=83=E5=9B=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 将 initialize_db 函数重命名为 initialize_sqlite_db，以明确该函数专用于 SQLite 数据库
- 更新相关模块和文件中的引用，以确保代码一致性
- 此修改旨在提高代码的可读性和维护性，特别是未来可能接入多种数据库时
---
 db/__init__.py | 4 ++--
 db/init_db.py  | 4 ++--
 main.py        | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

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("# 基于文档驱动的自适应编码大模型微调框架")