Skip to content

Commit 6dd0432

Browse files
Drop scheduler messages using default user_name
1 parent a9a8406 commit 6dd0432

File tree

1 file changed

+29
-16
lines changed

1 file changed

+29
-16
lines changed

src/memos/mem_scheduler/general_scheduler.py

Lines changed: 29 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -168,9 +168,12 @@ def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
168168

169169
# Process each message in the batch
170170
for msg in batch:
171-
if not (msg.user_name or "").strip():
171+
if (
172+
not (msg.user_name or "").strip()
173+
or (msg.user_name or "").strip() == "memosdefault"
174+
):
172175
logger.warning(
173-
"[AddConsumer] Dropping message with empty user_name; user_id=%s mem_cube_id=%s content=%s",
176+
"[AddConsumer] Dropping message with empty or default user_name; user_id=%s mem_cube_id=%s content=%s",
174177
msg.user_id,
175178
msg.mem_cube_id,
176179
msg.content,
@@ -220,9 +223,12 @@ def _query_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
220223
continue
221224
try:
222225
for msg in batch:
223-
if not (msg.user_name or "").strip():
226+
if (
227+
not (msg.user_name or "").strip()
228+
or (msg.user_name or "").strip() == "memosdefault"
229+
):
224230
logger.warning(
225-
"[QueryConsumer] Dropping message with empty user_name; user_id=%s mem_cube_id=%s content=%s",
231+
"[QueryConsumer] Dropping message with empty or default user_name; user_id=%s mem_cube_id=%s content=%s",
226232
msg.user_id,
227233
msg.mem_cube_id,
228234
msg.content,
@@ -273,9 +279,12 @@ def _answer_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
273279
continue
274280
try:
275281
for msg in batch:
276-
if not (msg.user_name or "").strip():
282+
if (
283+
not (msg.user_name or "").strip()
284+
or (msg.user_name or "").strip() == "memosdefault"
285+
):
277286
logger.warning(
278-
"[AnswerConsumer] Dropping message with empty user_name; user_id=%s mem_cube_id=%s content=%s",
287+
"[AnswerConsumer] Dropping message with empty or default user_name; user_id=%s mem_cube_id=%s content=%s",
279288
msg.user_id,
280289
msg.mem_cube_id,
281290
msg.content,
@@ -560,9 +569,12 @@ def _add_message_consumer(self, messages: list[ScheduleMessageItem]) -> None:
560569

561570
# Process each message in the batch
562571
for msg in batch:
563-
if not (msg.user_name or "").strip():
572+
if (
573+
not (msg.user_name or "").strip()
574+
or (msg.user_name or "").strip() == "memosdefault"
575+
):
564576
logger.warning(
565-
"[AddConsumer] Dropping message with empty user_name; user_id=%s mem_cube_id=%s content=%s",
577+
"[AddConsumer] Dropping message with empty or default user_name; user_id=%s mem_cube_id=%s content=%s",
566578
msg.user_id,
567579
msg.mem_cube_id,
568580
msg.content,
@@ -769,10 +781,10 @@ def process_message(message: ScheduleMessageItem):
769781
return
770782

771783
content = message.content
772-
user_name = message.user_name or ""
773-
if not user_name:
784+
user_name = (message.user_name or "").strip()
785+
if not user_name or user_name == "memosdefault":
774786
logger.warning(
775-
"[MemRead] Dropping message with empty user_name; user_id=%s mem_cube_id=%s mem_ids=%s",
787+
"[MemRead] Dropping message with empty or default user_name; user_id=%s mem_cube_id=%s mem_ids=%s",
776788
user_id,
777789
mem_cube_id,
778790
content,
@@ -1080,10 +1092,10 @@ def process_message(message: ScheduleMessageItem):
10801092
)
10811093
return
10821094
content = message.content
1083-
user_name = message.user_name or ""
1084-
if not user_name:
1095+
user_name = (message.user_name or "").strip()
1096+
if not user_name or user_name == "memosdefault":
10851097
logger.warning(
1086-
"[MemReorganize] Dropping message with empty user_name; user_id=%s mem_cube_id=%s mem_ids=%s",
1098+
"[MemReorganize] Dropping message with empty or default user_name; user_id=%s mem_cube_id=%s mem_ids=%s",
10871099
user_id,
10881100
mem_cube_id,
10891101
content,
@@ -1311,9 +1323,10 @@ def _pref_add_message_consumer(self, messages: list[ScheduleMessageItem]) -> Non
13111323

13121324
def process_message(message: ScheduleMessageItem):
13131325
try:
1314-
if not (message.user_name or "").strip():
1326+
user_name = (message.user_name or "").strip()
1327+
if not user_name or user_name == "memosdefault":
13151328
logger.warning(
1316-
"[PrefAdd] Dropping message with empty user_name; user_id=%s mem_cube_id=%s content=%s",
1329+
"[PrefAdd] Dropping message with empty or default user_name; user_id=%s mem_cube_id=%s content=%s",
13171330
message.user_id,
13181331
message.mem_cube_id,
13191332
message.content,

0 commit comments

Comments
 (0)