Skip to content

Commit 445f7b9

Browse files
authored
Scheduler: fix bugs in log (#746)
* fix bugs: try to fix bugs in _submit_web_logs * fix bugs: try to address bugs
1 parent 35a69b0 commit 445f7b9

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

src/memos/mem_scheduler/base_scheduler.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -842,12 +842,7 @@ def _submit_web_logs(
842842
messages = [messages] # transform single message to list
843843

844844
for message in messages:
845-
logger.info(
846-
f"[DIAGNOSTIC] base_scheduler._submit_web_logs called. Message to publish: {message.model_dump_json(indent=2)}"
847-
)
848-
849-
try:
850-
for message in messages:
845+
try:
851846
# Always call publish; the publisher now caches when offline and flushes after reconnect
852847
logger.info(
853848
f"[DIAGNOSTIC] base_scheduler._submit_web_logs: enqueue publish {message.model_dump_json(indent=2)}"
@@ -860,8 +855,10 @@ def _submit_web_logs(
860855
message.task_id,
861856
message.label,
862857
)
863-
except Exception as e:
864-
logger.error(f"[DIAGNOSTIC] base_scheduler._submit_web_logs failed: {e}", exc_info=True)
858+
except Exception as e:
859+
logger.error(
860+
f"[DIAGNOSTIC] base_scheduler._submit_web_logs failed: {e}", exc_info=True
861+
)
865862

866863
logger.debug(
867864
f"{len(messages)} submitted. {self._web_log_message_queue.qsize()} in queue. additional_log_info: {additional_log_info}"

src/memos/mem_scheduler/webservice_modules/rabbitmq_service.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77
from pathlib import Path
88
from queue import Empty
99

10+
from pyglet.libs.win32.constants import FALSE
11+
1012
from memos.configs.mem_scheduler import AuthConfig, RabbitMQConfig
1113
from memos.context.context import ContextThread
1214
from memos.dependency import require_python_package
@@ -325,14 +327,14 @@ def rabbitmq_publish_message(self, message: dict):
325327
f"[DIAGNOSTIC] Publishing {label} message in Cloud Env. "
326328
f"Exchange: {exchange_name}, Routing Key: '{routing_key}'."
327329
)
328-
logger.info(f" - Message Content: {json.dumps(message, indent=2)}")
330+
logger.info(f" - Message Content: {json.dumps(message, indent=2, ensure_ascii=FALSE)}")
329331
elif label == "knowledgeBaseUpdate":
330332
# Original diagnostic logging for knowledgeBaseUpdate if NOT in cloud env
331333
logger.info(
332334
f"[DIAGNOSTIC] Publishing knowledgeBaseUpdate message (Local Env). "
333335
f"Current configured Exchange: {exchange_name}, Routing Key: '{routing_key}'."
334336
)
335-
logger.info(f" - Message Content: {json.dumps(message, indent=2)}")
337+
logger.info(f" - Message Content: {json.dumps(message, indent=2, ensure_ascii=FALSE)}")
336338

337339
with self._rabbitmq_lock:
338340
logger.info(

0 commit comments

Comments
 (0)