manager.py

from typing import Dict
from base_memory import BaseMemoryModule
from memory_item import MemoryItem

class UnifiedMemoryManager:
    def __init__(self):
        self.modules: Dict[str, BaseMemoryModule] = {}

    def register_memory_type(self, name: str, module: BaseMemoryModule):
        self.modules[name] = module

    def add(self, module_name: str, item: MemoryItem):
        self.modules[module_name].add_item(item)

    def query(self, query: str, module_name: str = None):
        if module_name:
            return self.modules[module_name].retrieve(query)
        results = []
        for mod in self.modules.values():
            results.extend(mod.retrieve(query))
        return results