Skip to content

Commit 5850d7a

Browse files
author
glin1993@outlook.com
committed
fix: Resolve Ruff linting and formatting issues
1 parent 30109f6 commit 5850d7a

File tree

4 files changed

+17
-10
lines changed

4 files changed

+17
-10
lines changed

src/memos/mem_scheduler/base_scheduler.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -773,7 +773,7 @@ def _message_consumer(self) -> None:
773773
for msg in messages:
774774
enqueue_ts_obj = getattr(msg, "timestamp", None)
775775
enqueue_epoch = None
776-
if isinstance(enqueue_ts_obj, (int, float)):
776+
if isinstance(enqueue_ts_obj, int | float):
777777
enqueue_epoch = float(enqueue_ts_obj)
778778
elif hasattr(enqueue_ts_obj, "timestamp"):
779779
dt = enqueue_ts_obj
@@ -785,13 +785,15 @@ def _message_consumer(self) -> None:
785785
if enqueue_epoch is not None:
786786
queue_wait_ms = max(0.0, now - enqueue_epoch) * 1000
787787

788-
setattr(msg, "dequeue_ts", now)
788+
msg.dequeue_ts = now
789789
emit_monitor_event(
790790
"dequeue",
791791
msg,
792792
{
793793
"enqueue_ts": to_iso(enqueue_ts_obj),
794-
"dequeue_ts": datetime.fromtimestamp(now, tz=timezone.utc).isoformat(),
794+
"dequeue_ts": datetime.fromtimestamp(
795+
now, tz=timezone.utc
796+
).isoformat(),
795797
"queue_wait_ms": queue_wait_ms,
796798
},
797799
)

src/memos/mem_scheduler/task_schedule_modules/dispatcher.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
from collections import defaultdict
66
from collections.abc import Callable
7-
from datetime import timezone
7+
from datetime import datetime, timezone
88
from typing import Any
99

1010
from memos.context.context import (
@@ -168,7 +168,7 @@ def wrapped_handler(messages: list[ScheduleMessageItem]):
168168

169169
dequeue_ts = getattr(first_msg, "dequeue_ts", None)
170170
start_delay_ms = None
171-
if isinstance(dequeue_ts, (int, float)):
171+
if isinstance(dequeue_ts, int | float):
172172
start_delay_ms = max(0.0, start_time - dequeue_ts) * 1000
173173

174174
emit_monitor_event(
@@ -180,7 +180,7 @@ def wrapped_handler(messages: list[ScheduleMessageItem]):
180180
"enqueue_ts": to_iso(enq_ts),
181181
"dequeue_ts": to_iso(
182182
datetime.fromtimestamp(dequeue_ts, tz=timezone.utc)
183-
if isinstance(dequeue_ts, (int, float))
183+
if isinstance(dequeue_ts, int | float)
184184
else None
185185
),
186186
},
@@ -205,7 +205,9 @@ def wrapped_handler(messages: list[ScheduleMessageItem]):
205205
{
206206
"status": "ok",
207207
"start_ts": start_iso,
208-
"finish_ts": datetime.fromtimestamp(finish_time, tz=timezone.utc).isoformat(),
208+
"finish_ts": datetime.fromtimestamp(
209+
finish_time, tz=timezone.utc
210+
).isoformat(),
209211
"exec_duration_ms": duration * 1000,
210212
},
211213
)
@@ -233,7 +235,9 @@ def wrapped_handler(messages: list[ScheduleMessageItem]):
233235
{
234236
"status": "fail",
235237
"start_ts": start_iso,
236-
"finish_ts": datetime.fromtimestamp(finish_time, tz=timezone.utc).isoformat(),
238+
"finish_ts": datetime.fromtimestamp(
239+
finish_time, tz=timezone.utc
240+
).isoformat(),
237241
"exec_duration_ms": (finish_time - start_time) * 1000,
238242
"error_type": type(e).__name__,
239243
"error_msg": str(e),

src/memos/mem_scheduler/task_schedule_modules/task_queue.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,14 @@
55
the local memos_message_queue functionality in BaseScheduler.
66
"""
77

8+
from memos.context.context import get_current_trace_id
89
from memos.log import get_logger
910
from memos.mem_scheduler.schemas.message_schemas import ScheduleMessageItem
10-
from memos.context.context import get_current_trace_id
1111
from memos.mem_scheduler.task_schedule_modules.local_queue import SchedulerLocalQueue
1212
from memos.mem_scheduler.task_schedule_modules.redis_queue import SchedulerRedisQueue
1313
from memos.mem_scheduler.utils.db_utils import get_utc_now
14-
from memos.mem_scheduler.utils.monitor_event_utils import emit_monitor_event, to_iso
1514
from memos.mem_scheduler.utils.misc_utils import group_messages_by_user_and_mem_cube
15+
from memos.mem_scheduler.utils.monitor_event_utils import emit_monitor_event, to_iso
1616

1717

1818
logger = get_logger(__name__)

src/memos/mem_scheduler/utils/monitor_event_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import json
22
import os
33
import socket
4+
45
from datetime import datetime, timezone
56
from typing import Any
67

0 commit comments

Comments
 (0)