Skip to content

Commit 882c5d3

Browse files
fix(scheduler): Correct misleading logs in reorganize consumer
The and its helper function contained several logging errors due to being copied from the 'mem_read' consumer without modification. This commit corrects the following: - The handler info log now correctly uses the . - All log messages now refer to 'mem_reorganize' instead of 'mem_read'. - Exception logs now correctly cite and as the source of the error. These changes ensure that the logs for the reorganize functionality are accurate and not misleading, which is critical for monitoring and debugging. The core business logic of the function, which appears to be missing, has not been altered.
1 parent 4a68c3c commit 882c5d3

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/memos/mem_scheduler/general_scheduler.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -636,7 +636,7 @@ def _process_memories_with_reader(
636636
)
637637

638638
def _mem_reorganize_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
639-
logger.info(f"Messages {messages} assigned to {MEM_READ_LABEL} handler.")
639+
logger.info(f"Messages {messages} assigned to {MEM_ORGANIZE_LABEL} handler.")
640640

641641
def process_message(message: ScheduleMessageItem):
642642
try:
@@ -652,7 +652,7 @@ def process_message(message: ScheduleMessageItem):
652652
return
653653

654654
logger.info(
655-
f"Processing mem_read for user_id={user_id}, mem_cube_id={mem_cube_id}, mem_ids={mem_ids}"
655+
f"Processing mem_reorganize for user_id={user_id}, mem_cube_id={mem_cube_id}, mem_ids={mem_ids}"
656656
)
657657

658658
# Get the text memory from the mem_cube
@@ -795,11 +795,11 @@ def process_message(message: ScheduleMessageItem):
795795
self._submit_web_logs([event])
796796

797797
logger.info(
798-
f"Successfully processed mem_read for user_id={user_id}, mem_cube_id={mem_cube_id}"
798+
f"Successfully processed mem_reorganize for user_id={user_id}, mem_cube_id={mem_cube_id}"
799799
)
800800

801801
except Exception as e:
802-
logger.error(f"Error processing mem_read message: {e}", exc_info=True)
802+
logger.error(f"Error processing mem_reorganize message: {e}", exc_info=True)
803803

804804
with ContextThreadPoolExecutor(max_workers=min(8, len(messages))) as executor:
805805
futures = [executor.submit(process_message, msg) for msg in messages]
@@ -858,7 +858,7 @@ def _process_memories_with_reorganize(
858858

859859
except Exception:
860860
logger.error(
861-
f"Error in _process_memories_with_reader: {traceback.format_exc()}", exc_info=True
861+
f"Error in _process_memories_with_reorganize: {traceback.format_exc()}", exc_info=True
862862
)
863863

864864
def _pref_add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:

0 commit comments

Comments
 (0)