Skip to content

Commit d9d788c

Browse files
authored
Enable deduplication for trailing extraction indexing (#173)
1 parent b160585 commit d9d788c

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

agent_memory_server/long_term_memory.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,10 @@ async def run_delayed_extraction(
393393

394394
# Index the extracted memories
395395
if extracted_memories:
396-
await index_long_term_memories(extracted_memories)
396+
await index_long_term_memories(
397+
extracted_memories,
398+
deduplicate=True,
399+
)
397400
logger.info(
398401
f"Trailing extraction completed for session {session_id}: "
399402
f"{len(extracted_memories)} memories extracted and indexed"

tests/test_extraction_logic_fix.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -198,6 +198,7 @@ async def test_run_delayed_extraction_extracts_and_indexes(
198198
indexed_memories = mock_index.call_args[0][0]
199199
assert len(indexed_memories) == 1
200200
assert indexed_memories[0].id == "extracted-1"
201+
assert mock_index.call_args.kwargs.get("deduplicate") is True
201202

202203
# Verify working memory was updated - messages marked as extracted
203204
updated_wm = await get_working_memory(

0 commit comments

Comments
 (0)