From 05b012820b996843dd4c56966eeff88f6c7fc033 Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Mon, 20 Oct 2025 13:07:02 +0800
Subject: [PATCH] =?UTF-8?q?refactor(khaosz):=20=E9=87=8D=E6=9E=84=E6=A8=A1?=
 =?UTF-8?q?=E5=9D=97=E5=AF=BC=E5=87=BA=E7=BB=93=E6=9E=84=E5=B9=B6=E9=87=8D?=
 =?UTF-8?q?=E5=91=BD=E5=90=8D=E4=B8=BB=E6=A8=A1=E5=9D=97=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 khaosz/__init__.py           | 25 ++++++++++++-------------
 khaosz/{khaosz.py => api.py} |  0
 2 files changed, 12 insertions(+), 13 deletions(-)
 rename khaosz/{khaosz.py => api.py} (100%)

diff --git a/khaosz/__init__.py b/khaosz/__init__.py
index 8556926..a53b616 100644
--- a/khaosz/__init__.py
+++ b/khaosz/__init__.py
@@ -1,7 +1,7 @@
 __version__ = "1.3.0"
 __author__ = "ViperEkura"
 
-from khaosz.khaosz import Khaosz
+from khaosz.api import Khaosz
 from khaosz.config import (
     TransformerConfig,
     ParameterLoader,
@@ -33,14 +33,21 @@ from khaosz.trainer import (
 )
 
 __all__ = [
-    # model
     "Khaosz",
     
-    # module
     "Transformer",
+    
+    "Retriever",
+    "SemanticTextSplitter",
+    "PriorityTextSplitter",
+    
     "TransformerConfig",
-    "BpeTokenizer",
     "ParameterLoader",
+    "TrainConfig",
+    
+    "DatasetLoader",
+    "BpeTokenizer",
+    
     "TextGenerator",
     "ChatGenerator",
     "StreamGenerator",
@@ -48,15 +55,7 @@ __all__ = [
     "RetrievalGenerator",
     "EmbeddingEncoder",
     
-    # trainer
     "Trainer",
-    "DatasetLoader",  # 保持在 __all__ 中，但来源是 khaosz.data
-    "TrainConfig",
     "StrategyFactory",
-    "SchedulerFactory",
-    
-    # utils
-    "Retriever",
-    "SemanticTextSplitter",
-    "PriorityTextSplitter",
+    "SchedulerFactory"
 ]
\ No newline at end of file
diff --git a/khaosz/khaosz.py b/khaosz/api.py
similarity index 100%
rename from khaosz/khaosz.py
rename to khaosz/api.py