Skip to content

Commit 647b4c0

Browse files
Add cloud add-log handler fallback for schedulers
1 parent 3311832 commit 647b4c0

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

src/memos/mem_scheduler/general_scheduler.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -464,6 +464,25 @@ def send_add_log_messages_to_local_env(
464464
if events:
465465
self._submit_web_logs(events, additional_log_info="send_add_log_messages_to_cloud_env")
466466

467+
def send_add_log_messages_to_cloud_env(
468+
self, msg: ScheduleMessageItem, prepared_add_items, prepared_update_items_with_original
469+
):
470+
"""
471+
Cloud logging path for add/update events.
472+
473+
Currently reuses local env logging to avoid missing method errors in subclasses.
474+
"""
475+
logger.info(
476+
"send_add_log_messages_to_cloud_env fallback to local handler. user_id=%s mem_cube_id=%s task_id=%s item_id=%s",
477+
msg.user_id,
478+
msg.mem_cube_id,
479+
msg.task_id,
480+
msg.item_id,
481+
)
482+
return self.send_add_log_messages_to_local_env(
483+
msg, prepared_add_items, prepared_update_items_with_original
484+
)
485+
467486
def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
468487
logger.info(f"Messages {messages} assigned to {ADD_LABEL} handler.")
469488
# Process the query in a session turn

0 commit comments

Comments
 (0)