Skip to content

Commit dac18fc

Browse files
committed
Add test for MCP memory record defaults
- Add test_mcp_lenient_memory_record_defaults to verify LenientMemoryRecord and ExtractedMemoryRecord default to discrete_memory_extracted='t' - This ensures regression protection for the memory extraction flag fix
1 parent 0299d9f commit dac18fc

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

tests/test_mcp.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -437,3 +437,34 @@ async def test_set_working_memory_auto_id_generation(self, mcp_test_setup):
437437
memory = working_memory.memories[0]
438438
assert memory.id is not None
439439
assert len(memory.id) > 0 # ULID generates non-empty strings
440+
441+
@pytest.mark.asyncio
442+
async def test_mcp_lenient_memory_record_defaults(self, session, mcp_test_setup):
443+
"""Test that LenientMemoryRecord used by MCP has correct defaults for discrete_memory_extracted."""
444+
from agent_memory_server.models import (
445+
ExtractedMemoryRecord,
446+
LenientMemoryRecord,
447+
)
448+
449+
# Test 1: LenientMemoryRecord should default to discrete_memory_extracted='t'
450+
lenient_memory = LenientMemoryRecord(
451+
text="User likes green tea",
452+
memory_type="semantic",
453+
namespace="user_preferences",
454+
)
455+
456+
assert (
457+
lenient_memory.discrete_memory_extracted == "t"
458+
), f"LenientMemoryRecord should default to 't', got '{lenient_memory.discrete_memory_extracted}'"
459+
assert lenient_memory.memory_type.value == "semantic"
460+
assert lenient_memory.id is not None
461+
462+
# Test 2: ExtractedMemoryRecord should also default to discrete_memory_extracted='t'
463+
extracted_memory = ExtractedMemoryRecord(
464+
id="test_001", text="User prefers coffee", memory_type="semantic"
465+
)
466+
467+
assert (
468+
extracted_memory.discrete_memory_extracted == "t"
469+
), f"ExtractedMemoryRecord should default to 't', got '{extracted_memory.discrete_memory_extracted}'"
470+
assert extracted_memory.memory_type.value == "semantic"

0 commit comments

Comments
 (0)