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]:
    agents = [Text2MemAgent(), Mem0Agent(), LettaAgent(), ReMeAgent(), MemUAgent()]
    return {agent.id: agent for agent in agents}


agents = build_registry()