From ef3b93ab93f14de4d0d893a91fe441c7997c9102 Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Mon, 9 Dec 2024 09:58:15 -0300 Subject: [PATCH] feat: Add ContextualMemory to __init__ --- src/crewai/memory/__init__.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) 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", +]