fix(ci): Resolve ruff linting and style errors

Addresses all linting errors reported by ruff, including undefined names for 'os' and 'timezone' by correcting import statements in 'dispatcher.py' and 'status_tracker.py'. Also resolves various code style violations (line breaks, comment punctuation) to align with ruff standards. This ensures the code adheres to project standards and passes the CI quality checks.

Author: [email protected]

examples/mem_scheduler/api_w_scheduler.py

@@ -6,7 +6,6 @@
 )
 from memos.api.routers.server_router import mem_scheduler, status_tracker
 from memos.mem_scheduler.schemas.message_schemas import ScheduleMessageItem
-from memos.api.product_models import StatusRequest
 
 
 # Debug: Print scheduler configuration

src/memos/api/routers/server_router.py

@@ -23,8 +23,6 @@
 from memos.api.handlers.chat_handler import ChatHandler
 from memos.api.handlers.search_handler import SearchHandler
 from memos.api.product_models import (
-    AddStatusRequest,
-    AddStatusResponse,
     APIADDRequest,
     APIChatCompleteRequest,
     APISearchRequest,
@@ -36,10 +34,9 @@
     GetMemoryResponse,
     MemoryResponse,
     SearchResponse,
+    StatusResponse,
     SuggestionRequest,
     SuggestionResponse,
-    StatusRequest,
-    StatusResponse,
 )
 from memos.log import get_logger
 from memos.mem_scheduler.base_scheduler import BaseScheduler

src/memos/mem_scheduler/base_scheduler.py

@@ -4,7 +4,7 @@
 import time
 
 from collections.abc import Callable
-from datetime import datetime, timezone
+from datetime import datetime
 from pathlib import Path
 from typing import TYPE_CHECKING, Union
 
@@ -44,10 +44,12 @@
 from memos.mem_scheduler.task_schedule_modules.local_queue import SchedulerLocalQueue
 from memos.mem_scheduler.task_schedule_modules.redis_queue import SchedulerRedisQueue
 from memos.mem_scheduler.task_schedule_modules.task_queue import ScheduleTaskQueue
+from memos.mem_scheduler.utils import metrics
 from memos.mem_scheduler.utils.db_utils import get_utc_now
 from memos.mem_scheduler.utils.filter_utils import (
     transform_name_to_key,
 )
+from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker
 from memos.mem_scheduler.webservice_modules.rabbitmq_service import RabbitMQSchedulerModule
 from memos.mem_scheduler.webservice_modules.redis_service import RedisSchedulerModule
 from memos.memories.activation.kv import KVCacheMemory
@@ -59,14 +61,13 @@
     MemCubeID,
     UserID,
 )
-from memos.mem_scheduler.utils import metrics
-from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker
 
 
 if TYPE_CHECKING:
-    from memos.reranker.http_bge import HTTPBGEReranker
     import redis
 
+    from memos.reranker.http_bge import HTTPBGEReranker
+
 
 logger = get_logger(__name__)
 

src/memos/mem_scheduler/task_schedule_modules/dispatcher.py

@@ -1,4 +1,5 @@
 import concurrent
+import os
 import threading
 import time
 
@@ -11,11 +12,12 @@
 from memos.log import get_logger
 from memos.mem_scheduler.general_modules.base import BaseSchedulerModule
 from memos.mem_scheduler.general_modules.task_threads import ThreadManager
-from memos.mem_scheduler.schemas.general_schemas import DEFAULT_STOP_WAIT
-from memos.mem_scheduler.schemas.message_schemas import ScheduleMessageItem, ScheduleLogForWebItem
+from memos.mem_scheduler.schemas.general_schemas import (
+    DEFAULT_STOP_WAIT,
+)
+from memos.mem_scheduler.schemas.message_schemas import ScheduleLogForWebItem, ScheduleMessageItem
 from memos.mem_scheduler.schemas.task_schemas import RunningTaskItem
 from memos.mem_scheduler.utils.misc_utils import group_messages_by_user_and_mem_cube
-from memos.mem_scheduler.utils import metrics
 from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker
 
 
@@ -102,7 +104,6 @@ def on_messages_enqueued(self, msgs: list[ScheduleMessageItem]) -> None:
         if not msgs:
             return
         # This is handled in BaseScheduler now
-        pass
 
     def _create_task_wrapper(self, handler: Callable, task_item: RunningTaskItem):
         """
@@ -210,7 +211,7 @@ def wrapped_handler(messages: list[ScheduleMessageItem]):
                         mem_cube_id=task_item.mem_cube_id,
                         item_id=task_item.item_id,
                         label=m.label,
-                        log_content=f"Task {task_item.item_id} failed for user {task_item.user_id} with error: {str(e)}.",
+                        log_content=f"Task {task_item.item_id} failed for user {task_item.user_id} with error: {e!s}.",
                         status="failed",
                         exception=str(e)
                     )

src/memos/mem_scheduler/utils/metrics.py

@@ -1,7 +1,10 @@
 # src/memos/mem_scheduler/utils/metrics.py
 import time
+
 from contextlib import ContextDecorator
-from prometheus_client import Counter, Summary, Gauge, Histogram
+
+from prometheus_client import Counter, Gauge, Histogram, Summary
+
 
 # --- Metric Definitions ---
 

src/memos/mem_scheduler/utils/status_tracker.py

@@ -1,7 +1,7 @@
 # src/memos/mem_scheduler/utils/status_tracker.py
 import json
 
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from typing import TYPE_CHECKING
 
 from memos.dependency import require_python_package