diff --git a/src/crewai/memory/__init__.py b/src/crewai/memory/__init__.py index 3f7ca2ad6..eedf86bc7 100644 --- a/src/crewai/memory/__init__.py +++ b/src/crewai/memory/__init__.py @@ -1,6 +1,13 @@ +from .contextual.contextual_memory import ContextualMemory from .entity.entity_memory import EntityMemory from .long_term.long_term_memory import LongTermMemory from .short_term.short_term_memory import ShortTermMemory from .user.user_memory import UserMemory -__all__ = ["UserMemory", "EntityMemory", "LongTermMemory", "ShortTermMemory"] +__all__ = [ + "UserMemory", + "EntityMemory", + "LongTermMemory", + "ShortTermMemory", + "ContextualMemory", +]