carry/gzhu-biyesheji
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(global_var): 重构全局变量管理

Browse Source 
- 移除了 _docs 全局变量
- 更新了 get_docs() 函数，使其在每次调用时重新扫描文档目录
- 优化了全局变量初始化逻辑
This commit is contained in:
carry 2025-04-15 15:25:44 +08:00
parent d9abf08184
commit d0aebd17fa
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
global_var.py
View File

@ -3,16 +3,14 @@ from tools import scan_docs_directory
_prompt_store = None _prompt_store = None
_sql_engine = None _sql_engine = None
_docs = None
_datasets = None _datasets = None
_model = None _model = None
_tokenizer = None _tokenizer = None
_workdir = None _workdir = None
def init_global_var(workdir="workdir"): def init_global_var(workdir="workdir"):
global _prompt_store, _sql_engine, _docs, _datasets, _workdir global _prompt_store, _sql_engine, _datasets, _workdir
_prompt_store = get_prompt_tinydb(workdir) _prompt_store = get_prompt_tinydb(workdir)
_sql_engine = get_sqlite_engine(workdir) _sql_engine = get_sqlite_engine(workdir)
_docs = scan_docs_directory(workdir)
_datasets = get_all_dataset(workdir) _datasets = get_all_dataset(workdir)
_workdir = workdir _workdir = workdir
@ -34,12 +32,8 @@ def set_sql_engine(new_sql_engine):
_sql_engine = new_sql_engine _sql_engine = new_sql_engine
def get_docs(): def get_docs():
return _docs global _workdir
return scan_docs_directory(_workdir)
def set_docs(new_docs):
global _docs
_docs = new_docs
def get_datasets(): def get_datasets():
return _datasets return _datasets