This commit is contained in:
Brandon Hancock
2025-02-07 10:51:14 -05:00
parent f9b4e14173
commit 8659a531c5

View File

@@ -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