From 8659a531c59e27fa7664dcfd76687ff1d4c3bb6b Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Fri, 7 Feb 2025 10:51:14 -0500
Subject: [PATCH] wip

---
 src/crewai/memory/memory.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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