From 5094febcb48a7d77f940044687191ad6fbd1aa69 Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Sun, 13 Apr 2025 00:54:55 +0800
Subject: [PATCH] =?UTF-8?q?refactor(global=5Fvar):=20=E9=87=8D=E6=9E=84?=
 =?UTF-8?q?=E5=85=A8=E5=B1=80=E5=8F=98=E9=87=8F=E7=AE=A1=E7=90=86=E5=B9=B6?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=B7=A5=E4=BD=9C=E7=9B=AE=E5=BD=95=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 添加 _workdir 全局变量以存储工作目录路径
- 在 init_global_var 函数中初始化 _workdir
- 新增 get_workdir 函数以获取工作目录路径
- 调整全局变量的定义和初始化顺序
---
 global_var.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/global_var.py b/global_var.py
index 69d353a..5a449b4 100644
--- a/global_var.py
+++ b/global_var.py
@@ -5,14 +5,19 @@ _prompt_store = None
 _sql_engine = None
 _docs = None
 _datasets = None
-
+_model = None
+_tokenizer = None
+_workdir = 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)
+    _workdir = workdir
+
+def get_workdir():
+    return _workdir
 
 def get_prompt_store():
     return _prompt_store
@@ -42,9 +47,6 @@ def set_datasets(new_datasets):
     global _datasets
     _datasets = new_datasets
 
-_model = None
-_tokenizer = None
-
 def get_model():
     return _model