Skip to content

Commit 4e65f58

Browse files
committed
fix bugs: addresss some bugs
1 parent b56f57a commit 4e65f58

File tree

2 files changed

+1
-37
lines changed

2 files changed

+1
-37
lines changed

src/memos/mem_scheduler/general_scheduler.py

Lines changed: 0 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -562,42 +562,6 @@ def send_add_log_messages_to_cloud_env(
562562
event.task_id = msg.task_id
563563
self._submit_web_logs([event])
564564

565-
def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
566-
logger.info(f"Messages {messages} assigned to {ADD_TASK_LABEL} handler.")
567-
# Process the query in a session turn
568-
grouped_messages = group_messages_by_user_and_mem_cube(messages=messages)
569-
570-
self.validate_schedule_messages(messages=messages, label=ADD_TASK_LABEL)
571-
try:
572-
for user_id in grouped_messages:
573-
for mem_cube_id in grouped_messages[user_id]:
574-
batch = grouped_messages[user_id][mem_cube_id]
575-
if not batch:
576-
continue
577-
578-
# Process each message in the batch
579-
for msg in batch:
580-
prepared_add_items, prepared_update_items_with_original = (
581-
self.log_add_messages(msg=msg)
582-
)
583-
# Conditional Logging: Knowledge Base (Cloud Service) vs. Playground/Default
584-
is_cloud_env = (
585-
os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME")
586-
== "memos-memory-change"
587-
)
588-
589-
if is_cloud_env:
590-
self.send_add_log_messages_to_cloud_env(
591-
msg, prepared_add_items, prepared_update_items_with_original
592-
)
593-
else:
594-
self.send_add_log_messages_to_local_env(
595-
msg, prepared_add_items, prepared_update_items_with_original
596-
)
597-
598-
except Exception as e:
599-
logger.error(f"Error: {e}", exc_info=True)
600-
601565
def _mem_feedback_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
602566
try:
603567
if not messages:

src/memos/mem_scheduler/task_schedule_modules/redis_queue.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ def empty(self) -> bool:
491491
def full(self) -> bool:
492492
if self.max_len is None:
493493
return False
494-
return self.size() >= self.maxsize
494+
return self.size() >= self.max_len
495495

496496
def join(self) -> None:
497497
"""

0 commit comments

Comments
 (0)