Skip to content

Commit 11277b4

Browse files
committed
test: fix researcher test script
1 parent fa2d3e6 commit 11277b4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/memories/textual/test_tree_searcher.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
from memos.memories.textual.item import TextualMemoryItem, TreeNodeTextualMemoryMetadata
66
from memos.memories.textual.tree_text_memory.retrieve.searcher import Searcher
7+
from memos.reranker.base import BaseReranker
78

89

910
@pytest.fixture
@@ -12,12 +13,12 @@ def mock_searcher():
1213
graph_store = MagicMock()
1314
embedder = MagicMock()
1415

15-
s = Searcher(dispatcher_llm, graph_store, embedder)
16+
reranker = MagicMock(spec=BaseReranker)
17+
s = Searcher(dispatcher_llm, graph_store, embedder, reranker)
1618

1719
# Mock internals
1820
s.task_goal_parser = MagicMock()
1921
s.graph_retriever = MagicMock()
20-
s.reranker = MagicMock()
2122
s.reasoner = MagicMock()
2223

2324
return s

0 commit comments

Comments
 (0)