From cc5201ee5211fa5750dc750260930fa8ea8ef2d5 Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Fri, 7 Feb 2025 10:51:31 -0500 Subject: [PATCH] wip --- src/crewai/memory/memory.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/src/crewai/memory/memory.py b/src/crewai/memory/memory.py index 69df613e5..752a683b4 100644 --- a/src/crewai/memory/memory.py +++ b/src/crewai/memory/memory.py @@ -10,10 +10,8 @@ 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]): - super().__init__(storage=storage) + self.storage = storage def save( self, @@ -36,6 +34,3 @@ class Memory(BaseModel): return self.storage.search( query=query, limit=limit, score_threshold=score_threshold ) - - class Config: - arbitrary_types_allowed = True