fix bugs: address bugs caused by outdated test code

Author: tangg555

src/memos/mem_scheduler/base_scheduler.py

@@ -137,7 +137,6 @@ def __init__(self, config: BaseSchedulerConfig):
         self.dispatcher = SchedulerDispatcher(
             config=self.config,
             memos_message_queue=self.memos_message_queue,
-            use_redis_queue=self.use_redis_queue,
             max_workers=self.thread_pool_max_workers,
             enable_parallel_dispatch=self.enable_parallel_dispatch,
             status_tracker=self.status_tracker,

src/memos/mem_scheduler/task_schedule_modules/dispatcher.py

@@ -43,7 +43,6 @@ def __init__(
         self,
         max_workers: int = 30,
         memos_message_queue: ScheduleTaskQueue | None = None,
-        use_redis_queue: bool | None = None,
         enable_parallel_dispatch: bool = True,
         config=None,
         status_tracker: TaskStatusTracker | None = None,
@@ -56,8 +55,12 @@ def __init__(
         # Main dispatcher thread pool
         self.max_workers = max_workers
 
-        self.memos_message_queue = memos_message_queue.memos_message_queue
-        self.use_redis_queue = use_redis_queue
+        # Accept either a ScheduleTaskQueue wrapper or a concrete queue instance
+        self.memos_message_queue = (
+            memos_message_queue.memos_message_queue
+            if hasattr(memos_message_queue, "memos_message_queue")
+            else memos_message_queue
+        )
 
         # Get multi-task timeout from config
         self.multi_task_running_timeout = (

tests/mem_scheduler/test_dispatcher.py

@@ -157,7 +157,10 @@ def test_dispatch_serial(self):
         """Test dispatching messages in serial mode."""
         # Create a new dispatcher with parallel dispatch disabled
         serial_dispatcher = SchedulerDispatcher(
-            max_workers=2, enable_parallel_dispatch=False, metrics=MagicMock()
+            max_workers=2,
+            memos_message_queue=self.dispatcher.memos_message_queue,
+            enable_parallel_dispatch=False,
+            metrics=MagicMock(),
         )
 
         # Create fresh mock handlers for this test