Skip to content

Commit c1f0729

Browse files
abrookinsclaude
andcommitted
fix: update test imports after moving recency functions
- Update test imports to use utils.recency module for moved functions - Remove unnecessary comments from imports 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent b324f48 commit c1f0729

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tests/test_forgetting.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
from datetime import UTC, datetime, timedelta
22

33
from agent_memory_server.long_term_memory import (
4-
rerank_with_recency, # new: pure function
5-
score_recency, # new: pure function
6-
select_ids_for_forgetting, # new: pure function
4+
select_ids_for_forgetting,
75
)
86
from agent_memory_server.models import MemoryRecordResult, MemoryTypeEnum
7+
from agent_memory_server.utils.recency import (
8+
rerank_with_recency,
9+
score_recency,
10+
)
911

1012

1113
def make_result(

tests/test_long_term_memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
deduplicate_by_id,
1313
delete_long_term_memories,
1414
extract_memory_structure,
15-
generate_memory_hash,
1615
index_long_term_memories,
1716
merge_memories_with_llm,
1817
promote_working_memory_to_long_term,
@@ -24,6 +23,7 @@
2423
MemoryRecordResults,
2524
MemoryTypeEnum,
2625
)
26+
from agent_memory_server.utils.recency import generate_memory_hash
2727

2828

2929
# from agent_memory_server.utils.redis import ensure_search_index_exists # Not used currently

0 commit comments

Comments
 (0)