New version, updating dependencies, fixing memory

This commit is contained in:
João Moura
2024-05-13 22:26:41 -03:00
parent 615ac7f297
commit 27fddae286
9 changed files with 430 additions and 520 deletions

View File

@@ -10,9 +10,9 @@ class EntityMemory(Memory):
Inherits from the Memory class.
"""
def __init__(self, embedder_config=None):
def __init__(self, crew=None, embedder_config=None):
storage = RAGStorage(
type="entities", allow_reset=False, embedder_config=embedder_config
type="entities", allow_reset=False, embedder_config=embedder_config, crew=crew
)
super().__init__(storage)