registry.py 446 B

123456789101112131415
  1. from .adapters.base import MemoryAgent
  2. from .adapters.letta import LettaAgent
  3. from .adapters.mem0 import Mem0Agent
  4. from .adapters.memu import MemUAgent
  5. from .adapters.reme import ReMeAgent
  6. from .adapters.text2mem import Text2MemAgent
  7. def build_registry() -> dict[str, MemoryAgent]:
  8. agents = [Text2MemAgent(), Mem0Agent(), LettaAgent(), ReMeAgent(), MemUAgent()]
  9. return {agent.id: agent for agent in agents}
  10. agents = build_registry()