From 1c82312c294a0081a5e9cab117c40603d4aa7529 Mon Sep 17 00:00:00 2001
From: "glin1993@outlook.com" <>
Date: Mon, 15 Dec 2025 12:16:29 +0800
Subject: [PATCH] feat: Relax cloud env check to support any non-empty
 MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME

---
 src/memos/mem_scheduler/general_scheduler.py      | 15 ++++-----------
 .../task_schedule_modules/dispatcher.py           |  2 +-
 2 files changed, 5 insertions(+), 12 deletions(-)

diff --git a/src/memos/mem_scheduler/general_scheduler.py b/src/memos/mem_scheduler/general_scheduler.py
index bbcb2c379..5626b2c91 100644
--- a/src/memos/mem_scheduler/general_scheduler.py
+++ b/src/memos/mem_scheduler/general_scheduler.py
@@ -194,10 +194,7 @@ def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
                             f"prepared_add_items: {prepared_add_items};\n prepared_update_items_with_original: {prepared_update_items_with_original}"
                         )
                         # Conditional Logging: Knowledge Base (Cloud Service) vs. Playground/Default
-                        is_cloud_env = (
-                            os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME")
-                            == "memos-memory-change"
-                        )
+                        is_cloud_env = os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME")
 
                         if is_cloud_env:
                             self.send_add_log_messages_to_cloud_env(
@@ -618,7 +615,7 @@ def _mem_feedback_message_consumer(self, messages: list[ScheduleMessageItem]) ->
                 f"Successfully processed feedback for user_id={user_id}, mem_cube_id={mem_cube_id}"
             )
 
-            is_cloud_env = os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME") == "memos-memory-change"
+            is_cloud_env = os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME")
             if is_cloud_env:
                 record = feedback_result.get("record") if isinstance(feedback_result, dict) else {}
                 add_records = record.get("add") if isinstance(record, dict) else []
@@ -896,9 +893,7 @@ def _process_memories_with_reader(
 
                     # LOGGING BLOCK START
                     # This block is replicated from _add_message_consumer to ensure consistent logging
-                    is_cloud_env = (
-                        os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME") == "memos-memory-change"
-                    )
+                    is_cloud_env = os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME")
                     if is_cloud_env:
                         # New: Knowledge Base Logging (Cloud Service)
                         kb_log_content = []
@@ -1018,9 +1013,7 @@ def _process_memories_with_reader(
                 f"Error in _process_memories_with_reader: {traceback.format_exc()}", exc_info=True
             )
             with contextlib.suppress(Exception):
-                is_cloud_env = (
-                    os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME") == "memos-memory-change"
-                )
+                is_cloud_env = os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME")
                 if is_cloud_env:
                     if not kb_log_content:
                         trigger_source = (
diff --git a/src/memos/mem_scheduler/task_schedule_modules/dispatcher.py b/src/memos/mem_scheduler/task_schedule_modules/dispatcher.py
index 729345dc5..10d08a532 100644
--- a/src/memos/mem_scheduler/task_schedule_modules/dispatcher.py
+++ b/src/memos/mem_scheduler/task_schedule_modules/dispatcher.py
@@ -351,7 +351,7 @@ def _maybe_emit_task_completion(
         mem_cube_id = first.mem_cube_id
 
         try:
-            is_cloud_env = os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME") == "memos-memory-change"
+            is_cloud_env = os.getenv("MEMSCHEDULER_RABBITMQ_EXCHANGE_NAME")
             if not is_cloud_env:
                 return