refactor(khaosz): 重构模块导出结构并重命名主模块文件
This commit is contained in:
parent
e72e244df6
commit
05b012820b
|
|
@ -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",
|
|
||||||
]
|
]
|
||||||
Loading…
Reference in New Issue