feat: routes for adding/linking new memory blocks to agents + unlinking blocks from agents (#2083)

tests/test_memory.py

@@ -1,6 +1,7 @@
 import pytest
 
 # Import the classes here, assuming the above definitions are in a module named memory_module
+from letta.schemas.block import Block
 from letta.schemas.memory import ChatMemory, Memory
 
 
@@ -105,3 +106,37 @@ def test_memory_jinja2_set_template(sample_memory: Memory):
     )
     with pytest.raises(ValueError):
         sample_memory.set_prompt_template(prompt_template=template_bad_memory_structure)
+
+
+def test_link_unlink_block(sample_memory: Memory):
+    """Test linking and unlinking a block to the memory"""
+
+    # Link a new block
+
+    test_new_label = "test_new_label"
+    test_new_value = "test_new_value"
+    test_new_block = Block(label=test_new_label, value=test_new_value, limit=2000)
+
+    current_labels = sample_memory.list_block_labels()
+    assert test_new_label not in current_labels
+
+    sample_memory.link_block(block=test_new_block)
+    assert test_new_label in sample_memory.list_block_labels()
+    assert sample_memory.get_block(test_new_label).value == test_new_value
+
+    # Unlink the block
+    sample_memory.unlink_block(block_label=test_new_label)
+    assert test_new_label not in sample_memory.list_block_labels()
+
+
+def test_update_block_label(sample_memory: Memory):
+    """Test updating the label of a block"""
+
+    test_new_label = "test_new_label"
+    current_labels = sample_memory.list_block_labels()
+    assert test_new_label not in current_labels
+    test_old_label = current_labels[0]
+
+    sample_memory.update_block_label(current_label=test_old_label, new_label=test_new_label)
+    assert test_new_label in sample_memory.list_block_labels()
+    assert test_old_label not in sample_memory.list_block_labels()