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

feat: Add crew train cli (#624)

Browse Source 
* fix: fix crewai-tools cli command

* feat: add crewai train CLI command

* feat: add the tests

* fix: fix typing hinting issue on code

* fix: test.yml

* fix: fix test

* fix: removed fix since it didnt changed the test
This commit is contained in:
Eduardo Chiarotti
2024-05-23 18:46:45 -03:00
committed by GitHub
parent a336381849
commit 24ed8a2549
16 changed files with 278 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/crewai/memory/entity/entity_memory.py
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)

4
src/crewai/memory/short_term/short_term_memory.py
View File

@@ -13,7 +13,9 @@ class ShortTermMemory(Memory):
"""
def __init__(self, crew=None, embedder_config=None):
storage = RAGStorage(type="short_term", embedder_config=embedder_config, crew=crew)
storage = RAGStorage(
type="short_term", embedder_config=embedder_config, crew=crew
)
super().__init__(storage)
def save(self, item: ShortTermMemoryItem) -> None: # type: ignore # BUG?: Signature of "save" incompatible with supertype "Memory"