refactor: 优化工厂模式结构

2026-04-04 11:33:58 +08:00
parent 7861af12e4
commit e97536758f
9 changed files with 164 additions and 136 deletions
@@ -5,7 +5,8 @@ from astrai.config import (
    ModelConfig,
    TrainConfig,
 )
-from astrai.data import BpeTokenizer, DatasetLoader
+from astrai.core.factory import BaseFactory
+from astrai.data import BpeTokenizer, DatasetFactory
 from astrai.inference.generator import (
    BatchGenerator,
    EmbeddingEncoder,
@@ -21,7 +22,7 @@ __all__ = [
    "Transformer",
    "ModelConfig",
    "TrainConfig",
-    "DatasetLoader",
+    "DatasetFactory",
    "BpeTokenizer",
    "GenerationRequest",
    "LoopGenerator",
@@ -32,4 +33,5 @@ __all__ = [
    "Trainer",
    "StrategyFactory",
    "SchedulerFactory",
+    "BaseFactory",
 ]