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