refactor(khaosz): 重构模块导出结构并重命名主模块文件

This commit is contained in:
ViperEkura 2025-10-20 13:07:02 +08:00
parent e72e244df6
commit 05b012820b
2 changed files with 12 additions and 13 deletions

View File

@ -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"
]