diff --git a/src/crewai/memory/memory.py b/src/crewai/memory/memory.py index 752a683b4..69df613e5 100644 --- a/src/crewai/memory/memory.py +++ b/src/crewai/memory/memory.py @@ -10,8 +10,10 @@ class Memory(BaseModel): Base class for memory, now supporting agent tags and generic metadata. """ + storage: Union[RAGStorage, Any] + def __init__(self, storage: Union[RAGStorage, Any]): - self.storage = storage + super().__init__(storage=storage) def save( self, @@ -34,3 +36,6 @@ class Memory(BaseModel): return self.storage.search( query=query, limit=limit, score_threshold=score_threshold ) + + class Config: + arbitrary_types_allowed = True