from .adapters.base import MemoryAgent
from .adapters.letta import LettaAgent
from .adapters.mem0 import Mem0Agent
from .adapters.memu import MemUAgent
from .adapters.reme import ReMeAgent
from .adapters.text2mem import Text2MemAgent


def build_registry() -> dict[str, MemoryAgent]:
    """实例化所有适配器，并按系统 ID 建立索引。"""
    agents = [Text2MemAgent(), Mem0Agent(), LettaAgent(), ReMeAgent(), MemUAgent()]
    return {agent.id: agent for agent in agents}


agents = build_registry()