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" __version__ = "1.3.0"
__author__ = "ViperEkura" __author__ = "ViperEkura"
from khaosz.khaosz import Khaosz from khaosz.api import Khaosz
from khaosz.config import ( from khaosz.config import (
TransformerConfig, TransformerConfig,
ParameterLoader, ParameterLoader,
@ -33,14 +33,21 @@ from khaosz.trainer import (
) )
__all__ = [ __all__ = [
# model
"Khaosz", "Khaosz",
# module
"Transformer", "Transformer",
"Retriever",
"SemanticTextSplitter",
"PriorityTextSplitter",
"TransformerConfig", "TransformerConfig",
"BpeTokenizer",
"ParameterLoader", "ParameterLoader",
"TrainConfig",
"DatasetLoader",
"BpeTokenizer",
"TextGenerator", "TextGenerator",
"ChatGenerator", "ChatGenerator",
"StreamGenerator", "StreamGenerator",
@ -48,15 +55,7 @@ __all__ = [
"RetrievalGenerator", "RetrievalGenerator",
"EmbeddingEncoder", "EmbeddingEncoder",
# trainer
"Trainer", "Trainer",
"DatasetLoader", # 保持在 __all__ 中,但来源是 khaosz.data
"TrainConfig",
"StrategyFactory", "StrategyFactory",
"SchedulerFactory", "SchedulerFactory"
# utils
"Retriever",
"SemanticTextSplitter",
"PriorityTextSplitter",
] ]