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