f655936741
- 新增 init_global_var 函数，用于统一初始化所有全局变量 - 修改 get_prompt_store、get_sql_engine、get_docs 和 get_datasets 函数，使用新的全局变量初始化逻辑 - 更新 main.py 中的代码，使用新的 init_global_var 函数替代原有的单独初始化方法
60 lines
1.2 KiB
Python
60 lines
1.2 KiB
Python
from db import get_sqlite_engine, get_prompt_tinydb, get_all_dataset
|
|
from tools import scan_docs_directory
|
|
|
|
_prompt_store = None
|
|
_sql_engine = None
|
|
_docs = None
|
|
_datasets = None
|
|
|
|
def init_global_var(workdir="workdir"):
|
|
"""Initialize all global variables"""
|
|
global _prompt_store, _sql_engine, _docs, _datasets
|
|
_prompt_store = get_prompt_tinydb(workdir)
|
|
_sql_engine = get_sqlite_engine(workdir)
|
|
_docs = scan_docs_directory(workdir)
|
|
_datasets = get_all_dataset(workdir)
|
|
|
|
def get_prompt_store():
|
|
return _prompt_store
|
|
|
|
def set_prompt_store(new_prompt_store):
|
|
global _prompt_store
|
|
_prompt_store = new_prompt_store
|
|
|
|
def get_sql_engine():
|
|
return _sql_engine
|
|
|
|
def set_sql_engine(new_sql_engine):
|
|
global _sql_engine
|
|
_sql_engine = new_sql_engine
|
|
|
|
def get_docs():
|
|
return _docs
|
|
|
|
def set_docs(new_docs):
|
|
global _docs
|
|
_docs = new_docs
|
|
|
|
def get_datasets():
|
|
return _datasets
|
|
|
|
def set_datasets(new_datasets):
|
|
global _datasets
|
|
_datasets = new_datasets
|
|
|
|
_model = None
|
|
_tokenizer = None
|
|
|
|
def get_model():
|
|
return _model
|
|
|
|
def set_model(new_model):
|
|
global _model
|
|
_model = new_model
|
|
|
|
def get_tokenizer():
|
|
return _tokenizer
|
|
|
|
def set_tokenizer(new_tokenizer):
|
|
global _tokenizer
|
|
_tokenizer = new_tokenizer |