refactor: optimize memory update

tangg555 · tangg555 · commit f63470d1644c · 2025-12-17T21:00:26.000+08:00
diff --git a/src/memos/mem_scheduler/general_scheduler.py b/src/memos/mem_scheduler/general_scheduler.py
@@ -1379,6 +1379,7 @@ def process_session_turn(
         cur_working_memory: list[TextualMemoryItem] = text_mem_base.get_working_memory(
             user_name=mem_cube_id
         )
+        cur_working_memory = cur_working_memory[:top_k]
         text_working_memory: list[str] = [w_m.memory for w_m in cur_working_memory]
         intent_result = self.monitor.detect_intent(
             q_list=queries, text_working_memory=text_working_memory

Original file line number	Diff line number	Diff line change
`@@ -1379,6 +1379,7 @@ def process_session_turn(`
`1379`	`1379`	`cur_working_memory: list[TextualMemoryItem] = text_mem_base.get_working_memory(`
`1380`	`1380`	`user_name=mem_cube_id`
`1381`	`1381`	`)`
	`1382`	`+ cur_working_memory = cur_working_memory[:top_k]`
`1382`	`1383`	`text_working_memory: list[str] = [w_m.memory for w_m in cur_working_memory]`
`1383`	`1384`	`intent_result = self.monitor.detect_intent(`
`1384`	`1385`	`q_list=queries, text_working_memory=text_working_memory`