andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
Brandon Hancock
2025-02-07 10:51:14 -05:00
parent f9b4e14173
commit 8659a531c5
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai/memory/memory.py
View File

@@ -10,8 +10,10 @@ 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]):
self.storage = storage super().__init__(storage=storage)
def save( def save(
self, self,
@@ -34,3 +36,6 @@ 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