From d0aebd17fad24180b126e44a6012c2e6427f53f0 Mon Sep 17 00:00:00 2001
From: carry <luokairui@carry.fit>
Date: Tue, 15 Apr 2025 15:25:44 +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?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 移除了 _docs 全局变量
- 更新了 get_docs() 函数，使其在每次调用时重新扫描文档目录
- 优化了全局变量初始化逻辑
---
 global_var.py | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/global_var.py b/global_var.py
index 66b3118..594ea55 100644
--- a/global_var.py
+++ b/global_var.py
@@ -3,16 +3,14 @@ from tools import scan_docs_directory
 
 _prompt_store = None
 _sql_engine = None
-_docs = None
 _datasets = None
 _model = None
 _tokenizer = None
 _workdir = None
 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)
     _sql_engine = get_sqlite_engine(workdir)
-    _docs = scan_docs_directory(workdir)
     _datasets = get_all_dataset(workdir)
     _workdir = workdir
 
@@ -34,12 +32,8 @@ def set_sql_engine(new_sql_engine):
     _sql_engine = new_sql_engine
 
 def get_docs():
-    return _docs
-
-def set_docs(new_docs):
-    global _docs
-    _docs = new_docs
-
+    global _workdir
+    return scan_docs_directory(_workdir)
 def get_datasets():
     return _datasets