From feaea1fb647aa9d143c2d4de7671b3c73c99089d Mon Sep 17 00:00:00 2001
From: carry <luokairui@carry.fit>
Date: Fri, 18 Apr 2025 15:16:29 +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=BC=95=E6=93=8E=E5=8A=A0=E8=BD=BD?=
 =?UTF-8?q?=E5=87=BD=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 将 get_sqlite_engine 函数重命名为 load_sqlite_engine
- 更新了相关模块中的导入和调用
- 这个改动是为了更好地反映函数的实际功能，提高代码可读性
---
 db/__init__.py | 4 ++--
 db/init_db.py  | 4 ++--
 global_var.py  | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/db/__init__.py b/db/__init__.py
index 93ec022..ae45bae 100644
--- a/db/__init__.py
+++ b/db/__init__.py
@@ -1,9 +1,9 @@
-from .init_db import get_sqlite_engine, initialize_sqlite_db
+from .init_db import load_sqlite_engine, initialize_sqlite_db
 from .prompt_store import get_prompt_tinydb, initialize_prompt_store
 from .dataset_store import get_all_dataset
 
 __all__ = [
-    "get_sqlite_engine",
+    "load_sqlite_engine",
     "initialize_sqlite_db",
     "get_prompt_tinydb",
     "initialize_prompt_store",
diff --git a/db/init_db.py b/db/init_db.py
index 9fad3af..cb79475 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_sqlite_engine(workdir: str) -> Engine:
+def load_sqlite_engine(workdir: str) -> Engine:
     """
     获取数据库引擎实例。如果引擎尚未创建，则创建一个新的引擎并返回。
 
@@ -74,6 +74,6 @@ if __name__ == "__main__":
     # 定义工作目录路径
     workdir = os.path.join(os.path.dirname(__file__), "..", "workdir")
     # 获取数据库引擎
-    engine = get_sqlite_engine(workdir)
+    engine = load_sqlite_engine(workdir)
     # 初始化数据库
     initialize_sqlite_db(engine)
\ No newline at end of file
diff --git a/global_var.py b/global_var.py
index 594ea55..16866ac 100644
--- a/global_var.py
+++ b/global_var.py
@@ -1,4 +1,4 @@
-from db import get_sqlite_engine, get_prompt_tinydb, get_all_dataset
+from db import load_sqlite_engine, get_prompt_tinydb, get_all_dataset
 from tools import scan_docs_directory
 
 _prompt_store = None
@@ -10,7 +10,7 @@ _workdir = None
 def init_global_var(workdir="workdir"):
     global _prompt_store, _sql_engine, _datasets, _workdir
     _prompt_store = get_prompt_tinydb(workdir)
-    _sql_engine = get_sqlite_engine(workdir)
+    _sql_engine = load_sqlite_engine(workdir)
     _datasets = get_all_dataset(workdir)
     _workdir = workdir