feat: add crewai train CLI command

This commit is contained in:
Eduardo Chiarotti
2024-05-16 19:43:33 -03:00
parent 5de494c99b
commit a958b31768
11 changed files with 86 additions and 11 deletions

View File

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