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

View File

@@ -10,10 +10,8 @@ class Memory(BaseModel):
Base class for memory, now supporting agent tags and generic metadata. Base class for memory, now supporting agent tags and generic metadata.
""" """
storage: Union[RAGStorage, Any]
def __init__(self, storage: Union[RAGStorage, Any]): def __init__(self, storage: Union[RAGStorage, Any]):
super().__init__(storage=storage) self.storage = storage
def save( def save(
self, self,
@@ -36,6 +34,3 @@ class Memory(BaseModel):
return self.storage.search( return self.storage.search(
query=query, limit=limit, score_threshold=score_threshold query=query, limit=limit, score_threshold=score_threshold
) )
class Config:
arbitrary_types_allowed = True