From 868fcd45ba00676178cb0859abc74fccbf899e93 Mon Sep 17 00:00:00 2001
From: carry <luokairui@carry.fit>
Date: Sat, 19 Apr 2025 21:49:19 +0800
Subject: [PATCH] =?UTF-8?q?refactor(project):=20=E9=87=8D=E6=9E=84?=
 =?UTF-8?q?=E9=A1=B9=E7=9B=AE=E6=96=87=E4=BB=B6=E7=BB=84=E7=BB=87=E7=BB=93?=
 =?UTF-8?q?=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 修改模型管理和训练页面的导入路径
- 更新 main.py 中的导入模块
- 调整 tools 包的内容，移除 model 模块
- 新建 train 包，包含 model 模块
- 优化 __init__.py 文件，简化导入语句
---
 frontend/model_manage_page.py | 2 +-
 frontend/train_page.py        | 3 ++-
 main.py                       | 2 +-
 tools/__init__.py             | 3 +--
 train/__init__.py             | 1 +
 {tools => train}/model.py     | 0
 6 files changed, 6 insertions(+), 5 deletions(-)
 create mode 100644 train/__init__.py
 rename {tools => train}/model.py (100%)

diff --git a/frontend/model_manage_page.py b/frontend/model_manage_page.py
index fe2f0ce..7483049 100644
--- a/frontend/model_manage_page.py
+++ b/frontend/model_manage_page.py
@@ -7,7 +7,7 @@ import torch
 
 sys.path.append(str(Path(__file__).resolve().parent.parent))
 from global_var import get_model, get_tokenizer, set_model, set_tokenizer
-from tools.model import get_model_name
+from train import get_model_name
 
 def model_manage_page():
     workdir = "workdir"  # 假设workdir是当前工作目录下的一个文件夹
diff --git a/frontend/train_page.py b/frontend/train_page.py
index bcfd986..91e0e27 100644
--- a/frontend/train_page.py
+++ b/frontend/train_page.py
@@ -8,7 +8,8 @@ from transformers import TrainerCallback
 
 sys.path.append(str(Path(__file__).resolve().parent.parent))
 from global_var import get_model, get_tokenizer, get_datasets, get_workdir
-from tools import train_model, find_available_port
+from tools import find_available_port
+from train import train_model
 
 def train_page():
     with gr.Blocks() as demo:
diff --git a/main.py b/main.py
index f44429a..e7337f1 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,5 @@
 import gradio as gr
-import unsloth
+import train
 from frontend import *
 from db import initialize_sqlite_db, initialize_prompt_store
 from global_var import init_global_var, get_sql_engine, get_prompt_store
diff --git a/tools/__init__.py b/tools/__init__.py
index 6d15436..b384ee1 100644
--- a/tools/__init__.py
+++ b/tools/__init__.py
@@ -1,6 +1,5 @@
-from .parse_markdown import parse_markdown
+from .parse_markdown import *
 from .document import *
 from .json_example import generate_example_json
-from .model import *
 from .port import *
 from .reasoning import call_openai_api
\ No newline at end of file
diff --git a/train/__init__.py b/train/__init__.py
new file mode 100644
index 0000000..71a1d85
--- /dev/null
+++ b/train/__init__.py
@@ -0,0 +1 @@
+from .model import *
\ No newline at end of file
diff --git a/tools/model.py b/train/model.py
similarity index 100%
rename from tools/model.py
rename to train/model.py