Skip to content

Commit 3478f2c

Browse files
committed
test: update tree manager test script
1 parent 1b8a9b0 commit 3478f2c

File tree

1 file changed

+15
-4
lines changed

1 file changed

+15
-4
lines changed

tests/memories/textual/test_tree_manager.py

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212
def mock_graph_store():
1313
store = MagicMock()
1414
store.get_node.return_value = {
15-
"id": str(uuid.uuid4()),
16-
"memory": "old text",
15+
"id": "id",
16+
"memory": "text",
1717
"metadata": {"confidence": 90, "background": "", "tags": [], "sources": [], "usage": []},
1818
}
1919
store.search_by_embedding.return_value = [{"id": str(uuid.uuid4()), "score": 0.95}]
@@ -30,8 +30,19 @@ def mock_embedder():
3030

3131

3232
@pytest.fixture
33-
def memory_manager(mock_graph_store, mock_embedder):
34-
return MemoryManager(graph_store=mock_graph_store, embedder=mock_embedder)
33+
def mock_llm():
34+
llm = MagicMock()
35+
llm.run.side_effect = lambda *args, **kwargs: "mock_output"
36+
return llm
37+
38+
39+
@pytest.fixture
40+
def memory_manager(mock_graph_store, mock_embedder, mock_llm):
41+
return MemoryManager(
42+
graph_store=mock_graph_store,
43+
embedder=mock_embedder,
44+
llm=mock_llm,
45+
)
3546

3647

3748
def test_add_and_replace_working_memory(memory_manager):

0 commit comments

Comments
 (0)