Add in missing triple quote and execution time to resume agent functionality. (#1025)

* Add in missing triple quote and execution time to resume agent functionality

* Fixing broken kwargs and other issues causing our tests to fail
This commit is contained in:
Brandon Hancock (bhancock_ai)
2024-07-29 13:39:02 -04:00
committed by GitHub
parent 25c314befc
commit fa4393d77e
4 changed files with 22 additions and 43 deletions

View File

@@ -19,7 +19,7 @@ class ShortTermMemory(Memory):
super().__init__(storage)
def save(self, item: ShortTermMemoryItem) -> None:
super().save(item.data, item.metadata, item.agent)
super().save(value=item.data, metadata=item.metadata, agent=item.agent)
def search(self, query: str, score_threshold: float = 0.35):
return self.storage.search(query=query, score_threshold=score_threshold) # type: ignore # BUG? The reference is to the parent class, but the parent class does not have this parameters