From 162ddd0b138b83521e81981ea6a41b9246708663 Mon Sep 17 00:00:00 2001 From: "yuan.wang" Date: Fri, 14 Nov 2025 17:19:11 +0800 Subject: [PATCH] fix lack mem_cube_id bug in pref async --- src/memos/mem_os/core.py | 6 +++++- src/memos/mem_scheduler/general_scheduler.py | 4 +++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/src/memos/mem_os/core.py b/src/memos/mem_os/core.py index 1b6d4e126..3b53cef1a 100644 --- a/src/memos/mem_os/core.py +++ b/src/memos/mem_os/core.py @@ -796,7 +796,11 @@ def process_preference_memory(): pref_memories = self.mem_cubes[mem_cube_id].pref_mem.get_memory( messages_list, type="chat", - info={"user_id": target_user_id, "session_id": self.session_id}, + info={ + "user_id": target_user_id, + "session_id": self.session_id, + "mem_cube_id": mem_cube_id, + }, ) pref_ids = self.mem_cubes[mem_cube_id].pref_mem.add(pref_memories) logger.info( diff --git a/src/memos/mem_scheduler/general_scheduler.py b/src/memos/mem_scheduler/general_scheduler.py index 2b14887d6..6e916962e 100644 --- a/src/memos/mem_scheduler/general_scheduler.py +++ b/src/memos/mem_scheduler/general_scheduler.py @@ -521,7 +521,9 @@ def process_message(message: ScheduleMessageItem): # Use pref_mem.get_memory to process the memories pref_memories = pref_mem.get_memory( - messages_list, type="chat", info={"user_id": user_id, "session_id": session_id} + messages_list, + type="chat", + info={"user_id": user_id, "session_id": session_id, "mem_cube_id": mem_cube_id}, ) # Add pref_mem to vector db pref_ids = pref_mem.add(pref_memories)