hotfix: align KB knowledgeBaseUpdate logs with spec (#555)

* hotfix: align KB knowledgeBaseUpdate logs with spec

* style: apply ruff format to general_scheduler

---------

Co-authored-by: [email protected] <>

Author: glin93

src/memos/mem_scheduler/general_scheduler.py

@@ -367,7 +367,8 @@ def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
                             if kb_log_content:
                                 event = self.create_event_log(
                                     label="knowledgeBaseUpdate",
-                                    log_content=f"Knowledge Base Memory Update: {len(kb_log_content)} changes.",
+                                    from_memory_type=USER_INPUT_TYPE,
+                                    to_memory_type=LONG_TERM_MEMORY_TYPE,
                                     user_id=msg.user_id,
                                     mem_cube_id=msg.mem_cube_id,
                                     mem_cube=self.current_mem_cube,
@@ -376,6 +377,9 @@ def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
                                     memory_len=len(kb_log_content),
                                     memcube_name=self._map_memcube_name(msg.mem_cube_id),
                                 )
+                                event.log_content = (
+                                    f"Knowledge Base Memory Update: {len(kb_log_content)} changes."
+                                )
                                 event.task_id = msg.task_id
                                 self._submit_web_logs([event])
                         else:
@@ -633,7 +637,8 @@ def _process_memories_with_reader(
                         if kb_log_content:
                             event = self.create_event_log(
                                 label="knowledgeBaseUpdate",
-                                log_content=f"Knowledge Base Memory Update: {len(kb_log_content)} changes.",
+                                from_memory_type=USER_INPUT_TYPE,
+                                to_memory_type=LONG_TERM_MEMORY_TYPE,
                                 user_id=user_id,
                                 mem_cube_id=mem_cube_id,
                                 mem_cube=self.current_mem_cube,
@@ -642,6 +647,9 @@ def _process_memories_with_reader(
                                 memory_len=len(kb_log_content),
                                 memcube_name=self._map_memcube_name(mem_cube_id),
                             )
+                            event.log_content = (
+                                f"Knowledge Base Memory Update: {len(kb_log_content)} changes."
+                            )
                             event.task_id = task_id
                             self._submit_web_logs([event])
                     else:
@@ -742,7 +750,8 @@ def _process_memories_with_reader(
                         ]
                     event = self.create_event_log(
                         label="knowledgeBaseUpdate",
-                        log_content=f"Knowledge Base Memory Update failed: {exc!s}",
+                        from_memory_type=USER_INPUT_TYPE,
+                        to_memory_type=LONG_TERM_MEMORY_TYPE,
                         user_id=user_id,
                         mem_cube_id=mem_cube_id,
                         mem_cube=self.current_mem_cube,
@@ -751,6 +760,7 @@ def _process_memories_with_reader(
                         memory_len=len(kb_log_content),
                         memcube_name=self._map_memcube_name(mem_cube_id),
                     )
+                    event.log_content = f"Knowledge Base Memory Update failed: {exc!s}"
                     event.task_id = task_id
                     event.status = "failed"
                     self._submit_web_logs([event])