Method process_preference_memory

src/memos/mem_os/core.py:796–829

()

                            self.mem_scheduler.submit_messages(messages=[message_item])

        def process_preference_memory():
            if (
                (messages is not None)
                and self.config.enable_preference_memory
                and self.mem_cubes[mem_cube_id].pref_mem
            ):
                messages_list = [messages]
                if sync_mode == "sync":
                    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,
                            "mem_cube_id": mem_cube_id,
                        },
                    )
                    pref_ids = self.mem_cubes[mem_cube_id].pref_mem.add(pref_memories)
                    logger.info(
                        f"Added preferences user {target_user_id} to memcube {mem_cube_id}: {pref_ids}"
                    )
                elif sync_mode == "async":
                    assert self.mem_scheduler is not None, (
                        "Mem-Scheduler must be working when use asynchronous memory adding."
                    )
                    message_item = ScheduleMessageItem(
                        user_id=target_user_id,
                        session_id=target_session_id,
                        mem_cube_id=mem_cube_id,
                        label=PREF_ADD_TASK_LABEL,
                        content=json.dumps(messages_list),
                        timestamp=datetime.utcnow(),
                    )
                    self.mem_scheduler.submit_messages(messages=[message_item])

        # Execute both memory processing functions in parallel
        with ContextThreadPoolExecutor(max_workers=2) as executor: