Adding long term, short term, entity and contextual memory

tests/memory/long_term_memory_test.py

@@ -0,0 +1,29 @@
+import pytest
+
+from crewai.memory.long_term.long_term_memory import LongTermMemory
+from crewai.memory.long_term.long_term_memory_item import LongTermMemoryItem
+
+
+@pytest.fixture
+def long_term_memory():
+    """Fixture to create a LongTermMemory instance"""
+    return LongTermMemory()
+
+
+def test_save_and_search(long_term_memory):
+    memory = LongTermMemoryItem(
+        agent="test_agent",
+        task="test_task",
+        expected_output="test_output",
+        datetime="test_datetime",
+        quality=0.5,
+        metadata={"task": "test_task", "quality": 0.5},
+    )
+    long_term_memory.save(memory)
+    find = long_term_memory.search("test_task")[0]
+    assert find["score"] == 0.5
+    assert find["datetime"] == "test_datetime"
+    assert find["metadata"]["agent"] == "test_agent"
+    assert find["metadata"]["quality"] == 0.5
+    assert find["metadata"]["task"] == "test_task"
+    assert find["metadata"]["expected_output"] == "test_output"

tests/memory/short_term_memory_test.py

@@ -0,0 +1,24 @@
+import pytest
+
+from crewai.memory.short_term.short_term_memory import ShortTermMemory
+from crewai.memory.short_term.short_term_memory_item import ShortTermMemoryItem
+
+
+@pytest.fixture
+def short_term_memory():
+    """Fixture to create a ShortTermMemory instance"""
+    return ShortTermMemory()
+
+
+def test_save_and_search(short_term_memory):
+    memory = ShortTermMemoryItem(
+        data="""test value test value test value test value test value test value
+        test value test value test value test value test value test value
+        test value test value test value test value test value test value""",
+        agent="test_agent",
+        metadata={"task": "test_task"},
+    )
+    short_term_memory.save(memory)
+    find = short_term_memory.search("test value", score_threshold=0.01)[0]
+    assert find["context"] == memory.data, "Data value mismatch."
+    assert find["metadata"]["agent"] == "test_agent", "Agent value mismatch."