mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-23 07:33:53 +08:00
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
from typing import Dict, Any
|
|
from core.components.component_manager import ComponentManager, ComponentType
|
|
from core.components.adapters.tts_adapter import TTSFactory
|
|
from core.components.adapters.asr_adapter import ASRFactory
|
|
from core.components.adapters.vad_adapter import VADFactory
|
|
from core.components.adapters.llm_adapter import LLMFactory
|
|
from core.components.adapters.memory_adapter import MemoryFactory
|
|
from core.components.adapters.intent_adapter import IntentFactory
|
|
from config.logger import setup_logging
|
|
|
|
logger = setup_logging()
|
|
|
|
|
|
class ComponentRegistry:
|
|
"""组件注册器：统一管理所有组件工厂的注册"""
|
|
|
|
_factories_registered = False
|
|
|
|
@classmethod
|
|
def create_component_manager(cls, config: Dict[str, Any]) -> ComponentManager:
|
|
"""创建并配置组件管理器"""
|
|
manager = ComponentManager(config)
|
|
|
|
# 只在第一次时记录注册日志
|
|
if not cls._factories_registered:
|
|
logger.info("注册组件工厂")
|
|
cls._factories_registered = True
|
|
|
|
# 注册所有组件工厂（每个manager都需要注册，但不重复记录日志）
|
|
manager.register_factory(TTSFactory())
|
|
manager.register_factory(ASRFactory())
|
|
manager.register_factory(VADFactory())
|
|
manager.register_factory(LLMFactory())
|
|
manager.register_factory(MemoryFactory())
|
|
manager.register_factory(IntentFactory())
|
|
|
|
# 设置组件初始化顺序（考虑依赖关系）
|
|
# VAD -> ASR -> LLM -> Memory/Intent -> TTS
|
|
initialization_order = [
|
|
ComponentType.VAD,
|
|
ComponentType.ASR,
|
|
ComponentType.LLM,
|
|
ComponentType.MEMORY,
|
|
ComponentType.INTENT,
|
|
ComponentType.TTS,
|
|
]
|
|
manager.set_initialization_order(initialization_order)
|
|
|
|
if not cls._factories_registered:
|
|
logger.info("组件管理器创建完成，已注册所有组件工厂")
|
|
|
|
return manager
|
|
|
|
@staticmethod
|
|
def get_required_components(config: Dict[str, Any]) -> list[ComponentType]:
|
|
"""根据配置获取需要的组件类型"""
|
|
required = []
|
|
selected_modules = config.get("selected_module", {})
|
|
|
|
if selected_modules.get("VAD"):
|
|
required.append(ComponentType.VAD)
|
|
if selected_modules.get("ASR"):
|
|
required.append(ComponentType.ASR)
|
|
if selected_modules.get("LLM"):
|
|
required.append(ComponentType.LLM)
|
|
if selected_modules.get("TTS"):
|
|
required.append(ComponentType.TTS)
|
|
if selected_modules.get("Memory"):
|
|
required.append(ComponentType.MEMORY)
|
|
if selected_modules.get("Intent"):
|
|
required.append(ComponentType.INTENT)
|
|
|
|
return required
|
|
|
|
|