multi_agents/memory/__init__.py

from .draft import DraftState
from .research import ResearchState

__all__ = [
    "DraftState",
    "ResearchState"
]