Skip to content

Commit 119bbe2

Browse files
authored
feat: freeze usage update in Searcher (#466)
* feat: tmp-close usage update, maybe reopen it later * test: withdraw update usage test
1 parent f640855 commit 119bbe2

File tree

2 files changed

+2
-11
lines changed

2 files changed

+2
-11
lines changed

src/memos/memories/textual/tree_text_memory/retrieve/searcher.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,5 @@
1-
import json
21
import traceback
32

4-
from datetime import datetime
5-
63
from memos.context.context import ContextThreadPoolExecutor
74
from memos.embedders.factory import OllamaEmbedder
85
from memos.graph_dbs.factory import Neo4jGraphDB
@@ -508,7 +505,7 @@ def _sort_and_trim(self, results, top_k):
508505

509506
@timed
510507
def _update_usage_history(self, items, info, user_name: str | None = None):
511-
"""Update usage history in graph DB"""
508+
"""Update usage history in graph DB
512509
now_time = datetime.now().isoformat()
513510
info_copy = dict(info or {})
514511
info_copy.pop("chat_history", None)
@@ -532,6 +529,7 @@ def _update_usage_history(self, items, info, user_name: str | None = None):
532529
self._usage_executor.submit(
533530
self._update_usage_history_worker, payload, usage_record, user_name
534531
)
532+
"""
535533

536534
def _update_usage_history_worker(
537535
self, payload, usage_record: str, user_name: str | None = None

tests/memories/textual/test_tree_searcher.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -69,13 +69,6 @@ def test_searcher_fast_path(mock_searcher):
6969
assert len(result) <= 2
7070
assert all(isinstance(item, TextualMemoryItem) for item in result)
7171

72-
# Should update usage and call update_node
73-
for item in result:
74-
assert len(item.metadata.usage) > 0
75-
mock_searcher.graph_store.update_node.assert_any_call(
76-
item.id, {"usage": item.metadata.usage}, user_name=None
77-
)
78-
7972

8073
def test_searcher_fine_mode_triggers_reasoner(mock_searcher):
8174
parsed_goal = MagicMock()

0 commit comments

Comments
 (0)