mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
f2dfa79432
Co-authored-by: cpacker <packercharles@gmail.com> Co-authored-by: matthew zhou <matthewzhou@matthews-MacBook-Pro.local> Co-authored-by: Zack Field <field.zackery@gmail.com>
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
import pytest
|
|
|
|
# Import the classes here, assuming the above definitions are in a module named memory_module
|
|
from memgpt.schemas.memory import ChatMemory, Memory
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_memory():
|
|
return ChatMemory(persona="Chat Agent", human="User")
|
|
|
|
|
|
def test_create_chat_memory():
|
|
"""Test creating an instance of ChatMemory"""
|
|
chat_memory = ChatMemory(persona="Chat Agent", human="User")
|
|
assert chat_memory.get_block("persona").value == "Chat Agent"
|
|
assert chat_memory.get_block("human").value == "User"
|
|
|
|
|
|
def test_dump_memory_as_json(sample_memory: Memory):
|
|
"""Test dumping ChatMemory as JSON compatible dictionary"""
|
|
memory_dict = sample_memory.to_dict()
|
|
assert isinstance(memory_dict, dict)
|
|
assert "persona" in memory_dict
|
|
assert memory_dict["persona"]["value"] == "Chat Agent"
|
|
|
|
|
|
def test_load_memory_from_json(sample_memory: Memory):
|
|
"""Test loading ChatMemory from a JSON compatible dictionary"""
|
|
memory_dict = sample_memory.to_dict()
|
|
print(memory_dict)
|
|
new_memory = Memory.load(memory_dict)
|
|
assert new_memory.get_block("persona").value == "Chat Agent"
|
|
assert new_memory.get_block("human").value == "User"
|
|
|
|
|
|
# def test_memory_functionality(sample_memory):
|
|
# """Test memory modification functions"""
|
|
# # Get memory functions
|
|
# functions = get_memory_functions(ChatMemory)
|
|
# # Test core_memory_append function
|
|
# append_func = functions['core_memory_append']
|
|
# print("FUNCTIONS", functions)
|
|
# env = {}
|
|
# env.update(globals())
|
|
# for tool in functions:
|
|
# # WARNING: name may not be consistent?
|
|
# exec(tool.source_code, env)
|
|
#
|
|
# print(exec)
|
|
#
|
|
# append_func(sample_memory, 'persona', " is a test.")
|
|
# assert sample_memory.memory['persona'].value == "Chat Agent\n is a test."
|
|
# # Test core_memory_replace function
|
|
# replace_func = functions['core_memory_replace']
|
|
# replace_func(sample_memory, 'persona', " is a test.", " was a test.")
|
|
# assert sample_memory.memory['persona'].value == "Chat Agent\n was a test."
|
|
|
|
|
|
def test_memory_limit_validation(sample_memory: Memory):
|
|
"""Test exceeding memory limit"""
|
|
with pytest.raises(ValueError):
|
|
ChatMemory(persona="x" * 3000, human="y" * 3000)
|
|
|
|
with pytest.raises(ValueError):
|
|
sample_memory.get_block("persona").value = "x" * 3000
|