mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
wip
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user