@@ -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 :
0 commit comments