mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
feat(memory): adds support for customizable memory interface (#1339)
* feat(memory): adds support for customizing crew storage * chore: allow overwriting the crew memory configuration * docs: update custom storage usage * fix(lint): use correct syntax * fix: type check warning * fix: type check warnings * fix(test): address agent default failing test * fix(lint). address type checker error * Update crew.py --------- Co-authored-by: João Moura <joaomdmoura@gmail.com>
This commit is contained in:
@@ -10,12 +10,13 @@ class EntityMemory(Memory):
|
||||
Inherits from the Memory class.
|
||||
"""
|
||||
|
||||
def __init__(self, crew=None, embedder_config=None):
|
||||
storage = RAGStorage(
|
||||
type="entities",
|
||||
allow_reset=False,
|
||||
embedder_config=embedder_config,
|
||||
crew=crew,
|
||||
def __init__(self, crew=None, embedder_config=None, storage=None):
|
||||
storage = (
|
||||
storage
|
||||
if storage
|
||||
else RAGStorage(
|
||||
type="entities", allow_reset=False, embedder_config=embedder_config, crew=crew
|
||||
)
|
||||
)
|
||||
super().__init__(storage)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user