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