feat: 优化工厂模式的实现

This commit is contained in:
2026-04-04 15:49:46 +08:00
parent aa5e03d7f6
commit 3346c75584
11 changed files with 228 additions and 142 deletions
+1 -3
View File
@@ -6,7 +6,7 @@ from jinja2 import Template
from torch import Tensor
from astrai.config.param_config import ModelParameter
from astrai.core.factory import BaseFactory
from astrai.factory import BaseFactory
from astrai.inference.core import EmbeddingEncoderCore, GeneratorCore, KVCacheManager
HistoryType = List[Tuple[str, str]]
@@ -299,8 +299,6 @@ class GeneratorFactory(BaseFactory[GeneratorCore]):
result = generator.generate(request)
"""
_registry: Dict[str, type] = {}
@staticmethod
def create(parameter: ModelParameter, request: GenerationRequest) -> GeneratorCore:
"""Create a generator based on request characteristics.